koishi-plugin-dew-bot 1.0.188

package/lib/plugin/puppeteer.js

@@ -0,0 +1,81 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const fs_1 = __importDefault(require("fs"));
+const koishi_1 = require("koishi");
+const path = require('path');
+const console_1 = require("./console");
+class puppeteer extends console_1.console {
+    commonCss;
+    pageContents = new Map();
+    ctx;
+    tempDir;
+    init(ctx, config) {
+        super.init(ctx);
+        this.ctx = ctx;
+        this.log('初始化 puppeteer');
+        const pagesDir = path.resolve(__dirname, '../html/page');
+        const files = fs_1.default.readdirSync(pagesDir);
+        this.tempDir = path.resolve(path.resolve(__dirname, '../html/'), 'temp');
+        files.forEach(file => {
+            const filePath = path.join(pagesDir, file);
+            if (fs_1.default.lstatSync(filePath).isFile()) {
+                let content = fs_1.default.readFileSync(filePath, 'utf-8');
+                // 移除HTML注释
+                content = content.replace(/<!--[\s\S]*?-->/g, '');
+                let key = file.replace('.html', '');
+                this.pageContents.set(key, content);
+            }
+        });
+        // 如果tempDir不存在，则创建
+        if (!fs_1.default.existsSync(this.tempDir)) {
+            fs_1.default.mkdirSync(this.tempDir, { recursive: true });
+        }
+        else {
+            // 删除tempDir下的所有文件
+            fs_1.default.readdirSync(this.tempDir).forEach(file => {
+                fs_1.default.unlinkSync(path.resolve(this.tempDir, file));
+            });
+        }
+    }
+    async render(name, data) {
+        let tempHtml = this.pageContents.get(name).replace(/let data = \{[^]*?\};/, `let data = ${JSON.stringify(data)};`);
+        // 随机生成一个文件名
+        let randomName = `${name}_${Math.random().toString(36).substring(2, 15)}.html`;
+        const savePath = path.resolve(this.tempDir, randomName);
+        fs_1.default.writeFileSync(savePath, tempHtml, 'utf-8');
+        const page = await this.ctx.puppeteer.page();
+        const filePath = path.resolve(this.tempDir, randomName);
+        await page.goto(`file://${filePath}`, { waitUntil: 'networkidle2' });
+        const leaderboardElement = await page.$('body');
+        const boundingBox = await leaderboardElement.boundingBox();
+        if (boundingBox.height > 0 || boundingBox.width > 0) {
+            await page.setViewport({
+                width: Math.ceil(boundingBox.width),
+                height: Math.ceil(boundingBox.height),
+            });
+        }
+        try {
+            const imgBuf = await leaderboardElement.screenshot({ captureBeyondViewport: false, type: 'jpeg' });
+            let sendBuff = new Uint8Array(imgBuf);
+            page.close();
+            let leaderboardImage;
+            // if (sendBuff.length < 200 * 1024) { // 如果小于200kb
+            this.log(`not compress ${sendBuff.length / 1024}KB`);
+            leaderboardImage = koishi_1.h.image(sendBuff, 'image/jpeg');
+            // } else {
+            // let req = await server.api('open/CompressImg', { imgBuf: sendBuff });
+            // this.log(`compress ${req.imgBuf.length / 1024}KB`);
+            // leaderboardImage = h.image(req.imgBuf, 'image/jpeg');
+            // }
+            return leaderboardImage;
+        }
+        catch (error) {
+            page.close();
+            this.log(error);
+        }
+    }
+}
+exports.default = new puppeteer();

package/lib/plugin/rateLimiter.js

