[BACK]Return to menu.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / tmux

Annotation of src/usr.bin/tmux/menu.c, Revision 1.23

1.23    ! nicm        1: /* $OpenBSD: menu.c,v 1.22 2020/04/15 16:11:23 nicm Exp $ */
1.1       nicm        2:
                      3: /*
                      4:  * Copyright (c) 2019 Nicholas Marriott <nicholas.marriott@gmail.com>
                      5:  *
                      6:  * Permission to use, copy, modify, and distribute this software for any
                      7:  * purpose with or without fee is hereby granted, provided that the above
                      8:  * copyright notice and this permission notice appear in all copies.
                      9:  *
                     10:  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
                     11:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
                     12:  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
                     13:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
                     14:  * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
                     15:  * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
                     16:  * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
                     17:  */
                     18:
                     19: #include <sys/types.h>
                     20:
                     21: #include <stdlib.h>
                     22: #include <string.h>
                     23:
                     24: #include "tmux.h"
                     25:
                     26: struct menu_data {
                     27:        struct cmdq_item        *item;
                     28:        int                      flags;
                     29:
                     30:        struct cmd_find_state    fs;
                     31:        struct screen            s;
                     32:
                     33:        u_int                    px;
                     34:        u_int                    py;
                     35:
                     36:        struct menu             *menu;
                     37:        int                      choice;
                     38:
                     39:        menu_choice_cb           cb;
                     40:        void                    *data;
                     41: };
                     42:
1.8       nicm       43: void
                     44: menu_add_items(struct menu *menu, const struct menu_item *items,
                     45:     struct cmdq_item *qitem, struct client *c, struct cmd_find_state *fs)
                     46: {
                     47:        const struct menu_item  *loop;
                     48:
                     49:        for (loop = items; loop->name != NULL; loop++)
                     50:                menu_add_item(menu, loop, qitem, c, fs);
                     51: }
                     52:
                     53: void
                     54: menu_add_item(struct menu *menu, const struct menu_item *item,
1.6       nicm       55:     struct cmdq_item *qitem, struct client *c, struct cmd_find_state *fs)
1.1       nicm       56: {
                     57:        struct menu_item        *new_item;
1.8       nicm       58:        const char              *key, *cmd;
                     59:        char                    *s, *name;
1.1       nicm       60:        u_int                    width;
1.8       nicm       61:        int                      line;
                     62:
                     63:        line = (item == NULL || item->name == NULL || *item->name == '\0');
                     64:        if (line && menu->count == 0)
                     65:                return;
1.1       nicm       66:
                     67:        menu->items = xreallocarray(menu->items, menu->count + 1,
                     68:            sizeof *menu->items);
                     69:        new_item = &menu->items[menu->count++];
                     70:        memset(new_item, 0, sizeof *new_item);
                     71:
1.8       nicm       72:        if (line)
1.1       nicm       73:                return;
1.8       nicm       74:
                     75:        if (fs != NULL)
                     76:                s = format_single(qitem, item->name, c, fs->s, fs->wl, fs->wp);
                     77:        else
                     78:                s = format_single(qitem, item->name, c, NULL, NULL, NULL);
                     79:        if (*s == '\0') { /* no item if empty after format expanded */
1.1       nicm       80:                menu->count--;
                     81:                return;
                     82:        }
1.9       nicm       83:        if (*s != '-' && item->key != KEYC_UNKNOWN && item->key != KEYC_NONE) {
1.1       nicm       84:                key = key_string_lookup_key(item->key);
1.8       nicm       85:                xasprintf(&name, "%s#[default] #[align=right](%s)", s, key);
1.1       nicm       86:        } else
1.8       nicm       87:                xasprintf(&name, "%s", s);
                     88:        new_item->name = name;
                     89:        free(s);
                     90:
                     91:        cmd = item->command;
                     92:        if (cmd != NULL) {
                     93:                if (fs != NULL)
                     94:                        s = format_single(qitem, cmd, c, fs->s, fs->wl, fs->wp);
                     95:                else
                     96:                        s = format_single(qitem, cmd, c, NULL, NULL, NULL);
1.6       nicm       97:        } else
1.8       nicm       98:                s = NULL;
                     99:        new_item->command = s;
1.1       nicm      100:        new_item->key = item->key;
                    101:
                    102:        width = format_width(new_item->name);
                    103:        if (width > menu->width)
                    104:                menu->width = width;
                    105: }
                    106:
                    107: struct menu *
1.8       nicm      108: menu_create(const char *title)
1.1       nicm      109: {
                    110:        struct menu     *menu;
                    111:
                    112:        menu = xcalloc(1, sizeof *menu);
                    113:        menu->title = xstrdup(title);
                    114:
                    115:        return (menu);
                    116: }
                    117:
                    118: void
                    119: menu_free(struct menu *menu)
                    120: {
                    121:        u_int   i;
                    122:
                    123:        for (i = 0; i < menu->count; i++) {
1.8       nicm      124:                free((void *)menu->items[i].name);
                    125:                free((void *)menu->items[i].command);
1.1       nicm      126:        }
                    127:        free(menu->items);
                    128:
1.8       nicm      129:        free((void *)menu->title);
1.1       nicm      130:        free(menu);
                    131: }
                    132:
