add: 添加多平台的支持

Author: lrhh123

.env

@@ -1,4 +1,5 @@
 PY_HOSTNAME=localhost
 PY_PORT=9999
 VAR=1234
-BKEXE_PATH=./backend/__main__.exe
+BKEXE_PATH=./backend/__main__.exe
+PKG_VERSION=0.0.1

.eslintrc.js

@@ -30,6 +30,10 @@ module.exports = {
     'max-classes-per-file': 'off',
     'no-continue': 'off',
     'no-plusplus': 'off',
+    'no-underscore-dangle': 'off',
+    camelcase: 'off',
+    'no-use-before-define': 'off',
+    'no-dupe-class-members': 'off',
   },
   parserOptions: {
     ecmaVersion: 2022,

src/main/backend/backend.ts

@@ -17,7 +17,7 @@ import { BroadcastService } from './services/broadcastService';
 import { SystemService } from './services/systemService';
 import { ConfigService } from './services/configService';
 import { PlatformConfigController } from './controllers/platformConfigController';
-import { AutoReplyController } from './controllers/autoReplyController';
+import { AutoReplyController } from './controllers/keywordReplyController';
 
 const sessionController = new SessionController();
 const configController = new ConfigController();
@@ -93,7 +93,6 @@ class BKServer {
       console.log('Client connected registerHandlers');
       this.messageService.registerHandlers(socket);
       this.sessionService.registerHandlers(socket);
-      this.configService.registerHandlers(socket);
       this.broadcastService.registerHandlers(socket);
 
       socket.on('disconnect', () => {

src/main/backend/constants/index.ts

@@ -0,0 +1,71 @@
+export const ALL_PLATFORMS = [
+  {
+    id: 'bilibili',
+    name: 'bilibili',
+  },
+  {
+    id: 'douyin',
+    name: '抖音',
+  },
+  {
+    id: 'douyin_mp',
+    name: '抖音企业号',
+  },
+  {
+    id: 'win_jinmai',
+    name: '京卖',
+  },
+  {
+    id: 'jinritemai',
+    name: '抖店',
+  },
+  {
+    id: 'win_qianniu',
+    name: '千牛',
+  },
+  {
+    id: 'win_wechat',
+    name: '微信',
+  },
+  {
+    id: 'win_wecom',
+    name: '企微(Bate版)',
+  },
+  {
+    id: 'weibo',
+    name: '微博私信',
+  },
+  {
+    id: 'xiaohongshu',
+    name: '小红书评论',
+  },
+  {
+    id: 'xiaohongshu_pro',
+    name: '小红书私信',
+  },
+  {
+    id: 'zhihu',
+    name: '知乎',
+  },
+];
+
+// # 固定会传递的上下文参数
+// CTX_APP_NAME = "app_name"
+// CTX_APP_ID = "app_id"
+// CTX_INSTANCE_ID = "instance_id"
+
+// CTX_USERNAME = "username"
+// CTX_PLATFORM = "platform"
+// CTX_HAS_NEW_MESSAGE = "has_new_message"
+// CTX_HAS_GROUP_MESSAGE = "has_group_message"
+
+// # 千牛平台特有的上下文参数
+// CTX_CURRENT_GOODS = "CTX_CURRENT_GOODS"  # 当前商品
+// CTX_CURRENT_GOODS_ID = "CTX_CURRENT_GOODS_ID"  # 当前商品 ID
+// CTX_MEMBER_TAG = "CTX_MEMBER_TAG"  # 会员标签
+// CTX_FAN_TAG = "CTX_FAN_TAG"  # 粉丝标签
+// CTX_NEW_CUSTOMER_TAG = "CTX_NEW_CUSTOMER_TAG"  # 新客标签
+
+export const CTX_APP_NAME = 'app_name';
+export const CTX_APP_ID = 'app_id';
+export const CTX_INSTANCE_ID = 'instance_id';

src/main/backend/controllers/configController.ts

@@ -1,44 +1,37 @@
 import { Config } from '../entities/config';
+import { Plugin } from '../entities/plugin';
 
 export class ConfigController {
-  async getConfig(): Promise<Config> {
-    try {
-      const config = await Config.findByPk(1);
-      if (!config) {
-        throw new Error('Config not found');
-      }
+  async getByAppId(appId: string): Promise<Config | null> {
+    const config = await Config.findOne({
+      where: { platform_id: appId },
+    });
+    return config;
+  }
 
-      return config;
-    } catch (error) {
-      const newConfig = await Config.create({
-        extract_phone: true,
-        extract_product: true,
-        save_path: '',
-        reply_speed: 0,
-        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',
-        gpt_temperature: 0.7,
-        gpt_top_p: 0.9,
-        stream: true,
-        use_lazy: false,
-        lazy_key: 'your-key',
-      });
-      return newConfig;
-    }
+  async getByInstanceId(instanceId: string): Promise<Config | null> {
+    const config = await Config.findOne({
+      where: { instance_id: instanceId },
+    });
+    return config;
   }
 
-  async updateConfig(id: number, configData: Partial<Config>): Promise<Config> {
-    const config = await Config.findByPk(id);
+  async getGlobalConfig(): Promise<Config> {
+    let config = await Config.findOne({
+      where: { global: true },
+    });
+
     if (!config) {
-      throw new Error('Config not found');
+      config = await Config.create({
+        global: true,
+      });
     }
 
-    const data = await config.update(configData);
-    return data;
+    return config;
+  }
+
+  async getPluginConfig(pluginId: number): Promise<Plugin | null> {
+    const plugin = await Plugin.findByPk(pluginId);
+    return plugin;
   }
 }

src/main/backend/controllers/globalParamController.ts

@@ -1,25 +1,25 @@
 import ExcelJS from 'exceljs';
 import fs from 'fs';
 import { Op } from 'sequelize';
-import { AutoReply } from '../entities/autoReply';
+import { Keyword } from '../entities/keyword';
 import { ALL_PLATFORMS } from '../constants';
 import { getTempPath } from '../../utils';
 
 export class AutoReplyController {
   async create(autoReplyData: any) {
-    return AutoReply.create(autoReplyData);
+    return Keyword.create(autoReplyData);
   }
 
   async update(id: number, autoReplyData: any) {
-    const autoReply = await AutoReply.findByPk(id);
+    const autoReply = await Keyword.findByPk(id);
     if (!autoReply) {
       throw new Error('AutoReply not found');
     }
     await autoReply.update(autoReplyData);
   }
 
   async delete(id: number) {
-    const autoReply = await AutoReply.findByPk(id);
+    const autoReply = await Keyword.findByPk(id);
     if (!autoReply) {
       throw new Error('AutoReply not found');
     }
@@ -68,22 +68,22 @@ export class AutoReplyController {
       }
     });
 
-    const originalAutoReplies = await AutoReply.findAll();
+    const originalAutoReplies = await Keyword.findAll();
 
     try {
       // 先删除所有数据
-      await AutoReply.destroy({ where: {} });
-      await AutoReply.bulkCreate(autoReplies);
+      await Keyword.destroy({ where: {} });
+      await Keyword.bulkCreate(autoReplies);
     } catch (error) {
       // 如果插入失败，回滚数据
       // @ts-ignore
-      await AutoReply.bulkCreate(originalAutoReplies);
+      await Keyword.bulkCreate(originalAutoReplies);
       throw error;
     }
   }
 
   async exportExcel() {
-    const autoReplies = await AutoReply.findAll();
+    const autoReplies = await Keyword.findAll();
 
     const workbook = new ExcelJS.Workbook();
     const worksheet = workbook.addWorksheet('自动回复');
@@ -144,13 +144,13 @@ export class AutoReplyController {
   }
 
   async getKeywords(platformId: string) {
-    const autoReplies = await AutoReply.findAll({
+    const autoReplies = await Keyword.findAll({
       where: {
         platform_id: platformId,
       },
     });
 
-    const globalKeywords = await AutoReply.findAll({
+    const globalKeywords = await Keyword.findAll({
       where: {
         platform_id: {
           [Op.or]: [null, ''],
@@ -172,7 +172,7 @@ export class AutoReplyController {
   }) {
     try {
       const { rows: autoReplies, count: total } =
-        await AutoReply.findAndCountAll({
+        await Keyword.findAndCountAll({
           where: platformId
             ? {
                 platform_id: platformId,

src/main/backend/controllers/autoReplyController.ts renamed to src/main/backend/controllers/keywordReplyController.ts

@@ -17,9 +17,22 @@ export class SessionCreate {
 }
 
 export class SessionController {
-  async create(sessionData: SessionCreate) {
-    // @ts-ignore
-    return Session.create(sessionData);
+  async createSession(
+    platformId: string,
+    platformName: string,
+    username?: string,
+    goodsName?: string,
+    goodsAvatar?: string,
+  ): Promise<Session> {
+    return Session.create({
+      platform_id: platformId,
+      platform: platformName,
+      username: username || '',
+      last_active: new Date(),
+      goods_name: goodsName || '',
+      goods_avatar: goodsAvatar || '',
+      created_at: new Date(),
+    });
   }
 
   async update(id: number, sessionData: Session) {