✨ 重构UIGF导入导出备份恢复，支持UIGF4.2

Author: BTMuli

src/components/userGacha/ugo-uid.vue

@@ -1,3 +1,4 @@
+<!-- UIGF4导入导出组件 -->
 <template>
   <TOverlay v-model="visible" blur-val="5px">
     <div class="ugo-box">
@@ -32,7 +33,10 @@
         >
           <div class="ugo-item" @click="toggle">
             <div class="ugo-item-left">
-              <div class="ugo-item-title">{{ item.uid }} - {{ item.length }}条记录</div>
+              <div class="ugo-item-title">
+                <span>{{ item.uid }} - {{ item.length }}条</span>
+                <span>{{ item.isUgc ? "颂愿记录" : "祈愿记录" }}</span>
+              </div>
               <div class="ugo-item-sub">{{ item.time }}</div>
             </div>
             <div class="ugo-item-right">
@@ -55,6 +59,7 @@ import TOverlay from "@comp/app/t-overlay.vue";
 import showLoading from "@comp/func/loading.js";
 import showSnackbar from "@comp/func/snackbar.js";
 import TSUserGacha from "@Sqlm/userGacha.js";
+import TSUserGachaB from "@Sqlm/userGachaB.js";
 import { path } from "@tauri-apps/api";
 import { open } from "@tauri-apps/plugin-dialog";
 import TGLogger from "@utils/TGLogger.js";
@@ -68,7 +73,19 @@ type UgoUidProps = {
   /** filePathImport，导出路径 */
   fpi?: string;
 };
-type UgoUidItem = { uid: string; length: number; time: string };
+/**
+ * UID项
+ */
+type UgoUidItem = {
+  /** UID */
+  uid: string;
+  /** 数据条数 */
+  length: number;
+  /** 数据时间段 */
+  time: string;
+  /** 是否是颂愿数据 */
+  isUgc: boolean;
+};
 /** 兼容不同版本的导入 */
 type UgoUidImportRaw =
   | { isV4: true; data: TGApp.Plugins.UIGF.Schema4 }
