<reponame>nkk0/ec2-elastic-ip-manager
import logging
import os
from typing import List, Optional

import boto3

log = logging.getLogger()
log.setLevel(os.environ.get("LOG_LEVEL", "INFO"))
ec2 = boto3.client("ec2")


class EIP(dict):
    def __init__(self, instance: dict):
        self.update(instance)

    @property
    def allocation_id(self) -> str:
        return self.get("AllocationId")

    @property
    def association_id(self) -> Optional[str]:
        return self.get("AssociationId")

    @property
    def is_associated(self) -> bool:
        return self.association_id is not None

    @property
    def instance_id(self) -> Optional[str]:
        return self.get("InstanceId")

    @property
    def pool_name(self) -> Optional[str]:
        return self.tags.get("elastic-ip-manager-pool")

    @property
    def tags(self) -> dict:
        return {t["Key"]: t["Value"] for t in self["Tags"]}

    def __key(self):
        return self["AllocationId"]

    def __hash__(self):
        return hash(self.__key())

    def __eq__(self, other):
        return self.__key() == other.__key()

    def __str__(self):
        return str(self.__key())


def get_pool_addresses(pool_name: str) -> List[EIP]:
    response = ec2.describe_addresses(
        Filters=[
            {"Name": "domain", "Values": ["vpc"]},
            {"Name": "tag:elastic-ip-manager-pool", "Values": [pool_name]},
        ]
    )
    return [EIP(a) for a in response["Addresses"]]
