Annotation of src/usr.bin/tmux/menu.c, Revision 1.8
1.8 ! nicm 1: /* $OpenBSD: menu.c,v 1.7 2019/05/26 18:19:52 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.8 ! nicm 83: if (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:
133: static void
134: menu_draw_cb(struct client *c, __unused struct screen_redraw_ctx *ctx0)
135: {
136: struct menu_data *md = c->overlay_data;
137: struct tty *tty = &c->tty;
138: struct screen *s = &md->s;
139: struct menu *menu = md->menu;
140: struct screen_write_ctx ctx;
141: u_int i, px, py;
142:
143: screen_write_start(&ctx, NULL, s);
144: screen_write_clearscreen(&ctx, 8);
145: screen_write_menu(&ctx, menu, md->choice);
146: screen_write_stop(&ctx);
147:
148: px = md->px;
149: py = md->py;
150:
151: for (i = 0; i < screen_size_y(&md->s); i++)
152: tty_draw_line(tty, NULL, s, 0, i, menu->width + 4, px, py + i);
153:
154: if (~md->flags & MENU_NOMOUSE)
155: tty_update_mode(tty, MODE_MOUSE_ALL, NULL);
156: }
157:
158: static void
159: menu_free_cb(struct client *c)
160: {
161: struct menu_data *md = c->overlay_data;
162:
163: if (md->item != NULL)
164: md->item->flags &= ~CMDQ_WAITING;
165:
1.3 nicm 166: if (md->cb != NULL)
167: md->cb(md->menu, UINT_MAX, KEYC_NONE, md->data);
168:
1.1 nicm 169: screen_free(&md->s);
170: menu_free(md->menu);
171: free(md);
172: }
173:
174: static int
175: menu_key_cb(struct client *c, struct key_event *event)
176: {
177: struct menu_data *md = c->overlay_data;
178: struct menu *menu = md->menu;
179: struct mouse_event *m = &event->m;
180: u_int i;
181: int count = menu->count, old = md->choice;
182: const struct menu_item *item;
183: struct cmdq_item *new_item;
1.5 nicm 184: struct cmd_parse_result *pr;
1.1 nicm 185:
186: if (KEYC_IS_MOUSE(event->key)) {
187: if (md->flags & MENU_NOMOUSE)
188: return (0);
189: if (m->x < md->px ||
190: m->x > md->px + 4 + menu->width ||
191: m->y < md->py + 1 ||
192: m->y > md->py + 1 + count - 1) {
193: if (MOUSE_RELEASE(m->b))
194: return (1);
195: if (md->choice != -1) {
196: md->choice = -1;
197: c->flags |= CLIENT_REDRAWOVERLAY;
198: }
199: return (0);
200: }
201: if (MOUSE_RELEASE(m->b))
202: goto chosen;
1.7 nicm 203: md->choice = m->y - (md->py + 1);
1.1 nicm 204: if (md->choice != old)
205: c->flags |= CLIENT_REDRAWOVERLAY;
206: return (0);
207: }
208: switch (event->key) {
209: case KEYC_UP:
210: do {
211: if (md->choice == -1 || md->choice == 0)
212: md->choice = count - 1;
213: else
214: md->choice--;
215: } while (menu->items[md->choice].name == NULL);
216: c->flags |= CLIENT_REDRAWOVERLAY;
217: return (0);
218: case KEYC_DOWN:
219: do {
220: if (md->choice == -1 || md->choice == count - 1)
221: md->choice = 0;
222: else
223: md->choice++;
224: } while (menu->items[md->choice].name == NULL);
225: c->flags |= CLIENT_REDRAWOVERLAY;
226: return (0);
227: case '\r':
228: goto chosen;
229: case '\033': /* Escape */
230: case '\003': /* C-c */
231: case '\007': /* C-g */
232: case 'q':
233: return (1);
234: }
235: for (i = 0; i < (u_int)count; i++) {
236: if (event->key == menu->items[i].key) {
237: md->choice = i;
238: goto chosen;
239: }
240: }
241: return (0);
242:
243: chosen:
244: if (md->choice == -1)
245: return (1);
246: item = &menu->items[md->choice];
247: if (item->name == NULL)
248: return (1);
249: if (md->cb != NULL) {
250: md->cb(md->menu, md->choice, item->key, md->data);
1.3 nicm 251: md->cb = NULL;
1.1 nicm 252: return (1);
253: }
1.5 nicm 254:
255: pr = cmd_parse_from_string(item->command, NULL);
256: switch (pr->status) {
257: case CMD_PARSE_EMPTY:
258: new_item = NULL;
259: break;
260: case CMD_PARSE_ERROR:
261: new_item = cmdq_get_error(pr->error);
262: free(pr->error);
263: cmdq_append(c, new_item);
264: break;
265: case CMD_PARSE_SUCCESS:
1.6 nicm 266: if (md->item != NULL)
267: m = &md->item->shared->mouse;
268: else
269: m = NULL;
270: new_item = cmdq_get_command(pr->cmdlist, &md->fs, m, 0);
1.5 nicm 271: cmd_list_free(pr->cmdlist);
272: cmdq_append(c, new_item);
273: break;
1.1 nicm 274: }
275: return (1);
276: }
277:
278: int
279: menu_display(struct menu *menu, int flags, struct cmdq_item *item, u_int px,
280: u_int py, struct client *c, struct cmd_find_state *fs, menu_choice_cb cb,
281: void *data)
282: {
283: struct menu_data *md;
284:
285: if (c->tty.sx < menu->width + 4 || c->tty.sy < menu->count + 2)
286: return (-1);
287:
288: md = xcalloc(1, sizeof *md);
289: md->item = item;
290: md->flags = flags;
291:
1.3 nicm 292: if (fs != NULL)
293: cmd_find_copy_state(&md->fs, fs);
1.1 nicm 294: screen_init(&md->s, menu->width + 4, menu->count + 2, 0);
295:
296: md->px = px;
297: md->py = py;
298:
299: md->menu = menu;
300: md->choice = -1;
301:
302: md->cb = cb;
303: md->data = data;
304:
305: server_client_set_overlay(c, 0, menu_draw_cb, menu_key_cb, menu_free_cb,
306: md);
307: return (0);
308: }