Soulter
381 lines
16 KiB
Python
381 lines
16 KiB
Python
import botpy
|
||
import re
|
||
import time
|
||
import traceback
|
||
import asyncio
|
||
import botpy.message
|
||
import botpy.types
|
||
import botpy.types.message
|
||
|
||
from botpy.types.message import Reference, Media
|
||
from botpy import Client
|
||
from util.io import save_temp_img, download_image_by_url
|
||
from . import Platform
|
||
from type.astrbot_message import *
|
||
from type.message_event import *
|
||
from type.command import *
|
||
from typing import Union, List, Dict
|
||
from nakuru.entities.components import *
|
||
from util.log import LogManager
|
||
from logging import Logger
|
||
from astrbot.message.handler import MessageHandler
|
||
from util.cmd_config import PlatformConfig, QQOfficialPlatformConfig
|
||
|
||
logger: Logger = LogManager.GetLogger(log_name='astrbot')
|
||
|
||
# QQ 机器人官方框架
|
||
class botClient(Client):
|
||
def set_platform(self, platform: 'QQOfficial'):
|
||
self.platform = platform
|
||
|
||
# 收到群消息
|
||
async def on_group_at_message_create(self, message: botpy.message.GroupMessage):
|
||
abm = self.platform._parse_from_qqofficial(message, MessageType.GROUP_MESSAGE)
|
||
await self.platform.handle_msg(abm)
|
||
|
||
# 收到频道消息
|
||
async def on_at_message_create(self, message: botpy.message.Message):
|
||
# 转换层
|
||
abm = self.platform._parse_from_qqofficial(message, MessageType.GUILD_MESSAGE)
|
||
await self.platform.handle_msg(abm)
|
||
|
||
# 收到私聊消息
|
||
async def on_direct_message_create(self, message: botpy.message.DirectMessage):
|
||
# 转换层
|
||
abm = self.platform._parse_from_qqofficial(message, MessageType.FRIEND_MESSAGE)
|
||
await self.platform.handle_msg(abm)
|
||
|
||
# 收到 C2C 消息
|
||
async def on_c2c_message_create(self, message: botpy.message.C2CMessage):
|
||
abm = self.platform._parse_from_qqofficial(message, MessageType.FRIEND_MESSAGE)
|
||
await self.platform.handle_msg(abm)
|
||
|
||
|
||
class QQOfficial(Platform):
|
||
|
||
def __init__(self, context: Context,
|
||
message_handler: MessageHandler,
|
||
platform_config: PlatformConfig,
|
||
test_mode = False) -> None:
|
||
super().__init__("qqofficial", context)
|
||
assert isinstance(platform_config, QQOfficialPlatformConfig), "qq_official: 无法识别的配置类型。"
|
||
self.loop = asyncio.new_event_loop()
|
||
asyncio.set_event_loop(self.loop)
|
||
|
||
self.message_handler = message_handler
|
||
self.context = context
|
||
self.config = platform_config
|
||
|
||
self.appid = platform_config.appid
|
||
self.secret = platform_config.secret
|
||
self.unique_session = context.config_helper.platform_settings.unique_session
|
||
qq_group = platform_config.enable_group_c2c
|
||
guild_dm = platform_config.enable_guild_direct_message
|
||
|
||
if qq_group:
|
||
self.intents = botpy.Intents(
|
||
public_messages=True,
|
||
public_guild_messages=True,
|
||
direct_message=guild_dm
|
||
)
|
||
else:
|
||
self.intents = botpy.Intents(
|
||
public_guild_messages=True,
|
||
direct_message=guild_dm
|
||
)
|
||
self.client = botClient(
|
||
intents=self.intents,
|
||
bot_log=False,
|
||
timeout=20,
|
||
)
|
||
|
||
self.client.set_platform(self)
|
||
|
||
self.test_mode = os.environ.get('TEST_MODE', 'off') == 'on'
|
||
|
||
async def _parse_to_qqofficial(self, message: List[BaseMessageComponent], is_group: bool = False):
|
||
plain_text = ""
|
||
image_path = None # only one img supported
|
||
for i in message:
|
||
if isinstance(i, Plain):
|
||
plain_text += i.text
|
||
elif isinstance(i, Image) and not image_path:
|
||
if i.path:
|
||
image_path = i.path
|
||
elif i.file and i.file.startswith("base64://"):
|
||
img_data = base64.b64decode(i.file[9:])
|
||
image_path = save_temp_img(img_data)
|
||
elif i.file and i.file.startswith("http"):
|
||
# 如果是群消息，不需要下载
|
||
image_path = await download_image_by_url(i.file) if not is_group else i.file
|
||
return plain_text, image_path
|
||
|
||
def _parse_from_qqofficial(self, message: Union[botpy.message.Message, botpy.message.GroupMessage],
|
||
message_type: MessageType):
|
||
abm = AstrBotMessage()
|
||
abm.type = message_type
|
||
abm.timestamp = int(time.time())
|
||
abm.raw_message = message
|
||
abm.message_id = message.id
|
||
abm.tag = "qqofficial"
|
||
msg: List[BaseMessageComponent] = []
|
||
|
||
if isinstance(message, botpy.message.GroupMessage) or isinstance(message, botpy.message.C2CMessage):
|
||
if isinstance(message, botpy.message.GroupMessage):
|
||
abm.sender = MessageMember(
|
||
message.author.member_openid,
|
||
""
|
||
)
|
||
else:
|
||
abm.sender = MessageMember(
|
||
message.author.user_openid,
|
||
""
|
||
)
|
||
abm.message_str = message.content.strip()
|
||
abm.self_id = "unknown_selfid"
|
||
|
||
msg.append(Plain(abm.message_str))
|
||
if message.attachments:
|
||
for i in message.attachments:
|
||
if i.content_type.startswith("image"):
|
||
url = i.url
|
||
if not url.startswith("http"):
|
||
url = "https://"+url
|
||
img = Image.fromURL(url)
|
||
msg.append(img)
|
||
abm.message = msg
|
||
|
||
elif isinstance(message, botpy.message.Message) or isinstance(message, botpy.message.DirectMessage):
|
||
try:
|
||
abm.self_id = str(message.mentions[0].id)
|
||
except:
|
||
abm.self_id = ""
|
||
|
||
plain_content = message.content.replace(
|
||
"<@!"+str(abm.self_id)+">", "").strip()
|
||
msg.append(Plain(plain_content))
|
||
if message.attachments:
|
||
for i in message.attachments:
|
||
if i.content_type.startswith("image"):
|
||
url = i.url
|
||
if not url.startswith("http"):
|
||
url = "https://"+url
|
||
img = Image.fromURL(url)
|
||
msg.append(img)
|
||
abm.message = msg
|
||
abm.message_str = plain_content
|
||
abm.sender = MessageMember(
|
||
str(message.author.id),
|
||
str(message.author.username)
|
||
)
|
||
else:
|
||
raise ValueError(f"Unknown message type: {message_type}")
|
||
return abm
|
||
|
||
def run(self):
|
||
return self.client.start(
|
||
appid=self.appid,
|
||
secret=self.secret
|
||
)
|
||
|
||
async def handle_msg(self, message: AstrBotMessage):
|
||
assert isinstance(message.raw_message, (botpy.message.Message,
|
||
botpy.message.GroupMessage, botpy.message.DirectMessage, botpy.message.C2CMessage))
|
||
is_group = message.type != MessageType.FRIEND_MESSAGE
|
||
|
||
_t = "/私聊" if not is_group else ""
|
||
logger.info(
|
||
f"{message.sender.nickname}({message.sender.user_id}{_t}) -> {self.parse_message_outline(message)}")
|
||
|
||
# 解析出 session_id
|
||
if self.unique_session or not is_group:
|
||
session_id = message.sender.user_id
|
||
else:
|
||
if message.type == MessageType.GUILD_MESSAGE:
|
||
session_id = message.raw_message.channel_id
|
||
elif message.type == MessageType.GROUP_MESSAGE:
|
||
session_id = str(message.raw_message.group_openid)
|
||
else:
|
||
session_id = str(message.raw_message.author.id)
|
||
message.session_id = session_id
|
||
|
||
# construct astrbot message event
|
||
ame = AstrMessageEvent.from_astrbot_message(message, self.context, "qqofficial", session_id)
|
||
|
||
message_result = await self.message_handler.handle(ame)
|
||
if not message_result:
|
||
return
|
||
|
||
ret = await self.reply_msg(message, message_result.result_message, message_result.use_t2i)
|
||
if message_result.callback:
|
||
message_result.callback()
|
||
|
||
return ret
|
||
|
||
async def reply_msg(self,
|
||
message: AstrBotMessage,
|
||
result_message: List[BaseMessageComponent],
|
||
use_t2i: bool = None):
|
||
'''
|
||
回复频道消息
|
||
'''
|
||
source = message.raw_message
|
||
assert isinstance(source, (botpy.message.Message,
|
||
botpy.message.GroupMessage, botpy.message.DirectMessage, botpy.message.C2CMessage))
|
||
logger.info(
|
||
f"{message.sender.nickname}({message.sender.user_id}) <- {self.parse_message_outline(result_message)}")
|
||
|
||
plain_text = ''
|
||
image_path = ''
|
||
msg_ref = None
|
||
rendered_images = None
|
||
|
||
if use_t2i or (use_t2i == None and self.context.config_helper.t2i) and isinstance(result_message, list):
|
||
try:
|
||
rendered_images = await self.convert_to_t2i_chain(result_message)
|
||
except BaseException as e:
|
||
logger.warning(traceback.format_exc())
|
||
logger.warning(f"文本转图片时发生错误: {e}，将尝试默认方式。")
|
||
rendered_images = None
|
||
|
||
if isinstance(result_message, list):
|
||
plain_text, image_path = await self._parse_to_qqofficial(result_message, message.type == MessageType.GROUP_MESSAGE)
|
||
else:
|
||
plain_text = result_message
|
||
|
||
if source and not image_path: # file_image与message_reference不能同时传入
|
||
msg_ref = Reference(message_id=source.id,
|
||
ignore_get_message_error=False)
|
||
|
||
# 到这里，我们得到了 plain_text，image_path，msg_ref
|
||
data = {
|
||
'content': plain_text,
|
||
'msg_id': message.message_id,
|
||
'message_reference': msg_ref
|
||
}
|
||
|
||
if isinstance(message.raw_message, botpy.message.GroupMessage):
|
||
data['group_openid'] = str(source.group_openid)
|
||
elif isinstance(message.raw_message, botpy.message.Message):
|
||
data['channel_id'] = source.channel_id
|
||
elif isinstance(message.raw_message, botpy.message.DirectMessage):
|
||
data['guild_id'] = source.guild_id
|
||
elif isinstance(message.raw_message, botpy.message.C2CMessage):
|
||
data['openid'] = source.author.user_openid
|
||
if image_path:
|
||
data['file_image'] = image_path
|
||
if rendered_images:
|
||
# 文转图
|
||
_data = data.copy()
|
||
_data['content'] = ''
|
||
_data['file_image'] = rendered_images[0].file
|
||
_data['message_reference'] = None
|
||
|
||
try:
|
||
return await self._reply(**_data)
|
||
except BaseException as e:
|
||
logger.warn(traceback.format_exc())
|
||
logger.warn(f"以文本转图片的形式回复消息时发生错误: {e}，将尝试默认方式。")
|
||
|
||
try:
|
||
return await self._reply(**data)
|
||
except BaseException as e:
|
||
logger.error(traceback.format_exc())
|
||
# 分割过长的消息
|
||
if "msg over length" in str(e):
|
||
split_res = []
|
||
split_res.append(plain_text[:len(plain_text)//2])
|
||
split_res.append(plain_text[len(plain_text)//2:])
|
||
for i in split_res:
|
||
data['content'] = i
|
||
return await self._reply(**data)
|
||
else:
|
||
try:
|
||
# 防止被qq频道过滤消息
|
||
plain_text = plain_text.replace(".", " . ")
|
||
return await self._reply(**data)
|
||
except BaseException as e:
|
||
try:
|
||
data['content'] = str.join(" ", plain_text)
|
||
return await self._reply(**data)
|
||
except BaseException as e:
|
||
plain_text = re.sub(
|
||
r'(https|http)?:\/\/(\w|\.|\/|\?|\=|\&|\%)*\b', '[被隐藏的链接]', str(e), flags=re.MULTILINE)
|
||
plain_text = plain_text.replace(".", "·")
|
||
data['content'] = plain_text
|
||
return await self._reply(**data)
|
||
|
||
async def _reply(self, **kwargs):
|
||
await self.record_metrics()
|
||
if 'group_openid' in kwargs or 'openid' in kwargs:
|
||
# QQ群组消息
|
||
if 'file_image' in kwargs and kwargs['file_image']:
|
||
file_image_path = kwargs['file_image'].replace("file:///", "")
|
||
if file_image_path:
|
||
|
||
if file_image_path.startswith("http"):
|
||
image_url = file_image_path
|
||
else:
|
||
logger.debug(f"上传图片: {file_image_path}")
|
||
image_url = await self.context.image_uploader.upload_image(file_image_path)
|
||
logger.debug(f"上传成功: {image_url}")
|
||
if 'group_openid' in kwargs:
|
||
media = await self.client.api.post_group_file(kwargs['group_openid'], 1, image_url)
|
||
elif 'openid' in kwargs:
|
||
media = await self.client.api.post_c2c_file(kwargs['openid'], 1, image_url)
|
||
del kwargs['file_image']
|
||
kwargs['media'] = media
|
||
logger.debug(f"发送群图片: {media}")
|
||
kwargs['msg_type'] = 7 # 富媒体
|
||
if self.test_mode:
|
||
return kwargs
|
||
if 'group_openid' in kwargs:
|
||
await self.client.api.post_group_message(**kwargs)
|
||
elif 'openid' in kwargs:
|
||
await self.client.api.post_c2c_message(**kwargs)
|
||
elif 'channel_id' in kwargs:
|
||
# 频道消息
|
||
if 'file_image' in kwargs and kwargs['file_image']:
|
||
kwargs['file_image'] = kwargs['file_image'].replace("file:///", "")
|
||
# 频道消息发图只支持本地
|
||
if kwargs['file_image'].startswith("http"):
|
||
kwargs['file_image'] = await download_image_by_url(kwargs['file_image'])
|
||
if self.test_mode:
|
||
return kwargs
|
||
await self.client.api.post_message(**kwargs)
|
||
elif 'guild_id' in kwargs:
|
||
# 频道私聊消息
|
||
if 'file_image' in kwargs and kwargs['file_image']:
|
||
kwargs['file_image'] = kwargs['file_image'].replace("file:///", "")
|
||
if kwargs['file_image'].startswith("http"):
|
||
kwargs['file_image'] = await download_image_by_url(kwargs['file_image'])
|
||
if self.test_mode:
|
||
return kwargs
|
||
await self.client.api.post_dms(**kwargs)
|
||
else:
|
||
raise ValueError("Unknown target type.")
|
||
|
||
async def send_msg(self, target: Dict[str, str], result_message: CommandResult):
|
||
'''
|
||
以主动的方式给频道用户、群、频道或者消息列表用户（QQ用户）发送一条消息。
|
||
|
||
`target` 接收一个 dict 类型的值引用。
|
||
|
||
- 如果目标是 QQ 群，请添加 key `group_openid`。
|
||
- 如果目标是 频道消息，请添加 key `channel_id`。
|
||
- 如果目标是 频道私聊，请添加 key `guild_id`。
|
||
- 如果目标是 QQ 用户，请添加 key `openid`。
|
||
'''
|
||
plain_text, image_path = await self._parse_to_qqofficial(result_message.message_chain)
|
||
|
||
payload = {
|
||
'content': plain_text,
|
||
**target
|
||
}
|
||
if image_path:
|
||
payload['file_image'] = image_path
|
||
await self._reply(**payload)
|
||
|
||
async def send_msg_new(self, message_type: MessageType, target: str, result_message: CommandResult):
|
||
raise NotImplementedError("qqofficial 不支持此方法。")
|