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_ctx *);
  31
  32 const struct cmd_entry cmd_capture_pane_entry = {
  33         "capture-pane", "capturep",
  34         "b:CeE:JpS:t:", 0, 0,
  35         "[-CeJp] [-b buffer-index] [-E end-line] [-S start-line]"
  36         CMD_TARGET_PANE_USAGE,
  37         0,
  38         NULL,
  39         NULL,
  40         cmd_capture_pane_exec
  41 };
  42
  43 enum cmd_retval
  44 cmd_capture_pane_exec(struct cmd *self, struct cmd_ctx *ctx)
  45 {
  46         struct args             *args = self->args;
  47         struct client           *c;
  48         struct window_pane      *wp;
  49         char                    *buf, *line, *cause;
  50         struct screen           *s;
  51         struct grid             *gd;
  52         int                      buffer, n, with_codes, escape_c0, join_lines;
  53         u_int                    i, limit, top, bottom, tmp;
  54         size_t                   len, linelen;
  55         struct grid_cell        *gc;
  56         const struct grid_line  *gl;
  57
  58         if (cmd_find_pane(ctx, args_get(args, 't'), NULL, &wp) == NULL)
  59                 return (CMD_RETURN_ERROR);
  60         s = &wp->base;
  61         gd = s->grid;
  62
  63         buf = NULL;
  64         len = 0;
  65
  66         n = args_strtonum(args, 'S', INT_MIN, SHRT_MAX, &cause);
  67         if (cause != NULL) {
  68                 top = gd->hsize;
  69                 free(cause);
  70         } else if (n < 0 && (u_int) -n > gd->hsize)
  71                 top = 0;
  72         else
  73                 top = gd->hsize + n;
  74         if (top > gd->hsize + gd->sy - 1)
  75                 top = gd->hsize + gd->sy - 1;
  76
  77         n = args_strtonum(args, 'E', INT_MIN, SHRT_MAX, &cause);
  78         if (cause != NULL) {
  79                 bottom = gd->hsize + gd->sy - 1;
  80                 free(cause);
  81         } else if (n < 0 && (u_int) -n > gd->hsize)
  82                 bottom = 0;
  83         else
  84                 bottom = gd->hsize + n;
  85         if (bottom > gd->hsize + gd->sy - 1)
  86                 bottom = gd->hsize + gd->sy - 1;
  87
  88         if (bottom < top) {
  89                 tmp = bottom;
  90                 bottom = top;
  91                 top = tmp;
  92         }
  93
  94         with_codes = args_has(args, 'e');
  95         escape_c0 = args_has(args, 'C');
  96         join_lines = args_has(args, 'J');
  97
  98         gc = NULL;
  99         for (i = top; i <= bottom; i++) {
 100                 line = grid_string_cells(s->grid, 0, i, screen_size_x(s),
 101                     &gc, with_codes, escape_c0);
 102                 linelen = strlen(line);
 103
 104                 buf = xrealloc(buf, 1, len + linelen + 1);
 105                 memcpy(buf + len, line, linelen);
 106                 len += linelen;
 107
 108                 gl = grid_peek_line(s->grid, i);
 109                 if (!join_lines || !(gl->flags & GRID_LINE_WRAPPED))
 110                         buf[len++] = '\n';
 111
 112                 free(line);
 113         }
 114
 115         if (args_has(args, 'p')) {
 116                 c = ctx->curclient;
 117                 if (c == NULL || !(c->flags & CLIENT_CONTROL))
 118                         c = ctx->cmdclient;
 119                 if (c == NULL) {
 120                         ctx->error(ctx, "can't write to stdout");
 121                         return (CMD_RETURN_ERROR);
 122                 }
 123                 evbuffer_add(c->stdout_data, buf, len);
 124                 server_push_stdout(c);
 125         } else {
 126                 limit = options_get_number(&global_options, "buffer-limit");
 127                 if (!args_has(args, 'b')) {
 128                         paste_add(&global_buffers, buf, len, limit);
 129                         return (CMD_RETURN_NORMAL);
 130                 }
 131
 132                 buffer = args_strtonum(args, 'b', 0, INT_MAX, &cause);
 133                 if (cause != NULL) {
 134                         ctx->error(ctx, "buffer %s", cause);
 135                         free(buf);
 136                         free(cause);
 137                         return (CMD_RETURN_ERROR);
 138                 }
 139
 140                 if (paste_replace(&global_buffers, buffer, buf, len) != 0) {
 141                         ctx->error(ctx, "no buffer %d", buffer);
 142                         free(buf);
 143                         return (CMD_RETURN_ERROR);
 144                 }
 145         }
 146
 147         return (CMD_RETURN_NORMAL);
 148 }