Annotation of src/usr.bin/tmux/menu.c, Revision 1.32
1.32 ! nicm 1: /* $OpenBSD: menu.c,v 1.31 2020/09/16 18:37:55 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)
1.24 nicm 76: s = format_single_from_state(qitem, item->name, c, fs);
1.8 nicm 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.29 nicm 84: key = key_string_lookup_key(item->key, 0);
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)
1.24 nicm 94: s = format_single_from_state(qitem, cmd, c, fs);
1.8 nicm 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);
1.30 nicm 114: menu->width = format_width(title);
1.1 nicm 115:
116: return (menu);
117: }
118:
119: void
120: menu_free(struct menu *menu)
121: {
122: u_int i;
123:
124: for (i = 0; i < menu->count; i++) {
1.8 nicm 125: free((void *)menu->items[i].name);
126: free((void *)menu->items[i].command);
1.1 nicm 127: }
128: free(menu->items);
129:
1.8 nicm 130: free((void *)menu->title);
1.1 nicm 131: free(menu);
132: }
133:
1.27 nicm 134: static struct screen *
1.15 nicm 135: menu_mode_cb(struct client *c, __unused u_int *cx, __unused u_int *cy)
136: {
137: struct menu_data *md = c->overlay_data;
138:
1.27 nicm 139: return (&md->s);
1.15 nicm 140: }
141:
1.1 nicm 142: static void
143: menu_draw_cb(struct client *c, __unused struct screen_redraw_ctx *ctx0)
144: {
145: struct menu_data *md = c->overlay_data;
146: struct tty *tty = &c->tty;
147: struct screen *s = &md->s;
148: struct menu *menu = md->menu;
149: struct screen_write_ctx ctx;
1.13 nicm 150: u_int i, px = md->px, py = md->py;
1.22 nicm 151: struct grid_cell gc;
152:
1.25 nicm 153: style_apply(&gc, c->session->curw->window->options, "mode-style", NULL);
1.1 nicm 154:
1.27 nicm 155: screen_write_start(&ctx, s);
1.1 nicm 156: screen_write_clearscreen(&ctx, 8);
1.22 nicm 157: screen_write_menu(&ctx, menu, md->choice, &gc);
1.1 nicm 158: screen_write_stop(&ctx);
159:
1.27 nicm 160: for (i = 0; i < screen_size_y(&md->s); i++) {
161: tty_draw_line(tty, s, 0, i, menu->width + 4, px, py + i,
162: &grid_default_cell, NULL);
163: }
1.1 nicm 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.31 nicm 190: const char *name = NULL;
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) {
1.32 ! nicm 206: if (~md->flags & MENU_STAYOPEN) {
! 207: if (MOUSE_RELEASE(m->b))
! 208: return (1);
! 209: } else {
! 210: if (!MOUSE_RELEASE(m->b) &&
! 211: MOUSE_WHEEL(m->b) == 0 &&
! 212: !MOUSE_DRAG(m->b))
! 213: return (1);
! 214: }
1.1 nicm 215: if (md->choice != -1) {
216: md->choice = -1;
217: c->flags |= CLIENT_REDRAWOVERLAY;
218: }
219: return (0);
220: }
1.32 ! nicm 221: if (~md->flags & MENU_STAYOPEN) {
! 222: if (MOUSE_RELEASE(m->b))
! 223: goto chosen;
! 224: } else {
! 225: if (MOUSE_WHEEL(m->b) == 0 && !MOUSE_DRAG(m->b))
! 226: goto chosen;
! 227: }
1.7 nicm 228: md->choice = m->y - (md->py + 1);
1.1 nicm 229: if (md->choice != old)
230: c->flags |= CLIENT_REDRAWOVERLAY;
231: return (0);
232: }
1.11 nicm 233: for (i = 0; i < (u_int)count; i++) {
234: name = menu->items[i].name;
235: if (name == NULL || *name == '-')
236: continue;
237: if (event->key == menu->items[i].key) {
238: md->choice = i;
239: goto chosen;
240: }
241: }
1.29 nicm 242: switch (event->key & ~KEYC_MASK_FLAGS) {
1.1 nicm 243: case KEYC_UP:
1.11 nicm 244: case 'k':
1.9 nicm 245: if (old == -1)
246: old = 0;
1.1 nicm 247: do {
248: if (md->choice == -1 || md->choice == 0)
249: md->choice = count - 1;
250: else
251: md->choice--;
1.9 nicm 252: name = menu->items[md->choice].name;
253: } while ((name == NULL || *name == '-') && md->choice != old);
1.1 nicm 254: c->flags |= CLIENT_REDRAWOVERLAY;
255: return (0);
1.26 nicm 256: case KEYC_BSPACE:
257: if (~md->flags & MENU_TAB)
258: break;
259: return (1);
260: case '\011': /* Tab */
261: if (~md->flags & MENU_TAB)
262: break;
263: if (md->choice == count - 1)
264: return (1);
265: /* FALLTHROUGH */
1.1 nicm 266: case KEYC_DOWN:
1.11 nicm 267: case 'j':
1.9 nicm 268: if (old == -1)
269: old = 0;
1.1 nicm 270: do {
271: if (md->choice == -1 || md->choice == count - 1)
272: md->choice = 0;
1.9 nicm 273: else
274: md->choice++;
275: name = menu->items[md->choice].name;
276: } while ((name == NULL || *name == '-') && md->choice != old);
1.1 nicm 277: c->flags |= CLIENT_REDRAWOVERLAY;
278: return (0);
1.26 nicm 279: case 'g':
280: case KEYC_PPAGE:
281: case '\002': /* C-b */
282: if (md->choice > 5)
283: md->choice -= 5;
284: else
285: md->choice = 0;
286: while (md->choice != count && (name == NULL || *name == '-'))
287: md->choice++;
288: if (md->choice == count)
289: md->choice = -1;
290: c->flags |= CLIENT_REDRAWOVERLAY;
291: break;
292: case 'G':
293: case KEYC_NPAGE:
294: if (md->choice > count - 6)
295: md->choice = count - 1;
296: else
297: md->choice += 5;
298: while (md->choice != -1 && (name == NULL || *name == '-'))
299: md->choice--;
300: c->flags |= CLIENT_REDRAWOVERLAY;
301: break;
302: case '\006': /* C-f */
303: break;
1.1 nicm 304: case '\r':
305: goto chosen;
306: case '\033': /* Escape */
307: case '\003': /* C-c */
308: case '\007': /* C-g */
309: case 'q':
310: return (1);
311: }
312: return (0);
313:
314: chosen:
315: if (md->choice == -1)
316: return (1);
317: item = &menu->items[md->choice];
1.32 ! nicm 318: if ((md->flags & MENU_STAYOPEN) && item->name == NULL)
! 319: return (0);
1.9 nicm 320: if (item->name == NULL || *item->name == '-')
1.1 nicm 321: return (1);
322: if (md->cb != NULL) {
323: md->cb(md->menu, md->choice, item->key, md->data);
1.3 nicm 324: md->cb = NULL;
1.1 nicm 325: return (1);
326: }
1.5 nicm 327:
1.21 nicm 328: if (md->item != NULL)
329: event = cmdq_get_event(md->item);
330: else
331: event = NULL;
332: state = cmdq_new_state(&md->fs, event, 0);
333:
334: status = cmd_parse_and_append(item->command, NULL, c, state, &error);
335: if (status == CMD_PARSE_ERROR) {
336: cmdq_append(c, cmdq_get_error(error));
337: free(error);
1.1 nicm 338: }
1.21 nicm 339: cmdq_free_state(state);
340:
1.1 nicm 341: return (1);
342: }
343:
344: int
345: menu_display(struct menu *menu, int flags, struct cmdq_item *item, u_int px,
346: u_int py, struct client *c, struct cmd_find_state *fs, menu_choice_cb cb,
347: void *data)
348: {
349: struct menu_data *md;
1.23 nicm 350: u_int i;
351: const char *name;
1.1 nicm 352:
353: if (c->tty.sx < menu->width + 4 || c->tty.sy < menu->count + 2)
354: return (-1);
1.14 nicm 355: if (px + menu->width + 4 > c->tty.sx)
356: px = c->tty.sx - menu->width - 4;
357: if (py + menu->count + 2 > c->tty.sy)
358: py = c->tty.sy - menu->count - 2;
1.1 nicm 359:
360: md = xcalloc(1, sizeof *md);
361: md->item = item;
362: md->flags = flags;
363:
1.3 nicm 364: if (fs != NULL)
365: cmd_find_copy_state(&md->fs, fs);
1.1 nicm 366: screen_init(&md->s, menu->width + 4, menu->count + 2, 0);
1.27 nicm 367: if (~md->flags & MENU_NOMOUSE)
368: md->s.mode |= MODE_MOUSE_ALL;
1.28 nicm 369: md->s.mode &= ~MODE_CURSOR;
1.1 nicm 370:
371: md->px = px;
372: md->py = py;
373:
374: md->menu = menu;
1.23 nicm 375: if (md->flags & MENU_NOMOUSE) {
376: for (i = 0; i < menu->count; i++) {
377: name = menu->items[i].name;
378: if (name != NULL && *name != '-')
379: break;
380: }
381: if (i != menu->count)
382: md->choice = i;
383: else
384: md->choice = -1;
385: } else
386: md->choice = -1;
1.1 nicm 387:
388: md->cb = cb;
389: md->data = data;
390:
1.15 nicm 391: server_client_set_overlay(c, 0, NULL, menu_mode_cb, menu_draw_cb,
392: menu_key_cb, menu_free_cb, md);
1.1 nicm 393: return (0);
394: }