重写帮助，删除 详细帮助 命令

basic_plugins/help/utils.py

@@ -1,28 +1,15 @@
-from typing import List, Tuple, Dict, Optional, Union
+from typing import List, Tuple, Dict, Optional
 from configs.path_config import IMAGE_PATH
 from utils.image_utils import BuildImage
-from pydantic import BaseModel
+from configs.config import Config
 import os
 import random
 
+from utils.manager import plugin_data_manager, group_manager
+from utils.manager.models import PluginData, PluginType
+
 background_path = IMAGE_PATH / "background" / "help" / "simple_help"
 
-# class PluginType(BaseModel):
-#
-#
-# class Plugin(BaseModel):
-#     name: str
-#     plugin_type: PluginType                # 插件内部类型，根据name [Hidden] [Admin] [SUPERUSER]
-#     usage: Optional[str]
-#     des: Optional[str]
-#     cmd: Optional[List[str]]
-#     task: Optional[Dict[str, str]]
-#     type: Optional[Tuple[str, int]]                # 菜单类型
-#     version: Optional[Union[str, int]]
-#     author: Optional[str]
-
-
-
 
 async def build_help_image(image_group: List[List[BuildImage]], h: int):
     bk = None
@@ -36,6 +23,7 @@ async def build_help_image(image_group: List[List[BuildImage]], h: int):
         color="#FFEFD5",
         background=(background_path / bk) if bk else None,
     )
+    A.filter("GaussianBlur", 5)
     curr_w = 50
     for ig in image_group:
         curr_h = 180
@@ -121,8 +109,145 @@ def group_image(image_list: List[BuildImage]) -> Tuple[List[List[BuildImage]], i
     return image_group, max(max_h + 250, max_w + 70)
 
 
-# class HelpImageBuild:
-#
-#     def __init__(self):
-#         self._data: Dict[str, List[]] = {}
+class HelpImageBuild:
 
+    build: Optional["HelpImageBuild"] = None
+
+    def __init__(self):
+        self._data: Dict[str, PluginData] = plugin_data_manager.get_data()
+        self._sort_data: Dict[str, List[PluginData]] = {}
+        self._image_list = []
+
+    def __new__(cls, *args, **kwargs):
+        if not cls.build:
+            cls.build = super().__new__(cls)
+        return cls.build
+
+    def sort_type(self):
+        """
+        说明:
+            对插件按照菜单类型分类
+        """
+        for key in self._data.keys():
+            plugin_data = self._data[key]
+            if plugin_data.plugin_type == PluginType.NORMAL:
+                if not self._sort_data.get(plugin_data.menu_type[0]):
+                    self._sort_data[plugin_data.menu_type[0]] = []
+                self._sort_data[plugin_data.menu_type[0]].append(self._data[key])
+
+    async def build_name_image(
+        self, max_width: int, name: str, color: str, text_color: Tuple[int, int, int], pos: Optional[Tuple[int, int, int, int]]
+    ) -> BuildImage:
+        image = BuildImage(max_width - 5, 50, color=color, font_size=24)
+        await image.acircle_corner()
+        await image.atext((0, 0), name, text_color, center_type="center")
+        return image
+
+    async def build_image(self, group_id: Optional[int]) -> BuildImage:
+        """
+        说明:
+            构造帮助图片
+        参数:
+            :param group_id: 群号
+        """
+        self._image_list = []
+        self.sort_type()
+        font_size = 24
+        build_type = Config.get_config("help", "TYPE")
+        _image = BuildImage(0, 0, plain_text="1", font_size=font_size)
+        for idx, menu_type in enumerate(self._sort_data.keys()):
+            plugin_list = self._sort_data[menu_type]
+            wh_list = [_image.getsize(x.name) for x in plugin_list]
+            wh_list.append(_image.getsize(menu_type))
+            # sum_height = sum([x[1] for x in wh_list])
+            if build_type == 'VV':
+                sum_height = 50 * len(plugin_list) + 10
+            else:
+                sum_height = (font_size + 6) * len(plugin_list) + 10
+            max_width = max([x[0] for x in wh_list]) + 20
+            bk = BuildImage(
+                max_width + 40,
+                sum_height + 50,
+                font_size=30,
+                color="#a7d1fc",
+                font="CJGaoDeGuo.otf",
+            )
+            title_size = bk.getsize(menu_type)
+            max_width = max_width if max_width > title_size[0] else title_size[0]
+            B = BuildImage(
+                max_width + 40,
+                sum_height,
+                font_size=font_size,
+                color="white" if not idx % 2 else "black",
+            )
+            curr_h = 10
+            for i, plugin_data in enumerate(plugin_list):
+                text_color = (255, 255, 255) if idx % 2 else (0, 0, 0)
+                if group_id and not group_manager.get_plugin_status(
+                    plugin_data.model, group_id
+                ):
+                    text_color = (252, 75, 13)
+                pos = None
+                # 禁用状态划线
+                if (
+                    not plugin_data.plugin_status.status
+                    and plugin_data.plugin_status.block_type in ["group", "all"]
+                ):
+                    w = curr_h + int(B.getsize(plugin_data.name)[1] / 2) + 2
+                    pos = (
+                        7,
+                        w,
+                        B.getsize(plugin_data.name)[0] + 35,
+                        w,
+                    )
+                if build_type == "VV":
+                    name_image = await self.build_name_image(max_width, plugin_data.name, "black" if not idx % 2 else "white", text_color, pos)
+                    await B.apaste(name_image, (0, curr_h), True, center_type="by_width")
+                    curr_h += name_image.h + 5
+                else:
+                    await B.atext(
+                        (10, curr_h), f"{i + 1}.{plugin_data.name}", text_color
+                    )
+                    if pos:
+                        await B.aline(pos, (236, 66, 7), 3)
+                    curr_h += font_size + 5
+            if menu_type == "normal":
+                menu_type = "功能"
+            await bk.atext((0, 14), menu_type, center_type="by_width")
+            await bk.apaste(B, (0, 50))
+            await bk.atransparent(2)
+            # await bk.acircle_corner(point_list=['lt', 'rt'])
+            self._image_list.append(bk)
+        image_group, h = group_image(self._image_list)
+        B = await build_help_image(image_group, h)
+        w = 10
+        h = 10
+        for msg in [
+            "目前支持的功能列表:",
+            "可以通过 ‘帮助[功能名称]’ 来获取对应功能的使用方法",
+            "或者使用 ‘详细帮助’ 来获取所有功能方法",
+        ]:
+            text = BuildImage(
+                0,
+                0,
+                plain_text=msg,
+                font_size=24,
+                font="HYWenHei-85W.ttf",
+            )
+            B.paste(text, (w, h), True)
+            h += 50
+            if msg == "目前支持的功能列表:":
+                w += 50
+        await B.apaste(
+            BuildImage(
+                0,
+                0,
+                plain_text="注: 红字代表功能被群管理员禁用，红线代表功能正在维护",
+                font_size=24,
+                font="HYWenHei-85W.ttf",
+                font_color=(231, 74, 57),
+            ),
+            (300, 10),
+            True,
+        )
+        return B