#include <furi.h>
#include <gui/gui.h>
#include <gui/view_port.h>
#include <gui/view.h>
#include <gui/view_dispatcher.h>
#include <gui/modules/dialog.h>
#include <gui/modules/dialog_ex.h>
#include <gui/modules/submenu.h>
#include <gui/modules/text_input.h>
#include <gui/modules/byte_input.h>
#include <gui/modules/popup.h>

#define GUI_TEST_FLAG_EXIT 0x00000001U

typedef enum {
    GuiTesterViewTextInput = 0,
    GuiTesterViewSubmenu,
    GuiTesterViewDialog,
    GuiTesterViewDialogEx,
    GuiTesterViewPopup,
    GuiTesterViewByteInput,
    GuiTesterViewLast
} GuiTesterView;

typedef struct {
    ViewDispatcher* view_dispatcher;
    Dialog* dialog;
    DialogEx* dialog_ex;
    Submenu* submenu;
    TextInput* text_input;
    Popup* popup;
    ByteInput* byte_input;
    GuiTesterView view_index;
} GuiTester;

static GuiTester* gui_test_alloc(void) {
    GuiTester* gui_tester = furi_alloc(sizeof(GuiTester));
    gui_tester->view_dispatcher = view_dispatcher_alloc();
    gui_tester->view_index = GuiTesterViewByteInput;

    gui_tester->dialog = dialog_alloc();
    view_dispatcher_add_view(
        gui_tester->view_dispatcher, GuiTesterViewDialog, dialog_get_view(gui_tester->dialog));

    gui_tester->dialog_ex = dialog_ex_alloc();
    view_dispatcher_add_view(
        gui_tester->view_dispatcher,
        GuiTesterViewDialogEx,
        dialog_ex_get_view(gui_tester->dialog_ex));

    gui_tester->submenu = submenu_alloc();
    view_dispatcher_add_view(
        gui_tester->view_dispatcher, GuiTesterViewSubmenu, submenu_get_view(gui_tester->submenu));

    gui_tester->text_input = text_input_alloc();
    view_dispatcher_add_view(
        gui_tester->view_dispatcher,
        GuiTesterViewTextInput,
        text_input_get_view(gui_tester->text_input));

    gui_tester->popup = popup_alloc();
    view_dispatcher_add_view(
        gui_tester->view_dispatcher, GuiTesterViewPopup, popup_get_view(gui_tester->popup));

    gui_tester->byte_input = byte_input_alloc();
    view_dispatcher_add_view(
        gui_tester->view_dispatcher,
        GuiTesterViewByteInput,
        byte_input_get_view(gui_tester->byte_input));

    return gui_tester;
}

static void gui_test_free(GuiTester* gui_tester) {
    furi_assert(gui_tester);
    view_dispatcher_remove_view(gui_tester->view_dispatcher, GuiTesterViewDialog);
    dialog_free(gui_tester->dialog);
    view_dispatcher_remove_view(gui_tester->view_dispatcher, GuiTesterViewDialogEx);
    dialog_ex_free(gui_tester->dialog_ex);
    view_dispatcher_remove_view(gui_tester->view_dispatcher, GuiTesterViewSubmenu);
    submenu_free(gui_tester->submenu);
    view_dispatcher_remove_view(gui_tester->view_dispatcher, GuiTesterViewTextInput);
    text_input_free(gui_tester->text_input);
    view_dispatcher_remove_view(gui_tester->view_dispatcher, GuiTesterViewPopup);
    popup_free(gui_tester->popup);
    view_dispatcher_remove_view(gui_tester->view_dispatcher, GuiTesterViewByteInput);
    byte_input_free(gui_tester->byte_input);

    view_dispatcher_free(gui_tester->view_dispatcher);
    free(gui_tester);
}

static void next_view(void* context) {
    furi_assert(context);
    GuiTester* gui_tester = context;

    gui_tester->view_index++;
    if(gui_tester->view_index >= GuiTesterViewLast) {
        gui_tester->view_index = 0;
    }

    view_dispatcher_switch_to_view(gui_tester->view_dispatcher, gui_tester->view_index);
}

static void popup_callback(void* context) {
    next_view(context);
}

static void submenu_callback(void* context, uint32_t index) {
    next_view(context);
}

static void dialog_callback(DialogResult result, void* context) {
    next_view(context);
}

static void dialog_ex_callback(DialogExResult result, void* context) {
    next_view(context);
}

static void text_input_callback(void* context) {
    next_view(context);
}

static void byte_input_callback(void* context) {
    next_view(context);
}

static void event_cb(const void* value, void* ctx) {
    furi_assert(value);
    furi_assert(ctx);
    const InputEvent* event = value;
    if(event->key == InputKeyBack && event->type == InputTypeLong) {
        osThreadFlagsSet((osThreadId_t)ctx, GUI_TEST_FLAG_EXIT);
    }
}

