搞小槽的说话

konabot/plugins/memepack/__init__.py

@@ -1,12 +1,20 @@
 from io import BytesIO
+from typing import Iterable, cast
 
-from nonebot_plugin_alconna import (Alconna, Args, Field, MultiVar, UniMessage,
-                                    on_alconna)
+from nonebot import on_message
+from nonebot_plugin_alconna import (Alconna, Args, Field, MultiVar, Text,
+                                    UniMessage, UniMsg, on_alconna)
 
+from konabot.common.nb.extract_image import extract_image_from_message
+from konabot.plugins.memepack.drawing.display import draw_cao_display
 from konabot.plugins.memepack.drawing.saying import (draw_cute_ten,
                                                      draw_geimao, draw_mnk,
                                                      draw_pt, draw_suan)
 
+from nonebot.adapters import Bot, Event
+
+from returns.result import Success, Failure
+
 geimao = on_alconna(Alconna(
     "给猫说",
     Args["saying", MultiVar(str, '+'), Field(
@@ -52,7 +60,7 @@ async def _(saying: list[str]):
     img_bytes = BytesIO()
     img.save(img_bytes, format="PNG")
 
-    await pt.send(await UniMessage().image(raw=img_bytes).export())
+    await mnk.send(await UniMessage().image(raw=img_bytes).export())
 
 
 suan = on_alconna(Alconna(
@@ -68,7 +76,7 @@ async def _(saying: list[str]):
     img_bytes = BytesIO()
     img.save(img_bytes, format="PNG")
 
-    await pt.send(await UniMessage().image(raw=img_bytes).export())
+    await suan.send(await UniMessage().image(raw=img_bytes).export())
 
 
 dsuan = on_alconna(Alconna(
@@ -84,20 +92,50 @@ async def _(saying: list[str]):
     img_bytes = BytesIO()
     img.save(img_bytes, format="PNG")
 
-    await pt.send(await UniMessage().image(raw=img_bytes).export())
+    await dsuan.send(await UniMessage().image(raw=img_bytes).export())
 
 
-dsuan = on_alconna(Alconna(
+cutecat = on_alconna(Alconna(
     "乖猫说",
     Args["saying", MultiVar(str, '+'), Field(
         missing_tips=lambda: "你没有写十猫说了什么"
     )]
 ), use_cmd_start=True, use_cmd_sep=False, skip_for_unmatch=False, aliases={"十猫说"})
 
-@dsuan.handle()
+@cutecat.handle()
 async def _(saying: list[str]):
     img = await draw_cute_ten("\n".join(saying))
     img_bytes = BytesIO()
     img.save(img_bytes, format="PNG")
 
-    await pt.send(await UniMessage().image(raw=img_bytes).export())
+    await cutecat.send(await UniMessage().image(raw=img_bytes).export())
+
+
+cao_display_cmd = on_message()
+
+@cao_display_cmd.handle()
+async def _(msg: UniMsg, evt: Event, bot: Bot):
+    flag = False
+    for text in cast(Iterable[Text], msg.get(Text)):
+        if text.text.strip() == "小槽展示":
+            flag = True
+        elif text.text.strip() == '':
+            continue
+        else:
+            return
+    if not flag:
+        return
+    match await extract_image_from_message(evt.get_message(), evt, bot):
+        case Success(img):
+            img_handled = await draw_cao_display(img)
+            img_bytes = BytesIO()
+            img_handled.save(img_bytes, format="PNG")
+            await cao_display_cmd.send(await UniMessage().image(raw=img_bytes).export())
+        case Failure(err):
+            await cao_display_cmd.send(
+                await UniMessage()
+                    .at(user_id=evt.get_user_id())
+                    .text(' ')
+                    .text(err)
+                    .export()
+            )