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

Annotation of src/usr.bin/tmux/window-buffer.c, Revision 1.31

1.31    ! nicm        1: /* $OpenBSD: window-buffer.c,v 1.30 2020/05/16 16:02:24 nicm Exp $ */
1.1       nicm        2:
                      3: /*
                      4:  * Copyright (c) 2017 Nicholas Marriott <nicholas.marriott@gmail.com>
                      5:  *
                      6:  * Permission to use, copy, modify, and distribute this software for any
                      7:  * purpose with or without fee is hereby granted, provided that the above
                      8:  * copyright notice and this permission notice appear in all copies.
                      9:  *
                     10:  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
                     11:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
                     12:  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
                     13:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
                     14:  * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
                     15:  * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
                     16:  * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
                     17:  */
                     18:
                     19: #include <sys/types.h>
                     20:
                     21: #include <stdlib.h>
                     22: #include <string.h>
1.2       nicm       23: #include <time.h>
1.28      nicm       24: #include <unistd.h>
1.1       nicm       25: #include <vis.h>
                     26:
                     27: #include "tmux.h"
                     28:
1.15      nicm       29: static struct screen   *window_buffer_init(struct window_mode_entry *,
1.1       nicm       30:                             struct cmd_find_state *, struct args *);
1.15      nicm       31: static void             window_buffer_free(struct window_mode_entry *);
                     32: static void             window_buffer_resize(struct window_mode_entry *, u_int,
1.1       nicm       33:                             u_int);
1.15      nicm       34: static void             window_buffer_key(struct window_mode_entry *,
1.14      nicm       35:                             struct client *, struct session *,
                     36:                             struct winlink *, key_code, struct mouse_event *);
1.1       nicm       37:
                     38: #define WINDOW_BUFFER_DEFAULT_COMMAND "paste-buffer -b '%%'"
                     39:
1.9       nicm       40: #define WINDOW_BUFFER_DEFAULT_FORMAT \
1.27      nicm       41:        "#{t/p:buffer_created}: #{buffer_sample}"
1.9       nicm       42:
1.20      nicm       43: static const struct menu_item window_buffer_menu_items[] = {
                     44:        { "Paste", 'p', NULL },
                     45:        { "Paste Tagged", 'P', NULL },
                     46:        { "", KEYC_NONE, NULL },
                     47:        { "Tag", 't', NULL },
                     48:        { "Tag All", '\024', NULL },
                     49:        { "Tag None", 'T', NULL },
                     50:        { "", KEYC_NONE, NULL },
                     51:        { "Delete", 'd', NULL },
                     52:        { "Delete Tagged", 'D', NULL },
                     53:        { "", KEYC_NONE, NULL },
                     54:        { "Cancel", 'q', NULL },
                     55:
                     56:        { NULL, KEYC_NONE, NULL }
                     57: };
1.19      nicm       58:
1.1       nicm       59: const struct window_mode window_buffer_mode = {
                     60:        .name = "buffer-mode",
1.17      nicm       61:        .default_format = WINDOW_BUFFER_DEFAULT_FORMAT,
1.1       nicm       62:
                     63:        .init = window_buffer_init,
                     64:        .free = window_buffer_free,
                     65:        .resize = window_buffer_resize,
                     66:        .key = window_buffer_key,
                     67: };
                     68:
                     69: enum window_buffer_sort_type {
1.6       nicm       70:        WINDOW_BUFFER_BY_TIME,
1.1       nicm       71:        WINDOW_BUFFER_BY_NAME,
                     72:        WINDOW_BUFFER_BY_SIZE,
                     73: };
                     74: static const char *window_buffer_sort_list[] = {
1.6       nicm       75:        "time",
1.1       nicm       76:        "name",
                     77:        "size"
                     78: };
1.22      nicm       79: static struct mode_tree_sort_criteria *window_buffer_sort;
1.1       nicm       80:
                     81: struct window_buffer_itemdata {
                     82:        const char      *name;
                     83:        u_int            order;
                     84:        size_t           size;
                     85: };
                     86:
                     87: struct window_buffer_modedata {
1.19      nicm       88:        struct window_pane               *wp;
1.18      nicm       89:        struct cmd_find_state             fs;
1.19      nicm       90:
1.1       nicm       91:        struct mode_tree_data            *data;
                     92:        char                             *command;
1.9       nicm       93:        char                             *format;
1.1       nicm       94:
                     95:        struct window_buffer_itemdata   **item_list;
                     96:        u_int                             item_size;
                     97: };
                     98:
