source: mainline/uspace/lib/ui/src/menubar.c

/*
 * Copyright (c) 2024 Jiri Svoboda
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/** @addtogroup libui
 * @{
 */
/**
 * @file Menu bar
 */

#include <adt/list.h>
#include <ctype.h>
#include <errno.h>
#include <gfx/color.h>
#include <gfx/context.h>
#include <gfx/render.h>
#include <gfx/text.h>
#include <io/pos_event.h>
#include <stdlib.h>
#include <str.h>
#include <ui/control.h>
#include <ui/paint.h>
#include <ui/menubar.h>
#include <ui/menudd.h>
#include <ui/wdecor.h>
#include <ui/window.h>
#include "../private/menubar.h"
#include "../private/resource.h"
#include "../private/wdecor.h"
#include "../private/window.h"

enum {
        menubar_hpad = 4,
        menubar_vpad = 4,
        menubar_hpad_text = 1,
        menubar_vpad_text = 0
};

static void ui_menu_bar_ctl_destroy(void *);
static errno_t ui_menu_bar_ctl_paint(void *);
static ui_evclaim_t ui_menu_bar_ctl_kbd_event(void *, kbd_event_t *);
static ui_evclaim_t ui_menu_bar_ctl_pos_event(void *, pos_event_t *);
static void ui_menu_bar_activate_ev(ui_menu_bar_t *);
static void ui_menu_bar_deactivate_ev(ui_menu_bar_t *);

/** Menu bar control ops */
ui_control_ops_t ui_menu_bar_ops = {
        .destroy = ui_menu_bar_ctl_destroy,
        .paint = ui_menu_bar_ctl_paint,
        .kbd_event = ui_menu_bar_ctl_kbd_event,
        .pos_event = ui_menu_bar_ctl_pos_event
};

/** Create new menu bar.
 *
 * @param ui UI
 * @param window Window that will contain the menu bar
 * @param rmbar Place to store pointer to new menu bar
 * @return EOK on success, ENOMEM if out of memory
 */
errno_t ui_menu_bar_create(ui_t *ui, ui_window_t *window, ui_menu_bar_t **rmbar)
{
        ui_menu_bar_t *mbar;
        errno_t rc;

        mbar = calloc(1, sizeof(ui_menu_bar_t));
        if (mbar == NULL)
                return ENOMEM;

        rc = ui_control_new(&ui_menu_bar_ops, (void *) mbar, &mbar->control);
        if (rc != EOK) {
                free(mbar);
                return rc;
        }

        mbar->ui = ui;
        mbar->window = window;
        list_initialize(&mbar->menudds);

        if (window->mbar == NULL)
                window->mbar = mbar;

        *rmbar = mbar;
        return EOK;
}

/** Destroy menu bar
 *
 * @param mbar Menu bar or @c NULL
 */
void ui_menu_bar_destroy(ui_menu_bar_t *mbar)
{
        ui_menu_dd_t *mdd;

        if (mbar == NULL)
                return;

        if (mbar->window->mbar == mbar)
                mbar->window->mbar = NULL;

        /* Destroy menu drop-downs */
        mdd = ui_menu_dd_first(mbar);
        while (mdd != NULL) {
                ui_menu_dd_destroy(mdd);
                mdd = ui_menu_dd_first(mbar);
        }

        ui_control_delete(mbar->control);
        free(mbar);
}

/** Set menu bar callbacks.
 *
 * @param mbar Menu bar
 * @param cb Callbacks
 * @param arg Callback argument
 */
void ui_menu_bar_set_cb(ui_menu_bar_t *mbar, ui_menu_bar_cb_t *cb, void *arg)
{
        mbar->cb = cb;
        mbar->arg = arg;
}

/** Get base control from menu bar.
 *
 * @param mbar Menu bar
 * @return Control
 */
ui_control_t *ui_menu_bar_ctl(ui_menu_bar_t *mbar)
{
        return mbar->control;
}

/** Set menu bar rectangle.
 *
 * @param mbar Menu bar
 * @param rect New menu bar rectangle
 */
void ui_menu_bar_set_rect(ui_menu_bar_t *mbar, gfx_rect_t *rect)
{
        mbar->rect = *rect;
}

/** Paint menu bar.
 *
 * @param mbar Menu bar
 * @return EOK on success or an error code
 */
