#include "../../js_modules.h" // IWYU pragma: keep
#include "js_gui.h"
#include "../js_event_loop/js_event_loop.h"
#include <gui/modules/button_panel.h>
 
typedef struct {
    size_t matrix_x, matrix_y;
    int32_t next_index;
    char** owned_strings;
    size_t n_owned_strings;
 
    FuriMessageQueue* input_queue;
    JsEventLoopContract contract;
} JsBtnPanelContext;
 
typedef struct {
    int32_t index;
    InputType input_type;
} JsBtnPanelEvent;
 
// not using mlib to conserve code size
static const char* js_button_panel_own_string(JsBtnPanelContext* context, const char* str) {
    char* owned = strdup(str);
    context->n_owned_strings++;
    context->owned_strings =
        realloc(context->owned_strings, context->n_owned_strings * sizeof(const char*));
    context->owned_strings[context->n_owned_strings - 1] = owned;
    return owned;
}
 
static void js_button_panel_free_owned(JsBtnPanelContext* context) {
    for(size_t i = 0; i < context->n_owned_strings; i++) {
        free(context->owned_strings[i]);
    }
    free(context->owned_strings);
    context->owned_strings = NULL;
}
 
static const char* js_input_type_to_str(InputType type) {
    switch(type) {
    case InputTypePress:
        return "press";
    case InputTypeRelease:
        return "release";
    case InputTypeShort:
        return "short";
    case InputTypeLong:
        return "long";
    case InputTypeRepeat:
        return "repeat";
    default:
        furi_crash();
    }
}
 
static mjs_val_t
    input_transformer(struct mjs* mjs, FuriMessageQueue* queue, JsBtnPanelContext* context) {
    UNUSED(context);
    JsBtnPanelEvent event;
    furi_check(furi_message_queue_get(queue, &event, 0) == FuriStatusOk);
 
    mjs_val_t event_obj = mjs_mk_object(mjs);
    JS_ASSIGN_MULTI(mjs, event_obj) {
        JS_FIELD("index", mjs_mk_number(mjs, event.index));
        JS_FIELD("type", mjs_mk_string(mjs, js_input_type_to_str(event.input_type), ~0, false));
    }
 
    return event_obj;
}
 
static void input_callback(void* ctx, int32_t index, InputType type) {
    JsBtnPanelContext* context = ctx;
    JsBtnPanelEvent event = {
        .index = index,
        .input_type = type,
    };
    furi_check(furi_message_queue_put(context->input_queue, &event, 0) == FuriStatusOk);
}
 
static bool matrix_size_x_assign(
    struct mjs* mjs,
    ButtonPanel* panel,
    JsViewPropValue value,
    JsBtnPanelContext* context) {
    UNUSED(mjs);
    context->matrix_x = value.number;
    button_panel_reserve(panel, context->matrix_x, context->matrix_y);
    return true;
}
 
static bool matrix_size_y_assign(
    struct mjs* mjs,
    ButtonPanel* panel,
    JsViewPropValue value,
    JsBtnPanelContext* context) {
    UNUSED(mjs);
    context->matrix_y = value.number;
    button_panel_reserve(panel, context->matrix_x, context->matrix_y);
    return true;
}
 
