OmniNX-Installer-Payload/source/backup.c

/*
 * OmniNX Installer - Backup and Restore Operations
 */

#include "backup.h"
#include "fs.h"
#include <libs/fatfs/ff.h>
#include <string.h>
#include <utils/sprintf.h>

// Check if file/directory exists
static bool path_exists(const char *path) {
    FILINFO fno;
    return (f_stat(path, &fno) == FR_OK);
}

// Backup user data before clean install
int backup_user_data(void) {
    int res;
    
    // Create temp backup directory
    res = f_mkdir(TEMP_BACKUP_PATH);
    if (res != FR_OK && res != FR_EXIST) {
        return res;
    }
    
    // Backup DBI settings file only (not .nro installers)
    if (path_exists(DBI_CONFIG_PATH)) {
        res = file_copy(DBI_CONFIG_PATH, TEMP_BACKUP_DBI_CONFIG);
        if (res != FR_OK) {
            return res;
        }
    }
    
    // Backup prod.keys if it exists
    if (path_exists("sd:/switch/prod.keys")) {
        res = file_copy("sd:/switch/prod.keys", "sd:/temp_backup/prod.keys");
        if (res != FR_OK) {
            return res;
        }
    }

    // Backup CyberFoil (nro, config, remotes) if installed
    if (path_exists(CYBERFOIL_DIR)) {
        res = f_mkdir(TEMP_BACKUP_CYBERFOIL);
        if (res != FR_OK && res != FR_EXIST) {
            return res;
        }

        if (path_exists(CYBERFOIL_NRO)) {
            res = file_copy(CYBERFOIL_NRO, TEMP_BACKUP_CYBERFOIL_NRO);
            if (res != FR_OK) {
                return res;
            }
        }

        if (path_exists(CYBERFOIL_CONFIG)) {
            res = file_copy(CYBERFOIL_CONFIG, TEMP_BACKUP_CYBERFOIL_CONFIG);
            if (res != FR_OK) {
                return res;
            }
        }

        if (path_exists(CYBERFOIL_REMOTES)) {
            res = folder_copy(CYBERFOIL_REMOTES, TEMP_BACKUP_CYBERFOIL);
            if (res != FR_OK) {
                return res;
            }
        }
    }
    
    return FR_OK;
}

// Restore user data after clean install
int restore_user_data(void) {
    int res;
    
    // Recreate switch directory (should already exist, but be safe)
    res = f_mkdir("sd:/switch");
    if (res != FR_OK && res != FR_EXIST) {
        return res;
    }
    
    // Restore DBI settings if backed up
    if (path_exists(TEMP_BACKUP_DBI_CONFIG)) {
        res = f_mkdir("sd:/switch/DBI");
        if (res != FR_OK && res != FR_EXIST) {
            return res;
        }
        res = file_copy(TEMP_BACKUP_DBI_CONFIG, DBI_CONFIG_PATH);
        if (res != FR_OK) {
            return res;
        }
    }
    
    // Restore prod.keys if backup exists
    if (path_exists("sd:/temp_backup/prod.keys")) {
        res = file_copy("sd:/temp_backup/prod.keys", "sd:/switch/prod.keys");
        if (res != FR_OK) {
            return res;
        }
    }

    // Restore CyberFoil if backed up
    if (path_exists(TEMP_BACKUP_CYBERFOIL)) {
        res = f_mkdir(CYBERFOIL_DIR);
        if (res != FR_OK && res != FR_EXIST) {
            return res;
        }

        if (path_exists(TEMP_BACKUP_CYBERFOIL_NRO)) {
            res = file_copy(TEMP_BACKUP_CYBERFOIL_NRO, CYBERFOIL_NRO);
            if (res != FR_OK) {
                return res;
            }
        }

        if (path_exists(TEMP_BACKUP_CYBERFOIL_CONFIG)) {
            res = file_copy(TEMP_BACKUP_CYBERFOIL_CONFIG, CYBERFOIL_CONFIG);
            if (res != FR_OK) {
                return res;
            }
        }

        if (path_exists(TEMP_BACKUP_CYBERFOIL_REMOTES)) {
            res = folder_copy(TEMP_BACKUP_CYBERFOIL_REMOTES, CYBERFOIL_DIR);
            if (res != FR_OK) {
                return res;
            }
        }
    }
    
    return FR_OK;
}

// Clean up temporary backup directory
int cleanup_backup(void) {
    if (path_exists(TEMP_BACKUP_PATH)) {
        return folder_delete(TEMP_BACKUP_PATH);
    }
    return FR_OK;
}

int backup_horizon_oc_config_for_oc_update(void) {
    if (!path_exists(HORIZON_OC_CONFIG_PATH))
        return FR_OK;

    int res = f_mkdir(HORIZON_OC_UPDATE_BACKUP_DIR);
    if (res != FR_OK && res != FR_EXIST)
        return res;

    return file_copy(HORIZON_OC_CONFIG_PATH, HORIZON_OC_UPDATE_BACKUP_INI);
}

int restore_horizon_oc_config_after_oc_update(void) {
    if (!path_exists(HORIZON_OC_UPDATE_BACKUP_INI))
        return FR_OK;

    int res = f_mkdir("sd:/config");
    if (res != FR_OK && res != FR_EXIST)
        return res;
    res = f_mkdir("sd:/config/horizon-oc");
    if (res != FR_OK && res != FR_EXIST)
        return res;

    return file_copy(HORIZON_OC_UPDATE_BACKUP_INI, HORIZON_OC_CONFIG_PATH);
}

int cleanup_horizon_oc_update_backup(void) {
    if (!path_exists(HORIZON_OC_UPDATE_BACKUP_DIR))
        return FR_OK;
    return folder_delete(HORIZON_OC_UPDATE_BACKUP_DIR);
}