Miao-Yunzai/apps/gcLog.ts

import { plugin } from "yunzai/core"
import fs from "node:fs"
import GachaLog from "../model/gachaLog.js"
import ExportLog from "../model/exportLog.js"
import LogCount from "../model/logCount.js"
const _path = process.cwd() + "/plugins/genshin"
export class gcLog extends plugin {
  constructor() {
    /**
     * 
      name: "抽卡记录",
      dsc: "抽卡记录数据统计",
     */
    super({
      priority: 300,
      rule: [
        {
          reg: "(.*)authkey=(.*)",
          fnc: "logUrl"
        },
        {
          reg: "^#json文件导入记录$",
          fnc: "logJson"
        },
        {
          reg: "^#?(原神|星铁)?(全部)?(抽卡|抽奖|角色|武器|集录|常驻|up|新手|光锥|全部)池*(记录|祈愿|分析)$",
          fnc: "getLog"
        },
        {
          reg: "^#?(原神|星铁)?(强制)?导出记录(json)?$",
          fnc: "exportLog"
        },
        {
          reg: "^#?(记录帮助|抽卡帮助)$",
          fnc: "help"
        },
        {
          reg: "^#?(安卓|苹果|电脑|pc|ios)帮助$",
          fnc: "helpPort"
        },
        {
          reg: "^#?(原神|星铁)?(抽卡|抽奖|角色|武器|集录|常驻|up|新手|光锥)池*统计$",
          fnc: "logCount"
        }
      ]
    })

    this.androidUrl = "https://docs.qq.com/doc/DUWpYaXlvSklmVXlX"
    Object.defineProperty(this, "button", {
      get() {
        this.prefix = this.e?.isSr ? "*" : "#"
        return segment.button([
          { text: "角色记录", callback: `${this.prefix}角色记录` },
          { text: "角色统计", callback: `${this.prefix}角色统计` },
        ], [
          { text: "武器记录", callback: `${this.prefix}武器记录` },
          { text: "武器统计", callback: `${this.prefix}武器统计` },
        ], [
          { text: "集录记录", callback: `${this.prefix}集录记录` },
          { text: "集录统计", callback: `${this.prefix}集录统计` },
        ], [
          { text: "常驻记录", callback: `${this.prefix}常驻记录` },
          { text: "常驻统计", callback: `${this.prefix}常驻统计` },
        ])
      }
    })
  }

  async init() {
    let file = ["./data/gachaJson", "./data/srJson", "./temp/html/StarRail", "./temp/uigf"]
    for (let i of file) {
      if (!fs.existsSync(i)) {
        fs.mkdirSync(i)
      }
    }
  }

  accept() {
    if (this.e.file) {
      let name = this.e.file?.name
      if (/(.*)([1-9]|18)[0-9]{8}(.*).json/ig.test(name)) {
        this.e.msg = "#json文件导入记录"
        return true
      }
    }
    if (this.e.msg && /^#?(角色|武器)统计$/g.test(this.e.msg)) {
      this.e.msg = this.e.msg.replace("统计", "池统计")
      return true
    }
  }

  /** 抽卡记录链接 */
  async logUrl() {
    let data = await new GachaLog(this.e).logUrl()
    if (!data) return

    await this.renderImg("genshin", `html/gacha/gacha-log`, data)

    if (this.e.isGroup)
      this.e.reply("已收到链接，请撤回", false, { at: true })
  }

  /** #抽卡记录 */
  async getLog() {
    this.e.isAll = !!(this.e.msg.includes("全部"))
    let data = await new GachaLog(this.e).getLogData()
    if (!data) return
    let name = `html/gacha/gacha-log`
    if (this.e.isAll) {
      name = `html/gacha/gacha-all-log`
    }
    this.reply([await this.renderImg("genshin", name, data, { retType: "base64" }), this.button])
  }

  /** 导出记录 */
  exportLog() {
    if (this.e.isGroup && !this.e.msg.includes("强制")) {
      return this.reply("建议私聊导出，若你确认要在此导出，请发送【#强制导出记录】", false, { at: true })
    }

    return new ExportLog(this.e).exportJson()
  }

  async logJson() {
    if (!this.e.file)
      return this.e.reply("请发送Json文件")

    await new ExportLog(this.e).logJson()

    if (this.e.isGroup)
      this.e.reply("已收到文件，请撤回", false, { at: true })
  }

  help() {
    this.e.reply([segment.image(`file://${_path}/resources/logHelp/记录帮助.png`), segment.button([
      { text: "电脑", callback: "#电脑帮助" },
      { text: "安卓", callback: "#安卓帮助" },
      { text: "苹果", callback: "#苹果帮助" },
    ])])
  }

  helpPort() {
    let msg = this.e.msg.replace(/#|帮助/g, "")

    if (["电脑", "pc"].includes(msg)) {
      this.e.reply(segment.image(`file://${_path}/resources/logHelp/记录帮助-电脑.png`))
    } else if (["安卓"].includes(msg)) {
      this.e.reply(`安卓抽卡记录获取教程：${this.androidUrl}`)
    } else if (["苹果", "ios"].includes(msg)) {
      this.e.reply(segment.image(`file://${_path}/resources/logHelp/记录帮助-苹果.png`))
    }
  }

  async logCount() {
    let data = await new LogCount(this.e).count()
    if (!data) return

    this.reply([await this.renderImg("genshin", `html/gacha/log-count`, data, { retType: "base64" }), this.button])
  }
}