first commit

README.md

@@ -0,0 +1,10 @@
+一款优化coze-discord-proxy返回结果的chatgpt-on-wechat插件。
+
+安装方法：
+
+```sh
+#installp https://github.com/wangxyd/nicecoze.git
+#scanp
+```
+
+配置方法：无需任何配置！

__init__.py

@@ -0,0 +1 @@
+from .nicecoze import *

nicecoze.py

@@ -0,0 +1,68 @@
+# encoding:utf-8
+
+import json
+import os
+from urllib.parse import urlparse
+
+import plugins
+from bridge.context import ContextType
+from bridge.reply import Reply, ReplyType
+from common.log import logger
+from plugins import *
+
+
+@plugins.register(
+    name="Nicecoze",
+    desire_priority=100,
+    hidden=True,
+    desc="一款优化coze-discord-proxy返回结果的插件。",
+    version="1.0",
+    author="空心菜",
+)
+class Nicecoze(Plugin):
+    def __init__(self):
+        super().__init__()
+        try:
+            self.handlers[Event.ON_DECORATE_REPLY] = self.on_decorate_reply
+            logger.info("[Nicecoze] inited.")
+        except Exception as e:
+            logger.warn("[Nicecoze] init failed, ignore.")
+            raise e
+
+    def on_decorate_reply(self, e_context: EventContext):
+        if e_context["reply"].type != ReplyType.TEXT:
+            return
+        reply = e_context["reply"]
+        try:
+            content_list = reply.content.strip().split('\n')
+            # 提取CDP返回的Markdown图片链接中的网址，并修改ReplyType为IMAGE_URL，以便CoW自动下载Markdown链接中的图片
+            if len(content_list)==2 and self.is_url(content_list[0]):
+                if content_list[1] == f"![Image]({content_list[0]})":
+                    reply = Reply(ReplyType.IMAGE_URL, content_list[0])
+                    e_context["reply"] = reply
+                    e_context.action = EventAction.CONTINUE
+                    logger.info(f"[Nicecoze] Change ReplyType from TEXT to IMAGE_URL: {content_list[0]}")
+                else:
+                    logger.debug(f"[Nicecoze] URL in content but not a markdown image url.")
+            # 去掉每行结尾的Markdown链接中网址部分的小括号，避免微信误以为“)”是网址的一部分导致微信中无法打开该页面
+            else:
+                new_content_list = [re.sub(r'\((https?://[^\s]+)\)$', r' \1', line) for line in content_list]
+                reply = Reply(ReplyType.TEXT, '\n'.join(new_content_list))
+                e_context["reply"] = reply
+                e_context.action = EventAction.CONTINUE
+        except Exception:
+            pass
+
+    def is_url(self, string):
+        """
+        判断字符串是否为URL的urllib库方法
+        """
+        try:
+            result = urlparse(string)
+            return all([result.scheme, result.netloc])
+        except ValueError:
+            return False
+
+    def get_help_text(self, **kwargs):
+        return "一款优化coze-discord-proxy返回结果的插件。"
+