import socketserver
import ipaddr
import re

from lib.config import Config

class Daemon():
    def __init__(self, data):
        self.data = data
        self.domain_regexp = re.compile("(?!-)[A-Z\d-]{1,63}(?<!-)$", re.IGNORECASE)
        self.footer = None
        self.header = None

    def query(self, q):
        if self.is_ip(q):
            result = self.search_ip(q)
        elif self.is_domain(q):
            result = self.search_domain(q)            
        else:
            # Try to find a person with the given query
            person = self.search_person(q)
            if person:
                result = person
            else:
                result = self.get_help()

        return result.formatted_answer()

#    def format_result(self, values):
#        """Receive a dictionary and return a string"""
#        result = ""
#        for k, v in values.items():
#            if k == 'domains':
#                for i, dom in enumerate(v):
#                    if len(v) == 1:
#                        result += 'domain: {0}\n'.format(dom.name)
#                    else:
#                        result += 'domain-{0}: {1}\n'.format(i, dom.name)
#            elif k == 'admins':
#                for i, adm in enumerate(v):
#                    if len(v) == 1:
#                        result += 'Admin: {0} {1} <{2}>\n'.format(adm.name, adm.surname, adm.email)
#                    else:
#                        result += 'Admin-{0}: {1} {2} <{3}>\n'.format(i, adm.name, adm.surname, adm.email)
#            else:
#                result += '{0}: {1}\n'.format(k, v)
        
#        return result

    def search_ip(self, ip):
        # Iterate over all IP block elements
        networks = self.data.get_networks()
        for key in networks:
           for block in networks[key].ip_blocks:
               if ipaddr.IPAddress(ip) in ipaddr.IPNetwork(block):
                   return networks[key].as_answer(self.data)
                   
        return Answer(AnswerValue(1, 'Error', 'Network not found'))

    def search_domain(self, domain):
        # Iterate over all network and check its domains
        networks = self.data.get_networks()
        domains = self.data.get_domains()
        for network in networks.values():
            if any(domains[d].name == domain for d in network.domains):
                return network.as_answer(self.data)
                
        return Answer(AnswerValue(1, 'Error', 'Domain not found'))
                
    # TODO

    def search_person(self, query):
        pass

    def is_ip(self, query):
        try:
            ipaddr.IPAddress(query)
            return True
        except ValueError:
            return False

    def is_domain(self, hostname):
        if len(hostname) > 255:
            return False
        
        if hostname[-1:] == ".":
            hostname = hostname[:-1] # strip exactly one dot from the right, if present
            
        return all(self.domain_regexp.match(x) for x in hostname.split("."))

    # TODO
    def get_help(self):
        return "This will be the help"

    def get_footer(self):
        if not self.footer:
            f = open(Config().parser['Printing']['footer'])
            self.footer = f.read()
            f.close()

        return self.footer

    def get_header(self):
        if not self.header:
            f = open(Config().parser['Printing']['header'])
            self.header = f.read()
            f.close()

        return self.header

class Answer():
    def __init__(self):
        self.values = []

    def add_value(self, av):
        self.values.append(av)

    def formatted_answer(self):
        res = ""
        for av in sorted(self.values, key=lambda v: v.priority):
            res += av.formatted_value() + "\n"

        return res

class AnswerValue():
    def __init__(self, priority, key, val):
        self.priority = priority
        self.key = key
        self.val = val

    def value(self):
        return self.val

    def formatted_value(self):
        return "{0}: {1}".format(self.key, self.value())

class AnswerAdminValue(AnswerValue):
    def __init__(self, priority, admin):
        self.priority = priority
        self.key = 'Admin'
        self.adm = admin

    def value(self):
        return '{0} {1} <{2}>\n'.format(self.adm.name, self.adm.surname, self.adm.email)
        

class WhoisHandler(socketserver.BaseRequestHandler):

    def setup(self):
        self.daemon = self.server.daemon

    def handle(self):
        data = str(self.request.recv(100).strip(), 'utf-8')
        print('Received: {}'.format(data))

        response = self.daemon.get_header()
        response += self.daemon.query(data)
        response += self.daemon.get_footer()
        
        self.request.sendall(bytes(response, 'utf-8'))
    
class ClassicServer(socketserver.ThreadingTCPServer):    
    def __init__(self, daemon):
        host = Config().parser['Servers']['classic_host']
        port = int(Config().parser['Servers']['classic_port'])
        self.daemon = daemon

        socketserver.ThreadingTCPServer.__init__(self, (host, port), WhoisHandler)

class WebServer(socketserver.ThreadingTCPServer):
    def __init__(self, daemon):
        self.host = Config().parser['Servers']['web_host']
        self.port = int(Config().parser['Servers']['web_port'])
        self.daemon = daemon

        socketserver.ThreadingTCPServer.__init__(self, (self.host, self.port), WhoisHandler)