import mysql.connector
import yaml
import os

def initialize_database():
    # === Load config file ===
    with open("config.yml", "r") as file:
        config = yaml.safe_load(file)

    # === DB Connection ===
    conn = mysql.connector.connect(
        host=os.getenv("DB_HOST", "localhost"),
        port=os.getenv("DB_PORT", 3306),
        user=os.getenv("DB_USER", "emailuser"),
        password=os.getenv("DB_PASSWORD", "miguel33020"),
        database=os.getenv("DB_NAME", "emailassistant")
    )
    cursor = conn.cursor()

    # === Table: metadata (previously main_account) ===
    cursor.execute("""
        CREATE TABLE IF NOT EXISTS metadata (
            id INT AUTO_INCREMENT PRIMARY KEY,
            user VARCHAR(255),
            email VARCHAR(255) UNIQUE NOT NULL,
            token TEXT
        );
    """)
    print("✅ Table ready: metadata")

# === Table: emails ===
    cursor.execute("""
        CREATE TABLE IF NOT EXISTS emails (
            id INT AUTO_INCREMENT PRIMARY KEY,
            user VARCHAR(255),
            account VARCHAR(255),
            message_id VARCHAR(255) UNIQUE,
            thread_id VARCHAR(255),
            account_id VARCHAR(255),
            sender VARCHAR(255),
            cc TEXT,
            subject TEXT,
            body LONGTEXT,
            links LONGTEXT,
            unsubscribe_data TEXT,
            received_at DATETIME,
            folder VARCHAR(50),
            attachments LONGTEXT,
            is_read BOOLEAN DEFAULT FALSE,
            labels LONGTEXT,

            -- 🔍 AI-Generated Fields
            ai_category VARCHAR(100),              -- Top-level (e.g. 'promo')
            ai_confidence FLOAT,                   -- Confidence score
            ai_summary TEXT,                       -- Summary of subject/body
            ai_keywords TEXT,                      -- Comma-separated extracted keywords
            ai_label_source VARCHAR(100),          -- 'subject', 'body', 'combined', 'llm'
            summary_source VARCHAR(100),           -- Similar to above
            ai_model_version VARCHAR(100),         -- Versioning helps long-term debugging
            is_ai_reviewed BOOLEAN DEFAULT FALSE,  -- Was this fully processed by AI?
            processing_notes TEXT,                 -- Optional notes about fallback, etc.

            -- 🔄 Sync and Processing Status
            processing_status VARCHAR(50),
            sync_status VARCHAR(50),
            attachment_path TEXT,
            downloaded BOOLEAN DEFAULT FALSE
        );
    """)
    print("✅ Table ready: emails")


    # === Table: logs ===
    cursor.execute("""
        CREATE TABLE IF NOT EXISTS logs (
            id INT AUTO_INCREMENT PRIMARY KEY,
            timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
            level VARCHAR(20),
            source VARCHAR(255),
            message TEXT
        );
    """)
    print("✅ Table ready: logs")

    cursor.close()
    conn.close()

# if __name__ == "__main__":
#     initialize_database()


#if __name__ == "__main__":
#    initialize_database()