JKSV/source/tasks/saveimport.cpp

#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;
}