koishi-plugin-lutu 1.0.0

package/lib/command/tmpTraffic/tmpTrafficMap.js

@@ -0,0 +1,163 @@
+const truckyAppApi = require('../../api/truckyAppApi');
+const evmOpenApi = require('../../api/evmOpenApi');
+const baiduTranslate = require('../../util/baiduTranslate');
+const { resolve } = require("path");
+const common = require("../../util/common");
+const { segment } = require("koishi");
+/**
+ * 服务器别名
+ */
+const serverAlias = {
+    's1': {
+        name: 'sim1',
+        mapType: 'ets',
+        serverId: 2,
+        bounds: [[-94189, 93775], [79264, -78999]]
+    },
+    's2': {
+        name: 'sim2',
+        mapType: 'ets',
+        serverId: 41,
+        bounds: [[-94189, 93775], [79264, -78999]]
+    },
+    'p': {
+        name: 'eupromods1',
+        mapType: 'promods',
+        serverId: 50,
+        bounds: [[-96355, 16381], [205581, -70750]]
+    },
+    'a': {
+        name: 'arc1',
+        mapType: 'ets',
+        serverId: 7,
+        bounds: [[-94189, 93775], [79264, -78999]]
+    }
+};
+/**
+ * 路况程度转中文
+ */
+const severityToZh = {
+    'Fluid': {
+        text: '畅通',
+        color: '#00d26a'
+    },
+    'Moderate': {
+        text: '正常',
+        color: '#ff6723'
+    },
+    'Congested': {
+        text: '缓慢',
+        color: '#f8312f'
+    },
+    'Heavy': {
+        text: '拥堵',
+        color: '#8d67c5'
+    }
+};
+/**
+ * 位置类型转中文
+ */
+const typeToZh = {
+    'City': '城市',
+    'Road': '公路',
+    'Intersection': '十字路口'
+};
+/**
+ * 查询路况
+ */
+module.exports = async (ctx, cfg, serverName) => {
+    if (!ctx.puppeteer) {
+        return '未启用 puppeteer 服务';
+    }
+    // 根据别名获取服务器信息
+    let serverInfo = serverAlias[serverName];
+    if (!serverInfo) {
+        return '请输入正确的服务器名称 (s1, s2, p, a)';
+    }
+    // 查询路况信息
+    let trafficData = await truckyAppApi.trafficTop(ctx.http, serverInfo.name);
+    if (trafficData.error) {
+        return '查询路况信息失败';
+    }
+    // 查询地图玩家数据
+    let mapData = await evmOpenApi.mapPlayerList(ctx.http, serverInfo.serverId, serverInfo.bounds[0][0], serverInfo.bounds[0][1], serverInfo.bounds[1][0], serverInfo.bounds[1][1]);
+
+    switch (serverName) {
+        case 's1':
+        case 's2':
+        case 'a': {
+            let data = {
+                mapType: serverInfo.mapType,
+                trafficList: [],
+                playerCoordinateList: mapData.error && mapData.data ? [] : mapData.data.map(item => [item.axisX, item.axisY])
+            };
+            for (const traffic of trafficData.data) {
+                data.trafficList.push({
+                    country: await baiduTranslate(ctx, cfg, traffic.country),
+                    province: await baiduTranslate(ctx, cfg, traffic.name.substring(0, traffic.name.lastIndexOf('(') - 1)),
+                    playerCount: traffic.players,
+                    severity: severityToZh[traffic.newSeverity] || { text: '未知', color: '#ffffff' }
+                });
+            }
+            let page;
+            try {
+                page = await ctx.puppeteer.page();
+                await page.setViewport({ width: 1000, height: 1000 });
+                await page.goto(`file:///${resolve(__dirname, '../../resource/traffic.html')}`);
+                await page.evaluate(`setData(${JSON.stringify(data)})`);
+                await common.sleep(100);
+                await page.waitForNetworkIdle();
+                const element = await page.$("#container");
+                return (segment.image(await element.screenshot({
+                    encoding: "binary"
+                }), "image/jpg"));
+            }
+            catch {
+                return '渲染异常，请重试';
+            }
+            finally {
+                if (page) {
+                    await page.close();
+                }
+            }
+            break;
+        }
+        case 'p': {
+            let data = {
+                mapType: serverInfo.mapType,
+                trafficList: [],
+                playerCoordinateList: mapData.error && mapData.data ? [] : mapData.data.map(item => [item.axisX, item.axisY])
+            };
+            for (const traffic of trafficData.data) {
+                data.trafficList.push({
+                    country: await baiduTranslate(ctx, cfg, traffic.country),
+                    province: await baiduTranslate(ctx, cfg, traffic.name.substring(0, traffic.name.lastIndexOf('(') - 1)),
+                    playerCount: traffic.players,
+                    severity: severityToZh[traffic.newSeverity] || { text: '未知', color: '#ffffff' }
+                });
+            }
+            let page;
+            try {
+                page = await ctx.puppeteer.page();
+                await page.setViewport({ width: 1200, height: 1200 });
+                await page.goto(`file:///${resolve(__dirname, '../../resource/traffic.html')}`);
+                await page.evaluate(`setData(${JSON.stringify(data)})`);
+                await common.sleep(100);
+                await page.waitForNetworkIdle();
+                const element = await page.$("#container");
+                return (segment.image(await element.screenshot({
+                    encoding: "binary"
+                }), "image/jpg"));
+            }
+            catch {
+                return '渲染异常，请重试';
+            }
+            finally {
+                if (page) {
+                    await page.close();
+                }
+            }
+            break;
+        }
+    }
+};

