#include "../infrared_i.h"

static void infrared_scene_edit_button_select_submenu_callback(void* context, uint32_t index) {
    Infrared* infrared = context;
    view_dispatcher_send_custom_event(infrared->view_dispatcher, index);
}

void infrared_scene_edit_button_select_on_enter(void* context) {
    Infrared* infrared = context;
    Submenu* submenu = infrared->submenu;
    InfraredRemote* remote = infrared->remote;
    InfraredAppState* app_state = &infrared->app_state;

    const char* header = NULL;
    switch(infrared->app_state.edit_mode) {
    case InfraredEditModeRename:
        header = "Rename Button:";
        break;
    case InfraredEditModeDelete:
        header = "Delete Button:";
        break;
    case InfraredEditModeMove:
        header = "Select Button to Move:";
        break;
    case InfraredEditModeMoveSelectDest:
    case InfraredEditModeNone:
    default:
        header = "Move Button Before:";
        break;
    }
    submenu_set_header(submenu, header);

    const size_t button_count = infrared_remote_get_button_count(remote);
    for(size_t i = 0; i < button_count; ++i) {
        InfraredRemoteButton* button = infrared_remote_get_button(remote, i);
        submenu_add_item(
            submenu,
            infrared_remote_button_get_name(button),
            i,
            infrared_scene_edit_button_select_submenu_callback,
            context);
    }
    if(infrared->app_state.edit_mode == InfraredEditModeMoveSelectDest) {
        submenu_add_item(
            submenu,
            "-- Move to the end --",
            button_count,
            infrared_scene_edit_button_select_submenu_callback,
            context);
    }
    if(button_count && app_state->current_button_index != InfraredButtonIndexNone) {
        submenu_set_selected_item(submenu, app_state->current_button_index);
        app_state->current_button_index = InfraredButtonIndexNone;
    }

    view_dispatcher_switch_to_view(infrared->view_dispatcher, InfraredViewSubmenu);
}

bool infrared_scene_edit_button_select_on_event(void* context, SceneManagerEvent event) {
    Infrared* infrared = context;
    InfraredAppState* app_state = &infrared->app_state;
    SceneManager* scene_manager = infrared->scene_manager;
    bool consumed = false;

    if(event.type == SceneManagerEventTypeCustom) {
        app_state->current_button_index = event.event;
        const InfraredEditMode edit_mode = app_state->edit_mode;
        if(edit_mode == InfraredEditModeRename) {
            scene_manager_next_scene(scene_manager, InfraredSceneEditRename);
        } else if(edit_mode == InfraredEditModeDelete) {
            scene_manager_next_scene(scene_manager, InfraredSceneEditDelete);
        } else if(edit_mode == InfraredEditModeMove) {
            app_state->current_button_index_move_orig = event.event;
            app_state->edit_mode = InfraredEditModeMoveSelectDest;
            scene_manager_next_scene(scene_manager, InfraredSceneEditButtonSelect);
        } else if(edit_mode == InfraredEditModeMoveSelectDest) {
            scene_manager_next_scene(scene_manager, InfraredSceneEditMove);
        } else {
            furi_assert(0);
        }
        consumed = true;
    }

    return consumed;
}

void infrared_scene_edit_button_select_on_exit(void* context) {
    Infrared* infrared = context;
    submenu_reset(infrared->submenu);
}