import sqlite3, time

class History():
    def __init__(self):
        self.__db =  sqlite3.connect('history.sqlite', check_same_thread=False)
        self.__c = self.__db.cursor()
        self.__c.execute("""
                    CREATE TABLE IF NOT EXISTS numbers (
                        id INTEGER PRIMARY KEY AUTOINCREMENT,
                        number TEXT UNIQUE NOT NULL
                    );
        """)
        self.__c.execute("""
                    CREATE TABLE IF NOT EXISTS call_history (
                        call_id INTEGER PRIMARY KEY AUTOINCREMENT,
                        number_id INTEGER NOT NULL,
                        direction INTEGER NOT NULL,
                        time INTEGER NOT NULL,
                        duration INTEGER,
                        accepted INTEGER,
                        FOREIGN KEY(number_id) REFERENCES numbers(id)
                    );
        """)
        self.__db.commit()

    def log_call(self, number, incoming):
        call_time = int(time.time())
        try:
            self.__c.execute("INSERT INTO numbers(number) VALUES(?)", (number,))
            number_id = self.__c.lastrowid
        except sqlite3.IntegrityError:
            number_id = self.__c.execute("SELECT id FROM numbers WHERE number=?", (number,)).fetchone()[0]

        self.__c.execute("INSERT INTO call_history(direction, time, number_id) VALUES(?, ?, ?);", (int(incoming), call_time, number_id))
        new_call_history_id = self.__c.lastrowid
        self.__db.commit()
        return new_call_history_id

    def accept_call(self, call_id):
        self.update_call_acception(call_id, True)

    def decline_call(self, call_id):
        self.update_call_acception(call_id, False)

    def update_call_acception(self, call_id, accepted):
        self.__c.execute("UPDATE call_history SET accepted=? WHERE call_id=?;", (accepted, call_id))
        self.__db.commit()

    def set_call_duration(self, call_id, duration):
        self.__c.execute("UPDATE call_history SET duration=? WHERE call_id=?;", (duration, call_id))
        self.__db.commit()