@@ -98,7 +115,7 @@ watch(
 
 async function getDefaultSavePath(): Promise<string> {
   const tsNow = Math.floor(Date.now() / 1000);
-  return `${await path.downloadDir()}${path.sep()}UIGFv4.1_${tsNow}.json`;
+  return `${await path.downloadDir()}${path.sep()}UIGFv4.2_${tsNow}.json`;
 }
 
 async function refreshData(): Promise<void> {
@@ -147,7 +164,9 @@ async function refreshImport(): Promise<void> {
     if (isV4) {
       const read = await readUigf4Data(fp.value);
       importRaw.value = { isV4: true, data: read };
-      data.value = read.hk4e.map(parseData4);
+      const hk4eUids = read.hk4e?.map(parseData4) ?? [];
+      const ugcUids = read.hk4e_ugc?.map(parseUgc) ?? [];
+      data.value = [...hk4eUids, ...ugcUids];
     } else {
       const read = await readUigfData(fp.value);
       console.log(read.list.length);
@@ -172,6 +191,7 @@ function parseData(data: TGApp.Plugins.UIGF.Schema): UgoUidItem {
     uid: data.info.uid,
     length: data.list.length,
     time: `${timestampToDate(Math.min(...timeList))} ~ ${timestampToDate(Math.max(...timeList))}`,
+    isUgc: false,
   };
 }
 
@@ -181,25 +201,52 @@ function parseData4(data: TGApp.Plugins.UIGF.GachaHk4e): UgoUidItem {
     uid: data.uid.toString(),
     length: data.list.length,
     time: `${timestampToDate(Math.min(...timeList))} ~ ${timestampToDate(Math.max(...timeList))}`,
+    isUgc: false,
+  };
+}
+
+function parseUgc(data: TGApp.Plugins.UIGF.GachaUgc): UgoUidItem {
+  const timeList = data.list.map((item) => new Date(item.time).getTime());
+  return {
+    uid: data.uid.toString(),
+    length: data.list.length,
+    time: `${timestampToDate(Math.min(...timeList))} ~ ${timestampToDate(Math.max(...timeList))}`,
+    isUgc: true,
   };
 }
 
 async function refreshExport(): Promise<void> {
-  const uidList = await TSUserGacha.getUidList();
+  const uidHk4e = await TSUserGacha.getUidList();
+  const uidUgc = await TSUserGachaB.getUidList();
   const tmpData: Array<UgoUidItem> = [];
-  for (const uid of uidList) {
+  for (const uid of uidHk4e) {
     const dataRaw = await TSUserGacha.record.all(uid);
-    tmpData.push(parseDataRaw(dataRaw));
+    tmpData.push(parseLocalHk4e(dataRaw));
+  }
+  for (const uid of uidUgc) {
+    const dataRaw = await TSUserGachaB.getGachaRecords(uid);
+    tmpData.push(parseLocalUgc(dataRaw));
   }
   data.value = tmpData;
 }
 
-function parseDataRaw(data: Array<TGApp.Sqlite.Gacha.Gacha>): UgoUidItem {
+function parseLocalHk4e(data: Array<TGApp.Sqlite.Gacha.Gacha>): UgoUidItem {
   const timeList = data.map((item) => new Date(item.time).getTime());
   return {
     uid: data[0].uid,
     length: data.length,
     time: `${timestampToDate(Math.min(...timeList))} ~ ${timestampToDate(Math.max(...timeList))}`,
+    isUgc: false,
+  };
+}
+
+function parseLocalUgc(data: Array<TGApp.Sqlite.Gacha.GachaB>): UgoUidItem {
+  const timeList = data.map((item) => new Date(item.time).getTime());
+  return {
+    uid: data[0].uid,
+    length: data.length,
+    time: `${timestampToDate(Math.min(...timeList))} ~ ${timestampToDate(Math.max(...timeList))}`,
+    isUgc: true,
   };
 }
 
@@ -222,13 +269,23 @@ async function handleImport(): Promise<void> {
   if (importRaw.value.isV4) {
     for (const item of selectedData.value) {
       await showLoading.update(`正在导入UID: ${item.uid}`);
-      const dataFind = importRaw.value.data.hk4e.find((i) => i.uid.toString() === item.uid);
-      if (!dataFind) {
-        showSnackbar.error(`未找到UID: ${item.uid}`);
-        await new Promise<void>((resolve) => setTimeout(resolve, 1000));
-        continue;
+      if (!item.isUgc) {
+        const dataFind = importRaw.value.data.hk4e?.find((i) => i.uid.toString() === item.uid);
+        if (!dataFind) {
+          showSnackbar.error(`未找到UID: ${item.uid}`);
+          await new Promise<void>((resolve) => setTimeout(resolve, 1000));
+          continue;
+        }
+        await TSUserGacha.mergeUIGF4(dataFind, true);
+      } else {
+        const dataFind = importRaw.value.data.hk4e_ugc?.find((i) => i.uid.toString() === item.uid);
+        if (!dataFind) {
+          showSnackbar.error(`未找到UID: ${item.uid}`);
+          await new Promise<void>((resolve) => setTimeout(resolve, 1000));
+          continue;
+        }
+        await TSUserGachaB.mergeUIGF4(dataFind, true);
       }
-      await TSUserGacha.mergeUIGF4(dataFind, true);
     }
   } else {
     const iUid = selectedData.value[0].uid;
@@ -251,7 +308,8 @@ async function handleExport(): Promise<void> {
     `${selectedData.value.length}条UID - ${totalCnt}条记录`,
   );
   await exportUigf4Data(
-    selectedData.value.map((s) => s.uid.toString()),
+    selectedData.value.filter((i) => !i.isUgc).map((s) => s.uid.toString()),
+    selectedData.value.filter((i) => i.isUgc).map((s) => s.uid.toString()),
     fp.value,
   );
   await showLoading.end();
@@ -348,8 +406,17 @@ async function handleExport(): Promise<void> {
 }
 
 .ugo-item-title {
+  position: relative;
+  display: flex;
+  align-items: center;
+  justify-content: flex-start;
+  column-gap: 8px;
   font-family: var(--font-title);
   font-size: 16px;
+
+  :last-child {
+    color: var(--tgc-od-red);
+  }
 }
 
 .ugo-item-sub {

src/pages/User/GachaB.vue

@@ -1,4 +1,4 @@
-variantelevated<!-- 千星奇域祈愿记录页面 TODO：处理活动卡池次数共享 -->
+<!-- 千星奇域祈愿记录页面 TODO：处理活动卡池次数共享 -->
 <template>
   <v-app-bar>
     <template #prepend>
@@ -34,6 +34,31 @@ variantelevated<!-- 千星奇域祈愿记录页面 TODO：处理活动卡池次
         >
           全量刷新
         </v-btn>
+        <v-btn
+          class="gb-top-btn"
+          prepend-icon="mdi-import"
+          variant="elevated"
+          @click="importUigf()"
+        >
+          导入
+        </v-btn>
+        <v-btn
+          class="gb-top-btn"
+          prepend-icon="mdi-export"
+          variant="elevated"
+          @click="exportUigf()"
+        >
+          导出
+        </v-btn>
+        <v-btn
+          class="gb-top-btn"
+          prepend-icon="mdi-database"
+          title="将数据库中非中文数据转换为中文"
+          variant="elevated"
+          @click="checkData()"
+        >
+          检测数据
+        </v-btn>
         <v-btn
           class="gb-top-btn"
           prepend-icon="mdi-delete"
@@ -59,18 +84,22 @@ variantelevated<!-- 千星奇域祈愿记录页面 TODO：处理活动卡池次
       </v-window-item>
     </v-window>
   </div>
+  <UgoUid v-model="ovShow" :fpi="ovFpi" :mode="ovMode" />
 </template>
 <script lang="ts" setup>
 import showDialog from "@comp/func/dialog.js";
 import showLoading from "@comp/func/loading.js";
 import showSnackbar from "@comp/func/snackbar.js";
 import GbrOverview from "@comp/userGacha/gbr-overview.vue";
 import GbrTable from "@comp/userGacha/gbr-table.vue";
+import UgoUid from "@comp/userGacha/ugo-uid.vue";
 import hk4eReq from "@req/hk4eReq.js";
 import takumiReq from "@req/takumiReq.js";
 import TSUserGachaB from "@Sqlm/userGachaB.js";
 import useAppStore from "@store/app.js";
 import useUserStore from "@store/user.js";
+import { path } from "@tauri-apps/api";
+import { open } from "@tauri-apps/plugin-dialog";
 import TGLogger from "@utils/TGLogger.js";
 import { storeToRefs } from "pinia";
 import { onMounted, ref, shallowRef, watch } from "vue";
@@ -80,6 +109,10 @@ const router = useRouter();
 const { isLogin } = storeToRefs(useAppStore());
 const { account, cookie } = storeToRefs(useUserStore());
 
+const ovMode = ref<"export" | "import">("import");
+const ovShow = ref<boolean>(false);
+const ovFpi = ref<string>();
+
 const authkey = ref<string>("");
 const uidCur = ref<string>();
 const tab = ref<string>("overview");
@@ -219,27 +252,26 @@ async function refreshGachaPool(
       await new Promise<void>((resolve) => setTimeout(resolve, 1000));
       break;
     }
-    if (gachaRes.length === 0) {
-      // if (force) {
-      //   await showLoading.update(`正在清理${label}数据`);
-      //   if (gachaDataMap) {
-      //     await TSUserGacha.cleanGachaRecords(account.value.gameUid, type, gachaDataMap);
-      //   }
-      // }
-      break;
-    }
+    if (gachaRes.length === 0) break;
+    const uigfList: Array<TGApp.Plugins.UIGF.GachaItemB> = [];
     if (force) await showLoading.update(`[${gachaName}] 第${page}页，${gachaRes.length}条`);
     for (const item of gachaRes) {
       if (!force) {
         await showLoading.update(`[${item.item_type}][${item.time}] ${item.item_name}`);
       }
-      if (force) {
-        // if (!gachaDataMap) gachaDataMap = {};
-        // if (!gachaDataMap[item.time]) gachaDataMap[item.time] = [];
-        // gachaDataMap[item.time].push(item.id.toString());
-      }
+      const tempItem: TGApp.Plugins.UIGF.GachaItemB = {
+        id: item.id,
+        item_id: item.item_id,
+        item_name: item.item_name,
+        item_type: item.item_type,
+        op_gacha_type: item.op_gacha_type,
+        rank_type: item.rank_type,
+        schedule_id: item.schedule_id,
+        time: item.time,
+      };
+      uigfList.push(tempItem);
     }
-    await TSUserGachaB.insertGachaList(gachaRes);
+    await TSUserGachaB.insertGachaList(account.value.gameUid, uigfList);
     if (!force && gachaRes.some((i) => i.id.toString() === endId.toString())) break;
     reqId = gachaRes[gachaRes.length - 1].id.toString();
     if (force) await new Promise<void>((resolve) => setTimeout(resolve, 1000));
@@ -283,6 +315,39 @@ async function deleteGacha(): Promise<void> {
   await new Promise<void>((resolve) => setTimeout(resolve, 1500));
   window.location.reload();
 }
+
+async function importUigf(): Promise<void> {
+  await TGLogger.Info(`[UserGachaB][importUigf] 导入祈愿数据`);
+  const selectedFile = await open({
+    multiple: false,
+    title: "导入UIGF文件",
+    filters: [{ name: "UIGF JSON", extensions: ["json"] }],
+    defaultPath: await path.downloadDir(),
+    directory: false,
+  });
+  if (selectedFile === null) {
+    showSnackbar.cancel("已取消文件选择");
+    return;
+  }
+  ovFpi.value = selectedFile;
+  ovMode.value = "import";
+  ovShow.value = true;
+}
+
+async function exportUigf(): Promise<void> {
+  if (!uidCur.value) {
+    showSnackbar.error("未获取到 UID");
+    return;
+  }
+  await TGLogger.Info(`[UserGachaB][${uidCur.value}][exportUigf] 导出祈愿数据`);
+  ovMode.value = "export";
+  ovShow.value = true;
+}
+
+async function checkData(): Promise<void> {
+  // TODO
+  showSnackbar.warn("暂未支持");
+}
 </script>
 <style lang="scss" scoped>
 .gb-top-title {