509 lines
10 KiB
TypeScript
509 lines
10 KiB
TypeScript
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.js'
|
||
/**
|
||
* *******************
|
||
* Bot实际User用户类
|
||
* 主键QQ
|
||
*
|
||
* User可以注册UID，通过 getRegUid / setRegUid
|
||
* 一个User可以绑定多个MysUser CK，绑定MysUser
|
||
* *******************
|
||
*/
|
||
export default class NoteUser extends BaseModel {
|
||
db = null
|
||
|
||
qq = null
|
||
|
||
mysUsers: any = {}
|
||
|
||
_map = null
|
||
/**
|
||
*
|
||
* @param qq
|
||
* @returns
|
||
*/
|
||
constructor(qq) {
|
||
super()
|
||
// 检查实例缓存
|
||
let cacheObj = this._getThis('user', qq)
|
||
if (cacheObj) {
|
||
return cacheObj
|
||
}
|
||
this.qq = qq
|
||
return this._cacheThis()
|
||
}
|
||
|
||
/**
|
||
* OLD Func {{
|
||
*/
|
||
|
||
/**
|
||
*
|
||
*/
|
||
get uid() {
|
||
console.warn('NoteUser.uid 默认返回原神UID，可更改为 user.getUid(game)')
|
||
return this.getUid()
|
||
}
|
||
|
||
/**
|
||
* 获取绑定CK的UID列表，如未绑定CK则返回空数组
|
||
*/
|
||
get ckUids() {
|
||
console.warn(
|
||
'NoteUser.ckUids 默认返回原神UID，可更改为 user.getCkUidList(game)'
|
||
)
|
||
let uids = this.getCkUidList('gs')
|
||
return lodash.map(uids, ds => ds.uid)
|
||
}
|
||
|
||
/**
|
||
* 获取当前用户的所有ck
|
||
* @returns { {ltuid:{ckData, ck, uids}} }
|
||
*/
|
||
get cks() {
|
||
console.warn('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
|
||
}
|
||
|
||
/**
|
||
* End OLD Func }}
|
||
*/
|
||
|
||
// 当前用户是否具备CK
|
||
get hasCk() {
|
||
return !lodash.isEmpty(this.mysUsers)
|
||
}
|
||
|
||
/**
|
||
* 创建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 id = e.originalUserId || e.user_id
|
||
let mainId = await redis.get(`Yz:NoteUser:mainId:${e.user_id}`)
|
||
if (mainId) {
|
||
id = mainId
|
||
e.mainUserId = mainId
|
||
e.originalUserId = e.originalUserId || e.user_id
|
||
}
|
||
let user = await NoteUser.create(id)
|
||
e.user = user
|
||
return user
|
||
}
|
||
|
||
let user = new NoteUser(qq)
|
||
await user.initDB(db)
|
||
|
||
// 传入data则使用，否则读取
|
||
return user
|
||
}
|
||
|
||
/**
|
||
*
|
||
* @param fn
|
||
*/
|
||
static async forEach(fn) {
|
||
let dbs = await UserDB.findAll()
|
||
await Data.forEach(dbs, async db => {
|
||
let user = await NoteUser.create(db.id, db)
|
||
return await fn(user)
|
||
})
|
||
}
|
||
|
||
/**
|
||
* 初始化数据
|
||
* @param db
|
||
* @returns
|
||
*/
|
||
async initDB(db = false) {
|
||
if (this.db && !db) {
|
||
return
|
||
}
|
||
if (db && db !== true) {
|
||
this.db = db
|
||
} else {
|
||
this.db = await UserDB.find(this.qq, 'qq')
|
||
}
|
||
await this.initMysUser()
|
||
this._games = this.db.games
|
||
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
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
*
|
||
*/
|
||
async save() {
|
||
await this.db.saveDB(this)
|
||
}
|
||
|
||
/**
|
||
*
|
||
* @param game
|
||
* @param type
|
||
* @returns
|
||
*/
|
||
getUidMapList(game = 'gs', type = 'all') {
|
||
if (this._map?.[game]?.[type]) {
|
||
return this._map[game][type]
|
||
}
|
||
game = this.gameKey(game)
|
||
let uidMap = {}
|
||
let uidList = []
|
||
lodash.forEach(this.mysUsers, mys => {
|
||
if (!mys) {
|
||
return
|
||
}
|
||
lodash.forEach(mys.uids[game] || [], uid => {
|
||
uid = uid + ''
|
||
if (uid && !uidMap[uid]) {
|
||
uidMap[uid] = mys.getUidData(uid, game)
|
||
uidList.push(uidMap[uid])
|
||
}
|
||
})
|
||
})
|
||
if (type === 'all') {
|
||
let gameDs = this.getGameDs(game)
|
||
lodash.forEach(gameDs.data, ds => {
|
||
if (ds.uid && !uidMap[ds.uid]) {
|
||
uidMap[ds.uid] = ds
|
||
uidList.push(ds)
|
||
}
|
||
})
|
||
}
|
||
|
||
this._map = this._map || {}
|
||
this._map[game] = this._map[game] || {}
|
||
this._map[game][type] = {
|
||
map: uidMap,
|
||
list: uidList
|
||
}
|
||
return this._map[game][type]
|
||
}
|
||
|
||
/**
|
||
*
|
||
* @param uid
|
||
* @param game
|
||
* @returns
|
||
*/
|
||
getUidData(uid = '', game = 'gs') {
|
||
if (!uid) {
|
||
uid = this.getUid(game)
|
||
}
|
||
return this.getUidMapList(game, 'all').map[uid]
|
||
}
|
||
|
||
/**
|
||
* 有Uid
|
||
* @param uid
|
||
* @param game
|
||
* @returns
|
||
*/
|
||
hasUid(uid = '', game = '') {
|
||
if (!uid) {
|
||
return this.getUidMapList(game, 'all').list?.length > 0
|
||
}
|
||
return !!this.getUidData(uid, game)
|
||
}
|
||
|
||
/**
|
||
* 获取CK-Uid
|
||
* @param game
|
||
* @returns
|
||
*/
|
||
getCkUid(game = 'gs') {
|
||
let uid = this.getUid(game)
|
||
let { map, list } = this.getUidMapList(game, 'ck')
|
||
return (map[uid] ? uid : list[0]?.uid) || ''
|
||
}
|
||
|
||
/**
|
||
* 获取CK-Uid列表
|
||
* @param game
|
||
* @returns
|
||
*/
|
||
getCkUidList(game = 'gs') {
|
||
return this.getUidMapList(game, 'ck').list
|
||
}
|
||
|
||
/**
|
||
* 获取当前UID
|
||
* @param game
|
||
* @returns
|
||
*/
|
||
getUid(game = 'gs') {
|
||
game = this.gameKey(game)
|
||
// todo 刷新uid
|
||
let ds = this.getGameDs(game)
|
||
if (!ds.uid) {
|
||
this.setMainUid('', game)
|
||
}
|
||
return ds.uid || ''
|
||
}
|
||
|
||
/**
|
||
* 获取UID列表
|
||
* @param game
|
||
* @returns
|
||
*/
|
||
getUidList(game = 'gs') {
|
||
return this.getUidMapList(game, 'all').list
|
||
}
|
||
|
||
/**
|
||
* 获取当前的MysUser对象
|
||
* @param game
|
||
* @returns
|
||
*/
|
||
getMysUser(game = 'gs') {
|
||
if (lodash.isEmpty(this.mysUsers)) {
|
||
return false
|
||
}
|
||
let uid = this.getCkUid(game)
|
||
if (!uid) {
|
||
return false
|
||
}
|
||
let uidData = this.getUidData(uid, game)
|
||
return this.mysUsers[uidData.ltuid]
|
||
}
|
||
|
||
/**
|
||
* 添加UID
|
||
* @param uid
|
||
* @param game
|
||
* @param save
|
||
*/
|
||
addRegUid(uid, game = 'gs', save = true) {
|
||
game = this.gameKey(game)
|
||
uid = uid + ''
|
||
let gameDs = this.getGameDs(game)
|
||
gameDs.data[uid] = { uid, type: 'reg' }
|
||
this._map = false
|
||
this.setMainUid(uid, game, false)
|
||
if (save) {
|
||
this.save()
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 删除UID
|
||
* @param uid
|
||
* @param game
|
||
*/
|
||
delRegUid(uid, game = 'gs') {
|
||
game = this.gameKey(game)
|
||
let gameDs = this.getGameDs(game)
|
||
let dsData = gameDs.data
|
||
delete dsData[uid]
|
||
gameDs.data = dsData
|
||
this._map = false
|
||
if (gameDs.uid === uid) {
|
||
this.setMainUid('', game, false)
|
||
}
|
||
this.save()
|
||
}
|
||
|
||
_games = {}
|
||
|
||
/**
|
||
*
|
||
* @param game
|
||
* @returns
|
||
*/
|
||
getGameDs(game = 'gs') {
|
||
game = this.gameKey(game)
|
||
if (!this._games) {
|
||
this._games = {}
|
||
}
|
||
if (!this._games[game]) {
|
||
this._games[game] = {
|
||
uid: '',
|
||
data: {}
|
||
}
|
||
}
|
||
return this._games[game]
|
||
}
|
||
|
||
/**
|
||
* 设置当前用户的绑定uid
|
||
* @param uid 要绑定的uid
|
||
* @param game
|
||
*/
|
||
autoRegUid(uid = '', game = 'gs') {
|
||
if (this.getUid(game)) {
|
||
return uid
|
||
}
|
||
this.addRegUid(uid, game)
|
||
return uid
|
||
}
|
||
|
||
/**
|
||
* 切换绑定CK生效的UID
|
||
* @param uid
|
||
* @param game
|
||
* @param save
|
||
* @returns
|
||
*/
|
||
setMainUid(uid = '', game = 'gs', save = true) {
|
||
this._map = false
|
||
game = this.gameKey(game)
|
||
if (Number(uid) < 100 || !uid) {
|
||
let uids = this.getUidList(game)
|
||
uid = (uids?.[uid] || uids?.[0])?.uid || ''
|
||
}
|
||
if (!uid) {
|
||
return false
|
||
}
|
||
if (this.hasUid(uid, game)) {
|
||
let gameDs = this.getGameDs(game)
|
||
gameDs.uid = uid
|
||
}
|
||
if (save) {
|
||
this.save()
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 添加MysUser
|
||
* @param mysUser
|
||
*/
|
||
async addMysUser(mysUser) {
|
||
this.mysUsers[mysUser.ltuid] = mysUser
|
||
this._map = false
|
||
MysUtil.eachGame(game => {
|
||
let uid = mysUser.getUid(game)
|
||
if (uid && this.getUid(game) == '') {
|
||
this.setMainUid(uid, game, false)
|
||
}
|
||
})
|
||
this.save()
|
||
}
|
||
|
||
/**
|
||
* 删除当前用户绑定CK
|
||
* @param ltuid
|
||
* @returns
|
||
*/
|
||
async delCk(ltuid = '') {
|
||
console.warn('delCk即将废弃')
|
||
return await this.delMysUser(ltuid)
|
||
}
|
||
|
||
/**
|
||
*
|
||
* @param mysUser
|
||
*/
|
||
async delMysUser(mysUser: any = '') {
|
||
let ltuid = mysUser.ltuid || mysUser
|
||
if (ltuid && this.mysUsers[ltuid]) {
|
||
let mys = this.mysUsers[ltuid]
|
||
this.mysUsers[ltuid] = false
|
||
this._map = false
|
||
await mys.del()
|
||
}
|
||
this._map = false
|
||
await this.save()
|
||
}
|
||
|
||
/**
|
||
*
|
||
* @param fn
|
||
*/
|
||
async eachMysUser(fn) {
|
||
await Data.forEach(this.mysUsers, async (mys, ltuid) => {
|
||
if (!mys) {
|
||
return true
|
||
}
|
||
return fn(mys, ltuid)
|
||
})
|
||
}
|
||
|
||
/**
|
||
*
|
||
* @param fn
|
||
* @returns
|
||
*/
|
||
async eachAllMysUser(fn) {
|
||
return MysUser.forEach(fn)
|
||
}
|
||
|
||
/**
|
||
* 检查当前用户绑定的CK状态
|
||
* @returns
|
||
*/
|
||
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 && checkRet) {
|
||
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
|
||
}
|
||
}
|