koishi-plugin-tmp-bot 1.0.0

package/lib/api/TruckersMpApi.js

@@ -0,0 +1,50 @@
+const BASE_API = 'https://api.truckersmp.com/v2'
+
+module.exports = {
+  /**
+   * 查询玩家信息
+   */
+  async player (http, tmpId) {
+    let result = null
+    try {
+      result = await http.get(`${BASE_API}/player/${tmpId}`)
+    } catch {
+      return {
+        error: true
+      }
+    }
+
+    // 拼接返回数据
+    let data = {
+      error: JSON.parse(result.error)
+    }
+    if (!data.error) {
+      data.data = result.response
+    }
+
+    return data
+  },
+  /**
+   * 查询服务器列表
+   */
+  async servers (http) {
+    let result = null
+    try {
+      result = await http.get(`${BASE_API}/servers`)
+    } catch {
+      return {
+        error: true
+      }
+    }
+
+    // 拼接返回数据
+    let data = {
+      error: JSON.parse(result.error)
+    }
+    if (!data.error) {
+      data.data = result.response
+    }
+
+    return data
+  }
+}

package/lib/api/truckyAppApi.js

@@ -0,0 +1,26 @@
+const BASE_API_V3 = 'https://api.truckyapp.com/v3'
+
+module.exports = {
+  /**
+   * 查询线上信息
+   */
+  async online (http, tmpId) {
+    let result = null
+    try {
+      result = await http.get(`${BASE_API_V3}/map/online?playerID=${tmpId}`)
+    } catch {
+      return {
+        error: true
+      }
+    }
+
+    // 拼接返回数据
+    let data = {
+      error: !result || !result.response || result.response.error
+    }
+    if (!data.error) {
+      data.data = result.response
+    }
+    return data
+  }
+}

package/lib/command/tmpBind.js

@@ -0,0 +1,22 @@
+const guildBind = require('../database/guildBind')
+const truckersMpApi = require("../api/TruckersMpApi");
+
+/**
+ * 绑定 TMP ID
+ */
+module.exports = async (ctx, cfg, session, tmpId) => {
+  if (!tmpId || isNaN(tmpId)) {
+    return `请输入正确的玩家编号`
+  }
+
+  // 查询玩家信息
+  let playerInfo = await truckersMpApi.player(ctx.http, tmpId)
+  if (playerInfo.error) {
+    return '绑定失败 (查询玩家信息失败)'
+  }
+
+  // 更新数据库
+  guildBind.saveOrUpdate(ctx.database, session.platform, session.guildId, session.userId, session.author.username, tmpId)
+
+  return `绑定成功 ( ${playerInfo.data.name} )`
+}

package/lib/command/tmpQuery.js

