Remove old buffer when renaming rather than complaining, GitHub issue
[tmux-openbsd.git] / menu.c
blob288030b2188e1b6e4534469680304a1ad2999b4c
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;
67 if (line && menu->items[menu->count - 1].name == NULL)
68 return;
70 menu->items = xreallocarray(menu->items, menu->count + 1,
71 sizeof *menu->items);
72 new_item = &menu->items[menu->count++];
73 memset(new_item, 0, sizeof *new_item);
75 if (line)
76 return;
78 if (fs != NULL)
79 s = format_single_from_state(qitem, item->name, c, fs);
80 else
81 s = format_single(qitem, item->name, c, NULL, NULL, NULL);
82 if (*s == '\0') { /* no item if empty after format expanded */
83 menu->count--;
84 return;
86 max_width = c->tty.sx - 4;
88 slen = strlen(s);
89 if (*s != '-' && item->key != KEYC_UNKNOWN && item->key != KEYC_NONE) {
90 key = key_string_lookup_key(item->key, 0);
91 keylen = strlen(key) + 3; /* 3 = space and two brackets */
94 * Add the key if it is shorter than a quarter of the available
95 * space or there is space for the entire item text and the
96 * key.
98 if (keylen <= max_width / 4)
99 max_width -= keylen;
100 else if (keylen >= max_width || slen >= max_width - keylen)
101 key = NULL;
104 if (slen > max_width) {
105 max_width--;
106 suffix = ">";
108 trimmed = format_trim_right(s, max_width);
109 if (key != NULL) {
110 xasprintf(&name, "%s%s#[default] #[align=right](%s)",
111 trimmed, suffix, key);
112 } else
113 xasprintf(&name, "%s%s", trimmed, suffix);
114 free(trimmed);
116 new_item->name = name;
117 free(s);
119 cmd = item->command;
120 if (cmd != NULL) {
121 if (fs != NULL)
122 s = format_single_from_state(qitem, cmd, c, fs);
123 else
124 s = format_single(qitem, cmd, c, NULL, NULL, NULL);
125 } else
126 s = NULL;
127 new_item->command = s;
128 new_item->key = item->key;
130 width = format_width(new_item->name);
131 if (*new_item->name == '-')
132 width--;
133 if (width > menu->width)
134 menu->width = width;
137 struct menu *
138 menu_create(const char *title)
140 struct menu *menu;
142 menu = xcalloc(1, sizeof *menu);
143 menu->title = xstrdup(title);
144 menu->width = format_width(title);
146 return (menu);
149 void
150 menu_free(struct menu *menu)
152 u_int i;
154 for (i = 0; i < menu->count; i++) {
155 free((void *)menu->items[i].name);
156 free((void *)menu->items[i].command);
158 free(menu->items);
160 free((void *)menu->title);
161 free(menu);
164 struct screen *
165 menu_mode_cb(__unused struct client *c, void *data, u_int *cx, u_int *cy)
167 struct menu_data *md = data;
169 *cx = md->px + 2;
170 if (md->choice == -1)
171 *cy = md->py;
172 else
173 *cy = md->py + 1 + md->choice;
175 return (&md->s);
178 /* Return parts of the input range which are not obstructed by the menu. */
179 void
180 menu_check_cb(__unused struct client *c, void *data, u_int px, u_int py,
181 u_int nx, struct overlay_ranges *r)
183 struct menu_data *md = data;
184 struct menu *menu = md->menu;
186 server_client_overlay_range(md->px, md->py, menu->width + 4,
187 menu->count + 2, px, py, nx, r);
190 void
191 menu_draw_cb(struct client *c, void *data,
192 __unused struct screen_redraw_ctx *rctx)
194 struct menu_data *md = data;
195 struct tty *tty = &c->tty;
196 struct screen *s = &md->s;
197 struct menu *menu = md->menu;
198 struct screen_write_ctx ctx;
199 u_int i, px = md->px, py = md->py;
200 struct grid_cell gc;
202 style_apply(&gc, c->session->curw->window->options, "mode-style", NULL);
204 screen_write_start(&ctx, s);
205 screen_write_clearscreen(&ctx, 8);
206 screen_write_menu(&ctx, menu, md->choice, &gc);
207 screen_write_stop(&ctx);
209 for (i = 0; i < screen_size_y(&md->s); i++) {
210 tty_draw_line(tty, s, 0, i, menu->width + 4, px, py + i,
211 &grid_default_cell, NULL);
215 void
216 menu_free_cb(__unused struct client *c, void *data)
218 struct menu_data *md = data;
220 if (md->item != NULL)
221 cmdq_continue(md->item);
223 if (md->cb != NULL)
224 md->cb(md->menu, UINT_MAX, KEYC_NONE, md->data);
226 screen_free(&md->s);
227 menu_free(md->menu);
228 free(md);
232 menu_key_cb(struct client *c, void *data, struct key_event *event)
234 struct menu_data *md = data;
235 struct menu *menu = md->menu;
236 struct mouse_event *m = &event->m;
237 u_int i;
238 int count = menu->count, old = md->choice;
239 const char *name = NULL;
240 const struct menu_item *item;
241 struct cmdq_state *state;
242 enum cmd_parse_status status;
243 char *error;
245 if (KEYC_IS_MOUSE(event->key)) {
246 if (md->flags & MENU_NOMOUSE) {
247 if (MOUSE_BUTTONS(m->b) != MOUSE_BUTTON_1)
248 return (1);
249 return (0);
251 if (m->x < md->px ||
252 m->x > md->px + 4 + menu->width ||
253 m->y < md->py + 1 ||
254 m->y > md->py + 1 + count - 1) {
255 if (~md->flags & MENU_STAYOPEN) {
256 if (MOUSE_RELEASE(m->b))
257 return (1);
258 } else {
259 if (!MOUSE_RELEASE(m->b) &&
260 !MOUSE_WHEEL(m->b) &&
261 !MOUSE_DRAG(m->b))
262 return (1);
264 if (md->choice != -1) {
265 md->choice = -1;
266 c->flags |= CLIENT_REDRAWOVERLAY;
268 return (0);
270 if (~md->flags & MENU_STAYOPEN) {
271 if (MOUSE_RELEASE(m->b))
272 goto chosen;
273 } else {
274 if (!MOUSE_WHEEL(m->b) && !MOUSE_DRAG(m->b))
275 goto chosen;
277 md->choice = m->y - (md->py + 1);
278 if (md->choice != old)
279 c->flags |= CLIENT_REDRAWOVERLAY;
280 return (0);
282 for (i = 0; i < (u_int)count; i++) {
283 name = menu->items[i].name;
284 if (name == NULL || *name == '-')
285 continue;
286 if (event->key == menu->items[i].key) {
287 md->choice = i;
288 goto chosen;
291 switch (event->key & ~KEYC_MASK_FLAGS) {
292 case KEYC_UP:
293 case 'k':
294 if (old == -1)
295 old = 0;
296 do {
297 if (md->choice == -1 || md->choice == 0)
298 md->choice = count - 1;
299 else
300 md->choice--;
301 name = menu->items[md->choice].name;
302 } while ((name == NULL || *name == '-') && md->choice != old);
303 c->flags |= CLIENT_REDRAWOVERLAY;
304 return (0);
305 case KEYC_BSPACE:
306 if (~md->flags & MENU_TAB)
307 break;
308 return (1);
309 case '\011': /* Tab */
310 if (~md->flags & MENU_TAB)
311 break;
312 if (md->choice == count - 1)
313 return (1);
314 /* FALLTHROUGH */
315 case KEYC_DOWN:
316 case 'j':
317 if (old == -1)
318 old = 0;
319 do {
320 if (md->choice == -1 || md->choice == count - 1)
321 md->choice = 0;
322 else
323 md->choice++;
324 name = menu->items[md->choice].name;
325 } while ((name == NULL || *name == '-') && md->choice != old);
326 c->flags |= CLIENT_REDRAWOVERLAY;
327 return (0);
328 case KEYC_PPAGE:
329 case '\002': /* C-b */
330 if (md->choice < 6)
331 md->choice = 0;
332 else {
333 i = 5;
334 while (i > 0) {
335 md->choice--;
336 name = menu->items[md->choice].name;
337 if (md->choice != 0 &&
338 (name != NULL && *name != '-'))
339 i--;
340 else if (md->choice == 0)
341 break;
344 c->flags |= CLIENT_REDRAWOVERLAY;
345 break;
346 case KEYC_NPAGE:
347 if (md->choice > count - 6) {
348 md->choice = count - 1;
349 name = menu->items[md->choice].name;
350 } else {
351 i = 5;
352 while (i > 0) {
353 md->choice++;
354 name = menu->items[md->choice].name;
355 if (md->choice != count - 1 &&
356 (name != NULL && *name != '-'))
357 i++;
358 else if (md->choice == count - 1)
359 break;
362 while (name == NULL || *name == '-') {
363 md->choice--;
364 name = menu->items[md->choice].name;
366 c->flags |= CLIENT_REDRAWOVERLAY;
367 break;
368 case 'g':
369 case KEYC_HOME:
370 md->choice = 0;
371 name = menu->items[md->choice].name;
372 while (name == NULL || *name == '-') {
373 md->choice++;
374 name = menu->items[md->choice].name;
376 c->flags |= CLIENT_REDRAWOVERLAY;
377 break;
378 case 'G':
379 case KEYC_END:
380 md->choice = count - 1;
381 name = menu->items[md->choice].name;
382 while (name == NULL || *name == '-') {
383 md->choice--;
384 name = menu->items[md->choice].name;
386 c->flags |= CLIENT_REDRAWOVERLAY;
387 break;
388 case '\006': /* C-f */
389 break;
390 case '\r':
391 goto chosen;
392 case '\033': /* Escape */
393 case '\003': /* C-c */
394 case '\007': /* C-g */
395 case 'q':
396 return (1);
398 return (0);
400 chosen:
401 if (md->choice == -1)
402 return (1);
403 item = &menu->items[md->choice];
404 if (item->name == NULL || *item->name == '-') {
405 if (md->flags & MENU_STAYOPEN)
406 return (0);
407 return (1);
409 if (md->cb != NULL) {
410 md->cb(md->menu, md->choice, item->key, md->data);
411 md->cb = NULL;
412 return (1);
415 if (md->item != NULL)
416 event = cmdq_get_event(md->item);
417 else
418 event = NULL;
419 state = cmdq_new_state(&md->fs, event, 0);
421 status = cmd_parse_and_append(item->command, NULL, c, state, &error);
422 if (status == CMD_PARSE_ERROR) {
423 cmdq_append(c, cmdq_get_error(error));
424 free(error);
426 cmdq_free_state(state);
428 return (1);
431 struct menu_data *
432 menu_prepare(struct menu *menu, int flags, int starting_choice,
433 struct cmdq_item *item, u_int px, u_int py, struct client *c,
434 struct cmd_find_state *fs, menu_choice_cb cb, void *data)
436 struct menu_data *md;
437 int choice;
438 const char *name;
440 if (c->tty.sx < menu->width + 4 || c->tty.sy < menu->count + 2)
441 return (NULL);
442 if (px + menu->width + 4 > c->tty.sx)
443 px = c->tty.sx - menu->width - 4;
444 if (py + menu->count + 2 > c->tty.sy)
445 py = c->tty.sy - menu->count - 2;
447 md = xcalloc(1, sizeof *md);
448 md->item = item;
449 md->flags = flags;
451 if (fs != NULL)
452 cmd_find_copy_state(&md->fs, fs);
453 screen_init(&md->s, menu->width + 4, menu->count + 2, 0);
454 if (~md->flags & MENU_NOMOUSE)
455 md->s.mode |= (MODE_MOUSE_ALL|MODE_MOUSE_BUTTON);
456 md->s.mode &= ~MODE_CURSOR;
458 md->px = px;
459 md->py = py;
461 md->menu = menu;
462 md->choice = -1;
464 if (md->flags & MENU_NOMOUSE) {
465 if (starting_choice >= (int)menu->count) {
466 starting_choice = menu->count - 1;
467 choice = starting_choice + 1;
468 for (;;) {
469 name = menu->items[choice - 1].name;
470 if (name != NULL && *name != '-') {
471 md->choice = choice - 1;
472 break;
474 if (--choice == 0)
475 choice = menu->count;
476 if (choice == starting_choice + 1)
477 break;
479 } else if (starting_choice >= 0) {
480 choice = starting_choice;
481 for (;;) {
482 name = menu->items[choice].name;
483 if (name != NULL && *name != '-') {
484 md->choice = choice;
485 break;
487 if (++choice == (int)menu->count)
488 choice = 0;
489 if (choice == starting_choice)
490 break;
495 md->cb = cb;
496 md->data = data;
497 return (md);
501 menu_display(struct menu *menu, int flags, int starting_choice,
502 struct cmdq_item *item, u_int px, u_int py, struct client *c,
503 struct cmd_find_state *fs, menu_choice_cb cb, void *data)
505 struct menu_data *md;
507 md = menu_prepare(menu, flags, starting_choice, item, px, py, c, fs, cb,
508 data);
509 if (md == NULL)
510 return (-1);
511 server_client_set_overlay(c, 0, NULL, menu_mode_cb, menu_draw_cb,
512 menu_key_cb, menu_free_cb, NULL, md);
513 return (0);