"""Trend 业务工具 - 将趋势分析服务注册为 FunctionTool"""

import logging
from typing import Any

from agentkit.tools.function_tool import FunctionTool
from agentkit.tools.registry import ToolRegistry

logger = logging.getLogger(__name__)


async def trend_insight(
    brand_id: str,
    days: int = 30,
    platforms: list[str] | None = None,
    keywords: list[str] | None = None,
) -> dict:
    """执行趋势洞察分析"""
    from app.database import AsyncSessionLocal
    from app.services.trend.trend_analyzer_service import TrendAnalyzerService

    async with AsyncSessionLocal() as db:
        service = TrendAnalyzerService(db)
        result = await service.analyze_trends(
            brand_id=brand_id,
            days=days,
            platforms=platforms,
            keywords=keywords,
        )
    return result


async def trend_hotspot(
    brand_id: str,
    days: int = 30,
) -> dict:
    """检测引用量突增的热点话题"""
    from app.database import AsyncSessionLocal
    from app.services.trend.trend_analyzer_service import TrendAnalyzerService

    async with AsyncSessionLocal() as db:
        service = TrendAnalyzerService(db)
        result = await service.get_hotspots(
            brand_id=brand_id,
            days=days,
        )
    return result


def register_trend_tools(registry: ToolRegistry) -> None:
    """注册所有趋势分析相关工具"""
    registry.register(
        FunctionTool(
            name="trend_insight",
            description="分析品牌引用趋势，推断变化原因",
            func=trend_insight,
            tags=["trend", "insight"],
        )
    )
    registry.register(
        FunctionTool(
            name="trend_hotspot",
            description="检测引用量突增的热点话题",
            func=trend_hotspot,
            tags=["trend", "hotspot"],
        )
    )
    logger.info("Trend tools registered")