From af64f1829eb3d7a205a20e781c3405095eec2f05 Mon Sep 17 00:00:00 2001 From: thsrite Date: Mon, 7 Aug 2023 12:05:45 +0800 Subject: [PATCH] =?UTF-8?q?feat=20wechat=E6=B6=88=E6=81=AF=E8=BD=AC?= =?UTF-8?q?=E5=8F=91=E6=8F=92=E4=BB=B6--=E6=A0=B9=E6=8D=AE=E6=AD=A3?= =?UTF-8?q?=E5=88=99=E8=BD=AC=E5=8F=91=E9=80=9A=E7=9F=A5=E5=88=B0=E5=85=B6?= =?UTF-8?q?=E4=BB=96wechat=E5=BA=94=E7=94=A8=EF=BC=8C=E8=A7=A3=E5=86=B3?= =?UTF-8?q?=E5=8F=AA=E8=83=BD=E9=85=8D=E7=BD=AE=E4=B8=80=E4=B8=AAwechat?= =?UTF-8?q?=E9=80=9A=E7=9F=A5=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/modules/wechat/__init__.py | 10 + app/plugins/messageforward/__init__.py | 396 +++++++++++++++++++++++++ app/schemas/types.py | 2 + 3 files changed, 408 insertions(+) create mode 100644 app/plugins/messageforward/__init__.py diff --git a/app/modules/wechat/__init__.py b/app/modules/wechat/__init__.py index dfaa14df..51adbbec 100644 --- a/app/modules/wechat/__init__.py +++ b/app/modules/wechat/__init__.py @@ -3,11 +3,13 @@ from typing import Optional, Union, List, Tuple, Any from app.core.config import settings from app.core.context import Context, MediaInfo +from app.core.event import EventManager from app.log import logger from app.modules import _ModuleBase, checkMessage from app.modules.wechat.WXBizMsgCrypt3 import WXBizMsgCrypt from app.modules.wechat.wechat import WeChat from app.schemas import MessageChannel, CommingMessage, Notification +from app.schemas.types import EventType from app.utils.dom import DomUtils @@ -121,6 +123,14 @@ class WechatModule(_ModuleBase): :param message: 消息内容 :return: 成功或失败 """ + # 发送事件 + EventManager().send_event(etype=EventType.WechatMessage, + data={ + "title": message.title, + "text": message.text, + "image": message.image, + "userid": message.userid, + }) return self.wechat.send_msg(title=message.title, text=message.text, image=message.image, userid=message.userid) diff --git a/app/plugins/messageforward/__init__.py b/app/plugins/messageforward/__init__.py new file mode 100644 index 00000000..14b27de9 --- /dev/null +++ b/app/plugins/messageforward/__init__.py @@ -0,0 +1,396 @@ +import json +import re +from datetime import datetime + +from app.core.config import settings +from app.plugins import _PluginBase +from app.core.event import eventmanager +from app.schemas.types import EventType +from app.utils.http import RequestUtils +from typing import Any, List, Dict, Tuple, Optional +from app.log import logger + + +class MessageForward(_PluginBase): + # 插件名称 + plugin_name = "wechat消息转发" + # 插件描述 + plugin_desc = "根据正则转发通知到其他wechat应用。" + # 插件图标 + plugin_icon = "forward.png" + # 主题色 + plugin_color = "#32ABD1" + # 插件版本 + plugin_version = "1.0" + # 插件作者 + plugin_author = "thsrite" + # 作者主页 + author_url = "https://github.com/thsrite" + # 插件配置项ID前缀 + plugin_config_prefix = "messageforward_" + # 加载顺序 + plugin_order = 16 + # 可使用的用户级别 + auth_level = 2 + + # 私有属性 + _enabled = False + _wechat = None + _pattern = None + _pattern_token = {} + + # 企业微信发送消息URL + _send_msg_url = f"{settings.WECHAT_PROXY}/cgi-bin/message/send?access_token=%s" + # 企业微信获取TokenURL + _token_url = f"{settings.WECHAT_PROXY}/cgi-bin/gettoken?corpid=%s&corpsecret=%s" + + def init_plugin(self, config: dict = None): + if config: + self._enabled = config.get("enabled") + self._wechat = config.get("wechat") + self._pattern = config.get("pattern") + + # 获取token存库 + if self._enabled and self._wechat: + self.__save_wechat_token() + + def get_state(self) -> bool: + return self._enabled + + @staticmethod + def get_command() -> List[Dict[str, Any]]: + pass + + def get_api(self) -> List[Dict[str, Any]]: + pass + + def get_form(self) -> Tuple[List[dict], Dict[str, Any]]: + """ + 拼装插件配置页面，需要返回两块数据：1、页面配置；2、数据结构 + """ + return [ + { + 'component': 'VForm', + 'content': [ + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + 'md': 6 + }, + 'content': [ + { + 'component': 'VSwitch', + 'props': { + 'model': 'enabled', + 'label': '开启转发' + } + } + ] + } + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + }, + 'content': [ + { + 'component': 'VTextarea', + 'props': { + 'model': 'wechat', + 'rows': '3', + 'label': 'wechat应用配置', + 'placeholder': 'appid:corpid:appsecret（一行一个配置）' + } + } + ] + } + ] + }, + { + 'component': 'VRow', + 'content': [ + { + 'component': 'VCol', + 'props': { + 'cols': 12, + }, + 'content': [ + { + 'component': 'VTextarea', + 'props': { + 'model': 'pattern', + 'rows': '3', + 'label': '正则配置', + 'placeholder': '对应上方wechat配置，一行一个，一一对应' + } + } + ] + } + ] + }, + ] + } + ], { + "enabled": False, + "wechat": "", + "pattern": "" + } + + def get_page(self) -> List[dict]: + pass + + @eventmanager.register(EventType.WechatMessage) + def send(self, event): + """ + 向第三方Webhook发送请求 + """ + if not self._enabled: + return + + # 消息体 + data = event.event_data + title = data['title'] + text = data['text'] + image = data['image'] + userid = data['userid'] + + # 正则匹配 + patterns = self._pattern.split("\n") + for i, pattern in enumerate(patterns): + msg_match = re.search(pattern, title) + if msg_match: + access_token, appid = self.__flush_access_token(i) + if not access_token: + continue + + # 发送消息 + if image: + self.__send_image_message(title, text, image, userid, access_token, appid, i) + else: + self.__send_message(title, text, userid, access_token, appid, i) + + def __save_wechat_token(self): + """ + 获取并存储wechat token + """ + # 查询历史 + wechat_token_history = self.get_data("wechat_token") or {} + + # 解析配置 + wechats = self._wechat.split("\n") + for i, wechat in enumerate(wechats): + wechat_config = wechat.split(":") + if len(wechat_config) != 3: + logger.error(f"{wechat} 应用配置不正确") + continue + appid = wechat_config[0] + corpid = wechat_config[1] + appsecret = wechat_config[2] + + # 查询历史是否存储token + wechat_config = wechat_token_history.get("appid") + access_token = None + expires_in = None + access_token_time = None + if wechat_config: + access_token_time = wechat_config['access_token_time'] + expires_in = wechat_config['expires_in'] + # 判断token是否过期 + if (datetime.now() - access_token_time).seconds < expires_in: + # 重新获取token + access_token, expires_in, access_token_time = self.__get_access_token(corpid=corpid, + appsecret=appsecret) + if not access_token: + # 获取token + access_token, expires_in, access_token_time = self.__get_access_token(corpid=corpid, + appsecret=appsecret) + if access_token: + wechat_token_history[appid] = { + "access_token": access_token, + "expires_in": expires_in, + "access_token_time": str(access_token_time), + "corpid": corpid, + "appsecret": appsecret + } + self._pattern_token[i] = { + "appid": appid, + "corpid": corpid, + "appsecret": appsecret, + "access_token": access_token, + "expires_in": expires_in, + "access_token_time": access_token_time, + } + else: + logger.error(f"wechat配置 appid = {appid} 获取token失败，请检查配置") + + # 保存wechat token + if wechat_token_history: + self.save_data("wechat_token", wechat_token_history) + + def __flush_access_token(self, i: int): + """ + 获取第i个配置wechat token + """ + wechat_token = self._pattern_token[i] + if not wechat_token: + logger.error(f"未获取到第 {i} 条正则对应的wechat应用token，请检查配置") + return None + access_token = wechat_token['access_token'] + expires_in = wechat_token['expires_in'] + access_token_time = wechat_token['access_token_time'] + appid = wechat_token['appid'] + corpid = wechat_token['corpid'] + appsecret = wechat_token['appsecret'] + + # 判断token有效期 + if (datetime.now() - access_token_time).seconds < expires_in: + # 重新获取token + access_token, expires_in, access_token_time = self.__get_access_token(corpid=corpid, + appsecret=appsecret) + if not access_token: + logger.error(f"wechat配置 appid = {appid} 获取token失败，请检查配置") + return None, None + + self._pattern_token[i] = { + "appid": appid, + "corpid": corpid, + "appsecret": appsecret, + "access_token": access_token, + "expires_in": expires_in, + "access_token_time": access_token_time, + } + return access_token, appid + + def __send_message(self, title: str, text: str = None, userid: str = None, access_token: str = None, + appid: str = None, i: int = None) -> \ + Optional[bool]: + """ + 发送文本消息 + :param title: 消息标题 + :param text: 消息内容 + :param userid: 消息发送对象的ID，为空则发给所有人 + :return: 发送状态，错误信息 + """ + message_url = self._send_msg_url % access_token + if text: + conent = "%s\n%s" % (title, text.replace("\n\n", "\n")) + else: + conent = title + + if not userid: + userid = "@all" + req_json = { + "touser": userid, + "msgtype": "text", + "agentid": appid, + "text": { + "content": conent + }, + "safe": 0, + "enable_id_trans": 0, + "enable_duplicate_check": 0 + } + return self.__post_request(message_url, req_json, i, title) + + def __send_image_message(self, title: str, text: str, image_url: str, userid: str = None, access_token: str = None, + appid: str = None, i: int = None) -> Optional[bool]: + """ + 发送图文消息 + :param title: 消息标题 + :param text: 消息内容 + :param image_url: 图片地址 + :param userid: 消息发送对象的ID，为空则发给所有人 + :return: 发送状态，错误信息 + """ + message_url = self._send_msg_url % access_token + if text: + text = text.replace("\n\n", "\n") + if not userid: + userid = "@all" + req_json = { + "touser": userid, + "msgtype": "news", + "agentid": appid, + "news": { + "articles": [ + { + "title": title, + "description": text, + "picurl": image_url, + "url": '' + } + ] + } + } + return self.__post_request(message_url, req_json, i, title) + + def __post_request(self, message_url: str, req_json: dict, i: int, title: str) -> bool: + """ + 向微信发送请求 + """ + try: + res = RequestUtils(content_type='application/json').post( + message_url, + data=json.dumps(req_json, ensure_ascii=False).encode('utf-8') + ) + if res and res.status_code == 200: + ret_json = res.json() + if ret_json.get('errcode') == 0: + logger.info(f"转发消息 {title} 成功") + return True + else: + if ret_json.get('errcode') == 42001: + # 重新获取token + self.__flush_access_token(i) + logger.error(f"转发消息 {title} 失败，错误信息：{ret_json}") + return False + elif res is not None: + logger.error(f"转发消息 {title} 失败，错误码：{res.status_code}，错误原因：{res.reason}") + return False + else: + logger.error(f"转发消息 {title} 失败，未获取到返回信息") + return False + except Exception as err: + logger.error(f"转发消息 {title} 失败，错误信息：{err}") + return False + + def __get_access_token(self, corpid, appsecret): + """ + 获取微信Token + :return： 微信Token + """ + try: + token_url = self._token_url % (corpid, appsecret) + res = RequestUtils().get_res(token_url) + if res: + ret_json = res.json() + if ret_json.get('errcode') == 0: + access_token = ret_json.get('access_token') + expires_in = ret_json.get('expires_in') + access_token_time = datetime.now() + + return access_token, expires_in, access_token_time + else: + logger.error(f"{ret_json.get('errmsg')}") + return None, None, None + else: + logger.error(f"{corpid} {appsecret} 获取token失败") + return None, None, None + except Exception as e: + logger.error(f"获取微信access_token失败，错误信息：{e}") + return None, None, None + + def stop_service(self): + """ + 退出插件 + """ + pass diff --git a/app/schemas/types.py b/app/schemas/types.py index c8fc39a0..33da1142 100644 --- a/app/schemas/types.py +++ b/app/schemas/types.py @@ -36,6 +36,8 @@ class EventType(Enum): MediaDeleted = "media.deleted" # 用户外来消息 UserMessage = "user.message" + # wechat消息 + WechatMessage = "wechat.message" # 系统配置Key字典