errno_t ui_menu_bar_paint(ui_menu_bar_t *mbar)
{
        ui_resource_t *res;
        ui_text_fmt_t fmt;
        gfx_coord2_t pos;
        gfx_coord2_t tpos;
        gfx_rect_t rect;
        gfx_color_t *bg_color;
        ui_menu_dd_t *mdd;
        const char *caption;
        gfx_coord_t width;
        gfx_coord_t hpad;
        gfx_coord_t vpad;
        errno_t rc;

        res = ui_window_get_res(mbar->window);

        /* Paint menu bar background */

        rc = gfx_set_color(res->gc, res->wnd_face_color);
        if (rc != EOK)
                goto error;

        rc = gfx_fill_rect(res->gc, &mbar->rect);
        if (rc != EOK)
                goto error;

        if (res->textmode) {
                hpad = menubar_hpad_text;
                vpad = menubar_vpad_text;
        } else {
                hpad = menubar_hpad;
                vpad = menubar_vpad;
        }

        pos = mbar->rect.p0;

        ui_text_fmt_init(&fmt);
        fmt.font = res->font;
        fmt.halign = gfx_halign_left;
        fmt.valign = gfx_valign_top;

        mdd = ui_menu_dd_first(mbar);
        while (mdd != NULL) {
                caption = ui_menu_dd_caption(mdd);
                width = ui_text_width(res->font, caption) + 2 * hpad;
                tpos.x = pos.x + hpad;
                tpos.y = pos.y + vpad;

                rect.p0 = pos;
                rect.p1.x = rect.p0.x + width;
                rect.p1.y = mbar->rect.p1.y;

                if (mdd == mbar->selected) {
                        fmt.color = res->wnd_sel_text_color;
                        fmt.hgl_color = res->wnd_sel_text_hgl_color;
                        bg_color = res->wnd_sel_text_bg_color;
                } else {
                        fmt.color = res->wnd_text_color;
                        fmt.hgl_color = res->wnd_text_hgl_color;
                        bg_color = res->wnd_face_color;
                }

                rc = gfx_set_color(res->gc, bg_color);
                if (rc != EOK)
                        goto error;

                rc = gfx_fill_rect(res->gc, &rect);
                if (rc != EOK)
                        goto error;

                rc = ui_paint_text(&tpos, &fmt, caption);
                if (rc != EOK)
                        goto error;

                pos.x += width;
                mdd = ui_menu_dd_next(mdd);
        }

        rc = gfx_update(res->gc);
        if (rc != EOK)
                goto error;

        return EOK;
error:
        return rc;
}

/** Select or deselect menu from menu bar.
 *
 * Select @a menu. If @a menu is @c NULL or it is already selected,
 * then select none.
 *
 * @param mbar Menu bar
 * @param mdd Menu drop-down to select (or deselect if selected) or @c NULL
 * @param openup Open menu even if not currently open
 * @param idev_id Input device ID associated with the selecting seat
 */
void ui_menu_bar_select(ui_menu_bar_t *mbar, ui_menu_dd_t *mdd, bool openup,
    sysarg_t idev_id)
{
        ui_menu_dd_t *old_mdd;
        gfx_rect_t rect;
        bool was_open;

        old_mdd = mbar->selected;

        mbar->selected = mdd;

        /* Close previously open menu drop-down */
        if (old_mdd != NULL && ui_menu_dd_is_open(old_mdd)) {
                was_open = true;
                (void) ui_menu_dd_close(old_mdd);
        } else {
                was_open = false;
        }

        (void) ui_menu_bar_paint(mbar);

        if (mbar->selected != NULL) {
                ui_menu_bar_entry_rect(mbar, mbar->selected, &rect);
                if (openup || was_open) {
                        /*
                         * Open the newly selected menu drop-down if either
                         * the old menu drop-down was open or @a openup was
                         * specified.
                         */
                        (void) ui_menu_dd_open(mbar->selected, &rect, idev_id);
                }

                if (!mbar->active)
                        ui_menu_bar_activate_ev(mbar);
                mbar->active = true;
        } else {
                if (mbar->active)
                        ui_menu_bar_deactivate_ev(mbar);
                mbar->active = false;
        }
}

/** Select first drop-down.
 *
 * @param mbar Menu bar
 * @param openup @c true to open drop-down if it was not open
 * @param idev_id Input device ID
 */