1.28      nicm       99: struct window_buffer_editdata {
                    100:        u_int                    wp_id;
                    101:        char                    *name;
                    102:        struct paste_buffer     *pb;
                    103: };
                    104:
1.1       nicm      105: static struct window_buffer_itemdata *
                    106: window_buffer_add_item(struct window_buffer_modedata *data)
                    107: {
                    108:        struct window_buffer_itemdata   *item;
                    109:
                    110:        data->item_list = xreallocarray(data->item_list, data->item_size + 1,
                    111:            sizeof *data->item_list);
                    112:        item = data->item_list[data->item_size++] = xcalloc(1, sizeof *item);
                    113:        return (item);
                    114: }
                    115:
                    116: static void
                    117: window_buffer_free_item(struct window_buffer_itemdata *item)
                    118: {
                    119:        free((void *)item->name);
                    120:        free(item);
                    121: }
                    122:
                    123: static int
1.22      nicm      124: window_buffer_cmp(const void *a0, const void *b0)
1.1       nicm      125: {
1.22      nicm      126:        const struct window_buffer_itemdata *const      *a = a0;
                    127:        const struct window_buffer_itemdata *const      *b = b0;
                    128:        int                                              result = 0;
                    129:
                    130:        if (window_buffer_sort->field == WINDOW_BUFFER_BY_TIME)
                    131:                result = (*b)->order - (*a)->order;
                    132:        else if (window_buffer_sort->field == WINDOW_BUFFER_BY_SIZE)
                    133:                result = (*b)->size - (*a)->size;
                    134:
                    135:        /* Use WINDOW_BUFFER_BY_NAME as default order and tie breaker. */
                    136:        if (result == 0)
                    137:                result = strcmp((*a)->name, (*b)->name);
                    138:
                    139:        if (window_buffer_sort->reversed)
                    140:                result = -result;
                    141:        return (result);
1.1       nicm      142: }
                    143:
                    144: static void
1.22      nicm      145: window_buffer_build(void *modedata, struct mode_tree_sort_criteria *sort_crit,
                    146:     __unused uint64_t *tag, const char *filter)
1.1       nicm      147: {
                    148:        struct window_buffer_modedata   *data = modedata;
                    149:        struct window_buffer_itemdata   *item;
                    150:        u_int                            i;
                    151:        struct paste_buffer             *pb;
1.9       nicm      152:        char                            *text, *cp;
1.8       nicm      153:        struct format_tree              *ft;
1.18      nicm      154:        struct session                  *s = NULL;
                    155:        struct winlink                  *wl = NULL;
                    156:        struct window_pane              *wp = NULL;
1.1       nicm      157:
                    158:        for (i = 0; i < data->item_size; i++)
                    159:                window_buffer_free_item(data->item_list[i]);
                    160:        free(data->item_list);
                    161:        data->item_list = NULL;
                    162:        data->item_size = 0;
                    163:
                    164:        pb = NULL;
                    165:        while ((pb = paste_walk(pb)) != NULL) {
                    166:                item = window_buffer_add_item(data);
                    167:                item->name = xstrdup(paste_buffer_name(pb));
                    168:                paste_buffer_data(pb, &item->size);
                    169:                item->order = paste_buffer_order(pb);
                    170:        }
                    171:
1.22      nicm      172:        window_buffer_sort = sort_crit;
                    173:        qsort(data->item_list, data->item_size, sizeof *data->item_list,
                    174:            window_buffer_cmp);
1.1       nicm      175:
1.18      nicm      176:        if (cmd_find_valid_state(&data->fs)) {
                    177:                s = data->fs.s;
                    178:                wl = data->fs.wl;
                    179:                wp = data->fs.wp;
                    180:        }
                    181:
1.1       nicm      182:        for (i = 0; i < data->item_size; i++) {
                    183:                item = data->item_list[i];
1.8       nicm      184:
1.9       nicm      185:                pb = paste_get_name(item->name);
                    186:                if (pb == NULL)
                    187:                        continue;
                    188:                ft = format_create(NULL, NULL, FORMAT_NONE, 0);
1.18      nicm      189:                format_defaults(ft, NULL, s, wl, wp);
1.9       nicm      190:                format_defaults_paste_buffer(ft, pb);
                    191:
1.8       nicm      192:                if (filter != NULL) {
                    193:                        cp = format_expand(ft, filter);
                    194:                        if (!format_true(cp)) {
                    195:                                free(cp);
                    196:                                format_free(ft);
                    197:                                continue;
                    198:                        }
                    199:                        free(cp);
                    200:                }
1.1       nicm      201:
1.9       nicm      202:                text = format_expand(ft, data->format);
1.1       nicm      203:                mode_tree_add(data->data, NULL, item, item->order, item->name,
                    204:                    text, -1);
                    205:                free(text);
1.9       nicm      206:
                    207:                format_free(ft);
1.1       nicm      208:        }
                    209:
                    210: }
                    211:
