#include "input_event.h"

#include "diskop.h"

void return_from_keyboard_callback(void *ctx)
{
    FlizzerTrackerApp *tracker = (FlizzerTrackerApp *)ctx;

    if (!tracker->is_loading && !tracker->is_saving)
    {
        uint8_t string_length = 0;
        char *string = NULL;

        if (tracker->focus == EDIT_SONGINFO && tracker->mode == PATTERN_VIEW)
        {
            switch (tracker->selected_param)
            {
                case SI_SONGNAME:
                {
                    string_length = MUS_SONG_NAME_LEN;
                    string = (char *)&tracker->song.song_name;
                    break;
                }

                case SI_INSTRUMENTNAME:
                {
                    string_length = MUS_INST_NAME_LEN;
                    string = (char *)&tracker->song.instrument[tracker->current_instrument]->name;
                    break;
                }
            }
        }

        if (tracker->focus == EDIT_INSTRUMENT && tracker->mode == INST_EDITOR_VIEW)
        {
            switch (tracker->selected_param)
            {
                case INST_INSTRUMENTNAME:
                {
                    string_length = MUS_INST_NAME_LEN;
                    string = (char *)&tracker->song.instrument[tracker->current_instrument]->name;
                    break;
                }
            }
        }

        if (string == NULL || string_length == 0)
            return;

        for (uint8_t i = 0; i < string_length; i++) // I tinyfied the font by deleting lowercase chars, and I don't like the lowercase chars of any 3x5 pixels font
        {
            string[i] = toupper(string[i]);
        }
    }

    view_dispatcher_switch_to_view(tracker->view_dispatcher, VIEW_TRACKER);

    if (tracker->is_saving)
    {
        stop_song(tracker);

        tracker->filepath = furi_string_alloc();
        furi_string_cat_printf(tracker->filepath, "%s/%s%s", FLIZZER_TRACKER_FOLDER, tracker->filename, SONG_FILE_EXT);

        if (storage_file_exists(tracker->storage, furi_string_get_cstr(tracker->filepath)))
        {
            view_dispatcher_switch_to_view(tracker->view_dispatcher, VIEW_FILE_OVERWRITE);
            return;
        }

        else
        {
            FlizzerTrackerEvent event = {.type = EventTypeSaveSong, .input = {0}, .period = 0};
            furi_message_queue_put(tracker->event_queue, &event, FuriWaitForever);
        }
    }
}

void overwrite_file_widget_yes_input_callback(GuiButtonType result, InputType type, void *ctx)
{
    UNUSED(result);

    FlizzerTrackerApp *tracker = (FlizzerTrackerApp *)ctx;

    if (type == InputTypeShort)
    {
        tracker->is_saving = true;
        view_dispatcher_switch_to_view(tracker->view_dispatcher, VIEW_TRACKER);
        // save_song(tracker, tracker->filepath);
        static FlizzerTrackerEvent event = {.type = EventTypeSaveSong, .input = {0}, .period = 0};
        furi_message_queue_put(tracker->event_queue, &event, FuriWaitForever);
    }
}

void overwrite_file_widget_no_input_callback(GuiButtonType result, InputType type, void *ctx)
{
    UNUSED(result);

    FlizzerTrackerApp *tracker = (FlizzerTrackerApp *)ctx;

    if (type == InputTypeShort)
    {
        tracker->is_saving = false;
        furi_string_free(tracker->filepath);
        view_dispatcher_switch_to_view(tracker->view_dispatcher, VIEW_TRACKER);
    }
}

