import docker
import os


class DockerHandler:
    def __init__(self, base_url, port_ip_map):
        self.base_url = base_url
        self.client = docker.DockerClient(base_url=base_url)
        self.port_ip_map = port_ip_map
        self.current_network = self.get_current_network()

    def get_current_container_ip(self):
        # Get IP of current container
        current_container_name = os.environ.get('HOSTNAME')
        current_container = self.client.containers.get(current_container_name)
        networks = current_container.attrs['NetworkSettings']['Networks']
        current_network = list(networks.keys())[0]
        current_container_ip = networks[current_network]['IPAddress']
        return current_container_ip

    def get_current_network(self):
        # Get network of current container
        current_container_name = os.environ.get('HOSTNAME')
        current_container = self.client.containers.get(current_container_name)
        networks = current_container.attrs['NetworkSettings']['Networks']
        current_network = list(networks.keys())[0]
        return current_network

    def get_container_by_ip(self, ip):
        print('getting docker container by ip {} in network {}'.format(
            ip, self.current_network))
        containers = self.client.containers.list(all=True)

        for container in containers:
            networks = container.attrs['NetworkSettings']['Networks']
            if self.current_network in networks and networks[self.current_network]['IPAMConfig']['IPv4Address'] == ip:
                print('found docker container {} with ip {} in network {}'.format(
                    container.name, ip, self.current_network))
                return container
        print('no docker container found with ip {} in network {}'.format(
            ip, self.current_network))
        return None