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