import sqlite3
from config import DB_PATH 


def init_db(db_path=DB_PATH):
    conn = sqlite3.connect(db_path)
    c = conn.cursor()
    c.execute('''
        CREATE TABLE IF NOT EXISTS speed_tests (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            timestamp REAL NOT NULL,
            failed BOOLEAN NOT NULL,
            isp TEXT,
            ip TEXT,
            location_code TEXT,
            location_city TEXT,
            location_region TEXT,
            latency REAL,
            jitter REAL,
            down_100kB REAL,
            down_1MB REAL,
            down_10MB REAL,
            down_25MB REAL,
            down_90th REAL,
            up_100kB REAL,
            up_1MB REAL,
            up_10MB REAL,
            up_90th REAL
            )
    ''')
    conn.commit()
    conn.close()


def insert_result(results: dict|None, db_path=DB_PATH):
    conn = sqlite3.connect(db_path)
    c = conn.cursor()

    # If the test failed entirely, store it as a failure with timestamp now
    if results is None or "tests" not in results:
        from time import time
        c.execute("INSERT INTO speed_tests (timestamp, failed) VALUES (?, ?)", (time(), True))
        conn.commit()
        conn.close()
        return

    tests = results.get("tests", {})
    meta = results.get("meta", {})

    # Get a consistent timestamp from any TestResult (or fallback to now)
    from time import time as now
    sample_test = next(iter(tests.values()), None)
    timestamp = sample_test.time if sample_test else now()

    print(tests)
    print(meta)

    def get(tests, key):
        return tests[key].value if key in tests else None

    c.execute('''
        INSERT INTO speed_tests (
            timestamp, failed, isp, ip, location_code, location_city, location_region,
            latency, jitter,
            down_100kB, down_1MB, down_10MB, down_25MB, down_90th,
            up_100kB, up_1MB, up_10MB, up_90th
        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
    ''', (
        timestamp, False,
        get(tests, "isp"),
        get(meta, "ip"),
        get(meta, "location_code"),
        get(meta, "location_city"),
        get(meta, "location_region"),
        get(tests, "latency"),
        get(tests, "jitter"),
        get(tests, "100kB_down_mbps"),
        get(tests, "1MB_down_mbps"),
        get(tests, "10MB_down_mbps"),
        get(tests, "25MB_down_mbps"),
        get(tests, "90th_percentile_down_mbps"),
        get(tests, "100kB_up_mbps"),
        get(tests, "1MB_up_mbps"),
        get(tests, "10MB_up_mbps"),
        get(tests, "90th_percentile_up_mbps")
    ))
    conn.commit()
    conn.close()