Miao-Yunzai/plugins/genshin/model/mys/NoteUser.js

/**
 * Bot实际User用户类
 * 主键QQ
 *
 *  User可以注册UID，通过 getRegUid / setRegUid
 *  一个User可以绑定多个MysUser CK，绑定MysUser
 */
import BaseModel from './BaseModel.js'
import lodash from 'lodash'
import MysUser from './MysUser.js'
import MysUtil from './MysUtil.js'
import { UserDB } from '../db/index.js'
import { Data } from '#miao'

export default class NoteUser extends BaseModel {
  constructor (qq) {
    super()
    // 检查实例缓存
    let cacheObj = this._getThis('user', qq)
    if (cacheObj) {
      return cacheObj
    }
    this.qq = qq
    return this._cacheThis()
  }

  get uid () {
    console.log('NoteUser.uid 默认返回原神UID，可更改为 user.getUid(game)')
    return this.getUid()
  }

  /**
   * 当前用户是否具备CK
   */
  get hasCk () {
    return !lodash.isEmpty(this.mysUsers)
  }

  /**
   * 获取绑定CK的UID列表，如未绑定CK则返回空数组
   */
  get ckUids () {
    if (!this.hasCk) {
      return []
    }
    let ret = []
    return lodash.map(this.ckData, 'uid')
  }

  /**
   * 获取当前用户的所有ck
   * @returns { {ltuid:{ckData, ck, uids}} }
   */
  get cks () {
    console.log('NoteUser.cks 即将废弃')
    let game = 'gs'
    let cks = {}
    if (!this.hasCk) {
      return cks
    }
    for (let ltuid in this.mysUsers) {
      let mys = this.mysUsers[ltuid]
      if (mys && mys.ltuid && mys.uid) {
        cks[ltuid] = cks[ltuid] || {
          ckData: mys.getCkInfo(game),
          ck: mys.ck,
          uids: mys.getUids(game)
        }
      }
    }
    return cks
  }

  /**
   * 创建NoteUser实例
   * @param qq NoterUser对应id（qq）
   * @param db
   * * 若传入e对象则会识别e.user_id，并将user对象添加至e.user
   * @param data 用户对应MysCookie数据，为空则自动读取
   * @returns {Promise<NoteUser|*>}
   */
  static async create (qq, db = false) {
    // 兼容处理传入e
    if (qq && qq.user_id) {
      let e = qq
      let user = await NoteUser.create(e.user_id)
      e.user = user
      return user
    }

    let user = new NoteUser(qq)
    await user.initDB(db)

    // 传入data则使用，否则读取
    return user
  }

  static async forEach (fn) {
    let dbs = await UserDB.findAll()
    await Data.forEach(users, async (db) => {
      let user = await NoteUser.create(db.id, db)
      return await fn(user)
    })
  }

  // 初始化数据
  async initDB (db = false) {
    if (this.db && !db) {
      return
    }
    // 为后续多类型用户兼容
    this.db = db && db !== true ? db : await UserDB.find(this.qq, 'qq')
    await this.initMysUser()
    this.initUids()
    await this.save()
  }

  // 初始化MysUser对象
  async initMysUser () {
    let ltuids = this.db?.ltuids || ''
    this.mysUsers = {}
    for (let ltuid of ltuids.split(',')) {
      let mys = await MysUser.create(ltuid)
      if (mys) {
        this.mysUsers[ltuid] = mys
      }
    }
  }

  // 初始化Uid
  initUids (setMainUid = {}) {
    let self = this
    self.mainUid = self.mainUid || {}
    self.uidList = {}
    self.uidMap = self.uidMap || {}
    self.games = {}
    const { db, mainUid, uidList, games, uidMap, mysUsers } = self

    let gameDBs = {}
    lodash.forEach(db?.games, (gameDB) => {
      gameDBs[gameDB.game] = gameDB
    })

    MysUtil.eachGame((key) => {
      let gameDB = gameDBs[key]
      uidMap[key] = {}
      uidList[key] = []
      games[key] = gameDB
      // 优先设置CK UID
      lodash.forEach(mysUsers, (mys) => {
        lodash.forEach(mys.uids[key] || [], (uid) => {
          uid = uid + ''
          if (uid && !uidMap[key][uid]) {
            uidMap[key][uid] = { uid, type: 'ck', ltuid: mys.ltuid }
            uidList[key].push(uid)
          }
        })
      })

      let uidReg = /\d{9}/
      let regUidCount = 0

      // 存在数据库记录则进行设置
      if (gameDB) {
        let regUids = gameDB.data
        // 依次设置verify、reg uid数据
        lodash.forEach(['verify', 'reg'], (uidType) => {
          lodash.forEach(regUids, (ds, uid) => {
            uid = uid + ''
            if (regUidCount <= 5 && uid && uidReg.test(uid) && ds.type === uidType && !uidMap[key][uid]) {
              uidMap[key][uid] = { uid, type: ds.type }
              uidList[key].push(uid)
              regUidCount++
            }
          })
        })

        // 如果当前选中uid未在记录中，则补充为reg数据
        let uid = gameDB.uid
        if (uid && !uidMap[key][uid]) {
          uid = uid + ''
          uidMap[key][uid] = { uid, type: 'reg' }
          uidList[key].push(uid)
        }
      }
      // 设置选中uid
      if (setMainUid === false || setMainUid[key] === false) {
        mainUid[key] = uidList[key]?.[0] || ''
      } else {
        mainUid[key] = setMainUid[key] || mainUid[key] || gameDB?.uid || uidList[key]?.[0] || ''
      }
    })
  }