static bool js_button_panel_add_child(
    struct mjs* mjs,
    ButtonPanel* panel,
    JsBtnPanelContext* context,
    mjs_val_t child_obj) {
    typedef enum {
        JsButtonPanelChildTypeButton,
        JsButtonPanelChildTypeLabel,
        JsButtonPanelChildTypeIcon,
    } JsButtonPanelChildType;
    static const JsValueEnumVariant js_button_panel_child_type_variants[] = {
        {"button", JsButtonPanelChildTypeButton},
        {"label", JsButtonPanelChildTypeLabel},
        {"icon", JsButtonPanelChildTypeIcon},
    };
    static const JsValueDeclaration js_button_panel_child_type =
        JS_VALUE_ENUM(JsButtonPanelChildType, js_button_panel_child_type_variants);
 
    static const JsValueDeclaration js_button_panel_number = JS_VALUE_SIMPLE(JsValueTypeInt32);
    static const JsValueObjectField js_button_panel_common_fields[] = {
        {"type", &js_button_panel_child_type},
        {"x", &js_button_panel_number},
        {"y", &js_button_panel_number},
    };
    static const JsValueDeclaration js_button_panel_common =
        JS_VALUE_OBJECT(js_button_panel_common_fields);
 
    static const JsValueDeclaration js_button_panel_pointer =
        JS_VALUE_SIMPLE(JsValueTypeRawPointer);
    static const JsValueObjectField js_button_panel_button_fields[] = {
        {"matrixX", &js_button_panel_number},
        {"matrixY", &js_button_panel_number},
        {"icon", &js_button_panel_pointer},
        {"iconSelected", &js_button_panel_pointer},
    };
    static const JsValueDeclaration js_button_panel_button =
        JS_VALUE_OBJECT(js_button_panel_button_fields);
 
    static const JsValueDeclaration js_button_panel_string = JS_VALUE_SIMPLE(JsValueTypeString);
    static const JsValueObjectField js_button_panel_label_fields[] = {
        {"text", &js_button_panel_string},
        {"font", &js_gui_font_declaration},
    };
    static const JsValueDeclaration js_button_panel_label =
        JS_VALUE_OBJECT(js_button_panel_label_fields);
 
    static const JsValueObjectField js_button_panel_icon_fields[] = {
        {"icon", &js_button_panel_pointer},
    };
    static const JsValueDeclaration js_button_panel_icon =
        JS_VALUE_OBJECT(js_button_panel_icon_fields);
 
    JsButtonPanelChildType child_type;
    int32_t x, y;
    JsValueParseStatus status;
    JS_VALUE_PARSE(
        mjs,
        JS_VALUE_PARSE_SOURCE_VALUE(&js_button_panel_common),
        JsValueParseFlagReturnOnError,
        &status,
        &child_obj,
        &child_type,
        &x,
        &y);
    if(status != JsValueParseStatusOk) return false;
 
    switch(child_type) {
    case JsButtonPanelChildTypeButton: {
        int32_t matrix_x, matrix_y;
        const Icon *icon, *icon_selected;
        JS_VALUE_PARSE(
            mjs,
            JS_VALUE_PARSE_SOURCE_VALUE(&js_button_panel_button),
            JsValueParseFlagReturnOnError,
            &status,
            &child_obj,
            &matrix_x,
            &matrix_y,
            &icon,
            &icon_selected);
        if(status != JsValueParseStatusOk) return false;
        button_panel_add_item(
            panel,
            context->next_index++,
            matrix_x,
            matrix_y,
            x,
            y,
            icon,
            icon_selected,
            (ButtonItemCallback)input_callback,
            context);
        break;
    }
 
    case JsButtonPanelChildTypeLabel: {
        const char* text;
        Font font;
        JS_VALUE_PARSE(
            mjs,
            JS_VALUE_PARSE_SOURCE_VALUE(&js_button_panel_label),
            JsValueParseFlagReturnOnError,
            &status,
            &child_obj,
            &text,
            &font);
        if(status != JsValueParseStatusOk) return false;
        button_panel_add_label(panel, x, y, font, js_button_panel_own_string(context, text));
        break;
    }
 
    case JsButtonPanelChildTypeIcon: {
        const Icon* icon;
        JS_VALUE_PARSE(
            mjs,
            JS_VALUE_PARSE_SOURCE_VALUE(&js_button_panel_icon),
            JsValueParseFlagReturnOnError,
            &status,
            &child_obj,
            &icon);
        if(status != JsValueParseStatusOk) return false;
        button_panel_add_icon(panel, x, y, icon);
        break;
    }
    }
 
    return true;
}
 
static void js_button_panel_reset_children(ButtonPanel* panel, JsBtnPanelContext* context) {
    context->next_index = 0;
    button_panel_reset(panel);
    button_panel_reserve(panel, context->matrix_x, context->matrix_y);
    js_button_panel_free_owned(context);
}
 
static JsBtnPanelContext* ctx_make(struct mjs* mjs, ButtonPanel* panel, mjs_val_t view_obj) {
    UNUSED(panel);
    JsBtnPanelContext* context = malloc(sizeof(JsBtnPanelContext));
    *context = (JsBtnPanelContext){
        .matrix_x = 1,
        .matrix_y = 1,
        .next_index = 0,
        .owned_strings = NULL,
        .n_owned_strings = 0,
        .input_queue = furi_message_queue_alloc(1, sizeof(JsBtnPanelEvent)),
    };
    context->contract = (JsEventLoopContract){
        .magic = JsForeignMagic_JsEventLoopContract,
        .object_type = JsEventLoopObjectTypeQueue,
        .object = context->input_queue,
        .non_timer =
            {
                .event = FuriEventLoopEventIn,
                .transformer = (JsEventLoopTransformer)input_transformer,
                .transformer_context = context,
            },
    };
    mjs_set(mjs, view_obj, "input", ~0, mjs_mk_foreign(mjs, &context->contract));
    return context;
}
 
static void ctx_destroy(ButtonPanel* input, JsBtnPanelContext* context, FuriEventLoop* loop) {
    UNUSED(input);
    furi_event_loop_maybe_unsubscribe(loop, context->input_queue);
    furi_message_queue_free(context->input_queue);
    js_button_panel_free_owned(context);
    free(context);
}
 
static const JsViewDescriptor view_descriptor = {
    .alloc = (JsViewAlloc)button_panel_alloc,
    .free = (JsViewFree)button_panel_free,
    .get_view = (JsViewGetView)button_panel_get_view,
    .custom_make = (JsViewCustomMake)ctx_make,
    .custom_destroy = (JsViewCustomDestroy)ctx_destroy,
    .add_child = (JsViewAddChild)js_button_panel_add_child,
    .reset_children = (JsViewResetChildren)js_button_panel_reset_children,
    .prop_cnt = 2,
    .props = {
        (JsViewPropDescriptor){
            .name = "matrixSizeX",
            .type = JsViewPropTypeNumber,
            .assign = (JsViewPropAssign)matrix_size_x_assign},
        (JsViewPropDescriptor){
            .name = "matrixSizeY",
            .type = JsViewPropTypeNumber,
            .assign = (JsViewPropAssign)matrix_size_y_assign},
    }};
 
JS_GUI_VIEW_DEF(button_panel, &view_descriptor);