386 lines
9.7 KiB
JavaScript
386 lines
9.7 KiB
JavaScript
import base from './base.js'
|
||
import cfg from '../../../lib/config/config.js'
|
||
import common from '../../../lib/common/common.js'
|
||
import fs from 'node:fs'
|
||
import moment from 'moment'
|
||
import GachaLog from './gachaLog.js'
|
||
import lodash from 'lodash'
|
||
|
||
let xlsx = {}
|
||
|
||
export default class ExportLog extends base {
|
||
constructor (e) {
|
||
super(e)
|
||
this.model = 'gachaLog'
|
||
|
||
this.urlKey = `${this.prefix}url:`
|
||
/** 绑定的uid */
|
||
this.uidKey = `Yz:genshin:mys:qq-uid:${this.userId}`
|
||
|
||
this.path = `./data/gachaJson/${this.e.user_id}/`
|
||
|
||
this.pool = [
|
||
{ type: 301, typeName: '角色活动' },
|
||
{ type: 302, typeName: '武器活动' },
|
||
{ type: 200, typeName: '常驻' }
|
||
]
|
||
|
||
this.typeName = {
|
||
301: '角色',
|
||
302: '武器',
|
||
200: '常驻'
|
||
}
|
||
}
|
||
|
||
async initXlsx () {
|
||
if (!lodash.isEmpty(xlsx)) return xlsx
|
||
|
||
xlsx = await import('node-xlsx')
|
||
}
|
||
|
||
async exportJson () {
|
||
await this.getUid()
|
||
|
||
if (!this.uid) return false
|
||
|
||
let list = this.getAllList().list
|
||
|
||
let data = {
|
||
info: {
|
||
uid: this.uid,
|
||
lang: list[0].lang,
|
||
export_time: moment().format('YYYY-MM-DD HH:mm:ss'),
|
||
export_timestamp: moment().format('X'),
|
||
export_app: 'Yunzai-Bot',
|
||
export_app_version: cfg.package.version,
|
||
uigf_version: 'v2.2'
|
||
},
|
||
list
|
||
}
|
||
|
||
let saveFile = `${this.path}${this.uid}/${this.uid}.json`
|
||
|
||
fs.writeFileSync(saveFile, JSON.stringify(data, '', '\t'))
|
||
|
||
logger.mark(`${this.e.logFnc} 导出成功${this.uid}.json`)
|
||
|
||
this.e.reply(`导出成功：${this.uid}.json，共${list.length}条 \n请接收文件`)
|
||
|
||
await this.e.friend.sendFile(saveFile).catch((err) => {
|
||
logger.error(`${this.e.logFnc} 发送文件失败 ${JSON.stringify(err)}`)
|
||
})
|
||
|
||
/** 删除文件 */
|
||
fs.unlink(saveFile, () => {})
|
||
}
|
||
|
||
async exportXlsx () {
|
||
await this.getUid()
|
||
|
||
if (!this.uid) return false
|
||
|
||
await this.initXlsx()
|
||
|
||
logger.mark(`${this.e.logFnc} 开始导出${this.uid}.xlsx`)
|
||
|
||
let res = this.getAllList()
|
||
|
||
/** 处理卡池数据 */
|
||
let xlsxData = this.xlsxDataPool(res)
|
||
/** 处理所有数据 */
|
||
xlsxData.push(this.xlsxDataAll(res))
|
||
|
||
/** node-xlsx导出的buffer有点大.. */
|
||
let buffer = xlsx.build(xlsxData)
|
||
let saveFile = `${this.path}${this.uid}/${this.uid}.xlsx`
|
||
|
||
fs.writeFileSync(saveFile, Buffer.from(buffer))
|
||
|
||
logger.mark(`${this.e.logFnc} 导出成功${this.uid}.xlsx`)
|
||
|
||
this.e.reply(`记录文件${this.uid}.xlsx上传中，请耐心等待...`)
|
||
|
||
res = await this.e.friend.sendFile(saveFile).catch((err) => {
|
||
this.e.reply(`发送文件${this.uid}.xlsx失败，请稍后再试`)
|
||
logger.error(`${this.e.logFnc} 发送文件失败 ${JSON.stringify(err)}`)
|
||
})
|
||
|
||
let line = xlsxData[xlsxData.length - 1].data.length - 1
|
||
if (res) this.e.reply(`${this.uid}.xlsx上传成功，共${line}条\n请接收文件`)
|
||
|
||
/** 删除文件 */
|
||
fs.unlink(saveFile, () => {})
|
||
}
|
||
|
||
async getUid () {
|
||
let gachLog = new GachaLog(this.e)
|
||
let uid = await gachLog.getUid()
|
||
|
||
if (!uid) return false
|
||
|
||
this.uid = uid
|
||
return this.uid
|
||
}
|
||
|
||
getAllList () {
|
||
let res = {
|
||
list: []
|
||
}
|
||
let tmpId = {}
|
||
for (let v of this.pool) {
|
||
let json = `${this.path}${this.uid}/${v.type}.json`
|
||
if (fs.existsSync(json)) {
|
||
json = JSON.parse(fs.readFileSync(json, 'utf8'))
|
||
json = json.reverse()
|
||
} else {
|
||
json = []
|
||
}
|
||
res[v.type] = json
|
||
for (let v of json) {
|
||
if (v.gacha_type == 301 || v.gacha_type == 400) {
|
||
v.uigf_gacha_type = '301'
|
||
} else {
|
||
v.uigf_gacha_type = v.gacha_type
|
||
}
|
||
let id = v.id
|
||
if (!id) {
|
||
id = moment(v.time).format('x') + '000000'
|
||
v.id = id
|
||
} else {
|
||
if (id.length == 13) {
|
||
v.id = `${id}000000`
|
||
}
|
||
}
|
||
|
||
if (tmpId[id]) {
|
||
let newId = `${id}00000${tmpId[id].length}`
|
||
tmpId[id].push(newId)
|
||
v.id = newId
|
||
} else {
|
||
tmpId[id] = [id]
|
||
}
|
||
res.list.push(v)
|
||
}
|
||
}
|
||
res.list = lodash.orderBy(res.list, ['id', 'asc'])
|
||
return res
|
||
}
|
||
|
||
loadJson (json) {
|
||
if (!fs.existsSync(json)) return []
|
||
return JSON.parse(fs.readFileSync(json, 'utf8'))
|
||
}
|
||
|
||
xlsxDataPool (data) {
|
||
let xlsxData = []
|
||
|
||
for (let v of this.pool) {
|
||
let poolData = [
|
||
[
|
||
'时间', '名称', '物品类型', '星级', '祈愿类型'
|
||
]
|
||
]
|
||
for (let log of data[v.type]) {
|
||
poolData.push([
|
||
log.time, log.name, log.item_type, log.rank_type, log.gacha_type
|
||
])
|
||
}
|
||
|
||
let sheetOptions = {
|
||
'!cols': [{ wch: 20 }, { wch: 20 }, { wch: 10 }, { wch: 10 }, { wch: 10 }]
|
||
}
|
||
xlsxData.push({ name: `${v.typeName}祈愿`, data: poolData, options: sheetOptions })
|
||
}
|
||
|
||
return xlsxData
|
||
}
|
||
|
||
xlsxDataAll (data) {
|
||
let list = [
|
||
[
|
||
'count', 'gacha_type', 'id', 'item_id', 'item_type', 'lang', 'name', 'rank_type', 'time', 'uid', 'uigf_gacha_type'
|
||
]
|
||
]
|
||
for (let v of data.list) {
|
||
let tmp = []
|
||
for (let i of list[0]) {
|
||
if (i == 'id' || i == 'uigf_gacha_type') v[i] = String(v[i])
|
||
tmp.push(v[i])
|
||
}
|
||
list.push(tmp)
|
||
}
|
||
let sheetOptions = {
|
||
'!cols': [{ wch: 10 }, { wch: 10 }, { wch: 20 }, { wch: 10 }, { wch: 10 }, { wch: 10 }, { wch: 20 }, { wch: 10 }, { wch: 20 }, { wch: 10 }, { wch: 10 }]
|
||
}
|
||
|
||
return { name: '原始数据', data: list, options: sheetOptions }
|
||
}
|
||
|
||
/** xlsx导入抽卡记录 */
|
||
async logXlsx () {
|
||
await this.initXlsx()
|
||
|
||
let uid = /[1-9][0-9]{8}/g.exec(this.e.file.name)[0]
|
||
let textPath = `${this.path}${this.e.file.name}`
|
||
/** 获取文件下载链接 */
|
||
let fileUrl = await this.e.friend.getFileUrl(this.e.file.fid)
|
||
|
||
let ret = await common.downFile(fileUrl, textPath)
|
||
if (!ret) {
|
||
this.e.reply('下载xlsx文件错误')
|
||
return false
|
||
}
|
||
|
||
let list = xlsx.parse(textPath)
|
||
list = lodash.keyBy(list, 'name')
|
||
|
||
if (!list['原始数据']) {
|
||
this.e.reply('xlsx文件内容错误：非统一祈愿记录标准')
|
||
return false
|
||
}
|
||
|
||
/** 处理xlsx数据 */
|
||
let data = this.dealXlsx(list['原始数据'].data)
|
||
if (!data) return false
|
||
|
||
/** 保存json */
|
||
let msg = []
|
||
for (let type in data) {
|
||
if (!this.typeName[type]) continue
|
||
let gachLog = new GachaLog(this.e)
|
||
gachLog.uid = uid
|
||
gachLog.type = type
|
||
gachLog.writeJson(data[type])
|
||
|
||
msg.push(`${this.typeName[type]}记录：${data[type].length}条`)
|
||
}
|
||
|
||
/** 删除文件 */
|
||
fs.unlink(textPath, () => {})
|
||
|
||
await this.e.reply(`${this.e.file.name}，导入成功\n${msg.join('\n')}`)
|
||
}
|
||
|
||
dealXlsx (list) {
|
||
/** 必要字段 */
|
||
let reqField = ['uigf_gacha_type', 'gacha_type', 'item_type', 'name', 'time']
|
||
/** 不是必要字段 */
|
||
let noReqField = ['id', 'uid', 'count', 'item_id', 'lang', 'rank_type']
|
||
|
||
let field = {}
|
||
for (let i in list[0]) {
|
||
field[list[0][i]] = i
|
||
}
|
||
|
||
/** 判断字段 */
|
||
for (let v of reqField) {
|
||
if (!field[v]) {
|
||
this.e.reply(`xlsx文件内容错误：缺少必要字段${v}`)
|
||
return
|
||
}
|
||
}
|
||
|
||
/** 倒序 */
|
||
if (moment(list[1][field.time]).format('x') < moment(list[list.length - 1][field.time]).format('x')) {
|
||
list = list.reverse()
|
||
}
|
||
|
||
let data = {}
|
||
for (let v of list) {
|
||
if (v[field.name] == 'name') continue
|
||
if (!data[v[field.uigf_gacha_type]]) data[v[field.uigf_gacha_type]] = []
|
||
|
||
let tmp = {}
|
||
/** 加入必要字段 */
|
||
for (let re of reqField) {
|
||
tmp[re] = v[field[re]]
|
||
}
|
||
|
||
/** 加入非必要字段 */
|
||
for (let noRe of noReqField) {
|
||
if (field[noRe]) {
|
||
tmp[noRe] = v[field[noRe]]
|
||
} else {
|
||
tmp[noRe] = ''
|
||
}
|
||
}
|
||
|
||
data[v[field.uigf_gacha_type]].push(tmp)
|
||
}
|
||
|
||
return data
|
||
}
|
||
|
||
/** json导入抽卡记录 */
|
||
async logJson () {
|
||
let uid = /[1-9][0-9]{8}/g.exec(this.e.file.name)[0]
|
||
let textPath = `${this.path}${this.e.file.name}`
|
||
/** 获取文件下载链接 */
|
||
let fileUrl = await this.e.friend.getFileUrl(this.e.file.fid)
|
||
|
||
let ret = await common.downFile(fileUrl, textPath)
|
||
if (!ret) {
|
||
this.e.reply('下载json文件错误')
|
||
return false
|
||
}
|
||
let json = {}
|
||
try {
|
||
json = JSON.parse(fs.readFileSync(textPath, 'utf8'))
|
||
} catch (error) {
|
||
this.e.reply(`${this.e.file.name},json格式错误`)
|
||
return false
|
||
}
|
||
|
||
if (lodash.isEmpty(json) || !json.list) {
|
||
this.e.reply('json文件内容错误：非统一祈愿记录标准')
|
||
return false
|
||
}
|
||
|
||
let data = this.dealJson(json.list)
|
||
if (!data) return false
|
||
|
||
/** 保存json */
|
||
let msg = []
|
||
for (let type in data) {
|
||
if (!this.typeName[type]) continue
|
||
let gachLog = new GachaLog(this.e)
|
||
gachLog.uid = uid
|
||
gachLog.type = type
|
||
gachLog.writeJson(data[type])
|
||
|
||
msg.push(`${this.typeName[type]}记录：${data[type].length}条`)
|
||
}
|
||
|
||
/** 删除文件 */
|
||
fs.unlink(textPath, () => {})
|
||
|
||
await this.e.reply(`${this.e.file.name}，导入成功\n${msg.join('\n')}`)
|
||
}
|
||
|
||
dealJson (list) {
|
||
let data = {}
|
||
|
||
/** 必要字段 */
|
||
let reqField = ['gacha_type', 'item_type', 'name', 'time']
|
||
|
||
for (let v of reqField) {
|
||
if (!list[0][v]) {
|
||
this.e.reply(`json文件内容错误：缺少必要字段${v}`)
|
||
return false
|
||
}
|
||
}
|
||
|
||
/** 倒序 */
|
||
if (moment(list[0].time).format('x') < moment(list[list.length - 1].time).format('x')) {
|
||
list = list.reverse()
|
||
}
|
||
|
||
for (let v of list) {
|
||
if (!data[v.uigf_gacha_type]) data[v.uigf_gacha_type] = []
|
||
data[v.uigf_gacha_type].push(v)
|
||
}
|
||
|
||
return data
|
||
}
|
||
}
|