update v0.0.5.0

plugins/luxun/__init__.py

@@ -1,28 +1,41 @@
-from PIL import ImageFont, ImageDraw, Image
-import textwrap
-from configs.path_config import IMAGE_PATH, TTF_PATH
+from configs.path_config import IMAGE_PATH
 from nonebot import on_command
 from nonebot.typing import T_State
 from nonebot.adapters.cqhttp import Bot, MessageEvent, GroupMessageEvent
 from utils.message_builder import image
 from services.log import logger
-from utils.utils import UserExistLimiter, get_message_text
-from utils.image_utils import pic2b64
-
-__plugin_name__ = "鲁迅说"
-
-__plugin_usage__ = "用法：鲁迅说 [消息]"
-
-_ulmt = UserExistLimiter()
+from utils.utils import get_message_text
+from utils.image_utils import CreateImg
 
+__zx_plugin_name__ = "鲁迅说"
+__plugin_usage__ = """
+usage：
+    鲁迅说了啥？
+    指令：
+        鲁迅说 [文本]
+""".strip()
+__plugin_des__ = "鲁迅说他没说过这话！"
+__plugin_cmd__ = ["鲁迅说"]
+__plugin_version__ = 0.1
+__plugin_author__ = "HibiKier"
+__plugin_settings__ = {
+    "level": 5,
+    "default_status": True,
+    "limit_superuser": False,
+    "cmd": ["鲁迅说"],
+}
+__plugin_block_limit__ = {
+    "rst": "你的鲁迅正在说，等会"
+}
 
 luxun = on_command("鲁迅说过", aliases={"鲁迅说"}, priority=5, block=True)
 
 
+luxun_author = CreateImg(0, 0, plain_text="--鲁迅", font_size=30, font='msyh.ttf', font_color=(255, 255, 255))
+
+
 @luxun.handle()
 async def handle(bot: Bot, event: MessageEvent, state: T_State):
-    if _ulmt.check(event.user_id):
-        await luxun.finish("你的鲁迅正在说，等会", at_sender=True)
     args = get_message_text(event.json())
     if args:
         state["content"] = args if args else "烦了，不说了"
@@ -30,39 +43,24 @@ async def handle(bot: Bot, event: MessageEvent, state: T_State):
 
 @luxun.got("content", prompt="你让鲁迅说点啥?")
 async def handle_event(bot: Bot, event: MessageEvent, state: T_State):
-    filename = str(event.user_id) + "_.jpg"
     content = state["content"].strip()
     if content.startswith(",") or content.startswith("，"):
         content = content[1:]
-    _ulmt.set_true(event.user_id)
-    if len(content) > 20:
-        _ulmt.set_false(event.user_id)
-        await luxun.finish("太长了, 鲁迅说不完!", at_sender=True)
-    else:
-        if len(content) >= 12:
-            content = content[:12] + "\n" + content[12:]
-        img = image(b64=process_pic(content, filename))
-        logger.info(
-            f"USER {event.user_id} GROUP "
-            f"{event.group_id if isinstance(event, GroupMessageEvent) else 'private'} 鲁迅说过 {content}"
-        )
-        await luxun.send(img)
-        _ulmt.set_false(event.user_id)
-
-
-def process_pic(content, filename) -> str:
-    text = content
-    para = textwrap.wrap(text, width=15)
-    MAX_W, MAX_H = 480, 280
-    bk_img = Image.open(IMAGE_PATH + "other/luxun.jpg")
-    font_path = TTF_PATH + "/msyh.ttf"
-    font = ImageFont.truetype(font_path, 37)
-    font2 = ImageFont.truetype(font_path, 30)
-    draw = ImageDraw.Draw(bk_img)
-    current_h, pad = 300, 10
-    for line in para:
-        w, h = draw.textsize(line, font=font)
-        draw.text(((MAX_W - w) / 2, current_h), line, font=font)
-        current_h += h + pad
-    draw.text((320, 400), "——鲁迅", font=font2, fill=(255, 255, 255))
-    return pic2b64(bk_img)
+    A = CreateImg(0, 0, font_size=37, background=f'{IMAGE_PATH}/other/luxun.jpg', font='msyh.ttf')
+    x = ""
+    if len(content) > 40:
+        await luxun.finish('太长了，鲁迅说不完...')
+    while A.getsize(content)[0] > A.w - 50:
+        n = int(len(content) / 2)
+        x += content[:n] + '\n'
+        content = content[n:]
+    x += content
+    if len(x.split('\n')) > 2:
+        await luxun.finish('太长了，鲁迅说不完...')
+    A.text((int((480 - A.getsize(x.split("\n")[0])[0]) / 2), 300), x, (255, 255, 255))
+    A.paste(luxun_author, (320, 400), True)
+    await luxun.send(image(b64=A.pic2bs4()))
+    logger.info(
+        f"USER {event.user_id} GROUP "
+        f"{event.group_id if isinstance(event, GroupMessageEvent) else 'private'} 鲁迅说过 {content}"
+    )