feat: 支持发送转发消息

This commit is contained in:
Rock Chin
2023-04-24 12:46:33 +08:00
parent c7517d31a4
commit 471768e760
+56 -14
View File
@@ -10,6 +10,8 @@ import traceback
import logging import logging
import json import json
from pkg.qqbot.blob import Forward, ForwardMessageNode, ForwardMessageDiaplay
class NakuruProjectMessageConverter(MessageConverter): class NakuruProjectMessageConverter(MessageConverter):
"""消息转换器""" """消息转换器"""
@@ -28,7 +30,7 @@ class NakuruProjectMessageConverter(MessageConverter):
# 遍历并转换 # 遍历并转换
for component in msg_list: for component in msg_list:
if type(component) is mirai.Plain: if type(component) is mirai.Plain:
nakuru_msg_list.append(nkc.Plain(component.text)) nakuru_msg_list.append(nkc.Plain(component.text, False))
elif type(component) is mirai.Image: elif type(component) is mirai.Image:
if component.url is not None: if component.url is not None:
nakuru_msg_list.append(nkc.Image.fromURL(component.url)) nakuru_msg_list.append(nkc.Image.fromURL(component.url))
@@ -44,8 +46,29 @@ class NakuruProjectMessageConverter(MessageConverter):
nakuru_msg_list.append(nkc.AtAll()) nakuru_msg_list.append(nkc.AtAll())
elif type(component) is mirai.Voice: elif type(component) is mirai.Voice:
pass pass
elif type(component) is 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 as e:
import traceback
traceback.print_exc()
nakuru_msg_list.append(nakuru_forward_node_list)
else: else:
pass nakuru_msg_list.append(nkc.Plain(str(component)))
return nakuru_msg_list return nakuru_msg_list
@@ -163,15 +186,34 @@ class NakuruProjectAdapter(MessageSourceAdapter):
self, self,
target_type: str, target_type: str,
target_id: str, target_id: str,
message: mirai.MessageChain message: typing.Union[mirai.MessageChain, list],
converted: bool = False
): ):
task = None task = None
if target_type == "group":
task = self.bot.sendGroupMessage(int(target_id), self.message_converter.yiri2target(message)) converted_msg = self.message_converter.yiri2target(message) if not converted else message
elif target_type == "person":
task = self.bot.sendFriendMessage(int(target_id), self.message_converter.yiri2target(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: else:
raise Exception("Unknown target type: " + target_type) 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)
asyncio.run(task) asyncio.run(task)
@@ -189,22 +231,22 @@ class NakuruProjectAdapter(MessageSourceAdapter):
) )
) )
if type(message_source) is mirai.GroupMessage: if type(message_source) is mirai.GroupMessage:
task = self.bot.sendGroupMessage( self.send_message(
"group",
message_source.sender.group.id, message_source.sender.group.id,
message, message,
# quote=message_source.message_id if quote_origin else None converted=True
) )
elif type(message_source) is mirai.FriendMessage: elif type(message_source) is mirai.FriendMessage:
task = self.bot.sendFriendMessage( self.send_message(
"person",
message_source.sender.id, message_source.sender.id,
message, message,
# quote=message_source.message_id if quote_origin else None converted=True
) )
else: else:
raise Exception("Unknown message source type: " + str(type(message_source))) raise Exception("Unknown message source type: " + str(type(message_source)))
asyncio.run(task)
def is_muted(self, group_id: int) -> bool: def is_muted(self, group_id: int) -> bool:
import time import time
# 检查是否被禁言 # 检查是否被禁言