ForcePilot/backend/package/yuxi/channel/extensions/wecom/webhook_bot.py

import logging

import httpx

logger = logging.getLogger(__name__)

WEBHOOK_SEND_URL = "https://qyapi.weixin.qq.com/cgi-bin/webhook/send"


async def send_webhook_text(
    key: str,
    content: str,
    mentioned_list: list[str] | None = None,
    mentioned_mobile_list: list[str] | None = None,
) -> bool:
    url = f"{WEBHOOK_SEND_URL}?key={key}"
    text_payload: dict = {"content": content}
    if mentioned_list:
        text_payload["mentioned_list"] = mentioned_list
    if mentioned_mobile_list:
        text_payload["mentioned_mobile_list"] = mentioned_mobile_list

    payload: dict = {
        "msgtype": "text",
        "text": text_payload,
    }

    async with httpx.AsyncClient(timeout=10.0) as client:
        try:
            resp = await client.post(url, json=payload)
            data = resp.json()
            ok = data.get("errcode") == 0
            if not ok:
                logger.warning("WeCom webhook text send failed: %s", data)
            return ok
        except Exception:
            logger.exception("WeCom webhook text send error")
            return False


async def send_webhook_markdown(content: str, key: str) -> bool:
    url = f"{WEBHOOK_SEND_URL}?key={key}"
    payload = {
        "msgtype": "markdown",
        "markdown": {"content": content},
    }
    async with httpx.AsyncClient(timeout=10.0) as client:
        try:
            resp = await client.post(url, json=payload)
            data = resp.json()
            ok = data.get("errcode") == 0
            if not ok:
                logger.warning("WeCom webhook markdown send failed: %s", data)
            return ok
        except Exception:
            logger.exception("WeCom webhook markdown send error")
            return False


async def send_webhook_image(key: str, base64_content: str, md5_hash: str) -> bool:
    url = f"{WEBHOOK_SEND_URL}?key={key}"
    payload = {
        "msgtype": "image",
        "image": {"base64": base64_content, "md5": md5_hash},
    }
    async with httpx.AsyncClient(timeout=10.0) as client:
        try:
            resp = await client.post(url, json=payload)
            data = resp.json()
            ok = data.get("errcode") == 0
            if not ok:
                logger.warning("WeCom webhook image send failed: %s", data)
            return ok
        except Exception:
            logger.exception("WeCom webhook image send error")
            return False


async def send_webhook_news(key: str, articles: list[dict]) -> bool:
    url = f"{WEBHOOK_SEND_URL}?key={key}"
    payload = {
        "msgtype": "news",
        "news": {"articles": articles[:8]},
    }
    async with httpx.AsyncClient(timeout=10.0) as client:
        try:
            resp = await client.post(url, json=payload)
            data = resp.json()
            ok = data.get("errcode") == 0
            if not ok:
                logger.warning("WeCom webhook news send failed: %s", data)
            return ok
        except Exception:
            logger.exception("WeCom webhook news send error")
            return False