int32_t gui_test(void* param) {
    (void)param;
    PubSub* event_record = furi_record_open("input_events");
    PubSubItem* event_pubsub = subscribe_pubsub(event_record, event_cb, (void*)osThreadGetId());
    GuiTester* gui_tester = gui_test_alloc();

    Gui* gui = furi_record_open("gui");
    view_dispatcher_attach_to_gui(gui_tester->view_dispatcher, gui, ViewDispatcherTypeFullscreen);

    // Submenu
    submenu_add_item(gui_tester->submenu, "Read", 0, submenu_callback, gui_tester);
    submenu_add_item(gui_tester->submenu, "Saved", 0, submenu_callback, gui_tester);
    submenu_add_item(gui_tester->submenu, "Emulate", 0, submenu_callback, gui_tester);
    submenu_add_item(gui_tester->submenu, "Enter manually", 0, submenu_callback, gui_tester);
    submenu_add_item(gui_tester->submenu, "Blah blah", 0, submenu_callback, gui_tester);
    submenu_add_item(gui_tester->submenu, "Set time", 0, submenu_callback, gui_tester);
    submenu_add_item(gui_tester->submenu, "Gender-bender", 0, submenu_callback, gui_tester);
    submenu_add_item(
        gui_tester->submenu, "Hack American Elections", 0, submenu_callback, gui_tester);
    submenu_add_item(gui_tester->submenu, "Hack the White House", 0, submenu_callback, gui_tester);

    // Dialog
    dialog_set_result_callback(gui_tester->dialog, dialog_callback);
    dialog_set_context(gui_tester->dialog, gui_tester);
    dialog_set_header_text(gui_tester->dialog, "Delete Abc123?");
    dialog_set_text(gui_tester->dialog, "ID: F0 00 01 02 03 04\nAre you shure?");
    dialog_set_left_button_text(gui_tester->dialog, "Yes");
    dialog_set_right_button_text(gui_tester->dialog, "No");

    // Dialog extended
    dialog_ex_set_result_callback(gui_tester->dialog_ex, dialog_ex_callback);
    dialog_ex_set_context(gui_tester->dialog_ex, gui_tester);
    dialog_ex_set_header(gui_tester->dialog_ex, "Dallas", 95, 12, AlignCenter, AlignCenter);
    dialog_ex_set_text(
        gui_tester->dialog_ex, "F6 E5 D4\nC3 B2 A1", 95, 32, AlignCenter, AlignCenter);
    dialog_ex_set_icon(gui_tester->dialog_ex, 0, 1, &I_DolphinExcited_64x63);
    dialog_ex_set_left_button_text(gui_tester->dialog_ex, "More");
    dialog_ex_set_right_button_text(gui_tester->dialog_ex, "Save");

    // Popup
    popup_set_callback(gui_tester->popup, popup_callback);
    popup_set_context(gui_tester->popup, gui_tester);
    popup_set_icon(gui_tester->popup, 0, 2, &I_DolphinMafia_115x62);
    popup_set_text(gui_tester->popup, "Deleted", 83, 19, AlignLeft, AlignBottom);
    popup_set_timeout(gui_tester->popup, 5000);
    popup_enable_timeout(gui_tester->popup);

    // Text input
    const uint8_t text_input_text_len = 64;
    char* text_input_text = calloc(text_input_text_len + 1, 1);
    memcpy(text_input_text, "New_ke", strlen("New_ke"));

    text_input_set_result_callback(
        gui_tester->text_input,
        text_input_callback,
        gui_tester,
        text_input_text,
        text_input_text_len);
    text_input_set_header_text(gui_tester->text_input, "Name the key");

    const uint8_t byte_input_bytes_len = 16;
    uint8_t byte_input_bytes[16] = {
        0x00,
        0x01,
        0x02,
        0x03,
        0x04,
        0x05,
        0x06,
        0x07,
        0x08,
        0x09,
        0x0A,
        0x0B,
        0x0C,
        0x0D,
        0x0E,
        0x0F};

    byte_input_set_result_callback(
        gui_tester->byte_input,
        byte_input_callback,
        NULL,
        gui_tester,
        byte_input_bytes,
        byte_input_bytes_len);
    byte_input_set_header_text(gui_tester->byte_input, "Enter the key");

    view_dispatcher_switch_to_view(gui_tester->view_dispatcher, gui_tester->view_index);

    while(1) {
        if(osThreadFlagsWait(GUI_TEST_FLAG_EXIT, osFlagsWaitAny, osWaitForever)) {
            break;
        }
    }
    unsubscribe_pubsub(event_pubsub);
    free(text_input_text);
    gui_test_free(gui_tester);

    return 0;
}