"""任务记录工具类。"""
from __future__ import annotations

import json
from datetime import datetime
from typing import Any, Dict, Optional

from app.utils.db import db_session


class JobLogger:
    """任务记录器。"""
    
    def __init__(self, job_type: str) -> None:
        """初始化任务记录器。
        
        Args:
            job_type: 任务类型
        """
        self.job_type = job_type
        self.job_id: Optional[int] = None
        
    def __enter__(self) -> "JobLogger":
        """开始记录任务。"""
        with db_session() as session:
            cursor = session.execute(
                """
                INSERT INTO fetch_jobs (job_type, status, created_at, updated_at)
                VALUES (?, 'running', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)
                """,
                (self.job_type,)
            )
            self.job_id = cursor.lastrowid
            session.commit()
        return self
        
    def __exit__(self, exc_type: Any, exc_val: Any, exc_tb: Any) -> None:
        """结束任务记录。"""
        if exc_val:
            self.update_status("failed", str(exc_val))
        else:
            self.update_status("success")
            
    def update_status(self, status: str, error_msg: Optional[str] = None) -> None:
        """更新任务状态。
        
        Args:
            status: 新状态
            error_msg: 错误信息（如果有）
        """
        if not self.job_id:
            return
            
        with db_session() as session:
            session.execute(
                """
                UPDATE fetch_jobs
                SET status = ?,
                    error_msg = ?,
                    updated_at = CURRENT_TIMESTAMP
                WHERE id = ?
                """,
                (status, error_msg, self.job_id)
            )
            session.commit()
            
    def update_metadata(self, metadata: Dict[str, Any]) -> None:
        """更新任务元数据。
        
        Args:
            metadata: 元数据字典
        """
        if not self.job_id:
            return
            
        with db_session() as session:
            session.execute(
                """
                UPDATE fetch_jobs
                SET metadata = ?
                WHERE id = ?
                """,
                (json.dumps(metadata), self.job_id)
            )
            session.commit()