mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-17 11:14:19 +00:00
Junyan Qin (Chin)
209f16af76
* style: remove necessary imports * style: fix F841 * style: fix F401 * style: fix F811 * style: fix E402 * style: fix E721 * style: fix E722 * style: fix E722 * style: fix F541 * style: ruff format * style: all passed * style: add ruff in deps * style: more ignores in ruff.toml * style: add pre-commit
486 lines
19 KiB
Python
486 lines
19 KiB
Python
from __future__ import annotations
|
|
|
|
import gewechat_client
|
|
|
|
import typing
|
|
import asyncio
|
|
import traceback
|
|
import time
|
|
import re
|
|
import base64
|
|
import copy
|
|
import threading
|
|
|
|
import quart
|
|
import aiohttp
|
|
|
|
from .. import adapter
|
|
from ...core import app
|
|
from ..types import message as platform_message
|
|
from ..types import events as platform_events
|
|
from ..types import entities as platform_entities
|
|
from ...utils import image
|
|
import xml.etree.ElementTree as ET
|
|
|
|
|
|
class GewechatMessageConverter(adapter.MessageConverter):
|
|
def __init__(self, config: dict):
|
|
self.config = config
|
|
|
|
@staticmethod
|
|
async def yiri2target(message_chain: platform_message.MessageChain) -> list[dict]:
|
|
content_list = []
|
|
for component in message_chain:
|
|
if isinstance(component, platform_message.At):
|
|
content_list.append({'type': 'at', 'target': component.target})
|
|
elif isinstance(component, platform_message.Plain):
|
|
content_list.append({'type': 'text', 'content': component.text})
|
|
elif isinstance(component, platform_message.Image):
|
|
if not component.url:
|
|
pass
|
|
content_list.append({'type': 'image', 'image': component.url})
|
|
|
|
elif isinstance(component, platform_message.Voice):
|
|
content_list.append(
|
|
{'type': 'voice', 'url': component.url, 'length': component.length}
|
|
)
|
|
elif isinstance(component, platform_message.Forward):
|
|
for node in component.node_list:
|
|
content_list.extend(
|
|
await GewechatMessageConverter.yiri2target(node.message_chain)
|
|
)
|
|
|
|
return content_list
|
|
|
|
async def target2yiri(
|
|
self, message: dict, bot_account_id: str
|
|
) -> platform_message.MessageChain:
|
|
if message['Data']['MsgType'] == 1:
|
|
# 检查消息开头，如果有 wxid_sbitaz0mt65n22:\n 则删掉
|
|
regex = re.compile(r'^wxid_.*:')
|
|
# print(message)
|
|
|
|
line_split = message['Data']['Content']['string'].split('\n')
|
|
|
|
if len(line_split) > 0 and regex.match(line_split[0]):
|
|
message['Data']['Content']['string'] = '\n'.join(line_split[1:])
|
|
|
|
# 正则表达式模式，匹配'@'后跟任意数量的非空白字符
|
|
pattern = r'@\S+'
|
|
at_string = f'@{bot_account_id}'
|
|
content_list = []
|
|
if at_string in message['Data']['Content']['string']:
|
|
content_list.append(platform_message.At(target=bot_account_id))
|
|
content_list.append(
|
|
platform_message.Plain(
|
|
message['Data']['Content']['string'].replace(at_string, '', 1)
|
|
)
|
|
)
|
|
# 更优雅的替换改名后@机器人，仅仅限于单独AT的情况
|
|
elif (
|
|
'PushContent' in message['Data']
|
|
and '在群聊中@了你' in message['Data']['PushContent']
|
|
):
|
|
if (
|
|
'@所有人' in message['Data']['Content']['string']
|
|
): # at全员时候传入atll不当作at自己
|
|
content_list.append(platform_message.AtAll())
|
|
else:
|
|
content_list.append(platform_message.At(target=bot_account_id))
|
|
content_list.append(
|
|
platform_message.Plain(
|
|
re.sub(pattern, '', message['Data']['Content']['string'])
|
|
)
|
|
)
|
|
else:
|
|
content_list = [
|
|
platform_message.Plain(message['Data']['Content']['string'])
|
|
]
|
|
|
|
return platform_message.MessageChain(content_list)
|
|
|
|
elif message['Data']['MsgType'] == 3:
|
|
image_xml = message['Data']['Content']['string']
|
|
if not image_xml:
|
|
return platform_message.MessageChain(
|
|
[platform_message.Plain(text='[图片内容为空]')]
|
|
)
|
|
|
|
try:
|
|
base64_str, image_format = await image.get_gewechat_image_base64(
|
|
gewechat_url=self.config['gewechat_url'],
|
|
gewechat_file_url=self.config['gewechat_file_url'],
|
|
app_id=self.config['app_id'],
|
|
xml_content=image_xml,
|
|
token=self.config['token'],
|
|
image_type=2,
|
|
)
|
|
|
|
return platform_message.MessageChain(
|
|
[
|
|
platform_message.Image(
|
|
base64=f'data:image/{image_format};base64,{base64_str}'
|
|
)
|
|
]
|
|
)
|
|
except Exception as e:
|
|
print(f'处理图片消息失败: {str(e)}')
|
|
return platform_message.MessageChain(
|
|
[platform_message.Plain(text='[图片处理失败]')]
|
|
)
|
|
elif message['Data']['MsgType'] == 34:
|
|
audio_base64 = message['Data']['ImgBuf']['buffer']
|
|
return platform_message.MessageChain(
|
|
[
|
|
platform_message.Voice(
|
|
base64=f'data:audio/silk;base64,{audio_base64}'
|
|
)
|
|
]
|
|
)
|
|
elif message['Data']['MsgType'] == 49:
|
|
# 支持微信聊天记录的消息类型，将 XML 内容转换为 MessageChain 传递
|
|
try:
|
|
content = message['Data']['Content']['string']
|
|
# 有三种可能的消息结构weid开头，私聊直接<?xml>和直接<msg>
|
|
if content.startswith('wxid'):
|
|
xml_list = content.split('\n')[2:]
|
|
xml_data = '\n'.join(xml_list)
|
|
elif content.startswith('<?xml'):
|
|
xml_list = content.split('\n')[1:]
|
|
xml_data = '\n'.join(xml_list)
|
|
else:
|
|
xml_data = content
|
|
|
|
content_data = ET.fromstring(xml_data)
|
|
# print(xml_data)
|
|
# 拿到细分消息类型，按照gewe接口中描述
|
|
"""
|
|
小程序：33/36
|
|
引用消息：57
|
|
转账消息：2000
|
|
红包消息：2001
|
|
视频号消息：51
|
|
"""
|
|
appmsg_data = content_data.find('.//appmsg')
|
|
data_type = appmsg_data.find('.//type').text
|
|
if data_type == '57':
|
|
user_data = appmsg_data.find('.//title').text # 拿到用户消息
|
|
quote_data = (
|
|
appmsg_data.find('.//refermsg').find('.//content').text
|
|
) # 引用原文
|
|
sender_id = (
|
|
appmsg_data.find('.//refermsg').find('.//chatusr').text
|
|
) # 引用用户id
|
|
from_name = message['Data']['FromUserName']['string']
|
|
message_list = []
|
|
if (
|
|
message['Wxid'] == sender_id and from_name.endswith('@chatroom')
|
|
): # 因为引用机制暂时无法响应用户，所以当引用用户是机器人是构建一个at激活机器人
|
|
message_list.append(platform_message.At(target=bot_account_id))
|
|
message_list.append(
|
|
platform_message.Quote(
|
|
sender_id=sender_id,
|
|
origin=platform_message.MessageChain(
|
|
[platform_message.Plain(quote_data)]
|
|
),
|
|
)
|
|
)
|
|
message_list.append(platform_message.Plain(user_data))
|
|
return platform_message.MessageChain(message_list)
|
|
elif data_type == '51':
|
|
return platform_message.MessageChain(
|
|
[platform_message.Plain(text='[视频号消息]')]
|
|
)
|
|
# print(content_data)
|
|
elif data_type == '2000':
|
|
return platform_message.MessageChain(
|
|
[platform_message.Plain(text='[转账消息]')]
|
|
)
|
|
elif data_type == '2001':
|
|
return platform_message.MessageChain(
|
|
[platform_message.Plain(text='[红包消息]')]
|
|
)
|
|
elif data_type == '5':
|
|
return platform_message.MessageChain(
|
|
[platform_message.Plain(text='[公众号消息]')]
|
|
)
|
|
elif data_type == '33' or data_type == '36':
|
|
return platform_message.MessageChain(
|
|
[platform_message.Plain(text='[小程序消息]')]
|
|
)
|
|
# print(data_type.text)
|
|
else:
|
|
try:
|
|
content_bytes = content.encode('utf-8')
|
|
decoded_content = base64.b64decode(content_bytes)
|
|
return platform_message.MessageChain(
|
|
[platform_message.Unknown(content=decoded_content)]
|
|
)
|
|
except Exception:
|
|
return platform_message.MessageChain(
|
|
[platform_message.Plain(text=content)]
|
|
)
|
|
except Exception as e:
|
|
print(f'Error processing type 49 message: {str(e)}')
|
|
return platform_message.MessageChain(
|
|
[platform_message.Plain(text='[无法解析的消息]')]
|
|
)
|
|
|
|
|
|
class GewechatEventConverter(adapter.EventConverter):
|
|
def __init__(self, config: dict):
|
|
self.config = config
|
|
self.message_converter = GewechatMessageConverter(config)
|
|
|
|
@staticmethod
|
|
async def yiri2target(event: platform_events.MessageEvent) -> dict:
|
|
pass
|
|
|
|
async def target2yiri(
|
|
self, event: dict, bot_account_id: str
|
|
) -> platform_events.MessageEvent:
|
|
# print(event)
|
|
# 排除自己发消息回调回答问题
|
|
if event['Wxid'] == event['Data']['FromUserName']['string']:
|
|
return None
|
|
# 排除公众号以及微信团队消息
|
|
if event['Data']['FromUserName']['string'].startswith('gh_') or event['Data'][
|
|
'FromUserName'
|
|
]['string'].startswith('weixin'):
|
|
return None
|
|
message_chain = await self.message_converter.target2yiri(
|
|
copy.deepcopy(event), bot_account_id
|
|
)
|
|
|
|
if not message_chain:
|
|
return None
|
|
|
|
if '@chatroom' in event['Data']['FromUserName']['string']:
|
|
# 找出开头的 wxid_ 字符串，以:结尾
|
|
sender_wxid = event['Data']['Content']['string'].split(':')[0]
|
|
|
|
return platform_events.GroupMessage(
|
|
sender=platform_entities.GroupMember(
|
|
id=sender_wxid,
|
|
member_name=event['Data']['FromUserName']['string'],
|
|
permission=platform_entities.Permission.Member,
|
|
group=platform_entities.Group(
|
|
id=event['Data']['FromUserName']['string'],
|
|
name=event['Data']['FromUserName']['string'],
|
|
permission=platform_entities.Permission.Member,
|
|
),
|
|
special_title='',
|
|
join_timestamp=0,
|
|
last_speak_timestamp=0,
|
|
mute_time_remaining=0,
|
|
),
|
|
message_chain=message_chain,
|
|
time=event['Data']['CreateTime'],
|
|
source_platform_object=event,
|
|
)
|
|
else:
|
|
return platform_events.FriendMessage(
|
|
sender=platform_entities.Friend(
|
|
id=event['Data']['FromUserName']['string'],
|
|
nickname=event['Data']['FromUserName']['string'],
|
|
remark='',
|
|
),
|
|
message_chain=message_chain,
|
|
time=event['Data']['CreateTime'],
|
|
source_platform_object=event,
|
|
)
|
|
|
|
|
|
class GeWeChatAdapter(adapter.MessagePlatformAdapter):
|
|
name: str = 'gewechat' # 定义适配器名称
|
|
|
|
bot: gewechat_client.GewechatClient
|
|
quart_app: quart.Quart
|
|
|
|
bot_account_id: str
|
|
|
|
config: dict
|
|
|
|
ap: app.Application
|
|
|
|
message_converter: GewechatMessageConverter
|
|
event_converter: GewechatEventConverter
|
|
|
|
listeners: typing.Dict[
|
|
typing.Type[platform_events.Event],
|
|
typing.Callable[[platform_events.Event, adapter.MessagePlatformAdapter], None],
|
|
] = {}
|
|
|
|
def __init__(self, config: dict, ap: app.Application):
|
|
self.config = config
|
|
self.ap = ap
|
|
|
|
self.quart_app = quart.Quart(__name__)
|
|
|
|
self.message_converter = GewechatMessageConverter(config)
|
|
self.event_converter = GewechatEventConverter(config)
|
|
|
|
@self.quart_app.route('/gewechat/callback', methods=['POST'])
|
|
async def gewechat_callback():
|
|
data = await quart.request.json
|
|
# print(json.dumps(data, indent=4, ensure_ascii=False))
|
|
|
|
if 'data' in data:
|
|
data['Data'] = data['data']
|
|
if 'type_name' in data:
|
|
data['TypeName'] = data['type_name']
|
|
# print(json.dumps(data, indent=4, ensure_ascii=False))
|
|
|
|
if 'testMsg' in data:
|
|
return 'ok'
|
|
elif 'TypeName' in data and data['TypeName'] == 'AddMsg':
|
|
try:
|
|
event = await self.event_converter.target2yiri(
|
|
data.copy(), self.bot_account_id
|
|
)
|
|
except Exception:
|
|
traceback.print_exc()
|
|
|
|
if event.__class__ in self.listeners:
|
|
await self.listeners[event.__class__](event, self)
|
|
|
|
return 'ok'
|
|
|
|
async def send_message(
|
|
self, target_type: str, target_id: str, message: platform_message.MessageChain
|
|
):
|
|
geweap_msg = await self.message_converter.yiri2target(message)
|
|
# 此处加上群消息at处理
|
|
ats = [item['target'] for item in geweap_msg if item['type'] == 'at']
|
|
|
|
for msg in geweap_msg:
|
|
# at主动发送消息
|
|
if msg['type'] == 'text':
|
|
if ats:
|
|
member_info = self.bot.get_chatroom_member_detail(
|
|
self.config['app_id'], target_id, ats[::-1]
|
|
)['data']
|
|
|
|
for member in member_info:
|
|
msg['content'] = f'@{member["nickName"]} {msg["content"]}'
|
|
self.bot.post_text(
|
|
app_id=self.config['app_id'],
|
|
to_wxid=target_id,
|
|
content=msg['content'],
|
|
ats=','.join(ats),
|
|
)
|
|
|
|
elif msg['type'] == 'image':
|
|
self.bot.post_image(
|
|
app_id=self.config['app_id'],
|
|
to_wxid=target_id,
|
|
img_url=msg['image'],
|
|
)
|
|
|
|
async def reply_message(
|
|
self,
|
|
message_source: platform_events.MessageEvent,
|
|
message: platform_message.MessageChain,
|
|
quote_origin: bool = False,
|
|
):
|
|
content_list = await self.message_converter.yiri2target(message)
|
|
|
|
ats = [item['target'] for item in content_list if item['type'] == 'at']
|
|
|
|
for msg in content_list:
|
|
if msg['type'] == 'text':
|
|
if ats:
|
|
member_info = self.bot.get_chatroom_member_detail(
|
|
self.config['app_id'],
|
|
message_source.source_platform_object['Data']['FromUserName'][
|
|
'string'
|
|
],
|
|
ats[::-1],
|
|
)['data']
|
|
|
|
for member in member_info:
|
|
msg['content'] = f'@{member["nickName"]} {msg["content"]}'
|
|
|
|
self.bot.post_text(
|
|
app_id=self.config['app_id'],
|
|
to_wxid=message_source.source_platform_object['Data'][
|
|
'FromUserName'
|
|
]['string'],
|
|
content=msg['content'],
|
|
ats=','.join(ats),
|
|
)
|
|
|
|
async def is_muted(self, group_id: int) -> bool:
|
|
pass
|
|
|
|
def register_listener(
|
|
self,
|
|
event_type: typing.Type[platform_events.Event],
|
|
callback: typing.Callable[
|
|
[platform_events.Event, adapter.MessagePlatformAdapter], None
|
|
],
|
|
):
|
|
self.listeners[event_type] = callback
|
|
|
|
def unregister_listener(
|
|
self,
|
|
event_type: typing.Type[platform_events.Event],
|
|
callback: typing.Callable[
|
|
[platform_events.Event, adapter.MessagePlatformAdapter], None
|
|
],
|
|
):
|
|
pass
|
|
|
|
async def run_async(self):
|
|
if not self.config['token']:
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.post(
|
|
f'{self.config["gewechat_url"]}/v2/api/tools/getTokenId',
|
|
json={'app_id': self.config['app_id']},
|
|
) as response:
|
|
if response.status != 200:
|
|
raise Exception(
|
|
f'获取gewechat token失败: {await response.text()}'
|
|
)
|
|
self.config['token'] = (await response.json())['data']
|
|
|
|
self.bot = gewechat_client.GewechatClient(
|
|
f'{self.config["gewechat_url"]}/v2/api', self.config['token']
|
|
)
|
|
|
|
app_id, error_msg = self.bot.login(self.config['app_id'])
|
|
if error_msg:
|
|
raise Exception(f'Gewechat 登录失败: {error_msg}')
|
|
|
|
self.config['app_id'] = app_id
|
|
|
|
self.ap.logger.info(f'Gewechat 登录成功，app_id: {app_id}')
|
|
|
|
await self.ap.platform_mgr.write_back_config('gewechat', self, self.config)
|
|
|
|
# 获取 nickname
|
|
profile = self.bot.get_profile(self.config['app_id'])
|
|
self.bot_account_id = profile['data']['nickName']
|
|
|
|
def thread_set_callback():
|
|
time.sleep(3)
|
|
ret = self.bot.set_callback(
|
|
self.config['token'], self.config['callback_url']
|
|
)
|
|
print('设置 Gewechat 回调：', ret)
|
|
|
|
threading.Thread(target=thread_set_callback).start()
|
|
|
|
async def shutdown_trigger_placeholder():
|
|
while True:
|
|
await asyncio.sleep(1)
|
|
|
|
await self.quart_app.run_task(
|
|
host='0.0.0.0',
|
|
port=self.config['port'],
|
|
shutdown_trigger=shutdown_trigger_placeholder,
|
|
)
|
|
|
|
async def kill(self) -> bool:
|
|
pass
|