add: 添加插件校验逻辑

Author: lrhh123

src/main/backend/backend.ts

@@ -5,14 +5,15 @@ import bodyParser from 'body-parser';
 import http from 'http';
 import { Server } from 'socket.io';
 import { BrowserWindow, shell } from 'electron';
-import './ormconfig';
+import { sequelize } from './ormconfig';
 import { StrategyServiceStatusEnum } from './types';
 import { ConfigController } from './controllers/configController';
 import { MessageController } from './controllers/messageController';
 import { KeywordReplyController } from './controllers/keywordReplyController';
 import { MessageService } from './services/messageService';
 import { DispatchService } from './services/dispatchService';
 import { PluginService } from './services/pluginService';
+import { AppService } from './services/appService';
 
 const configController = new ConfigController();
 const messageController = new MessageController();
@@ -33,6 +34,8 @@ class BKServer {
 
   private dispatchService: DispatchService;
 
+  private appService: AppService;
+
   constructor(port: number, mainWindow: BrowserWindow) {
     this.app = express();
     this.app.use(bodyParser.json());
@@ -72,8 +75,14 @@ class BKServer {
       this.pluginService,
     );
 
+    this.appService = new AppService(this.dispatchService, sequelize);
+
     this.configureSocketIO();
     this.setupRoutes();
+    // 开启定时任务
+    setInterval(() => {
+      this.appService.initTasks();
+    }, 5 * 1000);
   }
 
   private configureSocketIO(): void {
@@ -383,10 +392,31 @@ class BKServer {
       // }
     });
 
+    // 检查插件是否正常工作
+    this.app.post('/api/v1/base/plugin/check', async (req, res) => {
+      try {
+        const { code, message, ctx } = req.body;
+        const ctxMap = new Map(Object.entries(ctx));
+        const resp = await this.pluginService.checkPlugin(
+          code,
+          // @ts-ignore
+          ctxMap,
+          message,
+        );
+        res.json(resp);
+      } catch (error) {
+        res.json({
+          success: false,
+          error: error instanceof Error ? error.message : String(error),
+          message: error instanceof Error ? error.message : String(error),
+        });
+      }
+    });
+
     // 获取任务列表
     this.app.get('/api/v1/strategy/tasks', async (req, res) => {
       try {
-        const tasks = await this.dispatchService.getTasks();
+        const tasks = await this.appService.getTasks();
         res.json({
           success: true,
           data: tasks,
@@ -404,7 +434,7 @@ class BKServer {
     this.app.post('/api/v1/strategy/tasks', async (req, res) => {
       const { appId } = req.body;
       try {
-        const task = await this.dispatchService.addTask(String(appId));
+        const task = await this.appService.addTask(String(appId));
         res.json({
           success: true,
           data: task,
@@ -422,7 +452,7 @@ class BKServer {
     this.app.post('/api/v1/strategy/task/remove', async (req, res) => {
       const { taskId } = req.body;
       try {
-        await this.dispatchService.removeTask(String(taskId));
+        await this.appService.removeTask(String(taskId));
         res.json({
           success: true,
         });

src/main/backend/entities/instance.ts

@@ -0,0 +1,41 @@
+import { DataTypes, Model, Sequelize } from 'sequelize';
+
+export class Instance extends Model {
+  declare id: number; // instance_id
+
+  declare app_id: string;
+
+  declare env_id: string;
+
+  declare created_at: string;
+}
+
+export function initInstance(sequelize: Sequelize) {
+  Instance.init(
+    {
+      id: {
+        type: DataTypes.INTEGER,
+        autoIncrement: true,
+        primaryKey: true,
+      },
+      app_id: {
+        type: DataTypes.STRING(255),
+        allowNull: false,
+      },
+      env_id: {
+        type: DataTypes.STRING(255),
+        allowNull: true,
+      },
+      created_at: {
+        type: DataTypes.DATE,
+        allowNull: false,
+      },
+    },
+    {
+      sequelize,
+      modelName: 'Instance',
+      tableName: 'instance',
+      timestamps: false,
+    },
+  );
+}

src/main/backend/ormconfig.ts

@@ -9,6 +9,7 @@ import { Config, initConfig } from './entities/config';
 import { initSession } from './entities/session';
 import { initMessage } from './entities/message';
 import { initPlugin } from './entities/plugin';
+import { initInstance } from './entities/instance';
 import { Keyword, initKeyword } from './entities/keyword';
 
 // Get user's documents directory path
@@ -25,7 +26,7 @@ const DB_FILE_PATH = path.join(APP_DIR, 'msg.db');
 
 console.log('DB_FILE_PATH:', DB_FILE_PATH);
 
-const sequelize = new Sequelize({
+export const sequelize = new Sequelize({
   dialect: 'sqlite',
   dialectModule: sqlite,
   storage: DB_FILE_PATH,
@@ -38,6 +39,7 @@ initSession(sequelize);
 initMessage(sequelize);
 initKeyword(sequelize);
 initPlugin(sequelize);
+initInstance(sequelize);
 
 // 异步初始化和数据填充函数
 async function initDb(): Promise<void> {

src/main/backend/services/appService.ts

@@ -0,0 +1,117 @@
+import { Sequelize, Transaction } from 'sequelize';
+import { DispatchService } from './dispatchService';
+import { Instance } from '../entities/instance';
+import { Config } from '../entities/config';
+import { Plugin } from '../entities/plugin';
+
+export class AppService {
+  private dispatchService: DispatchService;
+
+  private sequelize: Sequelize;
+
+  constructor(dispatchService: DispatchService, sequelize: Sequelize) {
+    this.dispatchService = dispatchService;
+    this.sequelize = sequelize;
+  }
+
+  public async getTasks(): Promise<
+    {
+      task_id: string;
+      env_id: string;
+      app_id: string;
+    }[]
+  > {
+    const instances = await Instance.findAll();
+    return instances.map((instance) => ({
+      task_id: String(instance.id),
+      env_id: instance.env_id,
+      app_id: instance.app_id,
+    }));
+  }
+
+  /**
+   * 初始化全部任务
+   */
+  public async initTasks(): Promise<void> {
+    const instances = await Instance.findAll();
+    await this.dispatchService.updateTasks(instances);
+  }
+
+  /**
+   * 添加一个任务
+   */
+  public async addTask(appId: string): Promise<Instance | null> {
+    // 使用事务
+    return this.sequelize
+      .transaction(async (t: Transaction) => {
+        const instance = await Instance.create(
+          {
+            app_id: appId,
+            created_at: new Date(),
+          },
+          { transaction: t },
+        );
+
+        // 取得全部 Tasks 然后全部更新
+        const tasks = await Instance.findAll();
+        tasks.push(instance);
+        const result = await this.dispatchService.updateTasks(tasks);
+        if (!result || result.length === 0) {
+          throw new Error('Failed to update tasks');
+        }
+
+        const target = result.find(
+          (task) => task.task_id === String(instance.id),
+        );
+        if (!target) {
+          throw new Error('Failed to find target task');
+        }
+
+        instance.env_id = target.env_id;
+        await instance.save({ transaction: t });
+        return instance;
+      })
+      .catch((error) => {
+        // 处理错误
+        console.error('Transaction failed:', error);
+        throw error; // 可根据需求自定义错误处理逻辑
+      });
+  }
+
+  /**
+   * 移除一个任务
+   */
+  public async removeTask(taskId: string): Promise<boolean> {
+    const instance = await Instance.findByPk(taskId);
+
+    if (!instance) {
+      return false;
+    }
+
+    await instance.destroy();
+
+    // 找到对应的 Config 删除
+    const config = await Config.findOne({
+      where: { instance_id: taskId },
+    });
+    if (config) {
+      // 检查是否使用插件
+      if (config.plugin_id) {
+        const plugin = await Plugin.findOne({
+          where: { id: config.plugin_id },
+        });
+        if (plugin) {
+          await plugin.destroy();
+        }
+      }
+
+      await config.destroy();
+    }
+
+    // 取得全部 Tasks 然后全部更新
+    const tasks = await Instance.findAll();
+    await this.dispatchService.updateTasks(tasks);
+
+    return true;
+  }
+}

src/main/backend/services/dispatchService.ts

@@ -6,6 +6,7 @@ import { MessageService } from './messageService';
 import PluginService from './pluginService';
 import { ConfigController } from '../controllers/configController';
 import { MessageController } from '../controllers/messageController';
+import { Instance } from '../entities/instance';
 
 export class DispatchService {
   private mainWindow: BrowserWindow;
@@ -83,37 +84,27 @@ export class DispatchService {
     }
   }
 
-  public async getTasks(): Promise<any> {
+  public async updateTasks(tasks: Instance[]): Promise<
+    | {
+        task_id: string;
+        env_id: string;
+      }[]
+    | null
+  > {
     try {
-      return await emitAndWait(this.io, 'strategyService-getTasks', {});
-    } catch (error) {
-      console.error('Failed to get tasks', error);
-      return null;
-    }
-  }
-
-  public async addTask(appId: string): Promise<any> {
-    try {
-      return await emitAndWait(this.io, 'strategyService-addTask', {
-        app_id: appId,
+      return await emitAndWait(this.io, 'strategyService-updateTasks', {
+        tasks: tasks.map((task) => ({
+          task_id: task.id,
+          app_id: task.app_id,
+          env_id: task.env_id,
+        })),
       });
     } catch (error) {
       console.error('Failed to add task', error);
       return null;
     }
   }
 
-  public async removeTask(taskId: string): Promise<any> {
-    try {
-      return await emitAndWait(this.io, 'strategyService-removeTask', {
-        task_id: taskId,
-      });
-    } catch (error) {
-      console.error('Failed to remove task', error);
-      return null;
-    }
-  }
-
   public async updateStatus(status: StrategyServiceStatusEnum): Promise<any> {
     try {
       return await emitAndWait(this.io, 'strategyService-updateStatus', {