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

Annotation of src/usr.bin/tmux/window-choose.c, Revision 1.82

1.82    ! nicm        1: /* $OpenBSD: window-choose.c,v 1.81 2016/10/16 17:55:14 nicm Exp $ */
1.1       nicm        2:
                      3: /*
1.74      nicm        4:  * Copyright (c) 2009 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:
1.19      nicm       21: #include <ctype.h>
1.21      nicm       22: #include <stdlib.h>
1.1       nicm       23: #include <string.h>
                     24:
                     25: #include "tmux.h"
                     26:
1.79      nicm       27: static struct screen *window_choose_init(struct window_pane *);
                     28: static void    window_choose_free(struct window_pane *);
                     29: static void    window_choose_resize(struct window_pane *, u_int, u_int);
                     30: static void    window_choose_key(struct window_pane *, struct client *,
                     31:                    struct session *, key_code, struct mouse_event *);
                     32:
                     33: static void    window_choose_default_callback(struct window_choose_data *);
                     34: static struct window_choose_mode_item *window_choose_get_item(
                     35:                    struct window_pane *, key_code, struct mouse_event *);
                     36:
                     37: static void    window_choose_fire_callback(struct window_pane *,
                     38:                    struct window_choose_data *);
                     39: static void    window_choose_redraw_screen(struct window_pane *);
                     40: static void    window_choose_write_line(struct window_pane *,
                     41:                    struct screen_write_ctx *, u_int);
                     42:
                     43: static void    window_choose_scroll_up(struct window_pane *);
                     44: static void    window_choose_scroll_down(struct window_pane *);
                     45:
                     46: static void    window_choose_collapse(struct window_pane *, struct session *,
                     47:                    u_int);
                     48: static void    window_choose_expand(struct window_pane *, struct session *,
                     49:                    u_int);
                     50: static void    window_choose_collapse_all(struct window_pane *);
1.1       nicm       51:
1.79      nicm       52: static void    window_choose_data_free(struct window_choose_data *);
1.27      nicm       53:
1.22      nicm       54: enum window_choose_input_type {
                     55:        WINDOW_CHOOSE_NORMAL = -1,
                     56:        WINDOW_CHOOSE_GOTO_ITEM,
                     57: };
                     58:
1.1       nicm       59: const struct window_mode window_choose_mode = {
1.78      nicm       60:        .init = window_choose_init,
                     61:        .free = window_choose_free,
                     62:        .resize = window_choose_resize,
                     63:        .key = window_choose_key,
1.64      nicm       64: };
                     65:
                     66: struct window_choose_mode_item {
                     67:        struct window_choose_data       *wcd;
                     68:        char                            *name;
                     69:        int                              pos;
                     70:        int                              state;
                     71: #define TREE_EXPANDED 0x1
1.1       nicm       72: };
                     73:
                     74: struct window_choose_mode_data {
                     75:        struct screen           screen;
                     76:
                     77:        struct mode_key_data    mdata;
                     78:
1.75      nicm       79:        struct window_choose_mode_item *list;
                     80:        u_int                   list_size;
                     81:        struct window_choose_mode_item *old_list;
                     82:        u_int                   old_list_size;
                     83:
1.22      nicm       84:        int                     width;
1.1       nicm       85:        u_int                   top;
                     86:        u_int                   selected;
1.22      nicm       87:        enum window_choose_input_type input_type;
                     88:        const char              *input_prompt;
                     89:        char                    *input_str;
1.1       nicm       90:
1.19      nicm       91:        void                    (*callbackfn)(struct window_choose_data *);
1.1       nicm       92: };
                     93:
1.79      nicm       94: static void    window_choose_free1(struct window_choose_mode_data *);
                     95: static int     window_choose_key_index(struct window_choose_mode_data *,
                     96:                    u_int);
                     97: static int     window_choose_index_key(struct window_choose_mode_data *,
                     98:                    key_code);
                     99: static void    window_choose_prompt_input(enum window_choose_input_type,
                    100:                    const char *, struct window_pane *, key_code);
                    101: static void    window_choose_reset_top(struct window_pane *, u_int);
1.11      nicm      102:
1.1       nicm      103: void
1.19      nicm      104: window_choose_add(struct window_pane *wp, struct window_choose_data *wcd)
1.1       nicm      105: {
                    106:        struct window_choose_mode_data  *data = wp->modedata;
                    107:        struct window_choose_mode_item  *item;
1.75      nicm      108:        char                             tmp[11];
1.1       nicm      109:
1.75      nicm      110:        data->list = xreallocarray(data->list, data->list_size + 1,
                    111:            sizeof *data->list);
                    112:        item = &data->list[data->list_size++];
1.1       nicm      113:
1.19      nicm      114:        item->name = format_expand(wcd->ft, wcd->ft_template);
                    115:        item->wcd = wcd;
1.75      nicm      116:        item->pos = data->list_size - 1;
1.27      nicm      117:        item->state = 0;
1.22      nicm      118:
1.75      nicm      119:        data->width = xsnprintf(tmp, sizeof tmp, "%d", item->pos);
1.1       nicm      120: }
                    121:
                    122: void
1.47      nicm      123: window_choose_set_current(struct window_pane *wp, u_int cur)
1.1       nicm      124: {
                    125:        struct window_choose_mode_data  *data = wp->modedata;
                    126:        struct screen                   *s = &data->screen;
                    127:
                    128:        data->selected = cur;
1.48      nicm      129:        window_choose_reset_top(wp, screen_size_y(s));
                    130: }
                    131:
1.79      nicm      132: static void
1.48      nicm      133: window_choose_reset_top(struct window_pane *wp, u_int sy)
                    134: {
                    135:        struct window_choose_mode_data  *data = wp->modedata;
                    136:
                    137:        data->top = 0;
                    138:        if (data->selected > sy - 1)
                    139:                data->top = data->selected - (sy - 1);
1.1       nicm      140:
1.47      nicm      141:        window_choose_redraw_screen(wp);
                    142: }
                    143:
                    144: void
                    145: window_choose_ready(struct window_pane *wp, u_int cur,
                    146:     void (*callbackfn)(struct window_choose_data *))
                    147: {
                    148:        struct window_choose_mode_data  *data = wp->modedata;
1.75      nicm      149:        u_int                            size;
1.47      nicm      150:
1.3       nicm      151:        data->callbackfn = callbackfn;
1.33      nicm      152:        if (data->callbackfn == NULL)
                    153:                data->callbackfn = window_choose_default_callback;
1.1       nicm      154:
1.75      nicm      155:        size = data->old_list_size;
                    156:        data->old_list_size += data->list_size;
                    157:        data->old_list = xreallocarray(data->old_list, data->old_list_size,
                    158:            sizeof *data->old_list);
                    159:        memcpy(data->old_list + size, data->list, data->list_size *
                    160:            sizeof *data->list);
1.27      nicm      161:
1.47      nicm      162:        window_choose_set_current(wp, cur);
1.27      nicm      163:        window_choose_collapse_all(wp);
1.1       nicm      164: }
                    165:
1.79      nicm      166: static struct screen *
1.1       nicm      167: window_choose_init(struct window_pane *wp)
                    168: {
                    169:        struct window_choose_mode_data  *data;
                    170:        struct screen                   *s;
1.5       nicm      171:        int                              keys;
1.1       nicm      172:
1.75      nicm      173:        wp->modedata = data = xcalloc(1, sizeof *data);
1.3       nicm      174:
                    175:        data->callbackfn = NULL;
1.22      nicm      176:        data->input_type = WINDOW_CHOOSE_NORMAL;
                    177:        data->input_str = xstrdup("");
                    178:        data->input_prompt = NULL;
1.3       nicm      179:
1.75      nicm      180:        data->list = NULL;
                    181:        data->list_size = 0;
                    182:
                    183:        data->old_list = NULL;
                    184:        data->old_list_size = 0;
                    185:
1.1       nicm      186:        data->top = 0;
                    187:
                    188:        s = &data->screen;
                    189:        screen_init(s, screen_size_x(&wp->base), screen_size_y(&wp->base), 0);
                    190:        s->mode &= ~MODE_CURSOR;
                    191:
1.67      nicm      192:        keys = options_get_number(wp->window->options, "mode-keys");
1.5       nicm      193:        if (keys == MODEKEY_EMACS)
1.6       nicm      194:                mode_key_init(&data->mdata, &mode_key_tree_emacs_choice);
1.5       nicm      195:        else
1.6       nicm      196:                mode_key_init(&data->mdata, &mode_key_tree_vi_choice);
1.1       nicm      197:
                    198:        return (s);
                    199: }
                    200:
1.19      nicm      201: struct window_choose_data *
1.33      nicm      202: window_choose_data_create(int type, struct client *c, struct session *s)
1.19      nicm      203: {
                    204:        struct window_choose_data       *wcd;
                    205:
                    206:        wcd = xmalloc(sizeof *wcd);
1.33      nicm      207:        wcd->type = type;
                    208:
1.72      nicm      209:        wcd->ft = format_create(NULL, 0);
1.19      nicm      210:        wcd->ft_template = NULL;
1.33      nicm      211:
1.20      nicm      212:        wcd->command = NULL;
1.33      nicm      213:
1.24      nicm      214:        wcd->wl = NULL;
1.33      nicm      215:        wcd->pane_id = -1;
                    216:        wcd->idx = -1;
                    217:
1.27      nicm      218:        wcd->tree_session = NULL;
1.33      nicm      219:
                    220:        wcd->start_client = c;
                    221:        wcd->start_client->references++;
                    222:        wcd->start_session = s;
                    223:        wcd->start_session->references++;
1.19      nicm      224:
                    225:        return (wcd);
                    226: }
                    227:
1.79      nicm      228: static void
1.33      nicm      229: window_choose_data_free(struct window_choose_data *wcd)
                    230: {
1.65      nicm      231:        server_client_unref(wcd->start_client);
                    232:        session_unref(wcd->start_session);
1.33      nicm      233:
                    234:        if (wcd->tree_session != NULL)
1.65      nicm      235:                session_unref(wcd->tree_session);
1.33      nicm      236:
                    237:        free(wcd->ft_template);
                    238:        format_free(wcd->ft);
                    239:
                    240:        free(wcd->command);
                    241:        free(wcd);
                    242: }
                    243:
                    244: void
                    245: window_choose_data_run(struct window_choose_data *cdata)
                    246: {
1.82    ! nicm      247:        struct cmd_list         *cmdlist;
        !           248:        char                    *cause;
        !           249:        struct cmdq_item        *item;
1.33      nicm      250:
                    251:        /*
                    252:         * The command template will have already been replaced. But if it's
                    253:         * NULL, bail here.
                    254:         */
                    255:        if (cdata->command == NULL)
                    256:                return;
                    257:
1.45      nicm      258:        if (cmd_string_parse(cdata->command, &cmdlist, NULL, 0, &cause) != 0) {
1.33      nicm      259:                if (cause != NULL) {
                    260:                        *cause = toupper((u_char) *cause);
                    261:                        status_message_set(cdata->start_client, "%s", cause);
                    262:                        free(cause);
                    263:                }
                    264:                return;
                    265:        }
                    266:
1.82    ! nicm      267:        item = cmdq_get_command(cmdlist, NULL, NULL, 0);
        !           268:        cmdq_append(cdata->start_client, item);
1.33      nicm      269:        cmd_list_free(cmdlist);
                    270: }
                    271:
1.79      nicm      272: static void
1.33      nicm      273: window_choose_default_callback(struct window_choose_data *wcd)
                    274: {
                    275:        if (wcd == NULL)
                    276:                return;
                    277:        if (wcd->start_client->flags & CLIENT_DEAD)
                    278:                return;
                    279:
                    280:        window_choose_data_run(wcd);
                    281: }
                    282:
1.79      nicm      283: static void
1.1       nicm      284: window_choose_free(struct window_pane *wp)
                    285: {
1.38      nicm      286:        if (wp->modedata != NULL)
                    287:                window_choose_free1(wp->modedata);
                    288: }
                    289:
1.79      nicm      290: static void
1.38      nicm      291: window_choose_free1(struct window_choose_mode_data *data)
                    292: {
1.19      nicm      293:        struct window_choose_mode_item  *item;
1.1       nicm      294:        u_int                            i;
                    295:
1.38      nicm      296:        if (data == NULL)
                    297:                return;
                    298:
1.75      nicm      299:        for (i = 0; i < data->old_list_size; i++) {
                    300:                item = &data->old_list[i];
1.33      nicm      301:                window_choose_data_free(item->wcd);
1.21      nicm      302:                free(item->name);
1.19      nicm      303:        }
1.75      nicm      304:        free(data->list);
                    305:        free(data->old_list);
                    306:
1.22      nicm      307:        free(data->input_str);
1.1       nicm      308:
                    309:        screen_free(&data->screen);
1.21      nicm      310:        free(data);
1.1       nicm      311: }
                    312:
1.79      nicm      313: static void
1.1       nicm      314: window_choose_resize(struct window_pane *wp, u_int sx, u_int sy)
                    315: {
                    316:        struct window_choose_mode_data  *data = wp->modedata;
                    317:        struct screen                   *s = &data->screen;
                    318:
1.48      nicm      319:        window_choose_reset_top(wp, sy);
1.32      nicm      320:        screen_resize(s, sx, sy, 0);
1.1       nicm      321:        window_choose_redraw_screen(wp);
                    322: }
                    323:
1.79      nicm      324: static void
1.73      nicm      325: window_choose_fire_callback(struct window_pane *wp,
                    326:     struct window_choose_data *wcd)
1.18      nicm      327: {
                    328:        struct window_choose_mode_data  *data = wp->modedata;
                    329:
1.38      nicm      330:        wp->modedata = NULL;
                    331:        window_pane_reset_mode(wp);
1.18      nicm      332:
1.19      nicm      333:        data->callbackfn(wcd);
1.18      nicm      334:
1.38      nicm      335:        window_choose_free1(data);
1.18      nicm      336: }
                    337:
1.79      nicm      338: static void
1.22      nicm      339: window_choose_prompt_input(enum window_choose_input_type input_type,
1.68      nicm      340:     const char *prompt, struct window_pane *wp, key_code key)
1.22      nicm      341: {
                    342:        struct window_choose_mode_data  *data = wp->modedata;
                    343:        size_t                           input_len;
                    344:
                    345:        data->input_type = input_type;
                    346:        data->input_prompt = prompt;
                    347:        input_len = strlen(data->input_str) + 2;
                    348:
1.55      nicm      349:        data->input_str = xrealloc(data->input_str, input_len);
1.22      nicm      350:        data->input_str[input_len - 2] = key;
                    351:        data->input_str[input_len - 1] = '\0';
                    352:
                    353:        window_choose_redraw_screen(wp);
                    354: }
                    355:
1.79      nicm      356: static void
1.61      nicm      357: window_choose_collapse(struct window_pane *wp, struct session *s, u_int pos)
1.27      nicm      358: {
                    359:        struct window_choose_mode_data  *data = wp->modedata;
1.75      nicm      360:        struct window_choose_mode_item  *item, *chosen, *copy = NULL;
1.27      nicm      361:        struct window_choose_data       *wcd;
1.75      nicm      362:        u_int                            i, copy_size = 0;
1.27      nicm      363:
1.75      nicm      364:        chosen = &data->list[pos];
1.27      nicm      365:        chosen->state &= ~TREE_EXPANDED;
                    366:
                    367:        /*
                    368:         * Trying to mangle the &data->list in-place has lots of problems, so
                    369:         * assign the actual result we want to render and copy the new one over
                    370:         * the top of it.
                    371:         */
1.75      nicm      372:        for (i = 0; i < data->list_size; i++) {
                    373:                item = &data->list[i];
1.27      nicm      374:                wcd = item->wcd;
                    375:
                    376:                if (s == wcd->tree_session) {
                    377:                        /* We only show the session when collapsed. */
                    378:                        if (wcd->type & TREE_SESSION) {
                    379:                                item->state &= ~TREE_EXPANDED;
1.75      nicm      380:
                    381:                                copy = xreallocarray(copy, copy_size + 1,
                    382:                                    sizeof *copy);
                    383:                                memcpy(&copy[copy_size], item, sizeof *copy);
                    384:                                copy_size++;
1.27      nicm      385:
                    386:                                /*
                    387:                                 * Update the selection to this session item so
                    388:                                 * we don't end up highlighting a non-existent
                    389:                                 * item.
                    390:                                 */
                    391:                                data->selected = i;
                    392:                        }
1.75      nicm      393:                } else {
                    394:                        copy = xreallocarray(copy, copy_size + 1, sizeof *copy);
                    395:                        memcpy(&copy[copy_size], item, sizeof *copy);
                    396:                        copy_size++;
                    397:                }
1.27      nicm      398:        }
                    399:
1.75      nicm      400:        if (copy_size != 0) {
                    401:                free(data->list);
                    402:                data->list = copy;
                    403:                data->list_size = copy_size;
1.27      nicm      404:        }
                    405: }
                    406:
1.79      nicm      407: static void
1.27      nicm      408: window_choose_collapse_all(struct window_pane *wp)
                    409: {
                    410:        struct window_choose_mode_data  *data = wp->modedata;
1.28      nicm      411:        struct window_choose_mode_item  *item;
1.48      nicm      412:        struct screen                   *scr = &data->screen;
1.28      nicm      413:        struct session                  *s, *chosen;
1.27      nicm      414:        u_int                            i;
                    415:
1.75      nicm      416:        chosen = data->list[data->selected].wcd->start_session;
1.27      nicm      417:
                    418:        RB_FOREACH(s, sessions, &sessions)
1.61      nicm      419:                window_choose_collapse(wp, s, data->selected);
1.27      nicm      420:
                    421:        /* Reset the selection back to the starting session. */
1.75      nicm      422:        for (i = 0; i < data->list_size; i++) {
                    423:                item = &data->list[i];
1.27      nicm      424:
1.28      nicm      425:                if (chosen != item->wcd->tree_session)
1.27      nicm      426:                        continue;
                    427:
                    428:                if (item->wcd->type & TREE_SESSION)
                    429:                        data->selected = i;
                    430:        }
1.48      nicm      431:        window_choose_reset_top(wp, screen_size_y(scr));
1.27      nicm      432: }
                    433:
                    434: void
                    435: window_choose_expand_all(struct window_pane *wp)
                    436: {
                    437:        struct window_choose_mode_data  *data = wp->modedata;
                    438:        struct window_choose_mode_item  *item;
1.48      nicm      439:        struct screen                   *scr = &data->screen;
1.27      nicm      440:        struct session                  *s;
                    441:        u_int                            i;
                    442:
                    443:        RB_FOREACH(s, sessions, &sessions) {
1.75      nicm      444:                for (i = 0; i < data->list_size; i++) {
                    445:                        item = &data->list[i];
1.27      nicm      446:
                    447:                        if (s != item->wcd->tree_session)
                    448:                                continue;
                    449:
                    450:                        if (item->wcd->type & TREE_SESSION)
                    451:                                window_choose_expand(wp, s, i);
                    452:                }
                    453:        }
                    454:
1.48      nicm      455:        window_choose_reset_top(wp, screen_size_y(scr));
1.27      nicm      456: }
                    457:
1.79      nicm      458: static void
1.27      nicm      459: window_choose_expand(struct window_pane *wp, struct session *s, u_int pos)
                    460: {
                    461:        struct window_choose_mode_data  *data = wp->modedata;
                    462:        struct window_choose_mode_item  *item, *chosen;
                    463:        struct window_choose_data       *wcd;
                    464:        u_int                            i, items;
                    465:
1.75      nicm      466:        chosen = &data->list[pos];
                    467:        items = data->old_list_size - 1;
1.27      nicm      468:
                    469:        /* It's not possible to expand anything other than sessions. */
                    470:        if (!(chosen->wcd->type & TREE_SESSION))
                    471:                return;
                    472:
                    473:        /* Don't re-expand a session which is already expanded. */
                    474:        if (chosen->state & TREE_EXPANDED)
                    475:                return;
                    476:
                    477:        /* Mark the session entry as expanded. */
                    478:        chosen->state |= TREE_EXPANDED;
                    479:
                    480:        /*
                    481:         * Go back through the original list of all sessions and windows, and
                    482:         * pull out the windows where the session matches the selection chosen
                    483:         * to expand.
                    484:         */
                    485:        for (i = items; i > 0; i--) {
1.75      nicm      486:                item = &data->old_list[i];
1.27      nicm      487:                item->state |= TREE_EXPANDED;
                    488:                wcd = item->wcd;
                    489:
                    490:                if (s == wcd->tree_session) {
                    491:                        /*
                    492:                         * Since the session is already displayed, we only care
                    493:                         * to add back in window for it.
                    494:                         */
                    495:                        if (wcd->type & TREE_WINDOW) {
                    496:                                /*
                    497:                                 * If the insertion point for adding the
                    498:                                 * windows to the session falls inside the
                    499:                                 * range of the list, then we insert these
                    500:                                 * entries in order *AFTER* the selected
                    501:                                 * session.
                    502:                                 */
1.75      nicm      503:                                if (pos < i) {
                    504:                                        data->list = xreallocarray(data->list,
                    505:                                            data->list_size + 1,
                    506:                                            sizeof *data->list);
                    507:                                        memmove(&data->list[pos + 2],
                    508:                                            &data->list[pos + 1],
1.76      nicm      509:                                            (data->list_size - (pos + 1)) *
1.75      nicm      510:                                            sizeof *data->list);
                    511:                                        memcpy(&data->list[pos + 1],
                    512:                                            &data->old_list[i],
                    513:                                            sizeof *data->list);
                    514:                                        data->list_size++;
1.27      nicm      515:                                } else {
                    516:                                        /* Ran out of room, add to the end. */
1.75      nicm      517:                                        data->list = xreallocarray(data->list,
                    518:                                            data->list_size + 1,
                    519:                                            sizeof *data->list);
                    520:                                        memcpy(&data->list[data->list_size],
                    521:                                            &data->old_list[i],
                    522:                                            sizeof *data->list);
                    523:                                        data->list_size++;
1.27      nicm      524:                                }
                    525:                        }
                    526:                }
                    527:        }
                    528: }
                    529:
1.79      nicm      530: static struct window_choose_mode_item *
1.68      nicm      531: window_choose_get_item(struct window_pane *wp, key_code key,
                    532:     struct mouse_event *m)
1.61      nicm      533: {
                    534:        struct window_choose_mode_data  *data = wp->modedata;
                    535:        u_int                            x, y, idx;
                    536:
                    537:        if (!KEYC_IS_MOUSE(key))
1.75      nicm      538:                return (&data->list[data->selected]);
1.61      nicm      539:
                    540:        if (cmd_mouse_at(wp, m, &x, &y, 0) != 0)
                    541:                return (NULL);
                    542:
                    543:        idx = data->top + y;
1.75      nicm      544:        if (idx >= data->list_size)
1.61      nicm      545:                return (NULL);
1.75      nicm      546:        return (&data->list[idx]);
1.61      nicm      547: }
                    548:
1.79      nicm      549: static void
1.70      nicm      550: window_choose_key(struct window_pane *wp, __unused struct client *c,
                    551:     __unused struct session *sess, key_code key, struct mouse_event *m)
1.1       nicm      552: {
                    553:        struct window_choose_mode_data  *data = wp->modedata;
                    554:        struct screen                   *s = &data->screen;
                    555:        struct screen_write_ctx          ctx;
                    556:        struct window_choose_mode_item  *item;
1.22      nicm      557:        size_t                           input_len;
1.27      nicm      558:        u_int                            items, n;
1.11      nicm      559:        int                              idx;
1.1       nicm      560:
1.75      nicm      561:        items = data->list_size;
1.1       nicm      562:
1.25      nicm      563:        if (data->input_type == WINDOW_CHOOSE_GOTO_ITEM) {
1.78      nicm      564:                switch (mode_key_lookup(&data->mdata, key)) {
1.25      nicm      565:                case MODEKEYCHOICE_CANCEL:
                    566:                        data->input_type = WINDOW_CHOOSE_NORMAL;
                    567:                        window_choose_redraw_screen(wp);
1.22      nicm      568:                        break;
1.25      nicm      569:                case MODEKEYCHOICE_CHOOSE:
1.22      nicm      570:                        n = strtonum(data->input_str, 0, INT_MAX, NULL);
1.25      nicm      571:                        if (n > items - 1) {
                    572:                                data->input_type = WINDOW_CHOOSE_NORMAL;
                    573:                                window_choose_redraw_screen(wp);
1.22      nicm      574:                                break;
1.25      nicm      575:                        }
1.75      nicm      576:                        window_choose_fire_callback(wp, data->list[n].wcd);
1.22      nicm      577:                        break;
1.25      nicm      578:                case MODEKEYCHOICE_BACKSPACE:
                    579:                        input_len = strlen(data->input_str);
                    580:                        if (input_len > 0)
                    581:                                data->input_str[input_len - 1] = '\0';
                    582:                        window_choose_redraw_screen(wp);
                    583:                        break;
                    584:                default:
                    585:                        if (key < '0' || key > '9')
                    586:                                break;
                    587:                        window_choose_prompt_input(WINDOW_CHOOSE_GOTO_ITEM,
                    588:                            "Goto Item", wp, key);
                    589:                        break;
1.22      nicm      590:                }
1.25      nicm      591:                return;
                    592:        }
                    593:
1.78      nicm      594:        switch (mode_key_lookup(&data->mdata, key)) {
1.25      nicm      595:        case MODEKEYCHOICE_CANCEL:
                    596:                window_choose_fire_callback(wp, NULL);
                    597:                break;
                    598:        case MODEKEYCHOICE_CHOOSE:
1.61      nicm      599:                if ((item = window_choose_get_item(wp, key, m)) == NULL)
                    600:                        break;
1.25      nicm      601:                window_choose_fire_callback(wp, item->wcd);
1.1       nicm      602:                break;
1.27      nicm      603:        case MODEKEYCHOICE_TREE_TOGGLE:
1.61      nicm      604:                if ((item = window_choose_get_item(wp, key, m)) == NULL)
                    605:                        break;
                    606:                if (item->state & TREE_EXPANDED) {
                    607:                        window_choose_collapse(wp, item->wcd->tree_session,
1.62      nicm      608:                            data->selected);
1.61      nicm      609:                } else {
1.27      nicm      610:                        window_choose_expand(wp, item->wcd->tree_session,
1.62      nicm      611:                            data->selected);
1.27      nicm      612:                }
                    613:                window_choose_redraw_screen(wp);
                    614:                break;
                    615:        case MODEKEYCHOICE_TREE_COLLAPSE:
1.61      nicm      616:                if ((item = window_choose_get_item(wp, key, m)) == NULL)
                    617:                        break;
1.27      nicm      618:                if (item->state & TREE_EXPANDED) {
1.61      nicm      619:                        window_choose_collapse(wp, item->wcd->tree_session,
                    620:                            data->selected);
1.27      nicm      621:                        window_choose_redraw_screen(wp);
                    622:                }
                    623:                break;
                    624:        case MODEKEYCHOICE_TREE_COLLAPSE_ALL:
                    625:                window_choose_collapse_all(wp);
                    626:                break;
                    627:        case MODEKEYCHOICE_TREE_EXPAND:
1.61      nicm      628:                if ((item = window_choose_get_item(wp, key, m)) == NULL)
                    629:                        break;
1.27      nicm      630:                if (!(item->state & TREE_EXPANDED)) {
                    631:                        window_choose_expand(wp, item->wcd->tree_session,
                    632:                            data->selected);
                    633:                        window_choose_redraw_screen(wp);
                    634:                }
                    635:                break;
                    636:        case MODEKEYCHOICE_TREE_EXPAND_ALL:
                    637:                window_choose_expand_all(wp);
                    638:                break;
1.5       nicm      639:        case MODEKEYCHOICE_UP:
1.1       nicm      640:                if (items == 0)
                    641:                        break;
                    642:                if (data->selected == 0) {
                    643:                        data->selected = items - 1;
                    644:                        if (data->selected > screen_size_y(s) - 1)
                    645:                                data->top = items - screen_size_y(s);
                    646:                        window_choose_redraw_screen(wp);
                    647:                        break;
                    648:                }
                    649:                data->selected--;
                    650:                if (data->selected < data->top)
                    651:                        window_choose_scroll_up(wp);
                    652:                else {
                    653:                        screen_write_start(&ctx, wp, NULL);
1.73      nicm      654:                        window_choose_write_line(wp, &ctx,
                    655:                            data->selected - data->top);
                    656:                        window_choose_write_line(wp, &ctx,
                    657:                            data->selected + 1 - data->top);
1.1       nicm      658:                        screen_write_stop(&ctx);
                    659:                }
                    660:                break;
1.5       nicm      661:        case MODEKEYCHOICE_DOWN:
1.1       nicm      662:                if (items == 0)
                    663:                        break;
                    664:                if (data->selected == items - 1) {
                    665:                        data->selected = 0;
                    666:                        data->top = 0;
                    667:                        window_choose_redraw_screen(wp);
                    668:                        break;
                    669:                }
                    670:                data->selected++;
1.5       nicm      671:
1.15      nicm      672:                if (data->selected < data->top + screen_size_y(s)) {
1.1       nicm      673:                        screen_write_start(&ctx, wp, NULL);
1.73      nicm      674:                        window_choose_write_line(wp, &ctx,
                    675:                            data->selected - data->top);
                    676:                        window_choose_write_line(wp, &ctx,
                    677:                            data->selected - 1 - data->top);
1.1       nicm      678:                        screen_write_stop(&ctx);
1.15      nicm      679:                } else
                    680:                        window_choose_scroll_down(wp);
                    681:                break;
                    682:        case MODEKEYCHOICE_SCROLLUP:
                    683:                if (items == 0 || data->top == 0)
                    684:                        break;
                    685:                if (data->selected == data->top + screen_size_y(s) - 1) {
                    686:                        data->selected--;
                    687:                        window_choose_scroll_up(wp);
                    688:                        screen_write_start(&ctx, wp, NULL);
1.73      nicm      689:                        window_choose_write_line(wp, &ctx,
                    690:                            screen_size_y(s) - 1);
1.15      nicm      691:                        screen_write_stop(&ctx);
                    692:                } else
                    693:                        window_choose_scroll_up(wp);
                    694:                break;
                    695:        case MODEKEYCHOICE_SCROLLDOWN:
                    696:                if (items == 0 ||
                    697:                    data->top + screen_size_y(&data->screen) >= items)
                    698:                        break;
                    699:                if (data->selected == data->top) {
                    700:                        data->selected++;
                    701:                        window_choose_scroll_down(wp);
                    702:                        screen_write_start(&ctx, wp, NULL);
                    703:                        window_choose_write_line(wp, &ctx, 0);
                    704:                        screen_write_stop(&ctx);
                    705:                } else
                    706:                        window_choose_scroll_down(wp);
1.1       nicm      707:                break;
1.5       nicm      708:        case MODEKEYCHOICE_PAGEUP:
1.1       nicm      709:                if (data->selected < screen_size_y(s)) {
                    710:                        data->selected = 0;
                    711:                        data->top = 0;
                    712:                } else {
                    713:                        data->selected -= screen_size_y(s);
                    714:                        if (data->top < screen_size_y(s))
                    715:                                data->top = 0;
                    716:                        else
                    717:                                data->top -= screen_size_y(s);
                    718:                }
1.13      nicm      719:                window_choose_redraw_screen(wp);
1.1       nicm      720:                break;
1.5       nicm      721:        case MODEKEYCHOICE_PAGEDOWN:
1.1       nicm      722:                data->selected += screen_size_y(s);
                    723:                if (data->selected > items - 1)
                    724:                        data->selected = items - 1;
                    725:                data->top += screen_size_y(s);
                    726:                if (screen_size_y(s) < items) {
                    727:                        if (data->top + screen_size_y(s) > items)
                    728:                                data->top = items - screen_size_y(s);
                    729:                } else
                    730:                        data->top = 0;
                    731:                if (data->selected < data->top)
                    732:                        data->top = data->selected;
                    733:                window_choose_redraw_screen(wp);
                    734:                break;
1.22      nicm      735:        case MODEKEYCHOICE_BACKSPACE:
                    736:                input_len = strlen(data->input_str);
                    737:                if (input_len > 0)
                    738:                        data->input_str[input_len - 1] = '\0';
                    739:                window_choose_redraw_screen(wp);
                    740:                break;
                    741:        case MODEKEYCHOICE_STARTNUMBERPREFIX:
1.25      nicm      742:                key &= KEYC_MASK_KEY;
                    743:                if (key < '0' || key > '9')
1.11      nicm      744:                        break;
1.25      nicm      745:                window_choose_prompt_input(WINDOW_CHOOSE_GOTO_ITEM,
                    746:                    "Goto Item", wp, key);
1.51      nicm      747:                break;
                    748:        case MODEKEYCHOICE_STARTOFLIST:
                    749:                data->selected = 0;
                    750:                data->top = 0;
                    751:                window_choose_redraw_screen(wp);
                    752:                break;
                    753:        case MODEKEYCHOICE_TOPLINE:
                    754:                data->selected = data->top;
                    755:                window_choose_redraw_screen(wp);
                    756:                break;
                    757:        case MODEKEYCHOICE_BOTTOMLINE:
                    758:                data->selected = data->top + screen_size_y(s) - 1;
                    759:                if (data->selected > items - 1)
                    760:                        data->selected = items - 1;
                    761:                window_choose_redraw_screen(wp);
                    762:                break;
                    763:        case MODEKEYCHOICE_ENDOFLIST:
                    764:                data->selected = items - 1;
                    765:                if (screen_size_y(s) < items)
                    766:                        data->top = items - screen_size_y(s);
                    767:                else
                    768:                        data->top = 0;
                    769:                window_choose_redraw_screen(wp);
1.22      nicm      770:                break;
                    771:        default:
1.25      nicm      772:                idx = window_choose_index_key(data, key);
1.75      nicm      773:                if (idx < 0 || (u_int) idx >= data->list_size)
1.25      nicm      774:                        break;
                    775:                data->selected = idx;
1.75      nicm      776:                window_choose_fire_callback(wp, data->list[idx].wcd);
1.1       nicm      777:                break;
                    778:        }
                    779: }
                    780:
1.79      nicm      781: static void
1.68      nicm      782: window_choose_write_line(struct window_pane *wp, struct screen_write_ctx *ctx,
                    783:     u_int py)
1.1       nicm      784: {
                    785:        struct window_choose_mode_data  *data = wp->modedata;
                    786:        struct window_choose_mode_item  *item;
1.67      nicm      787:        struct options                  *oo = wp->window->options;
1.1       nicm      788:        struct screen                   *s = &data->screen;
                    789:        struct grid_cell                 gc;
1.22      nicm      790:        size_t                           last, xoff = 0;
1.25      nicm      791:        char                             hdr[32], label[32];
1.69      nicm      792:        int                              key;
1.1       nicm      793:
1.3       nicm      794:        if (data->callbackfn == NULL)
1.1       nicm      795:                fatalx("called before callback assigned");
                    796:
1.22      nicm      797:        last = screen_size_y(s) - 1;
1.1       nicm      798:        memcpy(&gc, &grid_default_cell, sizeof gc);
1.22      nicm      799:        if (data->selected == data->top + py)
1.50      nicm      800:                style_apply(&gc, oo, "mode-style");
1.1       nicm      801:
                    802:        screen_write_cursormove(ctx, 0, py);
1.75      nicm      803:        if (data->top + py  < data->list_size) {
                    804:                item = &data->list[data->top + py];
1.24      nicm      805:                if (item->wcd->wl != NULL &&
                    806:                    item->wcd->wl->flags & WINLINK_ALERTFLAGS)
                    807:                        gc.attr |= GRID_ATTR_BRIGHT;
1.11      nicm      808:
1.25      nicm      809:                key = window_choose_key_index(data, data->top + py);
                    810:                if (key != -1)
1.56      nicm      811:                        xsnprintf(label, sizeof label, "(%c)", key);
1.25      nicm      812:                else
1.56      nicm      813:                        xsnprintf(label, sizeof label, "(%d)", item->pos);
1.69      nicm      814:                screen_write_nputs(ctx, screen_size_x(s) - 1, &gc,
1.27      nicm      815:                    "%*s %s %s", data->width + 2, label,
                    816:                    /*
                    817:                     * Add indication to tree if necessary about whether it's
                    818:                     * expanded or not.
                    819:                     */
                    820:                    (item->wcd->type & TREE_SESSION) ?
                    821:                    (item->state & TREE_EXPANDED ? "-" : "+") : "", item->name);
1.1       nicm      822:        }
1.40      nicm      823:        while (s->cx < screen_size_x(s) - 1)
1.1       nicm      824:                screen_write_putc(ctx, &gc, ' ');
1.11      nicm      825:
1.22      nicm      826:        if (data->input_type != WINDOW_CHOOSE_NORMAL) {
1.50      nicm      827:                style_apply(&gc, oo, "mode-style");
1.11      nicm      828:
1.22      nicm      829:                xoff = xsnprintf(hdr, sizeof hdr,
                    830:                        "%s: %s", data->input_prompt, data->input_str);
                    831:                screen_write_cursormove(ctx, 0, last);
                    832:                screen_write_puts(ctx, &gc, "%s", hdr);
                    833:                screen_write_cursormove(ctx, xoff, py);
                    834:                memcpy(&gc, &grid_default_cell, sizeof gc);
1.11      nicm      835:        }
1.22      nicm      836:
1.11      nicm      837: }
                    838:
