koishi-plugin-tmp-bot 1.8.2 → 1.8.4

package/lib/api/truckersMpApi.js

@@ -0,0 +1,73 @@
+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
+  },
+  /**
+   * 查询玩家封禁信息
+   */
+  async bans (http, tmpId) {
+    let result = null
+    try {
+      result = await http.get(`${BASE_API}/bans/${tmpId}`)
+    } catch {
+      return {
+        error: true
+      }
+    }
+
+    // 拼接返回数据
+    let data = {
+      error: JSON.parse(result.error)
+    }
+    if (!data.error) {
+      data.data = result.response
+    }
+
+    return data
+  }
+}

package/lib/api/truckersMpMapApi.js

@@ -0,0 +1,26 @@
+const BASE_API = 'https://tracker.ets2map.com'
+
+module.exports = {
+  /**
+   * 区域查询玩家
+   */
+  async area (http, serverId, x1, y1, x2, y2) {
+    let result = null
+    try {
+      result = await http.get(`${BASE_API}/v3/area?x1=${x1}&y1=${y1}&x2=${x2}&y2=${y2}&server=${serverId}`)
+    } catch {
+      return {
+        error: true
+      }
+    }
+
+    // 拼接返回数据
+    let data = {
+      error: !result || !result.Success
+    }
+    if (!data.error) {
+      data.data = result.Data
+    }
+    return data
+  }
+}

package/lib/api/truckyAppApi.js

@@ -0,0 +1,49 @@
+const BASE_API_V3 = 'https://api.truckyapp.com/v3'
+const BASE_API_V2 = 'https://api.truckyapp.com/v2'
+
+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
+  },
+  /**
+   * 查询热门交通数据
+   */
+  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/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.userId, session.author.username, tmpId)
+
+  return `绑定成功 ( ${playerInfo.data.name} )`
+}

package/lib/command/tmpPosition.js

