wechaty-web-panel 1.5.7 → 1.6.0

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 ## 更新日志
 
+### V1.6.0(2023-04-22)
+1、添加chatgpt自定义配置，支持自定义模型，自定义token，自定义代理，不同群组使用不同token，定义不同角色
+2、添加角色预设
+
+### V1.5.8(2023-04-07)
+1、去除报错上报，防止无限请求
+2、限制二维码推送次数超过50次后需要重启服务才能在平台扫码
+
 ### V1.5.7(2023-04-06)
 1、去除only esm的包，解决chatgpt中文回复会中断问题
 2、robotId获取问题修复

package/dist/cjs/src/common/command.d.ts

@@ -0,0 +1,14 @@
+declare class Command {
+    constructor(name: any, handler: any);
+    name: any;
+    handler: any;
+}
+declare class CommandManager {
+    commands: Map<any, any>;
+    register(command: any): void;
+    execute(wxid: any, name: any, args: any): Promise<any>;
+    getDb(wxid: any): any;
+    dbs: Map<any, any> | undefined;
+}
+declare const manager: CommandManager;
+//# sourceMappingURL=command.d.ts.map

package/dist/cjs/src/common/command.js

@@ -0,0 +1,58 @@
+"use strict";
+class Command {
+    constructor(name, handler) {
+        this.name = name;
+        this.handler = handler;
+    }
+}
+class CommandManager {
+    constructor() {
+        this.commands = new Map();
+    }
+    register(command) {
+        this.commands.set(command.name, command);
+    }
+    async execute(wxid, name, args) {
+        const db = this.getDb(wxid);
+        const command = this.commands.get(name);
+        if (command) {
+            return await command.handler(args, db);
+        }
+        else {
+            console.log(`Unknown command: ${name}`);
+        }
+    }
+    getDb(wxid) {
+        if (!this.dbs) {
+            this.dbs = new Map();
+        }
+        if (!this.dbs.has(wxid)) {
+            const db = new Nedb({ filename: `path/to/db/${wxid}.db`, autoload: true });
+            this.dbs.set(wxid, db);
+        }
+        return this.dbs.get(wxid);
+    }
+}
+const manager = new CommandManager();
+manager.register(new Command('/rssadd', async (args, db) => {
+    const url = args[0];
+    const id = generateUniqueId();
+    await db.insert({ id, url });
+    return '订阅源已添加';
+}));
+manager.register(new Command('/rsstime', async (args, db) => {
+    const interval = args[0];
+    await db.update({}, { $set: { interval } }, { multi: true });
+    return '更新间隔已设置';
+}));
+manager.register(new Command('/rssls', async (args, db) => {
+    const sources = await db.find({});
+    const urls = sources.map(source => source.url).join('\n');
+    return urls;
+}));
+manager.register(new Command('/rssdel', async (args, db) => {
+    const id = args[0];
+    await db.remove({ id });
+    return '订阅源已删除';
+}));
+//# sourceMappingURL=command.js.map

package/dist/cjs/src/common/index.d.ts

@@ -41,11 +41,11 @@ export function getNewsContent(sortId: any, endWord?: any, num?: number): Promis
 /**
  * 更新用户信息
  */
-export function updateContactInfo(that: any): Promise<void>;
+export function updateContactInfo(that: any, noCache?: boolean): Promise<void>;
 /**
  * 更新群列表
  */
-export function updateRoomInfo(that: any): Promise<void>;
+export function updateRoomInfo(that: any, noCache?: boolean): Promise<void>;
 /**
  * 统一邀请加群
  * @param that

package/dist/cjs/src/common/index.js

@@ -62,8 +62,12 @@ exports.getCountDownContent = getCountDownContent;
 /**
  * 更新用户信息
  */
