1 /* $OpenBSD$ */ 2 3 /* 4 * Copyright (c) 2017 Nicholas Marriott <nicholas.marriott@gmail.com> 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 #include <time.h> 24 25 #include "tmux.h" 26 27 static struct screen *window_buffer_init(struct window_pane *, 28 struct cmd_find_state *, struct args *); 29 static void window_buffer_free(struct window_pane *); 30 static void window_buffer_resize(struct window_pane *, u_int, 31 u_int); 32 static void window_buffer_key(struct window_pane *, 33 struct client *, struct session *, key_code, 34 struct mouse_event *); 35 36 #define WINDOW_BUFFER_DEFAULT_COMMAND "paste-buffer -b '%%'" 37 38 #define WINDOW_BUFFER_DEFAULT_FORMAT \ 39 "#{buffer_size} bytes (#{t:buffer_created})" 40 41 const struct window_mode window_buffer_mode = { 42 .name = "buffer-mode", 43 44 .init = window_buffer_init, 45 .free = window_buffer_free, 46 .resize = window_buffer_resize, 47 .key = window_buffer_key, 48 }; 49 50 enum window_buffer_sort_type { 51 WINDOW_BUFFER_BY_TIME, 52 WINDOW_BUFFER_BY_NAME, 53 WINDOW_BUFFER_BY_SIZE, 54 }; 55 static const char *window_buffer_sort_list[] = { 56 "time", 57 "name", 58 "size" 59 }; 60 61 struct window_buffer_itemdata { 62 const char *name; 63 u_int order; 64 size_t size; 65 }; 66 67 struct window_buffer_modedata { 68 struct mode_tree_data *data; 69 char *command; 70 char *format; 71 72 struct window_buffer_itemdata **item_list; 73 u_int item_size; 74 }; 75 76 static struct window_buffer_itemdata * 77 window_buffer_add_item(struct window_buffer_modedata *data) 78 { 79 struct window_buffer_itemdata *item; 80 81 data->item_list = xreallocarray(data->item_list, data->item_size + 1, 82 sizeof *data->item_list); 83 item = data->item_list[data->item_size++] = xcalloc(1, sizeof *item); 84 return (item); 85 } 86 87 static void 88 window_buffer_free_item(struct window_buffer_itemdata *item) 89 { 90 free(__UNCONST(item->name)); 91 free(item); 92 } 93 94 static int 95 window_buffer_cmp_name(const void *a0, const void *b0) 96 { 97 const struct window_buffer_itemdata *const *a = a0; 98 const struct window_buffer_itemdata *const *b = b0; 99 100 return (strcmp((*a)->name, (*b)->name)); 101 } 102 103 static int 104 window_buffer_cmp_time(const void *a0, const void *b0) 105 { 106 const struct window_buffer_itemdata *const *a = a0; 107 const struct window_buffer_itemdata *const *b = b0; 108 109 if ((*a)->order > (*b)->order) 110 return (-1); 111 if ((*a)->order < (*b)->order) 112 return (1); 113 return (strcmp((*a)->name, (*b)->name)); 114 } 115 116 static int 117 window_buffer_cmp_size(const void *a0, const void *b0) 118 { 119 const struct window_buffer_itemdata *const *a = a0; 120 const struct window_buffer_itemdata *const *b = b0; 121 122 if ((*a)->size > (*b)->size) 123 return (-1); 124 if ((*a)->size < (*b)->size) 125 return (1); 126 return (strcmp((*a)->name, (*b)->name)); 127 } 128 129 static void 130 window_buffer_build(void *modedata, u_int sort_type, __unused uint64_t *tag, 131 const char *filter) 132 { 133 struct window_buffer_modedata *data = modedata; 134 struct window_buffer_itemdata *item; 135 u_int i; 136 struct paste_buffer *pb; 137 char *text, *cp; 138 struct format_tree *ft; 139 140 for (i = 0; i < data->item_size; i++) 141 window_buffer_free_item(data->item_list[i]); 142 free(data->item_list); 143 data->item_list = NULL; 144 data->item_size = 0; 145 146 pb = NULL; 147 while ((pb = paste_walk(pb)) != NULL) { 148 item = window_buffer_add_item(data); 149 item->name = xstrdup(paste_buffer_name(pb)); 150 paste_buffer_data(pb, &item->size); 151 item->order = paste_buffer_order(pb); 152 } 153 154 switch (sort_type) { 155 case WINDOW_BUFFER_BY_NAME: 156 qsort(data->item_list, data->item_size, sizeof *data->item_list, 157 window_buffer_cmp_name); 158 break; 159 case WINDOW_BUFFER_BY_TIME: 160 qsort(data->item_list, data->item_size, sizeof *data->item_list, 161 window_buffer_cmp_time); 162 break; 163 case WINDOW_BUFFER_BY_SIZE: 164 qsort(data->item_list, data->item_size, sizeof *data->item_list, 165 window_buffer_cmp_size); 166 break; 167 } 168 169 for (i = 0; i < data->item_size; i++) { 170 item = data->item_list[i]; 171 172 pb = paste_get_name(item->name); 173 if (pb == NULL) 174 continue; 175 ft = format_create(NULL, NULL, FORMAT_NONE, 0); 176 format_defaults_paste_buffer(ft, pb); 177 178 if (filter != NULL) { 179 cp = format_expand(ft, filter); 180 if (!format_true(cp)) { 181 free(cp); 182 format_free(ft); 183 continue; 184 } 185 free(cp); 186 } 187 188 text = format_expand(ft, data->format); 189 mode_tree_add(data->data, NULL, item, item->order, item->name, 190 text, -1); 191 free(text); 192 193 format_free(ft); 194 } 195 196 } 197 198 static void 199 window_buffer_draw(__unused void *modedata, void *itemdata, 200 struct screen_write_ctx *ctx, u_int sx, u_int sy) 201 { 202 struct window_buffer_itemdata *item = itemdata; 203 struct paste_buffer *pb; 204 char line[1024]; 205 const char *pdata, *end, *cp; 206 size_t psize, at; 207 u_int i, cx = ctx->s->cx, cy = ctx->s->cy; 208 209 pb = paste_get_name(item->name); 210 if (pb == NULL) 211 return; 212 213 pdata = end = paste_buffer_data(pb, &psize); 214 for (i = 0; i < sy; i++) { 215 at = 0; 216 while (end != pdata + psize && *end != '\n') { 217 if ((sizeof line) - at > 5) { 218 cp = vis(line + at, *end, VIS_TAB|VIS_OCTAL, 0); 219 at = cp - line; 220 } 221 end++; 222 } 223 if (at > sx) 224 at = sx; 225 line[at] = '\0'; 226 227 if (*line != '\0') { 228 screen_write_cursormove(ctx, cx, cy + i); 229 screen_write_puts(ctx, &grid_default_cell, "%s", line); 230 } 231 232 if (end == pdata + psize) 233 break; 234 end++; 235 } 236 } 237 238 static int 239 window_buffer_search(__unused void *modedata, void *itemdata, const char *ss) 240 { 241 struct window_buffer_itemdata *item = itemdata; 242 struct paste_buffer *pb; 243 const char *bufdata; 244 size_t bufsize; 245 246 if ((pb = paste_get_name(item->name)) == NULL) 247 return (0); 248 if (strstr(item->name, ss) != NULL) 249 return (1); 250 bufdata = paste_buffer_data(pb, &bufsize); 251 return (memmem(bufdata, bufsize, ss, strlen(ss)) != NULL); 252 } 253 254 static struct screen * 255 window_buffer_init(struct window_pane *wp, __unused struct cmd_find_state *fs, 256 struct args *args) 257 { 258 struct window_buffer_modedata *data; 259 struct screen *s; 260 261 wp->modedata = data = xcalloc(1, sizeof *data); 262 263 if (args == NULL || !args_has(args, 'F')) 264 data->format = xstrdup(WINDOW_BUFFER_DEFAULT_FORMAT); 265 else 266 data->format = xstrdup(args_get(args, 'F')); 267 if (args == NULL || args->argc == 0) 268 data->command = xstrdup(WINDOW_BUFFER_DEFAULT_COMMAND); 269 else 270 data->command = xstrdup(args->argv[0]); 271 272 data->data = mode_tree_start(wp, args, window_buffer_build, 273 window_buffer_draw, window_buffer_search, data, 274 window_buffer_sort_list, nitems(window_buffer_sort_list), &s); 275 mode_tree_zoom(data->data, args); 276 277 mode_tree_build(data->data); 278 mode_tree_draw(data->data); 279 280 return (s); 281 } 282 283 static void 284 window_buffer_free(struct window_pane *wp) 285 { 286 struct window_buffer_modedata *data = wp->modedata; 287 u_int i; 288 289 if (data == NULL) 290 return; 291 292 mode_tree_free(data->data); 293 294 for (i = 0; i < data->item_size; i++) 295 window_buffer_free_item(data->item_list[i]); 296 free(data->item_list); 297 298 free(data->format); 299 free(data->command); 300 301 free(data); 302 } 303 304 static void 305 window_buffer_resize(struct window_pane *wp, u_int sx, u_int sy) 306 { 307 struct window_buffer_modedata *data = wp->modedata; 308 309 mode_tree_resize(data->data, sx, sy); 310 } 311 312 static void 313 window_buffer_do_delete(void* modedata, void *itemdata, 314 __unused struct client *c, __unused key_code key) 315 { 316 struct window_buffer_modedata *data = modedata; 317 struct window_buffer_itemdata *item = itemdata; 318 struct paste_buffer *pb; 319 320 if (item == mode_tree_get_current(data->data)) 321 mode_tree_down(data->data, 0); 322 if ((pb = paste_get_name(item->name)) != NULL) 323 paste_free(pb); 324 } 325 326 static void 327 window_buffer_do_paste(void* modedata, void *itemdata, struct client *c, 328 __unused key_code key) 329 { 330 struct window_buffer_modedata *data = modedata; 331 struct window_buffer_itemdata *item = itemdata; 332 struct paste_buffer *pb; 333 334 if ((pb = paste_get_name(item->name)) != NULL) 335 mode_tree_run_command(c, NULL, data->command, item->name); 336 } 337 338 static void 339 window_buffer_key(struct window_pane *wp, struct client *c, 340 __unused struct session *s, key_code key, struct mouse_event *m) 341 { 342 struct window_buffer_modedata *data = wp->modedata; 343 struct mode_tree_data *mtd = data->data; 344 struct window_buffer_itemdata *item; 345 int finished; 346 347 finished = mode_tree_key(mtd, c, &key, m, NULL, NULL); 348 switch (key) { 349 case 'd': 350 item = mode_tree_get_current(mtd); 351 window_buffer_do_delete(data, item, c, key); 352 mode_tree_build(mtd); 353 break; 354 case 'D': 355 mode_tree_each_tagged(mtd, window_buffer_do_delete, c, key, 0); 356 mode_tree_build(mtd); 357 break; 358 case 'P': 359 mode_tree_each_tagged(mtd, window_buffer_do_paste, c, key, 0); 360 finished = 1; 361 break; 362 case 'p': 363 case '\r': 364 item = mode_tree_get_current(mtd); 365 window_buffer_do_paste(data, item, c, key); 366 finished = 1; 367 break; 368 } 369 if (finished || paste_get_top(NULL) == NULL) 370 window_pane_reset_mode(wp); 371 else { 372 mode_tree_draw(mtd); 373 wp->flags |= PANE_REDRAW; 374 } 375 } 376