/*
 * OmniNX Installer - Update Mode
 * Selective cleanup and install when OmniNX is already installed.
 */

#include "install.h"
#include "deletion_lists_update.h"
#include "fs.h"
#include "version.h"
#include "gfx.h"
#include <libs/fatfs/ff.h>
#include <string.h>
#include <utils/sprintf.h>

#ifndef VERSION
#define VERSION "1.0.0"
#endif

#undef COLOR_CYAN
#undef COLOR_WHITE
#undef COLOR_GREEN
#undef COLOR_YELLOW
#undef COLOR_ORANGE
#undef COLOR_RED
#define COLOR_CYAN    0xFF00FFFF
#define COLOR_WHITE   0xFFFFFFFF
#define COLOR_GREEN   0xFF00FF00
#define COLOR_YELLOW  0xFFFFDD00
#define COLOR_ORANGE  0xFF00A5FF
#define COLOR_RED     0xFFFF0000

#define set_color install_set_color
#define check_and_clear_screen_if_needed install_check_and_clear_screen_if_needed
#define path_exists install_path_exists
#define count_directory_items install_count_directory_items

// Update mode: Cleanup specific directories/files
int update_mode_cleanup(omninx_variant_t variant) {
    (void)variant;
    check_and_clear_screen_if_needed();

    set_color(COLOR_WHITE);
    delete_path_lists_grouped("atmosphere/",
        atmosphere_dirs_to_delete,
        atmosphere_contents_dirs_to_delete,
        atmosphere_files_to_delete,
        NULL);

    delete_path_lists_grouped("bootloader/",
        bootloader_dirs_to_delete,
        bootloader_files_to_delete,
        NULL);

    delete_path_lists_grouped("config/",
        config_dirs_to_delete,
        NULL);

    delete_path_lists_grouped("switch/",
        switch_dirs_to_delete,
        switch_files_to_delete,
        NULL);

    delete_path_lists_grouped("Root-Dateien",
        root_files_to_delete,
        misc_dirs_to_delete,
        misc_files_to_delete,
        NULL);

    set_color(COLOR_GREEN);
    gfx_printf("  Bereinigung abgeschlossen!\n");
    set_color(COLOR_WHITE);

    return FR_OK;
}

// Update mode: Copy files from staging
int update_mode_install(omninx_variant_t variant) {
    int res;
    const char* staging = get_staging_path(variant);
    char src_path[256];

    if (!staging) {
        return FR_INVALID_PARAMETER;
    }

    check_and_clear_screen_if_needed();

    set_color(COLOR_YELLOW);
    gfx_printf("Dateien werden kopiert...\n");
    set_color(COLOR_WHITE);

    s_printf(src_path, "%s/atmosphere", staging);
    res = folder_copy_with_progress_v2(src_path, "sd:/", "atmosphere/");
    if (res != FR_OK && res != FR_NO_FILE) return res;

    s_printf(src_path, "%s/bootloader", staging);
    res = folder_copy_with_progress_v2(src_path, "sd:/", "bootloader/");
    if (res != FR_OK && res != FR_NO_FILE) return res;

    s_printf(src_path, "%s/config", staging);
    res = folder_copy_with_progress_v2(src_path, "sd:/", "config/");
    if (res != FR_OK && res != FR_NO_FILE) return res;

    s_printf(src_path, "%s/switch", staging);
    res = folder_copy_with_progress_v2(src_path, "sd:/", "switch/");
    if (res != FR_OK && res != FR_NO_FILE) return res;

    s_printf(src_path, "%s/themes", staging);
    res = folder_copy_with_progress_v2(src_path, "sd:/", "themes/");
    if (res != FR_OK && res != FR_NO_FILE) return res;

    s_printf(src_path, "%s/warmboot_mariko", staging);
    res = folder_copy_with_progress_v2(src_path, "sd:/", "warmboot_mariko/");
    if (res != FR_OK && res != FR_NO_FILE) return res;

    if (variant == VARIANT_OC) {
        s_printf(src_path, "%s/SaltySD", staging);
        res = folder_copy_with_progress_v2(src_path, "sd:/", "SaltySD/");
        if (res != FR_OK && res != FR_NO_FILE) return res;
    }

    set_color(COLOR_CYAN);
    gfx_printf("  Kopiere Root-Dateien...\n");
    set_color(COLOR_WHITE);

    res = install_copy_staging_root_files(staging, "sd:/");
    if (res != FR_OK)
        return res;

    set_color(COLOR_GREEN);
    gfx_printf("  Kopie abgeschlossen!\n");
    set_color(COLOR_WHITE);

    return FR_OK;
}

// Remove staging directory after installation
int cleanup_staging_directory(omninx_variant_t pack_variant) {
    const char* staging = get_staging_path(pack_variant);
    if (!staging) {
        return FR_INVALID_PARAMETER;
    }

    check_and_clear_screen_if_needed();

    if (path_exists(staging)) {
        set_color(COLOR_YELLOW);
        gfx_printf("\nEntferne Installationsordner...\n");
        set_color(COLOR_WHITE);

        int total = count_directory_items(staging);
        u32 start_x, start_y;
        gfx_con_getpos(&start_x, &start_y);

        const char *folder_name = strrchr(staging, '/');
        if (folder_name) folder_name++;
        else folder_name = staging;

        set_color(COLOR_CYAN);
        gfx_printf("  Loesche: %s [  0%%] (0/%d)", folder_name, total);
        set_color(COLOR_WHITE);

        int deleted = 0;
        int last_percent = -1;
        int res = folder_delete_progress_recursive(staging, &deleted, total, start_x, start_y, folder_name, &last_percent);

        gfx_con_setpos(start_x, start_y);
        if (res == FR_OK) {
            set_color(COLOR_GREEN);
            gfx_printf("  Loesche: %s [100%%] (%d/%d) - Fertig!\n", folder_name, deleted, total);
            set_color(COLOR_WHITE);
        } else {
            set_color(COLOR_ORANGE);
            gfx_printf("  Loesche: %s - Fehlgeschlagen!\n", folder_name);
            gfx_printf("  [WARN] Ordner konnte nicht entfernt werden (err=%d)\n", res);
            set_color(COLOR_WHITE);
        }
        return res;
    }

    return FR_OK;
}