wechaty-web-panel 1.6.111 → 1.6.113

package/dist/proxy/aibotk.js

@@ -0,0 +1,829 @@
+import { aiBotReq, req } from './superagent.js';
+import { updateConfig } from '../db/configDb.js';
+import { packageJson } from '../package-json.js';
+import { updateAllRssConfig, resetRssData } from "../db/rssConfig.js";
+import { getPuppetEol } from "../const/puppet-type.js";
+import globalConfig from "../db/global.js";
+/**
+ * 获取美女图片
+ */
+async function getMeiNv() {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/meinv',
+            params: {},
+        };
+        let content = await aiBotReq(option);
+        let pics = content.data.pics || [];
+        if (pics.length) {
+            let url = pics[0];
+            return url.includes('.jpg') ? url : 'https://tva2.sinaimg.cn/large/0072Vf1pgy1foxkcsx9rmj31hc0u0h9k.jpg';
+        }
+    }
+    catch (e) {
+        console.log('获取美女图片失败', e);
+        return 'https://tva2.sinaimg.cn/large/0072Vf1pgy1foxkcsx9rmj31hc0u0h9k.jpg';
+    }
+}
+/**
+ * 获取配置词云的所有群名
+ */
+export async function getWordCloudRoom() {
+    try {
+        let option = {
+            method: 'get',
+            url: '/wordcloudroom',
+            params: {},
+        };
+        let content = await aiBotReq(option);
+        const roomNames = content.data.map(item => item.roomName);
+        return roomNames || [];
+    }
+    catch (e) {
+        console.log('群词云配置拉取失败', e);
+        return [];
+    }
+}
+/**
+ * 获取群合影配置
+ */
+export async function getWordCloudConfig(roomName) {
+    try {
+        let option = {
+            method: 'get',
+            url: '/roomCloud',
+            params: { name: roomName },
+        };
+        let content = await aiBotReq(option);
+        return content.data || '';
+    }
+    catch (e) {
+        console.log('群词云配置拉取失败', e);
+    }
+}
+/**
+ * 获取每日一句
+ */
+async function getOne() {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/one',
+            params: {},
+            timeout: 5 * 60 * 1000
+        };
+        let content = await aiBotReq(option);
+        let word = content.data.word || '今日一句似乎已经消失';
+        return word;
+    }
+    catch (e) {
+        console.log('获取每日一句失败', e);
+        return '今日一句似乎已经消失';
+    }
+}
+/**
+ * 获取火灾新闻
+ */
+async function getFireNews(id, num) {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/firenews',
+            params: {
+                id,
+                num
+            },
+        };
+        let content = await aiBotReq(option);
+        let newList = content.data || [];
+        let news = '';
+        const eol = await getPuppetEol();
+        for (let i in newList) {
+            let num = parseInt(i) + 1;
+            const url = newList[i].shortUrl ? newList[i].shortUrl : newList[i].url;
+            news = `${news}${eol}${num}.${newList[i].title}${url ? `${eol}${url}${eol}` : `${eol}`}`;
+        }
+        const endMap = {
+            1001: '您可以 @消防小助手+新闻标题，通过ChatGPT为您分析时事新闻！',
+            1002: '您可以 @消防小助手+新闻标题，通过ChatGPT为您分析今日消防行业招标信息！',
+        };
+        return `${news}…………………………${eol}${eol}${endMap[id] || ''}`;
+    }
+    catch (e) {
+        console.log('获取每日一句失败', e);
+        return '今日一句似乎已经消失';
+    }
+}
+/**
+ * 获取自定义内容
+ * @param id
+ * @param taskId
+ * @returns {Promise<[{type: number, content: string}]|*[]>}
+ */
+export async function getCustomNews(id, taskId) {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/customnews',
+            params: {
+                id,
+                taskId
+            },
+            timeout: 5 * 60 * 1000
+        };
+        let content = await aiBotReq(option);
+        let newContent = content.data || [];
+        return newContent;
+    }
+    catch (e) {
+        console.log('定制内容获取失败', e);
+        return [{ type: 1, content: '定制内容获取失败' }];
+    }
+}
+/**
+ * 获取自定义技能
+ * @param params
+ * @returns {Promise<[{type: number, content: string}]|*[]>}
+ */
+export async function getCustomEvents(params) {
+    try {
+        let option = {
+            method: 'POST',
+            url: '/customevent',
+            params,
+            timeout: 5 * 60 * 1000
+        };
+        let content = await aiBotReq(option);
+        let newContent = content.data || [];
+        return newContent;
+    }
+    catch (e) {
+        console.log('自定义技能获取失败', e);
+        return [{ type: 1, content: '自定义技能获取失败' }];
+    }
+}
+/**
+ * 获取配置文件
+ * @returns {Promise<*>}
+ */
+async function getConfig() {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/wechat/config',
+            params: {},
+        };
+        let content = await aiBotReq(option);
+        const config = JSON.parse(content.data.config);
+        let cloudRoom = [];
+        if (config.userInfo.role === 'vip') {
+            await getGptConfig();
+            await getRssConfig();
+            await getTasks();
+            cloudRoom = await getWordCloudRoom();
+        }
+        let cres = await updateConfig({
+            puppetType: 'wechaty-puppet-wechat',
+            no_remind: false, // 不需要开启提醒
+            botScope: 'all',
+            parseMini: false,
+            openaiSystemMessage: '',
+            showQuestion: true,
+            openaiTimeout: 60,
+            chatDelay: 2,
+            openaiAccessToken: '',
+            openaiDebug: false,
+            openaiModel: 'gpt-3.5-turbo',
+            temperature: 0.8,
+            top_p: 1,
+            presence_penalty: 1,
+            maxToken: null,
+            modelMaxToken: null,
+            cozev3_token: '',
+            cozev3_botId: '',
+            cozev3_baseUrl: '',
+            qany_botId: '',
+            qany_token: '',
+            qany_baseUrl: '',
+            dify_token: '',
+            dify_baseUrl: '',
+            difyAgent: false,
+            coze_token: '',
+            coze_botId: '',
+            coze_baseUrl: '',
+            coze_showSuggestions: false,
+            proxyUrl: '',
+            proxyPassUrl: '',
+            chatFilter: 0,
+            filterType: 1, // 过滤引擎类型 1 百度文本审核
+            filterAppid: '',
+            filterApiKey: '',
+            filterSecretKey: '',
+            countDownTaskSchedule: [],
+            parseMiniRooms: [],
+            preventLength: 1000,
+            preventWords: '',
+            customBot: null,
+            roomAt: 1,
+            friendNoReplyInRoom: 0,
+            defaultReply: '',
+            forwards: [],
+            openRecord: false,
+            openWhisper: false,
+            whisperConfig: {},
+            ignoreRoomMentionAll: true,
+            noNeedAt: false,
+            uploadFileConfig: null,
+            ignoreFiles: false,
+            filterLinkContent: false,
+            ...config,
+            cloudRoom
+        });
+        return cres;
+    }
+    catch (e) {
+        console.log('获取配置文件失败:' + e);
+    }
+}
+/**
+ * 获取gpt配置
+ * @return {Promise<*>}
+ */
+export async function getGptConfig() {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/gpt/config',
+            params: {},
+        };
+        let content = await aiBotReq(option);
+        if (content.data) {
+            const list = content.data.map(item => ({ ...item, _id: item.id }));
+            globalConfig.updateAllGptConfig(list);
+        }
+    }
+    catch (error) {
+        console.log('获取gpt配置文件失败:' + error);
+    }
+}
+/**
+ * 获取批量任务
+ * @return {Promise<*>}
+ */
+export async function getTasks() {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/user/task',
+            params: {},
+        };
+        let content = await aiBotReq(option);
+        if (content.data) {
+            const list = content.data.map(item => ({ ...item, _id: item.id }));
+            globalConfig.updateAllTasks(list);
+        }
+    }
+    catch (error) {
+        console.log('获取批量任务失败:' + error);
+    }
+}
+/**
+ * 获取rss配置
+ * @return {Promise<*>}
+ */
+export async function getRssConfig() {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/rss/config',
+            params: {},
+        };
+        let content = await aiBotReq(option);
+        if (content.data) {
+            const list = content.data.map(item => ({ ...item, _id: item.id }));
+            resetRssData();
+            await updateAllRssConfig(list);
+        }
+    }
+    catch (error) {
+        console.log('获取rss配置文件失败:' + error);
+    }
+}
+/**
+ * 更新rss最后一条内容的配置
+ * @return {Promise<*>}
+ */
+async function updateRssLast(id, lastcontent) {
+    try {
+        console.log('id----', id, lastcontent);
+        let option = {
+            method: 'POST',
+            url: `/rss/lastcontent/${id}`,
+            params: {
+                content: lastcontent
+            },
+        };
+        let content = await aiBotReq(option);
+        return content;
+    }
+    catch (error) {
+        console.log('获取rss配置文件失败:' + error);
+    }
+}
+/**
+ * 更新rss最后一条内容的配置
+ * @return {Promise<*>}
+ */
+async function getRssLast(id) {
+    try {
+        let option = {
+            method: 'GET',
+            url: `/rss/lastcontent/${id}`,
+            params: {},
+        };
+        let content = await aiBotReq(option);
+        return content.data;
+    }
+    catch (error) {
+        console.log('获取rss配置文件失败:' + error);
+    }
+}
+/**
+ * 更新对话次数
+ * @param id
+ * @param num
+ * @return {Promise<*>}
+ */
+async function updateChatRecord(id, num) {
+    try {
+        let option = {
+            method: 'POST',
+            url: '/gpt/config',
+            params: {
+                id,
+                usedNum: num
+            },
+        };
+        let content = await aiBotReq(option);
+        return content.data;
+    }
+    catch (error) {
+        console.log('更新对话次数' + error);
+    }
+}
+/**
+ * 获取promotion信息
+ * @param id
+ * @return {Promise<*>}
+ */
+async function getPromotInfo(id) {
+    try {
+        let option = {
+            method: 'get',
+            url: '/promot/info',
+            params: {
+                id
+            },
+        };
+        let content = await aiBotReq(option);
+        return content.data;
+    }
+    catch (e) {
+        console.log("catch error:" + e);
+    }
+}
+/**
+ * 获取输入的验证码
+ * @param id
+ * @return {Promise<*>}
+ */
+async function getVerifyCode() {
+    try {
+        let option = {
+            method: 'get',
+            url: '/worker/verifycode',
+        };
+        let content = await aiBotReq(option);
+        if (content.data.code) {
+            globalConfig.setVerifyCode(content.data.code);
+        }
+    }
+    catch (e) {
+        console.log("catch error:" + e);
+    }
+}
+/**
+ * 清除使用过的验证码
+ * @returns {Promise<*>}
+ */
+async function clearVerifyCode() {
+    try {
+        let option = {
+            method: 'get',
+            url: '/worker/clearverifycode',
+        };
+        await aiBotReq(option);
+        globalConfig.setVerifyCode('');
+    }
+    catch (e) {
+        console.log("catch error:" + e);
+    }
+}
+/**
+ * 获取定时提醒任务列表
+ */
+async function getScheduleList() {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/task',
+            params: {},
+        };
+        let content = await aiBotReq(option);
+        let scheduleList = content.data;
+        console.log('获取定时任务成功:' + scheduleList);
+        return scheduleList;
+    }
+    catch (error) {
+        console.log('获取定时任务失败:' + error);
+    }
+}
+/**
+ * 设置定时提醒任务
+ * @param {*} obj 任务详情
+ * @returns {*} 任务详情
+ */
+async function setSchedule(obj) {
+    try {
+        let option = {
+            method: 'POST',
+            url: '/task',
+            params: obj,
+        };
+        let content = await aiBotReq(option);
+        return content.data;
+    }
+    catch (error) {
+        console.log('添加定时任务失败', error);
+    }
+}
+/**
+ * 更新定时提醒任务
+ */
+async function updateSchedule(id) {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/task/update',
+            params: { id: id },
+        };
+        let content = await aiBotReq(option);
+        console.log('更新定时任务成功');
+    }
+    catch (error) {
+        console.log('更新定时任务失败', error);
+    }
+}
+/**
+ * 登录二维码推送
+ * @param url
+ * @param status
+ * @returns {Promise<void>}
+ */
+async function setQrCode(url, status) {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/wechat/qrcode',
+            params: { qrUrl: url, qrStatus: status },
+        };
+        let content = await aiBotReq(option);
+        if (content) {
+            console.log('推送二维码成功');
+        }
+        else {
+            console.log('推送登录二维码失败');
+        }
+    }
+    catch (error) {
+        console.log('推送登录二维码失败', error);
+    }
+}
+/**
+ * 更新验证信息
+ * @param messages
+ * @returns {Promise<void>}
+ */
+async function setVerifyInfo(messages) {
+    try {
+        let option = {
+            method: 'POST',
+            url: '/wechat/setSecurityInfo',
+            params: { verifyMessage: messages },
+        };
+        let content = await aiBotReq(option);
+        if (content) {
+            console.log('推送验证信息成功');
+        }
+        else {
+            console.log('推送验证信息失败');
+        }
+    }
+    catch (error) {
+        console.log('推送验证信息失败', error);
+    }
+}
+/**
+ * 清除验证信息
+ * @param messages
+ * @returns {Promise<void>}
+ */
+async function clearVerifyInfo(messages) {
+    try {
+        let option = {
+            method: 'POST',
+            url: '/wechat/clearSecurityInfo',
+            params: { verifyMessage: '' },
+        };
+        let content = await aiBotReq(option);
+        if (content) {
+            console.log('清除验证信息成功');
+        }
+        else {
+            console.log('清除验证信息失败');
+        }
+    }
+    catch (error) {
+        console.log('清除验证信息失败', error);
+    }
+}
+/**
+ * 推送登录状态的心跳
+ * @param heart
+ * @returns {Promise<void>}
+ */
+async function sendHeartBeat(heart) {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/wechat/heart',
+            params: { heartBeat: heart },
+        };
+        let content = await aiBotReq(option);
+    }
+    catch (error) {
+        console.log('推送心跳失败', error);
+    }
+}
+/**
+ * 更新头像
+ * @returns {Promise<void>}
+ * @param url
+ * @param info 用户基本信息
+ */
+async function sendRobotInfo(url, name, id) {
+    try {
+        let option = {
+            method: 'POST',
+            url: '/wechat/info',
+            params: { avatar: url, robotName: name, robotId: id },
+        };
+        let content = await aiBotReq(option);
+    }
+    catch (error) {
+        console.log('推送头像失败', error);
+    }
+}
+/**
+ * 更新好友
+ * @returns {Promise<void>}
+ * @param url
+ */
+async function sendFriend(friend) {
+    try {
+        let option = {
+            method: 'POST',
+            url: '/wechat/friend',
+            params: { friend: friend },
+        };
+        let content = await aiBotReq(option);
+        if (!content.code === 200) {
+            console.log('推送失败', content.msg);
+        }
+    }
+    catch (error) {
+        console.log('推送好友列表失败');
+    }
+}
+/**
+ * 更新群
+ * @returns {Promise<void>}
+ * @param url
+ */
+async function sendRoom(room) {
+    try {
+        let option = {
+            method: 'POST',
+            url: '/wechat/room',
+            params: { room: room },
+        };
+        let content = await aiBotReq(option);
+        if (!content.code === 200) {
+            console.log('推送失败', content.msg);
+        }
+    }
+    catch (error) {
+        console.log('推送群列表失败', error);
+    }
+}
+/**
+ * 同步群和好友列表
+ * type： 1 好友 2 群组
+ */
+async function asyncData(robotId, type) {
+    try {
+        let option = {
+            method: 'get',
+            url: '/wechat/asyncData',
+            params: { type, robotId },
+        };
+        let content = await aiBotReq(option);
+    }
+    catch (error) {
+        console.log('同步好友列表失败', error);
+    }
+}
+/**
+ * 获取上传token
+ * @returns {Promise<*>}
+ */
+async function getQiToken() {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/wechat/qitoken',
+            params: {},
+            platform: 'qi',
+        };
+        let content = await aiBotReq(option);
+        return content.data.token;
+    }
+    catch (e) {
+        console.log('token error', e);
+    }
+}
+/**
+ * 上传base64图片到七牛云
+ * @param base
+ * @param name
+ * @returns {Promise<void>}
+ */
+async function putqn(base, name) {
+    try {
+        const token = await getQiToken();
+        const namebase = Buffer.from(name).toString('base64').replace(/=/g, '');
+        let filename = 'wechat/avatar/' + namebase + '.jpeg';
+        let base_file_name = Buffer.from(filename).toString('base64').replace('+', '-').replace('/', '_');
+        let options = {
+            method: 'POST',
+            url: 'http://upload.qiniup.com/putb64/-1/key/' + base_file_name,
+            contentType: 'application/octet-stream',
+            authorization: 'UpToken ' + token,
+            params: base,
+            platform: 'chuan',
+        };
+        let content = await req(options);
+        console.log('上传结果', content.key);
+        return 'https://img.aibotk.com/' + content.key;
+    }
+    catch (e) {
+        console.log('上传失败', e.Error);
+    }
+}
+/**
+ * 更新插件版本信息
+ * @param {*} version
+ */
+async function updatePanelVersion() {
+    try {
+        let option = {
+            method: 'POST',
+            url: '/webPanel/version',
+            params: { version: packageJson.version || '' },
+        };
+        console.log('更新插件版本号', packageJson.version);
+        let content = await aiBotReq(option);
+        return content.data;
+    }
+    catch (error) {
+        console.log('error', error);
+    }
+}
+/**
+ * 更新插件版本信息
+ * @param {*} version
+ */
+export async function getPanelVersion() {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/webPanel/version',
+        };
+        let content = await aiBotReq(option);
+        return content.data.version;
+    }
+    catch (error) {
+        console.log('error', error);
+    }
+}
+/**
+ * 获取mqtt信息
+ * @param {*} version
+ */
+async function getMqttConfig() {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/ws/mqtt/config',
+            params: {},
+        };
+        let content = await aiBotReq(option);
+        return content.data;
+    }
+    catch (error) {
+        console.log('获取mqtt配置错误', error);
+    }
+}
+/**
+ * 获取实时素材
+ * @param {*} version
+ */
+async function getMaterial(id) {
+    try {
+        let option = {
+            method: 'GET',
+            url: '/wechat/material',
+            params: {
+                id,
+            },
+        };
+        let content = await aiBotReq(option);
+        console.log('素材', content.data);
+        return content.data;
+    }
+    catch (error) {
+        console.log('获取mqtt配置错误', error);
+    }
+}
+export { getVerifyCode };
+export { clearVerifyCode };
+export { getConfig };
+export { getScheduleList };
+export { setSchedule };
+export { updateSchedule };
+export { setQrCode };
+export { sendHeartBeat };
+export { sendRobotInfo };
+export { putqn };
+export { sendFriend };
+export { sendRoom };
+export { asyncData };
+export { updatePanelVersion };
+export { getMqttConfig };
+export { getMeiNv };
+export { getOne };
+export { getMaterial };
+export { getFireNews };
+export { updateChatRecord };
+export { getPromotInfo };
+export { getRssLast };
+export { updateRssLast };
+export { clearVerifyInfo };
+export { setVerifyInfo };
+export default {
+    getConfig,
+    getScheduleList,
+    setSchedule,
+    updateSchedule,
+    setQrCode,
+    sendHeartBeat,
+    sendRobotInfo,
+    putqn,
+    sendFriend,
+    sendRoom,
+    asyncData,
+    updatePanelVersion,
+    getMqttConfig,
+    getMeiNv,
+    getOne,
+    getMaterial,
+    getFireNews,
+    clearVerifyCode,
+    getRssLast,
+    updateRssLast,
+    getVerifyCode,
+    clearVerifyInfo,
+    setVerifyInfo
+};
+//# sourceMappingURL=aibotk.js.map