feat: 优化各个平台的独立配置功能；添加关闭自动回复，只使用关键词回复的功能；添加延时随机时间

Author: lrhh123

.eslintrc.js

@@ -28,6 +28,8 @@ module.exports = {
     '@typescript-eslint/no-unused-vars': 'error',
     'no-console': 'off',
     'max-classes-per-file': 'off',
+    'no-continue': 'off',
+    'no-plusplus': 'off',
   },
   parserOptions: {
     ecmaVersion: 2022,

README.md

@@ -118,13 +118,14 @@
 - [x] 支持千牛平台的基础聊天功能
 - [x] 支持京麦平台的基础聊天功能
 - [ ] 支持多平台文章自动发送功能
-- [ ] 添加关闭自动回复，只使用关键词回复的功能
-- [ ] 添加延时随机时间
-- [ ] 优化各个平台的独立配置功能
+- [x] 添加关闭自动回复，只使用关键词回复的功能
+- [x] 添加延时随机时间
+- [x] 优化各个平台的独立配置功能
 - [x] 支持 Excel 导入回复内容
-- [ ] 支持导出回复内容到 Excel
+- [x] 支持导出回复内容到 Excel
 - [ ] 支持导出聊天记录到 Excel
-- [ ] 添加中文路径的支持
+- [x] 添加中文路径的支持
+- [ ] 实现小红书回复后，自动私聊的功能
 - [ ] 优化微信平台的回复速度
 - [ ] 支持企业微信外部群聊天功能
 - [ ] 支持企业微信自动加人功能

package.json

@@ -101,6 +101,7 @@
     "@emotion/styled": "^11.11.0",
     "@hw-hmscore/analytics-web": "6.9.9-301",
     "@tanstack/react-query": "4.36.1",
+    "@types/lodash": "^4.17.1",
     "ansi-styles": "^6.2.1",
     "axios": "^1.6.7",
     "body-parser": "^1.20.2",

pnpm-lock.yaml

@@ -68,15 +68,17 @@ class BKServer {
       transports: ['websocket'],
     });
 
-    this.messageService = new MessageService(
-      configController,
-      messageController,
-      autoReplyController,
-    );
     this.configService = new ConfigService(
       configController,
       platformConfigController,
     );