void ui_menu_bar_select_first(ui_menu_bar_t *mbar, bool openup,
    sysarg_t idev_id)
{
        ui_menu_dd_t *mdd;

        mdd = ui_menu_dd_first(mbar);
        ui_menu_bar_select(mbar, mdd, openup, idev_id);
}

/** Select last drop-down.
 *
 * @param mbar Menu bar
 * @param openup @c true to open drop-down if it was not open
 * @param idev_id Input device ID
 */
void ui_menu_bar_select_last(ui_menu_bar_t *mbar, bool openup,
    sysarg_t idev_id)
{
        ui_menu_dd_t *mdd;

        mdd = ui_menu_dd_last(mbar);
        ui_menu_bar_select(mbar, mdd, openup, idev_id);
}

/** Select system menu.
 *
 * @param mbar Menu bar
 * @param openup @c true to open drop-down if it was not open
 * @param idev_id Input device ID
 */
void ui_menu_bar_select_sysmenu(ui_menu_bar_t *mbar, bool openup,
    sysarg_t idev_id)
{
        ui_wdecor_sysmenu_hdl_set_active(mbar->window->wdecor, true);

        if (openup)
                ui_window_send_sysmenu(mbar->window, idev_id);
}

/** Move one entry left.
 *
 * If the selected menu is open, the newly selected menu will be open
 * as well. If we are already at the first entry, we wrap around.
 *
 * @param mbar Menu bar
 * @param idev_id Input device ID
 */
void ui_menu_bar_left(ui_menu_bar_t *mbar, sysarg_t idev_id)
{
        ui_menu_dd_t *nmdd;
        bool sel_sysmenu = false;
        bool was_open;

        if (mbar->selected == NULL)
                return;

        nmdd = ui_menu_dd_prev(mbar->selected);
        if (nmdd == NULL) {
                if ((mbar->window->wdecor->style & ui_wds_sysmenu_hdl) != 0) {
                        sel_sysmenu = true;
                } else {
                        nmdd = ui_menu_dd_last(mbar);
                }
        }

        was_open = mbar->selected != NULL &&
            ui_menu_dd_is_open(mbar->selected);

        if (nmdd != mbar->selected)
                ui_menu_bar_select(mbar, nmdd, false, idev_id);

        /*
         * Only open sysmenu *after* closing the previous menu, avoid
         * having multiple popup windows at the same time.
         */
        if (sel_sysmenu)
                ui_menu_bar_select_sysmenu(mbar, was_open, idev_id);
}

/** Move one entry right.
 *
 * If the selected menu is open, the newly selected menu will be open
 * as well. If we are already at the last entry, we wrap around.
 *
 * @param mbar Menu bar
 * @param idev_id Input device ID
 */
void ui_menu_bar_right(ui_menu_bar_t *mbar, sysarg_t idev_id)
{
        ui_menu_dd_t *nmdd;
        bool sel_sysmenu = false;
        bool was_open;

        if (mbar->selected == NULL)
                return;

        nmdd = ui_menu_dd_next(mbar->selected);
        if (nmdd == NULL) {
                if ((mbar->window->wdecor->style & ui_wds_sysmenu_hdl) != 0) {
                        sel_sysmenu = true;
                } else {
                        nmdd = ui_menu_dd_first(mbar);
                }
        }

        was_open = mbar->selected != NULL &&
            ui_menu_dd_is_open(mbar->selected);

        if (nmdd != mbar->selected)
                ui_menu_bar_select(mbar, nmdd, false, idev_id);

        /*
         * Only open sysmenu *after* closing the previous menu, avoid
         * having multiple popup windows at the same time.
         */
        if (sel_sysmenu)
                ui_menu_bar_select_sysmenu(mbar, was_open, idev_id);
}

/** Handle menu bar key press without modifiers.
 *
 * @param mbar Menu bar
 * @param kbd_event Keyboard event
 * @return @c ui_claimed iff the event is claimed
 */