-async function updateContactInfo(that) {
+async function updateContactInfo(that, noCache = false) {
     try {
+        if (noCache && that.puppet.syncContact) {
+            await that.puppet.syncContact();
+            await (0, index_js_1.delay)(5000);
+        }
         const contactSelf = await (0, userDb_js_1.getUser)();
         const contactList = await that.Contact.findAll() || [];
         let res = [];
@@ -111,8 +115,12 @@ async function updateFriendInfo(list, num) {
 /**
  * 更新群列表
  */
-async function updateRoomInfo(that) {
+async function updateRoomInfo(that, noCache = false) {
     try {
+        if (noCache && that.puppet.syncContact) {
+            await that.puppet.syncContact();
+            await (0, index_js_1.delay)(5000);
+        }
         const contactSelf = await (0, userDb_js_1.getUser)();
         const roomList = await that.Room.findAll() || [];
         let res = [];
@@ -328,7 +336,7 @@ async function updateContactAndRoom(that) {
     await (0, index_js_1.delay)(3000);
     await (0, aibotk_js_1.asyncData)(contactSelf.robotId, 2);
     await (0, index_js_1.delay)(3000);
-    await updateRoomInfo(that);
+    await updateRoomInfo(that, true);
     await (0, index_js_1.delay)(3000);
     await updateContactInfo(that);
 }
@@ -342,7 +350,7 @@ async function updateContactOnly(that) {
     const contactSelf = await (0, userDb_js_1.getUser)();
     await (0, aibotk_js_1.asyncData)(contactSelf.robotId, 1);
     await (0, index_js_1.delay)(3000);
-    await updateContactInfo(that);
+    await updateContactInfo(that, true);
 }
 exports.updateContactOnly = updateContactOnly;
 /**
@@ -354,7 +362,7 @@ async function updateRoomOnly(that) {
     const contactSelf = await (0, userDb_js_1.getUser)();
     await (0, aibotk_js_1.asyncData)(contactSelf.robotId, 2);
     await (0, index_js_1.delay)(3000);
-    await updateRoomInfo(that);
+    await updateRoomInfo(that, true);
 }
 exports.updateRoomOnly = updateRoomOnly;
 exports.default = {

package/dist/cjs/src/common/reply.d.ts

@@ -8,7 +8,7 @@ export function getContactTextReply(that: any, contact: any, msg: any): Promise<
  * @param {*} name 发消息者昵称
  * @param {*} id 发消息者id
  */
-export function getRoomTextReply({ that, content, name, id, avatar, room, isMention }: any): Promise<number>;
+export function getRoomTextReply({ that, content, name, id, avatar, room, isMention, roomName, roomId }: any): Promise<number>;
 declare namespace _default {
     export { getContactTextReply };
     export { getRoomTextReply };

package/dist/cjs/src/common/reply.js

@@ -19,8 +19,8 @@ exports.getContactTextReply = getContactTextReply;
  * @param {*} name 发消息者昵称
  * @param {*} id 发消息者id
  */
-async function getRoomTextReply({ that, content, name, id, avatar, room, isMention }) {
-    let result = await msg_filter_service_js_1.default.filterRoomMsg({ that, msg: content, name, id, avatar, room, isMention });
+async function getRoomTextReply({ that, content, name, id, avatar, room, isMention, roomName, roomId }) {
+    let result = await msg_filter_service_js_1.default.filterRoomMsg({ that, msg: content, name, id, avatar, room, isMention, roomName, roomId });
     return result;
 }
 exports.getRoomTextReply = getRoomTextReply;

package/dist/cjs/src/db/aichatDb.d.ts

@@ -0,0 +1,27 @@
+/**
+ * 记录群聊天记录 记录格式
+ * { contactName: '', contactId: '', roomName: '', roomId: '', input: '输入的问题', output: '输出内容', time: '时间' }
+ * @param info
+ * @returns {Promise<unknown>}
+ */
+export function addAichatRecord(info: any): Promise<unknown>;
+/**
+ * 获取指定群的聊天记录
+ * @param room
+ * @returns {Promise<*>}
+ */
+export function getAichatRecord(query: any): Promise<any>;
+/**
+ * 清楚指定群的聊天记录
+ * @param roomName
+ * @returns {Promise<void>}
+ */
+export function removeRecord(query: any): Promise<void>;
+/**
+ * 获取指定群聊的所有聊天内容
+ * @param rooName
+ * @param day 取的天数
+ * @returns {Promise<*>}
+ */
+export function getAichatQuestion(query: any, day: any): Promise<any>;
+//# sourceMappingURL=aichatDb.d.ts.map

package/dist/cjs/src/db/aichatDb.js

@@ -0,0 +1,83 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getAichatQuestion = exports.removeRecord = exports.getAichatRecord = exports.addAichatRecord = void 0;
+const nedb_js_1 = __importDefault(require("./nedb.js"));
+const path_1 = __importDefault(require("path"));
+const os_1 = __importDefault(require("os"));
+const baseDir = path_1.default.join(os_1.default.homedir(), path_1.default.sep, ".wechaty", "wechaty-panel-cache", path_1.default.sep);
+const dbpath = baseDir + "aichat.db";
+const rdb = (0, nedb_js_1.default)(dbpath);
+console.log('聊天记录路径：如果开启了记录会存到此处，未开启不会记录，所有记录都是存在本地', dbpath);
+/**
+ * 记录群聊天记录 记录格式
+ * { contactName: '', contactId: '', roomName: '', roomId: '', input: '输入的问题', output: '输出内容', time: '时间' }
+ * @param info
+ * @returns {Promise<unknown>}
+ */
+async function addAichatRecord(info) {
+    try {
+        let doc = await rdb.insert(info);
+        return doc;
+    }
+    catch (error) {
+        console.log("插入数据错误", error);
+    }
+}
+exports.addAichatRecord = addAichatRecord;
+/**
+ * 获取指定群的聊天记录
+ * @param room
+ * @returns {Promise<*>}
+ */
+async function getAichatRecord(query) {
+    try {
+        let search = await rdb.find(query);
+        return search;
+    }
+    catch (error) {
+        console.log("查询数据错误", error);
+    }
+}
+exports.getAichatRecord = getAichatRecord;
+/**
+ * 清楚指定群的聊天记录
+ * @param roomName
+ * @returns {Promise<void>}
+ */
+async function removeRecord(query) {
+    try {
+        let search = await rdb.remove(query, { multi: true });
+        return search;
+    }
+    catch (e) {
+        console.log("error", e);
+    }
+}
+exports.removeRecord = removeRecord;
+/**
+ * 获取指定群聊的所有聊天内容
+ * @param rooName
+ * @param day 取的天数
+ * @returns {Promise<*>}
+ */
+async function getAichatQuestion(query, day) {
+    try {
+        let list = await getAichatRecord(query);
+        list = list.filter(item => {
+            return item.time >= new Date().getTime() - day * 24 * 60 * 60 * 1000;
+        });
+        let question = "";
+        list.forEach((item) => {
+            question = question + "|" + item.input;
+        });
+        return question;
+    }
+    catch (e) {
+        console.log("error", e);
+    }
+}
+exports.getAichatQuestion = getAichatQuestion;
+//# sourceMappingURL=aichatDb.js.map

package/dist/cjs/src/db/gptConfig.d.ts

@@ -0,0 +1,17 @@
+/**
+ * 存储gpt配置
+ * { contactName: '', contactId: '', roomName: '', roomId: '', input: '输入的问题', output: '输出内容', time: '时间' }
+ * @param info
+ * @returns {Promise<unknown>}
+ */
+export function addGptConfig(info: any): Promise<unknown>;
+/**
+ * 获取指定群的聊天记录
+ * @param room
+ * @returns {Promise<*>}
+ */
+export function getAllGptConfig(): Promise<any>;
+export function getGptConfigById(id: any): Promise<any>;
+export function updateAllGptConfig(infos: any): Promise<void>;
+export function updateOneGptConfig(id: any, info: any): Promise<any>;
+//# sourceMappingURL=gptConfig.d.ts.map

package/dist/cjs/src/db/gptConfig.js

@@ -0,0 +1,75 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.updateOneGptConfig = exports.updateAllGptConfig = exports.getGptConfigById = exports.getAllGptConfig = exports.addGptConfig = void 0;
+const nedb_js_1 = __importDefault(require("./nedb.js"));
+const path_1 = __importDefault(require("path"));
+const os_1 = __importDefault(require("os"));
+const baseDir = path_1.default.join(os_1.default.homedir(), path_1.default.sep, ".wechaty", "wechaty-panel-cache", path_1.default.sep);
+const dbpath = baseDir + "gptconfig.db";
+const rdb = (0, nedb_js_1.default)(dbpath);
+/**
+ * 存储gpt配置
+ * { contactName: '', contactId: '', roomName: '', roomId: '', input: '输入的问题', output: '输出内容', time: '时间' }
+ * @param info
+ * @returns {Promise<unknown>}
+ */
+async function addGptConfig(info) {
+    try {
+        let doc = await rdb.insert(info);
+        return doc;
+    }
+    catch (error) {
+        console.log("插入数据错误", error);
+    }
+}
+exports.addGptConfig = addGptConfig;
+/**
+ * 获取指定群的聊天记录
+ * @param room
+ * @returns {Promise<*>}
+ */
+async function getAllGptConfig() {
+    try {
+        let search = await rdb.find({});
+        return search;
+    }
+    catch (error) {
+        console.log("查询数据错误", error);
+    }
+}
+exports.getAllGptConfig = getAllGptConfig;
+async function getGptConfigById(id) {
+    try {
+        let search = await rdb.find({ id });
+        return search[0];
+    }
+    catch (error) {
+        console.log("查询数据错误", error);
+    }
+}
+exports.getGptConfigById = getGptConfigById;
+async function updateAllGptConfig(infos) {
+    try {
+        for (const item of infos) {
+            await rdb.update({ _id: item._id }, item, { upsert: true });
+        }
+    }
+    catch (error) {
+        console.log('更新失败', error);
+    }
+}
+exports.updateAllGptConfig = updateAllGptConfig;
+async function updateOneGptConfig(id, info) {
+    try {
+        let search = await rdb.update({ id }, { ...info });
+        return search;
+    }
+    catch (error) {
+        console.log("查询数据错误", error);
+    }
+}
+exports.updateOneGptConfig = updateOneGptConfig;
+//# sourceMappingURL=gptConfig.js.map

package/dist/cjs/src/handlers/on-error.js

@@ -1,17 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-const aibotk_js_1 = require("../proxy/aibotk.js");
 async function onError(error) {
-    try {
-        console.log('错误', error);
-        await (0, aibotk_js_1.sendError)(error.message
-            .replace(/\ +/g, '')
-            .replace(/[\r\n]/g, '')
-            .replace('Error:type(){returnthis._type;}', ''));
-    }
-    catch (e) {
-        console.log('上报错误失败', e);
-    }
+    console.log('捕捉到🐛，如果还能正常运行，可以忽略', error);
 }
 exports.default = onError;
 //# sourceMappingURL=on-error.js.map

package/dist/cjs/src/handlers/on-login.js

@@ -1,6 +1,5 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-const index_js_1 = require("../lib/index.js");
 const aibotk_js_1 = require("../proxy/aibotk.js");
 const userDb_js_1 = require("../db/userDb.js");
 const mqtt_js_1 = require("../proxy/mqtt.js");
@@ -14,7 +13,6 @@ async function onLogin(user) {
         console.log(`贴心助理${user}登录了`);
         await (0, aibotk_js_1.updatePanelVersion)();
         await (0, aibotk_js_1.setQrCode)('', 4);
-        await (0, aibotk_js_1.sendError)('');
         await (0, aibotk_js_1.getConfig)(); // 获取配置文件
         const config = await (0, configDb_js_1.allConfig)();
         const { userId } = config && config.userInfo;

package/dist/cjs/src/handlers/on-message.js

@@ -146,8 +146,10 @@ async function dispatchRoomFilterByMsgType(that, room, msg) {
                     content,
                     name: contactName,
                     id: contactId,
+                    roomId: room.id,
                     avatar: contactAvatar,
                     room,
+                    roomName,
                     isMention: mentionSelf
                 });
                 for (let reply of replys) {
@@ -158,7 +160,7 @@ async function dispatchRoomFilterByMsgType(that, room, msg) {
                 if (role === "vip" && cloudRoom.includes(roomName) && !checkIgnore(content, ignoreRecord)) {
                     const regex = /(<([^>]+)>)/ig;
                     content = content.replace(regex, "");
-                    (0, roomDb_js_1.addRoomRecord)({
+                    void (0, roomDb_js_1.addRoomRecord)({
                         roomName,
                         roomId: room.id,
                         content,

package/dist/cjs/src/handlers/on-ready.js

@@ -18,6 +18,9 @@ async function onReady() {
         await (0, userDb_js_1.getUser)();
         console.log(`所有数据准备完毕`);
         (0, aibotk_js_1.sendHeartBeat)('live');
+        if (this.puppet.syncContact) {
+            await this.puppet.syncContact();
+        }
         await (0, index_js_2.delay)(5000);
         index_js_1.default.updateContactInfo(this);
         await (0, index_js_2.delay)(5000);

package/dist/cjs/src/handlers/on-scan.js

@@ -6,13 +6,23 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const qrcode_terminal_1 = __importDefault(require("qrcode-terminal"));
 const index_js_1 = require("../lib/index.js");
 const aibotk_js_1 = require("../proxy/aibotk.js");
+const aiDb_js_1 = require("../db/aiDb.js");
+// 限制推送二维码的次数，防止掉线后，无限推送二维码到服务器
+let scanTime = 0;
 /**
  * 扫描登录，显示二维码
  */
 async function onScan(qrcode, status) {
+    const aibotConfig = await (0, aiDb_js_1.getAibotConfig)();
     qrcode_terminal_1.default.generate(qrcode);
     console.log('扫描状态', status);
-    (0, index_js_1.throttle)((0, aibotk_js_1.setQrCode)(qrcode, status), 30000);
+    if (scanTime >= aibotConfig.scanTimes) {
+        console.log('长时间推送登录状态，平台二维码不再更新，请重启服务，或直接在终端扫码登录');
+    }
+    else {
+        scanTime++;
+        (0, index_js_1.throttle)((0, aibotk_js_1.setQrCode)(qrcode, status), 30000);
+    }
     const qrImgUrl = ['https://api.qrserver.com/v1/create-qr-code/?data=', encodeURIComponent(qrcode)].join('');
     console.log(qrImgUrl);
 }

package/dist/cjs/src/index.d.ts

@@ -1,5 +1,6 @@
 export function WechatyWebPanelPlugin(config?: {
     apiKey: any;
     apiSecret: any;
+    scanTimes: any;
 }): (bot: any) => void;
 //# sourceMappingURL=index.d.ts.map

package/dist/cjs/src/index.js

@@ -35,7 +35,7 @@ if (process.env['AIBOTK_SECRET']) {
     console.log('使用环境变量中的 aibotkSecret');
     envSecret = process.env['AIBOTK_SECRET'];
 }
-function WechatyWebPanelPlugin(config = { apiKey, apiSecret }) {
+function WechatyWebPanelPlugin(config = { apiKey, apiSecret, scanTimes }) {
     const initConfig = {
         apiKey: envKey || config.apiKey,
         apiSecret: envSecret || config.apiSecret,
@@ -43,6 +43,7 @@ function WechatyWebPanelPlugin(config = { apiKey, apiSecret }) {
         ignoreMessages: config.ignoreMessages || [],
         // 需要忽略的事件 ['scan', 'login', 'logout', 'friendship', 'room-join', 'room-topic', 'room-leave', 'message', 'ready', 'heartbeat', 'error']
         ignoreEvents: config.ignoreEvents || [],
+        scanTimes: config.scanTimes || 50
     };
     (0, aiDb_js_1.addAibotConfig)(initConfig);
     return function (bot) {

package/dist/cjs/src/lib/contentCensor.d.ts

@@ -0,0 +1,8 @@
+export class ContentCensor {
+    constructor(config: any);
+    client: any;
+    config: any;
+    init(): void;
+    checkText(text: any): Promise<boolean | undefined>;
+}
+//# sourceMappingURL=contentCensor.d.ts.map

package/dist/cjs/src/lib/contentCensor.js

@@ -0,0 +1,31 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ContentCensor = void 0;
+const baidu_aip_sdk_1 = __importDefault(require("baidu-aip-sdk"));
+const wechaty_1 = require("wechaty");
+class ContentCensor {
+    constructor(config) {
+        this.client = null;
+        this.config = config;
+    }
+    init() {
+        if (this.config.type === 1) {
+            this.client = new baidu_aip_sdk_1.default.contentCensor(this.config.appId, this.config.apiKey, this.config.secretKey);
+        }
+    }
+    async checkText(text) {
+        if (!this.client) {
+            this.init();
+        }
+        if (this.config.type === 1) {
+            const result = await this.client.textCensorUserDefined(text);
+            console.log('result', result);
+            return result.conclusionType ? result.conclusionType === 1 : true;
+        }
+    }
+}
+exports.ContentCensor = ContentCensor;
+//# sourceMappingURL=contentCensor.js.map

package/dist/cjs/src/lib/officialOpenAi.d.ts

@@ -0,0 +1,67 @@
+export default OfficialOpenAi;
+declare class OfficialOpenAi {
+    constructor(config?: {
+        token: string;
+        debug: number;
+        proxyPass: string;
+        proxyUrl: string;
+        showQuestion: boolean;
+        timeoutMs: number;
+        model: string;
+        promotId: string;
+        systemMessage: string;
+    });
+    chatGPT: {
+        _apiKey: any;
+        _apiOrg: any;
+        _apiBaseUrl: any;
+        _debug: boolean;
+        _fetch: any;
+        _completionParams: any;
+        _systemMessage: any;
+        _maxModelTokens: any;
+        _maxResponseTokens: any;
+        _getMessageById: any;
+        _upsertMessage: any;
+        _messageStore: any;
+        sendMessage(text: any, opts?: {}): Promise<any>;
+        apiKey: any;
+        apiOrg: any;
+        _buildMessages(text: any, opts: any): Promise<{
+            messages: {
+                role: string;
+                content: any;
+            }[];
+            maxTokens: number;
+            numTokens: number;
+        }>;
+        _getTokenCount(text: any): Promise<number>;
+        _defaultGetMessageById(id: any): Promise<any>;
+        _defaultUpsertMessage(message: any): Promise<void>;
+    } | null;
+    config: {
+        token: string;
+        debug: number;
+        proxyPass: string;
+        proxyUrl: string;
+        showQuestion: boolean;
+        timeoutMs: number;
+        model: string;
+        promotId: string;
+        systemMessage: string;
+    };
+    contentCensor: ContentCensor | null;
+    chatOption: {};
+    init(): Promise<void>;
+    /**
+     * 重置apikey
+     * @return {Promise<void>}
+     */
+    reset(): Promise<void>;
+    getReply(content: any, uid: any, adminId?: string): Promise<{
+        type: number;
+        content: any;
+    }[]>;
+}
+import { ContentCensor } from "./contentCensor.js";
+//# sourceMappingURL=officialOpenAi.d.ts.map