@@ -0,0 +1,104 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RateLimiter = void 0;
+class RateLimiter {
+    userMessageTimes = new Map();
+    blacklist = new Map();
+    cleanupInterval = null;
+    blacklistDuration; // 拉黑时长（分钟）
+    maxMessagesPerMinute = 60; // 每分钟最大消息数
+    ctx;
+    constructor(blacklistDurationMinutes = 5) {
+        this.blacklistDuration = blacklistDurationMinutes * 60 * 1000; // 转换为毫秒
+    }
+    // 设置 Koishi 上下文
+    setContext(ctx) {
+        this.ctx = ctx;
+    }
+    /**
+     * 检查用户是否可以发送消息
+     */
+    canSendMessage(userId) {
+        const now = Date.now();
+        // 检查是否在黑名单中
+        if (this.isBlacklisted(userId)) {
+            return false;
+        }
+        // 获取用户的消息时间记录
+        const userTimes = this.userMessageTimes.get(userId) || [];
+        // 清理1分钟之前的记录
+        const oneMinuteAgo = now - 60 * 1000;
+        const recentMessages = userTimes.filter(time => time > oneMinuteAgo);
+        // 检查是否超过频率限制
+        if (recentMessages.length >= this.maxMessagesPerMinute) {
+            // 拉黑用户
+            this.blacklist.set(userId, now + this.blacklistDuration);
+            return false;
+        }
+        // 记录这次消息时间
+        recentMessages.push(now);
+        this.userMessageTimes.set(userId, recentMessages);
+        return true;
+    }
+    /**
+     * 检查用户是否被拉黑
+     */
+    isBlacklisted(userId) {
+        const blacklistEnd = this.blacklist.get(userId);
+        if (!blacklistEnd)
+            return false;
+        const now = Date.now();
+        if (now > blacklistEnd) {
+            // 拉黑时间已过，移除黑名单
+            this.blacklist.delete(userId);
+            return false;
+        }
+        return true;
+    }
+    /**
+   * 启动定期清理
+   */
+    startCleanup() {
+        if (this.ctx) {
+            const scheduleNext = () => {
+                this.cleanupInterval = this.ctx?.setTimeout(() => {
+                    this.cleanup();
+                    if (this.cleanupInterval) {
+                        scheduleNext(); // 递归调度下一次清理
+                    }
+                }, 5 * 60 * 1000); // 每5分钟清理一次
+            };
+            scheduleNext();
+        }
+    }
+    /**
+     * 停止定期清理
+     */
+    stopCleanup() {
+        this.cleanupInterval = null; // 停止递归调度
+    }
+    /**
+     * 清理过期数据
+     */
+    cleanup() {
+        const now = Date.now();
+        const oneHourAgo = now - 60 * 60 * 1000;
+        // 清理消息时间记录
+        for (const [userId, times] of this.userMessageTimes.entries()) {
+            const recentTimes = times.filter(time => time > oneHourAgo);
+            if (recentTimes.length === 0) {
+                this.userMessageTimes.delete(userId);
+            }
+            else {
+                this.userMessageTimes.set(userId, recentTimes);
+            }
+        }
+        // 清理过期的黑名单
+        for (const [userId, blacklistEnd] of this.blacklist.entries()) {
+            if (now > blacklistEnd) {
+                this.blacklist.delete(userId);
+            }
+        }
+    }
+}
+exports.RateLimiter = RateLimiter;

package/lib/plugin/server.js