package/lib/command/tmpTraffic/tmpTrafficText.d.ts

@@ -0,0 +1,2 @@
+declare function _exports(ctx: any, cfg: any, serverName: any): Promise<string>;
+export = _exports;

package/lib/command/tmpTraffic/tmpTrafficText.js

@@ -0,0 +1,60 @@
+const truckyAppApi = require('../../api/truckyAppApi');
+const baiduTranslate = require('../../util/baiduTranslate');
+/**
+ * 服务器别名
+ */
+const serverNameAlias = {
+    's1': 'sim1',
+    's2': 'sim2',
+    'p': 'eupromods1',
+    'a': 'arc1',
+    'S1': 'sim1',
+    'S2': 'sim2',
+    'P': 'eupromods1',
+    'A': 'arc1'
+};
+/**
+ * 路况程度转中文
+ */
+const severityToZh = {
+    'Fluid': '🟢畅通',
+    'Moderate': '🟠正常',
+    'Congested': '🔴缓慢',
+    'Heavy': '🟣拥堵'
+};
+/**
+ * 位置类型转中文
+ */
+const typeToZh = {
+    'City': '城市',
+    'Road': '公路',
+    'Intersection': '十字路口'
+};
+/**
+ * 查询路况
+ */
+module.exports = async (ctx, cfg, serverName) => {
+    let serverQueryName = serverNameAlias[serverName];
+    if (!serverQueryName) {
+        return '请输入正确的服务器名称 (s1, s2, p, a)';
+    }
+    let trafficData = await truckyAppApi.trafficTop(ctx.http, serverQueryName);
+    if (trafficData.error) {
+        return '查询路况信息失败';
+    }
+    // 构建消息
+    let message = '';
+    for (const traffic of trafficData.data) {
+        if (message) {
+            message += '\n\n';
+        }
+        message += await baiduTranslate(ctx, cfg, traffic.country);
+        message += ' - ';
+        let name = traffic.name.substring(0, traffic.name.lastIndexOf('(') - 1);
+        let type = traffic.name.substring(traffic.name.lastIndexOf('(') + 1, traffic.name.lastIndexOf(')'));
+        message += await baiduTranslate(ctx, cfg, name) + ` (${typeToZh[type] || type})`;
+        message += '\n路况: ' + (severityToZh[traffic.newSeverity] || traffic.color);
+        message += ' | 人数: ' + traffic.players;
+    }
+    return message;
+};

package/lib/command/tmpVersion.d.ts

@@ -0,0 +1,2 @@
+declare function _exports(ctx: any): Promise<string>;
+export = _exports;

