69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
# 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"":
|
||
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返回结果的插件。"
|
||
|