Unescape the string for the literal operator (l:) so special characters
[tmux-openbsd.git] / menu.c
blob4aad1d8cfe65de52d1236e1135a18c0c8d466d29
1 /* $OpenBSD$ */
3 /*
4 * Copyright (c) 2019 Nicholas Marriott <nicholas.marriott@gmail.com>
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.
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.
19 #include <sys/types.h>
21 #include <stdlib.h>
22 #include <string.h>
24 #include "tmux.h"
26 struct menu_data {
27 struct cmdq_item *item;
28 int flags;
30 struct cmd_find_state fs;
31 struct screen s;
33 u_int px;
34 u_int py;
36 struct menu *menu;
37 int choice;
39 menu_choice_cb cb;
40 void *data;
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)
47 const struct menu_item *loop;
49 for (loop = items; loop->name != NULL; loop++)
50 menu_add_item(menu, loop, qitem, c, fs);
53 void
54 menu_add_item(struct menu *menu, const struct menu_item *item,
55 struct cmdq_item *qitem, struct client *c, struct cmd_find_state *fs)
57 struct menu_item *new_item;
58 const char *key = NULL, *cmd, *suffix = "";
59 char *s, *trimmed, *name;
60 u_int width, max_width;
61 int line;
62 size_t keylen, slen;
64 line = (item == NULL || item->name == NULL || *item->name == '\0');
65 if (line && menu->count == 0)
66 return;
68 menu->items = xreallocarray(menu->items, menu->count + 1,
69 sizeof *menu->items);
70 new_item = &menu->items[menu->count++];
71 memset(new_item, 0, sizeof *new_item);
73 if (line)
74 return;
76 if (fs != NULL)
77 s = format_single_from_state(qitem, item->name, c, fs);
78 else
79 s = format_single(qitem, item->name, c, NULL, NULL, NULL);
80 if (*s == '\0') { /* no item if empty after format expanded */
81 menu->count--;
82 return;
84 max_width = c->tty.sx - 4;
86 slen = strlen(s);
87 if (*s != '-' && item->key != KEYC_UNKNOWN && item->key != KEYC_NONE) {
88 key = key_string_lookup_key(item->key, 0);
89 keylen = strlen(key) + 3; /* 3 = space and two brackets */
92 * Add the key if it is shorter than a quarter of the available
93 * space or there is space for the entire item text and the
94 * key.
96 if (keylen <= max_width / 4)
97 max_width -= keylen;
98 else if (keylen >= max_width || slen >= max_width - keylen)
99 key = NULL;
102 if (slen > max_width) {
103 max_width--;
104 suffix = ">";
106 trimmed = format_trim_right(s, max_width);
107 if (key != NULL) {
108 xasprintf(&name, "%s%s#[default] #[align=right](%s)",
109 trimmed, suffix, key);
110 } else
111 xasprintf(&name, "%s%s", trimmed, suffix);
112 free(trimmed);
114 new_item->name = name;
115 free(s);
117 cmd = item->command;
118 if (cmd != NULL) {
119 if (fs != NULL)
120 s = format_single_from_state(qitem, cmd, c, fs);
121 else
122 s = format_single(qitem, cmd, c, NULL, NULL, NULL);
123 } else
124 s = NULL;
125 new_item->command = s;
126 new_item->key = item->key;
128 width = format_width(new_item->name);
129 if (*new_item->name == '-')
130 width--;
131 if (width > menu->width)
132 menu->width = width;
135 struct menu *
136 menu_create(const char *title)
138 struct menu *menu;
140 menu = xcalloc(1, sizeof *menu);
141 menu->title = xstrdup(title);
142 menu->width = format_width(title);
144 return (menu);
147 void
148 menu_free(struct menu *menu)
150 u_int i;
152 for (i = 0; i < menu->count; i++) {
153 free((void *)menu->items[i].name);
154 free((void *)menu->items[i].command);
156 free(menu->items);
158 free((void *)menu->title);
159 free(menu);
162 struct screen *
163 menu_mode_cb(__unused struct client *c, void *data, u_int *cx, u_int *cy)
165 struct menu_data *md = data;
167 *cx = md->px + 2;
168 if (md->choice == -1)
169 *cy = md->py;
170 else
171 *cy = md->py + 1 + md->choice;
173 return (&md->s);
176 /* Return parts of the input range which are not obstructed by the menu. */
177 void
178 menu_check_cb(__unused struct client *c, void *data, u_int px, u_int py,
179 u_int nx, struct overlay_ranges *r)
181 struct menu_data *md = data;
182 struct menu *menu = md->menu;
184 server_client_overlay_range(md->px, md->py, menu->width + 4,
185 menu->count + 2, px, py, nx, r);
188 void
189 menu_draw_cb(struct client *c, void *data,
190 __unused struct screen_redraw_ctx *rctx)
192 struct menu_data *md = data;
193 struct tty *tty = &c->tty;
194 struct screen *s = &md->s;
195 struct menu *menu = md->menu;
196 struct screen_write_ctx ctx;
197 u_int i, px = md->px, py = md->py;
198 struct grid_cell gc;
200 style_apply(&gc, c->session->curw->window->options, "mode-style", NULL);
202 screen_write_start(&ctx, s);
203 screen_write_clearscreen(&ctx, 8);
204 screen_write_menu(&ctx, menu, md->choice, &gc);
205 screen_write_stop(&ctx);
207 for (i = 0; i < screen_size_y(&md->s); i++) {
208 tty_draw_line(tty, s, 0, i, menu->width + 4, px, py + i,
209 &grid_default_cell, NULL);
213 void
214 menu_free_cb(__unused struct client *c, void *data)
216 struct menu_data *md = data;
218 if (md->item != NULL)
219 cmdq_continue(md->item);
221 if (md->cb != NULL)
222 md->cb(md->menu, UINT_MAX, KEYC_NONE, md->data);
224 screen_free(&md->s);
225 menu_free(md->menu);
226 free(md);
230 menu_key_cb(struct client *c, void *data, struct key_event *event)
232 struct menu_data *md = data;
233 struct menu *menu = md->menu;
234 struct mouse_event *m = &event->m;
235 u_int i;
236 int count = menu->count, old = md->choice;
237 const char *name = NULL;
238 const struct menu_item *item;
239 struct cmdq_state *state;
240 enum cmd_parse_status status;
241 char *error;
243 if (KEYC_IS_MOUSE(event->key)) {
244 if (md->flags & MENU_NOMOUSE) {
245 if (MOUSE_BUTTONS(m->b) != MOUSE_BUTTON_1)
246 return (1);
247 return (0);
249 if (m->x < md->px ||
250 m->x > md->px + 4 + menu->width ||
251 m->y < md->py + 1 ||
252 m->y > md->py + 1 + count - 1) {
253 if (~md->flags & MENU_STAYOPEN) {
254 if (MOUSE_RELEASE(m->b))
255 return (1);
256 } else {
257 if (!MOUSE_RELEASE(m->b) &&
258 !MOUSE_WHEEL(m->b) &&
259 !MOUSE_DRAG(m->b))
260 return (1);
262 if (md->choice != -1) {
263 md->choice = -1;
264 c->flags |= CLIENT_REDRAWOVERLAY;
266 return (0);
268 if (~md->flags & MENU_STAYOPEN) {
269 if (MOUSE_RELEASE(m->b))
270 goto chosen;
271 } else {
272 if (!MOUSE_WHEEL(m->b) && !MOUSE_DRAG(m->b))
273 goto chosen;
275 md->choice = m->y - (md->py + 1);
276 if (md->choice != old)
277 c->flags |= CLIENT_REDRAWOVERLAY;
278 return (0);
280 for (i = 0; i < (u_int)count; i++) {
281 name = menu->items[i].name;
282 if (name == NULL || *name == '-')
283 continue;
284 if (event->key == menu->items[i].key) {
285 md->choice = i;
286 goto chosen;
289 switch (event->key & ~KEYC_MASK_FLAGS) {
290 case KEYC_UP:
291 case 'k':
292 if (old == -1)
293 old = 0;
294 do {
295 if (md->choice == -1 || md->choice == 0)
296 md->choice = count - 1;
297 else
298 md->choice--;
299 name = menu->items[md->choice].name;
300 } while ((name == NULL || *name == '-') && md->choice != old);
301 c->flags |= CLIENT_REDRAWOVERLAY;
302 return (0);
303 case KEYC_BSPACE:
304 if (~md->flags & MENU_TAB)
305 break;
306 return (1);
307 case '\011': /* Tab */
308 if (~md->flags & MENU_TAB)
309 break;
310 if (md->choice == count - 1)
311 return (1);
312 /* FALLTHROUGH */
313 case KEYC_DOWN:
314 case 'j':
315 if (old == -1)
316 old = 0;
317 do {
318 if (md->choice == -1 || md->choice == count - 1)
319 md->choice = 0;
320 else
321 md->choice++;
322 name = menu->items[md->choice].name;
323 } while ((name == NULL || *name == '-') && md->choice != old);
324 c->flags |= CLIENT_REDRAWOVERLAY;
325 return (0);
326 case KEYC_PPAGE:
327 case '\002': /* C-b */
328 if (md->choice < 6)
329 md->choice = 0;
330 else {
331 i = 5;
332 while (i > 0) {
333 md->choice--;
334 name = menu->items[md->choice].name;
335 if (md->choice != 0 &&
336 (name != NULL && *name != '-'))
337 i--;
338 else if (md->choice == 0)
339 break;
342 c->flags |= CLIENT_REDRAWOVERLAY;
343 break;
344 case KEYC_NPAGE:
345 if (md->choice > count - 6) {
346 md->choice = count - 1;
347 name = menu->items[md->choice].name;
348 } else {
349 i = 5;
350 while (i > 0) {
351 md->choice++;
352 name = menu->items[md->choice].name;
353 if (md->choice != count - 1 &&
354 (name != NULL && *name != '-'))
355 i++;
356 else if (md->choice == count - 1)
357 break;
360 while (name == NULL || *name == '-') {
361 md->choice--;
362 name = menu->items[md->choice].name;
364 c->flags |= CLIENT_REDRAWOVERLAY;
365 break;
366 case 'g':
367 case KEYC_HOME:
368 md->choice = 0;
369 name = menu->items[md->choice].name;
370 while (name == NULL || *name == '-') {
371 md->choice++;
372 name = menu->items[md->choice].name;
374 c->flags |= CLIENT_REDRAWOVERLAY;
375 break;
376 case 'G':
377 case KEYC_END:
378 md->choice = count - 1;
379 name = menu->items[md->choice].name;
380 while (name == NULL || *name == '-') {
381 md->choice--;
382 name = menu->items[md->choice].name;
384 c->flags |= CLIENT_REDRAWOVERLAY;
385 break;
386 case '\006': /* C-f */
387 break;
388 case '\r':
389 goto chosen;
390 case '\033': /* Escape */
391 case '\003': /* C-c */
392 case '\007': /* C-g */
393 case 'q':
394 return (1);
396 return (0);
398 chosen:
399 if (md->choice == -1)
400 return (1);
401 item = &menu->items[md->choice];
402 if (item->name == NULL || *item->name == '-') {
403 if (md->flags & MENU_STAYOPEN)
404 return (0);
405 return (1);
407 if (md->cb != NULL) {
408 md->cb(md->menu, md->choice, item->key, md->data);
409 md->cb = NULL;
410 return (1);
413 if (md->item != NULL)
414 event = cmdq_get_event(md->item);
415 else
416 event = NULL;
417 state = cmdq_new_state(&md->fs, event, 0);
419 status = cmd_parse_and_append(item->command, NULL, c, state, &error);
420 if (status == CMD_PARSE_ERROR) {
421 cmdq_append(c, cmdq_get_error(error));
422 free(error);
424 cmdq_free_state(state);
426 return (1);
429 struct menu_data *
430 menu_prepare(struct menu *menu, int flags, struct cmdq_item *item, u_int px,
431 u_int py, struct client *c, struct cmd_find_state *fs, menu_choice_cb cb,
432 void *data)
434 struct menu_data *md;
435 u_int i;
436 const char *name;
438 if (c->tty.sx < menu->width + 4 || c->tty.sy < menu->count + 2)
439 return (NULL);
440 if (px + menu->width + 4 > c->tty.sx)
441 px = c->tty.sx - menu->width - 4;
442 if (py + menu->count + 2 > c->tty.sy)
443 py = c->tty.sy - menu->count - 2;
445 md = xcalloc(1, sizeof *md);
446 md->item = item;
447 md->flags = flags;
449 if (fs != NULL)
450 cmd_find_copy_state(&md->fs, fs);
451 screen_init(&md->s, menu->width + 4, menu->count + 2, 0);
452 if (~md->flags & MENU_NOMOUSE)
453 md->s.mode |= (MODE_MOUSE_ALL|MODE_MOUSE_BUTTON);
454 md->s.mode &= ~MODE_CURSOR;
456 md->px = px;
457 md->py = py;
459 md->menu = menu;
460 if (md->flags & MENU_NOMOUSE) {
461 for (i = 0; i < menu->count; i++) {
462 name = menu->items[i].name;
463 if (name != NULL && *name != '-')
464 break;
466 if (i != menu->count)
467 md->choice = i;
468 else
469 md->choice = -1;
470 } else
471 md->choice = -1;
473 md->cb = cb;
474 md->data = data;
475 return (md);
479 menu_display(struct menu *menu, int flags, struct cmdq_item *item, u_int px,
480 u_int py, struct client *c, struct cmd_find_state *fs, menu_choice_cb cb,
481 void *data)
483 struct menu_data *md;
485 md = menu_prepare(menu, flags, item, px, py, c, fs, cb, data);
486 if (md == NULL)
487 return (-1);
488 server_client_set_overlay(c, 0, NULL, menu_mode_cb, menu_draw_cb,
489 menu_key_cb, menu_free_cb, NULL, md);
490 return (0);