package/lib/command/tmpVersion.js

@@ -0,0 +1,14 @@
+const truckersMpApi = require("../api/truckersMpApi");
+module.exports = async (ctx) => {
+    // 查询版本信息
+    let result = await truckersMpApi.version(ctx.http);
+    if (result.error) {
+        return '查询失败，请稍后再试';
+    }
+    // 构建消息返回
+    let message = '';
+    message += `TMP版本：${result.data.name}\n`;
+    message += `欧卡支持版本: ${result.data.supported_game_version}\n`;
+    message += `美卡支持版本: ${result.data.supported_ats_game_version}`;
+    return message;
+};

package/lib/command/tmpVtc.js

@@ -0,0 +1,29 @@
+const truckersMpApi = require('../api/truckersMpApi')
+
+/**
+ * 查询玩家信息
+ */
+module.exports = async (ctx, cfg, session, vtcid) => {
+  if (!vtcid || isNaN(vtcid)) {
+    return `请输入正确的vtc编号`
+  }
+
+  // 查询玩家信息
+  let vtcInfo = await truckersMpApi.vtc(ctx.http, vtcid)
+  if (vtcInfo.error) {
+    return '查询vtc信息失败，请重试'
+  }
+
+  // 拼接消息模板
+  let message = ''
+  message += `<img src="${vtcInfo.data.logo}"/>\n`
+  message += '🆔VTC编号: ' + vtcInfo.data.id
+  message += '\n📑VTC名称: ' + vtcInfo.data.name
+  message += '\n📑VTC所有者id: ' + vtcInfo.data.owner_id
+  message += '\n📑VTC所有者名称: ' + vtcInfo.data.owner_username
+  message += '\n📑VTC创建日期: ' + vtcInfo.data.created + `(UTC)`
+  message += '\n🎮VTC人数: ' + vtcInfo.data.members_count
+  message += '\n🎮VTC前缀: ' + vtcInfo.data.tag
+  message += `\n💼VTC主页: ${vtcInfo.data.website}`
+  return message
+}

package/lib/database/guildBind.d.ts

@@ -0,0 +1,15 @@
+/**
+ * 获取绑定信息
+ * @param db 数据源
+ * @param platform 平台
+ * @param userId 用户编号
+ */
+export function get(db: any, platform: any, userId: any): Promise<any>;
+/**
+ * 新增或更新绑定信息
+ * @param db 数据源
+ * @param platform 平台
+ * @param userId 用户编号
+ * @param tmpId TMP ID
+ */
+export function saveOrUpdate(db: any, platform: any, userId: any, tmpId: any): void;

package/lib/database/guildBind.js

@@ -0,0 +1,41 @@
+module.exports = {
+    /**
+     * 获取绑定信息
+     * @param db 数据源
+     * @param platform 平台
+     * @param userId 用户编号
+     */
+    async get(db, platform, userId) {
+        const guildBindList = await db.get('tmp_guild_bind', {
+            platform,
+            user_id: userId
+        });
+        if (guildBindList && guildBindList.length > 0) {
+            return guildBindList[0];
+        }
+        return null;
+    },
+    /**
+     * 新增或更新绑定信息
+     * @param db 数据源
+     * @param platform 平台
+     * @param userId 用户编号
+     * @param tmpId TMP ID
+     */
+    saveOrUpdate(db, platform, userId, tmpId) {
+        this.get(db, platform, userId).then((data) => {
+            if (data) {
+                db.set('tmp_guild_bind', data.id, {
+                    tmp_id: tmpId
+                });
+            }
+            else {
+                db.create('tmp_guild_bind', {
+                    platform: platform,
+                    user_id: userId,
+                    tmp_id: tmpId
+                });
+            }
+        });
+    }
+};

package/lib/database/model.d.ts

@@ -0,0 +1,2 @@
+declare function _exports(ctx: any): void;
+export = _exports;

package/lib/database/model.js