ui_evclaim_t ui_menu_bar_key_press_unmod(ui_menu_bar_t *mbar, kbd_event_t *event)
{
        gfx_rect_t rect;

        if (event->key == KC_F10) {
                ui_menu_bar_activate(mbar);
                return ui_claimed;
        }

        if (!mbar->active)
                return ui_unclaimed;

        if (event->key == KC_ESCAPE) {
                ui_menu_bar_deactivate(mbar);
                return ui_claimed;
        }

        if (event->key == KC_LEFT)
                ui_menu_bar_left(mbar, event->kbd_id);

        if (event->key == KC_RIGHT)
                ui_menu_bar_right(mbar, event->kbd_id);

        if (event->key == KC_ENTER || event->key == KC_DOWN) {
                if (mbar->selected != NULL &&
                    !ui_menu_dd_is_open(mbar->selected)) {
                        ui_menu_bar_entry_rect(mbar, mbar->selected,
                            &rect);
                        ui_menu_dd_open(mbar->selected, &rect, event->kbd_id);
                }

                return ui_claimed;
        }

        if (event->c != '\0' && !ui_menu_dd_is_open(mbar->selected)) {
                /* Check if it is an accelerator. */
                ui_menu_bar_press_accel(mbar, event->c, event->kbd_id);
        }

        return ui_claimed;
}

/** Handle menu bar keyboard event.
 *
 * @param mbar Menu bar
 * @param kbd_event Keyboard event
 * @return @c ui_claimed iff the event is claimed
 */
ui_evclaim_t ui_menu_bar_kbd_event(ui_menu_bar_t *mbar, kbd_event_t *event)
{
        if ((event->mods & KM_ALT) != 0 &&
            (event->mods & (KM_CTRL | KM_SHIFT)) == 0 && event->c != '\0') {
                /* Check if it is an accelerator. */
                ui_menu_bar_press_accel(mbar, event->c, event->kbd_id);
        }

        if (event->type == KEY_PRESS && (event->mods &
            (KM_CTRL | KM_ALT | KM_SHIFT)) == 0) {
                return ui_menu_bar_key_press_unmod(mbar, event);
        }

        if (mbar->active)
                return ui_claimed;

        return ui_unclaimed;
}

/** Accelerator key press.
 *
 * If @a c matches an accelerator key, open the respective menu.
 *
 * @param mbar Menu bar
 * @param c Character
 * @param kbd_id Keyboard ID
 */
void ui_menu_bar_press_accel(ui_menu_bar_t *mbar, char32_t c, sysarg_t kbd_id)
{
        ui_menu_dd_t *mdd;
        char32_t maccel;

        mdd = ui_menu_dd_first(mbar);
        while (mdd != NULL) {
                maccel = ui_menu_dd_get_accel(mdd);
                if ((char32_t)tolower(c) == maccel) {
                        ui_menu_bar_select(mbar, mdd, true, kbd_id);
                        return;
                }

                mdd = ui_menu_dd_next(mdd);
        }
}

/** Handle menu bar position event.
 *
 * @param mbar Menu bar
 * @param pos_event Position event
 * @return @c ui_claimed iff the event is claimed
 */
ui_evclaim_t ui_menu_bar_pos_event(ui_menu_bar_t *mbar, pos_event_t *event)
{
        ui_resource_t *res;
        gfx_coord2_t pos;
        gfx_rect_t rect;
        ui_menu_dd_t *mdd;
        const char *caption;
        gfx_coord_t width;
        gfx_coord_t hpad;
        gfx_coord2_t ppos;
        sysarg_t pos_id;

        res = ui_window_get_res(mbar->window);

        ppos.x = event->hpos;
        ppos.y = event->vpos;

        if (res->textmode) {
                hpad = menubar_hpad_text;
        } else {
                hpad = menubar_hpad;
        }

        pos = mbar->rect.p0;
        pos_id = event->pos_id;

        mdd = ui_menu_dd_first(mbar);
        while (mdd != NULL) {
                caption = ui_menu_dd_caption(mdd);
                width = ui_text_width(res->font, caption) + 2 * hpad;

                rect.p0 = pos;
                rect.p1.x = rect.p0.x + width;
                rect.p1.y = mbar->rect.p1.y;

                /* Check if press is inside menu bar entry */
                if (event->type == POS_PRESS &&
                    gfx_pix_inside_rect(&ppos, &rect)) {
                        mbar->active = true;

                        /* Open the menu, if not already open. */
                        if (mdd != mbar->selected)
                                ui_menu_bar_select(mbar, mdd, true, pos_id);

                        return ui_claimed;
                }

                pos.x += width;
                mdd = ui_menu_dd_next(mdd);
        }

        return ui_unclaimed;
}