+
+    this.messageService = new MessageService(
+      this.configService,
+      messageController,
+      autoReplyController,
+    );
+
     this.broadcastService = new BroadcastService(mainWindow);
     this.strategyService = new StrategyService(this.io);
     this.sessionService = new SessionService(
@@ -173,22 +175,20 @@ class BKServer {
 
     // 获取平台设置
     this.app.get(
-      '/api/v1/base/platform/settings',
+      '/api/v1/base/platform/setting',
       asyncHandler(async (req, res) => {
-        // eslint-disable-next-line @typescript-eslint/no-unused-vars
-        const { total, data } = await platformConfigController.list({
-          page: 1,
-          pageSize: 100,
-        });
-        res.json({ success: true, data });
+        const { platformId } = req.query;
+        // @ts-ignore
+        const obj = await platformConfigController.getByPlatformId(platformId);
+        res.json({ success: true, data: obj.settings });
       }),
     );
 
     this.app.post(
-      '/api/v1/base/platform/settings',
+      '/api/v1/base/platform/setting',
       asyncHandler(async (req, res) => {
-        const { platform_id: platformId, config } = req.body;
-        await platformConfigController.updateByPlatformId(platformId, config);
+        const { platformId, settings } = req.body;
+        await platformConfigController.updateByPlatformId(platformId, settings);
         res.json({ success: true });
       }),
     );
@@ -198,6 +198,7 @@ class BKServer {
       const {
         is_paused: isPaused,
         is_keyword_match: isKeywordMatch,
+        is_use_gpt: isUseGptReply,
         ids,
       } = req.body;
       try {
@@ -212,7 +213,7 @@ class BKServer {
         }
 
         if (isKeywordMatch) {
-          this.messageService.updateKeywordMatch(isKeywordMatch);
+          this.messageService.updateKeywordMatch(isKeywordMatch, isUseGptReply);
         }
 
         await this.strategyService.updateStrategies(ids);
@@ -229,6 +230,8 @@ class BKServer {
     this.app.get('/api/v1/base/settings', async (req, res) => {
       try {
         const config = await configController.getConfig();
+        // @ts-ignore 兼容性处理
+        config.reply_speed = [config.reply_speed, config.reply_random_speed];
         res.json({ success: true, data: config });
       } catch (error) {
         if (error instanceof Error) {
@@ -240,9 +243,44 @@ class BKServer {
     // Endpoint to update configuration settings
     this.app.post('/api/v1/base/settings', async (req, res) => {
       try {
-        const cfg = req.body;
-        await configController.updateConfig(1, cfg); // Assuming the ID is known and static
-        // globalVariable.merged_message_num = cfg.merged_message_num;
+        const cfg = req.body as {
+          extract_phone: boolean; // 提取手机号
+          extract_product: boolean; // 提取商品
+          save_path?: string; // 保存路径
+          default_reply?: string; // 默认回复
+          reply_speed: number[]; // 回复速度
+          context_count: number; // 合并消息数量
+          wait_humans_time: number; // 等待人工时间
+          gpt_base_url?: string; // GPT服务地址
+          gpt_key?: string; // GPT服务key
+          gpt_model?: string; // GPT服务模型
+          gpt_temperature?: number; // GPT服务温度
+          gpt_top_p?: number; // GPT服务top_p
+          stream?: boolean; // 是否开启stream
+          use_dify?: boolean; // 是否使用 Dify 百宝箱
+        };
+
+        if (!cfg.reply_speed || cfg.reply_speed.length !== 2) {
+          cfg.reply_speed = [0, 0];
+        }
+
+        await configController.updateConfig(1, {
+          extract_phone: cfg.extract_phone,
+          extract_product: cfg.extract_product,
+          default_reply: cfg.default_reply,
+          save_path: cfg.save_path,
+          reply_speed: cfg.reply_speed[0],
+          reply_random_speed: cfg.reply_speed[1],
+          context_count: cfg.context_count,
+          wait_humans_time: cfg.wait_humans_time,
+          gpt_base_url: cfg.gpt_base_url,
+          gpt_key: cfg.gpt_key,
+          gpt_model: cfg.gpt_model,
+          gpt_temperature: cfg.gpt_temperature,
+          gpt_top_p: cfg.gpt_top_p,
+          stream: cfg.stream,
+          use_dify: cfg.use_dify,
+        });
         res.json({ success: true });
       } catch (error) {
         if (error instanceof Error) {

src/main/backend/backend.ts

@@ -15,8 +15,10 @@ export class ConfigController {
         extract_product: true,
         save_path: '',
         reply_speed: 0,
-        merged_message_num: 7,
+        reply_random_speed: 0,
+        default_reply: '',
         wait_humans_time: 60,
+        context_count: 1,
         gpt_base_url: 'https://api.openai.com/v1',
         gpt_key: 'your-key',
         gpt_model: 'gpt-3.5-turbo',

src/main/backend/controllers/configController.ts

@@ -3,29 +3,38 @@ import { PlatformConfig } from '../entities/platformConfig';
 export class PlatformConfigController {
   async updateByPlatformId(
     platformId: string,
-    platformConfigData: Partial<PlatformConfig>,
+    settings: any,
   ): Promise<PlatformConfig> {
     const platformConfig = await PlatformConfig.findOne({
       where: { platform_id: platformId },
     });
 
+    const active = settings && settings.active ? settings.active : false;
+
     if (platformConfig) {
       // @ts-ignore
-      return this.update(platformConfig.id, platformConfigData);
+      return this.update(platformConfig.id, { settings, active });
     }
+
     return this.create({
       platform_id: platformId,
-      ...platformConfigData,
+      settings,
+      active,
     });
   }
 
   async getByPlatformId(platformId: string) {
-    const data = await PlatformConfig.findOne({
+    let data = await PlatformConfig.findOne({
       where: { platform_id: platformId },
     });
 
     if (!data) {
-      throw new Error('PlatformConfig not found');
+      // 创建一个对象
+      data = await PlatformConfig.create({
+        platform_id: platformId,
+        active: false,
+        settings: {},
+      });
     }
 
     return data;
@@ -40,9 +49,11 @@ export class PlatformConfigController {
     const platformConfig = await PlatformConfig.findOne({
       where: { id },
     });
-    // @ts-ignore
-    const data = Object.assign(platformConfig, platformConfigData);
-    return data;
+    if (!platformConfig) {
+      return;
+    }
+
+    await platformConfig.update(platformConfigData);
   }
 
   async delete(id: number): Promise<void> {
@@ -54,18 +65,4 @@ export class PlatformConfigController {
     }
     await platformConfig.destroy();
   }
-
-  async list(page: { page: number; pageSize: number }): Promise<{
-    total: number;
-    data: PlatformConfig[];
-  }> {
-    const data = await PlatformConfig.findAll({
-      limit: page.pageSize,
-      offset: (page.page - 1) * page.pageSize,
-    });
-
-    const total = await PlatformConfig.count();
-
-    return { total, data };
-  }
 }