"""删除订阅工具"""

from typing import Optional, Type

from pydantic import BaseModel, Field

from app.agent.tools.base import MoviePilotTool
from app.agent.tools.tags import ToolTag
from app.runtime.events import eventmanager
from app.application.agentdata import SubscribePort as SubscribeOper
from app.adapters.external.server import MoviePilotServerHelper
from app.runtime.log import logger
from app.schemas.types import EventType


class DeleteSubscribeInput(BaseModel):
    """删除订阅工具的输入参数模型"""

    subscribe_id: int = Field(
        ...,
        description="The ID of the subscription to delete (can be obtained from query_subscribes tool)",
    )


class DeleteSubscribeTool(MoviePilotTool):
    """按订阅 ID 删除影视、单曲或专辑订阅。"""

    name: str = "delete_subscribe"
    tags: list[str] = [
        ToolTag.Write,
        ToolTag.Subscription,
        ToolTag.Admin,
    ]
    description: str = "Delete a media subscription by its ID. This will remove the subscription and stop automatic downloads for that media."
    args_schema: Type[BaseModel] = DeleteSubscribeInput
    require_admin: bool = True

    def get_tool_message(self, **kwargs) -> Optional[str]:
        """根据删除参数生成友好的提示消息"""
        subscribe_id = kwargs.get("subscribe_id")
        return f"删除订阅 (ID: {subscribe_id})"

    async def run(self, subscribe_id: int, **kwargs) -> str:
        """删除订阅并同步刷新带音乐实体维度的共享统计。"""
        logger.info(f"执行工具: {self.name}, 参数: subscribe_id={subscribe_id}")

        try:
            subscribe_oper = SubscribeOper()
            # 获取订阅信息
            subscribe = await subscribe_oper.async_get(subscribe_id)
            if not subscribe:
                return f"订阅 ID {subscribe_id} 不存在"

            # 在删除之前获取订阅信息（用于事件）
            subscribe_info = subscribe.to_dict()

            await subscribe_oper.async_delete(subscribe_id)
            # 分享订阅统计刷新本身已异步化，这里只需要在删除后触发即可。
            MoviePilotServerHelper.sub_done_async(
                {
                    "media_source": subscribe.media_source,
                    "media_id": subscribe.media_id,
                    "music_type": subscribe.music_type,
                    "total_tracks": subscribe.total_tracks,
                    "season": subscribe.season,
                }
            )

            # 发送事件
            await eventmanager.async_send_event(
                EventType.SubscribeDeleted,
                {"subscribe_id": subscribe_id, "subscribe_info": subscribe_info},
            )

            return f"成功删除订阅：{subscribe.name} ({subscribe.year})"
        except Exception as e:
            logger.error(f"删除订阅失败: {e}", exc_info=True)
            return f"删除订阅时发生错误: {str(e)}"