1.11      nicm      212: static void
                    213: window_buffer_draw(__unused void *modedata, void *itemdata,
                    214:     struct screen_write_ctx *ctx, u_int sx, u_int sy)
1.1       nicm      215: {
                    216:        struct window_buffer_itemdata   *item = itemdata;
                    217:        struct paste_buffer             *pb;
1.24      nicm      218:        const char                      *pdata, *start, *end;
                    219:        char                            *buf = NULL;
1.25      nicm      220:        size_t                           psize;
1.11      nicm      221:        u_int                            i, cx = ctx->s->cx, cy = ctx->s->cy;
1.1       nicm      222:
                    223:        pb = paste_get_name(item->name);
                    224:        if (pb == NULL)
1.11      nicm      225:                return;
1.1       nicm      226:
1.3       nicm      227:        pdata = end = paste_buffer_data(pb, &psize);
1.1       nicm      228:        for (i = 0; i < sy; i++) {
1.24      nicm      229:                start = end;
                    230:                while (end != pdata + psize && *end != '\n')
1.1       nicm      231:                        end++;
1.24      nicm      232:                buf = xreallocarray(buf, 4, end - start + 1);
1.29      nicm      233:                utf8_strvis(buf, start, end - start, VIS_OCTAL|VIS_CSTYLE|VIS_TAB);
1.24      nicm      234:                if (*buf != '\0') {
1.16      nicm      235:                        screen_write_cursormove(ctx, cx, cy + i, 0);
1.24      nicm      236:                        screen_write_nputs(ctx, sx, &grid_default_cell, "%s",
                    237:                            buf);
1.1       nicm      238:                }
                    239:
                    240:                if (end == pdata + psize)
                    241:                        break;
                    242:                end++;
                    243:        }
1.24      nicm      244:        free(buf);
1.1       nicm      245: }
                    246:
1.4       nicm      247: static int
                    248: window_buffer_search(__unused void *modedata, void *itemdata, const char *ss)
                    249: {
                    250:        struct window_buffer_itemdata   *item = itemdata;
                    251:        struct paste_buffer             *pb;
                    252:        const char                      *bufdata;
                    253:        size_t                           bufsize;
                    254:
                    255:        if ((pb = paste_get_name(item->name)) == NULL)
                    256:                return (0);
                    257:        if (strstr(item->name, ss) != NULL)
1.5       nicm      258:                return (1);
1.4       nicm      259:        bufdata = paste_buffer_data(pb, &bufsize);
                    260:        return (memmem(bufdata, bufsize, ss, strlen(ss)) != NULL);
                    261: }
                    262:
1.19      nicm      263: static void
                    264: window_buffer_menu(void *modedata, struct client *c, key_code key)
                    265: {
                    266:        struct window_buffer_modedata   *data = modedata;
                    267:        struct window_pane              *wp = data->wp;
                    268:        struct window_mode_entry        *wme;
                    269:
                    270:        wme = TAILQ_FIRST(&wp->modes);
                    271:        if (wme == NULL || wme->data != modedata)
                    272:                return;
                    273:        window_buffer_key(wme, c, NULL, NULL, key, NULL);
                    274: }
                    275:
1.1       nicm      276: static struct screen *
1.18      nicm      277: window_buffer_init(struct window_mode_entry *wme, struct cmd_find_state *fs,
                    278:     struct args *args)
