Momentum-Apps/applications/gui/modules/widget_elements/widget_element_button.c

#include "widget_element_i.h"
#include <gui/elements.h>
#include <m-string.h>

typedef struct {
    GuiButtonType button_type;
    string_t text;
    ButtonCallback callback;
    void* context;
} GuiButtonModel;

static void gui_button_draw(Canvas* canvas, WidgetElement* element) {
    furi_assert(canvas);
    furi_assert(element);
    GuiButtonModel* model = element->model;

    canvas_set_color(canvas, ColorBlack);
    canvas_set_font(canvas, FontSecondary);

    if(model->button_type == GuiButtonTypeLeft) {
        elements_button_left(canvas, string_get_cstr(model->text));
    } else if(model->button_type == GuiButtonTypeRight) {
        elements_button_right(canvas, string_get_cstr(model->text));
    } else if(model->button_type == GuiButtonTypeCenter) {
        elements_button_center(canvas, string_get_cstr(model->text));
    }
}

static bool gui_button_input(InputEvent* event, WidgetElement* element) {
    GuiButtonModel* model = element->model;
    bool consumed = false;

    if(model->callback == NULL) return consumed;

    if(event->key == InputKeyOk && event->type == InputTypePress &&
       model->button_type == GuiButtonTypeCenter) {
        model->callback(GuiButtonTypeCenterPress, model->context);
        consumed = true;
    } else if(
        event->key == InputKeyOk && event->type == InputTypeRelease &&
        model->button_type == GuiButtonTypeCenter) {
        model->callback(GuiButtonTypeCenterRelease, model->context);
        consumed = true;
    } else if(event->type == InputTypeShort) {
        if((model->button_type == GuiButtonTypeLeft) && (event->key == InputKeyLeft)) {
            model->callback(model->button_type, model->context);
            consumed = true;
        } else if((model->button_type == GuiButtonTypeRight) && (event->key == InputKeyRight)) {
            model->callback(model->button_type, model->context);
            consumed = true;
        } else if((model->button_type == GuiButtonTypeCenter) && (event->key == InputKeyOk)) {
            model->callback(model->button_type, model->context);
            consumed = true;
        }
    }

    return consumed;
}

static void gui_button_free(WidgetElement* gui_button) {
    furi_assert(gui_button);

    GuiButtonModel* model = gui_button->model;
    string_clear(model->text);
    free(gui_button->model);
    free(gui_button);
}

WidgetElement* widget_element_button_create(
    GuiButtonType button_type,
    const char* text,
    ButtonCallback callback,
    void* context) {
    // Allocate and init model
    GuiButtonModel* model = furi_alloc(sizeof(GuiButtonModel));
    model->button_type = button_type;
    model->callback = callback;
    model->context = context;
    string_init_set_str(model->text, text);

    // Allocate and init Element
    WidgetElement* gui_button = furi_alloc(sizeof(WidgetElement));
    gui_button->parent = NULL;
    gui_button->input = gui_button_input;
    gui_button->draw = gui_button_draw;
    gui_button->free = gui_button_free;
    gui_button->model = model;

    return gui_button;
}