✨ SuperuserShortcutExt

Author: RF-Tar-Railt

example/bot.py

@@ -13,8 +13,8 @@
 # driver.register_adapter(SatoriAdapter)
 
 # nonebot.require("nonebot_plugin_alconna")
-nonebot.load_plugins("plugins")
-# nonebot.load_plugin("plugins.demo1")
+# nonebot.load_plugins("plugins")
+nonebot.load_plugin("plugins.demo1")
 
 
 async def _():

example/plugins/demo.py

@@ -25,6 +25,7 @@
 
 from nonebot_plugin_waiter import waiter
 
+from nonebot_plugin_alconna.builtins.extensions.shortcut import SuperUserShortcutExtension
 from nonebot_plugin_alconna import (
     At,
     Check,
@@ -64,7 +65,7 @@ async def output_converter(self, output_type, content: str):
         return UniMessage(content)
 
 
-add_global_extension(DemoExtension())
+add_global_extension(DemoExtension(), SuperUserShortcutExtension())
 
 
 def get_dist_map() -> dict:

src/nonebot_plugin_alconna/builtins/extensions/shortcut.py

@@ -0,0 +1,37 @@
+from __future__ import annotations
+
+from arclet.alconna import Alconna
+from nonebot.permission import SuperUser
+from nonebot.internal.adapter import Bot, Event
+
+from nonebot_plugin_alconna import Extension, UniMessage
+
+
+class SuperUserShortcutExtension(Extension):
+    """
+    用于设置仅超级用户可使用内置选项 `--shortcut` 的扩展。
+
+    Example:
+        >>> from nonebot_plugin_alconna.builtins.extensions.shortcut import SuperUserShortcutExtension
+        >>>
+        >>> matcher = on_alconna("...", extensions=[SuperUserShortcutExtension()])
+    """
+
+    @property
+    def priority(self) -> int:
+        return 20
+
+    @property
+    def id(self) -> str:
+        return "builtins.extensions.shortcut:SuperUserShortcutExtension"
+
+    async def receive_wrapper(self, bot: Bot, event: Event, command: Alconna, receive: UniMessage) -> UniMessage:
+        su = SuperUser()
+        if await su(bot, event):
+            command.namespace_config.disable_builtin_options.discard("shortcut")
+        else:
+            command.namespace_config.disable_builtin_options.add("shortcut")
+        return receive
+
+
+__extension__ = SuperUserShortcutExtension