391 lines
11 KiB
JavaScript
391 lines
11 KiB
JavaScript
|
import base from './base.js'
|
|||
|
|
import MysInfo from './mys/mysInfo.js'
|
|||
|
|
import MysApi from './mys/mysApi.js'
|
|||
|
|
import lodash from 'lodash'
|
|||
|
|
import moment from 'moment'
|
|||
|
|
import fs from 'node:fs'
|
|||
|
|
import common from '../../../lib/common/common.js'
|
|||
|
|
|
|||
|
|
export default class Ledger extends base {
|
|||
|
|
constructor (e) {
|
|||
|
|
super(e)
|
|||
|
|
this.model = 'ledger'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async get () {
|
|||
|
|
this.getMonth()
|
|||
|
|
if (!this.month) return
|
|||
|
|
|
|||
|
|
let res = await MysInfo.get(this.e, 'ys_ledger', { month: this.month })
|
|||
|
|
if (!res || res.retcode !== 0) return false
|
|||
|
|
|
|||
|
|
this.e.nowData = res.data
|
|||
|
|
let data = this.dealData(res.data)
|
|||
|
|
|
|||
|
|
this.saveLedger(this.e.uid)
|
|||
|
|
|
|||
|
|
return data
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
getMonth () {
|
|||
|
|
let month = this.e.msg.replace(/#|原石|月|札记/g, '')
|
|||
|
|
let NowMonth = Number(moment().month()) + 1
|
|||
|
|
let monthData = ['一', '二', '三', '四', '五', '六', '七', '八', '九', '十', '十一', '十二']
|
|||
|
|
if (month) {
|
|||
|
|
if (isNaN(month)) {
|
|||
|
|
for (let i in monthData) {
|
|||
|
|
if (month == monthData[i]) {
|
|||
|
|
month = Number(i) + 1
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if (isNaN(month)) {
|
|||
|
|
month = NowMonth
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
month = NowMonth
|
|||
|
|
}
|
|||
|
|
if (month < 1 || month > 12) {
|
|||
|
|
month = NowMonth
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取前三个月
|
|||
|
|
let monthArr = [11, 12, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12].splice(NowMonth - 1, 3)
|
|||
|
|
if (!monthArr.includes(Number(month))) {
|
|||
|
|
this.e.reply('札记仅支持查询最近三个月的数据')
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
if ((NowMonth >= 3 && month > NowMonth) || (NowMonth < 3 && month > NowMonth && month <= 9 + month)) {
|
|||
|
|
month = NowMonth
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
this.NowMonth = NowMonth
|
|||
|
|
this.month = month
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
dealData (ledgerInfo) {
|
|||
|
|
let day
|
|||
|
|
if (this.month == this.NowMonth) {
|
|||
|
|
day = `${this.month}月${moment().date()}号`
|
|||
|
|
} else {
|
|||
|
|
day = `${this.month}月`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ledgerInfo.month_data.gacha = (ledgerInfo.month_data.current_primogems / 160).toFixed(0)
|
|||
|
|
ledgerInfo.month_data.last_gacha = (ledgerInfo.month_data.last_primogems / 160).toFixed(0)
|
|||
|
|
if (ledgerInfo.month_data.current_primogems > 10000) {
|
|||
|
|
ledgerInfo.month_data.current_primogems = (ledgerInfo.month_data.current_primogems / 10000).toFixed(2) + ' w'
|
|||
|
|
}
|
|||
|
|
if (ledgerInfo.month_data.last_primogems > 10000) {
|
|||
|
|
ledgerInfo.month_data.last_primogems = (ledgerInfo.month_data.last_primogems / 10000).toFixed(2) + ' w'
|
|||
|
|
}
|
|||
|
|
if (ledgerInfo.month_data.current_mora > 10000) {
|
|||
|
|
ledgerInfo.month_data.current_mora = (ledgerInfo.month_data.current_mora / 10000).toFixed(1) + ' w'
|
|||
|
|
}
|
|||
|
|
if (ledgerInfo.month_data.last_mora > 10000) {
|
|||
|
|
ledgerInfo.month_data.last_mora = (ledgerInfo.month_data.last_mora / 10000).toFixed(1) + ' w'
|
|||
|
|
}
|
|||
|
|
if (ledgerInfo.day_data.current_primogems > 10000) {
|
|||
|
|
ledgerInfo.day_data.current_primogems = (ledgerInfo.day_data.current_primogems / 10000).toFixed(1) + ' w'
|
|||
|
|
}
|
|||
|
|
if (ledgerInfo.day_data.current_mora > 10000) {
|
|||
|
|
ledgerInfo.day_data.current_mora = (ledgerInfo.day_data.current_mora / 10000).toFixed(1) + ' w'
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let color = ['#73a9c6', '#d56565', '#70b2b4', '#bd9a5a', '#739970', '#7a6da7', '#597ea0']
|
|||
|
|
for (let i in ledgerInfo.month_data.group_by) {
|
|||
|
|
ledgerInfo.month_data.group_by[i].color = color[ledgerInfo.month_data.group_by[i].action_id]
|
|||
|
|
}
|
|||
|
|
ledgerInfo.color = []
|
|||
|
|
ledgerInfo.month_data.group_by.forEach((item) => {
|
|||
|
|
ledgerInfo.color.push(['#73a9c6', '#d56565', '#70b2b4', '#bd9a5a', '#739970', '#7a6da7', '#597ea0'][item.action_id])
|
|||
|
|
})
|
|||
|
|
ledgerInfo.group_by = JSON.stringify(ledgerInfo.month_data.group_by)
|
|||
|
|
ledgerInfo.color = JSON.stringify(ledgerInfo.color)
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
saveId: this.e.uid,
|
|||
|
|
uid: this.e.uid,
|
|||
|
|
day,
|
|||
|
|
...ledgerInfo,
|
|||
|
|
...this.screenData
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 保存上两个原石数据
|
|||
|
|
async saveLedger (uid, ck = '') {
|
|||
|
|
if (ck) {
|
|||
|
|
uid = ck.uid
|
|||
|
|
} else {
|
|||
|
|
/** 获取个人ck */
|
|||
|
|
ck = await MysInfo.checkUidBing(uid)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (!ck || lodash.isEmpty(ck)) {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let dataPath = `./data/NoteData/${uid}.json`
|
|||
|
|
let NoteData = {}
|
|||
|
|
if (fs.existsSync(dataPath)) {
|
|||
|
|
NoteData = JSON.parse(fs.readFileSync(dataPath, 'utf8'))
|
|||
|
|
}
|
|||
|
|
// 当前年
|
|||
|
|
let NowYear = Number(moment().year())
|
|||
|
|
let NowMonth = Number(moment().month()) + 1
|
|||
|
|
|
|||
|
|
// 获取前三个月
|
|||
|
|
let monthArr = [11, 12, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12].splice(NowMonth - 1, 3)
|
|||
|
|
|
|||
|
|
for (let month of monthArr) {
|
|||
|
|
let year = NowYear
|
|||
|
|
// 上一年
|
|||
|
|
if (NowMonth <= 2 && month >= 11) {
|
|||
|
|
year--
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (!NoteData[year]) NoteData[year] = {}
|
|||
|
|
|
|||
|
|
if (NoteData[year][month] && NowMonth != month && NoteData[year][month].isUpdate) continue
|
|||
|
|
|
|||
|
|
let ledgerInfo
|
|||
|
|
if (NowMonth == month && this.e.nowData && this.e.nowData?.data?.data_month == NowMonth) {
|
|||
|
|
ledgerInfo = this.e.nowData
|
|||
|
|
} else {
|
|||
|
|
ledgerInfo = await this.ysLedger(ck, month)
|
|||
|
|
if (!ledgerInfo) continue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (NowMonth != month) {
|
|||
|
|
ledgerInfo.isUpdate = true
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
NoteData[year][month] = ledgerInfo
|
|||
|
|
|
|||
|
|
common.sleep(100)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
logger.mark(`[札记查询][自动保存] uid:${uid} 原石数据已保存`)
|
|||
|
|
|
|||
|
|
fs.writeFileSync(dataPath, JSON.stringify(NoteData, '', '\t'))
|
|||
|
|
return NoteData
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async ysLedger (ck, month) {
|
|||
|
|
let mysApi = new MysApi(ck.uid, ck.ck, { log: false })
|
|||
|
|
|
|||
|
|
let ledgerInfo = await mysApi.getData('ys_ledger', { month })
|
|||
|
|
|
|||
|
|
if (!ledgerInfo || ledgerInfo.retcode != 0) return false
|
|||
|
|
|
|||
|
|
return ledgerInfo.data
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async ledgerTask (manual) {
|
|||
|
|
let cks = await MysInfo.getBingCkUid()
|
|||
|
|
let uids = lodash.map(cks, 'uid')
|
|||
|
|
let finishTime = moment().add(uids.length * 0.7, 's').format('MM-DD HH:mm:ss')
|
|||
|
|
logger.mark(`札记ck:${uids.length}个，预计需要${this.countTime(uids.length)} ${finishTime} 完成`)
|
|||
|
|
|
|||
|
|
if (manual) {
|
|||
|
|
await this.e.reply('开始任务：保存原石数据，完成前请勿重复执行')
|
|||
|
|
await this.e.reply(`札记ck：${uids.length}个\n预计需要：${this.countTime(uids.length)}\n完成时间：${finishTime}`)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for (let uid of uids) {
|
|||
|
|
let ck = cks[uid]
|
|||
|
|
this.e.user_id = ck.qq
|
|||
|
|
|
|||
|
|
await this.saveLedger(uid, ck)
|
|||
|
|
await common.sleep(500)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (manual) {
|
|||
|
|
this.e.reply('原石任务完成')
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
countTime (num) {
|
|||
|
|
let time = num * 0.7
|
|||
|
|
let hour = Math.floor((time / 3600) % 24)
|
|||
|
|
let min = Math.floor((time / 60) % 60)
|
|||
|
|
let sec = Math.floor(time % 60)
|
|||
|
|
let msg = ''
|
|||
|
|
if (hour > 0) msg += `${hour}小时`
|
|||
|
|
if (min > 0) msg += `${min}分钟`
|
|||
|
|
if (sec > 0) msg += `${sec}秒`
|
|||
|
|
return msg
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async ledgerCount () {
|
|||
|
|
this.model = 'ledgerCount'
|
|||
|
|
|
|||
|
|
let mysInfo = await MysInfo.init(this.e, 'ys_ledger')
|
|||
|
|
let uid = mysInfo?.uid
|
|||
|
|
if (!uid) return false
|
|||
|
|
|
|||
|
|
/** 保存札记数据 */
|
|||
|
|
let NoteData = await this.saveLedger(uid)
|
|||
|
|
/** 处理数据 */
|
|||
|
|
return this.ledgerCountData(NoteData)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async ledgerCountHistory () {
|
|||
|
|
let nowYear
|
|||
|
|
if (this.e.msg.includes('去年')) {
|
|||
|
|
nowYear = moment().year() - 1
|
|||
|
|
} else if (this.e.msg.includes('今年')) {
|
|||
|
|
nowYear = moment().year()
|
|||
|
|
} else {
|
|||
|
|
// 获取年份
|
|||
|
|
nowYear = this.e.msg.match(/(\d{4})/)
|
|||
|
|
}
|
|||
|
|
if (nowYear) {
|
|||
|
|
nowYear = parseInt(nowYear)
|
|||
|
|
}
|
|||
|
|
if (!nowYear) {
|
|||
|
|
nowYear = moment().year()
|
|||
|
|
}
|
|||
|
|
this.model = 'ledgerCount'
|
|||
|
|
let mysInfo = await MysInfo.init(this.e, 'ys_ledger')
|
|||
|
|
let uid = mysInfo?.uid
|
|||
|
|
if (!uid) return false
|
|||
|
|
let dataPath = `./data/NoteData/${uid}.json`
|
|||
|
|
let NoteData = {}
|
|||
|
|
if (fs.existsSync(dataPath)) {
|
|||
|
|
NoteData = JSON.parse(fs.readFileSync(dataPath, 'utf8'))
|
|||
|
|
}
|
|||
|
|
// console.log(NoteData)
|
|||
|
|
if (!NoteData || lodash.isEmpty(NoteData)) {
|
|||
|
|
this.e.reply('暂无原石数据，请先发送 #原石', false, { at: true })
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
NoteData = NoteData[nowYear]
|
|||
|
|
if (!NoteData) {
|
|||
|
|
this.e.reply(`uid：${uid} ${nowYear}年无原石统计数据！`, false, { at: true })
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
lodash.forEach(NoteData, (val) => {
|
|||
|
|
val.year = nowYear
|
|||
|
|
})
|
|||
|
|
/** 处理数据 */
|
|||
|
|
return this.ledgerCountData(NoteData, String(nowYear))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ledgerCountData (NoteData, nowYear) {
|
|||
|
|
let hasMore = false
|
|||
|
|
let yearText
|
|||
|
|
if (!nowYear) {
|
|||
|
|
// 获取总长度
|
|||
|
|
if (NoteData && Object.keys(NoteData) && Object.keys(NoteData).length > 0) {
|
|||
|
|
let len = 0
|
|||
|
|
Object.keys(NoteData).forEach((year) => {
|
|||
|
|
let yearData = NoteData[year]
|
|||
|
|
len += Object.keys(yearData).length
|
|||
|
|
})
|
|||
|
|
hasMore = len >= 12
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 获取最近12个月的数据
|
|||
|
|
const newNoteData = []
|
|||
|
|
for (let i = 0; i < 12; i++) {
|
|||
|
|
let month = Number(moment().month()) + 1 - i
|
|||
|
|
let year = Number(moment().year())
|
|||
|
|
|
|||
|
|
if (month <= 0) {
|
|||
|
|
month = 12 + month
|
|||
|
|
year--
|
|||
|
|
}
|
|||
|
|
if (NoteData[year] && NoteData[year][month]) {
|
|||
|
|
NoteData[year][month].year = year
|
|||
|
|
newNoteData.push(NoteData[year][month])
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
NoteData = newNoteData
|
|||
|
|
} else {
|
|||
|
|
yearText = `${nowYear}年-`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if (!NoteData || lodash.isEmpty(NoteData)) return
|
|||
|
|
|
|||
|
|
let data = {
|
|||
|
|
allPrimogems: 0,
|
|||
|
|
allMora: 0,
|
|||
|
|
primogemsMonth: [],
|
|||
|
|
moraMonth: [],
|
|||
|
|
yearText
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
lodash.forEach(NoteData, (val) => {
|
|||
|
|
data.allPrimogems += val.month_data.current_primogems
|
|||
|
|
data.allMora += val.month_data.current_mora
|
|||
|
|
// 柱状图数据
|
|||
|
|
data.primogemsMonth.push({
|
|||
|
|
value: val.month_data.current_primogems,
|
|||
|
|
month: String(val.data_month),
|
|||
|
|
year: String(val.year),
|
|||
|
|
name: '原石'
|
|||
|
|
})
|
|||
|
|
data.moraMonth.push({
|
|||
|
|
value: (val.month_data.current_mora / 1000).toFixed(0),
|
|||
|
|
month: String(val.data_month),
|
|||
|
|
year: String(val.year),
|
|||
|
|
name: '摩拉'
|
|||
|
|
})
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
// 单位处理
|
|||
|
|
data.allMora = (data.allMora / 10000).toFixed(0) + 'w'
|
|||
|
|
data.allPrimogemsShow = (data.allPrimogems / 10000).toFixed(2) + 'w'
|
|||
|
|
data.allGacha = (data.allPrimogems / 160).toFixed(0)
|
|||
|
|
|
|||
|
|
// 原石最多
|
|||
|
|
data.maxPrimogems = lodash.maxBy(data.primogemsMonth, 'value')
|
|||
|
|
data.maxMora = lodash.maxBy(data.moraMonth, 'value')
|
|||
|
|
// 按年份月份排序
|
|||
|
|
data.primogemsMonth = lodash.sortBy(data.primogemsMonth, item => {
|
|||
|
|
return Number(item.year) * 100 + Number(item.month)
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
let groupBy = lodash(NoteData).map('month_data').map('group_by').flatMap().value()
|
|||
|
|
|
|||
|
|
let pieData = {}
|
|||
|
|
for (let val of groupBy) {
|
|||
|
|
if (!pieData[val.action]) {
|
|||
|
|
pieData[val.action] = {
|
|||
|
|
num: val.num,
|
|||
|
|
action: val.action,
|
|||
|
|
action_id: val.action_id
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
pieData[val.action].num += val.num
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
pieData = lodash.flatMap(pieData, (item) => {
|
|||
|
|
return item
|
|||
|
|
})
|
|||
|
|
pieData = lodash.orderBy(pieData, ['num'], ['desc'])
|
|||
|
|
|
|||
|
|
data.color = []
|
|||
|
|
pieData.forEach((item) => {
|
|||
|
|
data.color.push(['#73a9c6', '#d56565', '#70b2b4', '#bd9a5a', '#739970', '#7a6da7', '#597ea0'][item.action_id])
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
data.group_by = pieData
|
|||
|
|
|
|||
|
|
data.color = JSON.stringify(data.color)
|
|||
|
|
data.pieData = JSON.stringify(pieData)
|
|||
|
|
data.primogemsMonth = JSON.stringify(data.primogemsMonth)
|
|||
|
|
data.hasMore = hasMore
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
saveId: this.e.uid,
|
|||
|
|
uid: this.e.uid,
|
|||
|
|
...data,
|
|||
|
|
...this.screenData
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|