2021-08-16 16:07:04 +00:00
|
|
|
|
import asyncio
|
2021-08-14 15:26:10 +00:00
|
|
|
|
import re
|
|
|
|
|
import traceback
|
2021-08-30 18:53:39 +00:00
|
|
|
|
from typing import List
|
2021-08-14 15:26:10 +00:00
|
|
|
|
|
2021-08-18 13:11:14 +00:00
|
|
|
|
from core.bots.aiogram.client import dp, bot
|
2021-08-16 16:07:04 +00:00
|
|
|
|
from core.bots.aiogram.tasks import MessageTaskManager, FinishedTasks
|
2021-10-24 10:55:45 +00:00
|
|
|
|
from core.elements import Plain, Image, MessageSession as MS, MsgInfo, Session, Voice, FetchTarget as FT, \
|
|
|
|
|
ExecutionLockList
|
2021-12-25 14:01:33 +00:00
|
|
|
|
from core.elements.message.internal import Embed
|
2021-08-14 15:26:10 +00:00
|
|
|
|
from core.elements.others import confirm_command
|
2021-11-17 14:14:57 +00:00
|
|
|
|
from core.secret_check import Secret
|
2021-08-30 18:53:39 +00:00
|
|
|
|
from database import BotDBUtil
|
2021-08-14 15:26:10 +00:00
|
|
|
|
|
|
|
|
|
|
2021-09-10 18:05:27 +00:00
|
|
|
|
def convert2lst(s) -> list:
|
|
|
|
|
if isinstance(s, str):
|
|
|
|
|
return [Plain(s)]
|
|
|
|
|
elif isinstance(s, list):
|
|
|
|
|
return s
|
|
|
|
|
elif isinstance(s, tuple):
|
|
|
|
|
return list(s)
|
|
|
|
|
|
|
|
|
|
|
2021-08-14 15:26:10 +00:00
|
|
|
|
class MessageSession(MS):
|
|
|
|
|
class Feature:
|
|
|
|
|
image = True
|
2021-08-18 13:11:14 +00:00
|
|
|
|
voice = True
|
2021-10-11 14:45:28 +00:00
|
|
|
|
forward = False
|
2021-11-16 14:19:48 +00:00
|
|
|
|
delete = True
|
2021-08-14 15:26:10 +00:00
|
|
|
|
|
|
|
|
|
async def sendMessage(self, msgchain, quote=True):
|
2021-11-17 14:14:57 +00:00
|
|
|
|
if Secret.find(msgchain):
|
|
|
|
|
return await self.sendMessage('https://wdf.ink/6Oup')
|
2021-12-25 14:01:33 +00:00
|
|
|
|
if isinstance(msgchain, (Plain, Image, Voice)):
|
|
|
|
|
msgchain = [msgchain]
|
|
|
|
|
if isinstance(msgchain, Embed):
|
|
|
|
|
msgchain = msgchain.to_msgchain()
|
2021-08-14 15:26:10 +00:00
|
|
|
|
if isinstance(msgchain, str):
|
2021-10-09 13:32:54 +00:00
|
|
|
|
if msgchain == '':
|
2021-11-21 13:20:07 +00:00
|
|
|
|
msgchain = '发生错误:机器人尝试发送空文本消息,请联系机器人开发者解决问题。\n错误汇报地址:https://github.com/Teahouse-Studios/bot/issues/new?assignees=OasisAkari&labels=bug&template=report_bug.yaml&title=%5BBUG%5D%3A+'
|
2021-08-18 13:11:14 +00:00
|
|
|
|
send = await bot.send_message(self.session.target, msgchain,
|
|
|
|
|
reply_to_message_id=self.session.message.message_id if quote and self.session.message else None)
|
2021-10-09 13:32:54 +00:00
|
|
|
|
elif isinstance(msgchain, (list, tuple)):
|
2021-08-14 15:26:10 +00:00
|
|
|
|
count = 0
|
2021-10-09 13:32:54 +00:00
|
|
|
|
send = []
|
2021-08-14 15:26:10 +00:00
|
|
|
|
for x in msgchain:
|
|
|
|
|
if isinstance(x, Plain):
|
2021-10-09 13:32:54 +00:00
|
|
|
|
send_ = await bot.send_message(self.session.target, x.text,
|
2021-11-12 14:25:53 +00:00
|
|
|
|
reply_to_message_id=self.session.message.message_id if quote
|
2021-12-25 14:01:33 +00:00
|
|
|
|
and count == 0 and self.session.message else None)
|
2021-10-09 13:32:54 +00:00
|
|
|
|
elif isinstance(x, Image):
|
2021-08-14 15:26:10 +00:00
|
|
|
|
with open(await x.get(), 'rb') as image:
|
2021-10-09 13:32:54 +00:00
|
|
|
|
send_ = await bot.send_photo(self.session.target, image,
|
2021-11-12 14:25:53 +00:00
|
|
|
|
reply_to_message_id=self.session.message.message_id if quote
|
2021-12-25 14:01:33 +00:00
|
|
|
|
and count == 0
|
|
|
|
|
and self.session.message else None)
|
2021-10-09 13:32:54 +00:00
|
|
|
|
elif isinstance(x, Voice):
|
2021-08-18 13:11:14 +00:00
|
|
|
|
with open(x.path, 'rb') as voice:
|
2021-10-09 13:32:54 +00:00
|
|
|
|
send_ = await bot.send_audio(self.session.target, voice,
|
2021-11-12 14:25:53 +00:00
|
|
|
|
reply_to_message_id=self.session.message.message_id if quote
|
2021-12-25 14:01:33 +00:00
|
|
|
|
and count == 0 and self.session.message else None)
|
|
|
|
|
elif isinstance(x, Embed):
|
|
|
|
|
chains = x.to_msgchain()
|
|
|
|
|
for y in chains:
|
|
|
|
|
if isinstance(y, Plain):
|
|
|
|
|
send_ = await bot.send_message(self.session.target, y.text,
|
|
|
|
|
reply_to_message_id=self.session.message.message_id if quote
|
|
|
|
|
and count == 0 and self.session.message else None)
|
|
|
|
|
elif isinstance(y, Image):
|
|
|
|
|
with open(await y.get(), 'rb') as image:
|
|
|
|
|
send_ = await bot.send_photo(self.session.target, image,
|
|
|
|
|
reply_to_message_id=self.session.message.message_id if quote
|
|
|
|
|
and count == 0 and self.session.message else None)
|
2021-10-09 13:32:54 +00:00
|
|
|
|
else:
|
|
|
|
|
send_ = False
|
|
|
|
|
if send_:
|
|
|
|
|
send.append(send_)
|
2021-09-10 18:05:27 +00:00
|
|
|
|
count += 1
|
2021-10-09 13:32:54 +00:00
|
|
|
|
else:
|
2021-11-21 13:20:07 +00:00
|
|
|
|
msgchain = '发生错误:机器人尝试发送非法消息链,请联系机器人开发者解决问题。\n错误汇报地址:https://github.com/Teahouse-Studios/bot/issues/new?assignees=OasisAkari&labels=bug&template=report_bug.yaml&title=%5BBUG%5D%3A+'
|
2021-10-09 13:32:54 +00:00
|
|
|
|
send = await bot.send_message(self.session.target, msgchain,
|
|
|
|
|
reply_to_message_id=self.session.message.message_id if quote and self.session.message else None)
|
|
|
|
|
return MessageSession(target=MsgInfo(targetId=0, senderId=0, senderName='', targetFrom='Telegram|Bot',
|
2021-11-12 14:25:53 +00:00
|
|
|
|
senderFrom='Telegram|Bot'),
|
2021-11-17 14:20:51 +00:00
|
|
|
|
session=Session(message=send, target=send, sender=send))
|
2021-08-14 15:26:10 +00:00
|
|
|
|
|
2021-09-10 18:05:27 +00:00
|
|
|
|
async def waitConfirm(self, msgchain=None, quote=True):
|
2021-10-24 10:55:45 +00:00
|
|
|
|
ExecutionLockList.remove(self)
|
2021-09-10 18:05:27 +00:00
|
|
|
|
send = None
|
|
|
|
|
if msgchain is not None:
|
|
|
|
|
msgchain = convert2lst(msgchain)
|
|
|
|
|
msgchain.append(Plain('(发送“是”或符合确认条件的词语来确认)'))
|
|
|
|
|
send = await self.sendMessage(msgchain, quote)
|
2021-08-16 16:07:04 +00:00
|
|
|
|
flag = asyncio.Event()
|
2021-08-18 13:11:14 +00:00
|
|
|
|
MessageTaskManager.add_task(self.session.sender, flag)
|
2021-08-16 16:07:04 +00:00
|
|
|
|
await flag.wait()
|
2021-09-10 18:05:27 +00:00
|
|
|
|
if msgchain is not None:
|
|
|
|
|
await send.delete()
|
2021-08-18 13:11:14 +00:00
|
|
|
|
if FinishedTasks.get()[self.session.sender].text in confirm_command:
|
2021-08-17 15:35:54 +00:00
|
|
|
|
return True
|
|
|
|
|
return False
|
2021-08-16 16:07:04 +00:00
|
|
|
|
|
2021-08-14 15:26:10 +00:00
|
|
|
|
async def checkPermission(self):
|
2021-08-18 13:11:14 +00:00
|
|
|
|
if self.session.message.chat.type == 'private' or self.target.senderInfo.check_TargetAdmin(
|
2021-10-20 14:04:30 +00:00
|
|
|
|
self.target.targetId) or self.target.senderInfo.query.isSuperUser:
|
2021-08-14 15:26:10 +00:00
|
|
|
|
return True
|
|
|
|
|
admins = [member.user.id for member in await dp.bot.get_chat_administrators(self.session.message.chat.id)]
|
2021-08-18 13:11:14 +00:00
|
|
|
|
if self.session.sender in admins:
|
2021-08-14 15:26:10 +00:00
|
|
|
|
return True
|
|
|
|
|
return False
|
|
|
|
|
|
|
|
|
|
def checkSuperUser(self):
|
|
|
|
|
return True if self.target.senderInfo.query.isSuperUser else False
|
|
|
|
|
|
|
|
|
|
def asDisplay(self):
|
|
|
|
|
return self.session.message.text
|
|
|
|
|
|
2021-10-24 10:55:45 +00:00
|
|
|
|
async def sleep(self, s):
|
|
|
|
|
ExecutionLockList.remove(self)
|
|
|
|
|
await asyncio.sleep(s)
|
|
|
|
|
|
2021-08-14 15:26:10 +00:00
|
|
|
|
async def delete(self):
|
|
|
|
|
try:
|
|
|
|
|
if isinstance(self.session.message, list):
|
|
|
|
|
for x in self.session.message:
|
|
|
|
|
await x.delete()
|
|
|
|
|
else:
|
|
|
|
|
await self.session.message.delete()
|
2021-08-23 12:44:31 +00:00
|
|
|
|
except Exception:
|
2021-08-14 15:26:10 +00:00
|
|
|
|
traceback.print_exc()
|
|
|
|
|
|
|
|
|
|
class Typing:
|
|
|
|
|
def __init__(self, msg: MS):
|
|
|
|
|
self.msg = msg
|
|
|
|
|
|
|
|
|
|
async def __aenter__(self):
|
2021-11-12 14:25:53 +00:00
|
|
|
|
# await bot.answer_chat_action(self.msg.session.target, 'typing')
|
2021-08-23 16:49:54 +00:00
|
|
|
|
pass
|
2021-08-14 15:26:10 +00:00
|
|
|
|
|
|
|
|
|
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
2021-08-21 15:58:07 +00:00
|
|
|
|
class FetchTarget(FT):
|
2021-08-14 15:26:10 +00:00
|
|
|
|
@staticmethod
|
2021-08-30 18:53:39 +00:00
|
|
|
|
async def fetch_target(targetId) -> MessageSession:
|
2021-08-18 13:11:14 +00:00
|
|
|
|
matchChannel = re.match(r'^(Telegram\|.*?)\|(.*)', targetId)
|
2021-08-14 15:26:10 +00:00
|
|
|
|
if matchChannel:
|
|
|
|
|
return MessageSession(MsgInfo(targetId=targetId, senderId=targetId, senderName='',
|
|
|
|
|
targetFrom=matchChannel.group(1), senderFrom=matchChannel.group(1)),
|
2021-08-18 13:11:14 +00:00
|
|
|
|
Session(message=False, target=matchChannel.group(2), sender=matchChannel.group(2)))
|
2021-08-14 15:26:10 +00:00
|
|
|
|
else:
|
2021-08-18 13:11:14 +00:00
|
|
|
|
return False
|
2021-08-30 18:53:39 +00:00
|
|
|
|
|
2021-10-08 11:54:27 +00:00
|
|
|
|
@staticmethod
|
|
|
|
|
async def fetch_target_list(targetList: list) -> List[MessageSession]:
|
|
|
|
|
lst = []
|
|
|
|
|
for x in targetList:
|
|
|
|
|
fet = await FetchTarget.fetch_target(x)
|
|
|
|
|
if fet:
|
|
|
|
|
lst.append(fet)
|
|
|
|
|
return lst
|
|
|
|
|
|
2021-08-30 18:53:39 +00:00
|
|
|
|
@staticmethod
|
|
|
|
|
async def post_message(module_name, message, user_list: List[MessageSession] = None):
|
|
|
|
|
send_list = []
|
|
|
|
|
if user_list is not None:
|
|
|
|
|
for x in user_list:
|
|
|
|
|
try:
|
|
|
|
|
send = await x.sendMessage(message, quote=False)
|
|
|
|
|
send_list.append(send)
|
|
|
|
|
except Exception:
|
|
|
|
|
traceback.print_exc()
|
|
|
|
|
else:
|
|
|
|
|
get_target_id = BotDBUtil.Module.get_enabled_this(module_name)
|
|
|
|
|
for x in get_target_id:
|
|
|
|
|
fetch = await FetchTarget.fetch_target(x)
|
|
|
|
|
if fetch:
|
|
|
|
|
try:
|
|
|
|
|
send = await fetch.sendMessage(message, quote=False)
|
|
|
|
|
send_list.append(send)
|
|
|
|
|
except Exception:
|
2021-09-03 19:19:07 +00:00
|
|
|
|
traceback.print_exc()
|
|
|
|
|
return send_list
|