void submenu_callback(void *context, uint32_t index)
{
    FlizzerTrackerApp *tracker = (FlizzerTrackerApp *)context;

    switch (tracker->mode)
    {
        case PATTERN_VIEW:
        {
            switch (index)
            {
                case SUBMENU_PATTERN_RETURN:
                {
                    view_dispatcher_switch_to_view(tracker->view_dispatcher, VIEW_TRACKER);
                    break;
                }

                case SUBMENU_PATTERN_EXIT:
                {
                    tracker->quit = true;

                    static InputEvent inevent = {.sequence = 0, .key = InputKeyLeft, .type = InputTypeMAX};
                    FlizzerTrackerEvent event = {.type = EventTypeInput, .input = inevent, .period = 0}; // making an event so tracker does not wait for next keypress and exits immediately
                    furi_message_queue_put(tracker->event_queue, &event, FuriWaitForever);
                    view_dispatcher_switch_to_view(tracker->view_dispatcher, VIEW_TRACKER);
                    break;
                }

                case SUBMENU_PATTERN_SAVE_SONG:
                {
                    text_input_set_header_text(tracker->text_input, "Song filename:");
                    memset(&tracker->filename, 0, FILE_NAME_LEN);
                    text_input_set_result_callback(tracker->text_input, return_from_keyboard_callback, tracker, (char *)&tracker->filename, FILE_NAME_LEN, true);

                    tracker->is_saving = true;

                    view_dispatcher_switch_to_view(tracker->view_dispatcher, VIEW_KEYBOARD);
                    break;
                }

                case SUBMENU_PATTERN_LOAD_SONG:
                {
                    FlizzerTrackerEvent event = {.type = EventTypeLoadSong, .input = {0}, .period = 0};
                    furi_message_queue_put(tracker->event_queue, &event, FuriWaitForever);
                    view_dispatcher_switch_to_view(tracker->view_dispatcher, VIEW_TRACKER);
                    break;
                }

                default:
                    break;
            }

            break;
        }

        case INST_EDITOR_VIEW:
        {
            switch (index)
            {
                case SUBMENU_INSTRUMENT_RETURN:
                {
                    view_dispatcher_switch_to_view(tracker->view_dispatcher, VIEW_TRACKER);
                    break;
                }

                case SUBMENU_INSTRUMENT_EXIT:
                {
                    tracker->quit = true;

                    static InputEvent inevent = {.sequence = 0, .key = InputKeyLeft, .type = InputTypeMAX};
                    FlizzerTrackerEvent event = {.type = EventTypeInput, .input = inevent, .period = 0}; // making an event so tracker does not wait for next keypress and exits immediately
                    furi_message_queue_put(tracker->event_queue, &event, FuriWaitForever);
                    view_dispatcher_switch_to_view(tracker->view_dispatcher, VIEW_TRACKER);
                    break;
                }

                default:
                    break;
            }

            break;
        }

        default:
            break;
    }
}

void cycle_focus(FlizzerTrackerApp *tracker)
{
    switch (tracker->mode)
    {
        case PATTERN_VIEW:
        {
            tracker->focus++;

            if (tracker->focus > EDIT_SONGINFO)
            {
                tracker->focus = EDIT_PATTERN;
            }

            break;
        }

        case INST_EDITOR_VIEW:
        {
            tracker->focus++;

            if (tracker->focus > EDIT_PROGRAM)
            {
                tracker->focus = EDIT_INSTRUMENT;

                if (tracker->current_digit > 1)
                {
                    tracker->current_digit = 1;
                }
            }

            break;
        }

        default:
            break;
    }
}

void cycle_view(FlizzerTrackerApp *tracker)
{
    if (tracker->mode == PATTERN_VIEW)
    {
        tracker->mode = INST_EDITOR_VIEW;
        tracker->focus = EDIT_INSTRUMENT;

        tracker->selected_param = 0;
        tracker->current_digit = 0;

        return;
    }

    if (tracker->mode == INST_EDITOR_VIEW)
    {
        tracker->mode = PATTERN_VIEW;
        tracker->focus = EDIT_PATTERN;

        if (tracker->tracker_engine.song == NULL)
        {
            stop_song(tracker);
            tracker_engine_set_song(&tracker->tracker_engine, &tracker->song);
        }

        tracker->selected_param = 0;
        tracker->current_digit = 0;

        return;
    }
}

void process_input_event(FlizzerTrackerApp *tracker, FlizzerTrackerEvent *event)
{
    if (event->input.key == InputKeyBack && event->input.type == InputTypeShort && event->period > 0 && event->period < 300 && !(tracker->editing))
    {
        cycle_view(tracker);
        return;
    }

    else if (event->input.key == InputKeyBack && event->input.type == InputTypeShort && !(tracker->editing))
    {
        cycle_focus(tracker);
        return;
    }

    // Если нажата кнопка "назад", то выходим из цикла, а следовательно и из приложения
    if (event->input.key == InputKeyBack && event->input.type == InputTypeLong)
    {
        switch (tracker->mode)
        {
            case PATTERN_VIEW:
            {
                submenu_set_selected_item(tracker->pattern_submenu, SUBMENU_PATTERN_RETURN);
                view_dispatcher_switch_to_view(tracker->view_dispatcher, VIEW_SUBMENU_PATTERN);
                break;
            }

            case INST_EDITOR_VIEW:
            {
                submenu_set_selected_item(tracker->instrument_submenu, SUBMENU_INSTRUMENT_RETURN);
                view_dispatcher_switch_to_view(tracker->view_dispatcher, VIEW_SUBMENU_INSTRUMENT);
                break;
            }

            default:
                break;
        }

        return;
    }

    switch (tracker->focus)
    {
        case EDIT_PATTERN:
        {
            pattern_edit_event(tracker, event);
            break;
        }

        case EDIT_SEQUENCE:
        {
            sequence_edit_event(tracker, event);
            break;
        }

        case EDIT_SONGINFO:
        {
            songinfo_edit_event(tracker, event);
            break;
        }

        case EDIT_INSTRUMENT:
        {
            instrument_edit_event(tracker, event);
            break;
        }

        case EDIT_PROGRAM:
        {
            instrument_program_edit_event(tracker, event);
            break;
        }

        default:
            break;
    }
}