docs: 添加了命令手册

liteyuki/liteyuki_main/core.py

@@ -1,8 +1,12 @@
 from nonebot import require
 from nonebot.permission import SUPERUSER
+from git import Repo
 
 from liteyuki.utils.config import config
-from liteyuki.utils.ly_typing import T_Bot
+from liteyuki.utils.ly_typing import T_Bot, T_MessageEvent
+
+from .reloader import Reloader
+from ..utils.message import send_markdown
 
 require("nonebot_plugin_alconna")
 from nonebot_plugin_alconna import on_alconna, Alconna
@@ -14,7 +18,46 @@ cmd_liteyuki = on_alconna(
     permission=SUPERUSER
 )
 
+update_liteyuki = on_alconna(
+    Alconna(
+        ["update-liteyuki", "更新轻雪"]
+    ),
+    permission=SUPERUSER
+)
+
+reload_liteyuki = on_alconna(
+    Alconna(
+        ["reload-liteyuki", "重启轻雪"]
+    ),
+    permission=SUPERUSER
+)
+
 
 @cmd_liteyuki.handle()
 async def _(bot: T_Bot):
     await cmd_liteyuki.finish(f"Hello, Liteyuki!\nBot {bot.self_id}\nLiteyukiID {config.get('liteyuki_id', 'No')}")
+
+
+@update_liteyuki.handle()
+async def _(bot: T_Bot, event: T_MessageEvent):
+    # 使用git pull更新
+    origins = ["origin", "origin2"]
+    repo = Repo(".")
+    for origin in origins:
+        try:
+            repo.remotes[origin].pull()
+            break
+        except Exception as e:
+            print(f"Pull from {origin} failed: {e}")
+    logs = repo.git.log('--pretty=format:%H %s')
+    reply = "Liteyuki updated!\n"
+    reply += f"```\n{logs}\n```"
+    print(list(repo.iter_commits()))
+
+    await send_markdown(reply, bot, event=event, at_sender=False)
+
+
+@reload_liteyuki.handle()
+async def _():
+    await reload_liteyuki.send("Liteyuki reloading")
+    Reloader.reload(3)

liteyuki/liteyuki_main/reloader.py

@@ -0,0 +1,67 @@
+import threading
+from multiprocessing import get_context
+
+import nonebot
+from nonebot import logger
+from typing import List, Optional
+
+from nonebot import get_driver
+from pydantic import BaseSettings
+
+
+reboot_grace_time_limit: int = 20
+
+_nb_run = nonebot.run
+
+
+class Reloader:
+    event: threading.Event = None
+
+    @classmethod
+    def reload(cls, delay: int = 0):
+        if cls.event is None:
+            raise RuntimeError()
+        if delay > 0:
+            threading.Timer(delay, function=cls.event.set).start()
+            return
+        cls.event.set()
+
+
+def _run(ev: threading.Event, *args, **kwargs):
+    Reloader.event = ev
+    _nb_run(*args, **kwargs)
+
+
+def run(*args, **kwargs):
+    should_exit = False
+    ctx = get_context("spawn")
+    while not should_exit:
+        event = ctx.Event()
+        process = ctx.Process(
+            target=_run,
+            args=(
+                    event,
+                    *args,
+            ),
+            kwargs=kwargs,
+        )
+        process.start()
+        while not should_exit:
+            if event.wait(1):
+                logger.info("Receive reboot event")
+                process.terminate()
+                process.join(reboot_grace_time_limit)
+                if process.is_alive():
+                    logger.warning(
+                        f"Cannot shutdown gracefully in {reboot_grace_time_limit} second, force kill process."
+                    )
+                    process.kill()
+                break
+            elif process.is_alive():
+                continue
+            else:
+                # Process stoped without setting event
+                should_exit = True
+
+
+nonebot.run = run

liteyuki/plugins/liteyuki_npm/manager.py

@@ -29,22 +29,14 @@ list_plugins = on_alconna(
 
 toggle_plugin = on_alconna(
     Alconna(
-        ["enable", "disable"],
+        ["enable", "disable", "启用", "停用"],
         Args["plugin_name", str],
     )
 )
 
 toggle_plugin_global = on_alconna(
     Alconna(
-        ["enable-global", "disable-global"],
-        Args["plugin_name", str],
-    ),
-    permission=SUPERUSER
-)
-
-global_toggle = on_alconna(
-    Alconna(
-        ["toggle-global"],
+        ["enable-global", "disable-global", "全局启用", "全局停用"],
         Args["plugin_name", str],
     ),
     permission=SUPERUSER
@@ -126,7 +118,7 @@ async def _(result: Arparma, event: T_MessageEvent, bot: T_Bot):
     ulang = get_user_lang(str(event.user_id))
     plugin_module_name = result.args.get("plugin_name")
 
-    toggle = result.header_result == "enable-plugin"  # 判断是启用还是停用
+    toggle = result.header_result in ["enable-plugin", "启用"]  # 判断是启用还是停用
 
     session_enable = get_plugin_session_enable(event, plugin_module_name)  # 获取插件当前状态
 
@@ -189,7 +181,7 @@ async def _(result: Arparma, event: T_MessageEvent, bot: T_Bot):
     ulang = get_user_lang(str(event.user_id))
     plugin_module_name = result.args.get("plugin_name")
 
-    toggle = result.header_result == "enable-global"
+    toggle = result.header_result in ["enable-global", "全局启用"]
     can_be_toggled = get_plugin_can_be_toggle(plugin_module_name)
     if not can_be_toggled:
         await toggle_plugin_global.finish(ulang.get("npm.plugin_cannot_be_toggled", NAME=plugin_module_name))
@@ -238,6 +230,4 @@ async def pre_handle(event: Event, matcher: Matcher):
 
 @Bot.on_calling_api
 async def _(bot: Bot, api: str, data: dict[str, any]):
-    # TODO 插件启用/停用检查hook
-    plugin = current_plugin
     nonebot.logger.info(f"Plugin Callapi: {api}: {data}")