@@ -0,0 +1,206 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const tsrpc_1 = require("tsrpc");
+const console_1 = require("./console");
+const serviceProto_1 = require("../shared/serviceProto");
+class server extends console_1.console {
+    httpClient;
+    apiUrl;
+    wsClient;
+    ctx;
+    onConnectedCallback;
+    messageListenerSetup = false; // 标记消息监听器是否已设置
+    _init = false;
+    constructor() {
+        super();
+    }
+    async init(ctx, config) {
+        super.init(ctx, config);
+        this.ctx = ctx;
+        // 设置 dispose 事件监听器
+        ctx.on('dispose', () => {
+            // 修复热重载 监听残留
+            this.dispose();
+        });
+        // 确保重新初始化时先清理旧连接
+        if (this.wsClient) {
+            await this.dispose();
+        }
+        // 根据服务器环境选择对应的地址
+        const serverUrlMap = {
+            '测试服': 'ws://127.0.0.1:3000',
+            '开发服': 'ws://dev-bot.cn:3000',
+            '正式服': 'ws://dew-bot.cn:3000'
+        };
+        const serverUrl = serverUrlMap[config.服务器环境];
+        if (!serverUrl) {
+            throw new Error(`未知的服务器环境: ${config.服务器环境}`);
+        }
+        this.log(`选择环境: ${config.服务器环境}`);
+        this.log(`尝试连接到: ${serverUrl}`);
+        // 异步连接但不阻塞初始化
+        this.setWsUrl(serverUrl).catch(error => {
+            this.log('初始化连接失败:', error);
+            this.log('请检查服务器地址是否正确，以及服务器是否正在运行');
+        });
+    }
+    // 设置连接成功的回调
+    onConnected(callback) {
+        this.onConnectedCallback = callback;
+        // 如果已经连接且监听器未设置，立即调用回调
+        if (this._init && this.wsClient && !this.messageListenerSetup) {
+            callback();
+            this.messageListenerSetup = true;
+        }
+    }
+    // 卸载
+    async dispose() {
+        this.log('卸载:server');
+        this._init = false;
+        if (this.wsClient) {
+            this.wsClient.unlistenMsgAll(/.*/);
+            this.wsClient.disconnect();
+            // 清除对 wsClient 的引用
+            this.wsClient = null;
+        }
+        // 清除回调和监听器状态
+        this.onConnectedCallback = undefined;
+        this.messageListenerSetup = false;
+    }
+    /**
+ * 设置服务器地址
+ * @param link
+ */
+    async setApiUrl(link) {
+        return new Promise((resolve, reject) => {
+            this.apiUrl = `${link}`;
+            this.httpClient = new tsrpc_1.HttpClient(serviceProto_1.serviceProto, { server: this.apiUrl });
+            this.flowsToken(this.httpClient);
+            resolve(true);
+        });
+    }
+    flowsToken(client) {
+        client.flows.preApiReturnFlow.push((v) => {
+            if (v.return.isSucc) {
+                if (v.return.res.__token) {
+                    // tokenMap.set('_token', v.return.res.__token)
+                }
+            }
+            return v;
+        });
+    }
+    flowsResConnect(client) {
+        if (!client || !client.flows) {
+            this.log('客户端无效，无法设置断线重连');
+            return;
+        }
+        client.flows.postDisconnectFlow.push((v) => {
+            if (!v.isManual && this._init) {
+                this.connect();
+                this.log('开始断线重连');
+            }
+            // Bot_client.emit(Emitter_bot.server_off);
+            return v;
+        });
+    }
+    async connect() {
+        if (!this._init) {
+            this.log('插件已卸载，停止重连');
+            return;
+        }
+        if (!this.wsClient) {
+            this.log('WebSocket 客户端未初始化，无法重连');
+            return;
+        }
+        let res = await this.wsClient.connect();
+        if (!res.isSucc) {
+            this.log('重试断线准备重连');
+            if (this._init) {
+                this.ctx.setTimeout(async () => {
+                    this.connect();
+                }, 5000);
+            }
+        }
+        else {
+            this.log('断线重连成功');
+            // 断线重连成功时不需要重新设置监听器
+        }
+        return res;
+    }
+    lisentMsg(msgName, handler, self) {
+        if (!this._init || !this.wsClient) {
+            this.log('WebSocket 未连接，无法监听消息');
+            return;
+        }
+        return this.wsClient.listenMsg(msgName, ((data) => { handler.call(self, data); }));
+    }
+    async setWsUrl(link, maxRetries = 99999) {
+        this.log(`server link:${link}`);
+        this._init = true;
+        return new Promise(async (resolve, reject) => {
+            this.apiUrl = link;
+            if (this.wsClient && this.wsClient.unlistenMsgAll) {
+                this.wsClient.unlistenMsgAll(/.*/);
+                this.wsClient.disconnect();
+            }
+            this.wsClient = new tsrpc_1.WsClient(serviceProto_1.serviceProto, { server: this.apiUrl });
+            let connect;
+            let retryCount = 0;
+            const attemptConnect = async () => {
+                connect = await this.wsClient.connect();
+                if (connect.isSucc) {
+                    // this.flowsToken(this.wsClient);
+                    if (this.wsClient) {
+                        this.flowsResConnect(this.wsClient);
+                    }
+                    this.log('connect success');
+                    // 调用连接成功的回调（仅第一次连接）
+                    if (this.onConnectedCallback && !this.messageListenerSetup) {
+                        this.onConnectedCallback();
+                        this.messageListenerSetup = true;
+                    }
+                    resolve(true);
+                }
+                else {
+                    retryCount++;
+                    if (retryCount < maxRetries) {
+                        this.log(`连接失败，${retryCount}/${maxRetries}，5秒后重试...`);
+                        this.ctx.setTimeout(attemptConnect, 5000);
+                    }
+                    else {
+                        this.log('connect failed', connect.errMsg);
+                        this._init = false; // 连接失败时设置为未初始化状态
+                        reject(connect.errMsg);
+                    }
+                }
+            };
+            attemptConnect();
+        });
+    }
+    async api(apiName, posData) {
+        if (!this._init) {
+            this.log('插件未初始化，无法调用 API');
+            return;
+        }
+        let client = this.wsClient || this.httpClient;
+        if (!client) {
+            this.log('客户端未连接，无法调用 API');
+            return;
+        }
+        try {
+            let start = Date.now();
+            let req = await client.callApi(apiName, posData);
+            if (req.isSucc) {
+                this.log('apiReq', apiName, (Date.now() - start) / 1000, 's');
+                return req.res;
+            }
+            else {
+                this.log('请求出错', apiName, req.err.message);
+            }
+        }
+        catch (error) {
+            this.log('API 调用异常:', apiName, error.message);
+        }
+    }
+}
+exports.default = new server();

