koishi-plugin-ets2-tools-tmp 0.0.2 → 0.0.3

package/lib/command/tmpTraffic/tmpTrafficMap.js

@@ -0,0 +1,167 @@
+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,58 @@
+const truckyAppApi = require('../../api/truckyAppApi');
+const baiduTranslate = require('../../util/baiduTranslate');
+/**
+ * 服务器别名
+ */
+const serverNameAlias = {
+    '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,14 @@
+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;
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, cfg: Config): void;

package/lib/index.js

@@ -0,0 +1,75 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Config = exports.inject = exports.name = void 0;
+exports.apply = apply;
+const koishi_1 = require("koishi");
+const model = require('./database/model');
+const { MileageRankingType } = require('./util/constant');
+const tmpQuery = require('./command/tmpQuery/tmpQuery');
+const tmpServer = require('./command/tmpServer');
+const tmpBind = require('./command/tmpBind');
+const tmpTraffic = require('./command/tmpTraffic/tmpTraffic');
+const tmpPosition = require('./command/tmpPosition');
+const tmpVersion = require('./command/tmpVersion');
+const tmpDlcMap = require('./command/tmpDlcMap');
+const tmpMileageRanking = require('./command/tmpMileageRanking');
+const resetPassword = require('./command/ets-app/resetPassword');
+const queryPoint = require('./command/ets-app/queryPoint');
+const tmpVtc = require('./command/tmpVtc');
+
+exports.name = 'tmp-bot';
+exports.inject = {
+    required: ['database'],
+    optional: ['puppeteer']
+};
+exports.Config = koishi_1.Schema.intersect([
+    koishi_1.Schema.object({
+        baiduTranslateEnable: koishi_1.Schema.boolean().default(false).description('启用百度翻译'),
+        baiduTranslateAppId: koishi_1.Schema.string().description('百度翻译APP ID'),
+        baiduTranslateKey: koishi_1.Schema.string().description('百度翻译秘钥'),
+        baiduTranslateCacheEnable: koishi_1.Schema.boolean().default(false).description('启用百度翻译缓存')
+    }).description('基本配置'),
+    koishi_1.Schema.object({
+        queryShowAvatarEnable: koishi_1.Schema.boolean().default(false).description('查询指令展示头像，部分玩家的擦边头像可能导致封号'),
+        tmpTrafficType: koishi_1.Schema.union([
+            koishi_1.Schema.const(1).description('文字'),
+            koishi_1.Schema.const(2).description('热力图')
+        ]).default(1).description('路况信息展示方式'),
+        tmpQueryType: koishi_1.Schema.union([
+            koishi_1.Schema.const(1).description('文字'),
+            koishi_1.Schema.const(2).description('图片')
+        ]).default(1).description('玩家信息展示方式'),
+    }).description('指令配置'),
+    koishi_1.Schema.object({
+        mainSettings: koishi_1.Schema.object({
+            url: koishi_1.Schema.string().description("API服务器地址").required(),
+            token: koishi_1.Schema.string().description("API认证令牌").required(),
+            logOutput: koishi_1.Schema.boolean().description("是否输出日志").default(true)
+        }).description("车队平台设置")
+    }),
+    koishi_1.Schema.object({
+        resetPassword: koishi_1.Schema.object({
+            adminUsers: koishi_1.Schema.array(koishi_1.Schema.string()).description("管理员用户ID（拥有重置任意teamId权限）").default([])
+        }).description("重置密码设置")
+    }),
+]);
+function apply(ctx, cfg) {
+    // 初始化数据表
+    model(ctx);
+    // 注册指令
+    ctx.command('查询 <tmpId>').action(async ({ session }, tmpId) => await tmpQuery(ctx, cfg, session, tmpId));
+    ctx.command('美卡服务器').action(async () => await tmpServer(ctx, cfg, 'ATS'));
+    ctx.command('欧卡服务器').action(async () => await tmpServer(ctx, cfg, 'ETS2'));
+    ctx.command('绑定 <tmpId>').action(async ({ session }, tmpId) => await tmpBind(ctx, cfg, session, tmpId));
+    ctx.command('路况 <serverName>').action(async ({ session }, serverName) => await tmpTraffic(ctx, cfg, serverName));
+    ctx.command('定位 <tmpId>').action(async ({ session }, tmpId) => await tmpPosition(ctx, cfg, session, tmpId));
+    ctx.command('tmp版本').action(async () => await tmpVersion(ctx));
+    ctx.command('地图dlc价格').action(async ({ session }) => await tmpDlcMap(ctx, session));
+    ctx.command('里程排行榜').action(async ({ session }) => await tmpMileageRanking(ctx, session, MileageRankingType.total));
+    ctx.command('今日里程排行榜').action(async ({ session }) => await tmpMileageRanking(ctx, session, MileageRankingType.today));
+
+    ctx.command('vtc查询 <vtcid>').action(async ({ session }, vtcid) => await tmpVtc(ctx, cfg, session, vtcid));
+    ctx.command(`重置密码 [targetTeamId:string]`, "重置欧卡车队平台密码").usage("重置自己的密码，或管理员重置指定teamId的密码").example(`重置密码new - 重置自己的密码`).example(`重置密码new 123 - 管理员重置指定teamId的密码`).action(async ({ session }, targetTeamId) => await resetPassword(ctx, cfg, session, targetTeamId));
+    ctx.command(`查询积分 [targetQQ:string]`, "查询欧卡车队平台积分").usage("查询自己或指定QQ号的积分，在群聊中可@他人查询").example(`查询积分 - 查询自己的积分`).example(`查询积分 - 查询指定QQ号的积分`).action(async ({ session }, targetQQ) => await queryPoint(ctx, cfg, session, targetQQ));
+    ctx.command('规则查询').action(async ({ session }) => { return 'https://truckersmp.com/knowledge-base/article/746' })
+}