1.15      nicm      133: static int
                    134: menu_mode_cb(struct client *c, __unused u_int *cx, __unused u_int *cy)
                    135: {
                    136:        struct menu_data        *md = c->overlay_data;
                    137:
                    138:        if (~md->flags & MENU_NOMOUSE)
                    139:                return (MODE_MOUSE_ALL);
                    140:        return (0);
                    141: }
                    142:
1.1       nicm      143: static void
                    144: menu_draw_cb(struct client *c, __unused struct screen_redraw_ctx *ctx0)
                    145: {
                    146:        struct menu_data        *md = c->overlay_data;
                    147:        struct tty              *tty = &c->tty;
                    148:        struct screen           *s = &md->s;
                    149:        struct menu             *menu = md->menu;
                    150:        struct screen_write_ctx  ctx;
1.13      nicm      151:        u_int                    i, px = md->px, py = md->py;
1.22      nicm      152:        struct grid_cell         gc;
                    153:
                    154:        memcpy(&gc, &grid_default_cell, sizeof gc);
                    155:        style_apply(&gc, c->session->curw->window->options, "mode-style");
1.1       nicm      156:
                    157:        screen_write_start(&ctx, NULL, s);
                    158:        screen_write_clearscreen(&ctx, 8);
1.22      nicm      159:        screen_write_menu(&ctx, menu, md->choice, &gc);
1.1       nicm      160:        screen_write_stop(&ctx);
                    161:
                    162:        for (i = 0; i < screen_size_y(&md->s); i++)
                    163:                tty_draw_line(tty, NULL, s, 0, i, menu->width + 4, px, py + i);
                    164: }
                    165:
                    166: static void
                    167: menu_free_cb(struct client *c)
                    168: {
                    169:        struct menu_data        *md = c->overlay_data;
                    170:
                    171:        if (md->item != NULL)
1.10      nicm      172:                cmdq_continue(md->item);
1.1       nicm      173:
1.3       nicm      174:        if (md->cb != NULL)
                    175:                md->cb(md->menu, UINT_MAX, KEYC_NONE, md->data);
                    176:
1.1       nicm      177:        screen_free(&md->s);
                    178:        menu_free(md->menu);
                    179:        free(md);
                    180: }
                    181:
                    182: static int
                    183: menu_key_cb(struct client *c, struct key_event *event)
                    184: {
                    185:        struct menu_data                *md = c->overlay_data;
                    186:        struct menu                     *menu = md->menu;
                    187:        struct mouse_event              *m = &event->m;
                    188:        u_int                            i;
                    189:        int                              count = menu->count, old = md->choice;
1.21      nicm      190:        const char                      *name;
1.1       nicm      191:        const struct menu_item          *item;
1.21      nicm      192:        struct cmdq_state               *state;
                    193:        enum cmd_parse_status            status;
                    194:        char                            *error;
1.1       nicm      195:
                    196:        if (KEYC_IS_MOUSE(event->key)) {
1.12      nicm      197:                if (md->flags & MENU_NOMOUSE) {
                    198:                        if (MOUSE_BUTTONS(m->b) != 0)
                    199:                                return (1);
1.1       nicm      200:                        return (0);
1.12      nicm      201:                }
1.1       nicm      202:                if (m->x < md->px ||
                    203:                    m->x > md->px + 4 + menu->width ||
                    204:                    m->y < md->py + 1 ||
                    205:                    m->y > md->py + 1 + count - 1) {
                    206:                        if (MOUSE_RELEASE(m->b))
                    207:                                return (1);
                    208:                        if (md->choice != -1) {
                    209:                                md->choice = -1;
                    210:                                c->flags |= CLIENT_REDRAWOVERLAY;
                    211:                        }
                    212:                        return (0);
                    213:                }
                    214:                if (MOUSE_RELEASE(m->b))
                    215:                        goto chosen;
1.7       nicm      216:                md->choice = m->y - (md->py + 1);
1.1       nicm      217:                if (md->choice != old)
                    218:                        c->flags |= CLIENT_REDRAWOVERLAY;
                    219:                return (0);
                    220:        }
1.11      nicm      221:        for (i = 0; i < (u_int)count; i++) {
                    222:                name = menu->items[i].name;
                    223:                if (name == NULL || *name == '-')
                    224:                        continue;
                    225:                if (event->key == menu->items[i].key) {
                    226:                        md->choice = i;
                    227:                        goto chosen;
                    228:                }
                    229:        }
1.1       nicm      230:        switch (event->key) {
                    231:        case KEYC_UP:
1.11      nicm      232:        case 'k':
1.9       nicm      233:                if (old == -1)
                    234:                        old = 0;
1.1       nicm      235:                do {
                    236:                        if (md->choice == -1 || md->choice == 0)
                    237:                                md->choice = count - 1;
                    238:                        else
                    239:                                md->choice--;
1.9       nicm      240:                        name = menu->items[md->choice].name;
                    241:                } while ((name == NULL || *name == '-') && md->choice != old);
1.1       nicm      242:                c->flags |= CLIENT_REDRAWOVERLAY;
                    243:                return (0);
                    244:        case KEYC_DOWN:
1.11      nicm      245:        case 'j':
1.9       nicm      246:                if (old == -1)
                    247:                        old = 0;
1.1       nicm      248:                do {
                    249:                        if (md->choice == -1 || md->choice == count - 1)
                    250:                                md->choice = 0;
1.9       nicm      251:                        else
                    252:                                md->choice++;
                    253:                        name = menu->items[md->choice].name;
                    254:                } while ((name == NULL || *name == '-') && md->choice != old);
1.1       nicm      255:                c->flags |= CLIENT_REDRAWOVERLAY;
                    256:                return (0);
                    257:        case '\r':
                    258:                goto chosen;
                    259:        case '\033': /* Escape */
                    260:        case '\003': /* C-c */
                    261:        case '\007': /* C-g */
                    262:        case 'q':
                    263:                return (1);
                    264:        }
                    265:        return (0);
                    266:
                    267: chosen:
                    268:        if (md->choice == -1)
                    269:                return (1);
                    270:        item = &menu->items[md->choice];
1.9       nicm      271:        if (item->name == NULL || *item->name == '-')
1.1       nicm      272:                return (1);
                    273:        if (md->cb != NULL) {
                    274:            md->cb(md->menu, md->choice, item->key, md->data);
1.3       nicm      275:            md->cb = NULL;
1.1       nicm      276:            return (1);
                    277:        }
1.5       nicm      278:
1.21      nicm      279:        if (md->item != NULL)
                    280:                event = cmdq_get_event(md->item);
                    281:        else
                    282:                event = NULL;
                    283:        state = cmdq_new_state(&md->fs, event, 0);
                    284:
                    285:        status = cmd_parse_and_append(item->command, NULL, c, state, &error);
                    286:        if (status == CMD_PARSE_ERROR) {
                    287:                cmdq_append(c, cmdq_get_error(error));
                    288:                free(error);
1.1       nicm      289:        }
1.21      nicm      290:        cmdq_free_state(state);
                    291:
1.1       nicm      292:        return (1);
                    293: }
                    294:
                    295: int
                    296: menu_display(struct menu *menu, int flags, struct cmdq_item *item, u_int px,
                    297:     u_int py, struct client *c, struct cmd_find_state *fs, menu_choice_cb cb,
                    298:     void *data)
                    299: {
                    300:        struct menu_data        *md;
1.23    ! nicm      301:        u_int                    i;
        !           302:        const char              *name;
1.1       nicm      303:
                    304:        if (c->tty.sx < menu->width + 4 || c->tty.sy < menu->count + 2)
                    305:                return (-1);
1.14      nicm      306:        if (px + menu->width + 4 > c->tty.sx)
                    307:                px = c->tty.sx - menu->width - 4;
                    308:        if (py + menu->count + 2 > c->tty.sy)
                    309:                py = c->tty.sy - menu->count - 2;
1.1       nicm      310:
                    311:        md = xcalloc(1, sizeof *md);
                    312:        md->item = item;
                    313:        md->flags = flags;
                    314:
1.3       nicm      315:        if (fs != NULL)
                    316:                cmd_find_copy_state(&md->fs, fs);
1.1       nicm      317:        screen_init(&md->s, menu->width + 4, menu->count + 2, 0);
                    318:
                    319:        md->px = px;
                    320:        md->py = py;
                    321:
                    322:        md->menu = menu;
1.23    ! nicm      323:        if (md->flags & MENU_NOMOUSE) {
        !           324:                for (i = 0; i < menu->count; i++) {
        !           325:                        name = menu->items[i].name;
        !           326:                        if (name != NULL && *name != '-')
        !           327:                                break;
        !           328:                }
        !           329:                if (i != menu->count)
        !           330:                        md->choice = i;
        !           331:                else
        !           332:                        md->choice = -1;
        !           333:        } else
        !           334:                md->choice = -1;
1.1       nicm      335:
                    336:        md->cb = cb;
                    337:        md->data = data;
                    338:
1.15      nicm      339:        server_client_set_overlay(c, 0, NULL, menu_mode_cb, menu_draw_cb,
                    340:            menu_key_cb, menu_free_cb, md);
1.1       nicm      341:        return (0);
                    342: }