#include "../esp_flasher_app_i.h"

enum QuickState {
    QuickStart,
    QuickWifidevS2,
    QuickWifidevS2Blackmagic,
    QuickWifidevS2Marauder,
    QuickDevproWroom,
    QuickDevproWroomMarauder,
};

void esp_flasher_scene_quick_submenu_callback(void* context, uint32_t index) {
    furi_assert(context);
    EspFlasherApp* app = context;

    view_dispatcher_send_custom_event(app->view_dispatcher, index);
}

void esp_flasher_scene_quick_on_enter(void* context) {
    furi_assert(context);
    EspFlasherApp* app = context;
    Submenu* submenu = app->submenu;
    uint32_t state = scene_manager_get_scene_state(app->scene_manager, EspFlasherSceneQuick);

    switch(state) {
    case QuickStart:
    case QuickWifidevS2:
    case QuickDevproWroom:
        submenu_set_header(submenu, "Choose Board:");
        submenu_add_item(
            submenu,
            "WiFi Dev (ESP32-S2)",
            QuickWifidevS2,
            esp_flasher_scene_quick_submenu_callback,
            app);
        submenu_add_item(
            submenu,
            "Dev Pro (ESP32-WROOM)",
            QuickDevproWroom,
            esp_flasher_scene_quick_submenu_callback,
            app);
        break;
    case QuickWifidevS2Blackmagic:
    case QuickWifidevS2Marauder:
        submenu_set_header(submenu, "Choose Firmware:");
        submenu_add_item(
            submenu,
            "Black Magic",
            QuickWifidevS2Blackmagic,
            esp_flasher_scene_quick_submenu_callback,
            app);
        submenu_add_item(
            submenu,
            "Marauder (has Evil Portal)",
            QuickWifidevS2Marauder,
            esp_flasher_scene_quick_submenu_callback,
            app);
        break;
    case QuickDevproWroomMarauder:
        submenu_set_header(submenu, "Choose Firmware:");
        submenu_add_item(
            submenu,
            "Marauder (has Evil Portal)",
            QuickDevproWroomMarauder,
            esp_flasher_scene_quick_submenu_callback,
            app);
        break;
    default:
        break;
    }

    submenu_set_selected_item(submenu, state);

    view_dispatcher_switch_to_view(app->view_dispatcher, EspFlasherAppViewSubmenu);
}

bool esp_flasher_scene_quick_on_event(void* context, SceneManagerEvent event) {
    furi_assert(context);

    EspFlasherApp* app = context;
    bool consumed = false;
    if(event.type == SceneManagerEventTypeCustom) {
        consumed = true;

        bool flash = true;
        bool enter_bootloader = false;
        const char* boot = NULL; // 0x1000
        const char* part = NULL; // 0x8000
        const char* app0 = NULL; // 0xE000
        const char* firm = NULL; // 0x10000

        switch(event.event) {
        case QuickWifidevS2:
        case QuickDevproWroom:
            scene_manager_set_scene_state(
                app->scene_manager, EspFlasherSceneQuick, event.event + 1);
            scene_manager_next_scene(app->scene_manager, EspFlasherSceneQuick);
            flash = false;
            break;
        case QuickWifidevS2Blackmagic:
            boot = APP_DATA_PATH("assets/blackmagic/bootloader.bin");
            part = APP_DATA_PATH("assets/blackmagic/partition-table.bin");
            firm = APP_DATA_PATH("assets/blackmagic/blackmagic.bin");
            enter_bootloader = true;
            break;
        case QuickWifidevS2Marauder:
            boot = APP_DATA_PATH("assets/marauder/WifidevS2/esp32_marauder.ino.bootloader.bin");
            part = APP_DATA_PATH("assets/marauder/esp32_marauder.ino.partitions.bin");
            app0 = APP_DATA_PATH("assets/marauder/boot_app0.bin");
            firm = APP_DATA_PATH("assets/marauder/WifidevS2/esp32_marauder_flipper_sd_serial.bin");
            enter_bootloader = true;
            break;
        case QuickDevproWroomMarauder:
            boot = APP_DATA_PATH("assets/marauder/DevproWroom/esp32_marauder.ino.bootloader.bin");
            part = APP_DATA_PATH("assets/marauder/esp32_marauder.ino.partitions.bin");
            app0 = APP_DATA_PATH("assets/marauder/boot_app0.bin");
            firm = APP_DATA_PATH("assets/marauder/DevproWroom/esp32_marauder_dev_board_pro.bin");
            break;
        default:
            flash = false;
            consumed = false;
            break;
        }

        if(flash) {
            scene_manager_set_scene_state(app->scene_manager, EspFlasherSceneQuick, event.event);
            memset(app->selected_flash_options, 0, sizeof(app->selected_flash_options));
            app->bin_file_path_boot[0] = '\0';
            app->bin_file_path_part[0] = '\0';
            app->bin_file_path_nvs[0] = '\0';
            app->bin_file_path_boot_app0[0] = '\0';
            app->bin_file_path_app_a[0] = '\0';
            app->bin_file_path_app_b[0] = '\0';
            app->bin_file_path_custom[0] = '\0';

            if(boot) {
                app->selected_flash_options[SelectedFlashBoot] = true;
                strncpy(app->bin_file_path_boot, boot, sizeof(app->bin_file_path_boot));
            }
            if(part) {
                app->selected_flash_options[SelectedFlashPart] = true;
                strncpy(app->bin_file_path_part, part, sizeof(app->bin_file_path_part));
            }
            if(app0) {
                app->selected_flash_options[SelectedFlashBootApp0] = true;
                strncpy(app->bin_file_path_boot_app0, app0, sizeof(app->bin_file_path_boot_app0));
            }
            if(firm) {
                app->selected_flash_options[SelectedFlashAppA] = true;
                strncpy(app->bin_file_path_app_a, firm, sizeof(app->bin_file_path_app_a));
            }

            app->reset = false;
            app->quickflash = true;
            app->turbospeed = true;
            app->boot = enter_bootloader;
            scene_manager_next_scene(app->scene_manager, EspFlasherSceneConsoleOutput);
        }
    } else if(event.type == SceneManagerEventTypeBack) {
        uint32_t state = scene_manager_get_scene_state(app->scene_manager, EspFlasherSceneQuick);
        if(state > QuickDevproWroom)
            state = QuickDevproWroom;
        else if(state > QuickWifidevS2)
            state = QuickWifidevS2;
        scene_manager_set_scene_state(app->scene_manager, EspFlasherSceneQuick, state);
    }

    return consumed;
}

void esp_flasher_scene_quick_on_exit(void* context) {
    furi_assert(context);

    EspFlasherApp* app = context;
    submenu_reset(app->submenu);
}