import sqlite3
from typing import List, Tuple


class CounterBot:
    def __init__(self, db_name: str) -> None:
        self.con = sqlite3.connect(db_name)
        self.con.row_factory = sqlite3.Row
        self.try_create_tables()

    def try_create_tables(self) -> bool:
        try:
            with self.con:
                self.con.execute(
                    "CREATE TABLE presence (ID INTEGER PRIMARY KEY AUTOINCREMENT, NAME VARCHAR UNIQUE, IS_PRESENT BOOL)"
                )
                print("Creating new table")
        except sqlite3.OperationalError:
            print("Table already exists in database, nothing to create.")
        return True

    def update_presence(self, name: str, is_present: bool) -> bool:
        present_int = 1 if is_present else 0
        with self.con:
            self.con.execute(
                f"INSERT INTO presence (NAME, IS_PRESENT) VALUES ('{name}', '{present_int}') ON CONFLICT(name) DO UPDATE SET is_present=excluded.is_present"
            )
            return True
        return False

    def get_full_entries(self) -> List[Tuple]:
        with self.con:
            return [
                (x["name"], x["is_present"])
                for x in self.con.execute(
                    "SELECT name, is_present FROM presence ORDER BY name"
                )
            ]

    def get_present_names(self) -> List[str]:
        with self.con:
            return [
                x["name"]
                for x in self.con.execute(
                    "SELECT name FROM presence WHERE is_present ORDER BY name"
                )
            ]

    def get_present_count(self) -> int:
        with self.con:
            return self.con.execute(
                "SELECT COUNT(name) FROM presence WHERE is_present"
            ).fetchone()[0]

    def clear_rows(self) -> None:
        with self.con:
            self.con.execute("DELETE FROM presence")


def main() -> None:
    counter_bot = CounterBot("counter.db")


if __name__ == "__main__":
    main()