status.c revision 1.152 
1/* $OpenBSD: status.c,v 1.152 2016/10/11 07:11:40 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 char	*status_redraw_get_left(struct client *, time_t,
33		     struct grid_cell *, size_t *);
34static char	*status_redraw_get_right(struct client *, time_t,
35		     struct grid_cell *, size_t *);
36static char	*status_print(struct client *, struct winlink *, time_t,
37		     struct grid_cell *);
38static char	*status_replace(struct client *, struct winlink *, const char *,
39		     time_t);
40static void	 status_message_callback(int, short, void *);
41static void	 status_timer_callback(int, short, void *);
42
43static char	*status_prompt_find_history_file(void);
44static const char *status_prompt_up_history(u_int *);
45static const char *status_prompt_down_history(u_int *);
46static void	 status_prompt_add_history(const char *);
47
48static const char **status_prompt_complete_list(u_int *, const char *);
49static char	*status_prompt_complete_prefix(const char **, u_int);
50static char	*status_prompt_complete(struct session *, const char *);
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_STATUS;
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/* Get screen line of status line. -1 means off. */
196int
197status_at_line(struct client *c)
198{
199	struct session	*s = c->session;
200
201	if (!options_get_number(s->options, "status"))
202		return (-1);
203
204	if (options_get_number(s->options, "status-position") == 0)
205		return (0);
206	return (c->tty.sy - 1);
207}
208
209/* Retrieve options for left string. */
210static char *
211status_redraw_get_left(struct client *c, time_t t, struct grid_cell *gc,
212    size_t *size)
213{
214	struct session	*s = c->session;
215	const char	*template;
216	char		*left;
217	size_t		 leftlen;
218
219	style_apply_update(gc, s->options, "status-left-style");
220
221	template = options_get_string(s->options, "status-left");
222	left = status_replace(c, NULL, template, t);
223
224	*size = options_get_number(s->options, "status-left-length");
225	leftlen = screen_write_cstrlen("%s", left);
226	if (leftlen < *size)
227		*size = leftlen;
228	return (left);
229}
230
231/* Retrieve options for right string. */
232static char *
233status_redraw_get_right(struct client *c, time_t t, struct grid_cell *gc,
234    size_t *size)
235{
236	struct session	*s = c->session;
237	const char	*template;
238	char		*right;
239	size_t		 rightlen;
240
241	style_apply_update(gc, s->options, "status-right-style");
242
243	template = options_get_string(s->options, "status-right");
244	right = status_replace(c, NULL, template, t);
245
246	*size = options_get_number(s->options, "status-right-length");
247	rightlen = screen_write_cstrlen("%s", right);
248	if (rightlen < *size)
249		*size = rightlen;
250	return (right);
251}
252
253/* Get window at window list position. */
254struct window *
255status_get_window_at(struct client *c, u_int x)
256{
257	struct session	*s = c->session;
258	struct winlink	*wl;
259	struct options	*oo;
260	const char	*sep;
261	size_t		 seplen;
262
263	x += c->wlmouse;
264	RB_FOREACH(wl, winlinks, &s->windows) {
265		oo = wl->window->options;
266
267		sep = options_get_string(oo, "window-status-separator");
268		seplen = screen_write_cstrlen("%s", sep);
269
270		if (x < wl->status_width)
271			return (wl->window);
272		x -= wl->status_width + seplen;
273	}
274	return (NULL);
275}
276
277/* Draw status for client on the last lines of given context. */
278int
279status_redraw(struct client *c)
280{
281	struct screen_write_ctx	ctx;
282	struct session	       *s = c->session;
283	struct winlink	       *wl;
284	struct screen		old_status, window_list;
285	struct grid_cell	stdgc, lgc, rgc, gc;
286	struct options	       *oo;
287	time_t			t;
288	char		       *left, *right, *sep;
289	u_int			offset, needed;
290	u_int			wlstart, wlwidth, wlavailable, wloffset, wlsize;
291	size_t			llen, rlen, seplen;
292	int			larrow, rarrow;
293
294	/* No status line? */
295	if (c->tty.sy == 0 || !options_get_number(s->options, "status"))
296		return (1);
297	left = right = NULL;
298	larrow = rarrow = 0;
299
300	/* Store current time. */
301	t = time(NULL);
302
303	/* Set up default colour. */
304	style_apply(&stdgc, s->options, "status-style");
305
306	/* Create the target screen. */
307	memcpy(&old_status, &c->status, sizeof old_status);
308	screen_init(&c->status, c->tty.sx, 1, 0);
309	screen_write_start(&ctx, NULL, &c->status);
310	for (offset = 0; offset < c->tty.sx; offset++)
311		screen_write_putc(&ctx, &stdgc, ' ');
312	screen_write_stop(&ctx);
313
314	/* If the height is one line, blank status line. */
315	if (c->tty.sy <= 1)
316		goto out;
317
318	/* Work out left and right strings. */
319	memcpy(&lgc, &stdgc, sizeof lgc);
320	left = status_redraw_get_left(c, t, &lgc, &llen);
321	memcpy(&rgc, &stdgc, sizeof rgc);
322	right = status_redraw_get_right(c, t, &rgc, &rlen);
323
324	/*
325	 * Figure out how much space we have for the window list. If there
326	 * isn't enough space, just show a blank status line.
327	 */
328	needed = 0;
329	if (llen != 0)
330		needed += llen;
331	if (rlen != 0)
332		needed += rlen;
333	if (c->tty.sx == 0 || c->tty.sx <= needed)
334		goto out;
335	wlavailable = c->tty.sx - needed;
336
337	/* Calculate the total size needed for the window list. */
338	wlstart = wloffset = wlwidth = 0;
339	RB_FOREACH(wl, winlinks, &s->windows) {
340		free(wl->status_text);
341		memcpy(&wl->status_cell, &stdgc, sizeof wl->status_cell);
342		wl->status_text = status_print(c, wl, t, &wl->status_cell);
343		wl->status_width = screen_write_cstrlen("%s", wl->status_text);
344
345		if (wl == s->curw)
346			wloffset = wlwidth;
347
348		oo = wl->window->options;
349		sep = options_get_string(oo, "window-status-separator");
350		seplen = screen_write_cstrlen("%s", sep);
351		wlwidth += wl->status_width + seplen;
352	}
353
354	/* Create a new screen for the window list. */
355	screen_init(&window_list, wlwidth, 1, 0);
356
357	/* And draw the window list into it. */
358	screen_write_start(&ctx, NULL, &window_list);
359	RB_FOREACH(wl, winlinks, &s->windows) {
360		screen_write_cnputs(&ctx, -1, &wl->status_cell, "%s",
361		    wl->status_text);
362
363		oo = wl->window->options;
364		sep = options_get_string(oo, "window-status-separator");
365		screen_write_cnputs(&ctx, -1, &stdgc, "%s", sep);
366	}
367	screen_write_stop(&ctx);
368
369	/* If there is enough space for the total width, skip to draw now. */
370	if (wlwidth <= wlavailable)
371		goto draw;
372
373	/* Find size of current window text. */
374	wlsize = s->curw->status_width;
375
376	/*
377	 * If the current window is already on screen, good to draw from the
378	 * start and just leave off the end.
379	 */
380	if (wloffset + wlsize < wlavailable) {
381		if (wlavailable > 0) {
382			rarrow = 1;
383			wlavailable--;
384		}
385		wlwidth = wlavailable;
386	} else {
387		/*
388		 * Work out how many characters we need to omit from the
389		 * start. There are wlavailable characters to fill, and
390		 * wloffset + wlsize must be the last. So, the start character
391		 * is wloffset + wlsize - wlavailable.
392		 */
393		if (wlavailable > 0) {
394			larrow = 1;
395			wlavailable--;
396		}
397
398		wlstart = wloffset + wlsize - wlavailable;
399		if (wlavailable > 0 && wlwidth > wlstart + wlavailable + 1) {
400			rarrow = 1;
401			wlstart++;
402			wlavailable--;
403		}
404		wlwidth = wlavailable;
405	}
406
407	/* Bail if anything is now too small too. */
408	if (wlwidth == 0 || wlavailable == 0) {
409		screen_free(&window_list);
410		goto out;
411	}
412
413	/*
414	 * Now the start position is known, work out the state of the left and
415	 * right arrows.
416	 */
417	offset = 0;
418	RB_FOREACH(wl, winlinks, &s->windows) {
419		if (wl->flags & WINLINK_ALERTFLAGS &&
420		    larrow == 1 && offset < wlstart)
421			larrow = -1;
422
423		offset += wl->status_width;
424
425		if (wl->flags & WINLINK_ALERTFLAGS &&
426		    rarrow == 1 && offset > wlstart + wlwidth)
427			rarrow = -1;
428	}
429
430draw:
431	/* Begin drawing. */
432	screen_write_start(&ctx, NULL, &c->status);
433
434	/* Draw the left string and arrow. */
435	screen_write_cursormove(&ctx, 0, 0);
436	if (llen != 0)
437		screen_write_cnputs(&ctx, llen, &lgc, "%s", left);
438	if (larrow != 0) {
439		memcpy(&gc, &stdgc, sizeof gc);
440		if (larrow == -1)
441			gc.attr ^= GRID_ATTR_REVERSE;
442		screen_write_putc(&ctx, &gc, '<');
443	}
444
445	/* Draw the right string and arrow. */
446	if (rarrow != 0) {
447		screen_write_cursormove(&ctx, c->tty.sx - rlen - 1, 0);
448		memcpy(&gc, &stdgc, sizeof gc);
449		if (rarrow == -1)
450			gc.attr ^= GRID_ATTR_REVERSE;
451		screen_write_putc(&ctx, &gc, '>');
452	} else
453		screen_write_cursormove(&ctx, c->tty.sx - rlen, 0);
454	if (rlen != 0)
455		screen_write_cnputs(&ctx, rlen, &rgc, "%s", right);
456
457	/* Figure out the offset for the window list. */
458	if (llen != 0)
459		wloffset = llen;
460	else
461		wloffset = 0;
462	if (wlwidth < wlavailable) {
463		switch (options_get_number(s->options, "status-justify")) {
464		case 1:	/* centred */
465			wloffset += (wlavailable - wlwidth) / 2;
466			break;
467		case 2:	/* right */
468			wloffset += (wlavailable - wlwidth);
469			break;
470		}
471	}
472	if (larrow != 0)
473		wloffset++;
474
475	/* Copy the window list. */
476	c->wlmouse = -wloffset + wlstart;
477	screen_write_cursormove(&ctx, wloffset, 0);
478	screen_write_copy(&ctx, &window_list, wlstart, 0, wlwidth, 1);
479	screen_free(&window_list);
480
481	screen_write_stop(&ctx);
482
483out:
484	free(left);
485	free(right);
486
487	if (grid_compare(c->status.grid, old_status.grid) == 0) {
488		screen_free(&old_status);
489		return (0);
490	}
491	screen_free(&old_status);
492	return (1);
493}
494
495/* Replace special sequences in fmt. */
496static char *
497status_replace(struct client *c, struct winlink *wl, const char *fmt, time_t t)
498{
499	struct format_tree	*ft;
500	char			*expanded;
501
502	if (fmt == NULL)
503		return (xstrdup(""));
504
505	if (c->flags & CLIENT_STATUSFORCE)
506		ft = format_create(NULL, FORMAT_STATUS|FORMAT_FORCE);
507	else
508		ft = format_create(NULL, FORMAT_STATUS);
509	format_defaults(ft, c, NULL, wl, NULL);
510
511	expanded = format_expand_time(ft, fmt, t);
512
513	format_free(ft);
514	return (expanded);
515}
516
517/* Return winlink status line entry and adjust gc as necessary. */
518static char *
519status_print(struct client *c, struct winlink *wl, time_t t,
520    struct grid_cell *gc)
521{
522	struct options	*oo = wl->window->options;
523	struct session	*s = c->session;
524	const char	*fmt;
525	char   		*text;
526
527	style_apply_update(gc, oo, "window-status-style");
528	fmt = options_get_string(oo, "window-status-format");
529	if (wl == s->curw) {
530		style_apply_update(gc, oo, "window-status-current-style");
531		fmt = options_get_string(oo, "window-status-current-format");
532	}
533	if (wl == TAILQ_FIRST(&s->lastw))
534		style_apply_update(gc, oo, "window-status-last-style");
535
536	if (wl->flags & WINLINK_BELL)
537		style_apply_update(gc, oo, "window-status-bell-style");
538	else if (wl->flags & (WINLINK_ACTIVITY|WINLINK_SILENCE))
539		style_apply_update(gc, oo, "window-status-activity-style");
540
541	text = status_replace(c, wl, fmt, t);
542	return (text);
543}
544
545/* Set a status line message. */
546void
547status_message_set(struct client *c, const char *fmt, ...)
548{
549	struct timeval		 tv;
550	struct message_entry	*msg, *msg1;
551	va_list			 ap;
552	int			 delay;
553	u_int			 limit;
554
555	limit = options_get_number(global_options, "message-limit");
556
557	status_prompt_clear(c);
558	status_message_clear(c);
559
560	va_start(ap, fmt);
561	xvasprintf(&c->message_string, fmt, ap);
562	va_end(ap);
563
564	msg = xcalloc(1, sizeof *msg);
565	msg->msg_time = time(NULL);
566	msg->msg_num = c->message_next++;
567	msg->msg = xstrdup(c->message_string);
568	TAILQ_INSERT_TAIL(&c->message_log, msg, entry);
569
570	TAILQ_FOREACH_SAFE(msg, &c->message_log, entry, msg1) {
571		if (msg->msg_num + limit >= c->message_next)
572			break;
573		free(msg->msg);
574		TAILQ_REMOVE(&c->message_log, msg, entry);
575		free(msg);
576	}
577
578	delay = options_get_number(c->session->options, "display-time");
579	if (delay > 0) {
580		tv.tv_sec = delay / 1000;
581		tv.tv_usec = (delay % 1000) * 1000L;
582
583		if (event_initialized(&c->message_timer))
584			evtimer_del(&c->message_timer);
585		evtimer_set(&c->message_timer, status_message_callback, c);
586		evtimer_add(&c->message_timer, &tv);
587	}
588
589	c->tty.flags |= (TTY_NOCURSOR|TTY_FREEZE);
590	c->flags |= CLIENT_STATUS;
591}
592
593/* Clear status line message. */
594void
595status_message_clear(struct client *c)
596{
597	if (c->message_string == NULL)
598		return;
599
600	free(c->message_string);
601	c->message_string = NULL;
602
603	c->tty.flags &= ~(TTY_NOCURSOR|TTY_FREEZE);
604	c->flags |= CLIENT_REDRAW; /* screen was frozen and may have changed */
605
606	screen_reinit(&c->status);
607}
608
609/* Clear status line message after timer expires. */
610static void
611status_message_callback(__unused int fd, __unused short event, void *data)
612{
613	struct client	*c = data;
614
615	status_message_clear(c);
616}
617
618/* Draw client message on status line of present else on last line. */
619int
620status_message_redraw(struct client *c)
621{
622	struct screen_write_ctx		ctx;
623	struct session		       *s = c->session;
624	struct screen		        old_status;
625	size_t			        len;
626	struct grid_cell		gc;
627
628	if (c->tty.sx == 0 || c->tty.sy == 0)
629		return (0);
630	memcpy(&old_status, &c->status, sizeof old_status);
631	screen_init(&c->status, c->tty.sx, 1, 0);
632
633	len = screen_write_strlen("%s", c->message_string);
634	if (len > c->tty.sx)
635		len = c->tty.sx;
636
637	style_apply(&gc, s->options, "message-style");
638
639	screen_write_start(&ctx, NULL, &c->status);
640
641	screen_write_cursormove(&ctx, 0, 0);
642	screen_write_nputs(&ctx, len, &gc, "%s", c->message_string);
643	for (; len < c->tty.sx; len++)
644		screen_write_putc(&ctx, &gc, ' ');
645
646	screen_write_stop(&ctx);
647
648	if (grid_compare(c->status.grid, old_status.grid) == 0) {
649		screen_free(&old_status);
650		return (0);
651	}
652	screen_free(&old_status);
653	return (1);
654}
655
656/* Enable status line prompt. */
657void
658status_prompt_set(struct client *c, const char *msg, const char *input,
659    int (*callbackfn)(void *, const char *), void (*freefn)(void *),
660    void *data, int flags)
661{
662	struct format_tree	*ft;
663	int			 keys;
664	time_t			 t;
665	char			*tmp;
666
667	ft = format_create(NULL, 0);
668	format_defaults(ft, c, NULL, NULL, NULL);
669
670	t = time(NULL);
671	tmp = format_expand_time(ft, input, t);
672
673	status_message_clear(c);
674	status_prompt_clear(c);
675
676	c->prompt_string = format_expand_time(ft, msg, t);
677
678	c->prompt_buffer = utf8_fromcstr(tmp);
679	c->prompt_index = utf8_strlen(c->prompt_buffer);
680
681	c->prompt_callbackfn = callbackfn;
682	c->prompt_freefn = freefn;
683	c->prompt_data = data;
684
685	c->prompt_hindex = 0;
686
687	c->prompt_flags = flags;
688
689	keys = options_get_number(c->session->options, "status-keys");
690	if (keys == MODEKEY_EMACS)
691		mode_key_init(&c->prompt_mdata, &mode_key_tree_emacs_edit);
692	else
693		mode_key_init(&c->prompt_mdata, &mode_key_tree_vi_edit);
694
695	c->tty.flags |= (TTY_NOCURSOR|TTY_FREEZE);
696	c->flags |= CLIENT_STATUS;
697
698	free(tmp);
699	format_free(ft);
700}
701
702/* Remove status line prompt. */
703void
704status_prompt_clear(struct client *c)
705{
706	if (c->prompt_string == NULL)
707		return;
708
709	if (c->prompt_freefn != NULL && c->prompt_data != NULL)
710		c->prompt_freefn(c->prompt_data);
711
712	free(c->prompt_string);
713	c->prompt_string = NULL;
714
715	free(c->prompt_buffer);
716	c->prompt_buffer = NULL;
717
718	c->tty.flags &= ~(TTY_NOCURSOR|TTY_FREEZE);
719	c->flags |= CLIENT_REDRAW; /* screen was frozen and may have changed */
720
721	screen_reinit(&c->status);
722}
723
724/* Update status line prompt with a new prompt string. */
725void
726status_prompt_update(struct client *c, const char *msg, const char *input)
727{
728	struct format_tree	*ft;
729	time_t			 t;
730	char			*tmp;
731
732	ft = format_create(NULL, 0);
733	format_defaults(ft, c, NULL, NULL, NULL);
734
735	t = time(NULL);
736	tmp = format_expand_time(ft, input, t);
737
738	free(c->prompt_string);
739	c->prompt_string = format_expand_time(ft, msg, t);
740
741	free(c->prompt_buffer);
742	c->prompt_buffer = utf8_fromcstr(tmp);
743	c->prompt_index = utf8_strlen(c->prompt_buffer);
744
745	c->prompt_hindex = 0;
746
747	c->flags |= CLIENT_STATUS;
748
749	free(tmp);
750	format_free(ft);
751}
752
753/* Draw client prompt on status line of present else on last line. */
754int
755status_prompt_redraw(struct client *c)
756{
757	struct screen_write_ctx	 ctx;
758	struct session		*s = c->session;
759	struct screen		 old_status;
760	u_int			 i, offset, left, start, pcursor, pwidth, width;
761	struct grid_cell	 gc, cursorgc;
762
763	if (c->tty.sx == 0 || c->tty.sy == 0)
764		return (0);
765	memcpy(&old_status, &c->status, sizeof old_status);
766	screen_init(&c->status, c->tty.sx, 1, 0);
767
768	if (c->prompt_mdata.mode == 1)
769		style_apply(&gc, s->options, "message-command-style");
770	else
771		style_apply(&gc, s->options, "message-style");
772
773	memcpy(&cursorgc, &gc, sizeof cursorgc);
774	cursorgc.attr ^= GRID_ATTR_REVERSE;
775
776	start = screen_write_strlen("%s", c->prompt_string);
777	if (start > c->tty.sx)
778		start = c->tty.sx;
779
780	screen_write_start(&ctx, NULL, &c->status);
781	screen_write_cursormove(&ctx, 0, 0);
782	screen_write_nputs(&ctx, start, &gc, "%s", c->prompt_string);
783	while (c->status.cx < screen_size_x(&c->status))
784		screen_write_putc(&ctx, &gc, ' ');
785	screen_write_cursormove(&ctx, start, 0);
786
787	left = c->tty.sx - start;
788	if (left == 0)
789		goto finished;
790
791	pcursor = utf8_strwidth(c->prompt_buffer, c->prompt_index);
792	pwidth = utf8_strwidth(c->prompt_buffer, -1);
793	if (pcursor >= left) {
794		/*
795		 * The cursor would be outside the screen so start drawing
796		 * with it on the right.
797		 */
798		offset = (pcursor - left) + 1;
799		pwidth = left;
800	} else
801		offset = 0;
802	if (pwidth > left)
803		pwidth = left;
804
805	width = 0;
806	for (i = 0; c->prompt_buffer[i].size != 0; i++) {
807		if (width < offset) {
808			width += c->prompt_buffer[i].width;
809			continue;
810		}
811		if (width >= offset + pwidth)
812			break;
813		width += c->prompt_buffer[i].width;
814		if (width > offset + pwidth)
815			break;
816
817		if (i != c->prompt_index) {
818			utf8_copy(&gc.data, &c->prompt_buffer[i]);
819			screen_write_cell(&ctx, &gc);
820		} else {
821			utf8_copy(&cursorgc.data, &c->prompt_buffer[i]);
822			screen_write_cell(&ctx, &cursorgc);
823		}
824	}
825	if (c->status.cx < screen_size_x(&c->status) && c->prompt_index >= i)
826		screen_write_putc(&ctx, &cursorgc, ' ');
827
828finished:
829	screen_write_stop(&ctx);
830
831	if (grid_compare(c->status.grid, old_status.grid) == 0) {
832		screen_free(&old_status);
833		return (0);
834	}
835	screen_free(&old_status);
836	return (1);
837}
838
839/* Is this a separator? */
840static int
841status_prompt_in_list(const char *ws, const struct utf8_data *ud)
842{
843	if (ud->size != 1 || ud->width != 1)
844		return (0);
845	return (strchr(ws, *ud->data) != NULL);
846}
847
848/* Is this a space? */
849static int
850status_prompt_space(const struct utf8_data *ud)
851{
852	if (ud->size != 1 || ud->width != 1)
853		return (0);
854	return (*ud->data == ' ');
855}
856
857/* Handle keys in prompt. */
858void
859status_prompt_key(struct client *c, key_code key)
860{
861	struct options		*oo = c->session->options;
862	struct paste_buffer	*pb;
863	char			*s, word[64];
864	const char		*histstr, *bufdata, *ws = NULL;
865	u_char			 ch;
866	size_t			 size, n, off, idx, bufsize, used;
867	struct utf8_data	 tmp, *first, *last, *ud;
868
869	size = utf8_strlen(c->prompt_buffer);
870	switch (mode_key_lookup(&c->prompt_mdata, key, NULL, NULL)) {
871	case MODEKEYEDIT_CURSORLEFT:
872		if (c->prompt_index > 0) {
873			c->prompt_index--;
874			c->flags |= CLIENT_STATUS;
875		}
876		break;
877	case MODEKEYEDIT_SWITCHMODE:
878		c->flags |= CLIENT_STATUS;
879		break;
880	case MODEKEYEDIT_SWITCHMODEAPPEND:
881		c->flags |= CLIENT_STATUS;
882		/* FALLTHROUGH */
883	case MODEKEYEDIT_CURSORRIGHT:
884		if (c->prompt_index < size) {
885			c->prompt_index++;
886			c->flags |= CLIENT_STATUS;
887		}
888		break;
889	case MODEKEYEDIT_SWITCHMODEBEGINLINE:
890		c->flags |= CLIENT_STATUS;
891		/* FALLTHROUGH */
892	case MODEKEYEDIT_STARTOFLINE:
893		if (c->prompt_index != 0) {
894			c->prompt_index = 0;
895			c->flags |= CLIENT_STATUS;
896		}
897		break;
898	case MODEKEYEDIT_SWITCHMODEAPPENDLINE:
899		c->flags |= CLIENT_STATUS;
900		/* FALLTHROUGH */
901	case MODEKEYEDIT_ENDOFLINE:
902		if (c->prompt_index != size) {
903			c->prompt_index = size;
904			c->flags |= CLIENT_STATUS;
905		}
906		break;
907	case MODEKEYEDIT_COMPLETE:
908		if (c->prompt_buffer[0].size == 0)
909			break;
910
911		idx = c->prompt_index;
912		if (idx != 0)
913			idx--;
914
915		/* Find the word we are in. */
916		first = &c->prompt_buffer[idx];
917		while (first > c->prompt_buffer && !status_prompt_space(first))
918			first--;
919		while (first->size != 0 && status_prompt_space(first))
920			first++;
921		last = &c->prompt_buffer[idx];
922		while (last->size != 0 && !status_prompt_space(last))
923			last++;
924		while (last > c->prompt_buffer && status_prompt_space(last))
925			last--;
926		if (last->size != 0)
927			last++;
928		if (last <= first)
929			break;
930
931		used = 0;
932		for (ud = first; ud < last; ud++) {
933			if (used + ud->size >= sizeof word)
934				break;
935			memcpy(word + used, ud->data, ud->size);
936			used += ud->size;
937		}
938		if (ud != last)
939			break;
940		word[used] = '\0';
941
942		/* And try to complete it. */
943		if ((s = status_prompt_complete(c->session, word)) == NULL)
944			break;
945
946		/* Trim out word. */
947		n = size - (last - c->prompt_buffer) + 1; /* with \0 */
948		memmove(first, last, n * sizeof *c->prompt_buffer);
949		size -= last - first;
950
951		/* Insert the new word. */
952		size += strlen(s);
953		off = first - c->prompt_buffer;
954		c->prompt_buffer = xreallocarray(c->prompt_buffer, size + 1,
955		    sizeof *c->prompt_buffer);
956		first = c->prompt_buffer + off;
957		memmove(first + strlen(s), first, n * sizeof *c->prompt_buffer);
958		for (idx = 0; idx < strlen(s); idx++)
959			utf8_set(&first[idx], s[idx]);
960
961		c->prompt_index = (first - c->prompt_buffer) + strlen(s);
962		free(s);
963
964		c->flags |= CLIENT_STATUS;
965		break;
966	case MODEKEYEDIT_BACKSPACE:
967		if (c->prompt_index != 0) {
968			if (c->prompt_index == size)
969				c->prompt_buffer[--c->prompt_index].size = 0;
970			else {
971				memmove(c->prompt_buffer + c->prompt_index - 1,
972				    c->prompt_buffer + c->prompt_index,
973				    (size + 1 - c->prompt_index) *
974				    sizeof *c->prompt_buffer);
975				c->prompt_index--;
976			}
977			c->flags |= CLIENT_STATUS;
978		}
979		break;
980	case MODEKEYEDIT_DELETE:
981	case MODEKEYEDIT_SWITCHMODESUBSTITUTE:
982		if (c->prompt_index != size) {
983			memmove(c->prompt_buffer + c->prompt_index,
984			    c->prompt_buffer + c->prompt_index + 1,
985			    (size + 1 - c->prompt_index) *
986			    sizeof *c->prompt_buffer);
987			c->flags |= CLIENT_STATUS;
988		}
989		break;
990	case MODEKEYEDIT_DELETELINE:
991	case MODEKEYEDIT_SWITCHMODESUBSTITUTELINE:
992		c->prompt_buffer[0].size = 0;
993		c->prompt_index = 0;
994		c->flags |= CLIENT_STATUS;
995		break;
996	case MODEKEYEDIT_DELETETOENDOFLINE:
997	case MODEKEYEDIT_SWITCHMODECHANGELINE:
998		if (c->prompt_index < size) {
999			c->prompt_buffer[c->prompt_index].size = 0;
1000			c->flags |= CLIENT_STATUS;
1001		}
1002		break;
1003	case MODEKEYEDIT_DELETEWORD:
1004		ws = options_get_string(oo, "word-separators");
1005		idx = c->prompt_index;
1006
1007		/* Find a non-separator. */
1008		while (idx != 0) {
1009			idx--;
1010			if (!status_prompt_in_list(ws, &c->prompt_buffer[idx]))
1011				break;
1012		}
1013
1014		/* Find the separator at the beginning of the word. */
1015		while (idx != 0) {
1016			idx--;
1017			if (status_prompt_in_list(ws, &c->prompt_buffer[idx])) {
1018				/* Go back to the word. */
1019				idx++;
1020				break;
1021			}
1022		}
1023
1024		memmove(c->prompt_buffer + idx,
1025		    c->prompt_buffer + c->prompt_index,
1026		    (size + 1 - c->prompt_index) *
1027		    sizeof *c->prompt_buffer);
1028		memset(c->prompt_buffer + size - (c->prompt_index - idx),
1029		    '\0', (c->prompt_index - idx) * sizeof *c->prompt_buffer);
1030		c->prompt_index = idx;
1031
1032		c->flags |= CLIENT_STATUS;
1033		break;
1034	case MODEKEYEDIT_NEXTSPACE:
1035		ws = " ";
1036		/* FALLTHROUGH */
1037	case MODEKEYEDIT_NEXTWORD:
1038		if (ws == NULL)
1039			ws = options_get_string(oo, "word-separators");
1040
1041		/* Find a separator. */
1042		while (c->prompt_index != size) {
1043			idx = ++c->prompt_index;
1044			if (status_prompt_in_list(ws, &c->prompt_buffer[idx]))
1045				break;
1046		}
1047
1048		/* Find the word right after the separator. */
1049		while (c->prompt_index != size) {
1050			idx = ++c->prompt_index;
1051			if (!status_prompt_in_list(ws, &c->prompt_buffer[idx]))
1052				break;
1053		}
1054
1055		c->flags |= CLIENT_STATUS;
1056		break;
1057	case MODEKEYEDIT_NEXTSPACEEND:
1058		ws = " ";
1059		/* FALLTHROUGH */
1060	case MODEKEYEDIT_NEXTWORDEND:
1061		if (ws == NULL)
1062			ws = options_get_string(oo, "word-separators");
1063
1064		/* Find a word. */
1065		while (c->prompt_index != size) {
1066			idx = ++c->prompt_index;
1067			if (!status_prompt_in_list(ws, &c->prompt_buffer[idx]))
1068				break;
1069		}
1070
1071		/* Find the separator at the end of the word. */
1072		while (c->prompt_index != size) {
1073			idx = ++c->prompt_index;
1074			if (status_prompt_in_list(ws, &c->prompt_buffer[idx]))
1075				break;
1076		}
1077
1078		/* Back up to the end-of-word like vi. */
1079		if (options_get_number(oo, "status-keys") == MODEKEY_VI &&
1080		    c->prompt_index != 0)
1081			c->prompt_index--;
1082
1083		c->flags |= CLIENT_STATUS;
1084		break;
1085	case MODEKEYEDIT_PREVIOUSSPACE:
1086		ws = " ";
1087		/* FALLTHROUGH */
1088	case MODEKEYEDIT_PREVIOUSWORD:
1089		if (ws == NULL)
1090			ws = options_get_string(oo, "word-separators");
1091
1092		/* Find a non-separator. */
1093		while (c->prompt_index != 0) {
1094			idx = --c->prompt_index;
1095			if (!status_prompt_in_list(ws, &c->prompt_buffer[idx]))
1096				break;
1097		}
1098
1099		/* Find the separator at the beginning of the word. */
1100		while (c->prompt_index != 0) {
1101			idx = --c->prompt_index;
1102			if (status_prompt_in_list(ws, &c->prompt_buffer[idx])) {
1103				/* Go back to the word. */
1104				c->prompt_index++;
1105				break;
1106			}
1107		}
1108
1109		c->flags |= CLIENT_STATUS;
1110		break;
1111	case MODEKEYEDIT_HISTORYUP:
1112		histstr = status_prompt_up_history(&c->prompt_hindex);
1113		if (histstr == NULL)
1114			break;
1115		free(c->prompt_buffer);
1116		c->prompt_buffer = utf8_fromcstr(histstr);
1117		c->prompt_index = utf8_strlen(c->prompt_buffer);
1118		c->flags |= CLIENT_STATUS;
1119		break;
1120	case MODEKEYEDIT_HISTORYDOWN:
1121		histstr = status_prompt_down_history(&c->prompt_hindex);
1122		if (histstr == NULL)
1123			break;
1124		free(c->prompt_buffer);
1125		c->prompt_buffer = utf8_fromcstr(histstr);
1126		c->prompt_index = utf8_strlen(c->prompt_buffer);
1127		c->flags |= CLIENT_STATUS;
1128		break;
1129	case MODEKEYEDIT_PASTE:
1130		if ((pb = paste_get_top(NULL)) == NULL)
1131			break;
1132		bufdata = paste_buffer_data(pb, &bufsize);
1133		for (n = 0; n < bufsize; n++) {
1134			ch = (u_char)bufdata[n];
1135			if (ch < 32 || ch >= 127)
1136				break;
1137		}
1138
1139		c->prompt_buffer = xreallocarray(c->prompt_buffer, size + n + 1,
1140		    sizeof *c->prompt_buffer);
1141		if (c->prompt_index == size) {
1142			for (idx = 0; idx < n; idx++) {
1143				ud = &c->prompt_buffer[c->prompt_index + idx];
1144				utf8_set(ud, bufdata[idx]);
1145			}
1146			c->prompt_index += n;
1147			c->prompt_buffer[c->prompt_index].size = 0;
1148		} else {
1149			memmove(c->prompt_buffer + c->prompt_index + n,
1150			    c->prompt_buffer + c->prompt_index,
1151			    (size + 1 - c->prompt_index) *
1152			    sizeof *c->prompt_buffer);
1153			for (idx = 0; idx < n; idx++) {
1154				ud = &c->prompt_buffer[c->prompt_index + idx];
1155				utf8_set(ud, bufdata[idx]);
1156			}
1157			c->prompt_index += n;
1158		}
1159
1160		c->flags |= CLIENT_STATUS;
1161		break;
1162	case MODEKEYEDIT_TRANSPOSECHARS:
1163		idx = c->prompt_index;
1164		if (idx < size)
1165			idx++;
1166		if (idx >= 2) {
1167			utf8_copy(&tmp, &c->prompt_buffer[idx - 2]);
1168			utf8_copy(&c->prompt_buffer[idx - 2],
1169			    &c->prompt_buffer[idx - 1]);
1170			utf8_copy(&c->prompt_buffer[idx - 1], &tmp);
1171			c->prompt_index = idx;
1172			c->flags |= CLIENT_STATUS;
1173		}
1174		break;
1175	case MODEKEYEDIT_ENTER:
1176		s = utf8_tocstr(c->prompt_buffer);
1177		if (*s != '\0')
1178			status_prompt_add_history(s);
1179		if (c->prompt_callbackfn(c->prompt_data, s) == 0)
1180			status_prompt_clear(c);
1181		free(s);
1182		break;
1183	case MODEKEYEDIT_CANCEL:
1184		if (c->prompt_callbackfn(c->prompt_data, NULL) == 0)
1185			status_prompt_clear(c);
1186		break;
1187	case MODEKEY_OTHER:
1188		if (key <= 0x1f || key >= KEYC_BASE)
1189			break;
1190		if (utf8_split(key, &tmp) != UTF8_DONE)
1191			break;
1192
1193		c->prompt_buffer = xreallocarray(c->prompt_buffer, size + 2,
1194		    sizeof *c->prompt_buffer);
1195
1196		if (c->prompt_index == size) {
1197			utf8_copy(&c->prompt_buffer[c->prompt_index], &tmp);
1198			c->prompt_index++;
1199			c->prompt_buffer[c->prompt_index].size = 0;
1200		} else {
1201			memmove(c->prompt_buffer + c->prompt_index + 1,
1202			    c->prompt_buffer + c->prompt_index,
1203			    (size + 1 - c->prompt_index) *
1204			    sizeof *c->prompt_buffer);
1205			utf8_copy(&c->prompt_buffer[c->prompt_index], &tmp);
1206			c->prompt_index++;
1207		}
1208
1209		if (c->prompt_flags & PROMPT_SINGLE) {
1210			s = utf8_tocstr(c->prompt_buffer);
1211			if (strlen(s) != 1)
1212				status_prompt_clear(c);
1213			else if (c->prompt_callbackfn(c->prompt_data, s) == 0)
1214				status_prompt_clear(c);
1215			free(s);
1216		}
1217
1218		c->flags |= CLIENT_STATUS;
1219		break;
1220	default:
1221		break;
1222	}
1223}
1224
1225/* Get previous line from the history. */
1226static const char *
1227status_prompt_up_history(u_int *idx)
1228{
1229	/*
1230	 * History runs from 0 to size - 1. Index is from 0 to size. Zero is
1231	 * empty.
1232	 */
1233
1234	if (status_prompt_hsize == 0 || *idx == status_prompt_hsize)
1235		return (NULL);
1236	(*idx)++;
1237	return (status_prompt_hlist[status_prompt_hsize - *idx]);
1238}
1239
1240/* Get next line from the history. */
1241static const char *
1242status_prompt_down_history(u_int *idx)
1243{
1244	if (status_prompt_hsize == 0 || *idx == 0)
1245		return ("");
1246	(*idx)--;
1247	if (*idx == 0)
1248		return ("");
1249	return (status_prompt_hlist[status_prompt_hsize - *idx]);
1250}
1251
1252/* Add line to the history. */
1253static void
1254status_prompt_add_history(const char *line)
1255{
1256	size_t	size;
1257
1258	if (status_prompt_hsize > 0 &&
1259	    strcmp(status_prompt_hlist[status_prompt_hsize - 1], line) == 0)
1260		return;
1261
1262	if (status_prompt_hsize == PROMPT_HISTORY) {
1263		free(status_prompt_hlist[0]);
1264
1265		size = (PROMPT_HISTORY - 1) * sizeof *status_prompt_hlist;
1266		memmove(&status_prompt_hlist[0], &status_prompt_hlist[1], size);
1267
1268		status_prompt_hlist[status_prompt_hsize - 1] = xstrdup(line);
1269		return;
1270	}
1271
1272	status_prompt_hlist = xreallocarray(status_prompt_hlist,
1273	    status_prompt_hsize + 1, sizeof *status_prompt_hlist);
1274	status_prompt_hlist[status_prompt_hsize++] = xstrdup(line);
1275}
1276
1277/* Build completion list. */
1278static const char **
1279status_prompt_complete_list(u_int *size, const char *s)
1280{
1281	const char				**list = NULL, **layout;
1282	const struct cmd_entry			**cmdent;
1283	const struct options_table_entry	 *oe;
1284	const char				 *layouts[] = {
1285		"even-horizontal", "even-vertical", "main-horizontal",
1286		"main-vertical", "tiled", NULL
1287	};
1288
1289	*size = 0;
1290	for (cmdent = cmd_table; *cmdent != NULL; cmdent++) {
1291		if (strncmp((*cmdent)->name, s, strlen(s)) == 0) {
1292			list = xreallocarray(list, (*size) + 1, sizeof *list);
1293			list[(*size)++] = (*cmdent)->name;
1294		}
1295	}
1296	for (oe = options_table; oe->name != NULL; oe++) {
1297		if (strncmp(oe->name, s, strlen(s)) == 0) {
1298			list = xreallocarray(list, (*size) + 1, sizeof *list);
1299			list[(*size)++] = oe->name;
1300		}
1301	}
1302	for (layout = layouts; *layout != NULL; layout++) {
1303		if (strncmp(*layout, s, strlen(s)) == 0) {
1304			list = xreallocarray(list, (*size) + 1, sizeof *list);
1305			list[(*size)++] = *layout;
1306		}
1307	}
1308	return (list);
1309}
1310
1311/* Find longest prefix. */
1312static char *
1313status_prompt_complete_prefix(const char **list, u_int size)
1314{
1315	char	 *out;
1316	u_int	  i;
1317	size_t	  j;
1318
1319	out = xstrdup(list[0]);
1320	for (i = 1; i < size; i++) {
1321		j = strlen(list[i]);
1322		if (j > strlen(out))
1323			j = strlen(out);
1324		for (; j > 0; j--) {
1325			if (out[j - 1] != list[i][j - 1])
1326				out[j - 1] = '\0';
1327		}
1328	}
1329	return (out);
1330}
1331
1332/* Complete word. */
1333static char *
1334status_prompt_complete(struct session *session, const char *s)
1335{
1336	const char	**list = NULL, *colon;
1337	u_int		  size = 0, i;
1338	struct session	 *s_loop;
1339	struct winlink	 *wl;
1340	struct window	 *w;
1341	char		 *copy, *out, *tmp;
1342
1343	if (*s == '\0')
1344		return (NULL);
1345	out = NULL;
1346
1347	if (strncmp(s, "-t", 2) != 0 && strncmp(s, "-s", 2) != 0) {
1348		list = status_prompt_complete_list(&size, s);
1349		if (size == 0)
1350			out = NULL;
1351		else if (size == 1)
1352			xasprintf(&out, "%s ", list[0]);
1353		else
1354			out = status_prompt_complete_prefix(list, size);
1355		free(list);
1356		return (out);
1357	}
1358	copy = xstrdup(s);
1359
1360	colon = ":";
1361	if (copy[strlen(copy) - 1] == ':')
1362		copy[strlen(copy) - 1] = '\0';
1363	else
1364		colon = "";
1365	s = copy + 2;
1366
1367	RB_FOREACH(s_loop, sessions, &sessions) {
1368		if (strncmp(s_loop->name, s, strlen(s)) == 0) {
1369			list = xreallocarray(list, size + 2, sizeof *list);
1370			list[size++] = s_loop->name;
1371		}
1372	}
1373	if (size == 1) {
1374		out = xstrdup(list[0]);
1375		if (session_find(list[0]) != NULL)
1376			colon = ":";
1377	} else if (size != 0)
1378		out = status_prompt_complete_prefix(list, size);
1379	if (out != NULL) {
1380		xasprintf(&tmp, "-%c%s%s", copy[1], out, colon);
1381		out = tmp;
1382		goto found;
1383	}
1384
1385	colon = "";
1386	if (*s == ':') {
1387		RB_FOREACH(wl, winlinks, &session->windows) {
1388			xasprintf(&tmp, ":%s", wl->window->name);
1389			if (strncmp(tmp, s, strlen(s)) == 0){
1390				list = xreallocarray(list, size + 1,
1391				    sizeof *list);
1392				list[size++] = tmp;
1393				continue;
1394			}
1395			free(tmp);
1396
1397			xasprintf(&tmp, ":%d", wl->idx);
1398			if (strncmp(tmp, s, strlen(s)) == 0) {
1399				list = xreallocarray(list, size + 1,
1400				    sizeof *list);
1401				list[size++] = tmp;
1402				continue;
1403			}
1404			free(tmp);
1405		}
1406	} else {
1407		RB_FOREACH(s_loop, sessions, &sessions) {
1408			RB_FOREACH(wl, winlinks, &s_loop->windows) {
1409				w = wl->window;
1410
1411				xasprintf(&tmp, "%s:%s", s_loop->name, w->name);
1412				if (strncmp(tmp, s, strlen(s)) == 0) {
1413					list = xreallocarray(list, size + 1,
1414					    sizeof *list);
1415					list[size++] = tmp;
1416					continue;
1417				}
1418				free(tmp);
1419
1420				xasprintf(&tmp, "%s:%d", s_loop->name, wl->idx);
1421				if (strncmp(tmp, s, strlen(s)) == 0) {
1422					list = xreallocarray(list, size + 1,
1423					    sizeof *list);
1424					list[size++] = tmp;
1425					continue;
1426				}
1427				free(tmp);
1428			}
1429		}
1430	}
1431	if (size == 1) {
1432		out = xstrdup(list[0]);
1433		colon = " ";
1434	} else if (size != 0)
1435		out = status_prompt_complete_prefix(list, size);
1436	if (out != NULL) {
1437		xasprintf(&tmp, "-%c%s%s", copy[1], out, colon);
1438		out = tmp;
1439	}
1440
1441	for (i = 0; i < size; i++)
1442		free((void *)list[i]);
1443
1444found:
1445	free(copy);
1446	free(list);
1447	return (out);
1448}
1449