From 7ea47b46062be98a41421a0057ecb7776dbde098 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=9B=AE=E6=A3=83?= <BT-Muli@outlook.com>
Date: Thu, 20 Jun 2024 21:49:20 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20=E4=BF=AE=E5=A4=8D=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E6=81=A2=E5=A4=8D=E5=BC=82=E5=B8=B8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/config/tc-info.vue       |  6 ++---
 src/pages/common/Config.vue             |  4 +--
 src/plugins/Sqlite/index.ts             |  4 +--
 src/plugins/Sqlite/modules/userGacha.ts |  8 +++---
 src/utils/dataBS.ts                     | 34 +++++++++++++++++--------
 vite.config.ts                          |  2 +-
 6 files changed, 36 insertions(+), 22 deletions(-)

diff --git a/src/components/config/tc-info.vue b/src/components/config/tc-info.vue
index dca9acdd..88e315d6 100644
--- a/src/components/config/tc-info.vue
+++ b/src/components/config/tc-info.vue
@@ -76,9 +76,9 @@
 import { app, os } from "@tauri-apps/api";
 import { onMounted, ref } from "vue";
 
-import TGSqlite from "../../plugins/Sqlite";
-import { useAchievementsStore } from "../../store/modules/achievements";
-import showSnackbar from "../func/snackbar";
+import TGSqlite from "../../plugins/Sqlite/index.js";
+import { useAchievementsStore } from "../../store/modules/achievements.js";
+import showSnackbar from "../func/snackbar.js";
 
 const achievementsStore = useAchievementsStore();
 
diff --git a/src/pages/common/Config.vue b/src/pages/common/Config.vue
index fdca8890..4f507018 100644
--- a/src/pages/common/Config.vue
+++ b/src/pages/common/Config.vue
@@ -6,14 +6,14 @@
     <v-list class="config-list">
       <v-list-subheader :inset="true" class="config-header" title="设置" />
       <v-divider :inset="true" class="border-opacity-75" />
-      <v-list-item title="数据备份" @click="confirmBackup">
+      <v-list-item title="数据备份" @click="confirmBackup()">
         <template #prepend>
           <div class="config-icon">
             <v-icon>mdi-database-export</v-icon>
           </div>
         </template>
       </v-list-item>
-      <v-list-item title="数据恢复" @click="confirmRestore">
+      <v-list-item title="数据恢复" @click="confirmRestore()">
         <template #prepend>
           <div class="config-icon">
             <v-icon>mdi-database-import</v-icon>
diff --git a/src/plugins/Sqlite/index.ts b/src/plugins/Sqlite/index.ts
index a508a93e..d721b026 100644
--- a/src/plugins/Sqlite/index.ts
+++ b/src/plugins/Sqlite/index.ts
@@ -7,7 +7,7 @@
 import { app } from "@tauri-apps/api";
 import Database from "tauri-plugin-sql-api";
 
