[BACK]Return to cmd-capture-pane.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / tmux

Annotation of src/usr.bin/tmux/cmd-capture-pane.c, Revision 1.60

1.60    ! nicm        1: /* $OpenBSD: cmd-capture-pane.c,v 1.59 2022/09/28 07:55:29 nicm Exp $ */
1.1       nicm        2:
                      3: /*
                      4:  * Copyright (c) 2009 Jonathan Alvarado <radobobo@users.sourceforge.net>
                      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.4       nicm       21: #include <stdlib.h>
1.1       nicm       22: #include <string.h>
                     23:
                     24: #include "tmux.h"
                     25:
                     26: /*
1.14      nicm       27:  * Write the entire contents of a pane to a buffer or stdout.
1.1       nicm       28:  */
                     29:
1.41      nicm       30: static enum cmd_retval cmd_capture_pane_exec(struct cmd *, struct cmdq_item *);
1.1       nicm       31:
1.41      nicm       32: static char    *cmd_capture_pane_append(char *, size_t *, char *, size_t);
                     33: static char    *cmd_capture_pane_pending(struct args *, struct window_pane *,
                     34:                     size_t *);
                     35: static char    *cmd_capture_pane_history(struct args *, struct cmdq_item *,
                     36:                     struct window_pane *, size_t *);
1.23      nicm       37:
1.1       nicm       38: const struct cmd_entry cmd_capture_pane_entry = {
1.37      nicm       39:        .name = "capture-pane",
                     40:        .alias = "capturep",
                     41:
1.60    ! nicm       42:        .args = { "ab:CeE:JNpPqS:Tt:", 0, 0, NULL },
1.59      nicm       43:        .usage = "[-aCeJNpPqT] " CMD_BUFFER_USAGE " [-E end-line] "
1.47      nicm       44:                 "[-S start-line] " CMD_TARGET_PANE_USAGE,
1.37      nicm       45:
1.44      nicm       46:        .target = { 't', CMD_FIND_PANE, 0 },
1.38      nicm       47:
1.40      nicm       48:        .flags = CMD_AFTERHOOK,
1.37      nicm       49:        .exec = cmd_capture_pane_exec
1.1       nicm       50: };
                     51:
1.43      nicm       52: const struct cmd_entry cmd_clear_history_entry = {
                     53:        .name = "clear-history",
                     54:        .alias = "clearhist",
                     55:
1.57      nicm       56:        .args = { "Ht:", 0, 0, NULL },
                     57:        .usage = "[-H] " CMD_TARGET_PANE_USAGE,
1.43      nicm       58:
1.44      nicm       59:        .target = { 't', CMD_FIND_PANE, 0 },
1.43      nicm       60:
                     61:        .flags = CMD_AFTERHOOK,
                     62:        .exec = cmd_capture_pane_exec
                     63: };
                     64:
1.39      nicm       65: static char *
1.23      nicm       66: cmd_capture_pane_append(char *buf, size_t *len, char *line, size_t linelen)
                     67: {
1.29      nicm       68:        buf = xrealloc(buf, *len + linelen + 1);
1.23      nicm       69:        memcpy(buf + *len, line, linelen);
                     70:        *len += linelen;
                     71:        return (buf);
                     72: }
                     73:
1.39      nicm       74: static char *
1.23      nicm       75: cmd_capture_pane_pending(struct args *args, struct window_pane *wp,
                     76:     size_t *len)
                     77: {
1.32      nicm       78:        struct evbuffer *pending;
                     79:        char            *buf, *line, tmp[5];
                     80:        size_t           linelen;
                     81:        u_int            i;
1.23      nicm       82:
1.50      nicm       83:        pending = input_pending(wp->ictx);
1.32      nicm       84:        if (pending == NULL)
1.23      nicm       85:                return (xstrdup(""));
                     86:
1.32      nicm       87:        line = EVBUFFER_DATA(pending);
                     88:        linelen = EVBUFFER_LENGTH(pending);
1.23      nicm       89:
1.24      nicm       90:        buf = xstrdup("");
1.23      nicm       91:        if (args_has(args, 'C')) {
                     92:                for (i = 0; i < linelen; i++) {
1.42      nicm       93:                        if (line[i] >= ' ' && line[i] != '\\') {
1.23      nicm       94:                                tmp[0] = line[i];
                     95:                                tmp[1] = '\0';
                     96:                        } else
1.31      nicm       97:                                xsnprintf(tmp, sizeof tmp, "\\%03hho", line[i]);
1.23      nicm       98:                        buf = cmd_capture_pane_append(buf, len, tmp,
                     99:                            strlen(tmp));
                    100:                }
                    101:        } else
                    102:                buf = cmd_capture_pane_append(buf, len, line, linelen);
                    103:        return (buf);
                    104: }
                    105:
