miao-plugin/models/Avatars.js

import Base from './Base.js'
import lodash from 'lodash'
import { Data, Common } from '../components/index.js'
import { Artifact, Character } from './index.js'

export default class Avatars extends Base {
  constructor (uid, datas = []) {
    super()
    if (!uid) {
      return false
    }
    this.uid = uid
    let avatars = {}
    let abbr = Character.getAbbr()
    lodash.forEach(datas, (avatar) => {
      let data = Data.getData(avatar, 'id,name,level,star:rarity,cons:actived_constellation_num,fetter')
      data.elem = (avatar.element || '').toLowerCase() || 'anemo'
      let char = Character.get({ id: data.id, elem: data.elem })
      if (char.isTraveler) {
        char.setTraveler(uid)
      }
      data.face = char.face
      data.side = char.side
      data.abbr = char.abbr
      data.weapon = Data.getData(avatar.weapon, 'name,affix:affix_level,level,star:rarity')
      data.weapon.abbr = abbr[data?.weapon?.name || ''] || data?.weapon?.name
      if (data.star > 5) {
        data.star = 5
      }
      let artis = {}
      let setCount = {}
      lodash.forEach(avatar.reliquaries, (arti) => {
        artis[arti.pos] = Data.getData(arti, 'name,level,set:set.name')
        setCount[arti.set.name] = (setCount[arti.set.name] || 0) + 1
      })
      data.artis = artis
      data.sets = {}
      data.names = []
      for (let set in setCount) {
        if (setCount[set] >= 2) {
          data.sets[set] = setCount[set] >= 4 ? 4 : 2
          data.names.push(Artifact.getArtiBySet(set))
        }
      }
      avatars[data.id] = data
    })
    this.avatars = avatars
  }

  getData (ids) {
    let rets = {}
    let avatars = this.avatars
    lodash.forEach(ids, (id) => {
      rets[id] = avatars[id] || {}
    })
    return rets
  }

  getIds () {
    let rets = []
    lodash.forEach(this.avatars, (ds) => {
      rets.push(ds.id)
    })
    return rets
  }

  async getTalentData (ids, MysApi = false) {
    let avatarTalent = await Data.getCacheJSON(`genshin:avatar-talent:${this.uid}`)
    let needReq = {}
    lodash.forEach(ids, (id) => {
      if (!avatarTalent[id]) {
        needReq[id] = true
      }
    })
    let needReqIds = lodash.keys(needReq)
    if (needReqIds.length > 0 && MysApi && MysApi.isSelfCookie) {
      let num = 10
      let ms = 100
      let skillRet = []
      let avatarArr = lodash.chunk(needReqIds, num)
      for (let val of avatarArr) {
        for (let id of val) {
          skillRet.push(this.getAvatarTalent(id, MysApi))
        }
        skillRet = await Promise.all(skillRet)
        skillRet = skillRet.filter(item => item.id)
        await Common.sleep(ms)
      }
      lodash.forEach(skillRet, (talent) => {
        avatarTalent[talent.id] = talent
      })
      await Data.setCacheJSON(`genshin:avatar-talent:${this.uid}`, avatarTalent, 3600 * 2)
    }
    let ret = this.getData(ids)
    lodash.forEach(ret, (avatar, id) => {
      avatar.talent = avatarTalent[id] || { a: {}, e: {}, q: {} }
    })
    return ret
  }

  async getAvatarTalent (id, MysApi) {
    let talent = {}
    let talentRes = await MysApi.getDetail(id)
    let char = Character.get(id)
    let avatar = this.avatars[id]
    if (!char || !avatar) {
      return talent
    }
    if (talentRes && talentRes.skill_list) {
      talent.id = id
      let talentList = lodash.orderBy(talentRes.skill_list, ['id'], ['asc'])
      for (let val of talentList) {
        let { max_level: maxLv, level_current: lv } = val
        if (val.name.includes('普通攻击')) {
          talent.a = lv
          continue
        }
        if (maxLv >= 10 && !talent.e) {
          talent.e = lv
          continue
        }
        if (maxLv >= 10 && !talent.q) {
          talent.q = lv
          continue
        }
      }
    }
    let ret = char.getAvatarTalent(talent, avatar.cons, 'original')
    ret.id = id
    return ret
  }
}

Avatars.hasTalentCache = async function (uid) {
  return !!await redis.get(`genshin:avatar-talent:${uid}`)
}