package/lib/plugin/serverHandel/handel_base.js

@@ -0,0 +1,27 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const sessions_1 = __importDefault(require("../sessions"));
+class handel_base {
+    ctx;
+    constructor(data) {
+        console.log('handel_base:::', data);
+    }
+    set(ctx) {
+        this.ctx = ctx;
+    }
+    async send(data, content) {
+        let sendInfo = data.Message;
+        let session = sessions_1.default.get(sendInfo.userId);
+        if (session.bot.platform) {
+            content = '\n' + content;
+        }
+        await session.send(content);
+    }
+    start(data) {
+        console.log('handel_base:::', data);
+    }
+}
+exports.default = handel_base;

package/lib/plugin/serverHandel/image.js

@@ -0,0 +1,19 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const puppeteer_1 = __importDefault(require("../puppeteer"));
+const handel_base_1 = __importDefault(require("./handel_base"));
+class handelr_image extends handel_base_1.default {
+    constructor(data) {
+        super(data);
+    }
+    async start(data) {
+        let renderName = data.info.render;
+        puppeteer_1.default.render(renderName, data.data || {}).then(async (e) => {
+            await this.send(data, e);
+        });
+    }
+}
+exports.default = handelr_image;

package/lib/plugin/serverHandel/none.js

@@ -0,0 +1,15 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const handel_base_1 = __importDefault(require("./handel_base"));
+class handelr_none extends handel_base_1.default {
+    constructor(data) {
+        super(data);
+    }
+    async start(data) {
+        console.log('handelr_none:::', data);
+    }
+}
+exports.default = handelr_none;

package/lib/plugin/serverHandel/text.js

@@ -0,0 +1,20 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const handel_base_1 = __importDefault(require("./handel_base"));
+class handelr_text extends handel_base_1.default {
+    constructor(data) {
+        super(data);
+    }
+    async start(data) {
+        if (data.info.type == 'text') {
+            await this.send(data, data.data);
+        }
+        else {
+            console.log('未知类型的文本消息');
+        }
+    }
+}
+exports.default = handelr_text;

package/lib/plugin/sessions.js