1.39      nicm      106: static char *
1.41      nicm      107: cmd_capture_pane_history(struct args *args, struct cmdq_item *item,
1.23      nicm      108:     struct window_pane *wp, size_t *len)
1.1       nicm      109: {
1.7       nicm      110:        struct grid             *gd;
1.18      nicm      111:        const struct grid_line  *gl;
1.23      nicm      112:        struct grid_cell        *gc = NULL;
1.59      nicm      113:        int                      n, join_lines, flags = 0;
1.23      nicm      114:        u_int                    i, sx, top, bottom, tmp;
                    115:        char                    *cause, *buf, *line;
1.28      nicm      116:        const char              *Sflag, *Eflag;
1.23      nicm      117:        size_t                   linelen;
1.14      nicm      118:
1.23      nicm      119:        sx = screen_size_x(&wp->base);
1.20      nicm      120:        if (args_has(args, 'a')) {
1.51      nicm      121:                gd = wp->base.saved_grid;
1.23      nicm      122:                if (gd == NULL) {
                    123:                        if (!args_has(args, 'q')) {
1.41      nicm      124:                                cmdq_error(item, "no alternate screen");
1.23      nicm      125:                                return (NULL);
                    126:                        }
                    127:                        return (xstrdup(""));
1.20      nicm      128:                }
1.23      nicm      129:        } else
                    130:                gd = wp->base.grid;
                    131:
1.28      nicm      132:        Sflag = args_get(args, 'S');
                    133:        if (Sflag != NULL && strcmp(Sflag, "-") == 0)
1.23      nicm      134:                top = 0;
1.28      nicm      135:        else {
1.56      nicm      136:                n = args_strtonum_and_expand(args, 'S', INT_MIN, SHRT_MAX,
                    137:                        item, &cause);
1.28      nicm      138:                if (cause != NULL) {
                    139:                        top = gd->hsize;
                    140:                        free(cause);
                    141:                } else if (n < 0 && (u_int) -n > gd->hsize)
                    142:                        top = 0;
                    143:                else
                    144:                        top = gd->hsize + n;
                    145:                if (top > gd->hsize + gd->sy - 1)
                    146:                        top = gd->hsize + gd->sy - 1;
                    147:        }
1.23      nicm      148:
1.28      nicm      149:        Eflag = args_get(args, 'E');
                    150:        if (Eflag != NULL && strcmp(Eflag, "-") == 0)
1.23      nicm      151:                bottom = gd->hsize + gd->sy - 1;
1.28      nicm      152:        else {
1.56      nicm      153:                n = args_strtonum_and_expand(args, 'E', INT_MIN, SHRT_MAX,
                    154:                        item, &cause);
1.28      nicm      155:                if (cause != NULL) {
                    156:                        bottom = gd->hsize + gd->sy - 1;
                    157:                        free(cause);
                    158:                } else if (n < 0 && (u_int) -n > gd->hsize)
                    159:                        bottom = 0;
                    160:                else
                    161:                        bottom = gd->hsize + n;
                    162:                if (bottom > gd->hsize + gd->sy - 1)
                    163:                        bottom = gd->hsize + gd->sy - 1;
                    164:        }
1.23      nicm      165:
                    166:        if (bottom < top) {
                    167:                tmp = bottom;
                    168:                bottom = top;
                    169:                top = tmp;
1.20      nicm      170:        }
1.1       nicm      171:
1.23      nicm      172:        join_lines = args_has(args, 'J');
1.59      nicm      173:        if (args_has(args, 'e'))
                    174:                flags |= GRID_STRING_WITH_SEQUENCES;
                    175:        if (args_has(args, 'C'))
                    176:                flags |= GRID_STRING_ESCAPE_SEQUENCES;
                    177:        if (!join_lines && !args_has(args, 'T'))
                    178:                flags |= GRID_STRING_EMPTY_CELLS;
                    179:        if (!join_lines && !args_has(args, 'N'))
                    180:                flags |= GRID_STRING_TRIM_SPACES;
1.23      nicm      181:
1.1       nicm      182:        buf = NULL;
1.23      nicm      183:        for (i = top; i <= bottom; i++) {
1.59      nicm      184:                line = grid_string_cells(gd, 0, i, sx, &gc, flags, wp->screen);
1.23      nicm      185:                linelen = strlen(line);
                    186:
                    187:                buf = cmd_capture_pane_append(buf, len, line, linelen);
                    188:
                    189:                gl = grid_peek_line(gd, i);
                    190:                if (!join_lines || !(gl->flags & GRID_LINE_WRAPPED))
                    191:                        buf[(*len)++] = '\n';
                    192:
                    193:                free(line);
                    194:        }
                    195:        return (buf);
                    196: }