1.1       nicm      279: {
1.15      nicm      280:        struct window_pane              *wp = wme->wp;
1.1       nicm      281:        struct window_buffer_modedata   *data;
                    282:        struct screen                   *s;
                    283:
1.15      nicm      284:        wme->data = data = xcalloc(1, sizeof *data);
1.19      nicm      285:        data->wp = wp;
1.18      nicm      286:        cmd_find_copy_state(&data->fs, fs);
1.1       nicm      287:
1.9       nicm      288:        if (args == NULL || !args_has(args, 'F'))
                    289:                data->format = xstrdup(WINDOW_BUFFER_DEFAULT_FORMAT);
                    290:        else
                    291:                data->format = xstrdup(args_get(args, 'F'));
1.1       nicm      292:        if (args == NULL || args->argc == 0)
                    293:                data->command = xstrdup(WINDOW_BUFFER_DEFAULT_COMMAND);
                    294:        else
                    295:                data->command = xstrdup(args->argv[0]);
                    296:
1.7       nicm      297:        data->data = mode_tree_start(wp, args, window_buffer_build,
1.30      nicm      298:            window_buffer_draw, window_buffer_search, window_buffer_menu, NULL,
                    299:            data, window_buffer_menu_items, window_buffer_sort_list,
1.19      nicm      300:            nitems(window_buffer_sort_list), &s);
1.13      nicm      301:        mode_tree_zoom(data->data, args);
1.1       nicm      302:
                    303:        mode_tree_build(data->data);
                    304:        mode_tree_draw(data->data);
                    305:
                    306:        return (s);
                    307: }
                    308:
                    309: static void
1.15      nicm      310: window_buffer_free(struct window_mode_entry *wme)
1.1       nicm      311: {
1.15      nicm      312:        struct window_buffer_modedata   *data = wme->data;
1.1       nicm      313:        u_int                            i;
                    314:
                    315:        if (data == NULL)
                    316:                return;
                    317:
                    318:        mode_tree_free(data->data);
                    319:
                    320:        for (i = 0; i < data->item_size; i++)
                    321:                window_buffer_free_item(data->item_list[i]);
                    322:        free(data->item_list);
                    323:
1.9       nicm      324:        free(data->format);
1.1       nicm      325:        free(data->command);
1.9       nicm      326:
1.1       nicm      327:        free(data);
                    328: }
                    329:
                    330: static void
1.15      nicm      331: window_buffer_resize(struct window_mode_entry *wme, u_int sx, u_int sy)
1.1       nicm      332: {
1.15      nicm      333:        struct window_buffer_modedata   *data = wme->data;
1.1       nicm      334:
                    335:        mode_tree_resize(data->data, sx, sy);
                    336: }
                    337:
                    338: static void
1.23      nicm      339: window_buffer_do_delete(void *modedata, void *itemdata,
1.10      nicm      340:     __unused struct client *c, __unused key_code key)
1.1       nicm      341: {
                    342:        struct window_buffer_modedata   *data = modedata;
                    343:        struct window_buffer_itemdata   *item = itemdata;
                    344:        struct paste_buffer             *pb;
                    345:
                    346:        if (item == mode_tree_get_current(data->data))
                    347:                mode_tree_down(data->data, 0);
                    348:        if ((pb = paste_get_name(item->name)) != NULL)
                    349:                paste_free(pb);
                    350: }
                    351:
                    352: static void
1.23      nicm      353: window_buffer_do_paste(void *modedata, void *itemdata, struct client *c,
1.10      nicm      354:     __unused key_code key)
                    355: {
                    356:        struct window_buffer_modedata   *data = modedata;
                    357:        struct window_buffer_itemdata   *item = itemdata;
                    358:
1.26      nicm      359:        if (paste_get_name(item->name) != NULL)
1.10      nicm      360:                mode_tree_run_command(c, NULL, data->command, item->name);
                    361: }
                    362:
                    363: static void
1.28      nicm      364: window_buffer_finish_edit(struct window_buffer_editdata *ed)
                    365: {
                    366:        free(ed->name);
                    367:        free(ed);
                    368: }
                    369:
                    370: static void