-import initDataSql from "./sql/initData";
+import initDataSql from "./sql/initData.js";
 import {
   importAbyssData,
   insertAbyssData,
@@ -15,7 +15,7 @@ import {
   insertGameAccountData,
   insertRecordData,
   insertRoleData,
-} from "./sql/insertData";
+} from "./sql/insertData.js";
 
 class Sqlite {
   /**
diff --git a/src/plugins/Sqlite/modules/userGacha.ts b/src/plugins/Sqlite/modules/userGacha.ts
index 3cc7fcb3..cb7df47c 100644
--- a/src/plugins/Sqlite/modules/userGacha.ts
+++ b/src/plugins/Sqlite/modules/userGacha.ts
@@ -4,9 +4,9 @@
  * @since Beta v0.4.7
  */
 
-import { AppCharacterData, AppWeaponData } from "../../../data/index";
-import TGSqlite from "../index";
-import { importUIGFData } from "../sql/updateData";
+import { AppCharacterData, AppWeaponData } from "../../../data/index.js";
+import TGSqlite from "../index.js";
+import { importUIGFData } from "../sql/updateData.js";
 
 type gachaItemTypeRes =
   | ["角色", TGApp.App.Character.WikiBriefInfo]
@@ -34,7 +34,7 @@ function getGachaItemType(item_id: string): gachaItemTypeRes {
  * @return {TGApp.Plugins.UIGF.GachaItem} 转换后的数据
  */
 function transGacha(gacha: TGApp.Plugins.UIGF.GachaItem): TGApp.Plugins.UIGF.GachaItem {
-  const type = getGachaItemType(gacha.item_id);
+  const type = getGachaItemType(gacha.item_id!);
   let res = gacha;
   res.item_type = type[0];
   if (type[0] === "角色") {
diff --git a/src/utils/dataBS.ts b/src/utils/dataBS.ts
index 13f362f0..a2b578f5 100644
--- a/src/utils/dataBS.ts
+++ b/src/utils/dataBS.ts
@@ -1,16 +1,18 @@
 /**
  * @file utils/dataBS.ts
  * @description 用户数据的备份、恢复、迁移
- * @since Beta v0.4.7
+ * @since Beta v0.4.10
  */
 
 import { fs, path } from "@tauri-apps/api";
+import { FileEntry } from "@tauri-apps/api/fs";
 
-import showSnackbar from "../components/func/snackbar";
-import TGSqlite from "../plugins/Sqlite";
+import showSnackbar from "../components/func/snackbar.js";
+import TGSqlite from "../plugins/Sqlite/index.js";
 import TSUserAchi from "../plugins/Sqlite/modules/userAchi.js";
 import TSUserGacha from "../plugins/Sqlite/modules/userGacha.js";
 
+import TGLogger from "./TGLogger.js";
 import { exportUigfData, readUigfData, verifyUigfData } from "./UIGF.js";
 
 /**
@@ -56,7 +58,9 @@ export async function restoreUserData(dir: string): Promise<void> {
     });
     return;
   }
-  const files = (await fs.readDir(dir)).filter((item) => item.type === "File");
+  const filesRead = await fs.readDir(dir, { recursive: false });
+  const files: FileEntry[] = filesRead.filter((item: FileEntry) => item.path.endsWith(".json"));
+  await TGLogger.Info(`[DataBS][restoreUserData] files: ${JSON.stringify(files)}`);
   // 恢复成就数据
   const achiFind = files.find((item) => item.name === "UIAF.json");
   if (achiFind) {
@@ -65,11 +69,13 @@ export async function restoreUserData(dir: string): Promise<void> {
         await fs.readTextFile(achiFind.path),
       );
       await TSUserAchi.mergeUIAF(dataAchi);
+      await TGLogger.Info(`[DataBS][restoreUserData] 成就数据恢复成功`);
     } catch (e) {
       showSnackbar({
         text: `成就数据恢复失败 ${e}`,
         color: "error",
       });
+      await TGLogger.Error(`[DataBS][restoreUserData] 成就数据恢复失败 ${e}`);
       errNum++;
     }
   } else {
@@ -77,6 +83,8 @@ export async function restoreUserData(dir: string): Promise<void> {
       text: "成就数据恢复失败，备份文件不存在",
       color: "warn",
     });
+    await TGLogger.Warn(`[DataBS][restoreUserData] 未检测到成就数据备份文件`);
+    await new Promise((resolve) => setTimeout(resolve, 1500));
   }
   // 恢复 ck
   const ckFind = files.find((item) => item.name === "cookie.json");
@@ -84,11 +92,13 @@ export async function restoreUserData(dir: string): Promise<void> {
     try {
       const dataCK = await fs.readTextFile(ckFind.path);
       await TGSqlite.saveAppData("cookie", JSON.stringify(JSON.parse(dataCK)));
+      await TGLogger.Info(`[DataBS][restoreUserData] Cookie 数据恢复成功`);
     } catch (e) {
       showSnackbar({
         text: `Cookie 数据恢复失败 ${e}`,
         color: "error",
       });
+      await TGLogger.Error(`[DataBS][restoreUserData] Cookie 数据恢复失败 ${e}`);
       errNum++;
     }
   } else {
@@ -96,6 +106,8 @@ export async function restoreUserData(dir: string): Promise<void> {
       text: "Cookie 数据恢复失败，备份文件不存在",
       color: "warn",
     });
+    await TGLogger.Warn(`[DataBS][restoreUserData] 未检测到 Cookie 数据备份文件`);
+    await new Promise((resolve) => setTimeout(resolve, 1500));
   }
   // 恢复深渊数据
   const abyssFind = files.find((item) => item.name === "abyss.json");
@@ -117,27 +129,29 @@ export async function restoreUserData(dir: string): Promise<void> {
       text: "深渊数据恢复失败，备份文件不存在",
       color: "warn",
     });
+    await TGLogger.Warn(`[DataBS][restoreUserData] 未检测到深渊数据备份文件`);
+    await new Promise((resolve) => setTimeout(resolve, 1500));
   }
   // 恢复祈愿数据
   const reg = /UIGF_(\d+).json/;
-  const dataGachaList = files.filter((item) => reg.test(item.name));
+  const dataGachaList = files.filter((item) => reg.test(item.path));
   for (const item of dataGachaList) {
     const check = await verifyUigfData(item.path);
     if (!check) {
       errNum++;
       continue;
     }
+    const data = await readUigfData(item.path);
+    const uid = data.info.uid;
     try {
-      const data = await readUigfData(item.path);
-      const uid = data.info.uid;
-      for (const item of data.list) {
-        await TSUserGacha.mergeUIGF(uid, item);
-      }
+      await TSUserGacha.mergeUIGF(uid, data.list);
+      await TGLogger.Info(`[DataBS][restoreUserData] UID: ${uid} 祈愿数据恢复成功`);
     } catch (e) {
       showSnackbar({
         text: `UID: ${uid} 祈愿数据恢复失败`,
         color: "error",
       });
+      await TGLogger.Error(`[DataBS][restoreUserData] UID: ${uid} 祈愿数据恢复失败 ${e}`);
       errNum++;
     }
   }
diff --git a/vite.config.ts b/vite.config.ts
index a450d860..15dab8be 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -9,7 +9,7 @@ import { defineConfig } from "vite";
 // import VueDevtools from "vite-plugin-vue-devtools";
 import vuetify from "vite-plugin-vuetify";
 
-import buildTimePlugin from "./src/utils/TGBuild";
+import buildTimePlugin from "./src/utils/TGBuild.js";
 
 // https://vitejs.dev/config/
 export default defineConfig({