1.1       nicm      197:
1.39      nicm      198: static enum cmd_retval
1.41      nicm      199: cmd_capture_pane_exec(struct cmd *self, struct cmdq_item *item)
1.23      nicm      200: {
1.52      nicm      201:        struct args             *args = cmd_get_args(self);
1.53      nicm      202:        struct client           *c = cmdq_get_client(item);
                    203:        struct window_pane      *wp = cmdq_get_target(item)->wp;
1.23      nicm      204:        char                    *buf, *cause;
1.27      nicm      205:        const char              *bufname;
1.23      nicm      206:        size_t                   len;
1.43      nicm      207:
1.52      nicm      208:        if (cmd_get_entry(self) == &cmd_clear_history_entry) {
1.46      nicm      209:                window_pane_reset_mode_all(wp);
1.43      nicm      210:                grid_clear_history(wp->base.grid);
1.57      nicm      211:                if (args_has(args, 'H'))
                    212:                        screen_reset_hyperlinks(wp->screen);
1.43      nicm      213:                return (CMD_RETURN_NORMAL);
                    214:        }
1.1       nicm      215:
1.23      nicm      216:        len = 0;
                    217:        if (args_has(args, 'P'))
                    218:                buf = cmd_capture_pane_pending(args, wp, &len);
                    219:        else
1.41      nicm      220:                buf = cmd_capture_pane_history(args, item, wp, &len);
1.23      nicm      221:        if (buf == NULL)
                    222:                return (CMD_RETURN_ERROR);
1.1       nicm      223:
1.14      nicm      224:        if (args_has(args, 'p')) {
1.54      nicm      225:                if (len > 0 && buf[len - 1] == '\n')
                    226:                        len--;
                    227:                if (c->flags & CLIENT_CONTROL)
                    228:                        control_write(c, "%.*s", (int)len, buf);
                    229:                else {
                    230:                        if (!file_can_print(c)) {
                    231:                                cmdq_error(item, "can't write to client");
                    232:                                free(buf);
                    233:                                return (CMD_RETURN_ERROR);
                    234:                        }
                    235:                        file_print_buffer(c, buf, len);
                    236:                        file_print(c, "\n");
1.33      nicm      237:                        free(buf);
1.14      nicm      238:                }
                    239:        } else {
1.27      nicm      240:                bufname = NULL;
                    241:                if (args_has(args, 'b'))
                    242:                        bufname = args_get(args, 'b');
                    243:
                    244:                if (paste_set(buf, len, bufname, &cause) != 0) {
1.41      nicm      245:                        cmdq_error(item, "%s", cause);
1.33      nicm      246:                        free(cause);
1.14      nicm      247:                        free(buf);
                    248:                        return (CMD_RETURN_ERROR);
                    249:                }
1.1       nicm      250:        }
1.6       nicm      251:
1.12      nicm      252:        return (CMD_RETURN_NORMAL);
1.1       nicm      253: }