feat: message-api for mission via wecomm agent

notifier/backends.py

@@ -1,7 +1,14 @@
 import logging
+import json
 
 from api_v1.utils.wecom_webhook import send_wecom_webhook_message
 from notifier.models import NotifierChannelEnum
+from notifier.message_api import send_message_api_news_to_agents, send_message_api_text_message
+from notifier.serializers import (
+    MessageNewsRequestSerializer,
+    MessageTemplatePayloadSerializer,
+    MessageTextRequestSerializer,
+)
 
 logger = logging.getLogger(__name__)
 
@@ -17,6 +24,9 @@ class WeComWebhookNotifierBackend(BaseNotifierBackend):
     channel = NotifierChannelEnum.WECOM_WEBHOOK
 
     def notify(self, *, notifier, content: str, context: dict) -> dict:
+        content = (content or "").strip()
+        if not content:
+            raise ValueError("企业微信机器人消息 content 不能为空")
         msgtype = str(notifier.get_config_value("msgtype", "markdown") or "markdown").strip().lower()
         timeout_seconds = float(notifier.get_config_value("timeout_seconds", 10.0) or 10.0)
         key = str(notifier.get_config_value("key", "") or "").strip()
@@ -36,3 +46,89 @@ class WeComWebhookNotifierBackend(BaseNotifierBackend):
         }
         logger.info("[notifier.backends] wecom webhook sent: notifier_id=%s result=%s", notifier.id, result)
         return result
+
+
+class MessageAPINotifierBackend(BaseNotifierBackend):
+    channel = NotifierChannelEnum.MESSAGE_API
+
+    def _load_rendered_payload(self, *, content: str) -> dict:
+        content = (content or "").strip()
+        if not content:
+            raise ValueError("消息发送 API 模板渲染结果不能为空")
+        try:
+            payload = json.loads(content)
+        except json.JSONDecodeError as exc:
+            raise ValueError(f"消息发送 API 模板渲染结果不是合法 JSON: {exc}") from exc
+        if not isinstance(payload, dict):
+            raise ValueError("消息发送 API 模板渲染结果必须是 JSON 对象")
+
+        serializer = MessageTemplatePayloadSerializer(data=payload)
+        serializer.is_valid(raise_exception=True)
+        return serializer.validated_data
+
+    def notify(self, *, notifier, content: str, context: dict) -> dict:
+        rendered_payload = self._load_rendered_payload(content=content)
+        msgtype = rendered_payload["msgtype"]
+
+        if msgtype == "text":
+            payload = {
+                "agent_id": rendered_payload.get("agent_id", notifier.get_config_value("agent_id")),
+                "content": rendered_payload["content"],
+            }
+            resolved_user_ids = rendered_payload.get("user_ids") or notifier.get_config_value("user_ids")
+            if resolved_user_ids:
+                payload["user_ids"] = resolved_user_ids
+            serializer = MessageTextRequestSerializer(data=payload)
+            serializer.is_valid(raise_exception=True)
+            data = serializer.validated_data
+            response = send_message_api_text_message(
+                agent_id=data["agent_id"],
+                content=data["content"],
+                user_ids=data.get("user_ids"),
+                timeout_seconds=float(notifier.get_config_value("timeout_seconds", 10.0) or 10.0),
+            )
+            result = {
+                "channel": self.channel,
+                "msgtype": "text",
+                "agent_id": data["agent_id"],
+                "errcode": response.errcode,
+                "errmsg": response.errmsg,
+            }
+            logger.info("[notifier.backends] message api text sent: notifier_id=%s result=%s", notifier.id, result)
+            return result
+
+        payload = {
+            "agent_ids": rendered_payload.get("agent_ids", notifier.get_config_value("agent_ids")),
+            "title": rendered_payload["title"],
+            "description": rendered_payload["description"],
+            "url": rendered_payload["url"],
+            "image_url": rendered_payload["image_url"],
+        }
+        resolved_user_ids = rendered_payload.get("user_ids") or notifier.get_config_value("user_ids")
+        if resolved_user_ids:
+            payload["user_ids"] = resolved_user_ids
+        serializer = MessageNewsRequestSerializer(data=payload)
+        serializer.is_valid(raise_exception=True)
+        data = serializer.validated_data
+        results = send_message_api_news_to_agents(
+            agent_ids=data["agent_ids"],
+            title=data["title"],
+            description=data["description"],
+            url=data["url"],
+            image_url=data["image_url"],
+            user_ids=data.get("user_ids"),
+            timeout_seconds=float(notifier.get_config_value("timeout_seconds", 10.0) or 10.0),
+        )
+        if not any(item.get("ok") for item in results):
+            raise RuntimeError("All agent sends failed")
+
+        result = {
+            "channel": self.channel,
+            "msgtype": "news",
+            "agent_ids": data["agent_ids"],
+            "ok_count": sum(1 for item in results if item.get("ok")),
+            "total_count": len(results),
+            "results": results,
+        }
+        logger.info("[notifier.backends] message api news sent: notifier_id=%s result=%s", notifier.id, result)
+        return result