@@ -0,0 +1,73 @@
+const dayjs = require('dayjs')
+const guildBind = require('../database/guildBind')
+const truckersMpApi = require('../api/TruckersMpApi')
+const truckyAppApi = require('../api/truckyAppApi')
+const baiduTranslate = require('../util/baiduTranslate')
+
+/**
+ * 用户组
+ */
+const userGroup = {
+  'Player': '玩家',
+  'Retired Legend': '退役',
+  'Game Developer': '游戏开发者'
+}
+
+/**
+ * 查询玩家信息
+ */
+module.exports = async (ctx, cfg, session, tmpId) => {
+  if (tmpId && isNaN(tmpId)) {
+    return `请输入正确的玩家编号`
+  }
+
+  // 如果没有传入tmpId，尝试从数据库查询绑定信息
+  if (!tmpId) {
+    let guildBindData = await guildBind.get(ctx.database, session.platform, session.guildId, session.userId)
+    if (!guildBindData) {
+      return `请输入正确的玩家编号`
+    }
+    tmpId = guildBindData.tmp_id
+  }
+
+  // 查询玩家信息
+  let playerInfo = await truckersMpApi.player(ctx.http, tmpId)
+  if (playerInfo.error) {
+    return '查询玩家信息失败，请重试'
+  }
+
+  // 查询线上信息
+  let playerMapInfo = await truckyAppApi.online(ctx.http, tmpId)
+  if (playerMapInfo.error) {
+    return '查询玩家信息失败，请重试'
+  }
+
+  // 拼接消息模板 正常4763167 永久5396563 暂时封号5118166
+  let message = `<img src="${playerInfo.data.avatar}"/>`
+  message += '\n😀玩家名称: ' + playerInfo.data.name
+  message += '\n📑注册日期: ' + dayjs(playerInfo.data.joinDate + 'Z').format('YYYY年MM月DD日')
+  message += '\n💼所属分组: ' + (userGroup[playerInfo.data.groupName] || playerInfo.data.groupName) // 🪪💼📂🚹
+  if (playerInfo.data.vtc && playerInfo.data.vtc.inVTC) {
+    message += '\n🚚所属车队: ' + playerInfo.data.vtc.name
+  }
+  message += '\n🚫是否封禁: ' + (playerInfo.data.banned ? '是' : '否')
+  if (playerInfo.data.banned) {
+    message += '\n🚫封禁截止: '
+    if (!playerInfo.data.displayBans) {
+      message += '隐藏'
+    } else if (!playerInfo.data.bannedUntil) {
+      message += '永久'
+    } else {
+      message += daiyjs(playerInfo.data.bannedUntil + 'Z').format('YYYY年MM月DD日 HH:mm')
+    }
+  }
+  message += '\n🚫封禁次数: ' + playerInfo.data.bansCount || 0
+  message += '\n📶在线状态: ' + (playerMapInfo.data.online ? `在线🟢 (${playerMapInfo.data.serverDetails.name})` : '离线⚫')
+  if (playerMapInfo.data.online) {
+    message += '\n🌍线上位置: '
+    message += await baiduTranslate(ctx, cfg, playerMapInfo.data.location.poi.country)
+    message += ' - '
+    message += await baiduTranslate(ctx, cfg, playerMapInfo.data.location.poi.realName)
+  }
+  return message
+}

package/lib/command/tmpServer.js

@@ -0,0 +1,39 @@
+const truckersMpApi = require('../api/TruckersMpApi')
+
+module.exports = async (ctx, cfg) => {
+  // 查询服务器信息
+  let serverData = await truckersMpApi.servers(ctx.http)
+  if (serverData.error) {
+    return '查询服务器失败，请稍后重试'
+  }
+
+  // 过滤非欧卡2服务器
+  let etsServerList = serverData.data.filter(server => server.game === 'ETS2')
+
+  // 构建消息
+  let message = ''
+  for (let server of etsServerList) {
+    // 如果前面有内容，换行
+    if (message) {
+      message += '\n\n'
+    }
+
+    message += '服务器: ' + ( server.online ? '🟢' : '⚫' ) + server.name
+    message += `\n玩家人数: ${server.players}/${server.maxplayers}`
+    if (server.queue) {
+      message += ` (队列: ${server.queue})`
+    }
+    // 服务器特性
+    let characteristicList = []
+    if (!server.afkenabled) {
+      characteristicList.push('⏱挂机')
+    }
+    if (server.collisions) {
+      characteristicList.push('💥碰撞')
+    }
+    if (characteristicList && characteristicList.length > 0) {
+      message += '\n服务器特性: ' + characteristicList.join(' ')
+    }
+  }
+  return message
+}

package/lib/database/guildBind.js

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

package/lib/database/model.js

