[BACK]Return to status.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / tmux

Annotation of src/usr.bin/tmux/status.c, Revision 1.231

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