import json import re import xml.dom.minidom from typing import Optional, Union, List, Tuple, Any, Dict from urllib.parse import quote from app.domain.context import Context, MediaInfo from app.application.messaging.agent import ( matches_channel_admin, register_channel_admin_resolver, resolve_config_principal_ids, ) from app.runtime.log import logger from app.modules._base import _MessageChannelModuleBase from app.adapters.external.wechat_crypt import WXBizMsgCrypt from app.modules.wechat.wechat import WeChat from app.modules.wechat.wechatbot import WeChatBot from app.schemas.notification import NotificationChannel from app.schemas.message import IncomingMessage from app.schemas.message import Message from app.schemas.types import ModuleType from app.foundation.dom import DomUtils def _resolve_wechat_admin_ids(config: Optional[dict]) -> set[str]: """解析企业微信管理员及机器人模式下的主用户 ID。""" config_keys = ["WECHAT_ADMINS"] if (config or {}).get("WECHAT_MODE", "app") == "bot": config_keys.append("WECHAT_BOT_CHAT_ID") return resolve_config_principal_ids(config, *config_keys) register_channel_admin_resolver(NotificationChannel.Wechat, _resolve_wechat_admin_ids) class WechatModule(_MessageChannelModuleBase[WeChat]): # 管理员配置键，与渠道 resolver 保持一致 _admin_config_key = "WECHAT_ADMINS" # 命令注册事件源标识固定为 WeChat（get_name 为“企业微信”） _command_origin = "WeChat" def init_module(self) -> None: """ 初始化模块 """ super().init_service(service_name=WeChat.__name__.lower(), service_type=self._create_client) self._channel = NotificationChannel.Wechat @staticmethod def get_name() -> str: return "企业微信" @staticmethod def get_type() -> ModuleType: """ 获取模块类型 """ return ModuleType.Notification @staticmethod def get_subtype() -> NotificationChannel: """ 获取模块的子类型 """ return NotificationChannel.Wechat @staticmethod def get_priority() -> int: """ 获取模块优先级，数字越小优先级越高，只有同一接口下优先级才生效 """ return 1 def stop(self) -> None: """停止模块""" for client in self.get_instances().values(): try: if hasattr(client, "stop"): client.stop() except Exception as err: logger.error(f"停止微信模块实例失败：{err}") @staticmethod def _is_bot_mode(config: dict) -> bool: return (config or {}).get("WECHAT_MODE", "app") == "bot" @classmethod def _create_client(cls, conf): if cls._is_bot_mode(conf.config): return WeChatBot(name=conf.name, **conf.config) return WeChat(name=conf.name, **conf.config) def init_setting(self) -> Tuple[str, Union[str, bool]]: pass def message_parser(self, source: str, body: Any, form: Any, args: Any) -> Optional[IncomingMessage]: """ 解析消息内容，返回字典，注意以下约定值： userid: 用户ID username: 用户名 text: 内容 :param source: 消息来源 :param body: 请求体 :param form: 表单 :param args: 参数 :return: 渠道、消息体 """ try: # 获取服务配置 client_config = self.get_config(source) if not client_config: return None if self._is_bot_mode(client_config.config): return self._parse_bot_message(source=source, body=body, client_config=client_config) client: WeChat = self.get_instance(client_config.name) # URL参数 sVerifyMsgSig = args.get("msg_signature") sVerifyTimeStamp = args.get("timestamp") sVerifyNonce = args.get("nonce") if not sVerifyMsgSig or not sVerifyTimeStamp or not sVerifyNonce: logger.debug(f"微信请求参数错误：{args}") return None # 解密模块 wxcpt = WXBizMsgCrypt(sToken=client_config.config.get('WECHAT_TOKEN'), sEncodingAESKey=client_config.config.get('WECHAT_ENCODING_AESKEY'), sReceiveId=client_config.config.get('WECHAT_CORPID')) # 报文数据 if not body: logger.debug(f"微信请求数据为空") return None logger.debug(f"收到微信请求：{body}") ret, sMsg = wxcpt.DecryptMsg(sPostData=body, sMsgSignature=sVerifyMsgSig, sTimeStamp=sVerifyTimeStamp, sNonce=sVerifyNonce) if ret != 0: logger.error(f"解密微信消息失败 DecryptMsg ret = {ret}") return None # 解析XML报文 """ 1、消息格式： 1348831860 1234567890123456 1 2、事件格式： 1348831860 1 """ dom_tree = xml.dom.minidom.parseString(sMsg.decode('UTF-8')) root_node = dom_tree.documentElement # 消息类型 msg_type = DomUtils.tag_value(root_node, "MsgType") # Event event事件只有click才有效,enter_agent无效 event = DomUtils.tag_value(root_node, "Event") # 用户ID user_id = DomUtils.tag_value(root_node, "FromUserName") # 没的消息类型和用户ID的消息不要 if not msg_type or not user_id: logger.warn(f"解析不到消息类型和用户ID") return None # 解析消息内容 content = None images = None audio_refs = None files = None if msg_type == "event" and event == "click": # 企业微信菜单最终会转成命令文本，需与斜杠命令使用一致的管理员校验。 if self._should_reject_admin_command(client_config.config, user_id): client.send_msg(title="只有管理员才有权限执行此命令", userid=user_id) return None # 根据EventKey执行命令 content = DomUtils.tag_value(root_node, "EventKey") logger.info(f"收到来自 {client_config.name} 的微信事件：userid={user_id}, event={content}") elif msg_type == "text": # 文本消息 content = DomUtils.tag_value(root_node, "Content", default="") logger.info(f"收到来自 {client_config.name} 的微信消息：userid={user_id}, text={content}") elif msg_type == "image": media_id = DomUtils.tag_value(root_node, "MediaId") pic_url = DomUtils.tag_value(root_node, "PicUrl") if media_id: images = [IncomingMessage.MessageImage(ref=f"wxwork://media_id/{media_id}")] elif pic_url: images = [IncomingMessage.MessageImage(ref=pic_url)] logger.info( f"收到来自 {client_config.name} 的微信图片消息：userid={user_id}, images={len(images) if images else 0}" ) elif msg_type == "voice": media_id = DomUtils.tag_value(root_node, "MediaId") recognition = DomUtils.tag_value(root_node, "Recognition", default="") content = (recognition or "").strip() if media_id: audio_refs = [f"wxwork://voice_media_id/{media_id}"] logger.info( f"收到来自 {client_config.name} 的微信语音消息：userid={user_id}, " f"text={content}, audios={len(audio_refs) if audio_refs else 0}" ) elif msg_type == "file": media_id = DomUtils.tag_value(root_node, "MediaId") file_name = DomUtils.tag_value(root_node, "FileName") if media_id: files = [ IncomingMessage.MessageAttachment( ref=f"wxwork://file_media_id/{media_id}", name=file_name, ) ] logger.info( f"收到来自 {client_config.name} 的微信文件消息：userid={user_id}, files={len(files) if files else 0}" ) else: return None if content and content.startswith("/") and self._should_reject_admin_command( client_config.config, user_id ): client.send_msg(title="只有管理员才有权限执行此命令", userid=user_id) return None if content or images or audio_refs or files: # 处理消息内容 return IncomingMessage(channel=NotificationChannel.Wechat, source=client_config.name, userid=user_id, username=user_id, is_channel_admin=matches_channel_admin( NotificationChannel.Wechat, client_config.config, user_id, ), text=content or "", images=images, audio_refs=audio_refs, files=files) except Exception as err: logger.error(f"微信消息处理发生错误：{str(err)}") return None def _parse_bot_message(self, source: str, body: Any, client_config) -> Optional[IncomingMessage]: try: if isinstance(body, bytes): msg_json = json.loads(body) elif isinstance(body, dict): msg_json = body else: msg_json = json.loads(body) while isinstance(msg_json, str): msg_json = json.loads(msg_json) except Exception as err: logger.debug(f"解析企业微信智能机器人消息失败：{err}") return None if not isinstance(msg_json, dict): return None payload_body = msg_json.get("body") or {} sender = ((payload_body.get("from") or {}).get("userid") or "").strip() if not sender: return None if payload_body.get("chattype") == "group": return None text = WeChatBot._extract_text_from_body(payload_body) images = WeChatBot._extract_images_from_body(payload_body) audio_refs = ["wxbot://voice"] if payload_body.get("msgtype") == "voice" else None files = None if payload_body.get("msgtype") == "file": file_payload = payload_body.get("file") or {} download_url = file_payload.get("download_url") if download_url: files = [ IncomingMessage.MessageAttachment( ref=f"wxbot://file/{quote(download_url, safe='')}", name=file_payload.get("name") or file_payload.get("filename"), mime_type=file_payload.get("content_type") or file_payload.get("mime_type"), size=file_payload.get("size"), ) ] if text: text = re.sub(r"@\S+", "", text).strip() if text and text.startswith("/") and self._should_reject_admin_command( client_config.config, sender ): client: WeChatBot = self.get_instance(client_config.name) if client: client.send_msg(title="只有管理员才有权限执行此命令", userid=sender) return None if not text and not images and not audio_refs and not files: return None logger.info( f"收到来自 {client_config.name} 的企业微信智能机器人消息：" f"userid={sender}, text={text}, images={len(images) if images else 0}" ) return IncomingMessage( channel=NotificationChannel.Wechat, source=client_config.name, userid=sender, username=sender, is_channel_admin=matches_channel_admin( NotificationChannel.Wechat, client_config.config, sender, ), text=text or "", images=images, audio_refs=audio_refs, files=files, ) def post_message(self, message: Message, **kwargs) -> None: """ 发送消息 :param message: 消息内容 :return: 成功或失败 """ for conf in self.get_configs().values(): if not self.check_message(message, conf.name): continue targets = message.targets userid = message.userid if not userid and targets is not None: userid = targets.get('wechat_userid') if not userid: logger.warn(f"用户没有指定 微信用户ID，消息无法发送") return client: WeChat = self.get_instance(conf.name) if client: if message.voice_path and hasattr(client, "send_voice"): sent = client.send_voice( voice_path=message.voice_path, userid=userid, ) if not sent: client.send_msg(title=message.title, text=message.text, image=message.image, userid=userid, link=message.link) else: client.send_msg(title=message.title, text=message.text, image=message.image, userid=userid, link=message.link) def download_wechat_image_to_data_url(self, image_ref: str, source: str) -> Optional[str]: """ 下载企业微信渠道图片并转换为 data URL """ if not image_ref: return None client_config = self.get_config(source) if not client_config: return None client = self.get_instance(client_config.name) if not client: return None if image_ref.startswith("wxwork://media_id/") and hasattr(client, "download_media_to_data_url"): media_id = image_ref.replace("wxwork://media_id/", "", 1) return client.download_media_to_data_url(media_id) if image_ref.startswith("wxbot://image/") and hasattr(client, "download_image_to_data_url"): return client.download_image_to_data_url(image_ref) return None def download_wechat_media_bytes(self, media_ref: str, source: str) -> Optional[bytes]: """ 下载企业微信语音媒体并返回原始字节。 """ if not media_ref: return None client_config = self.get_config(source) if not client_config: return None client = self.get_instance(client_config.name) if not client or not hasattr(client, "download_media_bytes"): return None if media_ref.startswith("wxwork://voice_media_id/"): media_id = media_ref.replace("wxwork://voice_media_id/", "", 1) return client.download_media_bytes(media_id) if media_ref.startswith("wxwork://file_media_id/"): media_id = media_ref.replace("wxwork://file_media_id/", "", 1) return client.download_media_bytes(media_id) return None def post_medias_message(self, message: Message, medias: List[MediaInfo]) -> None: """ 发送媒体信息选择列表 :param message: 消息内容 :param medias: 媒体列表 :return: 成功或失败 """ for conf in self.get_configs().values(): if not self.check_message(message, conf.name): continue client: WeChat = self.get_instance(conf.name) if client: # 先发送标题 client.send_msg(title=message.title, userid=message.userid, link=message.link) # 再发送内容 client.send_medias_msg(medias=medias, userid=message.userid) def post_torrents_message(self, message: Message, torrents: List[Context]) -> None: """ 发送种子信息选择列表 :param message: 消息内容 :param torrents: 种子列表 :return: 成功或失败 """ for conf in self.get_configs().values(): if not self.check_message(message, conf.name): continue client: WeChat = self.get_instance(conf.name) if client: client.send_torrents_msg(title=message.title, torrents=torrents, userid=message.userid, link=message.link) def _commands_enabled(self, config: Optional[dict]) -> bool: """ 菜单注册前置条件：智能机器人模式无传统菜单，缺少解密参数时无法调用菜单 API。 """ if self._is_bot_mode(config): logger.debug("智能机器人模式，跳过传统菜单初始化") return False if not config.get("WECHAT_ENCODING_AESKEY") or not config.get("WECHAT_TOKEN"): logger.debug("缺少消息解密参数，跳过菜单初始化") return False return True def _delete_commands(self, client) -> None: """企业微信使用自定义菜单 API 清理命令。""" client.delete_menus() def _apply_commands(self, client, commands: Dict[str, dict]) -> None: """企业微信使用自定义菜单 API 注册命令。""" client.create_menus(commands)