"""查询整理历史记录工具"""

import json
from typing import Optional, Type

import jieba
from pydantic import BaseModel, Field

from app.agent.tools.base import MoviePilotTool
from app.db import AsyncSessionFactory
from app.db.models.transferhistory import TransferHistory
from app.log import logger
from app.schemas.types import media_type_to_agent


class QueryTransferHistoryInput(BaseModel):
    """查询整理历史记录工具的输入参数模型"""
    explanation: str = Field(..., description="Clear explanation of why this tool is being used in the current context")
    title: Optional[str] = Field(None, description="Search by title (optional, supports partial match)")
    status: Optional[str] = Field("all",
                                  description="Filter by status: 'success' for successful transfers, 'failed' for failed transfers, 'all' for all records (default: 'all')")
    page: Optional[int] = Field(1, description="Page number for pagination (default: 1, each page contains 30 records)")


class QueryTransferHistoryTool(MoviePilotTool):
    name: str = "query_transfer_history"
    description: str = "Query file transfer history records. Shows transfer status, source and destination paths, media information, and transfer details. Supports filtering by title and status."
    args_schema: Type[BaseModel] = QueryTransferHistoryInput

    def get_tool_message(self, **kwargs) -> Optional[str]:
        """根据查询参数生成友好的提示消息"""
        title = kwargs.get("title")
        status = kwargs.get("status", "all")
        page = kwargs.get("page", 1)

        parts = ["正在查询整理历史"]

        if title:
            parts.append(f"标题: {title}")
        if status != "all":
            status_map = {"success": "成功", "failed": "失败"}
            parts.append(f"状态: {status_map.get(status, status)}")
        if page > 1:
            parts.append(f"第{page}页")

        return " | ".join(parts) if len(parts) > 1 else parts[0]

    async def run(self, title: Optional[str] = None,
                  status: Optional[str] = "all",
                  page: Optional[int] = 1, **kwargs) -> str:
        logger.info(f"执行工具: {self.name}, 参数: title={title}, status={status}, page={page}")

        try:
            # 处理状态参数
            status_bool = None
            if status == "success":
                status_bool = True
            elif status == "failed":
                status_bool = False

            # 处理页码参数
            if page is None or page < 1:
                page = 1

            # 每页记录数
            count = 50

            # 获取数据库会话
            async with AsyncSessionFactory() as db:
                # 处理标题搜索
                if title:
                    # 使用 jieba 分词处理标题
                    words = jieba.cut(title, HMM=False)
                    title_search = "%".join(words)
                    # 查询记录
                    result = await TransferHistory.async_list_by_title(
                        db, title=title_search, page=page, count=count, status=status_bool
                    )
                    total = await TransferHistory.async_count_by_title(
                        db, title=title_search, status=status_bool
                    )
                else:
                    # 查询所有记录
                    result = await TransferHistory.async_list_by_page(
                        db, page=page, count=count, status=status_bool
                    )
                    total = await TransferHistory.async_count(db, status=status_bool)

                if not result:
                    return "未找到相关整理历史记录"

                # 转换为字典格式，只保留关键信息
                simplified_records = []
                for record in result:
                    simplified = {
                        "id": record.id,
                        "title": record.title,
                        "year": record.year,
                        "type": media_type_to_agent(record.type),
                        "category": record.category,
                        "seasons": record.seasons,
                        "episodes": record.episodes,
                        "src": record.src,
                        "dest": record.dest,
                        "mode": record.mode,
                        "status": "成功" if record.status else "失败",
                        "date": record.date,
                        "downloader": record.downloader,
                        "download_hash": record.download_hash
                    }
                    # 如果失败，添加错误信息
                    if not record.status and record.errmsg:
                        simplified["errmsg"] = record.errmsg
                    # 添加媒体ID信息（如果有）
                    if record.tmdbid:
                        simplified["tmdbid"] = record.tmdbid
                    if record.imdbid:
                        simplified["imdbid"] = record.imdbid
                    if record.doubanid:
                        simplified["doubanid"] = record.doubanid
                    simplified_records.append(simplified)

                result_json = json.dumps(simplified_records, ensure_ascii=False, indent=2)

                # 计算总页数
                total_pages = (total + count - 1) // count if total > 0 else 1

                # 构建分页信息
                pagination_info = f"第 {page}/{total_pages} 页，共 {total} 条记录（每页 {count} 条）"

                return f"{pagination_info}\n\n{result_json}"
        except Exception as e:
            logger.error(f"查询整理历史记录失败: {e}", exc_info=True)
            return f"查询整理历史记录时发生错误: {str(e)}"