1.31    ! nicm      371: window_buffer_edit_close_cb(char *buf, size_t len, void *arg)
1.28      nicm      372: {
                    373:        struct window_buffer_editdata   *ed = arg;
                    374:        size_t                           oldlen;
                    375:        const char                      *oldbuf;
                    376:        struct paste_buffer             *pb;
                    377:        struct window_pane              *wp;
                    378:        struct window_buffer_modedata   *data;
                    379:        struct window_mode_entry        *wme;
                    380:
1.31    ! nicm      381:        if (buf == NULL || len == 0) {
1.28      nicm      382:                window_buffer_finish_edit(ed);
                    383:                return;
                    384:        }
                    385:
                    386:        pb = paste_get_name(ed->name);
                    387:        if (pb == NULL || pb != ed->pb) {
                    388:                window_buffer_finish_edit(ed);
                    389:                return;
                    390:        }
                    391:
1.31    ! nicm      392:        oldbuf = paste_buffer_data(pb, &oldlen);
        !           393:        if (oldlen != '\0' &&
        !           394:            oldbuf[oldlen - 1] != '\n' &&
        !           395:            buf[len - 1] == '\n')
        !           396:                len--;
        !           397:        if (len != 0)
        !           398:                paste_replace(pb, buf, len);
1.28      nicm      399:
                    400:        wp = window_pane_find_by_id(ed->wp_id);
                    401:        if (wp != NULL) {
                    402:                wme = TAILQ_FIRST(&wp->modes);
                    403:                if (wme->mode == &window_buffer_mode) {
                    404:                        data = wme->data;
                    405:                        mode_tree_build(data->data);
                    406:                        mode_tree_draw(data->data);
                    407:                }
                    408:                wp->flags |= PANE_REDRAW;
                    409:        }
                    410:        window_buffer_finish_edit(ed);
                    411: }
                    412:
                    413: static void
                    414: window_buffer_start_edit(struct window_buffer_modedata *data,
                    415:     struct window_buffer_itemdata *item, struct client *c)
                    416: {
                    417:        struct paste_buffer             *pb;
                    418:        const char                      *buf;
                    419:        size_t                           len;
                    420:        struct window_buffer_editdata   *ed;
                    421:
                    422:        if ((pb = paste_get_name(item->name)) == NULL)
                    423:                return;
                    424:        buf = paste_buffer_data(pb, &len);
                    425:
                    426:        ed = xcalloc(1, sizeof *ed);
                    427:        ed->wp_id = data->wp->id;
                    428:        ed->name = xstrdup(paste_buffer_name(pb));
                    429:        ed->pb = pb;
                    430:
1.31    ! nicm      431:        if (popup_editor(c, buf, len, window_buffer_edit_close_cb, ed) != 0)
1.28      nicm      432:                window_buffer_finish_edit(ed);
                    433: }
                    434:
                    435: static void
1.15      nicm      436: window_buffer_key(struct window_mode_entry *wme, struct client *c,
1.14      nicm      437:     __unused struct session *s, __unused struct winlink *wl, key_code key,
                    438:     struct mouse_event *m)
1.1       nicm      439: {
1.15      nicm      440:        struct window_pane              *wp = wme->wp;
                    441:        struct window_buffer_modedata   *data = wme->data;
1.10      nicm      442:        struct mode_tree_data           *mtd = data->data;
1.1       nicm      443:        struct window_buffer_itemdata   *item;
                    444:        int                              finished;
                    445:
1.12      nicm      446:        finished = mode_tree_key(mtd, c, &key, m, NULL, NULL);
1.1       nicm      447:        switch (key) {
1.28      nicm      448:        case 'e':
                    449:                item = mode_tree_get_current(mtd);
                    450:                window_buffer_start_edit(data, item, c);
                    451:                break;
1.1       nicm      452:        case 'd':
1.10      nicm      453:                item = mode_tree_get_current(mtd);
                    454:                window_buffer_do_delete(data, item, c, key);
                    455:                mode_tree_build(mtd);
1.1       nicm      456:                break;
                    457:        case 'D':
1.10      nicm      458:                mode_tree_each_tagged(mtd, window_buffer_do_delete, c, key, 0);
                    459:                mode_tree_build(mtd);
                    460:                break;
                    461:        case 'P':
                    462:                mode_tree_each_tagged(mtd, window_buffer_do_paste, c, key, 0);
                    463:                finished = 1;
1.1       nicm      464:                break;
1.10      nicm      465:        case 'p':
1.1       nicm      466:        case '\r':
1.10      nicm      467:                item = mode_tree_get_current(mtd);
                    468:                window_buffer_do_paste(data, item, c, key);
                    469:                finished = 1;
                    470:                break;
1.1       nicm      471:        }
                    472:        if (finished || paste_get_top(NULL) == NULL)
                    473:                window_pane_reset_mode(wp);
                    474:        else {
1.10      nicm      475:                mode_tree_draw(mtd);
1.1       nicm      476:                wp->flags |= PANE_REDRAW;
                    477:        }
                    478: }