mirror of
https://github.com/langbot-app/LangBot.git
synced 2025-11-25 19:37:36 +08:00
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
|