update v0.0.5.0

plugins/help/data_source.py

@@ -1,175 +1,289 @@
 from utils.image_utils import CreateImg
-from configs.path_config import IMAGE_PATH, DATA_PATH
+from configs.path_config import IMAGE_PATH
+from utils.manager import plugins2settings_manager
+from typing import Optional
+from services.log import logger
 from pathlib import Path
-from .config import *
-from configs.config import (
-    INITIAL_OPEN_CASE_COUNT,
-    INITIAL_SETU_PROBABILITY,
-    ADMIN_DEFAULT_AUTH,
-)
-from configs.config import plugins2info_dict, NICKNAME
-from utils.static_data import group_manager
+from utils.utils import get_matchers
+import random
+import asyncio
 import nonebot
+import os
 
 
-width = 1600
+random_bk_path = Path(IMAGE_PATH) / "background" / "help" / "simple_help"
+
+background = Path(IMAGE_PATH) / "background" / "0.png"
 
 
-def create_help_img():
-    help_img_file = Path(IMAGE_PATH) / 'help.png'
-    if help_img_file.exists():
-        help_img_file.unlink()
-    h = (
-        100
-        + len(utility_help) * 24
-        + len(entertainment_help) * 24
-        + len(other_help) * 24
-    ) + 800
-    A = CreateImg(width, h - 200, font_size=24)
-    e = CreateImg(width, len(entertainment_help) * 26 + 100, font_size=24)
-    rst = ""
-    i = 0
-    for cmd in entertainment_help:
-        rst += f"{i + 1}.{entertainment_help[cmd]}\n"
-        i += 1
-    e.text((10, 10), "娱乐功能：")
-    e.text((40, 40), rst)
-    u = CreateImg(width, len(utility_help) * 26 + 100, font_size=24, color="black")
-    rst = ""
-    i = 0
-    for cmd in utility_help:
-        rst += f"{i + 1}.{utility_help[cmd]}\n"
-        i += 1
-    u.text((10, 10), "实用功能：", fill=(255, 255, 255))
-    u.text((40, 40), rst, fill=(255, 255, 255))
-    o = CreateImg(width, len(other_help) * 26 + 100, font_size=24)
-    rst = ""
-    for i in range(len(other_help)):
-        rst += f"{i + 1}.{other_help[i]}\n"
-        i += 1
-    o.text((10, 10), "其他功能：")
-    o.text((40, 40), rst)
-    A.paste(e, (0, 0))
-    A.paste(u, (0, e.h))
-    A.paste(o, (0, e.h + u.h))
-    A.text(
-        (10, e.h + u.h + o.h + 50),
-        f"大部分交互功能可以通过输入‘取消’，‘算了’来取消当前交互\n对{NICKNAME}说 “{NICKNAME}帮助 指令名” 获取对应详细帮助\n"
-        "可以通过 “滴滴滴- [消息]” 联系管理员（有趣的想法尽管来吧！<还有Bug和建议>）"
-        "\n[群管理员请看 管理员帮助（群主与管理员自带 5 级权限）]\n\n"
-        f"\t「如果{NICKNAME}回复了一些不符合人设的话，那是因为每日白嫖的图灵次数已用完，使用的是备用接口【QAQ】」",
+async def create_help_img(help_image: Path, simple_help_image: Path):
+    """
+    生成帮助图片
+    :param help_image: 图片路径
+    :param simple_help_image: 简易帮助图片路径
+    """
+    return await asyncio.get_event_loop().run_in_executor(
+        None, _create_help_img, help_image, simple_help_image
     )
 
-    A.save(IMAGE_PATH + "help.png")
 
-
-def create_group_help_img(group_id: int):
-    group_id = str(group_id)
-    h = (
-        100
-        + len(utility_help) * 24
-        + len(entertainment_help) * 24
-        + len(other_help) * 24
-    ) + 800
-    A = CreateImg(width, h, font_size=24)
-    u = CreateImg(width, len(utility_help) * 26 + 100, font_size=24, color="black")
-    o = CreateImg(width, len(other_help) * 26 + 100, font_size=24)
-    e = CreateImg(width, len(entertainment_help) * 26 + 100, font_size=24)
-    rst = ""
-    i = 1
-    for cmd in entertainment_help.keys():
-        flag, dfg = parse_cmd(cmd, group_id)
-        if dfg:
-            cmd = rcmd(dfg)
-        rst += f"【{flag}】{i}.{entertainment_help[cmd]}\n"
-        i += 1
-    e.text((10, 10), "娱乐功能：")
-    e.text((40, 40), rst)
-
-    rst = ""
-    i = 1
-    for cmd in utility_help.keys():
-        flag, dfg = parse_cmd(cmd, group_id)
-        rst += f"【{flag}】{i}.{utility_help[cmd]}\n"
-        i += 1
-    u.text((10, 10), "实用功能：", fill=(255, 255, 255))
-    u.text((40, 40), rst, fill=(255, 255, 255))
-
-    rst = ""
-    for i in range(len(other_help)):
-        rst += f"{i + 1}.{other_help[i]}\n"
-    o.text((10, 10), "其他功能：")
-    o.text((40, 40), rst)
-
-    A.paste(e, (0, 0))
-    A.paste(u, (0, e.h))
-    A.paste(o, (0, e.h + u.h))
-    # A.text((width, 10), f'总开关【{"√" if data["总开关"] else "×"}】')
-    A.text(
-        (10, e.h + u.h + o.h + 50),
-        f"大部分交互功能可以通过输入‘取消’，‘算了’来取消当前交互\n对{NICKNAME}说 “{NICKNAME}帮助 指令名” 获取对应详细帮助\n"
-        "可以通过 “滴滴滴- [消息]” 联系管理员（有趣的想法尽管来吧！<还有Bug和建议>）"
-        f"\n[群管理员请看 管理员帮助（群主与管理员自带 {ADMIN_DEFAULT_AUTH} 级权限）]",
+def _create_help_img(help_image: Path, simple_help_image: Path):
+    """
+    生成帮助图片
+    :param help_image: 图片路径
+    :param simple_help_image: 简易帮助图片路径
+    """
+    _matchers = get_matchers()
+    width = 0
+    matchers_data = {}
+    _des_tmp = {}
+    _plugin_name_tmp = []
+    _tmp = []
+    tmp_img = CreateImg(0, 0, plain_text="1", font_size=24)
+    font_height = tmp_img.h
+    for matcher in _matchers:
+        plugin_name = None
+        _plugin = nonebot.plugin.get_plugin(matcher.module)
+        _module = _plugin.module
+        try:
+            plugin_name = _module.__getattribute__("__zx_plugin_name__")
+            try:
+                plugin_des = _module.__getattribute__("__plugin_des__")
+            except AttributeError:
+                plugin_des = "_"
+            if (
+                "[hidden]" in plugin_name.lower()
+                or "[admin]" in plugin_name.lower()
+                or "[superuser]" in plugin_name.lower()
+                or plugin_name in _plugin_name_tmp
+                or plugin_name == "帮助"
+            ):
+                continue
+            plugin_type = ("normal",)
+            text_type = 0
+            if plugins2settings_manager.get(
+                matcher.module
+            ) and plugins2settings_manager[matcher.module].get("plugin_type"):
+                plugin_type = tuple(
+                    plugins2settings_manager.get_plugin_data(matcher.module)[
+                        "plugin_type"
+                    ]
+                )
+            else:
+                try:
+                    plugin_type = _module.__getattribute__("__plugin_type__")
+                except AttributeError:
+                    pass
+            if len(plugin_type) > 1:
+                try:
+                    text_type = int(plugin_type[1])
+                except ValueError as e:
+                    logger.warning(f"生成列向帮助排列失败 {plugin_name}: {type(e)}: {e}")
+                plugin_type = plugin_type[0]
+            else:
+                plugin_type = plugin_type[0]
+            try:
+                plugin_cmd = _module.__getattribute__("__plugin_cmd__")
+                plugin_cmd = [x for x in plugin_cmd if "[_superuser]" not in x]
+            except AttributeError:
+                plugin_cmd = []
+            if plugin_type not in matchers_data.keys():
+                matchers_data[plugin_type] = {}
+            if plugin_des in _des_tmp.keys():
+                try:
+                    matchers_data[plugin_type][_des_tmp[plugin_des]]["cmd"] = (
+                        matchers_data[plugin_type][_des_tmp[plugin_des]]["cmd"]
+                        + plugin_cmd
+                    )
+                except KeyError as e:
+                    logger.warning(f"{type(e)}: {e}")
+            else:
+                matchers_data[plugin_type][plugin_name] = {
+                    "des": plugin_des,
+                    "cmd": plugin_cmd,
+                    "text_type": text_type,
+                }
+            try:
+                if text_type == 0:
+                    x = tmp_img.getsize(
+                        f'{plugin_name}: {matchers_data[plugin_type][plugin_name]["des"]} ->'
+                        + " / ".join(matchers_data[plugin_type][plugin_name]["cmd"])
+                    )[0]
+                    width = width if width > x else x
+            except KeyError:
+                pass
+            if plugin_des not in _des_tmp:
+                _des_tmp[plugin_des] = plugin_name
+        except AttributeError as e:
+            if plugin_name not in _plugin_name_tmp:
+                logger.warning(f"获取功能 {matcher.module}: {plugin_name} 设置失败...e：{e}")
+        if plugin_name not in _plugin_name_tmp:
+            _plugin_name_tmp.append(plugin_name)
+    help_img_list = []
+    simple_help_img_list = []
+    types = list(matchers_data.keys())
+    types.sort()
+    ix = 0
+    for type_ in types:
+        keys = list(matchers_data[type_].keys())
+        keys.sort()
+        help_str = f"{type_ if type_ != 'normal' else '功能'}:\n\n"
+        simple_help_str = f"{type_ if type_ != 'normal' else '功能'}:\n\n"
+        for i, k in enumerate(keys):
+            simple_help_str += f"{i+1}.{k}\n"
+            if matchers_data[type_][k]["text_type"] == 1:
+                _x = tmp_img.getsize(
+                    f"{i+1}".rjust(5)
+                    + f'.{k}: {matchers_data[type_][k]["des"]} {"->" if matchers_data[type_][k]["cmd"] else ""} '
+                )[0]
+                _str = (
+                    f"{i+1}".rjust(5)
+                    + f'.{k}: {matchers_data[type_][k]["des"]} {"->" if matchers_data[type_][k]["cmd"] else ""} '
+                )
+                _str += matchers_data[type_][k]["cmd"][0] + "\n"
+                for c in matchers_data[type_][k]["cmd"][1:]:
+                    _str += "".rjust(int(_x * 0.125) + 1) + f"{c}\n"
+                help_str += _str
+            else:
+                help_str += (
+                    f"{i+1}".rjust(5)
+                    + f'.{k}: {matchers_data[type_][k]["des"]} {"->" if matchers_data[type_][k]["cmd"] else ""} '
+                    + " / ".join(matchers_data[type_][k]["cmd"])
+                    + "\n"
+                )
+        height = len(help_str.split("\n")) * (font_height + 5)
+        simple_height = len(simple_help_str.split("\n")) * (font_height + 5)
+        A = CreateImg(
+            width + 150, height, font_size=24, color="white" if not ix % 2 else "black"
+        )
+        A.text((10, 10), help_str, (255, 255, 255) if ix % 2 else (0, 0, 0))
+        simple_width = 0
+        for x in [tmp_img.getsize(x)[0] for x in simple_help_str.split("\n")]:
+            simple_width = simple_width if simple_width > x else x
+        bk = CreateImg(
+            simple_width + 20, simple_height, font_size=24, color="#6495ED"
+        )
+        B = CreateImg(
+            simple_width + 20,
+            simple_height,
+            font_size=24,
+            color="white" if not ix % 2 else "black",
+        )
+        B.text((10, 10), simple_help_str, (255, 255, 255) if ix % 2 else (0, 0, 0))
+        bk.paste(B, center_type="center")
+        bk.transparent(2)
+        ix += 1
+        help_img_list.append(A)
+        simple_help_img_list.append(bk)
+    height = 0
+    for img in help_img_list:
+        height += img.h
+    A = CreateImg(width + 150, height + 50, font_size=24)
+    A.text((10, 10), '*  注: ‘*’ 代表可有多个相同参数 ‘?’ 代表可省略该参数  *\n\n" "功能名: 功能简介 -> 指令\n\n')
+    current_height = 50
+    for img in help_img_list:
+        A.paste(img, (0, current_height))
+        current_height += img.h
+    A.save(help_image)
+    height = 0
+    width = 0
+    for img in simple_help_img_list:
+        if img.h > height:
+            height = img.h
+        width += img.w + 10
+    B = CreateImg(width + 100, height + 250, font_size=24)
+    width, _ = get_max_width_or_paste(simple_help_img_list, B)
+    bk = None
+    random_bk = os.listdir(random_bk_path)
+    if random_bk:
+        bk = random.choice(random_bk)
+    x = max(width + 50, height + 250)
+    B = CreateImg(
+        x,
+        x,
+        font_size=24,
+        color="#FFEFD5",
+        background=random_bk_path / bk,
     )
-    A.text(
-        (10, e.h + u.h + o.h + 250),
-        f"【注】「色图概率：好感度 + {int(INITIAL_SETU_PROBABILITY*100)}%\n"
-        f"\t\t每 3 点好感度 + 1次开箱，初始 {INITIAL_OPEN_CASE_COUNT} 次\n"
-        f"\t\t开启/关闭功能只需输入‘开启/关闭 指令名’（每个功能的第一个指令）」\n"
-        f"\t\t示例：开启签到\n"
-        f"\t\t可以通过管理员开关自动发送消息(早晚安等)\n"
-        f"\t\t^请查看管理员帮助^\n\n"
-        f"\t「如果{NICKNAME}回复了一些不符合人设的话，那是因为每日白嫖的图灵次数已用完，使用的是备用接口【QAQ】」",
-    )
-    A.save(DATA_PATH + f"group_help/{group_id}.png")
+    B.filter("GaussianBlur", 10)
+    _, B = get_max_width_or_paste(simple_help_img_list, B, True)
+    w = 10
+    h = 10
+    for msg in ['目前支持的功能列表:', '可以通过 ‘帮助[功能名称]’ 来获取对应功能的使用方法', '或者使用 ‘详细帮助’ 来获取所有功能方法']:
+        text = CreateImg(
+            0,
+            0,
+            plain_text=msg,
+            font_size=24,
+            color=(255, 255, 255, 0),
+            font='yuanshen.ttf'
+        )
+        B.paste(text, (w, h), True)
+        h += 50
+        if msg == '目前支持的功能列表:':
+            w += 50
+    B.save(simple_help_image)
 
 
-def parse_cmd(cmd, group_id):
-    flag = "√"
-    dfg = None
-    if cmd.find("draw_card") != -1:
-        lst = cmd.split("_")
-        cmd = lst[0] + "_" + lst[1]
-        dfg = lst[-1]
-    elif cmd == "pixiv_r":
-        cmd = "pixiv"
-        dfg = "r"
-    elif cmd == "pixiv_s":
-        cmd = "pixiv"
-        dfg = "s"
-    if not group_manager.get_plugin_status(cmd, group_id):
-        flag = "×"
-    if cmd in ["bt", "reimu", "nickname"]:
-        flag = "- "
-    return flag, dfg
+def get_max_width_or_paste(
+    simple_help_img_list: list, B: CreateImg = None, is_paste: bool = False
+) -> "int, CreateImg":
+    """
+    获取最大宽度，或直接贴图
+    :param simple_help_img_list: 简单帮助图片列表
+    :param B: 背景图
+    :param is_paste: 是否直接贴图
+    """
+    current_width = 50
+    current_height = 180
+    max_width = simple_help_img_list[0].w
+    for i in range(len(simple_help_img_list)):
+        try:
+            if is_paste and B:
+                B.paste(simple_help_img_list[i], (current_width, current_height), True)
+            current_height += simple_help_img_list[i].h + 40
+            if current_height + simple_help_img_list[i + 1].h > B.h - 10:
+                current_height = 180
+                current_width += max_width + 30
+                max_width = 0
+            elif simple_help_img_list[i].w > max_width:
+                max_width = simple_help_img_list[i].w
+        except IndexError:
+            pass
+    if current_width > simple_help_img_list[0].w + 50:
+        current_width += simple_help_img_list[-1].w
+    return current_width, B
 
 
-def rcmd(dfg):
-    if dfg in [
-        "prts",
-        "genshin",
-        "pretty",
-        "guardian",
-        "pcr",
-        "azur",
-        "onmyoji",
-        "fgo",
-    ]:
-        return "draw_card_" + dfg
-    if dfg == "r":
-        return "pixiv_r"
-    if dfg == "s":
-        return "pixiv_s"
-
-
-def get_plugin_help(msg: str) -> str:
-    plugin = None
-    for p in plugins2info_dict.keys():
-        if msg in plugins2info_dict[p]["cmd"]:
-            plugin = nonebot.plugin.get_plugin(p)
-            break
-    if plugin:
-        result = plugin.module.__getattribute__("__plugin_usage__")
-        return result
-    else:
-        return "没有此功能的帮助信息..."
+def get_plugin_help(msg: str, is_super: bool = False) -> Optional[str]:
+    """
+    获取功能的帮助信息
+    :param msg: 功能cmd
+    :param is_super: 是否为超级用户
+    """
+    module = plugins2settings_manager.get_plugin_module(msg)
+    if module:
+        try:
+            plugin = nonebot.plugin.get_plugin(module)
+            if plugin:
+                if is_super:
+                    result = plugin.module.__getattribute__(
+                        "__plugin_superuser_usage__"
+                    )
+                else:
+                    result = plugin.module.__getattribute__("__plugin_usage__")
+                width = 0
+                for x in result.split("\n"):
+                    _width = len(x) * 24
+                    width = width if width > _width else _width
+                height = len(result.split("\n")) * 45
+                A = CreateImg(width, height, font_size=24)
+                bk = CreateImg(
+                    width, height, background=Path(IMAGE_PATH) / "background" / "1.png"
+                )
+                A.paste(bk, alpha=True)
+                A.text((int(width * 0.048), int(height * 0.21)), result)
+                return A.pic2bs4()
+        except AttributeError:
+            pass
+    return None