miao-plugin/models/MysApi.js

import { User } from './index.js'
import { Version } from '#miao'

export default class MysApi {
  constructor (e, uid, mysInfo) {
    this.e = e
    this.mysInfo = mysInfo
    this.ckInfo = mysInfo.ckInfo
    this.ckUser = mysInfo.ckUser
    this.uid = uid
    e.targetUser = this.targetUser
    e.selfUser = this.selfUser
    e.isSelfCookie = this.isSelfCookie
  }

  get isSelfCookie () {
    return this.uid * 1 === this.ckUid * 1 || this?.mysInfo?.isSelf
  }

  get ckUid () {
    return this.ckInfo.uid
  }

  get ck () {
    return this.ckInfo.ck
  }

  get selfUser () {
    return new User({ id: this.e.user_id, uid: this.uid })
  }

  get targetUser () {
    return new User({ id: this.e.user_id, uid: this.uid })
  }

  static async init (e, auth = 'all') {
    if (!e.runtime) {
      Version.runtime()
      return false
    }
    let mys = await e.runtime.getMysInfo(auth)
    if (!mys) {
      return false
    }
    let uid = mys.uid
    e._mys = new MysApi(e, uid, mys)
    return e._mys
  }

  static async initUser (e, auth = 'all') {
    let { runtime } = e
    if (!runtime) {
      Version.runtime()
      return false
    }
    let uid
    if (runtime.getUid) {
      uid = await runtime.getUid()
    } else {
      // 兼容处理老版本Yunzai
      uid = runtime.uid || e.uid
      if (e.at) {
        // 暂时使用MysApi.init替代
        let mys = await MysApi.init(e, auth)
        if (!mys) {
          return false
        }
        uid = mys.uid || uid
      }
    }
    if (uid) {
      return new User({ id: e.user_id, uid })
    } else {
      e.reply('请先发送【#绑定+你的UID】来绑定查询目标\n星铁请使用【#星铁绑定+UID】')
      e._replyNeedUid = true
      return false
    }
  }

  async getMysApi (e, targetType = 'all', option = {}) {
    if (this.mys) {
      return this.mys
    }
    this.mys = await e.runtime.getMysApi(targetType, option)
    return this.mys
  }

  async getData (api, data) {
    if (!this.mysInfo) {
      return false
    }
    let e = this.e
    let mys = await this.getMysApi(e, api, { log: false })
    if (!mys) {
      return false
    }
    let mysInfo = this.mysInfo || {}
    // 暂时先在plugin侧阻止错误，防止刷屏
    e._original_reply = e._original_reply || e.reply
    e._reqCount = e._reqCount || 0
    e.reply = function (msg) {
      if (!e._isReplyed) {
        e._isReplyed = true
        return e._original_reply(msg)
      } else {
        // console.log('请求错误')
      }
    }
    e._reqCount++
    let ret = await mys.getData(api, data)
    if (mysInfo && mysInfo.checkCode) {
      ret = await mysInfo.checkCode(ret, api, this.mys)
    }
    e._reqCount--
    if (e._reqCount === 0) {
      e.reply = e._original_reply
    }
    if (!ret) {
      return false
    }
    if (ret.retcode !== 0) {
      e._retcode = ret.retcode
    }
    return ret.data || ret
  }

  // 获取角色信息
  async getCharacter () {
    return await this.getData('character')
  }

  // 获取角色详情
  async getAvatar (id) {
    return await this.getData('detail', { avatar_id: id })
  }

  // 首页宝箱信息
  async getIndex () {
    return await this.getData('index')
  }

  // 获取深渊信息
  async getSpiralAbyss (type = 1) {
    return await this.getData('spiralAbyss', { schedule_type: type })
  }

  async getDetail (id) {
    return await this.getData('detail', { avatar_id: id })
  }

  async getCompute (data) {
    return await this.getData('compute', data)
  }

  async getAvatarSkill (id) {
    return await this.getData('avatarSkill', { avatar_id: id })
  }
}