import sqlite3
from pathlib import Path

CACHE_DB_PATH = Path("agentm/data/agentM.db")
CACHE_DB_PATH.parent.mkdir(parents=True, exist_ok=True)

def get_db_conn():
    return sqlite3.connect(CACHE_DB_PATH)

def initialize_database():
    with get_db_conn() as conn:
        conn.execute("""
            CREATE TABLE IF NOT EXISTS roms (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                title TEXT NOT NULL,
                rom_file TEXT NOT NULL UNIQUE,
                game_id TEXT NOT NULL,
                sha256 TEXT,
                difficulty_min INTEGER,
                difficulty_max INTEGER,
                characters TEXT,
                keywords TEXT,
                verified BOOLEAN NOT NULL DEFAULT 0,
                verified_at TEXT
            );
        """)
        conn.commit()