SaveImportState progress.

2026-08-21 18:05:23 -05:00 · 2025-09-24 15:32:57 -04:00
parent a78dd3456a
commit 9d542b4314
9 changed files with 222 additions and 34 deletions
--- a/source/tasks/saveimport.cpp
+++ b/source/tasks/saveimport.cpp
@@ -0,0 +1,104 @@
+#include "tasks/saveimport.hpp"
+
+#include "appstates/SaveImportState.hpp"
+#include "error.hpp"
+#include "fs/fs.hpp"
+#include "fslib.hpp"
+#include "strings/strings.hpp"
+#include "tasks/backup.hpp"
+#include "ui/PopMessageManager.hpp"
+
+// Defined at bottom.
+static bool read_save_meta(const fslib::Path &path, fs::SaveMetaData &metaOut);
+static bool test_for_save(data::User *user, const fs::SaveMetaData &saveMeta);
+
+void tasks::saveimport::import_save_backup(sys::threadpool::JobData taskData)
+{
+    auto castData = std::static_pointer_cast<SaveImportState::DataStruct>(taskData);
+
+    sys::ProgressTask *task = static_cast<sys::ProgressTask *>(castData->task);
+    if (error::is_null(task)) { return; }
+
+    data::User *user          = castData->user;
+    const fslib::Path &target = castData->path;
+
+    fs::SaveMetaData metaData{};
+    const bool metaRead = read_save_meta(target, metaData);
+    if (!metaRead) { TASK_FINISH_RETURN(task); }
+
+    const bool saveExists = test_for_save(user, metaData);
+    if (!saveExists) { fs::create_save_data_for(user, metaData); }
+}
+
+static bool read_save_meta(const fslib::Path &path, fs::SaveMetaData &metaOut)
+{
+    const bool isDir = fslib::directory_exists(path);
+    if (isDir)
+    {
+        const fslib::Path metaPath{path / fs::NAME_SAVE_META};
+        const bool exists = fslib::file_exists(metaPath);
+        if (!exists) { return false; }
+
+        fslib::File metaFile{metaPath, FsOpenMode_Read};
+        const bool metaRead = metaFile.is_open() && metaFile.get_size() <= fs::SIZE_SAVE_META &&
+                              metaFile.read(&metaOut, fs::SIZE_SAVE_META) <= fs::SIZE_SAVE_META;
+        const bool magicMatch = metaRead && metaOut.magic == fs::SAVE_META_MAGIC;
+        if (!metaRead || !magicMatch) { return false; }
+    }
+    else
+    {
+        fs::MiniUnzip zip{path};
+        if (!zip.is_open()) { return false; }
+
+        const bool located    = zip.locate_file(fs::NAME_SAVE_META);
+        const bool sizeMatch  = located && zip.get_uncompressed_size() <= fs::SIZE_SAVE_META;
+        const bool metaRead   = sizeMatch && zip.read(&metaOut, fs::SIZE_SAVE_META) <= fs::SIZE_SAVE_META;
+        const bool magicMatch = metaRead && metaOut.magic == fs::SAVE_META_MAGIC;
+        if (!metaRead || !magicMatch) { return false; }
+    }
+
+    return true;
+}
+
+static bool test_for_save(data::User *user, const fs::SaveMetaData &saveMeta)
+{
+    const FsSaveDataType saveType = user->get_account_save_type();
+
+    bool mounted{};
+    switch (saveType)
+    {
+        case FsSaveDataType_Account:
+        {
+            mounted =
+                fslib::open_account_save_file_system(fs::DEFAULT_SAVE_MOUNT, saveMeta.applicationID, user->get_account_id());
+        }
+        break;
+
+        case FsSaveDataType_Bcat:
+        {
+            mounted = fslib::open_bcat_save_file_system(fs::DEFAULT_SAVE_MOUNT, saveMeta.applicationID);
+        }
+        break;
+
+        case FsSaveDataType_Device:
+        {
+            mounted = fslib::open_device_save_file_system(fs::DEFAULT_SAVE_MOUNT, saveMeta.applicationID);
+        }
+        break;
+
+        case FsSaveDataType_Cache:
+        {
+            mounted = fslib::open_cache_save_file_system(fs::DEFAULT_SAVE_MOUNT,
+                                                         saveMeta.applicationID,
+                                                         saveMeta.saveDataIndex,
+                                                         static_cast<FsSaveDataSpaceId>(saveMeta.saveDataIndex));
+        }
+        break;
+
+        default: return false;
+    }
+
+    if (mounted) { fslib::close_file_system(fs::DEFAULT_SAVE_MOUNT); }
+
+    return mounted;
+}