import lodash from 'lodash' import moment from 'moment' import { Common, Data } from '../components/index.js' export default class ProfileRank { constructor (data) { this.groupId = data.groupId || data.groupId || '' if (!this.groupId || this.groupId === 'undefined') { return false } this.qq = data.qq this.uid = data.uid + '' this.allowRank = false } static async create (data) { let rank = new ProfileRank(data) rank.allowRank = await ProfileRank.checkRankLimit(rank.uid) return rank } key (profile, type) { return `miao:rank:${this.groupId}:${type}:${profile.id}` } /** * 获取排行信息 * @param profile * @param force * @returns {Promise<{}|boolean>} */ async getRank (profile, force = false) { if (!profile || !this.groupId || !this.allowRank || !profile.hasData) { return false } let ret = {} for (let typeKey of ['mark', 'dmg']) { let typeRank = await this.getTypeRank(profile, typeKey, force) ret[typeKey] = typeRank if (!ret.rank || ret.rank >= typeRank.rank) { ret.rank = typeRank.rank ret.rankType = typeKey } } return ret } async getTypeRank (profile, type, force) { if (!profile || !profile.hasData || !type) { return false } if (type === 'dmg' && !profile.hasDmg) { return false } const typeKey = this.key(profile, type) let value let rank if (force) { value = await this.getTypeValue(profile, type) } else { rank = await redis.zRevRank(typeKey, this.uid) if (!lodash.isNumber(rank)) { value = await this.getTypeValue(profile, type) } } if (value && !lodash.isUndefined(value.score)) { await redis.zAdd(typeKey, { score: value.score, value: this.uid }) } if (!lodash.isNumber(rank)) { rank = await redis.zRevRank(typeKey, this.uid) } if (rank === null) { rank = 99 } if (force) { return { rank: rank + 1, value: value.score, data: value.data } } return { rank: rank + 1 } } async getTypeValue (profile, type) { if (!profile || !profile.hasData) { return false } if (type === 'mark') { if (!profile?.artis?.hasArtis) { return false } let mark = profile.getArtisMark(false) if (mark && mark._mark) { return { score: mark._mark * 1, data: mark } } } if (type === 'dmg' && profile.hasDmg) { let dmg = await profile.calcDmg({ mode: 'single' }) if (dmg && dmg.avg) { return { score: dmg.avg, data: dmg } } } return false } /** * 获取群排行UID * @param groupId * @param charId * @param type * @returns {Promise} */ static async getGroupMaxUid (groupId, charId, type = 'mark') { let uids = await redis.zRange(`miao:rank:${groupId}:${type}:${charId}`, -1, -1) return uids ? uids[0] : false } static async getGroupMaxUidList (groupId, type = 'mark') { let keys = await redis.keys(`miao:rank:${groupId}:${type}:*`) let ret = [] for (let key of keys) { let keyRet = /^miao:rank:\d+:(?:mark|dmg):(\d{8})$/.exec(key) if (keyRet && keyRet[1]) { let charId = keyRet[1] let uid = await ProfileRank.getGroupMaxUid(groupId, charId, type) if (uid) { ret.push({ uid, charId }) } } } return ret } /** * 获取排行榜 * @param groupId * @param charId * @param type * @returns {Promise[]|boolean>} */ static async getGroupUidList (groupId, charId, type = 'mark') { let uids = await redis.zRangeWithScores(`miao:rank:${groupId}:${type}:${charId}`, -20, -1) return uids ? uids.reverse() : false } /** * 重置群排行 * @param groupId * @param charId * @returns {Promise} */ static async resetRank (groupId, groupMemList, charId = '') { let keys = await redis.keys(`miao:rank:${groupId}:*`) for (let key of keys) { let charRet = /^miao:rank:\d+:(?:mark|dmg):(\d{8})$/.exec(key) if (charRet) { if (charId === '' || charId * 1 === charRet[1] * 1) { await redis.del(key) } } } if (charId === '') { await redis.del(`miao:rank:${groupId}:cfg`) } } static async getGroupCfg (groupId) { const rankLimitTxt = { 1: '无限制', 2: '绑定有CK的用户', 3: '绑定CK,或列表有16个角色数据', 4: '绑定CK,或列表有安柏&凯亚&丽莎的数据', 5: '绑定CK,或列表有16个角色数据且包含安柏&凯亚&丽莎' } let rankLimit = Common.cfg('groupRankLimit') * 1 || 1 let ret = await Data.redisGet(`miao:rank:${groupId}:cfg`, { timestamp: (new Date()) * 1, status: 0 }) await Data.redisSet(`miao:rank:${groupId}:cfg`, ret, 3600 * 24 * 365) ret.limitTxt = rankLimitTxt[rankLimit] ret.time = moment(new Date(ret.timestamp)).format('MM-DD HH:mm') return ret } /** * 设置群开关状态 * @param groupId * @param status:0开启,1关闭 * @returns {Promise} */ static async setGroupStatus (groupId, status = 0) { let cfg = await Data.redisGet(`miao:rank:${groupId}:cfg`, { timestamp: (new Date()) * 1, status }) cfg.status = status await Data.redisSet(`miao:rank:${groupId}:cfg`, cfg, 3600 * 24 * 365) } static async setUidInfo ({ uid, qq, profiles, uidType = 'bind' }) { if (!uid) { return false } let basicCount = 0 let totalCount = 0 for (let charId in profiles) { let profile = profiles[charId] if (!profile || !profile.hasData) { continue } if (['安柏', '凯亚', '丽莎'].includes(profile.name)) { basicCount++ } totalCount++ } let data = {} try { let uData = await redis.get(`miao:rank:uid-info:${uid}`) if (uData) { data = JSON.parse(uData) } } catch (e) { data = {} } data.totalCount = totalCount data.basicCount = basicCount if (data.isSelfUid) { delete data.isSelfUid data.uidType = 'ck' } if (uidType === 'ck') { data.uidType = 'ck' data.qq = qq || data.qq || '' } else { data.uidType = data.uidType || 'bind' if (data.uidType === 'bind') { data.qq = data.qq || qq || '' } else { data.qq = qq || data.qq || '' } } await redis.set(`miao:rank:uid-info:${uid}`, JSON.stringify(data), { EX: 3600 * 24 * 365 }) } static async getUidInfo (uid) { try { let data = await redis.get(`miao:rank:uid-info:${uid}`) return JSON.parse(data) } catch (e) { } return false } /** * 1: '无限制', * 2: '绑定有CK的用户', * 3: '面板列表有16个角色数据,或绑定CK', * 4: '面板列表有安柏&凯亚&丽莎的数据,或绑定CK', * 5: '面板列表有16个角色数据且包含安柏&凯亚&丽莎,或绑定CK' * @param uid * @returns {Promise} */ static async checkRankLimit (uid) { if (!uid) { return false } try { let rankLimit = Common.cfg('groupRankLimit') * 1 || 1 if (rankLimit === 1) { return true } let data = await redis.get(`miao:rank:uid-info:${uid}`) data = JSON.parse(data) if (data.isSelfUid || data.uidType === 'ck') { return true } if (rankLimit === 2) { return false } if ((data.totalCount || 0) < 16 && [3, 5].includes(rankLimit)) { return false } if ((data.basicCount || 0) < 3 && [4, 5].includes(rankLimit)) { return false } return true } catch (e) { return false } } }