__author__ = 'vanzhiganov'

"""
import dnsmasq
dnsmasq.Dnsmasq().exists_name('test')
dnsmasq.Dnsmasq().add('test', '10.10.10.10')
dnsmasq.Dnsmasq().exists_name('test')

"""


import shutil
import subprocess
import os.path


class Error(Exception):
    pass


class NameAlreadyExists(Exception):
    pass


class NameNotExists(Exception):
    pass


class IPAlreadyExists(Exception):
    pass


class IPNotExists(Exception):
    pass


class Dnsmasq:
    def __init__(self, dnsmasq_conf='/etc/lxc/dnsmasq.conf'):
        """

        :param dnsmasq_conf:
        :return:
        """
        if not self.__exists_conf(dnsmasq_conf):
            raise Error("Config not exists")

        self.dnsmasq_conf = dnsmasq_conf
        self.list = self.__read_conf()

    def exists_name(self, name):
        """

        :param name:
        :return:
        """
        # origin = self.__read_conf()
        for i in self.list:
            if name in i:
                return True
        return False

    # @staticmethod
    def exists_ip(self, ip):
        """

        :param ip:
        :return:
        """
        # origin = self.__read_conf()
        for i in self.list:
            if ip in i:
                return True
        return False

    def add(self, name, ip):
        """

        :param name:
        :param ip:
        :return:
        """
        if not self.__exists_conf(self.dnsmasq_conf):
            return False

        if self.exists_name(name):
            raise NameAlreadyExists("Already exists: %s" % name)
        if self.exists_ip(ip):
            raise IPAlreadyExists("Already exists: %s" % ip)
        self.list.append("dhcp-host=%s,%s" % (name, ip))
        result_write = self.__write_conf()
        if not result_write:
            return False
        return True

    def delete(self, name):
        if not self.__exists_conf(self.dnsmasq_conf):
            return False

        clean = []
        for i in self.__read_conf():
            if name not in i:
                clean.append(i)
        result_write = self.__write_conf()
        if not result_write:
            return False
        return True

    def __read_conf(self):
        """

        :return:
        """
        origin = []
        for i in open(self.dnsmasq_conf, "r"):
            origin.append(i.rstrip())
        return origin

    def __write_conf(self):
        """

        :return:
        """
        if not self.__exists_conf(self.dnsmasq_conf):
            return False

        f = open("/tmp/dnsmasq.conf", 'w')
        # f.writelines(self.list)
        for i in self.list:
            f.write("%s\n" % i)
        f.close()
        shutil.copyfile("/tmp/dnsmasq.conf", "/etc/lxc/dnsmasq.conf")
        return True

    def __exists_conf(self, conf):
        if os.path.isfile(conf):
            return True
        return False


class Service:
    def restart(self):
        subprocess.call("service dnsmasq restart", shell=True)
        return True

    def pid(self):
        return None