status.c revision 1.219 
1/* $OpenBSD: status.c,v 1.219 2021/01/08 08:22:10 nicm Exp $ */
2
3/*
4 * Copyright (c) 2007 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#include <sys/time.h>
21
22#include <errno.h>
23#include <limits.h>
24#include <stdarg.h>
25#include <stdlib.h>
26#include <string.h>
27#include <time.h>
28#include <unistd.h>
29
30#include "tmux.h"
31
32static void	 status_message_callback(int, short, void *);
33static void	 status_timer_callback(int, short, void *);
34
35static char	*status_prompt_find_history_file(void);
36static const char *status_prompt_up_history(u_int *);
37static const char *status_prompt_down_history(u_int *);
38static void	 status_prompt_add_history(const char *);
39
40static char	*status_prompt_complete(struct client *, const char *, u_int);
41static char	*status_prompt_complete_window_menu(struct client *,
42		     struct session *, const char *, u_int, char);
43
44struct status_prompt_menu {
45	struct client	 *c;
46	u_int		  start;
47	u_int		  size;
48	char		**list;
49	char		  flag;
50};
51
52/* Status prompt history. */
53#define PROMPT_HISTORY 100
54static char	**status_prompt_hlist;
55static u_int	  status_prompt_hsize;
56
57/* Find the history file to load/save from/to. */
58static char *
59status_prompt_find_history_file(void)
60{
61	const char	*home, *history_file;
62	char		*path;
63
64	history_file = options_get_string(global_options, "history-file");
65	if (*history_file == '\0')
66		return (NULL);
67	if (*history_file == '/')
68		return (xstrdup(history_file));
69
70	if (history_file[0] != '~' || history_file[1] != '/')
71		return (NULL);
72	if ((home = find_home()) == NULL)
73		return (NULL);
74	xasprintf(&path, "%s%s", home, history_file + 1);
75	return (path);
76}
77
78/* Load status prompt history from file. */
79void
80status_prompt_load_history(void)
81{
82	FILE	*f;
83	char	*history_file, *line, *tmp;
84	size_t	 length;
85
86	if ((history_file = status_prompt_find_history_file()) == NULL)
87		return;
88	log_debug("loading history from %s", history_file);
89
90	f = fopen(history_file, "r");
91	if (f == NULL) {
92		log_debug("%s: %s", history_file, strerror(errno));
93		free(history_file);
94		return;
95	}
96	free(history_file);
97
98	for (;;) {
99		if ((line = fgetln(f, &length)) == NULL)
100			break;
101
102		if (length > 0) {
103			if (line[length - 1] == '\n') {
104				line[length - 1] = '\0';
105				status_prompt_add_history(line);
106			} else {
107				tmp = xmalloc(length + 1);
108				memcpy(tmp, line, length);
109				tmp[length] = '\0';
110				status_prompt_add_history(tmp);
111				free(tmp);
112			}
113		}
114	}
115	fclose(f);
116}
117
118/* Save status prompt history to file. */
119void
120status_prompt_save_history(void)
121{
122	FILE	*f;
123	u_int	 i;
124	char	*history_file;
125
126	if ((history_file = status_prompt_find_history_file()) == NULL)
127		return;
128	log_debug("saving history to %s", history_file);
129
130	f = fopen(history_file, "w");
131	if (f == NULL) {
132		log_debug("%s: %s", history_file, strerror(errno));
133		free(history_file);
134		return;
135	}
136	free(history_file);
137
138	for (i = 0; i < status_prompt_hsize; i++) {
139		fputs(status_prompt_hlist[i], f);
140		fputc('\n', f);
141	}
142	fclose(f);
143
144}
145
146/* Status timer callback. */
147static void
148status_timer_callback(__unused int fd, __unused short events, void *arg)
149{
150	struct client	*c = arg;
151	struct session	*s = c->session;
152	struct timeval	 tv;
153
154	evtimer_del(&c->status.timer);
155
156	if (s == NULL)
157		return;
158
159	if (c->message_string == NULL && c->prompt_string == NULL)
160		c->flags |= CLIENT_REDRAWSTATUS;
161
162	timerclear(&tv);
163	tv.tv_sec = options_get_number(s->options, "status-interval");
164
165	if (tv.tv_sec != 0)
166		evtimer_add(&c->status.timer, &tv);
167	log_debug("client %p, status interval %d", c, (int)tv.tv_sec);
168}
169
170/* Start status timer for client. */
171void
172status_timer_start(struct client *c)
173{
174	struct session	*s = c->session;
175
176	if (event_initialized(&c->status.timer))
177		evtimer_del(&c->status.timer);
178	else
179		evtimer_set(&c->status.timer, status_timer_callback, c);
180
181	if (s != NULL && options_get_number(s->options, "status"))
182		status_timer_callback(-1, 0, c);
183}
184
185/* Start status timer for all clients. */
186void
187status_timer_start_all(void)
188{
189	struct client	*c;
190
191	TAILQ_FOREACH(c, &clients, entry)
192		status_timer_start(c);
193}
194
195/* Update status cache. */
196void
197status_update_cache(struct session *s)
198{
199	s->statuslines = options_get_number(s->options, "status");
200	if (s->statuslines == 0)
201		s->statusat = -1;
202	else if (options_get_number(s->options, "status-position") == 0)
203		s->statusat = 0;
204	else
205		s->statusat = 1;
206}
207
208/* Get screen line of status line. -1 means off. */
209int
210status_at_line(struct client *c)
211{
212	struct session	*s = c->session;
213
214	if (c->flags & (CLIENT_STATUSOFF|CLIENT_CONTROL))
215		return (-1);
216	if (s->statusat != 1)
217		return (s->statusat);
218	return (c->tty.sy - status_line_size(c));
219}
220
221/* Get size of status line for client's session. 0 means off. */
222u_int
223status_line_size(struct client *c)
224{
225	struct session	*s = c->session;
226
227	if (c->flags & (CLIENT_STATUSOFF|CLIENT_CONTROL))
228		return (0);
229	return (s->statuslines);
230}
231
232/* Get window at window list position. */
233struct style_range *
234status_get_range(struct client *c, u_int x, u_int y)
235{
236	struct status_line	*sl = &c->status;
237	struct style_range	*sr;
238
239	if (y >= nitems(sl->entries))
240		return (NULL);
241	TAILQ_FOREACH(sr, &sl->entries[y].ranges, entry) {
242		if (x >= sr->start && x < sr->end)
243			return (sr);
244	}
245	return (NULL);
246}
247
248/* Free all ranges. */
249static void
250status_free_ranges(struct style_ranges *srs)
251{
252	struct style_range	*sr, *sr1;
253
254	TAILQ_FOREACH_SAFE(sr, srs, entry, sr1) {
255		TAILQ_REMOVE(srs, sr, entry);
256		free(sr);
257	}
258}
259
260/* Save old status line. */
261static void
262status_push_screen(struct client *c)
263{
264	struct status_line *sl = &c->status;
265
266	if (sl->active == &sl->screen) {
267		sl->active = xmalloc(sizeof *sl->active);
268		screen_init(sl->active, c->tty.sx, status_line_size(c), 0);
269	}
270	sl->references++;
271}
272
273/* Restore old status line. */
274static void
275status_pop_screen(struct client *c)
276{
277	struct status_line *sl = &c->status;
278
279	if (--sl->references == 0) {
280		screen_free(sl->active);
281		free(sl->active);
282		sl->active = &sl->screen;
283	}
284}
285
286/* Initialize status line. */
287void
288status_init(struct client *c)
289{
290	struct status_line	*sl = &c->status;
291	u_int			 i;
292
293	for (i = 0; i < nitems(sl->entries); i++)
294		TAILQ_INIT(&sl->entries[i].ranges);
295
296	screen_init(&sl->screen, c->tty.sx, 1, 0);
297	sl->active = &sl->screen;
298}
299
300/* Free status line. */
301void
302status_free(struct client *c)
303{
304	struct status_line	*sl = &c->status;
305	u_int			 i;
306
307	for (i = 0; i < nitems(sl->entries); i++) {
308		status_free_ranges(&sl->entries[i].ranges);
309		free((void *)sl->entries[i].expanded);
310	}
311
312	if (event_initialized(&sl->timer))
313		evtimer_del(&sl->timer);
314
315	if (sl->active != &sl->screen) {
316		screen_free(sl->active);
317		free(sl->active);
318	}
319	screen_free(&sl->screen);
320}
321
322/* Draw status line for client. */
323int
324status_redraw(struct client *c)
325{
326	struct status_line		*sl = &c->status;
327	struct status_line_entry	*sle;
328	struct session			*s = c->session;
329	struct screen_write_ctx		 ctx;
330	struct grid_cell		 gc;
331	u_int				 lines, i, n, width = c->tty.sx;
332	int				 flags, force = 0, changed = 0, fg, bg;
333	struct options_entry		*o;
334	union options_value		*ov;
335	struct format_tree		*ft;
336	char				*expanded;
337
338	log_debug("%s enter", __func__);
339
340	/* Shouldn't get here if not the active screen. */
341	if (sl->active != &sl->screen)
342		fatalx("not the active screen");
343
344	/* No status line? */
345	lines = status_line_size(c);
346	if (c->tty.sy == 0 || lines == 0)
347		return (1);
348
349	/* Create format tree. */
350	flags = FORMAT_STATUS;
351	if (c->flags & CLIENT_STATUSFORCE)
352		flags |= FORMAT_FORCE;
353	ft = format_create(c, NULL, FORMAT_NONE, flags);
354	format_defaults(ft, c, NULL, NULL, NULL);
355
356	/* Set up default colour. */
357	style_apply(&gc, s->options, "status-style", ft);
358	fg = options_get_number(s->options, "status-fg");
359	if (fg != 8)
360		gc.fg = fg;
361	bg = options_get_number(s->options, "status-bg");
362	if (bg != 8)
363		gc.bg = bg;
364	if (!grid_cells_equal(&gc, &sl->style)) {
365		force = 1;
366		memcpy(&sl->style, &gc, sizeof sl->style);
367	}
368
369	/* Resize the target screen. */
370	if (screen_size_x(&sl->screen) != width ||
371	    screen_size_y(&sl->screen) != lines) {
372		screen_resize(&sl->screen, width, lines, 0);
373		changed = force = 1;
374	}
375	screen_write_start(&ctx, &sl->screen);
376
377	/* Write the status lines. */
378	o = options_get(s->options, "status-format");
379	if (o == NULL) {
380		for (n = 0; n < width * lines; n++)
381			screen_write_putc(&ctx, &gc, ' ');
382	} else {
383		for (i = 0; i < lines; i++) {
384			screen_write_cursormove(&ctx, 0, i, 0);
385
386			ov = options_array_get(o, i);
387			if (ov == NULL) {
388				for (n = 0; n < width; n++)
389					screen_write_putc(&ctx, &gc, ' ');
390				continue;
391			}
392			sle = &sl->entries[i];
393
394			expanded = format_expand_time(ft, ov->string);
395			if (!force &&
396			    sle->expanded != NULL &&
397			    strcmp(expanded, sle->expanded) == 0) {
398				free(expanded);
399				continue;
400			}
401			changed = 1;
402
403			for (n = 0; n < width; n++)
404				screen_write_putc(&ctx, &gc, ' ');
405			screen_write_cursormove(&ctx, 0, i, 0);
406
407			status_free_ranges(&sle->ranges);
408			format_draw(&ctx, &gc, width, expanded, &sle->ranges);
409
410			free(sle->expanded);
411			sle->expanded = expanded;
412		}
413	}
414	screen_write_stop(&ctx);
415
416	/* Free the format tree. */
417	format_free(ft);
418
419	/* Return if the status line has changed. */
420	log_debug("%s exit: force=%d, changed=%d", __func__, force, changed);
421	return (force || changed);
422}
423
424/* Set a status line message. */
425void
426status_message_set(struct client *c, int delay, int ignore_styles,
427    const char *fmt, ...)
428{
429	struct timeval	tv;
430	va_list		ap;
431
432	status_message_clear(c);
433	status_push_screen(c);
434
435	va_start(ap, fmt);
436	c->message_ignore_styles = ignore_styles;
437	xvasprintf(&c->message_string, fmt, ap);
438	va_end(ap);
439
440	server_add_message("%s message: %s", c->name, c->message_string);
441
442	/*
443	 * With delay -1, the display-time option is used; zero means wait for
444	 * key press; more than zero is the actual delay time in milliseconds.
445	 */
446	if (delay == -1)
447		delay = options_get_number(c->session->options, "display-time");
448	if (delay > 0) {
449		tv.tv_sec = delay / 1000;
450		tv.tv_usec = (delay % 1000) * 1000L;
451
452		if (event_initialized(&c->message_timer))
453			evtimer_del(&c->message_timer);
454		evtimer_set(&c->message_timer, status_message_callback, c);
455
456		evtimer_add(&c->message_timer, &tv);
457	}
458
459	c->tty.flags |= (TTY_NOCURSOR|TTY_FREEZE);
460	c->flags |= CLIENT_REDRAWSTATUS;
461}
462
463/* Clear status line message. */
464void
465status_message_clear(struct client *c)
466{
467	if (c->message_string == NULL)
468		return;
469
470	free(c->message_string);
471	c->message_string = NULL;
472
473	if (c->prompt_string == NULL)
474		c->tty.flags &= ~(TTY_NOCURSOR|TTY_FREEZE);
475	c->flags |= CLIENT_ALLREDRAWFLAGS; /* was frozen and may have changed */
476
477	status_pop_screen(c);
478}
479
480/* Clear status line message after timer expires. */
481static void
482status_message_callback(__unused int fd, __unused short event, void *data)
483{
484	struct client	*c = data;
485
486	status_message_clear(c);
487}
488
489/* Draw client message on status line of present else on last line. */
490int
491status_message_redraw(struct client *c)
492{
493	struct status_line	*sl = &c->status;
494	struct screen_write_ctx	 ctx;
495	struct session		*s = c->session;
496	struct screen		 old_screen;
497	size_t			 len;
498	u_int			 lines, offset;
499	struct grid_cell	 gc;
500	struct format_tree	*ft;
501
502	if (c->tty.sx == 0 || c->tty.sy == 0)
503		return (0);
504	memcpy(&old_screen, sl->active, sizeof old_screen);
505
506	lines = status_line_size(c);
507	if (lines <= 1)
508		lines = 1;
509	screen_init(sl->active, c->tty.sx, lines, 0);
510
511	len = screen_write_strlen("%s", c->message_string);
512	if (len > c->tty.sx)
513		len = c->tty.sx;
514
515	ft = format_create_defaults(NULL, c, NULL, NULL, NULL);
516	style_apply(&gc, s->options, "message-style", ft);
517	format_free(ft);
518
519	screen_write_start(&ctx, sl->active);
520	screen_write_fast_copy(&ctx, &sl->screen, 0, 0, c->tty.sx, lines - 1);
521	screen_write_cursormove(&ctx, 0, lines - 1, 0);
522	for (offset = 0; offset < c->tty.sx; offset++)
523		screen_write_putc(&ctx, &gc, ' ');
524	screen_write_cursormove(&ctx, 0, lines - 1, 0);
525	if (c->message_ignore_styles)
526		screen_write_nputs(&ctx, len, &gc, "%s", c->message_string);
527	else
528		format_draw(&ctx, &gc, c->tty.sx, c->message_string, NULL);
529	screen_write_stop(&ctx);
530
531	if (grid_compare(sl->active->grid, old_screen.grid) == 0) {
532		screen_free(&old_screen);
533		return (0);
534	}
535	screen_free(&old_screen);
536	return (1);
537}
538
539/* Enable status line prompt. */
540void
541status_prompt_set(struct client *c, struct cmd_find_state *fs,
542    const char *msg, const char *input, prompt_input_cb inputcb,
543    prompt_free_cb freecb, void *data, int flags)
544{
545	struct format_tree	*ft;
546	char			*tmp;
547
548	if (fs != NULL)
549		ft = format_create_from_state(NULL, c, fs);
550	else
551		ft = format_create_defaults(NULL, c, NULL, NULL, NULL);
552
553	if (input == NULL)
554		input = "";
555	if (flags & PROMPT_NOFORMAT)
556		tmp = xstrdup(input);
557	else
558		tmp = format_expand_time(ft, input);
559
560	status_message_clear(c);
561	status_prompt_clear(c);
562	status_push_screen(c);
563
564	c->prompt_string = format_expand_time(ft, msg);
565
566	if (flags & PROMPT_INCREMENTAL) {
567		c->prompt_last = xstrdup(tmp);
568		c->prompt_buffer = utf8_fromcstr("");
569	} else {
570		c->prompt_last = NULL;
571		c->prompt_buffer = utf8_fromcstr(tmp);
572	}
573	c->prompt_index = utf8_strlen(c->prompt_buffer);
574
575	c->prompt_inputcb = inputcb;
576	c->prompt_freecb = freecb;
577	c->prompt_data = data;
578
579	c->prompt_hindex = 0;
580
581	c->prompt_flags = flags;
582	c->prompt_mode = PROMPT_ENTRY;
583
584	if (~flags & PROMPT_INCREMENTAL)
585		c->tty.flags |= (TTY_NOCURSOR|TTY_FREEZE);
586	c->flags |= CLIENT_REDRAWSTATUS;
587
588	if (flags & PROMPT_INCREMENTAL)
589		c->prompt_inputcb(c, c->prompt_data, "=", 0);
590
591	free(tmp);
592	format_free(ft);
593}
594
595/* Remove status line prompt. */
596void
597status_prompt_clear(struct client *c)
598{
599	if (c->prompt_string == NULL)
600		return;
601
602	if (c->prompt_freecb != NULL && c->prompt_data != NULL)
603		c->prompt_freecb(c->prompt_data);
604
605	free(c->prompt_last);
606	c->prompt_last = NULL;
607
608	free(c->prompt_string);
609	c->prompt_string = NULL;
610
611	free(c->prompt_buffer);
612	c->prompt_buffer = NULL;
613
614	free(c->prompt_saved);
615	c->prompt_saved = NULL;
616
617	c->tty.flags &= ~(TTY_NOCURSOR|TTY_FREEZE);
618	c->flags |= CLIENT_ALLREDRAWFLAGS; /* was frozen and may have changed */
619
620	status_pop_screen(c);
621}
622
623/* Update status line prompt with a new prompt string. */
624void
625status_prompt_update(struct client *c, const char *msg, const char *input)
626{
627	struct format_tree	*ft;
628	char			*tmp;
629
630	ft = format_create(c, NULL, FORMAT_NONE, 0);
631	format_defaults(ft, c, NULL, NULL, NULL);
632
633	tmp = format_expand_time(ft, input);
634
635	free(c->prompt_string);
636	c->prompt_string = format_expand_time(ft, msg);
637
638	free(c->prompt_buffer);
639	c->prompt_buffer = utf8_fromcstr(tmp);
640	c->prompt_index = utf8_strlen(c->prompt_buffer);
641
642	c->prompt_hindex = 0;
643
644	c->flags |= CLIENT_REDRAWSTATUS;
645
646	free(tmp);
647	format_free(ft);
648}
649
650/* Draw client prompt on status line of present else on last line. */
651int
652status_prompt_redraw(struct client *c)
653{
654	struct status_line	*sl = &c->status;
655	struct screen_write_ctx	 ctx;
656	struct session		*s = c->session;
657	struct screen		 old_screen;
658	u_int			 i, lines, offset, left, start, width;
659	u_int			 pcursor, pwidth;
660	struct grid_cell	 gc, cursorgc;
661	struct format_tree	*ft;
662
663	if (c->tty.sx == 0 || c->tty.sy == 0)
664		return (0);
665	memcpy(&old_screen, sl->active, sizeof old_screen);
666
667	lines = status_line_size(c);
668	if (lines <= 1)
669		lines = 1;
670	screen_init(sl->active, c->tty.sx, lines, 0);
671
672	ft = format_create_defaults(NULL, c, NULL, NULL, NULL);
673	if (c->prompt_mode == PROMPT_COMMAND)
674		style_apply(&gc, s->options, "message-command-style", ft);
675	else
676		style_apply(&gc, s->options, "message-style", ft);
677	format_free(ft);
678
679	memcpy(&cursorgc, &gc, sizeof cursorgc);
680	cursorgc.attr ^= GRID_ATTR_REVERSE;
681
682	start = screen_write_strlen("%s", c->prompt_string);
683	if (start > c->tty.sx)
684		start = c->tty.sx;
685
686	screen_write_start(&ctx, sl->active);
687	screen_write_fast_copy(&ctx, &sl->screen, 0, 0, c->tty.sx, lines - 1);
688	screen_write_cursormove(&ctx, 0, lines - 1, 0);
689	for (offset = 0; offset < c->tty.sx; offset++)
690		screen_write_putc(&ctx, &gc, ' ');
691	screen_write_cursormove(&ctx, 0, lines - 1, 0);
692	screen_write_nputs(&ctx, start, &gc, "%s", c->prompt_string);
693	screen_write_cursormove(&ctx, start, lines - 1, 0);
694
695	left = c->tty.sx - start;
696	if (left == 0)
697		goto finished;
698
699	pcursor = utf8_strwidth(c->prompt_buffer, c->prompt_index);
700	pwidth = utf8_strwidth(c->prompt_buffer, -1);
701	if (pcursor >= left) {
702		/*
703		 * The cursor would be outside the screen so start drawing
704		 * with it on the right.
705		 */
706		offset = (pcursor - left) + 1;
707		pwidth = left;
708	} else
709		offset = 0;
710	if (pwidth > left)
711		pwidth = left;
712
713	width = 0;
714	for (i = 0; c->prompt_buffer[i].size != 0; i++) {
715		if (width < offset) {
716			width += c->prompt_buffer[i].width;
717			continue;
718		}
719		if (width >= offset + pwidth)
720			break;
721		width += c->prompt_buffer[i].width;
722		if (width > offset + pwidth)
723			break;
724
725		if (i != c->prompt_index) {
726			utf8_copy(&gc.data, &c->prompt_buffer[i]);
727			screen_write_cell(&ctx, &gc);
728		} else {
729			utf8_copy(&cursorgc.data, &c->prompt_buffer[i]);
730			screen_write_cell(&ctx, &cursorgc);
731		}
732	}
733	if (sl->active->cx < screen_size_x(sl->active) && c->prompt_index >= i)
734		screen_write_putc(&ctx, &cursorgc, ' ');
735
736finished:
737	screen_write_stop(&ctx);
738
739	if (grid_compare(sl->active->grid, old_screen.grid) == 0) {
740		screen_free(&old_screen);
741		return (0);
742	}
743	screen_free(&old_screen);
744	return (1);
745}
746
747/* Is this a separator? */
748static int
749status_prompt_in_list(const char *ws, const struct utf8_data *ud)
750{
751	if (ud->size != 1 || ud->width != 1)
752		return (0);
753	return (strchr(ws, *ud->data) != NULL);
754}
755
756/* Is this a space? */
757static int
758status_prompt_space(const struct utf8_data *ud)
759{
760	if (ud->size != 1 || ud->width != 1)
761		return (0);
762	return (*ud->data == ' ');
763}
764
765/*
766 * Translate key from emacs to vi. Return 0 to drop key, 1 to process the key
767 * as an emacs key; return 2 to append to the buffer.
768 */
769static int
770status_prompt_translate_key(struct client *c, key_code key, key_code *new_key)
771{
772	if (c->prompt_mode == PROMPT_ENTRY) {
773		switch (key) {
774		case '\003': /* C-c */
775		case '\007': /* C-g */
776		case '\010': /* C-h */
777		case '\011': /* Tab */
778		case '\025': /* C-u */
779		case '\027': /* C-w */
780		case '\n':
781		case '\r':
782		case KEYC_BSPACE:
783		case KEYC_DC:
784		case KEYC_DOWN:
785		case KEYC_END:
786		case KEYC_HOME:
787		case KEYC_LEFT:
788		case KEYC_RIGHT:
789		case KEYC_UP:
790			*new_key = key;
791			return (1);
792		case '\033': /* Escape */
793			c->prompt_mode = PROMPT_COMMAND;
794			c->flags |= CLIENT_REDRAWSTATUS;
795			return (0);
796		}
797		*new_key = key;
798		return (2);
799	}
800
801	switch (key) {
802	case 'A':
803	case 'I':
804	case 'C':
805	case 's':
806	case 'a':
807		c->prompt_mode = PROMPT_ENTRY;
808		c->flags |= CLIENT_REDRAWSTATUS;
809		break; /* switch mode and... */
810	case 'S':
811		c->prompt_mode = PROMPT_ENTRY;
812		c->flags |= CLIENT_REDRAWSTATUS;
813		*new_key = '\025'; /* C-u */
814		return (1);
815	case 'i':
816	case '\033': /* Escape */
817		c->prompt_mode = PROMPT_ENTRY;
818		c->flags |= CLIENT_REDRAWSTATUS;
819		return (0);
820	}
821
822	switch (key) {
823	case 'A':
824	case '$':
825		*new_key = KEYC_END;
826		return (1);
827	case 'I':
828	case '0':
829	case '^':
830		*new_key = KEYC_HOME;
831		return (1);
832	case 'C':
833	case 'D':
834		*new_key = '\013'; /* C-k */
835		return (1);
836	case KEYC_BSPACE:
837	case 'X':
838		*new_key = KEYC_BSPACE;
839		return (1);
840	case 'b':
841	case 'B':
842		*new_key = 'b'|KEYC_META;
843		return (1);
844	case 'd':
845		*new_key = '\025';
846		return (1);
847	case 'e':
848	case 'E':
849	case 'w':
850	case 'W':
851		*new_key = 'f'|KEYC_META;
852		return (1);
853	case 'p':
854		*new_key = '\031'; /* C-y */
855		return (1);
856	case 'q':
857		*new_key = '\003'; /* C-c */
858		return (1);
859	case 's':
860	case KEYC_DC:
861	case 'x':
862		*new_key = KEYC_DC;
863		return (1);
864	case KEYC_DOWN:
865	case 'j':
866		*new_key = KEYC_DOWN;
867		return (1);
868	case KEYC_LEFT:
869	case 'h':
870		*new_key = KEYC_LEFT;
871		return (1);
872	case 'a':
873	case KEYC_RIGHT:
874	case 'l':
875		*new_key = KEYC_RIGHT;
876		return (1);
877	case KEYC_UP:
878	case 'k':
879		*new_key = KEYC_UP;
880		return (1);
881	case '\010' /* C-h */:
882	case '\003' /* C-c */:
883	case '\n':
884	case '\r':
885		return (1);
886	}
887	return (0);
888}
889
890/* Paste into prompt. */
891static int
892status_prompt_paste(struct client *c)
893{
894	struct paste_buffer	*pb;
895	const char		*bufdata;
896	size_t			 size, n, bufsize;
897	u_int			 i;
898	struct utf8_data	*ud, *udp;
899	enum utf8_state		 more;
900
901	size = utf8_strlen(c->prompt_buffer);
902	if (c->prompt_saved != NULL) {
903		ud = c->prompt_saved;
904		n = utf8_strlen(c->prompt_saved);
905	} else {
906		if ((pb = paste_get_top(NULL)) == NULL)
907			return (0);
908		bufdata = paste_buffer_data(pb, &bufsize);
909		ud = xreallocarray(NULL, bufsize + 1, sizeof *ud);
910		udp = ud;
911		for (i = 0; i != bufsize; /* nothing */) {
912			more = utf8_open(udp, bufdata[i]);
913			if (more == UTF8_MORE) {
914				while (++i != bufsize && more == UTF8_MORE)
915					more = utf8_append(udp, bufdata[i]);
916				if (more == UTF8_DONE) {
917					udp++;
918					continue;
919				}
920				i -= udp->have;
921			}
922			if (bufdata[i] <= 31 || bufdata[i] >= 127)
923				break;
924			utf8_set(udp, bufdata[i]);
925			udp++;
926			i++;
927		}
928		udp->size = 0;
929		n = udp - ud;
930	}
931	if (n == 0)
932		return (0);
933
934	c->prompt_buffer = xreallocarray(c->prompt_buffer, size + n + 1,
935	    sizeof *c->prompt_buffer);
936	if (c->prompt_index == size) {
937		memcpy(c->prompt_buffer + c->prompt_index, ud,
938		    n * sizeof *c->prompt_buffer);
939		c->prompt_index += n;
940		c->prompt_buffer[c->prompt_index].size = 0;
941	} else {
942		memmove(c->prompt_buffer + c->prompt_index + n,
943		    c->prompt_buffer + c->prompt_index,
944		    (size + 1 - c->prompt_index) * sizeof *c->prompt_buffer);
945		memcpy(c->prompt_buffer + c->prompt_index, ud,
946		    n * sizeof *c->prompt_buffer);
947		c->prompt_index += n;
948	}
949
950	if (ud != c->prompt_saved)
951		free(ud);
952	return (1);
953}
954
955/* Finish completion. */
956static int
957status_prompt_replace_complete(struct client *c, const char *s)
958{
959	char			 word[64], *allocated = NULL;
960	size_t			 size, n, off, idx, used;
961	struct utf8_data	*first, *last, *ud;
962
963	/* Work out where the cursor currently is. */
964	idx = c->prompt_index;
965	if (idx != 0)
966		idx--;
967	size = utf8_strlen(c->prompt_buffer);
968
969	/* Find the word we are in. */
970	first = &c->prompt_buffer[idx];
971	while (first > c->prompt_buffer && !status_prompt_space(first))
972		first--;
973	while (first->size != 0 && status_prompt_space(first))
974		first++;
975	last = &c->prompt_buffer[idx];
976	while (last->size != 0 && !status_prompt_space(last))
977		last++;
978	while (last > c->prompt_buffer && status_prompt_space(last))
979		last--;
980	if (last->size != 0)
981		last++;
982	if (last < first)
983		return (0);
984	if (s == NULL) {
985		used = 0;
986		for (ud = first; ud < last; ud++) {
987			if (used + ud->size >= sizeof word)
988				break;
989			memcpy(word + used, ud->data, ud->size);
990			used += ud->size;
991		}
992		if (ud != last)
993			return (0);
994		word[used] = '\0';
995	}
996
997	/* Try to complete it. */
998	if (s == NULL) {
999		allocated = status_prompt_complete(c, word,
1000		    first - c->prompt_buffer);
1001		if (allocated == NULL)
1002			return (0);
1003		s = allocated;
1004	}
1005
1006	/* Trim out word. */
1007	n = size - (last - c->prompt_buffer) + 1; /* with \0 */
1008	memmove(first, last, n * sizeof *c->prompt_buffer);
1009	size -= last - first;
1010
1011	/* Insert the new word. */
1012	size += strlen(s);
1013	off = first - c->prompt_buffer;
1014	c->prompt_buffer = xreallocarray(c->prompt_buffer, size + 1,
1015	    sizeof *c->prompt_buffer);
1016	first = c->prompt_buffer + off;
1017	memmove(first + strlen(s), first, n * sizeof *c->prompt_buffer);
1018	for (idx = 0; idx < strlen(s); idx++)
1019		utf8_set(&first[idx], s[idx]);
1020	c->prompt_index = (first - c->prompt_buffer) + strlen(s);
1021
1022	free(allocated);
1023	return (1);
1024}
1025
1026/* Handle keys in prompt. */
1027int
1028status_prompt_key(struct client *c, key_code key)
1029{
1030	struct options		*oo = c->session->options;
1031	char			*s, *cp, prefix = '=';
1032	const char		*histstr, *ws = NULL, *keystring;
1033	size_t			 size, idx;
1034	struct utf8_data	 tmp;
1035	int			 keys;
1036
1037	if (c->prompt_flags & PROMPT_KEY) {
1038		keystring = key_string_lookup_key(key, 0);
1039		c->prompt_inputcb(c, c->prompt_data, keystring, 1);
1040		status_prompt_clear(c);
1041		return (0);
1042	}
1043	size = utf8_strlen(c->prompt_buffer);
1044
1045	if (c->prompt_flags & PROMPT_NUMERIC) {
1046		if (key >= '0' && key <= '9')
1047			goto append_key;
1048		s = utf8_tocstr(c->prompt_buffer);
1049		c->prompt_inputcb(c, c->prompt_data, s, 1);
1050		status_prompt_clear(c);
1051		free(s);
1052		return (1);
1053	}
1054	key &= ~KEYC_MASK_FLAGS;
1055
1056	keys = options_get_number(c->session->options, "status-keys");
1057	if (keys == MODEKEY_VI) {
1058		switch (status_prompt_translate_key(c, key, &key)) {
1059		case 1:
1060			goto process_key;
1061		case 2:
1062			goto append_key;
1063		default:
1064			return (0);
1065		}
1066	}
1067
1068process_key:
1069	switch (key) {
1070	case KEYC_LEFT:
1071	case '\002': /* C-b */
1072		if (c->prompt_index > 0) {
1073			c->prompt_index--;
1074			break;
1075		}
1076		break;
1077	case KEYC_RIGHT:
1078	case '\006': /* C-f */
1079		if (c->prompt_index < size) {
1080			c->prompt_index++;
1081			break;
1082		}
1083		break;
1084	case KEYC_HOME:
1085	case '\001': /* C-a */
1086		if (c->prompt_index != 0) {
1087			c->prompt_index = 0;
1088			break;
1089		}
1090		break;
1091	case KEYC_END:
1092	case '\005': /* C-e */
1093		if (c->prompt_index != size) {
1094			c->prompt_index = size;
1095			break;
1096		}
1097		break;
1098	case '\011': /* Tab */
1099		if (status_prompt_replace_complete(c, NULL))
1100			goto changed;
1101		break;
1102	case KEYC_BSPACE:
1103	case '\010': /* C-h */
1104		if (c->prompt_index != 0) {
1105			if (c->prompt_index == size)
1106				c->prompt_buffer[--c->prompt_index].size = 0;
1107			else {
1108				memmove(c->prompt_buffer + c->prompt_index - 1,
1109				    c->prompt_buffer + c->prompt_index,
1110				    (size + 1 - c->prompt_index) *
1111				    sizeof *c->prompt_buffer);
1112				c->prompt_index--;
1113			}
1114			goto changed;
1115		}
1116		break;
1117	case KEYC_DC:
1118	case '\004': /* C-d */
1119		if (c->prompt_index != size) {
1120			memmove(c->prompt_buffer + c->prompt_index,
1121			    c->prompt_buffer + c->prompt_index + 1,
1122			    (size + 1 - c->prompt_index) *
1123			    sizeof *c->prompt_buffer);
1124			goto changed;
1125		}
1126		break;
1127	case '\025': /* C-u */
1128		c->prompt_buffer[0].size = 0;
1129		c->prompt_index = 0;
1130		goto changed;
1131	case '\013': /* C-k */
1132		if (c->prompt_index < size) {
1133			c->prompt_buffer[c->prompt_index].size = 0;
1134			goto changed;
1135		}
1136		break;
1137	case '\027': /* C-w */
1138		ws = options_get_string(oo, "word-separators");
1139		idx = c->prompt_index;
1140
1141		/* Find a non-separator. */
1142		while (idx != 0) {
1143			idx--;
1144			if (!status_prompt_in_list(ws, &c->prompt_buffer[idx]))
1145				break;
1146		}
1147
1148		/* Find the separator at the beginning of the word. */
1149		while (idx != 0) {
1150			idx--;
1151			if (status_prompt_in_list(ws, &c->prompt_buffer[idx])) {
1152				/* Go back to the word. */
1153				idx++;
1154				break;
1155			}
1156		}
1157
1158		free(c->prompt_saved);
1159		c->prompt_saved = xcalloc(sizeof *c->prompt_buffer,
1160		    (c->prompt_index - idx) + 1);
1161		memcpy(c->prompt_saved, c->prompt_buffer + idx,
1162		    (c->prompt_index - idx) * sizeof *c->prompt_buffer);
1163
1164		memmove(c->prompt_buffer + idx,
1165		    c->prompt_buffer + c->prompt_index,
1166		    (size + 1 - c->prompt_index) *
1167		    sizeof *c->prompt_buffer);
1168		memset(c->prompt_buffer + size - (c->prompt_index - idx),
1169		    '\0', (c->prompt_index - idx) * sizeof *c->prompt_buffer);
1170		c->prompt_index = idx;
1171
1172		goto changed;
1173	case 'f'|KEYC_META:
1174	case KEYC_RIGHT|KEYC_CTRL:
1175		ws = options_get_string(oo, "word-separators");
1176
1177		/* Find a word. */
1178		while (c->prompt_index != size) {
1179			idx = ++c->prompt_index;
1180			if (!status_prompt_in_list(ws, &c->prompt_buffer[idx]))
1181				break;
1182		}
1183
1184		/* Find the separator at the end of the word. */
1185		while (c->prompt_index != size) {
1186			idx = ++c->prompt_index;
1187			if (status_prompt_in_list(ws, &c->prompt_buffer[idx]))
1188				break;
1189		}
1190
1191		/* Back up to the end-of-word like vi. */
1192		if (options_get_number(oo, "status-keys") == MODEKEY_VI &&
1193		    c->prompt_index != 0)
1194			c->prompt_index--;
1195
1196		goto changed;
1197	case 'b'|KEYC_META:
1198	case KEYC_LEFT|KEYC_CTRL:
1199		ws = options_get_string(oo, "word-separators");
1200
1201		/* Find a non-separator. */
1202		while (c->prompt_index != 0) {
1203			idx = --c->prompt_index;
1204			if (!status_prompt_in_list(ws, &c->prompt_buffer[idx]))
1205				break;
1206		}
1207
1208		/* Find the separator at the beginning of the word. */
1209		while (c->prompt_index != 0) {
1210			idx = --c->prompt_index;
1211			if (status_prompt_in_list(ws, &c->prompt_buffer[idx])) {
1212				/* Go back to the word. */
1213				c->prompt_index++;
1214				break;
1215			}
1216		}
1217		goto changed;
1218	case KEYC_UP:
1219	case '\020': /* C-p */
1220		histstr = status_prompt_up_history(&c->prompt_hindex);
1221		if (histstr == NULL)
1222			break;
1223		free(c->prompt_buffer);
1224		c->prompt_buffer = utf8_fromcstr(histstr);
1225		c->prompt_index = utf8_strlen(c->prompt_buffer);
1226		goto changed;
1227	case KEYC_DOWN:
1228	case '\016': /* C-n */
1229		histstr = status_prompt_down_history(&c->prompt_hindex);
1230		if (histstr == NULL)
1231			break;
1232		free(c->prompt_buffer);
1233		c->prompt_buffer = utf8_fromcstr(histstr);
1234		c->prompt_index = utf8_strlen(c->prompt_buffer);
1235		goto changed;
1236	case '\031': /* C-y */
1237		if (status_prompt_paste(c))
1238			goto changed;
1239		break;
1240	case '\024': /* C-t */
1241		idx = c->prompt_index;
1242		if (idx < size)
1243			idx++;
1244		if (idx >= 2) {
1245			utf8_copy(&tmp, &c->prompt_buffer[idx - 2]);
1246			utf8_copy(&c->prompt_buffer[idx - 2],
1247			    &c->prompt_buffer[idx - 1]);
1248			utf8_copy(&c->prompt_buffer[idx - 1], &tmp);
1249			c->prompt_index = idx;
1250			goto changed;
1251		}
1252		break;
1253	case '\r':
1254	case '\n':
1255		s = utf8_tocstr(c->prompt_buffer);
1256		if (*s != '\0')
1257			status_prompt_add_history(s);
1258		if (c->prompt_inputcb(c, c->prompt_data, s, 1) == 0)
1259			status_prompt_clear(c);
1260		free(s);
1261		break;
1262	case '\033': /* Escape */
1263	case '\003': /* C-c */
1264	case '\007': /* C-g */
1265		if (c->prompt_inputcb(c, c->prompt_data, NULL, 1) == 0)
1266			status_prompt_clear(c);
1267		break;
1268	case '\022': /* C-r */
1269		if (~c->prompt_flags & PROMPT_INCREMENTAL)
1270			break;
1271		if (c->prompt_buffer[0].size == 0) {
1272			prefix = '=';
1273			free (c->prompt_buffer);
1274			c->prompt_buffer = utf8_fromcstr(c->prompt_last);
1275			c->prompt_index = utf8_strlen(c->prompt_buffer);
1276		} else
1277			prefix = '-';
1278		goto changed;
1279	case '\023': /* C-s */
1280		if (~c->prompt_flags & PROMPT_INCREMENTAL)
1281			break;
1282		if (c->prompt_buffer[0].size == 0) {
1283			prefix = '=';
1284			free (c->prompt_buffer);
1285			c->prompt_buffer = utf8_fromcstr(c->prompt_last);
1286			c->prompt_index = utf8_strlen(c->prompt_buffer);
1287		} else
1288			prefix = '+';
1289		goto changed;
1290	default:
1291		goto append_key;
1292	}
1293
1294	c->flags |= CLIENT_REDRAWSTATUS;
1295	return (0);
1296
1297append_key:
1298	if (key <= 0x1f || key >= KEYC_BASE)
1299		return (0);
1300	if (key <= 0x7f)
1301		utf8_set(&tmp, key);
1302	else
1303		utf8_to_data(key, &tmp);
1304
1305	c->prompt_buffer = xreallocarray(c->prompt_buffer, size + 2,
1306	    sizeof *c->prompt_buffer);
1307
1308	if (c->prompt_index == size) {
1309		utf8_copy(&c->prompt_buffer[c->prompt_index], &tmp);
1310		c->prompt_index++;
1311		c->prompt_buffer[c->prompt_index].size = 0;
1312	} else {
1313		memmove(c->prompt_buffer + c->prompt_index + 1,
1314		    c->prompt_buffer + c->prompt_index,
1315		    (size + 1 - c->prompt_index) *
1316		    sizeof *c->prompt_buffer);
1317		utf8_copy(&c->prompt_buffer[c->prompt_index], &tmp);
1318		c->prompt_index++;
1319	}
1320
1321	if (c->prompt_flags & PROMPT_SINGLE) {
1322		s = utf8_tocstr(c->prompt_buffer);
1323		if (strlen(s) != 1)
1324			status_prompt_clear(c);
1325		else if (c->prompt_inputcb(c, c->prompt_data, s, 1) == 0)
1326			status_prompt_clear(c);
1327		free(s);
1328	}
1329
1330changed:
1331	c->flags |= CLIENT_REDRAWSTATUS;
1332	if (c->prompt_flags & PROMPT_INCREMENTAL) {
1333		s = utf8_tocstr(c->prompt_buffer);
1334		xasprintf(&cp, "%c%s", prefix, s);
1335		c->prompt_inputcb(c, c->prompt_data, cp, 0);
1336		free(cp);
1337		free(s);
1338	}
1339	return (0);
1340}
1341
1342/* Get previous line from the history. */
1343static const char *
1344status_prompt_up_history(u_int *idx)
1345{
1346	/*
1347	 * History runs from 0 to size - 1. Index is from 0 to size. Zero is
1348	 * empty.
1349	 */
1350
1351	if (status_prompt_hsize == 0 || *idx == status_prompt_hsize)
1352		return (NULL);
1353	(*idx)++;
1354	return (status_prompt_hlist[status_prompt_hsize - *idx]);
1355}
1356
1357/* Get next line from the history. */
1358static const char *
1359status_prompt_down_history(u_int *idx)
1360{
1361	if (status_prompt_hsize == 0 || *idx == 0)
1362		return ("");
1363	(*idx)--;
1364	if (*idx == 0)
1365		return ("");
1366	return (status_prompt_hlist[status_prompt_hsize - *idx]);
1367}
1368
1369/* Add line to the history. */
1370static void
1371status_prompt_add_history(const char *line)
1372{
1373	size_t	size;
1374
1375	if (status_prompt_hsize > 0 &&
1376	    strcmp(status_prompt_hlist[status_prompt_hsize - 1], line) == 0)
1377		return;
1378
1379	if (status_prompt_hsize == PROMPT_HISTORY) {
1380		free(status_prompt_hlist[0]);
1381
1382		size = (PROMPT_HISTORY - 1) * sizeof *status_prompt_hlist;
1383		memmove(&status_prompt_hlist[0], &status_prompt_hlist[1], size);
1384
1385		status_prompt_hlist[status_prompt_hsize - 1] = xstrdup(line);
1386		return;
1387	}
1388
1389	status_prompt_hlist = xreallocarray(status_prompt_hlist,
1390	    status_prompt_hsize + 1, sizeof *status_prompt_hlist);
1391	status_prompt_hlist[status_prompt_hsize++] = xstrdup(line);
1392}
1393
1394/* Build completion list. */
1395static char **
1396status_prompt_complete_list(u_int *size, const char *s, int at_start)
1397{
1398	char					**list = NULL;
1399	const char				**layout, *value, *cp;
1400	const struct cmd_entry			**cmdent;
1401	const struct options_table_entry	 *oe;
1402	size_t					  slen = strlen(s), valuelen;
1403	struct options_entry			 *o;
1404	struct options_array_item		 *a;
1405	const char				 *layouts[] = {
1406		"even-horizontal", "even-vertical", "main-horizontal",
1407		"main-vertical", "tiled", NULL
1408	};
1409
1410	*size = 0;
1411	for (cmdent = cmd_table; *cmdent != NULL; cmdent++) {
1412		if (strncmp((*cmdent)->name, s, slen) == 0) {
1413			list = xreallocarray(list, (*size) + 1, sizeof *list);
1414			list[(*size)++] = xstrdup((*cmdent)->name);
1415		}
1416		if ((*cmdent)->alias != NULL &&
1417		    strncmp((*cmdent)->alias, s, slen) == 0) {
1418			list = xreallocarray(list, (*size) + 1, sizeof *list);
1419			list[(*size)++] = xstrdup((*cmdent)->alias);
1420		}
1421	}
1422	o = options_get_only(global_options, "command-alias");
1423	if (o != NULL) {
1424		a = options_array_first(o);
1425		while (a != NULL) {
1426			value = options_array_item_value(a)->string;
1427			if ((cp = strchr(value, '=')) == NULL)
1428				goto next;
1429			valuelen = cp - value;
1430			if (slen > valuelen || strncmp(value, s, slen) != 0)
1431				goto next;
1432
1433			list = xreallocarray(list, (*size) + 1, sizeof *list);
1434			list[(*size)++] = xstrndup(value, valuelen);
1435
1436		next:
1437			a = options_array_next(a);
1438		}
1439	}
1440	if (at_start)
1441		return (list);
1442
1443	for (oe = options_table; oe->name != NULL; oe++) {
1444		if (strncmp(oe->name, s, slen) == 0) {
1445			list = xreallocarray(list, (*size) + 1, sizeof *list);
1446			list[(*size)++] = xstrdup(oe->name);
1447		}
1448	}
1449	for (layout = layouts; *layout != NULL; layout++) {
1450		if (strncmp(*layout, s, slen) == 0) {
1451			list = xreallocarray(list, (*size) + 1, sizeof *list);
1452			list[(*size)++] = xstrdup(*layout);
1453		}
1454	}
1455	return (list);
1456}
1457
1458/* Find longest prefix. */
1459static char *
1460status_prompt_complete_prefix(char **list, u_int size)
1461{
1462	char	 *out;
1463	u_int	  i;
1464	size_t	  j;
1465
1466	if (list == NULL || size == 0)
1467		return (NULL);
1468	out = xstrdup(list[0]);
1469	for (i = 1; i < size; i++) {
1470		j = strlen(list[i]);
1471		if (j > strlen(out))
1472			j = strlen(out);
1473		for (; j > 0; j--) {
1474			if (out[j - 1] != list[i][j - 1])
1475				out[j - 1] = '\0';
1476		}
1477	}
1478	return (out);
1479}
1480
1481/* Complete word menu callback. */
1482static void
1483status_prompt_menu_callback(__unused struct menu *menu, u_int idx, key_code key,
1484    void *data)
1485{
1486	struct status_prompt_menu	*spm = data;
1487	struct client			*c = spm->c;
1488	u_int				 i;
1489	char				*s;
1490
1491	if (key != KEYC_NONE) {
1492		idx += spm->start;
1493		if (spm->flag == '\0')
1494			s = xstrdup(spm->list[idx]);
1495		else
1496			xasprintf(&s, "-%c%s", spm->flag, spm->list[idx]);
1497		if (c->prompt_flags & PROMPT_WINDOW) {
1498			free(c->prompt_buffer);
1499			c->prompt_buffer = utf8_fromcstr(s);
1500			c->prompt_index = utf8_strlen(c->prompt_buffer);
1501			c->flags |= CLIENT_REDRAWSTATUS;
1502		} else if (status_prompt_replace_complete(c, s))
1503			c->flags |= CLIENT_REDRAWSTATUS;
1504		free(s);
1505	}
1506
1507	for (i = 0; i < spm->size; i++)
1508		free(spm->list[i]);
1509	free(spm->list);
1510}
1511
1512/* Show complete word menu. */
1513static int
1514status_prompt_complete_list_menu(struct client *c, char **list, u_int size,
1515    u_int offset, char flag)
1516{
1517	struct menu			*menu;
1518	struct menu_item		 item;
1519	struct status_prompt_menu	*spm;
1520	u_int				 lines = status_line_size(c), height, i;
1521	u_int				 py;
1522
1523	if (size <= 1)
1524		return (0);
1525	if (c->tty.sy - lines < 3)
1526		return (0);
1527
1528	spm = xmalloc(sizeof *spm);
1529	spm->c = c;
1530	spm->size = size;
1531	spm->list = list;
1532	spm->flag = flag;
1533
1534	height = c->tty.sy - lines - 2;
1535	if (height > 10)
1536		height = 10;
1537	if (height > size)
1538		height = size;
1539	spm->start = size - height;
1540
1541	menu = menu_create("");
1542	for (i = spm->start; i < size; i++) {
1543		item.name = list[i];
1544		item.key = '0' + (i - spm->start);
1545		item.command = NULL;
1546		menu_add_item(menu, &item, NULL, NULL, NULL);
1547	}
1548
1549	if (options_get_number(c->session->options, "status-position") == 0)
1550		py = lines;
1551	else
1552		py = c->tty.sy - 3 - height;
1553	offset += utf8_cstrwidth(c->prompt_string);
1554	if (offset > 2)
1555		offset -= 2;
1556	else
1557		offset = 0;
1558
1559	if (menu_display(menu, MENU_NOMOUSE|MENU_TAB, NULL, offset,
1560	    py, c, NULL, status_prompt_menu_callback, spm) != 0) {
1561		menu_free(menu);
1562		free(spm);
1563		return (0);
1564	}
1565	return (1);
1566}
1567
1568/* Show complete word menu. */
1569static char *
1570status_prompt_complete_window_menu(struct client *c, struct session *s,
1571    const char *word, u_int offset, char flag)
1572{
1573	struct menu			 *menu;
1574	struct menu_item		  item;
1575	struct status_prompt_menu	 *spm;
1576	struct winlink			 *wl;
1577	char				**list = NULL, *tmp;
1578	u_int				  lines = status_line_size(c), height;
1579	u_int				  py, size = 0;
1580
1581	if (c->tty.sy - lines < 3)
1582		return (NULL);
1583
1584	spm = xmalloc(sizeof *spm);
1585	spm->c = c;
1586	spm->flag = flag;
1587
1588	height = c->tty.sy - lines - 2;
1589	if (height > 10)
1590		height = 10;
1591	spm->start = 0;
1592
1593	menu = menu_create("");
1594	RB_FOREACH(wl, winlinks, &s->windows) {
1595		if (word != NULL && *word != '\0') {
1596			xasprintf(&tmp, "%d", wl->idx);
1597			if (strncmp(tmp, word, strlen(word)) != 0) {
1598				free(tmp);
1599				continue;
1600			}
1601			free(tmp);
1602		}
1603
1604		list = xreallocarray(list, size + 1, sizeof *list);
1605		if (c->prompt_flags & PROMPT_WINDOW) {
1606			xasprintf(&tmp, "%d (%s)", wl->idx, wl->window->name);
1607			xasprintf(&list[size++], "%d", wl->idx);
1608		} else {
1609			xasprintf(&tmp, "%s:%d (%s)", s->name, wl->idx,
1610			    wl->window->name);
1611			xasprintf(&list[size++], "%s:%d", s->name, wl->idx);
1612		}
1613		item.name = tmp;
1614		item.key = '0' + size - 1;
1615		item.command = NULL;
1616		menu_add_item(menu, &item, NULL, NULL, NULL);
1617		free(tmp);
1618
1619		if (size == height)
1620			break;
1621	}
1622	if (size == 0) {
1623		menu_free(menu);
1624		return (NULL);
1625	}
1626	if (size == 1) {
1627		menu_free(menu);
1628		if (flag != '\0') {
1629			xasprintf(&tmp, "-%c%s", flag, list[0]);
1630			free(list[0]);
1631		} else
1632			tmp = list[0];
1633		free(list);
1634		return (tmp);
1635	}
1636	if (height > size)
1637		height = size;
1638
1639	spm->size = size;
1640	spm->list = list;
1641
1642	if (options_get_number(c->session->options, "status-position") == 0)
1643		py = lines;
1644	else
1645		py = c->tty.sy - 3 - height;
1646	offset += utf8_cstrwidth(c->prompt_string);
1647	if (offset > 2)
1648		offset -= 2;
1649	else
1650		offset = 0;
1651
1652	if (menu_display(menu, MENU_NOMOUSE|MENU_TAB, NULL, offset,
1653	    py, c, NULL, status_prompt_menu_callback, spm) != 0) {
1654		menu_free(menu);
1655		free(spm);
1656		return (NULL);
1657	}
1658	return (NULL);
1659}
1660
1661/* Sort complete list. */
1662static int
1663status_prompt_complete_sort(const void *a, const void *b)
1664{
1665	const char	**aa = (const char **)a, **bb = (const char **)b;
1666
1667	return (strcmp(*aa, *bb));
1668}
1669
1670/* Complete a session. */
1671static char *
1672status_prompt_complete_session(char ***list, u_int *size, const char *s,
1673    char flag)
1674{
1675	struct session	*loop;
1676	char		*out, *tmp, n[11];
1677
1678	RB_FOREACH(loop, sessions, &sessions) {
1679		if (*s == '\0' || strncmp(loop->name, s, strlen(s)) == 0) {
1680			*list = xreallocarray(*list, (*size) + 2,
1681			    sizeof **list);
1682			xasprintf(&(*list)[(*size)++], "%s:", loop->name);
1683		} else if (*s == '$') {
1684			xsnprintf(n, sizeof n, "%u", loop->id);
1685			if (s[1] == '\0' ||
1686			    strncmp(n, s + 1, strlen(s) - 1) == 0) {
1687				*list = xreallocarray(*list, (*size) + 2,
1688				    sizeof **list);
1689				xasprintf(&(*list)[(*size)++], "$%s:", n);
1690			}
1691		}
1692	}
1693	out = status_prompt_complete_prefix(*list, *size);
1694	if (out != NULL && flag != '\0') {
1695		xasprintf(&tmp, "-%c%s", flag, out);
1696		free(out);
1697		out = tmp;
1698	}
1699	return (out);
1700}
1701
1702/* Complete word. */
1703static char *
1704status_prompt_complete(struct client *c, const char *word, u_int offset)
1705{
1706	struct session	 *session;
1707	const char	 *s, *colon;
1708	char		**list = NULL, *copy = NULL, *out = NULL;
1709	char		  flag = '\0';
1710	u_int		  size = 0, i;
1711
1712	if (*word == '\0' &&
1713	    ((c->prompt_flags & (PROMPT_TARGET|PROMPT_WINDOW)) == 0))
1714		return (NULL);
1715
1716	if (((c->prompt_flags & (PROMPT_TARGET|PROMPT_WINDOW)) == 0) &&
1717	    strncmp(word, "-t", 2) != 0 &&
1718	    strncmp(word, "-s", 2) != 0) {
1719		list = status_prompt_complete_list(&size, word, offset == 0);
1720		if (size == 0)
1721			out = NULL;
1722		else if (size == 1)
1723			xasprintf(&out, "%s ", list[0]);
1724		else
1725			out = status_prompt_complete_prefix(list, size);
1726		goto found;
1727	}
1728
1729	if (c->prompt_flags & (PROMPT_TARGET|PROMPT_WINDOW)) {
1730		s = word;
1731		flag = '\0';
1732	} else {
1733		s = word + 2;
1734		flag = word[1];
1735		offset += 2;
1736	}
1737
1738	/* If this is a window completion, open the window menu. */
1739	if (c->prompt_flags & PROMPT_WINDOW) {
1740		out = status_prompt_complete_window_menu(c, c->session, s,
1741		    offset, '\0');
1742		goto found;
1743	}
1744	colon = strchr(s, ':');
1745
1746	/* If there is no colon, complete as a session. */
1747	if (colon == NULL) {
1748		out = status_prompt_complete_session(&list, &size, s, flag);
1749		goto found;
1750	}
1751
1752	/* If there is a colon but no period, find session and show a menu. */
1753	if (strchr(colon + 1, '.') == NULL) {
1754		if (*s == ':')
1755			session = c->session;
1756		else {
1757			copy = xstrdup(s);
1758			*strchr(copy, ':') = '\0';
1759			session = session_find(copy);
1760			free(copy);
1761			if (session == NULL)
1762				goto found;
1763		}
1764		out = status_prompt_complete_window_menu(c, session, colon + 1,
1765		    offset, flag);
1766		if (out == NULL)
1767			return (NULL);
1768	}
1769
1770found:
1771	if (size != 0) {
1772		qsort(list, size, sizeof *list, status_prompt_complete_sort);
1773		for (i = 0; i < size; i++)
1774			log_debug("complete %u: %s", i, list[i]);
1775	}
1776
1777	if (out != NULL && strcmp(word, out) == 0) {
1778		free(out);
1779		out = NULL;
1780	}
1781	if (out != NULL ||
1782	    !status_prompt_complete_list_menu(c, list, size, offset, flag)) {
1783		for (i = 0; i < size; i++)
1784			free(list[i]);
1785		free(list);
1786	}
1787	return (out);
1788}
1789