Annotation of src/usr.bin/tmux/cmd-capture-pane.c, Revision 1.18
1.18 ! nicm 1: /* $OpenBSD: cmd-capture-pane.c,v 1.17 2013/03/22 15:53:24 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.12 nicm 30: enum cmd_retval cmd_capture_pane_exec(struct cmd *, struct cmd_ctx *);
1.1 nicm 31:
32: const struct cmd_entry cmd_capture_pane_entry = {
33: "capture-pane", "capturep",
1.18 ! nicm 34: "b:CeE:JpS:t:", 0, 0,
! 35: "[-CeJp] [-b buffer-index] [-E end-line] [-S start-line]"
1.13 nicm 36: CMD_TARGET_PANE_USAGE,
1.6 nicm 37: 0,
38: NULL,
39: NULL,
40: cmd_capture_pane_exec
1.1 nicm 41: };
42:
1.12 nicm 43: enum cmd_retval
1.1 nicm 44: cmd_capture_pane_exec(struct cmd *self, struct cmd_ctx *ctx)
45: {
1.6 nicm 46: struct args *args = self->args;
1.17 nicm 47: struct client *c;
1.6 nicm 48: struct window_pane *wp;
1.16 nicm 49: char *buf, *line, *cause;
1.6 nicm 50: struct screen *s;
1.7 nicm 51: struct grid *gd;
1.18 ! nicm 52: int buffer, n, with_codes, escape_c0, join_lines;
1.7 nicm 53: u_int i, limit, top, bottom, tmp;
1.16 nicm 54: size_t len, linelen;
55: struct grid_cell *gc;
1.18 ! nicm 56: const struct grid_line *gl;
1.14 nicm 57:
1.6 nicm 58: if (cmd_find_pane(ctx, args_get(args, 't'), NULL, &wp) == NULL)
1.12 nicm 59: return (CMD_RETURN_ERROR);
1.1 nicm 60: s = &wp->base;
1.7 nicm 61: gd = s->grid;
1.1 nicm 62:
63: buf = NULL;
64: len = 0;
65:
1.10 nicm 66: n = args_strtonum(args, 'S', INT_MIN, SHRT_MAX, &cause);
1.8 nicm 67: if (cause != NULL) {
1.7 nicm 68: top = gd->hsize;
1.11 nicm 69: free(cause);
1.8 nicm 70: } else if (n < 0 && (u_int) -n > gd->hsize)
1.7 nicm 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:
1.10 nicm 77: n = args_strtonum(args, 'E', INT_MIN, SHRT_MAX, &cause);
1.8 nicm 78: if (cause != NULL) {
1.7 nicm 79: bottom = gd->hsize + gd->sy - 1;
1.11 nicm 80: free(cause);
1.8 nicm 81: } else if (n < 0 && (u_int) -n > gd->hsize)
1.7 nicm 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:
1.18 ! nicm 94: with_codes = args_has(args, 'e');
! 95: escape_c0 = args_has(args, 'C');
! 96: join_lines = args_has(args, 'J');
! 97:
1.16 nicm 98: gc = NULL;
1.7 nicm 99: for (i = top; i <= bottom; i++) {
1.16 nicm 100: line = grid_string_cells(s->grid, 0, i, screen_size_x(s),
1.18 ! nicm 101: &gc, with_codes, escape_c0);
1.16 nicm 102: linelen = strlen(line);
103:
104: buf = xrealloc(buf, 1, len + linelen + 1);
105: memcpy(buf + len, line, linelen);
106: len += linelen;
1.18 ! nicm 107:
! 108: gl = grid_peek_line(s->grid, i);
! 109: if (!join_lines || !(gl->flags & GRID_LINE_WRAPPED))
! 110: buf[len++] = '\n';
1.1 nicm 111:
1.16 nicm 112: free(line);
1.1 nicm 113: }
114:
1.14 nicm 115: if (args_has(args, 'p')) {
1.17 nicm 116: c = ctx->curclient;
117: if (c == NULL || !(c->flags & CLIENT_CONTROL))
118: c = ctx->cmdclient;
1.14 nicm 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: }
1.1 nicm 145: }
1.6 nicm 146:
1.12 nicm 147: return (CMD_RETURN_NORMAL);
1.1 nicm 148: }