import logging
from fastapi import APIRouter, HTTPException
from pydantic import BaseModel
from typing import Optional

# 引入 DB 函數
from database import (
    insert_config_backup,
    get_config_backup_list,
    get_config_backup_detail,
    delete_config_backup
)

# ✅ 修正：引入我們剛剛在 cmts_scraper.py 新增的函數
from cmts_scraper import fetch_raw_config, parse_config_to_tree

logger = logging.getLogger(__name__)

router = APIRouter(
    prefix="/backups",
    tags=["Backups"]
)

class SnapshotRequest(BaseModel):
    host: str
    username: str
    password: str
    snapshot_name: str
    config_type: str = "running"

@router.get("/", summary="取得歷史快照列表")
async def list_backups(host: str, config_type: str = "running"):
    records = await get_config_backup_list(host, config_type)
    if records is None:
        return {"status": "error", "message": "資料庫查詢失敗"}
    return {"status": "success", "data": records}

@router.get("/{backup_id}", summary="取得特定快照詳細內容")
async def get_backup_detail(backup_id: str):
    record = await get_config_backup_detail(backup_id)
    if not record:
        return {"status": "error", "message": "找不到該筆備份資料"}
    return {"status": "success", "data": record}

@router.delete("/{backup_id}", summary="刪除特定快照")
async def delete_backup(backup_id: str):
    success = await delete_config_backup(backup_id)
    if not success:
        return {"status": "error", "message": "刪除失敗或找不到該筆資料"}
    return {"status": "success", "message": "快照已成功刪除"}

@router.post("/snapshot", summary="手動建立設備快照")
async def create_snapshot(req: SnapshotRequest):
    try:
        # ✅ 修正：使用函數式呼叫來抓取與解析設定檔
        raw_cli = await fetch_raw_config(req.host, req.username, req.password, req.config_type)
        parsed_tree = parse_config_to_tree(raw_cli)

        # 寫入資料庫
        backup_id = await insert_config_backup(
            host=req.host,
            config_type=req.config_type,
            raw_cli=raw_cli,
            parsed_tree=parsed_tree,
            snapshot_name=req.snapshot_name,
            is_auto=False
        )

        if not backup_id:
            return {"status": "error", "message": "資料庫寫入失敗，請檢查系統日誌"}

        return {
            "status": "success", 
            "message": f"快照 '{req.snapshot_name}' 建立成功！",
            "backup_id": backup_id
        }

    except Exception as e:
        logger.error(f"❌ 建立快照失敗: {e}")
        return {"status": "error", "message": f"設備連線或解析失敗: {str(e)}"}