/** Handle menu bar position event.
 *
 * @param mbar Menu bar
 * @param mdd Menu drop-down whose entry's rectangle is to be returned
 * @param rrect Place to store entry rectangle
 */
void ui_menu_bar_entry_rect(ui_menu_bar_t *mbar, ui_menu_dd_t *mdd,
    gfx_rect_t *rrect)
{
        ui_resource_t *res;
        gfx_coord2_t pos;
        gfx_rect_t rect;
        ui_menu_dd_t *cur;
        const char *caption;
        gfx_coord_t width;
        gfx_coord_t hpad;

        res = ui_window_get_res(mbar->window);

        if (res->textmode) {
                hpad = menubar_hpad_text;
        } else {
                hpad = menubar_hpad;
        }

        pos = mbar->rect.p0;

        cur = ui_menu_dd_first(mbar);
        while (cur != NULL) {
                caption = ui_menu_dd_caption(cur);
                width = ui_text_width(res->font, caption) + 2 * hpad;

                rect.p0 = pos;
                rect.p1.x = rect.p0.x + width;
                rect.p1.y = mbar->rect.p1.y;

                if (cur == mdd) {
                        *rrect = rect;
                        return;
                }

                pos.x += width;
                cur = ui_menu_dd_next(cur);
        }

        /* We should never get here */
        assert(false);
}

/** Activate menu bar.
 *
 * @param mbar Menu bar
 */
void ui_menu_bar_activate(ui_menu_bar_t *mbar)
{
        if (mbar->active)
                return;

        mbar->active = true;
        if (mbar->selected == NULL)
                mbar->selected = ui_menu_dd_first(mbar);

        (void) ui_menu_bar_paint(mbar);
        ui_menu_bar_activate_ev(mbar);
}

/** Deactivate menu bar.
 *
 * @param mbar Menu bar
 */
void ui_menu_bar_deactivate(ui_menu_bar_t *mbar)
{
        ui_menu_bar_select(mbar, NULL, false, 0);
        ui_menu_bar_deactivate_ev(mbar);
}

/** Destroy menu bar control.
 *
 * @param arg Argument (ui_menu_bar_t *)
 */
void ui_menu_bar_ctl_destroy(void *arg)
{
        ui_menu_bar_t *mbar = (ui_menu_bar_t *) arg;

        ui_menu_bar_destroy(mbar);
}

/** Paint menu bar control.
 *
 * @param arg Argument (ui_menu_bar_t *)
 * @return EOK on success or an error code
 */
errno_t ui_menu_bar_ctl_paint(void *arg)
{
        ui_menu_bar_t *mbar = (ui_menu_bar_t *) arg;

        return ui_menu_bar_paint(mbar);
}

/** Handle menu bar control keyboard event.
 *
 * @param arg Argument (ui_menu_bar_t *)
 * @param pos_event Position event
 * @return @c ui_claimed iff the event is claimed
 */
ui_evclaim_t ui_menu_bar_ctl_kbd_event(void *arg, kbd_event_t *event)
{
        ui_menu_bar_t *mbar = (ui_menu_bar_t *) arg;

        return ui_menu_bar_kbd_event(mbar, event);
}

/** Handle menu bar control position event.
 *
 * @param arg Argument (ui_menu_bar_t *)
 * @param pos_event Position event
 * @return @c ui_claimed iff the event is claimed
 */
ui_evclaim_t ui_menu_bar_ctl_pos_event(void *arg, pos_event_t *event)
{
        ui_menu_bar_t *mbar = (ui_menu_bar_t *) arg;

        return ui_menu_bar_pos_event(mbar, event);
}

/** Send menu bar activate event.
 *
 * @param mbar Menu bar
 */
static void ui_menu_bar_activate_ev(ui_menu_bar_t *mbar)
{
        if (mbar->cb != NULL && mbar->cb->activate != NULL)
                mbar->cb->activate(mbar, mbar->arg);
}

/** Send menu bar deactivate event.
 *
 * @param mbar Menu bar
 */
static void ui_menu_bar_deactivate_ev(ui_menu_bar_t *mbar)
{
        if (mbar->cb != NULL && mbar->cb->deactivate != NULL)
                mbar->cb->deactivate(mbar, mbar->arg);
}

/** @}
 */