♻️ UIAF重构，支持祈愿备份/恢复

close #109

src/utils/dataBS.ts

@@ -1,17 +1,21 @@
 /**
  * @file utils/dataBS.ts
  * @description 用户数据的备份、恢复、迁移
- * @since Beta v0.4.1
+ * @since Beta v0.4.7
  */
 
 import { fs, path } from "@tauri-apps/api";
 
 import showSnackbar from "../components/func/snackbar";
 import TGSqlite from "../plugins/Sqlite";
+import TSUserAchi from "../plugins/Sqlite/modules/userAchi.js";
+import TSUserGacha from "../plugins/Sqlite/modules/userGacha.js";
+
+import { exportUigfData, readUigfData, verifyUigfData } from "./UIGF.js";
 
 /**
  * @description 备份用户数据
- * @since Beta v0.4.1
+ * @since Beta v0.4.7
  * @param {string} dir 备份目录路径
  * @returns {Promise<void>}
  */
@@ -20,8 +24,7 @@ export async function backUpUserData(dir: string): Promise<void> {
     console.log("备份目录不存在，创建备份目录");
     await fs.createDir(dir, { recursive: true });
   }
-  // 备份成就数据
-  const dataAchi = await TGSqlite.getUIAF();
+  const dataAchi = await TSUserAchi.getUIAF();
   await fs.writeTextFile(`${dir}${path.sep}UIAF.json`, JSON.stringify(dataAchi));
   // 备份 ck
   const dataCK = await TGSqlite.getCookie();
@@ -29,12 +32,17 @@ export async function backUpUserData(dir: string): Promise<void> {
   // 备份深渊数据
   const dataAbyss = await TGSqlite.getAbyss();
   await fs.writeTextFile(`${dir}${path.sep}abyss.json`, JSON.stringify(dataAbyss));
-  // todo 添加祈愿数据备份支持
+  // 备份祈愿数据
+  const uidList = await TSUserGacha.getUidList();
+  for (const uid of uidList) {
+    const dataGacha = await TSUserGacha.getGachaRecords(uid);
+    await exportUigfData(uid, dataGacha);
+  }
 }
 
 /**
  * @description 恢复用户数据
- * @since Beta v0.4.1
+ * @since Beta v0.4.7
  * @param {string} dir 备份目录路径
  * @returns {Promise<void>}
  */
@@ -47,17 +55,18 @@ export async function restoreUserData(dir: string): Promise<void> {
     });
     return;
   }
+  const files = (await fs.readDir(dir)).filter((item) => item.type === "File");
   // 恢复成就数据
-  const dataAchiPath = `${dir}${path.sep}UIAF.json`;
-  if (await fs.exists(dataAchiPath)) {
+  const achiFind = files.find((item) => item.name === "UIAF.json");
+  if (achiFind) {
     try {
       const dataAchi: TGApp.Plugins.UIAF.Achievement[] = JSON.parse(
-        await fs.readTextFile(dataAchiPath),
+        await fs.readTextFile(achiFind.path),
       );
-      await TGSqlite.mergeUIAF(dataAchi);
+      await TSUserAchi.mergeUIAF(dataAchi);
     } catch (e) {
       showSnackbar({
-        text: "成就数据恢复失败",
+        text: `成就数据恢复失败 ${e}`,
         color: "error",
       });
       errNum++;
@@ -69,14 +78,14 @@ export async function restoreUserData(dir: string): Promise<void> {
     });
   }
   // 恢复 ck
-  const dataCKPath = `${dir}${path.sep}cookie.json`;
-  if (await fs.exists(dataCKPath)) {
+  const ckFind = files.find((item) => item.name === "cookie.json");
+  if (ckFind) {
     try {
-      const dataCK = await fs.readTextFile(dataCKPath);
+      const dataCK = await fs.readTextFile(ckFind.path);
       await TGSqlite.saveAppData("cookie", JSON.stringify(JSON.parse(dataCK)));
     } catch (e) {
       showSnackbar({
-        text: "Cookie 数据恢复失败",
+        text: `Cookie 数据恢复失败 ${e}`,
         color: "error",
       });
       errNum++;
@@ -88,11 +97,11 @@ export async function restoreUserData(dir: string): Promise<void> {
     });
   }
   // 恢复深渊数据
-  const dataAbyssPath = `${dir}${path.sep}abyss.json`;
-  if (await fs.exists(dataAbyssPath)) {
+  const abyssFind = files.find((item) => item.name === "abyss.json");
+  if (abyssFind) {
     try {
       const dataAbyss: TGApp.Sqlite.Abyss.SingleTable[] = JSON.parse(
-        await fs.readTextFile(dataAbyssPath),
+        await fs.readTextFile(abyssFind.path),
       );
       await TGSqlite.restoreAbyss(dataAbyss);
     } catch (e) {
@@ -108,6 +117,29 @@ export async function restoreUserData(dir: string): Promise<void> {
       color: "warn",
     });
   }
+  // 恢复祈愿数据
+  const reg = /UIGF_(\d+).json/;
+  const dataGachaList = files.filter((item) => reg.test(item.name));
+  for (const item of dataGachaList) {
+    const check = await verifyUigfData(item.path);
+    if (!check) {
+      errNum++;
+      continue;
+    }
+    try {
+      const data = await readUigfData(item.path);
+      const uid = data.info.uid;
+      for (const item of data.list) {
+        await TSUserGacha.mergeUIGF(uid, item);
+      }
+    } catch (e) {
+      showSnackbar({
+        text: `UID: ${uid} 祈愿数据恢复失败`,
+        color: "error",
+      });
+      errNum++;
+    }
+  }
   if (errNum === 0) {
     showSnackbar({
       text: "数据恢复成功",