Miao-Yunzai/plugins/genshin/model/calculator.js

import base from './base.js'
import MysInfo from './mys/mysInfo.js'
import MysApi from './mys/mysApi.js'
import lodash from 'lodash'
import gsCfg from './gsCfg.js'

export default class Calculator extends base {
  constructor (e) {
    super(e)
    this.model = 'calculator'
    this.checkMsg = `设置角色、武器、技能等级有误\n指令：${e.isSr ? '*克拉拉养成\n示例：*克拉拉养成75 80 6 9 9 9\n参数为角色、武器、普攻、战技、终结技、天赋' : '#刻晴养成\n示例：#刻晴养成81 90 9 9 9\n参数为角色、武器、技能等级'}`
  }

  async get (role) {
    this.role = role
    /** 获取绑定uid */
    let uid = await MysInfo.getUid(this.e)
    if (!uid) return false

    /** 判断是否绑定了ck */
    let ck = await MysInfo.checkUidBing(uid, this.e)
    if (!ck) {
      await this.e.reply(MysInfo.tips)
      return false
    }

    this.mysApi = new MysApi(uid, ck.ck, { log: true })

    let device_fp = await MysInfo.get(this.e, 'getFp')
    this.headers = {
      'x-rpc-device_fp': device_fp?.data?.device_fp
    }

    /** 获取角色数据 */
    let character = await MysInfo.get(this.e, this.e.isSr ? 'avatarInfo' : 'character', {
      headers: this.headers
    })
    if (!character || character.retcode !== 0) return false
    character = character.data

    /** 获取设置参数 */
    await this.getSet()

    /** 获取计算角色 */
    this.dataCharacter = character[this.e.isSr ? 'avatar_list' : 'avatars'].find((item) => item.id == role.roleId)

    /** 获取计算参数 */
    let body = await this.getBody()
    if (!body) return false

    /** 计算 */
    let computes = await this.computes(body)
    if (!computes) return false

    return {
      saveId: uid,
      uid,
      dataCharacter: this.dataCharacter,
      setSkill: this.setSkill,
      skillList: this.skillList,
      computes,
      ...this.screenData
    }
  }

