From 888f1f2bdd8d02c877999955a8d639b4f4865fa6 Mon Sep 17 00:00:00 2001
From: BTMuli <BT-Muli@outlook.com>
Date: Tue, 12 Dec 2023 13:53:32 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9A=A1=EF=B8=8F=20=E5=A2=9E=E5=8A=A0?=
 =?UTF-8?q?=E7=9B=AE=E5=BD=95=E5=AD=98=E5=9C=A8=E5=88=A4=E6=96=AD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/pages/User/Gacha.vue    |  5 ++++-
 src/pages/common/Config.vue | 15 +++++++++++++--
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/src/pages/User/Gacha.vue b/src/pages/User/Gacha.vue
index e848acb1..a68b37d1 100644
--- a/src/pages/User/Gacha.vue
+++ b/src/pages/User/Gacha.vue
@@ -47,7 +47,7 @@
   </div>
 </template>
 <script lang="ts" setup>
-import { dialog, path } from "@tauri-apps/api";
+import { dialog, fs, path } from "@tauri-apps/api";
 import { computed, onMounted, ref, watch } from "vue";
 
 import showConfirm from "../../components/func/confirm";
@@ -352,6 +352,9 @@ async function backupGacha(): Promise<void> {
   }
   loadingTitle.value = "正在备份祈愿数据";
   loading.value = true;
+  if (!(await fs.exists("userData", { dir: fs.BaseDirectory.AppLocalData }))) {
+    await fs.createDir("userData", { dir: fs.BaseDirectory.AppLocalData, recursive: true });
+  }
   await backupUigfData(uidCur.value, gachaListCur.value);
   loading.value = false;
   showSnackbar({
diff --git a/src/pages/common/Config.vue b/src/pages/common/Config.vue
index 5bc5f322..ff359fea 100644
--- a/src/pages/common/Config.vue
+++ b/src/pages/common/Config.vue
@@ -149,7 +149,7 @@
 
 <script lang="ts" setup>
 import { app, fs, invoke, os } from "@tauri-apps/api";
-import { exit } from "@tauri-apps/api/process";
+import { relaunch } from "@tauri-apps/api/types/process";
 import { computed, onMounted, ref } from "vue";
 
 import showConfirm from "../../components/func/confirm";
@@ -378,6 +378,10 @@ async function confirmBackup(): Promise<void> {
   }
   loadingTitle.value = "正在备份数据...";
   loading.value = true;
+  if (!(await fs.exists("userData", { dir: fs.BaseDirectory.AppLocalData }))) {
+    await fs.createDir("userData", { dir: fs.BaseDirectory.AppLocalData, recursive: true });
+  }
+  console.info("数据文件夹创建完成！");
   loadingSub.value = "正在获取成就数据";
   const achievements = await TGSqlite.getUIAF();
   loadingSub.value = "正在备份成就数据";
@@ -410,6 +414,13 @@ async function confirmRestore(): Promise<void> {
   }
   loadingTitle.value = "正在恢复数据...";
   loading.value = true;
+  if (!(await fs.exists("userData", { dir: fs.BaseDirectory.AppLocalData }))) {
+    showSnackbar({
+      color: "error",
+      text: "数据文件夹不存在！",
+    });
+    return;
+  }
   const fail: string[] = [];
   let res: boolean;
   loadingSub.value = "正在恢复成就数据";
@@ -523,7 +534,7 @@ async function confirmDelCache(): Promise<void> {
   });
   await new Promise(() => {
     setTimeout(async () => {
-      await exit();
+      await relaunch();
     }, 1500);
   });
 }