1/* $OpenBSD$ */
2
3/*
4 * Copyright (c) 2009 Jonathan Alvarado <[email protected]>
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
30static enum cmd_retval cmd_capture_pane_exec(struct cmd *, struct cmdq_item *);
31
32static char *cmd_capture_pane_append(char *, size_t *, char *, size_t);
33static char *cmd_capture_pane_pending(struct args *, struct window_pane *,
34 size_t *);
35static char *cmd_capture_pane_history(struct args *, struct cmdq_item *,
36 struct window_pane *, size_t *);
37
38const struct cmd_entry cmd_capture_pane_entry = {
39 .name = "capture-pane",
40 .alias = "capturep",
41
42 .args = { "ab:CeE:JNpPqS:t:", 0, 0 },
43 .usage = "[-aCeJNpPq] " CMD_BUFFER_USAGE " [-E end-line] "
44 "[-S start-line] " CMD_TARGET_PANE_USAGE,
45
46 .target = { 't', CMD_FIND_PANE, 0 },
47
48 .flags = CMD_AFTERHOOK,
49 .exec = cmd_capture_pane_exec
50};
51
52const struct cmd_entry cmd_clear_history_entry = {
53 .name = "clear-history",
54 .alias = "clearhist",
55
56 .args = { "t:", 0, 0 },
57 .usage = CMD_TARGET_PANE_USAGE,
58
59 .target = { 't', CMD_FIND_PANE, 0 },
60
61 .flags = CMD_AFTERHOOK,
62 .exec = cmd_capture_pane_exec
63};
64
65static char *
66cmd_capture_pane_append(char *buf, size_t *len, char *line, size_t linelen)
67{
68 buf = xrealloc(buf, *len + linelen + 1);
69 memcpy(buf + *len, line, linelen);
70 *len += linelen;
71 return (buf);
72}
73
74static char *
75cmd_capture_pane_pending(struct args *args, struct window_pane *wp,
76 size_t *len)
77{
78 struct evbuffer *pending;
79 char *buf, *line, tmp[5];
80 size_t linelen;
81 u_int i;
82
83 pending = input_pending(wp);
84 if (pending == NULL)
85 return (xstrdup(""));
86
87 line = EVBUFFER_DATA(pending);
88 linelen = EVBUFFER_LENGTH(pending);
89
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));
100 }
101 } else
102 buf = cmd_capture_pane_append(buf, len, line, linelen);
103 return (buf);
104}
105
106static char *
107cmd_capture_pane_history(struct args *args, struct cmdq_item *item,
108 struct window_pane *wp, size_t *len)
109{
110 struct grid *gd;
111 const struct grid_line *gl;
112 struct grid_cell *gc = NULL;
113 int n, with_codes, escape_c0, join_lines, no_trim;
114 u_int i, sx, top, bottom, tmp;
115 char *cause, *buf, *line;
116 const char *Sflag, *Eflag;
117 size_t linelen;
118
119 sx = screen_size_x(&wp->base);
120 if (args_has(args, 'a')) {
121 gd = wp->saved_grid;
122 if (gd == NULL) {
123 if (!args_has(args, 'q')) {
124 cmdq_error(item, "no alternate screen");
125 return (NULL);
126 }
127 return (xstrdup(""));
128 }
129 } else
130 gd = wp->base.grid;
131
132 Sflag = args_get(args, 'S');
133 if (Sflag != NULL && strcmp(Sflag, "-") == 0)
134 top = 0;
135 else {
136 n = args_strtonum(args, 'S', INT_MIN, SHRT_MAX, &cause);
137 if (cause != NULL) {
138 top = gd->hsize;
139 free(cause);
140 } else if (n < 0 && (u_int) -n > gd->hsize)
141 top = 0;
142 else
143 top = gd->hsize + n;
144 if (top > gd->hsize + gd->sy - 1)
145 top = gd->hsize + gd->sy - 1;
146 }
147
148 Eflag = args_get(args, 'E');
149 if (Eflag != NULL && strcmp(Eflag, "-") == 0)
150 bottom = gd->hsize + gd->sy - 1;
151 else {
152 n = args_strtonum(args, 'E', INT_MIN, SHRT_MAX, &cause);
153 if (cause != NULL) {
154 bottom = gd->hsize + gd->sy - 1;
155 free(cause);
156 } else if (n < 0 && (u_int) -n > gd->hsize)
157 bottom = 0;
158 else
159 bottom = gd->hsize + n;
160 if (bottom > gd->hsize + gd->sy - 1)
161 bottom = gd->hsize + gd->sy - 1;
162 }
163
164 if (bottom < top) {
165 tmp = bottom;
166 bottom = top;
167 top = tmp;
168 }
169
170 with_codes = args_has(args, 'e');
171 escape_c0 = args_has(args, 'C');
172 join_lines = args_has(args, 'J');
173 no_trim = args_has(args, 'N');
174
175 buf = NULL;
176 for (i = top; i <= bottom; i++) {
177 line = grid_string_cells(gd, 0, i, sx, &gc, with_codes,
178 escape_c0, !join_lines && !no_trim);
179 linelen = strlen(line);
180
181 buf = cmd_capture_pane_append(buf, len, line, linelen);
182
183 gl = grid_peek_line(gd, i);
184 if (!join_lines || !(gl->flags & GRID_LINE_WRAPPED))
185 buf[(*len)++] = '\n';
186
187 free(line);
188 }
189 return (buf);
190}
191
192static enum cmd_retval
193cmd_capture_pane_exec(struct cmd *self, struct cmdq_item *item)
194{
195 struct args *args = self->args;
196 struct client *c = item->client;
197 struct window_pane *wp = item->target.wp;
198 char *buf, *cause;
199 const char *bufname;
200 size_t len;
201
202 if (self->entry == &cmd_clear_history_entry) {
203 window_pane_reset_mode_all(wp);
204 grid_clear_history(wp->base.grid);
205 return (CMD_RETURN_NORMAL);
206 }
207
208 len = 0;
209 if (args_has(args, 'P'))
210 buf = cmd_capture_pane_pending(args, wp, &len);
211 else
212 buf = cmd_capture_pane_history(args, item, wp, &len);
213 if (buf == NULL)
214 return (CMD_RETURN_ERROR);
215
216 if (args_has(args, 'p')) {
217 if (!file_can_print(c)) {
218 cmdq_error(item, "can't write output to client");
219 free(buf);
220 return (CMD_RETURN_ERROR);
221 }
222 file_print_buffer(c, buf, len);
223 if (args_has(args, 'P') && len > 0)
224 file_print(c, "\n");
225 free(buf);
226 } else {
227 bufname = NULL;
228 if (args_has(args, 'b'))
229 bufname = args_get(args, 'b');
230
231 if (paste_set(buf, len, bufname, &cause) != 0) {
232 cmdq_error(item, "%s", cause);
233 free(cause);
234 free(buf);
235 return (CMD_RETURN_ERROR);
236 }
237 }
238
239 return (CMD_RETURN_NORMAL);
240}
241