koishi-plugin-tmp-bot 1.0.0 → 1.1.0

package/lib/api/truckyAppApi.js

@@ -1,4 +1,5 @@
 const BASE_API_V3 = 'https://api.truckyapp.com/v3'
+const BASE_API_V2 = 'https://api.truckyapp.com/v2'
 
 module.exports = {
   /**
@@ -22,5 +23,27 @@ module.exports = {
       data.data = result.response
     }
     return data
+  },
+  /**
+   * 查询热门交通数据
+   */
+  async trafficTop (http, serverName) {
+    let result = null
+    try {
+      result = await http.get(`${BASE_API_V2}/traffic/top?game=ets2&server=${serverName}`)
+    } catch {
+      return {
+        error: true
+      }
+    }
+
+    // 拼接返回数据
+    let data = {
+      error: !result || !result.response || result.response.length <= 0
+    }
+    if (!data.error) {
+      data.data = result.response
+    }
+    return data
   }
 }

package/lib/command/tmpQuery.js

@@ -10,7 +10,8 @@ const baiduTranslate = require('../util/baiduTranslate')
 const userGroup = {
   'Player': '玩家',
   'Retired Legend': '退役',
-  'Game Developer': '游戏开发者'
+  'Game Developer': '游戏开发者',
+  'Retired Team Member': '退休团队成员'
 }
 
 /**

package/lib/command/tmpTraffic.js

@@ -0,0 +1,65 @@
+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': '🔴拥堵'
+}
+
+/**
+ * 位置类型转中文
+ */
+const typeToZh = {
+  'City': '城市',
+  'Road': '公路'
+}
+
+/**
+ * 查询路况
+ */
+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
+    console.info(traffic)
+  }
+
+  return message
+}

package/lib/index.js

@@ -6,6 +6,7 @@ const model = require('./database/model');
 const tmpQuery = require('./command/tmpQuery');
 const tmpServer = require('./command/tmpServer');
 const tmpBind = require('./command/tmpBind');
+const tmpTraffic = require('./command/tmpTraffic');
 exports.name = 'tmp-bot';
 exports.Config = koishi_1.Schema.object({
     baiduTranslateEnable: koishi_1.Schema.boolean().default(false).description('启用百度翻译'),
@@ -18,6 +19,7 @@ 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));
+    ctx.command('tmptraffic <serverName>').action(async ({ session }, serverName) => await tmpTraffic(ctx, cfg, serverName));
     // 等待数据库模块准备完毕后初始化数据库表
     let databaseTime = setInterval(() => {
         if (ctx.database) {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-tmp-bot",
   "description": "欧洲卡车模拟2 TMP查询机器人",
-  "version": "1.0.0",
+  "version": "1.1.0",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [
@@ -21,7 +21,9 @@
   ],
   "koishi": {
     "service": {
-      "required": ["database"]
+      "required": [
+        "database"
+      ]
     }
   },
   "peerDependencies": {