feat: check app_code dynamically

plugins/linkai/linkai.py

@@ -8,7 +8,7 @@ from bridge.context import ContextType
 from bridge.reply import Reply, ReplyType
 from channel.chat_message import ChatMessage
 from common.log import logger
-from config import conf
+from config import conf, global_config
 from plugins import *
 from .midjourney import MJBot, TaskType
 
@@ -46,14 +46,48 @@ class LinkAI(Plugin):
             self.mj_bot.process_mj_task(mj_type, e_context)
             return
 
+        if context.content.startswith(f"{_get_trigger_prefix()}linkai"):
+            # 应用管理功能
+            self._process_admin_cmd(e_context)
+            return
+
         if self._is_chat_task(e_context):
+            # 文本对话任务处理
             self._process_chat_task(e_context)
 
+    # 插件管理功能
+    def _process_admin_cmd(self, e_context: EventContext):
+        context = e_context['context']
+        cmd = context.content.split()
+        if len(cmd) == 1 or (len(cmd) == 2 and cmd[1] == "help"):
+            _set_reply_text(self.get_help_text(verbose=True), e_context, level=ReplyType.INFO)
+            return
+        if len(cmd) == 3 and cmd[1] == "app":
+            if not context.kwargs.get("isgroup"):
+                _set_reply_text("该指令需在群聊中使用", e_context, level=ReplyType.ERROR)
+                return
+            if e_context["context"]["session_id"] not in global_config["admin_users"]:
+                _set_reply_text("需要管理员权限执行", e_context, level=ReplyType.ERROR)
+                return
+            app_code = cmd[2]
+            group_name = context.kwargs.get("msg").from_user_nickname
+            group_mapping = self.config.get("group_app_map")
+            if group_mapping:
+                group_mapping[group_name] = app_code
+            else:
+                self.config["group_app_map"] = {group_name: app_code}
+            # 保存插件配置
+            super().save_config(self.config)
+            _set_reply_text(f"应用设置成功: {app_code}", e_context, level=ReplyType.INFO)
+        else:
+            _set_reply_text(f"指令错误，请输入{_get_trigger_prefix()}linkai help 获取帮助", e_context, level=ReplyType.INFO)
+            return
+
     # LinkAI 对话任务处理
     def _is_chat_task(self, e_context: EventContext):
         context = e_context['context']
         # 群聊应用管理
-        return self.config.get("knowledge_base") and context.kwargs.get("isgroup")
+        return self.config.get("group_app_map") and context.kwargs.get("isgroup")
 
     def _process_chat_task(self, e_context: EventContext):
         """
@@ -73,21 +107,27 @@ class LinkAI(Plugin):
         :param group_name: 群聊名称
         :return: 应用code
         """
-        knowledge_base_config = self.config.get("knowledge_base")
-        if knowledge_base_config and knowledge_base_config.get("group_mapping"):
-            app_code = knowledge_base_config.get("group_mapping").get(group_name) \
-                       or knowledge_base_config.get("group_mapping").get("ALL_GROUP")
+        group_mapping = self.config.get("group_app_map")
+        if group_mapping:
+            app_code = group_mapping.get(group_name) or group_mapping.get("ALL_GROUP")
             return app_code
 
     def get_help_text(self, verbose=False, **kwargs):
-        trigger_prefix = conf().get("plugin_trigger_prefix", "$")
-        help_text = "利用midjourney来画图。\n"
+        trigger_prefix = _get_trigger_prefix()
+        help_text = "用于集成 LinkAI 提供的文本对话、知识库、绘画等能力。\n"
         if not verbose:
             return help_text
-        help_text += f"{trigger_prefix}mj 描述词1,描述词2 ... ： 利用描述词作画，参数请放在提示词之后。\n{trigger_prefix}mjimage 描述词1,描述词2 ... ： 利用描述词进行图生图，参数请放在提示词之后。\n{trigger_prefix}mjr ID: 对指定ID消息重新生成图片。\n{trigger_prefix}mju ID 图片序号: 对指定ID消息中的第x张图片进行放大。\n{trigger_prefix}mjv ID 图片序号: 对指定ID消息中的第x张图片进行变换。\n例如：\n\"{trigger_prefix}mj a little cat, white --ar 9:16\"\n\"{trigger_prefix}mjimage a white cat --ar 9:16\"\n\"{trigger_prefix}mju 1105592717188272288 2\""
+        help_text += ""
+        help_text += f"{trigger_prefix}mj 描述词1,描述词2 ... ： 利用描述词作画，参数请放在提示词之后。\n\n{trigger_prefix}mju ID 图片序号: 对指定ID消息中的第x张图片进行放大。\n例如：\n\"{trigger_prefix}mj a little cat, white --ar 9:16\"\n\"{trigger_prefix}mjimage a white cat --ar 9:16\"\n\"{trigger_prefix}mju 1105592717188272288 2\""
         return help_text
 
-    def _set_reply_text(self, content: str, e_context: EventContext, level: ReplyType=ReplyType.ERROR):
-        reply = Reply(level, content)
-        e_context["reply"] = reply
-        e_context.action = EventAction.BREAK_PASS
+
+# 静态方法
+def _set_reply_text(content: str, e_context: EventContext, level: ReplyType = ReplyType.ERROR):
+    reply = Reply(level, content)
+    e_context["reply"] = reply
+    e_context.action = EventAction.BREAK_PASS
+
+
+def _get_trigger_prefix():
+    return conf().get("plugin_trigger_prefix", "$")