TeyvatGuide/src/utils/UIGF.ts

/**
 * @file utils UIGF.ts
 * @description UIGF工具类
 * @author BTMuli <bt-muli@outlook.com>
 * @since Alpha v0.2.3
 */

// tauri
import { app, fs, path } from "@tauri-apps/api";
// utils
import TGSqlite from "../plugins/Sqlite";
import { timestampToDate } from "./t2D";

/**
 * @description 获取 UIGF 头部信息
 * @since Alpha v0.2.3
 * @param {string} uid - UID
 * @returns {Promise<TGApp.Plugins.UIGF.Export>}
 */
export async function getUigfHeader(uid: string): Promise<TGApp.Plugins.UIGF.Export> {
  const stamp = Date.now();
  return {
    uid,
    lang: "zh-cn",
    uigf_version: "2.3.0",
    export_timestamp: Math.floor(stamp / 1000),
    export_time: timestampToDate(stamp),
    export_app: "Tauri.Genshin",
    export_app_version: await app.getVersion(),
  };
}

/**
 * @description 数据转换-数据库到 UIGF
 * @since Alpha v0.2.3
 * @param {TGApp.Sqlite.GachaRecords.SingleTable[]} data - 数据库数据
 * @returns {TGApp.Plugins.UIGF.GachaItem[]} UIGF 数据
 */
export function convertDataToUigf(
  data: TGApp.Sqlite.GachaRecords.SingleTable[],
): TGApp.Plugins.UIGF.GachaItem[] {
  return data.map((gacha) => {
    return {
      gacha_type: gacha.gachaType,
      item_id: gacha.itemId,
      count: gacha.count,
      time: gacha.time,
      name: gacha.name,
      item_type: gacha.type,
      rank_type: gacha.rank,
      id: gacha.id,
      uigf_gacha_type: gacha.uigfType,
    };
  });
}

/**
 * @description 检测是否存在 UIGF 数据
 * @description 粗略检测，不保证数据完整性
 * @since Alpha v0.2.3
 * @param {string} path - UIGF 数据路径
 * @returns {Promise<boolean>} 是否存在 UIGF 数据
 */
export async function verifyUigfData(path: string): Promise<boolean> {
  const fileData: string = await fs.readTextFile(path);
  const UigfData: TGApp.Plugins.UIGF.Export = JSON.parse(fileData)?.info;
  return UigfData?.uigf_version !== undefined;
}

/**
 * @description 读取 UIGF 数据
 * @since Alpha v0.2.3
 * @param {string} userPath - UIGF 数据路径
 * @returns {Promise<TGApp.Plugins.UIGF.FullData>} UIGF 数据
 */
export async function readUigfData(userPath: string): Promise<TGApp.Plugins.UIGF.FullData> {
  const fileData = await fs.readTextFile(userPath);
  return <TGApp.Plugins.UIGF.FullData>JSON.parse(fileData);
}

/**
 * @description 导出 UIGF 数据
 * @since Alpha v0.2.3
 * @param {string} uid - UID
 * @param {TGApp.Sqlite.GachaRecords.SingleTable[]} gachaList - 祈愿列表
 * @param {string} savePath - 保存路径
 * @returns {Promise<void>}
 */
export async function exportUigfData(
  uid: string,
  gachaList: TGApp.Sqlite.GachaRecords.SingleTable[],
  savePath?: string,
): Promise<void> {
  const UigfData = {
    info: await getUigfHeader(uid),
    list: convertDataToUigf(gachaList),
  };
  const filePath = savePath ?? `${await path.appLocalDataDir()}userData\\UIGF_${uid}.json`;
  await fs.writeTextFile(filePath, JSON.stringify(UigfData));
}

/**
 * @description 备份 UIGF 数据
 * @since Alpha v0.2.3
 * @param {string} uid - UID
 * @param {TGApp.Sqlite.GachaRecords.SingleTable[]} gachaList - 祈愿列表
 * @returns {Promise<void>}
 */
export async function backupUigfData(
  uid: string,
  gachaList: TGApp.Sqlite.GachaRecords.SingleTable[],
): Promise<void> {
  const savePath = `${await path.appLocalDataDir()}userData\\UIGF_${uid}.json`;
  await exportUigfData(uid, gachaList, savePath);
}

/**
 * @description 恢复 UIGF 数据
 * @since Alpha v0.2.3
 * @param {string} uid - UID
 * @returns {Promise<boolean>} UIGF 数据
 */
export async function restoreUigfData(uid: string): Promise<boolean> {
  const uigfPath = `${await path.appLocalDataDir()}userData\\UIGF_${uid}.json`;
  if (!(await fs.exists(uigfPath))) {
    return false;
  }
  const uigfData = await readUigfData(uigfPath);
  await TGSqlite.mergeUIGF(uid, uigfData.list);
  return true;
}