from http.server import HTTPServer, BaseHTTPRequestHandler
from urllib.parse import parse_qs
from json import dumps

from router import run
from utils import HTTP_STATUS_CODES


class MyHTTPRequestHandler(BaseHTTPRequestHandler):
    def _set_response(self, code, content_type):
        self.send_response(code)
        self.send_header('Content-type', content_type)
        self.end_headers()

    def do_GET(self):
        try:
            content_length = int(self.headers['Content-Length'])
            get_data = parse_qs(self.rfile.read(content_length).decode('utf-8'))
        except TypeError:
            get_data = {}

        for key in get_data:
            get_data[key] = get_data[key][0]

        self.finalize_request(run({
            'url': self.path,
            'method': 'GET',
            'query': get_data
        }))

    def do_POST(self):
        content_length = int(self.headers['Content-Length'])
        post_data = parse_qs(self.rfile.read(content_length).decode('utf-8'))

        for key in post_data:
            post_data[key] = post_data[key][0]

        self.finalize_request(run({
            'url': self.path,
            'method': 'POST',
            'query': post_data
        }))

    def finalize_request(self, response):
        if isinstance(response, int):
            self._set_response(response, 'text/html')
            response = f'<center><h1>ERROR {response} {HTTP_STATUS_CODES[response].upper()}</h1></center>'
        elif isinstance(response, (dict, list, tuple)):
            self._set_response(200, 'application/json')
            response = dumps(response)
        else:
            self._set_response(200, 'text/html')

        self.wfile.write(response.encode('utf-8'))


def start_server(host, port):
    server_address = (host, port)
    httpd = HTTPServer(server_address, MyHTTPRequestHandler)
    httpd.serve_forever()