cmd-capture-pane.c

   1 /* $OpenBSD$ */
   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
  21 #include <stdlib.h>
  22 #include <string.h>
  23
  24 #include "tmux.h"
  25
  26 /*
  27  * Write the entire contents of a pane to a buffer or stdout.
  28  */
  29
  30 enum cmd_retval  cmd_capture_pane_exec(struct cmd *, struct cmd_q *);
  31
  32 char            *cmd_capture_pane_append(char *, size_t *, char *, size_t);
  33 char            *cmd_capture_pane_pending(struct args *, struct window_pane *,
  34                      size_t *);
  35 char            *cmd_capture_pane_history(struct args *, struct cmd_q *,
  36                      struct window_pane *, size_t *);
  37
  38 const struct cmd_entry cmd_capture_pane_entry = {
  39         "capture-pane", "capturep",
  40         "ab:CeE:JpPqS:t:", 0, 0,
  41         "[-aCeJpPq] [-b buffer-index] [-E end-line] [-S start-line]"
  42         CMD_TARGET_PANE_USAGE,
  43         0,
  44         NULL,
  45         cmd_capture_pane_exec
  46 };
  47
  48 char *
  49 cmd_capture_pane_append(char *buf, size_t *len, char *line, size_t linelen)
  50 {
  51         buf = xrealloc(buf, 1, *len + linelen + 1);
  52         memcpy(buf + *len, line, linelen);
  53         *len += linelen;
  54         return (buf);
  55 }
  56
  57 char *
  58 cmd_capture_pane_pending(struct args *args, struct window_pane *wp,
  59     size_t *len)
  60 {
  61         char    *buf, *line, tmp[5];
  62         size_t   linelen;
  63         u_int    i;
  64
  65         if (wp->ictx.since_ground == NULL)
  66                 return (xstrdup(""));
  67
  68         line = EVBUFFER_DATA(wp->ictx.since_ground);
  69         linelen = EVBUFFER_LENGTH(wp->ictx.since_ground);
  70
  71         buf = xstrdup("");
  72         if (args_has(args, 'C')) {
  73                 for (i = 0; i < linelen; i++) {
  74                         if (line[i] >= ' ') {
  75                                 tmp[0] = line[i];
  76                                 tmp[1] = '\0';
  77                         } else
  78                                 xsnprintf(tmp, sizeof tmp, "\\%03o", line[i]);
  79                         buf = cmd_capture_pane_append(buf, len, tmp,
  80                             strlen(tmp));
  81                 }
  82         } else
  83                 buf = cmd_capture_pane_append(buf, len, line, linelen);
  84         return (buf);
  85 }
  86
  87 char *
  88 cmd_capture_pane_history(struct args *args, struct cmd_q *cmdq,
  89     struct window_pane *wp, size_t *len)
  90 {
  91         struct grid             *gd;
  92         const struct grid_line  *gl;
  93         struct grid_cell        *gc = NULL;
  94         int                      n, with_codes, escape_c0, join_lines;
  95         u_int                    i, sx, top, bottom, tmp;
  96         char                    *cause, *buf, *line;
  97         size_t                   linelen;
  98
  99         sx = screen_size_x(&wp->base);
 100         if (args_has(args, 'a')) {
 101                 gd = wp->saved_grid;
 102                 if (gd == NULL) {
 103                         if (!args_has(args, 'q')) {
 104                                 cmdq_error(cmdq, "no alternate screen");
 105                                 return (NULL);
 106                         }
 107                         return (xstrdup(""));
 108                 }
 109         } else
 110                 gd = wp->base.grid;
 111
 112         n = args_strtonum(args, 'S', INT_MIN, SHRT_MAX, &cause);
 113         if (cause != NULL) {
 114                 top = gd->hsize;
 115                 free(cause);
 116         } else if (n < 0 && (u_int) -n > gd->hsize)
 117                 top = 0;
 118         else
 119                 top = gd->hsize + n;
 120         if (top > gd->hsize + gd->sy - 1)
 121                 top = gd->hsize + gd->sy - 1;
 122
 123         n = args_strtonum(args, 'E', INT_MIN, SHRT_MAX, &cause);
 124         if (cause != NULL) {
 125                 bottom = gd->hsize + gd->sy - 1;
 126                 free(cause);
 127         } else if (n < 0 && (u_int) -n > gd->hsize)
 128                 bottom = 0;
 129         else
 130                 bottom = gd->hsize + n;
 131         if (bottom > gd->hsize + gd->sy - 1)
 132                 bottom = gd->hsize + gd->sy - 1;
 133
 134         if (bottom < top) {
 135                 tmp = bottom;
 136                 bottom = top;
 137                 top = tmp;
 138         }
 139
 140         with_codes = args_has(args, 'e');
 141         escape_c0 = args_has(args, 'C');
 142         join_lines = args_has(args, 'J');
 143
 144         buf = NULL;
 145         for (i = top; i <= bottom; i++) {
 146                 line = grid_string_cells(gd, 0, i, sx, &gc, with_codes,
 147                     escape_c0, !join_lines);
 148                 linelen = strlen(line);
 149
 150                 buf = cmd_capture_pane_append(buf, len, line, linelen);
 151
 152                 gl = grid_peek_line(gd, i);
 153                 if (!join_lines || !(gl->flags & GRID_LINE_WRAPPED))
 154                         buf[(*len)++] = '\n';
 155
 156                 free(line);
 157         }
 158         return (buf);
 159 }
 160
 161 enum cmd_retval
 162 cmd_capture_pane_exec(struct cmd *self, struct cmd_q *cmdq)
 163 {
 164         struct args             *args = self->args;
 165         struct client           *c;
 166         struct window_pane      *wp;
 167         char                    *buf, *cause;
 168         int                      buffer;
 169         u_int                    limit;
 170         size_t                   len;
 171
 172         if (cmd_find_pane(cmdq, args_get(args, 't'), NULL, &wp) == NULL)
 173                 return (CMD_RETURN_ERROR);
 174
 175         len = 0;
 176         if (args_has(args, 'P'))
 177                 buf = cmd_capture_pane_pending(args, wp, &len);
 178         else
 179                 buf = cmd_capture_pane_history(args, cmdq, wp, &len);
 180         if (buf == NULL)
 181                 return (CMD_RETURN_ERROR);
 182
 183         if (args_has(args, 'p')) {
 184                 c = cmdq->client;
 185                 if (c == NULL ||
 186                     (c->session != NULL && !(c->flags & CLIENT_CONTROL))) {
 187                         cmdq_error(cmdq, "can't write to stdout");
 188                         return (CMD_RETURN_ERROR);
 189                 }
 190                 evbuffer_add(c->stdout_data, buf, len);
 191                 if (args_has(args, 'P') && len > 0)
 192                     evbuffer_add(c->stdout_data, "\n", 1);
 193                 server_push_stdout(c);
 194         } else {
 195                 limit = options_get_number(&global_options, "buffer-limit");
 196                 if (!args_has(args, 'b')) {
 197                         paste_add(&global_buffers, buf, len, limit);
 198                         return (CMD_RETURN_NORMAL);
 199                 }
 200
 201                 buffer = args_strtonum(args, 'b', 0, INT_MAX, &cause);
 202                 if (cause != NULL) {
 203                         cmdq_error(cmdq, "buffer %s", cause);
 204                         free(buf);
 205                         free(cause);
 206                         return (CMD_RETURN_ERROR);
 207                 }
 208
 209                 if (paste_replace(&global_buffers, buffer, buf, len) != 0) {
 210                         cmdq_error(cmdq, "no buffer %d", buffer);
 211                         free(buf);
 212                         return (CMD_RETURN_ERROR);
 213                 }
 214         }
 215
 216         return (CMD_RETURN_NORMAL);
 217 }