@@ -0,0 +1,107 @@
+const { segment } = require('koishi')
+const { resolve } = require('path')
+const guildBind = require('../database/guildBind')
+const truckyAppApi = require('../api/truckyAppApi')
+const truckersMpApi = require('../api/truckersMpApi')
+const truckersMpMapApi = require('../api/truckersMpMapApi')
+const baiduTranslate = require('../util/baiduTranslate')
+const common = require('../util/common')
+
+/**
+ * 定位
+ */
+module.exports = async (ctx, cfg, session, tmpId) => {
+  if (ctx.puppeteer) {
+    if (tmpId && isNaN(tmpId)) {
+      return `请输入正确的玩家编号`
+    }
+
+    // 如果没有传入tmpId，尝试从数据库查询绑定信息
+    if (!tmpId) {
+      let guildBindData = await guildBind.get(ctx.database, session.platform, 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 '查询玩家信息失败，请重试'
+    }
+    if (!playerMapInfo.data.online) {
+      return '玩家离线'
+    }
+
+    // 查询周边玩家，并处理数据
+    let areaPlayersData = await truckersMpMapApi.area(ctx.http, playerMapInfo.data.server,
+        playerMapInfo.data.x - 4000,
+        playerMapInfo.data.y + 2500,
+        playerMapInfo.data.x + 4000,
+        playerMapInfo.data.y - 2500)
+    let areaPlayerList = []
+    if (!areaPlayersData.error) {
+      areaPlayerList = areaPlayersData.data
+      let index = areaPlayerList.findIndex((player) => {
+        return player.MpId.toString() === tmpId
+      })
+      if (index !== -1) {
+        areaPlayerList.splice(index, 1)
+      }
+    }
+    areaPlayerList.push({
+      X: playerMapInfo.data.x,
+      Y: playerMapInfo.data.y,
+      MpId: tmpId
+    })
+
+    // promods服ID集合
+    let promodsServerIdList = [50, 51]
+
+    // 构建地图数据
+    let data = {
+      mapType: promodsServerIdList.indexOf(playerMapInfo.data.server) !== -1 ? 'promods' : 'ets',
+      avatar: playerInfo.data.smallAvatar,
+      username: playerInfo.data.name,
+      serverName: playerMapInfo.data.serverDetails.name,
+      country: await baiduTranslate(ctx, cfg, playerMapInfo.data.location.poi.country),
+      realName: await baiduTranslate(ctx, cfg, playerMapInfo.data.location.poi.realName),
+      currentPlayerId: tmpId,
+      centerX: playerMapInfo.data.x,
+      centerY: playerMapInfo.data.y,
+      playerList: areaPlayerList
+    }
+
+    let page
+    try {
+      page = await ctx.puppeteer.page()
+      await page.setViewport({ width: 1000, height: 1000 })
+      await page.goto(`file:///${resolve(__dirname, '../resource/position.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 (e) {
+      return '渲染异常，请重试'
+    } finally {
+      if (page) {
+        await page.close()
+      }
+    }
+
+  } else {
+    return '未启用 puppeteer 服务'
+  }
+}

package/lib/command/tmpQuery.js

@@ -0,0 +1,85 @@
+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': '游戏开发者',
+  'Retired Team Member': '退休团队成员',
+  'Add-On Team': 'Add-On Team',
+  'Game Moderator': 'Game Moderator'
+}
+
+/**
+ * 查询玩家信息
+ */
+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.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 '查询玩家信息失败，请重试'
+  }
+
+  // 拼接消息模板
+  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 {
+      let banData = await truckersMpApi.bans(ctx.http, tmpId)
+      if (!banData.error) {
+        let ban = banData.data[0]
+        if (!ban.expiration) {
+          message += '永久'
+        } else {
+          message += dayjs(ban.expiration + 'Z').format('YYYY年MM月DD日 HH:mm')
+        }
+        message += "\n🚫封禁原因: " + await baiduTranslate(ctx, cfg, ban.reason, false)
+      } else {
+        message += '查询失败'
+      }
+    }
+  }
+  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, game) => {
+  // 查询服务器信息
+  let serverData = await truckersMpApi.servers(ctx.http)
+  if (serverData.error) {
+    return '查询服务器失败，请稍后重试'
+  }
+
+  // 过滤服务器
+  let etsServerList = serverData.data.filter(server => server.game === game)
+
+  // 构建消息
+  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/command/tmpTraffic/tmpTraffic.js

@@ -0,0 +1,16 @@
+const tmpTrafficMap = require("./tmpTrafficMap");
+const tmpTrafficText = require("./tmpTrafficText");
+
+/**
+ * 查询路况
+ */
+module.exports = async (ctx, cfg, serverName) => {
+  switch (cfg.tmpTrafficType) {
+    case 1:
+      return await tmpTrafficText(ctx, cfg, serverName)
+    case 2:
+      return await tmpTrafficMap(ctx, cfg, serverName)
+    default:
+      return '指令配置错误'
+  }
+}

package/lib/command/tmpTraffic/tmpTrafficMap.js

@@ -0,0 +1,128 @@
+const truckyAppApi = require('../../api/truckyAppApi')
+const truckersMpMapApi = require('../../api/truckersMpMapApi')
+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 truckersMpMapApi.area(ctx.http, serverInfo.serverId, serverInfo.bounds[0][0], serverInfo.bounds[0][1], serverInfo.bounds[1][0], serverInfo.bounds[1][1])
+
+  // 构建路况数据
+  let data = {
+    mapType: serverInfo.mapType,
+    trafficList: [],
+    playerCoordinateList: mapData.error && mapData.data ? [] : mapData.data.map(item => [item.X, item.Y])
+  }
+  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()
+    }
+  }
+}

package/lib/command/tmpTraffic/tmpTrafficText.js

@@ -0,0 +1,66 @@
+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/database/guildBind.js

@@ -0,0 +1,45 @@
+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 userName 用户昵称
+   * @param tmpId TMP ID
+   */
+  saveOrUpdate (db, platform, userId, userName, tmpId) {
+    this.get(db, platform, 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,
+          user_id: userId,
+          user_name: userName,
+          tmp_id: tmpId
+        })
+      }
+    })
+  }
+}