mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 20:14:36 +00:00
254 lines
8.5 KiB
Python
254 lines
8.5 KiB
Python
from __future__ import annotations
|
||
|
||
import json
|
||
import os
|
||
import logging
|
||
import asyncio
|
||
|
||
from mirai import At, GroupMessage, MessageEvent, StrangerMessage, \
|
||
FriendMessage, Image, MessageChain, Plain
|
||
import mirai
|
||
import func_timeout
|
||
|
||
from ..openai import session as openai_session
|
||
|
||
from ..qqbot import process as processor
|
||
from ..utils import context
|
||
from ..plugin import host as plugin_host
|
||
from ..plugin import models as plugin_models
|
||
import tips as tips_custom
|
||
from ..qqbot import adapter as msadapter
|
||
from .ratelim import ratelim
|
||
|
||
from ..core import app, entities as core_entities
|
||
|
||
|
||
# 控制QQ消息输入输出的类
|
||
class QQBotManager:
|
||
retry = 3
|
||
|
||
adapter: msadapter.MessageSourceAdapter = None
|
||
|
||
bot_account_id: int = 0
|
||
|
||
ban_person = []
|
||
ban_group = []
|
||
|
||
# modern
|
||
ap: app.Application = None
|
||
|
||
ratelimiter: ratelim.RateLimiter = None
|
||
|
||
def __init__(self, first_time_init=True, ap: app.Application = None):
|
||
config = context.get_config_manager().data
|
||
|
||
self.ap = ap
|
||
self.ratelimiter = ratelim.RateLimiter(ap)
|
||
|
||
self.timeout = config['process_message_timeout']
|
||
self.retry = config['retry_times']
|
||
|
||
async def initialize(self):
|
||
await self.ratelimiter.initialize()
|
||
|
||
config = context.get_config_manager().data
|
||
|
||
logging.debug("Use adapter:" + config['msg_source_adapter'])
|
||
if config['msg_source_adapter'] == 'yirimirai':
|
||
from pkg.qqbot.sources.yirimirai import YiriMiraiAdapter
|
||
|
||
mirai_http_api_config = config['mirai_http_api_config']
|
||
self.bot_account_id = config['mirai_http_api_config']['qq']
|
||
self.adapter = YiriMiraiAdapter(mirai_http_api_config)
|
||
elif config['msg_source_adapter'] == 'nakuru':
|
||
from pkg.qqbot.sources.nakuru import NakuruProjectAdapter
|
||
self.adapter = NakuruProjectAdapter(config['nakuru_config'])
|
||
self.bot_account_id = self.adapter.bot_account_id
|
||
|
||
# 保存 account_id 到审计模块
|
||
from ..utils.center import apigroup
|
||
apigroup.APIGroup._runtime_info['account_id'] = "{}".format(self.bot_account_id)
|
||
|
||
context.set_qqbot_manager(self)
|
||
|
||
# 注册诸事件
|
||
# Caution: 注册新的事件处理器之后，请务必在unsubscribe_all中编写相应的取消订阅代码
|
||
async def on_friend_message(event: FriendMessage):
|
||
|
||
await self.ap.query_pool.add_query(
|
||
launcher_type=core_entities.LauncherTypes.PERSON,
|
||
launcher_id=event.sender.id,
|
||
sender_id=event.sender.id,
|
||
message_event=event,
|
||
message_chain=event.message_chain
|
||
)
|
||
|
||
self.adapter.register_listener(
|
||
FriendMessage,
|
||
on_friend_message
|
||
)
|
||
|
||
async def on_stranger_message(event: StrangerMessage):
|
||
|
||
await self.ap.query_pool.add_query(
|
||
launcher_type=core_entities.LauncherTypes.PERSON,
|
||
launcher_id=event.sender.id,
|
||
sender_id=event.sender.id,
|
||
message_event=event,
|
||
message_chain=event.message_chain
|
||
)
|
||
|
||
# nakuru不区分好友和陌生人，故仅为yirimirai注册陌生人事件
|
||
if config['msg_source_adapter'] == 'yirimirai':
|
||
self.adapter.register_listener(
|
||
StrangerMessage,
|
||
on_stranger_message
|
||
)
|
||
|
||
async def on_group_message(event: GroupMessage):
|
||
|
||
await self.ap.query_pool.add_query(
|
||
launcher_type=core_entities.LauncherTypes.GROUP,
|
||
launcher_id=event.group.id,
|
||
sender_id=event.sender.id,
|
||
message_event=event,
|
||
message_chain=event.message_chain
|
||
)
|
||
|
||
self.adapter.register_listener(
|
||
GroupMessage,
|
||
on_group_message
|
||
)
|
||
|
||
async def send(self, event, msg, check_quote=True, check_at_sender=True):
|
||
config = context.get_config_manager().data
|
||
|
||
if check_at_sender and config['at_sender']:
|
||
msg.insert(
|
||
0,
|
||
Plain(" \n")
|
||
)
|
||
|
||
# 当回复的正文中包含换行时，quote可能会自带at，此时就不再单独添加at，只添加换行
|
||
if "\n" not in str(msg[1]) or config['msg_source_adapter'] == 'nakuru':
|
||
msg.insert(
|
||
0,
|
||
At(
|
||
event.sender.id
|
||
)
|
||
)
|
||
|
||
await self.adapter.reply_message(
|
||
event,
|
||
msg,
|
||
quote_origin=True if config['quote_origin'] and check_quote else False
|
||
)
|
||
|
||
async def common_process(
|
||
self,
|
||
launcher_type: str,
|
||
launcher_id: int,
|
||
text_message: str,
|
||
message_chain: MessageChain,
|
||
sender_id: int
|
||
) -> mirai.MessageChain:
|
||
"""
|
||
私聊群聊通用消息处理方法
|
||
"""
|
||
# 检查bansess
|
||
if await self.bansess_mgr.is_banned(launcher_type, launcher_id, sender_id):
|
||
self.ap.logger.info("根据禁用列表忽略{}_{}的消息".format(launcher_type, launcher_id))
|
||
return []
|
||
|
||
if mirai.Image in message_chain:
|
||
return []
|
||
elif sender_id == self.bot_account_id:
|
||
return []
|
||
else:
|
||
# 超时则重试，重试超过次数则放弃
|
||
failed = 0
|
||
for i in range(self.retry):
|
||
try:
|
||
reply = await processor.process_message(launcher_type, launcher_id, text_message, message_chain,
|
||
sender_id)
|
||
return reply
|
||
|
||
# TODO openai 超时处理
|
||
except func_timeout.FunctionTimedOut:
|
||
logging.warning("{}_{}: 超时，重试中({})".format(launcher_type, launcher_id, i))
|
||
openai_session.get_session("{}_{}".format(launcher_type, launcher_id)).release_response_lock()
|
||
if "{}_{}".format(launcher_type, launcher_id) in processor.processing:
|
||
processor.processing.remove("{}_{}".format(launcher_type, launcher_id))
|
||
failed += 1
|
||
continue
|
||
|
||
if failed == self.retry:
|
||
openai_session.get_session("{}_{}".format(launcher_type, launcher_id)).release_response_lock()
|
||
await self.notify_admin("{} 请求超时".format("{}_{}".format(launcher_type, launcher_id)))
|
||
reply = [tips_custom.reply_message]
|
||
|
||
# 私聊消息处理
|
||
async def on_person_message(self, event: MessageEvent):
|
||
reply = ''
|
||
|
||
reply = await self.common_process(
|
||
launcher_type="person",
|
||
launcher_id=event.sender.id,
|
||
text_message=str(event.message_chain),
|
||
message_chain=event.message_chain,
|
||
sender_id=event.sender.id
|
||
)
|
||
|
||
if reply:
|
||
await self.send(event, reply, check_quote=False, check_at_sender=False)
|
||
|
||
# 群消息处理
|
||
async def on_group_message(self, event: GroupMessage):
|
||
reply = ''
|
||
|
||
text = str(event.message_chain).strip()
|
||
|
||
rule_check_res = await self.resprule_chkr.check(
|
||
text,
|
||
event.message_chain,
|
||
event.group.id,
|
||
event.sender.id
|
||
)
|
||
|
||
if rule_check_res.matching:
|
||
text = str(rule_check_res.replacement).strip()
|
||
reply = await self.common_process(
|
||
launcher_type="group",
|
||
launcher_id=event.group.id,
|
||
text_message=text,
|
||
message_chain=rule_check_res.replacement,
|
||
sender_id=event.sender.id
|
||
)
|
||
|
||
if reply:
|
||
await self.send(event, reply)
|
||
|
||
# 通知系统管理员
|
||
async def notify_admin(self, message: str):
|
||
await self.notify_admin_message_chain(MessageChain([Plain("[bot]{}".format(message))]))
|
||
|
||
async def notify_admin_message_chain(self, message: mirai.MessageChain):
|
||
config = context.get_config_manager().data
|
||
if config['admin_qq'] != 0 and config['admin_qq'] != []:
|
||
logging.info("通知管理员:{}".format(message))
|
||
|
||
admin_list = []
|
||
|
||
if type(config['admin_qq']) == int:
|
||
admin_list.append(config['admin_qq'])
|
||
|
||
for adm in admin_list:
|
||
self.adapter.send_message(
|
||
"person",
|
||
adm,
|
||
message
|
||
)
|
||
|
||
async def run(self):
|
||
await self.adapter.run_async()
|