feat: 修复插件服务

Author: lrhh123

src/main/backend/backend.ts

@@ -395,13 +395,13 @@ class BKServer {
     // 检查插件是否正常工作
     this.app.post('/api/v1/base/plugin/check', async (req, res) => {
       try {
-        const { code, message, ctx } = req.body;
+        const { code, messages, ctx } = req.body;
         const ctxMap = new Map(Object.entries(ctx));
         const resp = await this.pluginService.checkPlugin(
           code,
           // @ts-ignore
           ctxMap,
-          message,
+          messages,
         );
         res.json(resp);
       } catch (error) {

src/main/backend/constants/index.ts

@@ -65,3 +65,63 @@ export const CTX_CURRENT_GOODS_ID = 'CTX_CURRENT_GOODS_ID'; // 当前商品 ID
 export const CTX_MEMBER_TAG = 'CTX_MEMBER_TAG'; // 会员标签
 export const CTX_FAN_TAG = 'CTX_FAN_TAG'; // 粉丝标签
 export const CTX_NEW_CUSTOMER_TAG = 'CTX_NEW_CUSTOMER_TAG'; // 新客标签
+
+export const PluginDefaultRunCode = `
+const cc = require('config_srv');
+const rp = require('reply_srv');
+
+/**
+ * 插件主函数
+ * @param {AppContext} ctx - 上下文信息
+ * @param {Message[]} messages - 消息数组
+ * @returns {Reply} 插件执行结果
+ */
+async function main(ctx, messages) {
+  const appId = ctx.get('app_id');
+  const instanceId = ctx.get('instance_id');
+
+  const cfg = cc.get(ctx);
+
+  console.log('插件开始执行....');
+  console.log('当前应用 ID:', appId);
+  console.log('当前客服实例 ID:', instanceId);
+  console.log('当前配置:', ctx, messages);
+
+  // 先检查是否存在用户的消息
+  const lastUserMsg = messages
+    .slice()
+    .reverse()
+    .find((msg) => msg.role === 'OTHER');
+
+  if (!lastUserMsg) {
+    return {
+      type: 'TEXT',
+      content: cfg.default_reply, // 默认回复
+    };
+  }
+
+  // 等待随机时间
+  await new Promise((resolve) => {
+    const min = cfg.reply_speed;
+    const max = cfg.reply_random_speed + cfg.reply_speed;
+    const randomTime = min + Math.random() * (max - min);
+    setTimeout(resolve, randomTime * 1000);
+  });
+
+  // 再检查是否使用关键词匹配
+  if (rp.isKeywordMatch) {
+    const data = await rp.matchKeyword(ctx, lastUserMsg);
+    if (data) return data;
+  }
+
+  // 最后检查是否使用 GPT 生成回复
+  if (rp.isUseGptReply) {
+    const data = await rp.getLLMResponse(cfg, ctx, messages);
+    if (data) return data;
+  }
+
+  return {
+    type: 'TEXT',
+    content: cfg.default_reply,
+  };
+}`;

src/main/backend/entities/config.ts

@@ -43,6 +43,8 @@ export class Config extends Model {
   declare model: string;
 
   declare activation_code: string;
+
+  declare version: string;
 }
 
 export function initConfig(sequelize: Sequelize) {
@@ -140,6 +142,11 @@ export function initConfig(sequelize: Sequelize) {
         type: DataTypes.STRING,
         allowNull: true,
       },
+      version: {
+        type: DataTypes.STRING(255),
+        defaultValue: '1.0.0',
+        allowNull: true,
+      },
     },
     {
       sequelize,

src/main/backend/entities/plugin.ts

@@ -14,6 +14,8 @@ export class Plugin extends Model {
   declare instance_id: string; // 可能是作用于单个实例的插件
 
   declare created_at: Date;
+
+  declare version: string;
 }
 
 export function initPlugin(sequelize: Sequelize) {
@@ -48,6 +50,11 @@ export function initPlugin(sequelize: Sequelize) {
         type: DataTypes.DATE,
         allowNull: true,
       },
+      version: {
+        type: DataTypes.STRING(255),
+        defaultValue: '1.0.0',
+        allowNull: true,
+      },
     },
     {
       sequelize,

src/main/backend/services/dispatchService.ts

@@ -7,6 +7,7 @@ import PluginService from './pluginService';
 import { ConfigController } from '../controllers/configController';
 import { MessageController } from '../controllers/messageController';
 import { Instance } from '../entities/instance';
+import { PluginDefaultRunCode } from '../constants';
 
 export class DispatchService {
   private mainWindow: BrowserWindow;
@@ -58,10 +59,19 @@ export class DispatchService {
 
       // 检查是否使用插件
       const cfg = await this.configController.get(ctxMap);
+      await this.messageService.extractMsgInfo(cfg, ctxMap, messages);
       if (cfg.use_plugin && cfg.plugin_id) {
-        reply = this.pluginService.executePlugin(cfg.plugin_id, ctx, messages);
+        reply = await this.pluginService.executePlugin(
+          cfg.plugin_id,
+          ctx,
+          messages,
+        );
       } else {
-        reply = await this.messageService.getReply(ctxMap, messages);
+        reply = await this.pluginService.executePluginCode(
+          PluginDefaultRunCode,
+          ctxMap,
+          messages,
+        );
       }
 
       callback(reply);

src/main/backend/services/messageService.ts

@@ -58,66 +58,13 @@ export class MessageService {
     this.isUseGptReply = isUseGptReply;
   }
 
-  /**
-   * 获取回复消息
-   * @param ctx
-   * @param messages
-   * @returns
-   */
-  public async getReply(
-    ctx: Context,
-    messages: MessageDTO[],
-  ): Promise<ReplyDTO> {
-    const cfg = await this.configController.get(ctx);
-
-    // 提取消息中的信息
-    await this.extractMsgInfo(cfg, ctx, messages);
-
-    // 先检查是否存在用户的消息
-    const lastUserMsg = messages
-      .slice()
-      .reverse()
-      .find((msg) => msg.role === 'OTHER');
-    if (!lastUserMsg) {
-      return {
-        type: 'TEXT',
-        content: cfg.default_reply,
-      };
-    }
-
-    // 先等待随机时间
-    await new Promise((resolve) => {
-      const min = cfg.reply_speed; // 5 seconds
-      const max = cfg.reply_random_speed + cfg.reply_speed; // 10 seconds
-      const randomTime = min + Math.random() * (max - min);
-      setTimeout(resolve, randomTime * 1000);
-    });
-
-    // 再检查是否使用关键词匹配
-    if (this.isKeywordMatch) {
-      const data = await this.matchKeyword(ctx, lastUserMsg);
-      if (data) return data;
-    }
-
-    // 最后检查是否使用 GPT 生成回复
-    if (this.isUseGptReply) {
-      const data = await this.getLLMResponse(cfg, ctx, messages);
-      if (data) return data;
-    }
-
-    return {
-      type: 'TEXT',
-      content: cfg.default_reply,
-    };
-  }
-
   /**
    * 匹配关键词
    * @param ctx
    * @param message
    * @returns
    */
-  private async matchKeyword(
+  public async matchKeyword(
     ctx: Context,
     message: MessageDTO,
   ): Promise<ReplyDTO | null> {
@@ -167,7 +114,7 @@ export class MessageService {
    * @param messages
    * @returns
    */
-  private async getLLMResponse(
+  public async getLLMResponse(
     cfg: Config,
     ctx: Context,
     messages: MessageDTO[],
@@ -262,7 +209,11 @@ export class MessageService {
    * @param messages
    * @returns
    */
-  async extractMsgInfo(cfg: Config, ctx: Context, messages: MessageDTO[]) {
+  public async extractMsgInfo(
+    cfg: Config,
+    ctx: Context,
+    messages: MessageDTO[],
+  ) {
     if (!cfg.extract_phone && !cfg.extract_product) return;
     if (cfg.save_path === '') return;