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