  async save () {
    await this.db.saveDB(this)
  }

  // 获取当前UID
  getUid (game = 'gs') {
    let gameKey = this.gameKey(game)
    return this.mainUid[gameKey] || this.uidList[gameKey][0] || ''
  }

  getSelfUid (game = 'gs') {
    let gameKey = this.gameKey(game)
    let uidList = this.uidMap[gameKey].filter((v) => v.type === 'ck')
    if (uidList.length === 0) {
      return false
    }
    let find = lodash.find(uidList, (v) => v.uid + '' === uid + '', 0)
    return find ? find.uid : uidList[0].uid
  }

  // 获取UID列表
  getUidList (game = 'gs') {
    let ret = []
    let gameKey = this.gameKey(game)
    lodash.forEach(this.uidList[gameKey], (uid) => {
      ret.push(this.uidMap[gameKey][uid])
    })
    return ret
  }

  // 获取当前UID数据
  getUidData (game = 'gs') {
    let gameKey = this.gameKey(game)
    let uid = this.getUid(game)
    return this.uidMap[gameKey]?.[uid]
  }

  // 获取当前的MysUser对象
  getMysUser (game = 'gs') {
    if (lodash.isEmpty(this.mysUsers)) {
      return false
    }
    let uidData = this.getUidData(game)
    let ltuid = lodash.keys(this.mysUsers)[0]
    if (uidData.type === 'ck') {
      ltuid = uidData.ltuid
    }
    return this.mysUsers[ltuid]
  }


  // 添加UID
  async addRegUid (uid, game = 'gs') {
    let gameKey = this.gameKey(game)
    uid = uid + ''
    if (!this.uidMap[gameKey][uid]) {
      this.uidMap[gameKey][uid] = { uid, type: 'reg' }
    }
    await this.save()
    this.setMainUid(uid, game)
    // todo 优化保存
    await this.save()
  }

  // 删除UID
  async delRegUid (uid, game = 'gs') {
    let gameKey = this.gameKey(game)
    if (this.uidMap[gameKey][uid] && this.uidMap[gameKey][uid].type !== 'ck') {
      lodash.remove(this.uidList[gameKey], (u) => u + '' === uid + '')
      delete this.uidMap[gameKey][uid]
      if (this.mainUid[gameKey] === uid) {
        this.mainUid[gameKey] = ''
      }
    }
    await this.save()
    if (this.mainUid[gameKey] === '') {
      this.setMainUid(this.uidList[gameKey][0], game)
      await this.save()
    }
  }

  /**
   * 获取当前用户的绑定UID
   * 主要供内部调用，建议使用 user.uid 获取用户uid
   * @returns {Promise<*>}
   */
  async getRegUid (game = 'gs') {
    let gameKey = this.gameKey(game)
    return this.mainUid[gameKey] || ''
  }

  /**
   * 设置当前用户的绑定uid
   * @param uid 要绑定的uid
   * @param game
   * @param force 若已存在绑定uid关系是否强制更新
   */
  async setRegUid (uid = '', game = 'gs', force = false) {
    if (this.getRegUid(game) && !force) {
      return uid
    }
    await this.addRegUid(uid, game)
    return uid
  }

  // 切换绑定CK生效的UID
  setMainUid (uid = '', game = 'gs') {
    let gameKey = this.gameKey(game)
    // 兼容传入index
    if (uid < 100 && this.uidList[gameKey][uid]) {
      uid = this.uidList[gameKey][uid]
    }
    if (this.uidMap[gameKey][uid]) {
      this.mainUid[gameKey] = uid
    }
    let mainUid = {}
    mainUid[gameKey] = uid
    this.initUids(mainUid)
  }

  // 添加MysUser
  addMysUser (mysUser) {
    this.mysUsers[mysUser.ltuid] = mysUser
    this.initUids(mysUser.getMainUid())
  }

  // 删除当前用户绑定CK
  async delCk (ltuid = '') {
    console.log('delCk即将废弃')
    return await this.delMysUser(ltuid)

  }

  async delMysUser (ltuid = '') {
    if (ltuid && this.mysUsers[ltuid]) {
      let mys = this.mysUsers[ltuid]
      delete this.mysUsers[ltuid]
      await mys.del()
    }
    this.initUids(false)
    await this.save()
  }

  /**
   * 检查当前用户绑定的CK状态
   */
  async checkCk () {
    // TODO:待完善文案
    let cks = this.cks
    let ret = []
    for (let ltuid in cks) {
      let ck = cks[ltuid].ck
      if (!ltuid || !ck) {
        continue
      }
      let checkRet = await MysUser.checkCkStatus(ck)
      // TODO: 若checkRet中返回了不同的uid，进行CK保存更新
      // 失效
      let mysUser = await MysUser.create(ck)
      if (mysUser) {
        let status = checkRet.status
        if (status === 0 || status === 1) {
          // status为1时无法查询天赋，但仍可查询角色，保留CK
          await mysUser.initCache()
        } else if (status === 2) {
          // status为2时无法查询角色，删除ck cache
          // 因仍能查询体力，故保留ck记录不直接删除
          await mysUser.del()
        } else if (status === 3) {
          // status为3时CK完全失效，用户删除此CK
          await this.delCk(ltuid)
        }
      }
      ret.push({
        ltuid,
        ...checkRet
      })
    }
    return ret
  }
}