import asyncio
import asyncpg
import logging

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# DB 設定
DB_CONFIG = {
    "database": "cmts_nms",
    "user": "swpa",
    "password": "swpa4920",
    "host": "127.0.0.1",
    "port": "5432"
}

async def init_database():
    try:
        logger.info("🔄 正在連線到 PostgreSQL (asyncpg)...")
        conn = await asyncpg.connect(
            database=DB_CONFIG["database"],
            user=DB_CONFIG["user"],
            password=DB_CONFIG["password"],
            host=DB_CONFIG["host"],
            port=int(DB_CONFIG["port"])
        )

        # 1. 建立選項快取表 cmts_options
        logger.info("🛠️ 正在建立 cmts_options 資料表...")
        await conn.execute("""
            CREATE TABLE IF NOT EXISTS cmts_options (
                host VARCHAR(255) NOT NULL,
                config_type VARCHAR(50) NOT NULL,
                path VARCHAR(500) NOT NULL,
                data JSONB NOT NULL,
                updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
                PRIMARY KEY (host, config_type, path)
            );
        """)

        # 2. 建立設備狀態表 device_status
        logger.info("🛠️ 正在建立 device_status 資料表...")
        await conn.execute("""
            CREATE TABLE IF NOT EXISTS device_status (
                host VARCHAR(255) NOT NULL,
                config_type VARCHAR(50) NOT NULL,
                cmts_version VARCHAR(100) DEFAULT 'unknown',
                last_scanned VARCHAR(100),
                updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
                PRIMARY KEY (host, config_type)
            );
        """)

        # 3. 建立過濾器設定表 system_filters
        logger.info("🛠️ 正在建立 system_filters 資料表...")
        await conn.execute("""
            CREATE TABLE IF NOT EXISTS system_filters (
                config_type VARCHAR(50) PRIMARY KEY,
                hidden_keys TEXT[] DEFAULT '{}',
                updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
            );
        """)

        # 4. 建立設備配置備份表 config_backups (Phase 1 新增)
        logger.info("🛠️ 正在建立 config_backups 資料表與索引...")
        await conn.execute("""
            CREATE TABLE IF NOT EXISTS config_backups (
                id UUID PRIMARY KEY,
                host VARCHAR(255) NOT NULL,
                config_type VARCHAR(50) NOT NULL DEFAULT 'running',
                timestamp TIMESTAMPTZ NOT NULL DEFAULT CURRENT_TIMESTAMP,
                snapshot_name VARCHAR(255),
                is_auto BOOLEAN NOT NULL DEFAULT FALSE,
                raw_cli TEXT,
                parsed_tree JSONB,
                CONSTRAINT uq_snapshot_name UNIQUE (host, config_type, snapshot_name)
            );
        """)
        
        # 建立複合索引以加速列表查詢與排序
        await conn.execute("""
            CREATE INDEX IF NOT EXISTS idx_config_backups_host_type_ts 
            ON config_backups (host, config_type, timestamp DESC);
        """)

        await conn.close()
        logger.info("✅ 資料庫初始化完成！所有資料表已準備就緒。")

    except Exception as e:
        logger.error(f"❌ 資料庫初始化失敗: {e}")

if __name__ == "__main__":
    asyncio.run(init_database())