191 lines
7.2 KiB
Python
191 lines
7.2 KiB
Python
import logging
|
||
import os
|
||
import re
|
||
import sys
|
||
|
||
from aiocqhttp import Event
|
||
|
||
from bots.lagrange.client import bot
|
||
from bots.lagrange.info import client_name
|
||
from bots.lagrange.message import MessageSession, FetchTarget
|
||
from bots.lagrange.utils import get_plain_msg
|
||
from config import Config
|
||
from core.builtins import EnableDirtyWordCheck, PrivateAssets, Url
|
||
from core.logger import Logger
|
||
from core.parser.message import parser
|
||
from core.queue import check_job_queue
|
||
from core.scheduler import Scheduler, IntervalTrigger
|
||
from core.types import MsgInfo, Session
|
||
from core.utils.bot import load_prompt, init_async
|
||
from core.utils.info import Info
|
||
from core.queue import JobQueue
|
||
|
||
PrivateAssets.set(os.path.abspath(os.path.dirname(__file__) + '/assets'))
|
||
EnableDirtyWordCheck.status = True if Config('enable_dirty_check') else False
|
||
Url.disable_mm = False if Config('enable_urlmanager') else True
|
||
qq_account = str(Config("qq_account"))
|
||
|
||
|
||
@Scheduler.scheduled_job(IntervalTrigger(seconds=1))
|
||
async def job():
|
||
await check_job_queue()
|
||
|
||
|
||
@bot.on_startup
|
||
async def startup():
|
||
# await init_async()
|
||
Scheduler.start()
|
||
bot.logger.setLevel(logging.WARNING)
|
||
|
||
|
||
@bot.on_meta_event('heartbeat')
|
||
async def _(event: Event):
|
||
if event['status']['online']:
|
||
await JobQueue.add_job('QQ', 'lagrange_keepalive', event['status'], wait=False)
|
||
glist = await bot.call_action('get_group_list')
|
||
g = []
|
||
for i in glist:
|
||
g.append(i['group_id'])
|
||
await JobQueue.add_job('QQ', 'lagrange_available_groups', g, wait=False)
|
||
|
||
|
||
async def message_handler(event: Event):
|
||
if event.detail_type == 'private':
|
||
if event.sub_type == 'group':
|
||
if Config('qq_disable_temp_session'):
|
||
return await bot.send(event, '请先添加好友后再进行命令交互。')
|
||
|
||
message = get_plain_msg(event.message)
|
||
"""
|
||
filter_msg = re.match(r'.*?\\[CQ:(?:json|xml).*?\\].*?|.*?<\\?xml.*?>.*?', event.message, re.MULTILINE | re.DOTALL)
|
||
if filter_msg:
|
||
match_json = re.match(r'.*?\\[CQ:json,data=(.*?)\\].*?', event.message, re.MULTILINE | re.DOTALL)
|
||
if match_json:
|
||
load_json = json.loads(html.unescape(match_json.group(1)))
|
||
if load_json['app'] == 'com.tencent.multimsg':
|
||
event.message = f'[CQ:forward,id={load_json["meta"]["detail"]["resid"]}]'
|
||
else:
|
||
return
|
||
else:
|
||
return
|
||
reply_id = None
|
||
match_reply = re.match(r'^\\[CQ:reply,id=(.*?)].*', event.message)
|
||
if match_reply:
|
||
reply_id = int(match_reply.group(1))
|
||
|
||
prefix = None
|
||
if match_at := re.match(r'^\\[CQ:at,qq=(.*?)](.*)', event.message):
|
||
if match_at.group(1) == qq_account:
|
||
event.message = match_at.group(2)
|
||
if event.message in ['', ' ']:
|
||
event.message = 'help'
|
||
prefix = ['']"""
|
||
if event.detail_type == 'group':
|
||
if event.group_id not in Config('lagrange_avaliable_groups'):
|
||
return
|
||
|
||
target_id = 'QQ|' + (f'Group|{str(event.group_id)}' if event.detail_type == 'group' else str(event.user_id))
|
||
|
||
msg = MessageSession(MsgInfo(target_id=target_id,
|
||
sender_id=f'QQ|{str(event.user_id)}',
|
||
target_from='QQ|Group' if event.detail_type == 'group' else 'QQ',
|
||
sender_from='QQ', sender_name=event.sender['nickname'], client_name=client_name,
|
||
message_id=event.message_id,
|
||
reply_id=None),
|
||
Session(message=message,
|
||
target=event.group_id if event.detail_type == 'group' else event.user_id,
|
||
sender=event.user_id))
|
||
await parser(msg, running_mention=True, prefix=['N~'])
|
||
|
||
"""@bot.on_message('group', 'private')
|
||
async def _(event: Event):
|
||
await message_handler(event)"""
|
||
|
||
"""
|
||
if enable_listening_self_message:
|
||
@bot.on('message_sent')
|
||
async def _(event: Event):
|
||
await message_handler(event)
|
||
|
||
|
||
|
||
|
||
|
||
class GuildAccountInfo:
|
||
tiny_id = None
|
||
|
||
|
||
@bot.on_message('guild')
|
||
async def _(event):
|
||
if GuildAccountInfo.tiny_id is None:
|
||
profile = await bot.call_action('get_guild_service_profile')
|
||
GuildAccountInfo.tiny_id = profile['tiny_id']
|
||
tiny_id = event.user_id
|
||
if tiny_id == GuildAccountInfo.tiny_id:
|
||
return
|
||
reply_id = None
|
||
match_reply = re.match(r'^\\[CQ:reply,id=(.*?)].*', event.message)
|
||
if match_reply:
|
||
reply_id = int(match_reply.group(1))
|
||
target_id = f'QQ|Guild|{str(event.guild_id)}|{str(event.channel_id)}'
|
||
msg = MessageSession(MsgInfo(target_id=target_id,
|
||
sender_id=f'QQ|Tiny|{str(event.user_id)}',
|
||
target_from='QQ|Guild',
|
||
sender_from='QQ|Tiny', sender_name=event.sender['nickname'], client_name=client_name,
|
||
message_id=event.message_id,
|
||
reply_id=reply_id),
|
||
Session(message=event,
|
||
target=f'{str(event.guild_id)}|{str(event.channel_id)}',
|
||
sender=event.user_id))
|
||
await parser(msg, running_mention=True)
|
||
|
||
|
||
@bot.on('request.friend')
|
||
async def _(event: Event):
|
||
if BotDBUtil.SenderInfo('QQ|' + str(event.user_id)).query.isInBlockList:
|
||
return {'approve': False}
|
||
return {'approve': True}
|
||
|
||
|
||
@bot.on('request.group.invite')
|
||
async def _(event: Event):
|
||
if BotDBUtil.SenderInfo('QQ|' + str(event.user_id)).query.isSuperUser:
|
||
return {'approve': True}
|
||
if not Config('allow_bot_auto_approve_group_invite'):
|
||
await bot.send_private_msg(user_id=event.user_id,
|
||
message='你好！本机器人暂时不主动同意入群请求。\n'
|
||
f'请至{Config("qq_join_group_application_link")}申请入群。')
|
||
else:
|
||
return {'approve': True}
|
||
|
||
|
||
@bot.on_notice('group_ban')
|
||
async def _(event: Event):
|
||
if event.user_id == int(qq_account):
|
||
result = BotDBUtil.UnfriendlyActions(target_id=event.group_id,
|
||
sender_id=event.operator_id).add_and_check('mute', str(event.duration))
|
||
if event.duration >= 259200:
|
||
result = True
|
||
if result:
|
||
await bot.call_action('set_group_leave', group_id=event.group_id)
|
||
BotDBUtil.SenderInfo('QQ|' + str(event.operator_id)).edit('isInBlockList', True)
|
||
await bot.call_action('delete_friend', friend_id=event.operator_id)
|
||
|
||
|
||
|
||
@bot.on_message('group')
|
||
async def _(event: Event):
|
||
result = BotDBUtil.isGroupInAllowList(f'QQ|Group|{str(event.group_id)}')
|
||
if not result:
|
||
await bot.send(event=event, message='此群不在白名单中，已自动退群。'
|
||
'\n如需申请白名单，请至https://github.com/Teahouse-Studios/bot/issues/new/choose发起issue。')
|
||
await bot.call_action('set_group_leave', group_id=event.group_id)
|
||
"""
|
||
|
||
qq_host = Config("lagrange_host")
|
||
if qq_host:
|
||
argv = sys.argv
|
||
if 'subprocess' in sys.argv:
|
||
Info.subprocess = True
|
||
host, port = qq_host.split(':')
|
||
bot.run(host=host, port=port, debug=False)
|