1.79      nicm      839: static int
1.25      nicm      840: window_choose_key_index(struct window_choose_mode_data *data, u_int idx)
                    841: {
                    842:        static const char       keys[] = "0123456789"
                    843:                                         "abcdefghijklmnopqrstuvwxyz"
                    844:                                         "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
                    845:        const char             *ptr;
                    846:        int                     mkey;
                    847:
                    848:        for (ptr = keys; *ptr != '\0'; ptr++) {
1.78      nicm      849:                mkey = mode_key_lookup(&data->mdata, *ptr);
1.25      nicm      850:                if (mkey != MODEKEY_NONE && mkey != MODEKEY_OTHER)
                    851:                        continue;
                    852:                if (idx-- == 0)
                    853:                        return (*ptr);
                    854:        }
                    855:        return (-1);
                    856: }
                    857:
1.79      nicm      858: static int
1.68      nicm      859: window_choose_index_key(struct window_choose_mode_data *data, key_code key)
1.11      nicm      860: {
1.25      nicm      861:        static const char       keys[] = "0123456789"
                    862:                                         "abcdefghijklmnopqrstuvwxyz"
                    863:                                         "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
1.11      nicm      864:        const char             *ptr;
1.25      nicm      865:        int                     mkey;
1.11      nicm      866:        u_int                   idx = 0;
                    867:
                    868:        for (ptr = keys; *ptr != '\0'; ptr++) {
1.78      nicm      869:                mkey = mode_key_lookup(&data->mdata, *ptr);
1.25      nicm      870:                if (mkey != MODEKEY_NONE && mkey != MODEKEY_OTHER)
                    871:                        continue;
1.68      nicm      872:                if (key == (key_code)*ptr)
1.11      nicm      873:                        return (idx);
                    874:                idx++;
                    875:        }
                    876:        return (-1);
1.1       nicm      877: }
                    878:
1.79      nicm      879: static void
1.1       nicm      880: window_choose_redraw_screen(struct window_pane *wp)
                    881: {
                    882:        struct window_choose_mode_data  *data = wp->modedata;
                    883:        struct screen                   *s = &data->screen;
                    884:        struct screen_write_ctx          ctx;
                    885:        u_int                            i;
                    886:
                    887:        screen_write_start(&ctx, wp, NULL);
                    888:        for (i = 0; i < screen_size_y(s); i++)
                    889:                window_choose_write_line(wp, &ctx, i);
                    890:        screen_write_stop(&ctx);
                    891: }
                    892:
1.79      nicm      893: static void
1.1       nicm      894: window_choose_scroll_up(struct window_pane *wp)
                    895: {
                    896:        struct window_choose_mode_data  *data = wp->modedata;
                    897:        struct screen_write_ctx          ctx;
                    898:
                    899:        if (data->top == 0)
                    900:                return;
                    901:        data->top--;
                    902:
                    903:        screen_write_start(&ctx, wp, NULL);
                    904:        screen_write_cursormove(&ctx, 0, 0);
1.80      nicm      905:        screen_write_insertline(&ctx, 1, 8);
1.1       nicm      906:        window_choose_write_line(wp, &ctx, 0);
                    907:        if (screen_size_y(&data->screen) > 1)
                    908:                window_choose_write_line(wp, &ctx, 1);
                    909:        screen_write_stop(&ctx);
                    910: }
                    911:
1.79      nicm      912: static void
1.1       nicm      913: window_choose_scroll_down(struct window_pane *wp)
                    914: {
                    915:        struct window_choose_mode_data  *data = wp->modedata;
                    916:        struct screen                   *s = &data->screen;
                    917:        struct screen_write_ctx          ctx;
                    918:
1.75      nicm      919:        if (data->top >= data->list_size)
1.1       nicm      920:                return;
                    921:        data->top++;
                    922:
                    923:        screen_write_start(&ctx, wp, NULL);
                    924:        screen_write_cursormove(&ctx, 0, 0);
1.80      nicm      925:        screen_write_deleteline(&ctx, 1, 8);
1.1       nicm      926:        window_choose_write_line(wp, &ctx, screen_size_y(s) - 1);
                    927:        if (screen_size_y(&data->screen) > 1)
                    928:                window_choose_write_line(wp, &ctx, screen_size_y(s) - 2);
                    929:        screen_write_stop(&ctx);
1.19      nicm      930: }
                    931:
1.20      nicm      932: struct window_choose_data *
1.35      nicm      933: window_choose_add_session(struct window_pane *wp, struct client *c,
1.36      nicm      934:     struct session *s, const char *template, const char *action, u_int idx)
1.20      nicm      935: {
                    936:        struct window_choose_data       *wcd;
                    937:
1.33      nicm      938:        wcd = window_choose_data_create(TREE_SESSION, c, c->session);
1.46      nicm      939:        wcd->idx = s->id;
1.33      nicm      940:
1.27      nicm      941:        wcd->tree_session = s;
1.33      nicm      942:        wcd->tree_session->references++;
                    943:
1.20      nicm      944:        wcd->ft_template = xstrdup(template);
                    945:        format_add(wcd->ft, "line", "%u", idx);
1.59      nicm      946:        format_defaults(wcd->ft, NULL, s, NULL, NULL);
1.20      nicm      947:
1.33      nicm      948:        wcd->command = cmd_template_replace(action, s->name, 1);
1.20      nicm      949:
                    950:        window_choose_add(wp, wcd);
                    951:
                    952:        return (wcd);
                    953: }
                    954:
                    955: struct window_choose_data *
1.35      nicm      956: window_choose_add_window(struct window_pane *wp, struct client *c,
1.22      nicm      957:     struct session *s, struct winlink *wl, const char *template,
1.36      nicm      958:     const char *action, u_int idx)
1.20      nicm      959: {
                    960:        struct window_choose_data       *wcd;
1.33      nicm      961:        char                            *expanded;
1.20      nicm      962:
1.33      nicm      963:        wcd = window_choose_data_create(TREE_WINDOW, c, c->session);
                    964:        wcd->idx = wl->idx;
1.20      nicm      965:
1.33      nicm      966:        wcd->wl = wl;
1.20      nicm      967:
1.27      nicm      968:        wcd->tree_session = s;
1.33      nicm      969:        wcd->tree_session->references++;
                    970:
1.20      nicm      971:        wcd->ft_template = xstrdup(template);
                    972:        format_add(wcd->ft, "line", "%u", idx);
1.59      nicm      973:        format_defaults(wcd->ft, NULL, s, wl, NULL);
1.20      nicm      974:
1.33      nicm      975:        xasprintf(&expanded, "%s:%d", s->name, wl->idx);
                    976:        wcd->command = cmd_template_replace(action, expanded, 1);
                    977:        free(expanded);
1.20      nicm      978:
                    979:        window_choose_add(wp, wcd);
                    980:
                    981:        return (wcd);
1.1       nicm      982: }