feat✨: 更新内置插件

zhenxun/builtin_plugins/hooks/withdraw_hook.py

@@ -0,0 +1,46 @@
+import asyncio
+from typing import Optional
+
+from nonebot.adapters import Bot
+from nonebot.adapters.discord import Bot as DiscordBot
+from nonebot.adapters.dodo import Bot as DodoBot
+from nonebot.adapters.kaiheila import Bot as KaiheilaBot
+from nonebot.adapters.onebot.v11 import Bot as v11Bot
+from nonebot.adapters.onebot.v12 import Bot as v12Bot
+from nonebot.matcher import Matcher
+from nonebot.message import run_postprocessor
+
+from zhenxun.services.log import logger
+from zhenxun.utils.utils import WithdrawManager
+
+# TODO: 其他平台撤回消息
+
+
+# 消息撤回
+@run_postprocessor
+async def _(
+    matcher: Matcher,
+    exception: Optional[Exception],
+    bot: Bot,
+):
+    tasks = []
+    for message_id in WithdrawManager._data:
+        second = WithdrawManager._data[message_id]
+        tasks.append(asyncio.ensure_future(_withdraw_message(bot, message_id, second)))
+        WithdrawManager.remove(message_id)
+    await asyncio.gather(*tasks)
+
+
+async def _withdraw_message(bot: Bot, message_id: str, time: int):
+    await asyncio.sleep(time)
+    logger.debug(f"撤回消息ID: {message_id}", "HOOK")
+    if isinstance(bot, v11Bot):
+        await bot.delete_msg(message_id=int(message_id))
+    elif isinstance(bot, v12Bot):
+        await bot.delete_message(message_id=message_id)
+    elif isinstance(bot, DodoBot):
+        pass
+    elif isinstance(bot, KaiheilaBot):
+        pass
+    elif isinstance(bot, DiscordBot):
+        pass