mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-17 11:14:19 +00:00
Junyan Qin
331 lines
13 KiB
Python
331 lines
13 KiB
Python
# 加了之后会导致：https://github.com/Lxns-Network/nakuru-project/issues/25
|
|
# from __future__ import annotations
|
|
|
|
import asyncio
|
|
import typing
|
|
import traceback
|
|
|
|
|
|
import nakuru
|
|
import nakuru.entities.components as nkc
|
|
|
|
from .. import adapter as adapter_model
|
|
from ...pipeline.longtext.strategies import forward
|
|
from ...platform.types import message as platform_message
|
|
from ...platform.types import entities as platform_entities
|
|
from ...platform.types import events as platform_events
|
|
from ..logger import EventLogger
|
|
|
|
|
|
class NakuruProjectMessageConverter(adapter_model.MessageConverter):
|
|
"""消息转换器"""
|
|
|
|
@staticmethod
|
|
def yiri2target(message_chain: platform_message.MessageChain) -> list:
|
|
msg_list = []
|
|
if type(message_chain) is platform_message.MessageChain:
|
|
msg_list = message_chain.__root__
|
|
elif type(message_chain) is list:
|
|
msg_list = message_chain
|
|
elif type(message_chain) is str:
|
|
msg_list = [platform_message.Plain(message_chain)]
|
|
else:
|
|
raise Exception('Unknown message type: ' + str(message_chain) + str(type(message_chain)))
|
|
|
|
nakuru_msg_list = []
|
|
|
|
# 遍历并转换
|
|
for component in msg_list:
|
|
if type(component) is platform_message.Plain:
|
|
nakuru_msg_list.append(nkc.Plain(component.text, False))
|
|
elif type(component) is platform_message.Image:
|
|
if component.url is not None:
|
|
nakuru_msg_list.append(nkc.Image.fromURL(component.url))
|
|
elif component.base64 is not None:
|
|
nakuru_msg_list.append(nkc.Image.fromBase64(component.base64))
|
|
elif component.path is not None:
|
|
nakuru_msg_list.append(nkc.Image.fromFileSystem(component.path))
|
|
elif type(component) is platform_message.At:
|
|
nakuru_msg_list.append(nkc.At(qq=component.target))
|
|
elif type(component) is platform_message.AtAll:
|
|
nakuru_msg_list.append(nkc.AtAll())
|
|
elif type(component) is platform_message.Voice:
|
|
if component.url is not None:
|
|
nakuru_msg_list.append(nkc.Record.fromURL(component.url))
|
|
elif component.path is not None:
|
|
nakuru_msg_list.append(nkc.Record.fromFileSystem(component.path))
|
|
elif type(component) is forward.Forward:
|
|
# 转发消息
|
|
yiri_forward_node_list = component.node_list
|
|
nakuru_forward_node_list = []
|
|
|
|
# 遍历并转换
|
|
for yiri_forward_node in yiri_forward_node_list:
|
|
try:
|
|
content_list = NakuruProjectMessageConverter.yiri2target(yiri_forward_node.message_chain)
|
|
nakuru_forward_node = nkc.Node(
|
|
name=yiri_forward_node.sender_name,
|
|
uin=yiri_forward_node.sender_id,
|
|
time=int(yiri_forward_node.time.timestamp())
|
|
if yiri_forward_node.time is not None
|
|
else None,
|
|
content=content_list,
|
|
)
|
|
nakuru_forward_node_list.append(nakuru_forward_node)
|
|
except Exception:
|
|
import traceback
|
|
|
|
traceback.print_exc()
|
|
|
|
nakuru_msg_list.append(nakuru_forward_node_list)
|
|
else:
|
|
nakuru_msg_list.append(nkc.Plain(str(component)))
|
|
|
|
return nakuru_msg_list
|
|
|
|
@staticmethod
|
|
def target2yiri(message_chain: typing.Any, message_id: int = -1) -> platform_message.MessageChain:
|
|
"""将Yiri的消息链转换为YiriMirai的消息链"""
|
|
assert type(message_chain) is list
|
|
|
|
yiri_msg_list = []
|
|
import datetime
|
|
|
|
# 添加Source组件以标记message_id等信息
|
|
yiri_msg_list.append(platform_message.Source(id=message_id, time=datetime.datetime.now()))
|
|
for component in message_chain:
|
|
if type(component) is nkc.Plain:
|
|
yiri_msg_list.append(platform_message.Plain(text=component.text))
|
|
elif type(component) is nkc.Image:
|
|
yiri_msg_list.append(platform_message.Image(url=component.url))
|
|
elif type(component) is nkc.At:
|
|
yiri_msg_list.append(platform_message.At(target=component.qq))
|
|
elif type(component) is nkc.AtAll:
|
|
yiri_msg_list.append(platform_message.AtAll())
|
|
else:
|
|
pass
|
|
# logging.debug("转换后的消息链: " + str(yiri_msg_list))
|
|
chain = platform_message.MessageChain(yiri_msg_list)
|
|
return chain
|
|
|
|
|
|
class NakuruProjectEventConverter(adapter_model.EventConverter):
|
|
"""事件转换器"""
|
|
|
|
@staticmethod
|
|
def yiri2target(event: typing.Type[platform_events.Event]):
|
|
if event is platform_events.GroupMessage:
|
|
return nakuru.GroupMessage
|
|
elif event is platform_events.FriendMessage:
|
|
return nakuru.FriendMessage
|
|
else:
|
|
raise Exception('未支持转换的事件类型: ' + str(event))
|
|
|
|
@staticmethod
|
|
def target2yiri(event: typing.Any) -> platform_events.Event:
|
|
yiri_chain = NakuruProjectMessageConverter.target2yiri(event.message, event.message_id)
|
|
if type(event) is nakuru.FriendMessage: # 私聊消息事件
|
|
return platform_events.FriendMessage(
|
|
sender=platform_entities.Friend(
|
|
id=event.sender.user_id,
|
|
nickname=event.sender.nickname,
|
|
remark=event.sender.nickname,
|
|
),
|
|
message_chain=yiri_chain,
|
|
time=event.time,
|
|
)
|
|
elif type(event) is nakuru.GroupMessage: # 群聊消息事件
|
|
permission = 'MEMBER'
|
|
|
|
if event.sender.role == 'admin':
|
|
permission = 'ADMINISTRATOR'
|
|
elif event.sender.role == 'owner':
|
|
permission = 'OWNER'
|
|
|
|
return platform_events.GroupMessage(
|
|
sender=platform_entities.GroupMember(
|
|
id=event.sender.user_id,
|
|
member_name=event.sender.nickname,
|
|
permission=permission,
|
|
group=platform_entities.Group(
|
|
id=event.group_id,
|
|
name=event.sender.nickname,
|
|
permission=platform_entities.Permission.Member,
|
|
),
|
|
special_title=event.sender.title,
|
|
join_timestamp=0,
|
|
last_speak_timestamp=0,
|
|
mute_time_remaining=0,
|
|
),
|
|
message_chain=yiri_chain,
|
|
time=event.time,
|
|
)
|
|
else:
|
|
raise Exception('未支持转换的事件类型: ' + str(event))
|
|
|
|
|
|
class NakuruAdapter(adapter_model.MessagePlatformAdapter):
|
|
"""nakuru-project适配器"""
|
|
|
|
bot: nakuru.CQHTTP
|
|
bot_account_id: int
|
|
|
|
message_converter: NakuruProjectMessageConverter = NakuruProjectMessageConverter()
|
|
event_converter: NakuruProjectEventConverter = NakuruProjectEventConverter()
|
|
|
|
listener_list: list[dict]
|
|
|
|
# ap: app.Application
|
|
|
|
cfg: dict
|
|
|
|
def __init__(self, cfg: dict, ap, logger: EventLogger):
|
|
"""初始化nakuru-project的对象"""
|
|
cfg['port'] = cfg['ws_port']
|
|
del cfg['ws_port']
|
|
self.cfg = cfg
|
|
self.ap = ap
|
|
self.logger = logger
|
|
self.listener_list = []
|
|
self.bot = nakuru.CQHTTP(**self.cfg)
|
|
|
|
async def send_message(
|
|
self,
|
|
target_type: str,
|
|
target_id: str,
|
|
message: typing.Union[platform_message.MessageChain, list],
|
|
converted: bool = False,
|
|
):
|
|
task = None
|
|
|
|
converted_msg = self.message_converter.yiri2target(message) if not converted else message
|
|
|
|
# 检查是否有转发消息
|
|
has_forward = False
|
|
for msg in converted_msg:
|
|
if type(msg) is list: # 转发消息，仅回复此消息组件
|
|
has_forward = True
|
|
converted_msg = msg
|
|
break
|
|
if has_forward:
|
|
if target_type == 'group':
|
|
task = self.bot.sendGroupForwardMessage(int(target_id), converted_msg)
|
|
elif target_type == 'person':
|
|
task = self.bot.sendPrivateForwardMessage(int(target_id), converted_msg)
|
|
else:
|
|
raise Exception('Unknown target type: ' + target_type)
|
|
else:
|
|
if target_type == 'group':
|
|
task = self.bot.sendGroupMessage(int(target_id), converted_msg)
|
|
elif target_type == 'person':
|
|
task = self.bot.sendFriendMessage(int(target_id), converted_msg)
|
|
else:
|
|
raise Exception('Unknown target type: ' + target_type)
|
|
|
|
await task
|
|
|
|
async def reply_message(
|
|
self,
|
|
message_source: platform_events.MessageEvent,
|
|
message: platform_message.MessageChain,
|
|
quote_origin: bool = False,
|
|
):
|
|
message = self.message_converter.yiri2target(message)
|
|
if quote_origin:
|
|
# 在前方添加引用组件
|
|
message.insert(
|
|
0,
|
|
nkc.Reply(
|
|
id=message_source.message_chain.message_id,
|
|
),
|
|
)
|
|
if type(message_source) is platform_events.GroupMessage:
|
|
await self.send_message('group', message_source.sender.group.id, message, converted=True)
|
|
elif type(message_source) is platform_events.FriendMessage:
|
|
await self.send_message('person', message_source.sender.id, message, converted=True)
|
|
else:
|
|
raise Exception('Unknown message source type: ' + str(type(message_source)))
|
|
|
|
def is_muted(self, group_id: int) -> bool:
|
|
import time
|
|
|
|
# 检查是否被禁言
|
|
group_member_info = asyncio.run(self.bot.getGroupMemberInfo(group_id, self.bot_account_id))
|
|
return group_member_info.shut_up_timestamp > int(time.time())
|
|
|
|
def register_listener(
|
|
self,
|
|
event_type: typing.Type[platform_events.Event],
|
|
callback: typing.Callable[[platform_events.Event, adapter_model.MessagePlatformAdapter], None],
|
|
):
|
|
try:
|
|
source_cls = NakuruProjectEventConverter.yiri2target(event_type)
|
|
|
|
# 包装函数
|
|
async def listener_wrapper(app: nakuru.CQHTTP, source: source_cls):
|
|
await callback(self.event_converter.target2yiri(source), self)
|
|
|
|
# 将包装函数和原函数的对应关系存入列表
|
|
self.listener_list.append(
|
|
{
|
|
'event_type': event_type,
|
|
'callable': callback,
|
|
'wrapper': listener_wrapper,
|
|
}
|
|
)
|
|
|
|
# 注册监听器
|
|
self.bot.receiver(source_cls.__name__)(listener_wrapper)
|
|
except Exception as e:
|
|
self.logger.error(f'Error in nakuru register_listener: {traceback.format_exc()}')
|
|
raise e
|
|
|
|
def unregister_listener(
|
|
self,
|
|
event_type: typing.Type[platform_events.Event],
|
|
callback: typing.Callable[[platform_events.Event, adapter_model.MessagePlatformAdapter], None],
|
|
):
|
|
nakuru_event_name = self.event_converter.yiri2target(event_type).__name__
|
|
|
|
new_event_list = []
|
|
|
|
# 从本对象的监听器列表中查找并删除
|
|
target_wrapper = None
|
|
for listener in self.listener_list:
|
|
if listener['event_type'] == event_type and listener['callable'] == callback:
|
|
target_wrapper = listener['wrapper']
|
|
self.listener_list.remove(listener)
|
|
break
|
|
|
|
if target_wrapper is None:
|
|
raise Exception('未找到对应的监听器')
|
|
|
|
for func in self.bot.event[nakuru_event_name]:
|
|
if func.callable != target_wrapper:
|
|
new_event_list.append(func)
|
|
|
|
self.bot.event[nakuru_event_name] = new_event_list
|
|
|
|
async def run_async(self):
|
|
try:
|
|
import requests
|
|
|
|
resp = requests.get(
|
|
url='http://{}:{}/get_login_info'.format(self.cfg['host'], self.cfg['http_port']),
|
|
headers={'Authorization': 'Bearer ' + self.cfg['token'] if 'token' in self.cfg else ''},
|
|
timeout=5,
|
|
proxies=None,
|
|
)
|
|
if resp.status_code == 403:
|
|
raise Exception('go-cqhttp拒绝访问，请检查配置文件中nakuru适配器的配置')
|
|
self.bot_account_id = int(resp.json()['data']['user_id'])
|
|
except Exception:
|
|
raise Exception('获取go-cqhttp账号信息失败, 请检查是否已启动go-cqhttp并配置正确')
|
|
await self.bot._run()
|
|
self.ap.logger.info('运行 Nakuru 适配器')
|
|
while True:
|
|
await asyncio.sleep(1)
|
|
|
|
async def kill(self) -> bool:
|
|
return False
|