Merge branch 'obsd-master'
[tmux.git] / cmd-capture-pane.c
blob8f7250e84205470a902b0c97e6215bef5b7f5e52
1 /* $OpenBSD$ */
3 /*
4 * Copyright (c) 2009 Jonathan Alvarado <radobobo@users.sourceforge.net>
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.
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.
19 #include <sys/types.h>
21 #include <stdlib.h>
22 #include <string.h>
24 #include "tmux.h"
27 * Write the entire contents of a pane to a buffer or stdout.
30 static enum cmd_retval cmd_capture_pane_exec(struct cmd *, struct cmdq_item *);
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 *);
38 const struct cmd_entry cmd_capture_pane_entry = {
39 .name = "capture-pane",
40 .alias = "capturep",
42 .args = { "ab:CeE:JNpPqS:Tt:", 0, 0, NULL },
43 .usage = "[-aCeJNpPqT] " CMD_BUFFER_USAGE " [-E end-line] "
44 "[-S start-line] " CMD_TARGET_PANE_USAGE,
46 .target = { 't', CMD_FIND_PANE, 0 },
48 .flags = CMD_AFTERHOOK,
49 .exec = cmd_capture_pane_exec
52 const struct cmd_entry cmd_clear_history_entry = {
53 .name = "clear-history",
54 .alias = "clearhist",
56 .args = { "Ht:", 0, 0, NULL },
57 .usage = "[-H] " CMD_TARGET_PANE_USAGE,
59 .target = { 't', CMD_FIND_PANE, 0 },
61 .flags = CMD_AFTERHOOK,
62 .exec = cmd_capture_pane_exec
65 static char *
66 cmd_capture_pane_append(char *buf, size_t *len, char *line, size_t linelen)
68 buf = xrealloc(buf, *len + linelen + 1);
69 memcpy(buf + *len, line, linelen);
70 *len += linelen;
71 return (buf);
74 static char *
75 cmd_capture_pane_pending(struct args *args, struct window_pane *wp,
76 size_t *len)
78 struct evbuffer *pending;
79 char *buf, *line, tmp[5];
80 size_t linelen;
81 u_int i;
83 pending = input_pending(wp->ictx);
84 if (pending == NULL)
85 return (xstrdup(""));
87 line = EVBUFFER_DATA(pending);
88 linelen = EVBUFFER_LENGTH(pending);
90 buf = xstrdup("");
91 if (args_has(args, 'C')) {
92 for (i = 0; i < linelen; i++) {
93 if (line[i] >= ' ' && line[i] != '\\') {
94 tmp[0] = line[i];
95 tmp[1] = '\0';
96 } else
97 xsnprintf(tmp, sizeof tmp, "\\%03hho", line[i]);
98 buf = cmd_capture_pane_append(buf, len, tmp,
99 strlen(tmp));
101 } else
102 buf = cmd_capture_pane_append(buf, len, line, linelen);
103 return (buf);
106 static char *
107 cmd_capture_pane_history(struct args *args, struct cmdq_item *item,
108 struct window_pane *wp, size_t *len)
110 struct grid *gd;
111 const struct grid_line *gl;
112 struct grid_cell *gc = NULL;
113 int n, join_lines, flags = 0;
114 u_int i, sx, top, bottom, tmp;
115 char *cause, *buf, *line;
116 const char *Sflag, *Eflag;
117 size_t linelen;
119 sx = screen_size_x(&wp->base);
120 if (args_has(args, 'a')) {
121 gd = wp->base.saved_grid;
122 if (gd == NULL) {
123 if (!args_has(args, 'q')) {
124 cmdq_error(item, "no alternate screen");
125 return (NULL);
127 return (xstrdup(""));
129 } else
130 gd = wp->base.grid;
132 Sflag = args_get(args, 'S');
133 if (Sflag != NULL && strcmp(Sflag, "-") == 0)
134 top = 0;
135 else {
136 n = args_strtonum_and_expand(args, 'S', INT_MIN, SHRT_MAX,
137 item, &cause);
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;
149 Eflag = args_get(args, 'E');
150 if (Eflag != NULL && strcmp(Eflag, "-") == 0)
151 bottom = gd->hsize + gd->sy - 1;
152 else {
153 n = args_strtonum_and_expand(args, 'E', INT_MIN, SHRT_MAX,
154 item, &cause);
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;
166 if (bottom < top) {
167 tmp = bottom;
168 bottom = top;
169 top = tmp;
172 join_lines = args_has(args, 'J');
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;
182 buf = NULL;
183 for (i = top; i <= bottom; i++) {
184 line = grid_string_cells(gd, 0, i, sx, &gc, flags, wp->screen);
185 linelen = strlen(line);
187 buf = cmd_capture_pane_append(buf, len, line, linelen);
189 gl = grid_peek_line(gd, i);
190 if (!join_lines || !(gl->flags & GRID_LINE_WRAPPED))
191 buf[(*len)++] = '\n';
193 free(line);
195 return (buf);
198 static enum cmd_retval
199 cmd_capture_pane_exec(struct cmd *self, struct cmdq_item *item)
201 struct args *args = cmd_get_args(self);
202 struct client *c = cmdq_get_client(item);
203 struct window_pane *wp = cmdq_get_target(item)->wp;
204 char *buf, *cause;
205 const char *bufname;
206 size_t len;
208 if (cmd_get_entry(self) == &cmd_clear_history_entry) {
209 window_pane_reset_mode_all(wp);
210 grid_clear_history(wp->base.grid);
211 if (args_has(args, 'H'))
212 screen_reset_hyperlinks(wp->screen);
213 return (CMD_RETURN_NORMAL);
216 len = 0;
217 if (args_has(args, 'P'))
218 buf = cmd_capture_pane_pending(args, wp, &len);
219 else
220 buf = cmd_capture_pane_history(args, item, wp, &len);
221 if (buf == NULL)
222 return (CMD_RETURN_ERROR);
224 if (args_has(args, 'p')) {
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);
235 file_print_buffer(c, buf, len);
236 file_print(c, "\n");
237 free(buf);
239 } else {
240 bufname = NULL;
241 if (args_has(args, 'b'))
242 bufname = args_get(args, 'b');
244 if (paste_set(buf, len, bufname, &cause) != 0) {
245 cmdq_error(item, "%s", cause);
246 free(cause);
247 free(buf);
248 return (CMD_RETURN_ERROR);
252 return (CMD_RETURN_NORMAL);