from typing import Any

from app.config import settings
from app.wechat.client import WeChatClient


class NotificationService:
    def __init__(self):
        self.client = WeChatClient()

    async def send(self, announcements: list[dict[str, Any]]) -> int:
        if not settings.wechat_enabled:
            return 0
        if not announcements:
            return 0

        sent = 0
        for ann in announcements:
            try:
                # 发送普通 textcard
                title = ann.get("title", "")
                if len(title) > 128:
                    title = title[:125] + "..."

                purchase_name = ann.get("purchase_name", "")
                if len(purchase_name) > 25:
                    purchase_name = purchase_name[:22] + "..."

                pub_date = ann.get("publish_date")
                time_str = pub_date.strftime("%Y-%m-%d %H:%M") if pub_date else "时间未知"

                source_name = ann.get("source_name", "")

                description = f"{source_name} | {purchase_name} | {time_str}"

                url = ann.get("content_url", "")

                if await self.client.send_textcard(title, description, url):
                    sent += 1

                # AI 标记为可承接的，额外发送着重通知
                ai_result = ann.get("ai_result")
                if ai_result and ai_result.get("is_relevant"):
                    await self._send_ai_emphasis(ann, ai_result)

            except Exception:
                continue

        return sent

    async def _send_ai_emphasis(
        self, ann: dict[str, Any], ai_result: dict[str, Any]
    ) -> bool:
        """发送 AI 分析的着重通知（markdown 格式）"""
        title = ann.get("title", "")
        purchase_name = ann.get("purchase_name", "")
        pub_date = ann.get("publish_date")
        time_str = pub_date.strftime("%Y-%m-%d %H:%M") if pub_date else "时间未知"
        url = ann.get("content_url", "")

        reason = ai_result.get("reason", "")
        business_type = ai_result.get("business_type", "")

        # 企业微信 markdown 格式
        md = (
            f"{settings.ai_analysis_title}\n"
            f"---\n"
            f"**标题：** [{title}]({url})\n"
            f"> 采购人：{purchase_name}\n"
            f"> 发布时间：{time_str}\n\n"
            f"**🤖 AI 分析：**\n"
            f"> {reason}\n\n"
            f"**🏷 业务分类：** {business_type}\n"
            f"---\n"
            f"[📄 查看公告原文]({url})"
        )

        return await self.client.send_markdown(md)