@@ -0,0 +1,33 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+class sessions {
+    sessions = new Map();
+    constructor() {
+        this.sessions = new Map();
+    }
+    init(ctx) {
+        ctx.on('dispose', () => {
+        });
+    }
+    /**
+     *
+     * @param key userid
+     * @param value 会话
+     */
+    set(key, value) {
+        this.sessions.set(key, value);
+    }
+    get(key) {
+        return this.sessions.get(key);
+    }
+    delete(key) {
+        this.sessions.delete(key);
+    }
+    clear() {
+        this.sessions.clear();
+    }
+    size() {
+        return this.sessions.size;
+    }
+}
+exports.default = new sessions();

package/lib/shared/IBody.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/lib/shared/IMassage.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ClientAction = void 0;
+var ClientAction;
+(function (ClientAction) {
+    ClientAction["none"] = "none";
+    ClientAction["image"] = "image";
+    ClientAction["text"] = "text";
+})(ClientAction || (exports.ClientAction = ClientAction = {}));

package/lib/shared/MsgMessage.js

@@ -0,0 +1,4 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.conf = void 0;
+exports.conf = {};

package/lib/shared/PtlMessage.js

@@ -0,0 +1,4 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.conf = void 0;
+exports.conf = {};

package/lib/shared/PtlTest.js

@@ -0,0 +1,4 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.conf = void 0;
+exports.conf = {};

package/lib/shared/base.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/lib/shared/face/index.js

@@ -0,0 +1,47 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.quality = exports.EquipAttType = exports.UserLevel = exports.ItemType = void 0;
+var ItemType;
+(function (ItemType) {
+    ItemType["\u8D44\u6E90"] = "prop_\u8D44\u6E90";
+    // 装备
+    ItemType["\u88C5\u5907"] = "equipment";
+    // 消耗品
+    ItemType["\u9053\u5177"] = "prop_\u9053\u5177";
+    // 消耗品 - 不可叠加
+    ItemType["\u6280\u80FD\u4E66"] = "skill_book";
+    // 宠物蛋
+    ItemType["\u5BA0\u7269\u86CB"] = "pet_egg";
+    // 血统卷轴
+    ItemType["\u8840\u7EDF\u5377\u8F74"] = "talent_book";
+    // 盲盒
+    ItemType["\u76F2\u76D2"] = "prop_\u76F2\u76D2";
+    // 形象卡
+    ItemType["\u5F62\u8C61\u5361"] = "image_card";
+})(ItemType || (exports.ItemType = ItemType = {}));
+var UserLevel;
+(function (UserLevel) {
+    UserLevel[UserLevel["\u73A9\u5BB6"] = 0] = "\u73A9\u5BB6";
+    UserLevel[UserLevel["\u8D5E\u52A9\u4F1A\u5458"] = 1] = "\u8D5E\u52A9\u4F1A\u5458";
+    UserLevel[UserLevel["\u7BA1\u7406\u5458"] = 2] = "\u7BA1\u7406\u5458";
+    UserLevel[UserLevel["\u5F00\u53D1\u8005"] = 3] = "\u5F00\u53D1\u8005";
+})(UserLevel || (exports.UserLevel = UserLevel = {}));
+var EquipAttType;
+(function (EquipAttType) {
+    EquipAttType["\u9B54\u6CD5\u6B66\u5668"] = "\u9B54\u6CD5\u6B66\u5668";
+})(EquipAttType || (exports.EquipAttType = EquipAttType = {}));
+//品质
+var quality;
+(function (quality) {
+    quality[quality["F"] = 0] = "F";
+    quality[quality["D"] = 1] = "D";
+    quality[quality["C"] = 2] = "C";
+    quality[quality["B"] = 3] = "B";
+    quality[quality["A"] = 4] = "A";
+    quality[quality["S"] = 5] = "S";
+    // 常规顶级
+    quality[quality["SS"] = 6] = "SS";
+    quality[quality["SSS"] = 7] = "SSS";
+    quality[quality["SR"] = 8] = "SR";
+    quality[quality["SSR"] = 9] = "SSR";
+})(quality || (exports.quality = quality = {}));