✨ 重构帮助，限制普通用户查询管理插件 (#1626)

2026-10-05 20:00:04 +08:00 · 2024-09-14 05:23:55 +08:00
parent d97e437e80
commit 029a731fb9
9 changed files with 491 additions and 367 deletions
@@ -0,0 +1,136 @@
+import os
+import random
+
+from pydantic import BaseModel
+from nonebot_plugin_htmlrender import template_to_pic
+
+from zhenxun.utils.enum import BlockType
+from zhenxun.models.plugin_info import PluginInfo
+from zhenxun.configs.path_config import TEMPLATE_PATH
+from zhenxun.models.group_console import GroupConsole
+
+from ._utils import classify_plugin
+
+LOGO_PATH = TEMPLATE_PATH / "menu" / "res" / "logo"
+
+
+class Item(BaseModel):
+    plugin_name: str
+    """插件名称"""
+    sta: int
+    """插件状态"""
+
+
+class PluginList(BaseModel):
+    plugin_type: str
+    """菜单名称"""
+    icon: str
+    """图标"""
+    logo: str
+    """logo"""
+    items: list[Item]
+    """插件列表"""
+
+
+ICON2STR = {
+    "normal": "fa fa-cog",
+    "原神相关": "fa fa-circle-o",
+    "常规插件": "fa fa-cubes",
+    "联系管理员": "fa fa-envelope-o",
+    "抽卡相关": "fa fa-credit-card-alt",
+    "来点好康的": "fa fa-picture-o",
+    "数据统计": "fa fa-bar-chart",
+    "一些工具": "fa fa-shopping-cart",
+    "商店": "fa fa-shopping-cart",
+    "其它": "fa fa-tags",
+    "群内小游戏": "fa fa-gamepad",
+}
+
+
+def __handle_item(plugin: PluginInfo, group: GroupConsole | None) -> Item:
+    """构造Item
+
+    参数:
+        plugin: PluginInfo
+        group: 群组
+
+    返回:
+        Item: Item
+    """
+    sta = 0
+    if not plugin.status:
+        if group and plugin.block_type in [
+            BlockType.ALL,
+            BlockType.GROUP,
+        ]:
+            sta = 2
+        if not group and plugin.block_type in [
+            BlockType.ALL,
+            BlockType.PRIVATE,
+        ]:
+            sta = 2
+    if group:
+        if f"{plugin.module}:super," in group.block_plugin:
+            sta = 2
+        if f"{plugin.module}," in group.block_plugin:
+            sta = 1
+    return Item(plugin_name=plugin.name, sta=sta)
+
+
+def build_plugin_data(classify: dict[str, list[Item]]) -> list[dict[str, str]]:
+    """构建前端插件数据
+
+    参数:
+        classify: 插件数据
+
+    返回:
+        list[dict[str, str]]: 前端插件数据
+    """
+    lengths = [len(classify[c]) for c in classify]
+    index = lengths.index(max(lengths))
+    menu_key = list(classify.keys())[index]
+    max_data = classify[menu_key]
+    del classify[menu_key]
+    plugin_list = []
+    for menu_type in classify:
+        icon = "fa fa-pencil-square-o"
+        if menu_type in ICON2STR.keys():
+            icon = ICON2STR[menu_type]
+        logo = LOGO_PATH / random.choice(os.listdir(LOGO_PATH))
+        data = {
+            "name": menu_type if menu_type != "normal" else "功能",
+            "items": classify[menu_type],
+            "icon": icon,
+            "logo": str(logo.absolute()),
+        }
+        plugin_list.append(data)
+    plugin_list.insert(
+        0,
+        {
+            "name": menu_key if menu_key != "normal" else "功能",
+            "items": max_data,
+            "icon": "fa fa-pencil-square-o",
+            "logo": str((LOGO_PATH / random.choice(os.listdir(LOGO_PATH))).absolute()),
+        },
+    )
+    return plugin_list
+
+
+async def build_html_image(group_id: str | None) -> bytes:
+    """构造HTML帮助图片
+
+    参数:
+        group_id: 群号
+    """
+    classify = await classify_plugin(group_id, __handle_item)
+    plugin_list = build_plugin_data(classify)
+    return await template_to_pic(
+        template_path=str((TEMPLATE_PATH / "menu").absolute()),
+        template_name="zhenxun_menu.html",
+        templates={"plugin_list": plugin_list},
+        pages={
+            "viewport": {"width": 1903, "height": 975},
+            "base_url": f"file://{TEMPLATE_PATH}",
+        },
+        wait=2,
+    )