mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-10-21 16:27:15 +08:00
100 lines
3.5 KiB
Python
100 lines
3.5 KiB
Python
import re
|
||
import base64
|
||
|
||
from PIL import Image
|
||
from pathlib import Path
|
||
from typing import Union, Optional, Tuple
|
||
from io import BytesIO
|
||
|
||
from nonebot import get_bot, logger
|
||
from nonebot.adapters.onebot.v11 import MessageEvent, Message, MessageSegment
|
||
|
||
from .db_util import get_last_query, update_last_query
|
||
from .file_handler import load_image
|
||
from . import aiorequests
|
||
|
||
|
||
class MessageBuild:
|
||
|
||
@classmethod
|
||
def Image(cls,
|
||
img: Union[Image.Image, Path, str],
|
||
*,
|
||
size: Optional[Union[Tuple[int, int], float]] = None,
|
||
crop: Optional[Tuple[int, int, int, int]] = None,
|
||
quality: Optional[int] = 100,
|
||
mode: Optional[str] = 'RGB'
|
||
) -> MessageSegment:
|
||
if isinstance(img, str) or isinstance(img, Path):
|
||
img = load_image(path=img, size=size, mode=mode, crop=crop)
|
||
bio = BytesIO()
|
||
img.save(bio, format='JPEG' if mode == 'RGB' else 'PNG', quality=quality)
|
||
img_b64 = 'base64://' + base64.b64encode(bio.getvalue()).decode()
|
||
return MessageSegment.image(img_b64)
|
||
|
||
@classmethod
|
||
def Text(cls, text: str) -> MessageSegment:
|
||
# TODO 过滤负面文本
|
||
return MessageSegment.text(text)
|
||
|
||
@classmethod
|
||
def Record(cls, path: str) -> MessageSegment:
|
||
# TODO 网络语音
|
||
return MessageSegment.record(path)
|
||
|
||
|
||
async def get_at_target(msg):
|
||
for msg_seg in msg:
|
||
if msg_seg.type == "at":
|
||
return msg_seg.data['qq']
|
||
return None
|
||
|
||
|
||
# message预处理，获取uid、干净的msg、user_id、是否缓存
|
||
async def get_uid_in_msg(event: MessageEvent, msg: Message):
|
||
msg = str(msg).strip()
|
||
if not msg:
|
||
uid = await get_last_query(str(event.user_id))
|
||
return uid, '', str(event.user_id), True
|
||
user_id = await get_at_target(event.message) or str(event.user_id)
|
||
msg = re.sub(r'\[CQ.*?\]', '', msg)
|
||
use_cache = False if '-r' in msg else True
|
||
msg = msg.replace('-r', '').strip()
|
||
find_uid = r'(?P<uid>(1|2|5)\d{8})'
|
||
for msg_seg in event.message:
|
||
if msg_seg.type == 'text':
|
||
match = re.search(find_uid, msg_seg.data['text'])
|
||
if match:
|
||
await update_last_query(user_id, match.group('uid'), 'uid')
|
||
return match.group('uid'), msg.replace(match.group('uid'), '').strip(), user_id, use_cache
|
||
uid = await get_last_query(user_id)
|
||
return uid, msg.strip(), user_id, use_cache
|
||
|
||
|
||
# 向超级用户私聊发送cookie删除信息
|
||
async def send_cookie_delete_msg(cookie_info):
|
||
msg = ''
|
||
if cookie_info['type'] == 'public':
|
||
msg = f'公共池的{cookie_info["no"]}号cookie已失效'
|
||
elif cookie_info['type'] == 'private':
|
||
if cookie_info['uid']:
|
||
msg = f'用户{cookie_info["user_id"]}的uid{cookie_info["uid"]}的cookie已失效'
|
||
elif cookie_info['mys_id']:
|
||
msg = f'用户{cookie_info["user_id"]}的mys_id{cookie_info["mys_id"]}的cookie已失效'
|
||
if msg:
|
||
logger.info(f'---{msg}---')
|
||
for superuser in get_bot().config.superusers:
|
||
try:
|
||
await get_bot().send_private_msg(user_id=superuser, message=msg + '，派蒙帮你删除啦!')
|
||
except Exception as e:
|
||
logger.error(f'发送cookie删除消息失败: {e}')
|
||
|
||
|
||
def get_message_id(event):
|
||
if event.message_type == 'private':
|
||
return event.user_id
|
||
elif event.message_type == 'group':
|
||
return event.group_id
|
||
elif event.message_type == 'guild':
|
||
return event.channel_id
|