@@ -0,0 +1,78 @@
+/**
+ * 数据表声明
+ */
+const modelArray = {
+  tmp_guild_bind: {
+    id: {
+      type: 'unsigned',
+      length: 10,
+      nullable: false,
+      comment: '主键'
+    },
+    platform: {
+      type: 'string',
+      length: 50,
+      nullable: false,
+      comment: '所属平台'
+    },
+    guild_id: {
+      type: 'string',
+      length: 50,
+      nullable: false,
+      comment: '群组编号'
+    },
+    user_id: {
+      type: 'string',
+      length: 50,
+      nullable: false,
+      comment: '用户编号'
+    },
+    user_name: {
+      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.js

@@ -0,0 +1,33 @@
+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,10 @@
+import { Context, Schema } from 'koishi';
+export declare const name = "tmp-bot";
+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,30 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.apply = exports.Config = exports.name = void 0;
+const koishi_1 = require("koishi");
+const model = require('./database/model');
+const tmpQuery = require('./command/tmpQuery');
+const tmpServer = require('./command/tmpServer');
+const tmpBind = require('./command/tmpBind');
+exports.name = 'tmp-bot';
+exports.Config = 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('启用百度翻译缓存')
+});
+function apply(ctx, cfg) {
+    // 注册指令
+    ctx.command('tmpquery <tmpId>').action(async ({ session }, tmpId) => await tmpQuery(ctx, cfg, session, tmpId));
+    ctx.command('tmpserver').action(async () => await tmpServer(ctx, cfg));
+    ctx.command('tmpbind <tmpId>').action(async ({ session }, tmpId) => await tmpBind(ctx, cfg, session, tmpId));
+    // 等待数据库模块准备完毕后初始化数据库表
+    let databaseTime = setInterval(() => {
+        if (ctx.database) {
+            clearInterval(databaseTime);
+            databaseTime = null;
+            model(ctx);
+        }
+    }, 100);
+}
+exports.apply = apply;

package/lib/util/baiduTranslate.js

@@ -0,0 +1,37 @@
+const md5 = require('js-md5')
+const translateCache = require('../database/translateCache')
+const TRANSLATE_API = 'https://fanyi-api.baidu.com/api/trans/vip/translate'
+
+module.exports = async (ctx, cfg, content) => {
+  // 没有开启百度翻译功能，直接返回文本
+  if (!cfg.baiduTranslateEnable) {
+    return content
+  }
+
+  // 如果开启了缓存，尝试从缓存中查询翻译
+  if (cfg.baiduTranslateCacheEnable) {
+    let translateContent = await translateCache.getTranslate(ctx.database, md5(content))
+    if (translateContent) {
+      return translateContent
+    }
+  }
+
+  // 创建请求秘钥
+  let randomInt = Math.floor(Math.random() * 10000)
+  let sign = md5(cfg.baiduTranslateAppId + content + randomInt + cfg.baiduTranslateKey)
+
+  // 调用请求
+  let result = await ctx.http.get(`${TRANSLATE_API}?q=${encodeURI(content)}&from=auto&to=zh&appid=${cfg.baiduTranslateAppId}&salt=${randomInt}&sign=${sign}`);
+
+  // 如果翻译失败，直接返回内容
+  if (result.error_code) {
+    return content
+  }
+
+  // 如果开启了缓存，将翻译内容缓存到数据库
+  if (cfg.baiduTranslateCacheEnable) {
+    translateCache.save(ctx.database, md5(content), content, result.trans_result[0].dst)
+  }
+
+  return result.trans_result[0].dst
+}

package/package.json

@@ -0,0 +1,34 @@
+{
+  "name": "koishi-plugin-tmp-bot",
+  "description": "欧洲卡车模拟2 TMP查询机器人",
+  "version": "1.0.0",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "dist"
+  ],
+  "license": "MIT",
+  "scripts": {},
+  "keywords": [
+    "chatbot",
+    "koishi",
+    "plugin",
+    "truckersmp",
+    "gametool",
+    "ets2",
+    "tmp"
+  ],
+  "koishi": {
+    "service": {
+      "required": ["database"]
+    }
+  },
+  "peerDependencies": {
+    "koishi": "^4.17.1"
+  },
+  "dependencies": {
+    "dayjs": "^1.11.10",
+    "js-md5": "^0.8.3"
+  }
+}

package/readme.md

@@ -0,0 +1,5 @@
+# koishi-plugin-tmp-bot
+
+[![npm](https://img.shields.io/npm/v/koishi-plugin-tmp-bot?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-tmp-bot)
+
+欧洲卡车模拟2 TMP查询机器人