Miao-Yunzai/model/ledger.ts

import base from './base.js'
import { MysInfo, MysApi } from 'yunzai/mys'
import lodash from 'lodash'
import moment from 'moment'
import fs from 'node:fs'
import { sleep } from 'yunzai/utils'
export default class Ledger extends base {
  constructor(e) {
    super(e)
    this.e = e
    this.model = 'ledger'
    if (this.e.msg?.includes('星琼')) this.e.isSr = true

    this.color = [
      '#73a9c6',
      '#d56565',
      '#70b2b4',
      '#bd9a5a',
      '#739970',
      '#7a6da7',
      '#597ea0'
    ]
    this.action = {
      other: 0,
      adventure_reward: 1,
      space_reward: 2,
      daily_reward: 3,
      abyss_reward: 4,
      mail_reward: 5,
      event_reward: 6
    }
  }

  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
    }
    if (this.e.isSr) {
      this.NowMonth =
        moment().year().toString() +
        (NowMonth < 10 ? '0' : '') +
        NowMonth.toString()
      this.month =
        moment().year().toString() + (month < 10 ? '0' : '') + month.toString()
      this.srmonth = month
    } else {
      this.NowMonth = NowMonth
      this.month = month
    }
  }

  dealData(ledgerInfo) {
    let day
    if (this.month == this.NowMonth) {
      day = `${this[this.e.isSr ? 'srmonth' : 'month']}月${moment().date()}号`
    } else {
      day = `${this[this.e.isSr ? 'srmonth' : 'month']}月`
    }

    let gacha = this.e.isSr ? 'current_hcoin' : 'current_primogems'
    let last_gacha = this.e.isSr ? 'last_hcoin' : 'last_primogems'
    let mora = this.e.isSr ? 'current_rails_pass' : 'current_mora'
    let last_mora = this.e.isSr ? 'last_rails_pass' : 'last_mora'

    ledgerInfo.month_data.gacha = (ledgerInfo.month_data[gacha] / 160).toFixed(
      0
    )
    ledgerInfo.month_data.last_gacha = (
      ledgerInfo.month_data[last_gacha] / 160
    ).toFixed(0)
    if (ledgerInfo.month_data[gacha] > 10000) {
      ledgerInfo.month_data[gacha] =
        (ledgerInfo.month_data[gacha] / 10000).toFixed(2) + ' w'
    }
    if (ledgerInfo.month_data[last_gacha] > 10000) {
      ledgerInfo.month_data[last_gacha] =
        (ledgerInfo.month_data[last_gacha] / 10000).toFixed(2) + ' w'
    }
    if (ledgerInfo.month_data[mora] > 10000) {
      ledgerInfo.month_data[mora] =
        (ledgerInfo.month_data[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[gacha] > 10000) {
      ledgerInfo.day_data[gacha] =
        (ledgerInfo.day_data[gacha] / 10000).toFixed(1) + ' w'
    }
    if (ledgerInfo.day_data[mora] > 10000) {
      ledgerInfo.day_data[mora] =
        (ledgerInfo.day_data[mora] / 10000).toFixed(1) + ' w'
    }

    ledgerInfo.color = []
    ledgerInfo.month_data.group_by.forEach(item => {
      if (this.e.isSr) {
        item.color = this.color[this.action[item.action]]
        item.action_name = item.action_name.slice(0, 4)
      } else {
        item.color = this.color[item.action_id]
      }
      ledgerInfo.color.push(item.color)
    })

    ledgerInfo.group_by = JSON.stringify(ledgerInfo.month_data.group_by)
    ledgerInfo.color = JSON.stringify(ledgerInfo.color)

    let icon = ''
    if (this.e.isSr)
      icon = lodash.sample(
        fs
          .readdirSync(
            `${this._path}/plugins/genshin/resources/StarRail/img/role`
          )
          .filter(file => file.endsWith('.webp'))
      )

    let week = [
      '星期日',
      '星期一',
      '星期二',
      '星期三',
      '星期四',
      '星期五',
      '星期六'
    ]

    return {
      saveId: this.e.uid,
      uid: this.e.uid,
      day,
      icon,
      srday: `${week[moment().day()]}`,
      nowDay: moment(new Date()).format('YYYY年MM月DD日'),
      ...ledgerInfo,
      ...this.screenData
    }
  }

  // 保存上两个原石数据
  async saveLedger(uid, ck = '', isTask = false) {
    if (ck) {
      uid = ck.uid
    } else {
      /** 获取个人ck */
      ck = await MysInfo.checkUidBing(uid, this.e.isSr ? 'sr' : 'gs')
    }

    if (!ck || lodash.isEmpty(ck)) {
      return false
    }

    let dataPath = `./data/${this.e?.isSr ? 'SR_NoteData' : '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 {
        let months = month
        if (this.e.isSr)
          months = String(year) + (month < 10 ? '0' : '') + String(month)

        ledgerInfo = await this.ysLedger(ck, months, isTask)
        if (!ledgerInfo) continue
      }

      if (NowMonth != month) {
        ledgerInfo.isUpdate = true
      }

      NoteData[year][month] = ledgerInfo

      sleep(100)
    }

    logger.mark(`[札记查询][自动保存] uid:${uid} 数据已保存`)

    fs.writeFileSync(dataPath, JSON.stringify(NoteData, '', '\t'))
    return NoteData
  }

  async ysLedger(ck, month, isTask) {
    let ledgerInfo = {}
    if (isTask) {
      let mysApi = new MysApi(ck.uid, ck.ck, { log: false }, this.e?.isSr)
      ledgerInfo = await mysApi.getData('ys_ledger', { month })
      ledgerInfo = await new MysInfo(this.e).checkCode(
        ledgerInfo,
        'ys_ledger',
        mysApi,
        { month },
        isTask
      )
    } else {
      ledgerInfo = await MysInfo.get(this.e, 'ys_ledger', { month })
    }

    if (!ledgerInfo || ledgerInfo.retcode != 0) return false

    return ledgerInfo.data
  }

  async ledgerTask(manual) {
    let cks = (await GsCfg.getBingCk(this.e?.isSr ? 'sr' : 'gs')).ck
    let uids = lodash.map(cks, 'uid')
    let finishTime = moment()
      .add(uids.length * 0.7, 's')
      .format('MM-DD HH:mm:ss')
    logger.mark(
      `${this.e?.isSr ? '开拓月历' : '札记'}ck:${uids.length}个，预计需要${this.countTime(uids.length)} ${finishTime} 完成`
    )

    if (manual) {
      await this.e.reply(
        `开始任务：保存${this.e?.isSr ? '星琼' : '原石'}数据，完成前请勿重复执行`
      )
      await this.e.reply(
        `${this.e?.isSr ? '开拓月历' : '札记'}ck：${uids.length}个\n预计需要：${this.countTime(uids.length)}\n完成时间：${finishTime}`
      )
    }

    for (const uid of uids) {
      const ck = cks[uid]

      await this.saveLedger(uid, ck, true)
      await sleep(500)
    }

    if (manual) {
      this.e.reply(`${this.e?.isSr ? '星琼' : '原石'}任务完成`)
    }
  }

  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/${this.e?.isSr ? 'SR_NoteData' : '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(
        this.e?.isSr
          ? [
              '暂无星琼数据，请先发送 *星琼',
              segment.button([{ text: '星琼', input: '*星琼' }])
            ]
          : [
              '暂无原石数据，请先发送 #原石',
              segment.button([{ text: '原石', input: '#原石' }])
            ],
        false,
        { at: true }
      )
      return false
    }
    NoteData = NoteData[nowYear]
    if (!NoteData) {
      this.e.reply(
        `uid：${uid} ${nowYear}年无${this.e?.isSr ? '星琼' : '原石'}统计数据！`,
        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
    }

    let Primogems = this.e.isSr ? 'current_hcoin' : 'current_primogems'
    let Mora = this.e.isSr ? 'current_rails_pass' : 'current_mora'
    lodash.forEach(NoteData, val => {
      data.allPrimogems += val.month_data[Primogems]
      data.allMora += val.month_data[Mora]
      // 柱状图数据
      if (this.e.isSr)
        val.data_month =
          val.data_month.slice(-2, -1) == '0'
            ? val.data_month.slice(-1)
            : val.data_month.slice(-2)

      data.primogemsMonth.push({
        value: val.month_data[Primogems],
        month: String(val.data_month),
        year: String(val.year),
        name: this.e.isSr ? '星琼' : '原石'
      })
      data.moraMonth.push({
        value: (val.month_data[Mora] / 1000).toFixed(0),
        month: String(val.data_month),
        year: String(val.year),
        name: this.e.isSr ? '专&通票' : '摩拉'
      })
    })

    // 单位处理
    data.allPrimogemsShow = (data.allPrimogems / 10000).toFixed(2) + 'w'
    data.allGacha = (data.allPrimogems / 160).toFixed(0)
    if (this.e.isSr) {
      data.allGacha = (data.allPrimogems / 160 + data.allMora).toFixed(0)
      data.allMora = data.allMora.toFixed(0) + '张'
    } else {
      data.allMora = (data.allMora / 10000).toFixed(0) + 'w'
    }

    // 原石最多
    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()

    if (this.e.isSr)
      groupBy.forEach(item => {
        item.action_id = this.action[item.action]
        item.action = item.action_name.slice(0, 4)
      })

    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(this.color[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
    }
  }
}