mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 00:32:06 +08:00
81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
from nonebot.adapters import Bot, Event
|
|
from nonebot.adapters.onebot.v11 import Bot as OneBotV11Bot
|
|
from nonebot.adapters.onebot.v12 import Bot as OneBotV12Bot
|
|
from nonebot.plugin import PluginMetadata
|
|
from nonebot.rule import Rule
|
|
from nonebot_plugin_alconna import Alconna, Arparma, on_alconna
|
|
from nonebot_plugin_alconna.uniseg.tools import reply_fetch
|
|
from nonebot_plugin_uninfo import Uninfo
|
|
|
|
from zhenxun.configs.utils import Command, PluginExtraData
|
|
from zhenxun.services.log import logger
|
|
from zhenxun.utils.manager.message_manager import MessageManager
|
|
from zhenxun.utils.message import MessageUtils
|
|
from zhenxun.utils.platform import PlatformUtils
|
|
|
|
__plugin_meta__ = PluginMetadata(
|
|
name="消息撤回",
|
|
description="撤回自己触发的消息撤回，不允许撤回其他人触发消息的撤回哦",
|
|
usage="""
|
|
引用消息 撤回
|
|
""".strip(),
|
|
extra=PluginExtraData(
|
|
author="HibiKier",
|
|
version="0.1",
|
|
menu_type="其他",
|
|
commands=[Command(command="[引用消息] 撤回")],
|
|
).to_dict(),
|
|
)
|
|
|
|
|
|
def reply_check() -> Rule:
|
|
"""
|
|
检查是否存在回复消息
|
|
|
|
返回:
|
|
Rule: Rule
|
|
"""
|
|
|
|
async def _rule(bot: Bot, event: Event, session: Uninfo):
|
|
if event.get_type() == "message":
|
|
return (
|
|
bool(await reply_fetch(event, bot))
|
|
and PlatformUtils.get_platform(session) == "qq"
|
|
)
|
|
return False
|
|
|
|
return Rule(_rule)
|
|
|
|
|
|
_matcher = on_alconna(Alconna("撤回"), priority=5, block=True, rule=reply_check())
|
|
|
|
|
|
def _resolve_delete_message_id(bot: Bot, reply_id: str | int):
|
|
if isinstance(bot, OneBotV11Bot):
|
|
return int(reply_id)
|
|
if isinstance(bot, OneBotV12Bot):
|
|
return str(reply_id)
|
|
return reply_id
|
|
|
|
|
|
@_matcher.handle()
|
|
async def _(bot: Bot, event: Event, session: Uninfo, arparma: Arparma):
|
|
if reply := await reply_fetch(event, bot):
|
|
message_id = _resolve_delete_message_id(bot, reply.id)
|
|
if session.user.id in bot.config.superusers:
|
|
try:
|
|
await bot.delete_msg(message_id=message_id)
|
|
logger.info("撤回消息", arparma.header_result, session=session)
|
|
except Exception:
|
|
await MessageUtils.build_message("撤回失败，可能消息已过期...").send()
|
|
elif MessageManager.check(session.user.id, reply.id):
|
|
try:
|
|
await bot.delete_msg(message_id=message_id)
|
|
logger.info("撤回消息", arparma.header_result, session=session)
|
|
except Exception:
|
|
await MessageUtils.build_message("撤回失败，可能消息已过期...").send()
|
|
else:
|
|
await MessageUtils.build_message(
|
|
"权限不足，不是你触发的消息不要胡乱撤回哦..."
|
|
).send()
|