@@ -0,0 +1,65 @@
+/**
+ * 数据表声明
+ */
+const modelArray = {
+    tmp_guild_bind: {
+        id: {
+            type: 'unsigned',
+            length: 10,
+            nullable: false,
+            comment: '主键'
+        },
+        platform: {
+            type: 'string',
+            length: 50,
+            nullable: false,
+            comment: '所属平台'
+        },
+        user_id: {
+            type: 'string',
+            length: 50,
+            nullable: false,
+            comment: '用户编号'
+        },
+        tmp_id: {
+            type: 'unsigned',
+            length: 50,
+            nullable: false,
+            comment: 'TMP ID'
+        }
+    },
+    tmp_translate_cache: {
+        id: {
+            type: 'unsigned',
+            length: 10,
+            nullable: false,
+            comment: '主键'
+        },
+        content: {
+            type: 'string',
+            nullable: false,
+            length: 200,
+            comment: '原文文本'
+        },
+        content_md5: {
+            type: 'string',
+            nullable: false,
+            length: 32,
+            comment: '原文文本md5'
+        },
+        translate_content: {
+            type: 'string',
+            nullable: false,
+            length: 200,
+            comment: '翻译文本'
+        }
+    }
+};
+/**
+ * 初始化数据库
+ */
+module.exports = (ctx) => {
+    for (let modelName in modelArray) {
+        ctx.model.extend(modelName, modelArray[modelName], { autoInc: true });
+    }
+};

package/lib/database/translateCache.d.ts

@@ -0,0 +1,14 @@
+/**
+ * 查询翻译
+ * @param db 数据源
+ * @param contentMd5 文本MD5
+ */
+export function getTranslate(db: any, contentMd5: any): Promise<any>;
+/**
+ * 保存翻译缓存信息
+ * @param db 数据源
+ * @param contentMd5 原文文本MD5
+ * @param content 原文文本
+ * @param translateContent 翻译文本
+ */
+export function save(db: any, contentMd5: any, content: any, translateContent: any): void;

package/lib/database/translateCache.js

@@ -0,0 +1,31 @@
+module.exports = {
+    /**
+     * 查询翻译
+     * @param db 数据源
+     * @param contentMd5 文本MD5
+     */
+    async getTranslate(db, contentMd5) {
+        const translateCacheList = await db.get('tmp_translate_cache', {
+            content_md5: contentMd5
+        });
+        // 如果查询到了缓存，直接返回翻译文本
+        if (translateCacheList && translateCacheList.length > 0) {
+            return translateCacheList[0].translate_content;
+        }
+        return null;
+    },
+    /**
+     * 保存翻译缓存信息
+     * @param db 数据源
+     * @param contentMd5 原文文本MD5
+     * @param content 原文文本
+     * @param translateContent 翻译文本
+     */
+    save(db, contentMd5, content, translateContent) {
+        db.create('tmp_translate_cache', {
+            content,
+            content_md5: contentMd5,
+            translate_content: translateContent
+        });
+    }
+};

package/lib/index.d.ts

@@ -0,0 +1,35 @@
+import { Context, Schema } from 'koishi';
+export declare const name = "tmp-bot";
+export declare const inject: {
+    required: string[];
+    optional: string[];
+};
+export interface Config {
+    baiduTranslateEnable: boolean;
+    baiduTranslateAppId: string;
+    baiduTranslateKey: string;
+    baiduTranslateCacheEnable: boolean;
+    adminUseHttps: boolean;
+    adminApiUrl: string;
+    adminApiToken: string;
+    adminVtcId: string;
+    adminCheckTimes: string[];
+    adminSendTimes: string[];
+    adminGroups: string[];
+    adminServerSource: string;
+    adminStartPointSource: string;
+    adminEndPointSource: string;
+    adminShowBanner: boolean;
+    adminProfileUploadedMessage: string;
+    adminProfileNotUploadedMessage: string;
+    mainGroups: string[];
+    mainActivityReminderMessage: string;
+    mainActivityReminderTimes: number[];
+    debugMode: boolean;
+    logApiResponses: boolean;
+    logTimingDetails: boolean;
+    logActivityMatching: boolean;
+    logMessageSending: boolean;
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, cfg: Config): void;