  async getSet () {
    let defSetSkill = this.e.isSr ? '80,80,6,10,10,10'.split(',') : '90,90,10,10,10'.split(',')

    let set = this.e.msg.replace(/#|＃|星铁|养成|计算/g, '').trim()

    set = set.replace(/，| /g, ',')

    set = set.replace(this.role.alias, '')

    let setSkill = []
    let length = this.e.isSr ? 6 : 5
    if (set) {
      setSkill = set.split(',')
      setSkill = lodash.compact(setSkill)
      for (let i = 0; i < length; i++) {
        if (!setSkill[i]) setSkill[i] = defSetSkill[i]
      }
    } else {
      setSkill = defSetSkill
    }

    if (setSkill.length != length) {
      let reMsg = this.checkMsg.replace(/刻晴|克拉拉/g, this.role.alias)
      await this.e.reply(reMsg)
      return false
    }

    /** 检查参数 */
    let check = this.e.isSr ? [80, 80, 6, 10, 10, 10] : [90, 90, 10, 10, 10]
    for (const key in check) {
      if (check[key] < Number(setSkill[key])) {
        setSkill[key] = check[key]
      }
    }

    this.setSkill = setSkill
  }

  async getBody () {
    // 技能
    let skillList = []
    if (this.dataCharacter) {
      /** 角色存在获取技能数据 */
      let data = this.e.isSr ? { avatar_id: this.role.roleId, tab_from: 'TabOwned' } : { avatar_id: this.role.roleId }
      let detail = await MysInfo.get(this.e, 'detail', {
        headers: this.headers,
        ...data
      })
      if (!detail || detail.retcode !== 0) return false

      skillList = detail.data.skill_list || detail.data.skills
    } else {
      /** 尚未拥有的角色 */
      if (this.e.isSr) {
        let detail = await MysInfo.get(this.e, 'detail', {
          headers: this.headers,
          avatar_id: this.role.roleId,
          tab_from: 'TabAll'
        })
        if (!detail || detail.retcode !== 0) return false

        this.avatar = detail.data.avatar
        skillList = detail.data.skills
      } else {
        skillList = await this.getSkillId(this.role.roleId)
      }

      if (!skillList) {
        this.e.reply('暂无角色数据,请稍后再试\n星铁角色养成请使用*开头')
        return false
      }

      let four = gsCfg.getdefSet('role', 'other').four

      this.dataCharacter = {
        level: 1,
        name: this.role.name,
        icon: this.e.isSr ? this.avatar.icon_url : `${this.screenData.pluResPath}img/role/${this.role.name}.png`,
        rarity: this.e.isSr ? this.avatar.rarity : four.includes(Number(this.role.roleId)) ? 4 : 5
      }
    }

    /** 拼接计算参数 */
    let body
    if (this.e.isSr) {
      body = {
        game: 'hkrpg',
        avatar: {
          item_id: Number(this.role.roleId),
          cur_level: Number(this.dataCharacter.level),
          target_level: Number(this.setSkill[0])
        },
        skill_list: []
      }

      let srSkills = this.setSkill.slice(2)
      skillList.forEach((v, k) => {
        body.skill_list.push({
          item_id: v.point_id,
          cur_level: v.cur_level,
          target_level: Number(srSkills[k]) || v.target_level
        })
      })
    } else {
      skillList = skillList.filter((item) => item.max_level != 1)

      body = {
        avatar_id: Number(this.role.roleId),
        avatar_level_current: Number(this.dataCharacter.level),
        avatar_level_target: Number(this.setSkill[0]),
        skill_list: [
          {
            id: Number(skillList[0].group_id),
            level_current: Number(skillList[0].level_current),
            level_target: Number(this.setSkill[2])
          },
          {
            id: Number(skillList[1].group_id),
            level_current: Number(skillList[1].level_current),
            level_target: Number(this.setSkill[3])
          },
          {
            id: Number(skillList[2].group_id),
            level_current: Number(skillList[2].level_current),
            level_target: Number(this.setSkill[4])
          }
        ]
      }
    }

    if (this.mysApi.getServer().startsWith('os')) {
      body.lang = 'zh-cn'
    }

    if (this.e.isSr) {
      if (this.dataCharacter.equip) {
        body.equipment = {
          item_id: Number(this.dataCharacter.equip.id),
          cur_level: Number(this.dataCharacter.equip.level),
          target_level: Number(this.setSkill[1])
        }
      }
    } else {
      if (this.dataCharacter.weapon) {
        if (Number(this.dataCharacter.weapon.rarity) < 3) {
          this.setSkill[1] = 70
        }

        body.weapon = {
          id: Number(this.dataCharacter.weapon.id),
          level_current: Number(this.dataCharacter.weapon.level),
          level_target: Number(this.setSkill[1])
        }
      }
    }

    skillList = skillList.filter((item) => item.max_level != 1)
    this.skillList = skillList
    return this.e.isSr ? body : { items: [body] }
  }

  async getSkillId (roleId) {
    let avatarSkill = await MysInfo.get(this.e, 'avatarSkill', {
      headers: this.headers,
      avatar_id: roleId
    })
    if (!avatarSkill || avatarSkill.retcode !== 0) return false
    avatarSkill = avatarSkill.data
    avatarSkill.list.forEach((item) => {
      item.level_current = 1
    })

    return avatarSkill.list
  }

  async computes (body) {
    let computes = await MysInfo.get(this.e, 'compute', {
      body,
      headers: this.headers
    })
    if (!computes || computes.retcode !== 0) return false
    computes = this.e.isSr ? computes.data : computes.data.overall_material_consume
    let computeList = {}

    let formart = (num) => {
      return num > 10000 ? (num / 10000).toFixed(1) + ' w' : num
    }
    if (this.e.isSr) delete computes.coin_id
    for (let i in computes) {
      computeList[i] = []
      if (!this.e.isSr) {
        computes[i].forEach(({ consume }) => {
          consume.forEach(val => {
            computeList[i].push(val)
            val.num = formart(val.num)
            if (val.name.includes('「')) {
              val.isTalent = true
            }
          })
        })
      } else {
        for (let j in computes[i]) {
          computes[i][j].num = formart(computes[i][j].num)

          if (computes[i][j].item_name.includes('「')) {
            computes[i][j].isTalent = true
          }
          computeList[i].push(computes[i][j])
        }
      }
    }

    return computeList
  }
}