package/lib/resource/dlc.html

@@ -0,0 +1,115 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <title>DLC</title>
+  <style>
+    #dlc-info-container {
+      width: 600px;
+      background-color: #222d33;
+      padding: 14px;
+    }
+
+    .dlc-box {
+      display: flex;
+      flex-direction: row;
+      box-sizing: border-box;
+      padding: 12px;
+      margin-top: 12px;
+      background-size: cover;
+      background-repeat: no-repeat;
+      background-position: center;
+    }
+    .dlc-box:nth-of-type(1) {
+      margin-top: 0;
+    }
+
+    .dlc-box .header-image {
+      width: 210px;
+      display: inline-block;
+    }
+
+    .dlc-box .dlc-info {
+      flex: 1;
+      width: 0;
+      //border: 1px solid red;
+      box-sizing: border-box;
+      padding: 2px 12px;
+    }
+    .dlc-info .name {
+      color: #ffffff;
+      font-size: 18px;
+      font-weight: 600;
+      overflow: hidden;
+      text-overflow: ellipsis;
+      white-space: nowrap;
+    }
+    .dlc-info .desc {
+      color: #e5e5e5;
+      font-size: 14px;
+      overflow: hidden;
+      text-overflow: ellipsis;
+      display: -webkit-box;
+      -webkit-line-clamp: 2;
+      -webkit-box-orient: vertical;
+    }
+    .dlc-info .price-box {
+      margin-top: 8px;
+    }
+
+    .dlc-info .price-box span {
+      display: inline-block;
+      color: #BEEE11;
+      font-size: 16px;
+    }
+    .dlc-info .price-box .discount-price {
+      color: #cbcbcb;
+      text-decoration: line-through;
+    }
+    .dlc-info .price-box .discount {
+      font-size: 14px;
+      color: #BEEE11;
+      background-color: #4c6b22;
+      padding: 2px 6px;
+      margin-left: 4px;
+    }
+  </style>
+</head>
+<body>
+  <div id="dlc-info-container">
+  </div>
+
+  <script>
+    function setData(dlcList) {
+      // 遍历渲染DLC列表
+      for (let dlc of dlcList) {
+        let dom = document.createElement(`div`);
+        dom.className = 'dlc-box';
+        dom.style.backgroundImage = `url('${dlc.backgroundImageUrl}')`;
+
+        let priceDiscountDom = ''
+        if (dlc.discount > 0) {
+          priceDiscountDom = `
+          <span class="discount-price">￥${Math.ceil(dlc.originalPrice / 100)}</span>
+          <span class="discount">-${dlc.discount}%</span>
+        `;
+        }
+
+        dom.innerHTML = `
+          <img class="header-image" src="${dlc.headerImageUrl}"/>
+          <div class="dlc-info">
+            <div class="name">${dlc.name}</div>
+            <div class="desc">${dlc.desc}</div>
+            <div class="price-box">
+              <span>￥${Math.ceil(dlc.finalPrice / 100)}</span>
+              ${priceDiscountDom}
+            </div>
+          </div>
+        `;
+
+        document.querySelector('#dlc-info-container').appendChild(dom);
+      }
+    }
+  </script>
+</body>
+</html>