Soulter
217 lines
9.9 KiB
Python
217 lines
9.9 KiB
Python
import io
|
|
import botpy
|
|
from PIL import Image as PILImage
|
|
from botpy.message import Message, DirectMessage
|
|
import re
|
|
import asyncio
|
|
import requests
|
|
from cores.qqbot.personality import personalities
|
|
from util import general_utils as gu
|
|
from nakuru.entities.components import Plain, At, Image
|
|
from botpy.types.message import Reference
|
|
from botpy import Client
|
|
import time
|
|
|
|
class NakuruGuildMember():
|
|
tiny_id: int # 发送者识别号
|
|
user_id: int # 发送者识别号
|
|
title: str
|
|
nickname: str # 昵称
|
|
role: int # 角色
|
|
icon_url: str # 头像url
|
|
|
|
class NakuruGuildMessage():
|
|
type: str = "GuildMessage"
|
|
self_id: int # bot的qq号
|
|
self_tiny_id: int # bot的qq号
|
|
sub_type: str # 消息类型
|
|
message_id: str # 消息id
|
|
guild_id: int # 频道号
|
|
channel_id: int # 子频道号
|
|
user_id: int # 发送者qq号
|
|
message: list # 消息内容
|
|
sender: NakuruGuildMember # 发送者信息
|
|
raw_message: Message
|
|
|
|
def __str__(self) -> str:
|
|
return str(self.__dict__)
|
|
|
|
class QQChan():
|
|
def __init__(self, cnt: dict = None) -> None:
|
|
self.qqchan_cnt = 0
|
|
self.waiting: dict = {}
|
|
|
|
def get_cnt(self):
|
|
return self.qqchan_cnt
|
|
|
|
def set_cnt(self, cnt):
|
|
self.qqchan_cnt = cnt
|
|
|
|
def run_bot(self, botclient: Client, appid, token):
|
|
intents = botpy.Intents(public_guild_messages=True, direct_message=True)
|
|
self.client = botclient
|
|
self.client.run(appid=appid, token=token)
|
|
|
|
# gocq-频道SDK兼容层（发）
|
|
def gocq_compatible_send(self, gocq_message_chain: list):
|
|
plain_text = ""
|
|
image_path = None # only one img supported
|
|
for i in gocq_message_chain:
|
|
if isinstance(i, Plain):
|
|
plain_text += i.text
|
|
elif isinstance(i, Image) and image_path == None:
|
|
if i.path is not None:
|
|
image_path = i.path
|
|
else:
|
|
image_path = i.file
|
|
return plain_text, image_path
|
|
|
|
# gocq-频道SDK兼容层（收）
|
|
def gocq_compatible_receive(self, message: Message) -> NakuruGuildMessage:
|
|
ngm = NakuruGuildMessage()
|
|
try:
|
|
ngm.self_id = message.mentions[0].id
|
|
ngm.self_tiny_id = message.mentions[0].id
|
|
except:
|
|
ngm.self_id = 0
|
|
ngm.self_tiny_id = 0
|
|
|
|
ngm.sub_type = "normal"
|
|
ngm.message_id = message.id
|
|
ngm.guild_id = int(message.guild_id)
|
|
ngm.channel_id = int(message.channel_id)
|
|
ngm.user_id = int(message.author.id)
|
|
msg = []
|
|
plain_content = message.content.replace("<@!"+str(ngm.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)
|
|
ngm.message = msg
|
|
ngm.sender = NakuruGuildMember()
|
|
ngm.sender.tiny_id = int(message.author.id)
|
|
ngm.sender.user_id = int(message.author.id)
|
|
ngm.sender.title = ""
|
|
ngm.sender.nickname = message.author.username
|
|
ngm.sender.role = 0
|
|
ngm.sender.icon_url = message.author.avatar
|
|
ngm.raw_message = message
|
|
return ngm
|
|
|
|
def send_qq_msg(self,
|
|
message: NakuruGuildMessage,
|
|
res: list):
|
|
'''
|
|
回复频道消息
|
|
'''
|
|
gu.log("回复QQ频道消息: "+str(res), level=gu.LEVEL_INFO, tag="QQ频道", max_len=500)
|
|
self.qqchan_cnt += 1
|
|
plain_text = ""
|
|
image_path = None
|
|
if isinstance(res, list):
|
|
# 兼容gocq
|
|
plain_text, image_path = self.gocq_compatible_send(res)
|
|
elif isinstance(res, str):
|
|
plain_text = res
|
|
|
|
# print(plain_text, image_path)
|
|
msg_ref = None
|
|
if message.raw_message is not None:
|
|
msg_ref = Reference(message_id=message.raw_message.id, ignore_get_message_error=False)
|
|
if image_path is not None:
|
|
msg_ref = None
|
|
if image_path.startswith("http"):
|
|
pic_res = requests.get(image_path, stream = True)
|
|
if pic_res.status_code == 200:
|
|
image = PILImage.open(io.BytesIO(pic_res.content))
|
|
image_path = gu.save_temp_img(image)
|
|
|
|
|
|
|
|
try:
|
|
# reply_res = asyncio.run_coroutine_threadsafe(message.raw_message.reply(content=str(plain_text), message_reference = msg_ref, file_image=image_path), self.client.loop)
|
|
reply_res = asyncio.run_coroutine_threadsafe(self.client.api.post_message(channel_id=str(message.channel_id),
|
|
content=str(plain_text),
|
|
msg_id=message.message_id,
|
|
file_image=image_path,
|
|
message_reference=msg_ref), self.client.loop)
|
|
reply_res.result()
|
|
except BaseException as e:
|
|
# 分割过长的消息
|
|
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:
|
|
reply_res = asyncio.run_coroutine_threadsafe(self.client.api.post_message(channel_id=str(message.channel_id),
|
|
content=str(i),
|
|
msg_id=message.message_id,
|
|
file_image=image_path,
|
|
message_reference=msg_ref), self.client.loop)
|
|
reply_res.result()
|
|
else:
|
|
# 发送qq信息
|
|
try:
|
|
# 防止被qq频道过滤消息
|
|
plain_text = plain_text.replace(".", " . ")
|
|
reply_res = asyncio.run_coroutine_threadsafe(self.client.api.post_message(channel_id=str(message.channel_id),
|
|
content=str(plain_text),
|
|
msg_id=message.message_id,
|
|
file_image=image_path,
|
|
message_reference=msg_ref), self.client.loop).result() # 发送信息
|
|
except BaseException as e:
|
|
print("QQ频道API错误: \n"+str(e))
|
|
try:
|
|
# reply_res = asyncio.run_coroutine_threadsafe(message.raw_message.reply(content=str(str.join(" ", plain_text)), message_reference = msg_ref, file_image=image_path), self.client.loop)
|
|
reply_res = asyncio.run_coroutine_threadsafe(self.client.api.post_message(channel_id=str(message.channel_id),
|
|
content=str(str.join(" ", plain_text)),
|
|
msg_id=message.message_id,
|
|
file_image=image_path,
|
|
message_reference=msg_ref), self.client.loop).result()
|
|
except BaseException as e:
|
|
plain_text = re.sub(r'(https|http)?:\/\/(\w|\.|\/|\?|\=|\&|\%)*\b', '[被隐藏的链接]', str(e), flags=re.MULTILINE)
|
|
plain_text = plain_text.replace(".", "·")
|
|
reply_res = asyncio.run_coroutine_threadsafe(self.client.api.post_message(channel_id=str(message.channel_id),
|
|
content=plain_text,
|
|
msg_id=message.message_id,
|
|
file_image=image_path,
|
|
message_reference=msg_ref), self.client.loop).result()
|
|
# send(message, f"QQ频道API错误：{str(e)}\n下面是格式化后的回答：\n{f_res}")
|
|
|
|
def push_message(self, channel_id: int, message_chain: list, message_id: int = None):
|
|
'''
|
|
推送消息, 如果有 message_id，那么就是回复消息。
|
|
'''
|
|
_n = NakuruGuildMessage()
|
|
_n.channel_id = channel_id
|
|
_n.message_id = message_id
|
|
self.send_qq_msg(_n, message_chain)
|
|
|
|
def send(self, message_obj, message_chain: list):
|
|
'''
|
|
发送信息
|
|
'''
|
|
self.send_qq_msg(message_obj, message_chain)
|
|
|
|
def wait_for_message(self, channel_id: int) -> NakuruGuildMessage:
|
|
'''
|
|
等待指定 channel_id 的下一条信息，超时 300s 后抛出异常
|
|
'''
|
|
self.waiting[channel_id] = ''
|
|
cnt = 0
|
|
while True:
|
|
if channel_id in self.waiting and self.waiting[channel_id] != '':
|
|
# 去掉
|
|
ret = self.waiting[channel_id]
|
|
del self.waiting[channel_id]
|
|
return ret
|
|
cnt += 1
|
|
if cnt > 300:
|
|
raise Exception("等待消息超时。")
|
|
time.sleep(1)
|
|
|