from urllib.parse import urlparse, parse_qs
from datetime import datetime
from http.server import HTTPServer, BaseHTTPRequestHandler
import json
import psycopg2
import re

class TrackerHTTPRequestHandler(BaseHTTPRequestHandler):
    def __init__(self, *args):
        BaseHTTPRequestHandler.__init__(self, *args)

    def insert(self, datatype=None, key=None, value=None, created=None):
        if value:
            print("inserting", datatype, key, value)
            with conn.cursor() as cur:
                if key:
                    cur.execute("INSERT INTO datapoint (datatype, key, value, created) VALUES (%s, %s, %s, %s)", (datatype, key, value, created))
                else:
                    cur.execute("INSERT INTO datapoint (datatype, value, created) VALUES (%s, %s, %s)", (datatype, value, created))
            conn.commit()

    def insert_book_datapoint(self, title, pages, complete, created=None):
        with conn.cursor() as cur:
            cur.execute("SELECT id FROM book where title = %s", (title,))
            row = cur.fetchone()
            book_id = row[0]
            print(book_id, pages, created)
            cur.execute("INSERT INTO book_datapoint (book_id, pages, created) VALUES (%s, %s, %s)", (book_id, pages, created))
            if complete:
                cur.execute("UPDATE book SET completed = TRUE where id = %s", (book_id,))
        conn.commit()

    def do_GET(self):
        u = urlparse(self.path)
        print("GET", u.path)
        if u.path == "/":
            with open("index.html", "rb") as f:
                self.send_response(200)
                self.end_headers()
                self.wfile.write(f.read())
        elif date_pattern.match(u.path[1:]):
            with conn.cursor() as cur:
                cur.execute("SELECT datatype, key, value FROM datapoint where date(created) = %s", (u.path[1:],))
                items = []
                for datatype, key, value in cur:
                    items.append({
                        "datatype": datatype,
                        "key": key,
                        "value": value
                    })
                self.send_response(200)
                self.end_headers()
                self.wfile.write(json.dumps(items).encode("utf-8"))
        elif u.path == "/books":
            with conn.cursor() as cur:
                cur.execute("SELECT title FROM book WHERE completed = FALSE")
                items = []
                for row in cur:
                    items.append(row[0])
                self.send_response(200)
                self.end_headers()
                self.wfile.write(json.dumps(items).encode("utf-8"))
        elif u.path == "/forms":
            with conn.cursor() as cur:
                cur.execute("SELECT type, prompt, prompt_id, extra FROM form ORDER BY id")
                items = []
                for row in cur:
                    items.append({
                        "type": row[0],
                        "prompt": row[1],
                        "prompt_id": row[2],
                        "extra": row[3],
                    })
                self.send_response(200)
                self.end_headers()
                self.wfile.write(json.dumps(items).encode("utf-8"))
        else:
            self.send_response(404)
            self.end_headers()
            self.wfile.write(b'not found')

    def do_POST(self):
        u = urlparse(self.path)
        length = int(self.headers['Content-Length'])
        post_data = json.loads(self.rfile.read(length).decode('utf-8'))
        print("POST", u.path)
        if u.path == "/submit_payload":
            timestamp = datetime.fromtimestamp(int(post_data["timestamp"]/1000))
            del post_data["timestamp"]

            for book in post_data.get("books", []):
                try:
                    self.insert_book_datapoint(**book, created=timestamp)
                except Exception as e:
                    # Eventually add error messaging
                    print("ERROR", e)
                    pass
            post_data.pop("books", None)

            for k in post_data.keys():
                if isinstance(post_data[k], dict):
                    for key, value in post_data[k].items():
                        self.insert(datatype=k, key=key, value=value, created=timestamp)
                else:
                    self.insert(datatype=k, value=post_data[k], created=timestamp)
            self.send_response(204)
            self.end_headers()
            self.wfile.write(b"")
        else:
            self.send_response(404)
            self.end_headers()
            self.wfile.write(b'not found')


def setup_db():
    with conn.cursor() as cur:
        cur.execute("CREATE TABLE IF NOT EXISTS datapoint (id SERIAL PRIMARY KEY, created TIMESTAMP, datatype TEXT, key TEXT, value TEXT);")
        cur.execute("CREATE TABLE IF NOT EXISTS outside_weather (id SERIAL PRIMARY KEY, created TIMESTAMP, temp FLOAT8, humidity FLOAT8, pressure FLOAT8, uvi FLOAT8, dew_point FLOAT8, wind_speed FLOAT8, wind_guest FLOAT8, wind_deg FLOAT8);")
        cur.execute("CREATE TABLE IF NOT EXISTS book (id SERIAL PRIMARY KEY, title TEXT, completed BOOLEAN);")
        cur.execute("CREATE TABLE IF NOT EXISTS book_datapoint (id SERIAL PRIMARY KEY, created TIMESTAMP, book_id SERIAL, pages TEXT, CONSTRAINT fk_book FOREIGN KEY(book_id) REFERENCES book(id));")
        cur.execute("CREATE TABLE IF NOT EXISTS email (id SERIAL PRIMARY KEY, created TIMESTAMP, username TEXT, domain TEXT);")
        cur.execute("CREATE TABLE IF NOT EXISTS form (id SERIAL PRIMARY KEY, type TEXT, prompt TEXT, prompt_id TEXT, extra JSON);")
    conn.commit()

date_pattern = re.compile(r"(\d{4})-(\d{2})-(\d{2})")
conn = psycopg2.connect(host="db", dbname="tracking", user="tracking", password="password")
if __name__ == "__main__":
    setup_db()
    print("Starting http server")
    http = HTTPServer(("", 8000), TrackerHTTPRequestHandler)
    print("serving forever")
    http.serve_forever()