import os
from pathlib import Path
import sqlite3

tam_data_dir = os.getenv("TAM_DATA_DIR", "./data")

def session():
    data_dir = Path(tam_data_dir)
    if not data_dir.is_dir():
        data_dir.mkdir(parents=True)
    db_file = data_dir / "tam-remote.db"
    conn = sqlite3.connect(db_file)
    cur = conn.cursor()
    return conn, cur

def init_db():
    conn, cur = session()
    cur.execute("CREATE TABLE IF NOT EXISTS auth_keys (auth_key TEXT PRIMARY KEY, description TEXT)")
    cur.execute("CREATE TABLE IF NOT EXISTS prefixes (prefix TEXT PRIMARY KEY, color TEXT, weight INTEGER)")
    cur.execute("""CREATE TABLE IF NOT EXISTS tickets (prefix TEXT, t_id INTEGER, first_name TEXT, last_name TEXT,
        phone_number TEXT, pref TEXT, PRIMARY KEY (prefix, t_id))""")
    cur.execute("""CREATE TABLE IF NOT EXISTS baskets (prefix TEXT, b_id INTEGER, description TEXT, donors TEXT,
        winning_ticket INTEGER, PRIMARY KEY (prefix, b_id))""")
    cur.execute("""CREATE VIEW IF NOT EXISTS drawing AS
        SELECT b.prefix, b.b_id, b.description, b.winning_ticket, t.last_name, t.first_name, t.phone_number
        FROM baskets b LEFT JOIN tickets t ON b.prefix = t.prefix AND b.winning_ticket = t.t_id
        ORDER BY b.prefix, b.b_id""")
    cur.execute("""CREATE VIEW IF NOT EXISTS report_by_name AS
        SELECT t.last_name, t.first_name, t.phone_number, t.pref, b.*
        FROM baskets b LEFT JOIN tickets t ON b.prefix = t.prefix AND b.winning_ticket = t.t_id
        ORDER BY t.last_name, t.first_name, t.phone_number, b.prefix, b.b_id""")
    cur.execute("""CREATE VIEW IF NOT EXISTS report_by_basket AS
        SELECT b.*, t.last_name, t.first_name, t.phone_number, t.pref
        FROM baskets b LEFT JOIN tickets t ON b.prefix = t.prefix AND b.winning_ticket = t.t_id
        ORDER BY b.prefix, b.b_id""")
    cur.execute("""CREATE VIEW IF NOT EXISTS report_counts AS
        SELECT prefix, COUNT(DISTINCT(CONCAT(first_name, last_name, phone_number))) AS unique_buyers, COUNT(*) AS total_buys
        FROM tickets
        GROUP BY prefix
        UNION ALL
        SELECT 'Totals', COUNT(DISTINCT(CONCAT(first_name, last_name, phone_number))), COUNT(*)
        FROM tickets""")
    conn.commit()
    conn.close()

class RepoTemplate:
    def __init__(self):
        self.conn, self.cur = session()
    def __del__(self):
        self.conn.close()