import json
from iexceptions import RunCommandException, NotCaliDevException, \
    MountDirNotFoundException, InspectDockerError, GetDockerNetDevIDError, \
    ExecBashError
from libs.misc import system


class Docker(object):
    def __init__(self):
        self._ip_cmd = "ip"
        self._docker = "docker"
        self._cgroup_blkio_dir = "/host/sys/fs/cgroup/blkio"
        self._cgroup_cpu_dir = "/host/sys/fs/cgroup/cpu"
        self._cgroup_mem_dir = "/host/sys/fs/cgroup/memory"

    def get_netdev_for_ip(self, address):
        if not self._valid_ip(address):
            raise InvalidateIpException(address=address)
        cmd = "{ip_cmd} route get {address}".format(
            ip_cmd=self._ip_cmd, address=address)
        rc, so, se = system(cmd)
        if rc:
            raise RunCommandException(cmd=cmd, rc=rc, so=so, se=se)
        netdev = so.strip().split()[2]
        if not netdev.startswith("cali"):
            raise NotCaliDevException(dev=netdev)
        return netdev

    def get_netdev_for_container_id(self, container_id):
        dev_id = self.get_netdev_id(container_id)
        cmd = "ip addr"
        rc, so, se = system(cmd)
        if rc: 
            raise ExecBashError(cmd=cmd, msg=so)

        for line in so.split("\n"):
            line = line.strip()
            if not line:
                continue
            ls = line.split(":")
            if len(ls) < 1: 
                continue
            if ls[0].strip() == dev_id.strip():
                return ls[1].strip().split("@")[0]
        return None

    def get_netdev_id(self, container_id):
        cmd = "{docker} exec {container_id} ip addr | grep 'eth0@if'".format(
            docker=self._docker, container_id=container_id)
        rc, so, se = system(cmd)
        if rc: 
            raise GetDockerNetDevIDError(docker=container_id, msg=so)
        
        ls = so.strip().split(":")
        if len(ls) < 2: 
            raise GetDockerNetDevIDError(docker=container_id, msg=so)
        
        ls_dev = ls[1].strip().split("@if")
        if len(ls_dev) < 2: 
            raise GetDockerNetDevIDError(docker=container_id, msg=so)
        return ls_dev[1].strip()

    def is_exist(self, container_id):
        cmd = "{docker} inspect {container_id}".format(
            docker=self._docker, container_id=container_id)
        rc, so, se = system(cmd)
        if rc:
            return False
        return True

    @staticmethod
    def _valid_ip(ip):
        return True

    def get_cgroup_path(self, base_dir, container_id):
        cmd = "{docker} inspect {container_id}".format(
            docker=self._docker, container_id=container_id)
        rc, so, se = system(cmd)
        if rc:
            raise InspectDockerError(docker=container_id, msg=so)
        so = json.loads(so)[0]["HostConfig"]["CgroupParent"]
        dirname = base_dir + so + "/" + container_id
        return dirname

    def get_cpu_cgroup_path(self, container_id):
        return self.get_cgroup_path(self._cgroup_cpu_dir, container_id)

    def get_mem_cgroup_path(self, container_id):
        return self.get_cgroup_path(self._cgroup_mem_dir, container_id)

    def get_blkio_cgroup_path(self, container_id):
        return self.get_cgroup_path(self._cgroup_blkio_dir, container_id)

    def get_mount_dir(self, container_id, dirname):
        # "Mounts"
        cmd = "docker inspect {container_id}".format(container_id=container_id)
        rc, so, se = system(cmd)
        mounts = json.loads(so)[0]["Mounts"]
        for mount in mounts:
            if mount["Destination"] == dirname:
                return mount["Source"]
            if mount["Destination"] == "/".join(dirname.split("/")[:-1]):
                return mount["Source"]
        raise MountDirNotFoundException(dirname=dirname)


docker = Docker()