import sqlite3

DB_PATH = "/home/matrixhasyou/legal-swarm-core/services/domovoy_bot/database/domovoy.db"

def migrate():
    print(f"[*] Adding has_ai_access column to users...")
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()
    try:
        cursor.execute("ALTER TABLE users ADD COLUMN has_ai_access BOOLEAN DEFAULT FALSE;")
        print("[+] Column has_ai_access added.")
        # Даем доступ админу (тебе) сразу
        cursor.execute("UPDATE users SET has_ai_access = TRUE WHERE is_admin = TRUE;")
        print("[+] AI access granted to all admins.")
    except Exception as e:
        print(f"[!] Error or already exists: {e}")
    conn.commit()
    conn.close()

if __name__ == "__main__":
    migrate()