| /* $OpenBSD$ */ |
| |
| /* |
| * Copyright (c) 2017 Nicholas Marriott <nicholas.marriott@gmail.com> |
| * |
| * Permission to use, copy, modify, and distribute this software for any |
| * purpose with or without fee is hereby granted, provided that the above |
| * copyright notice and this permission notice appear in all copies. |
| * |
| * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES |
| * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
| * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR |
| * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
| * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER |
| * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING |
| * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. |
| */ |
| |
| #include <sys/types.h> |
| |
| #include <stdlib.h> |
| #include <string.h> |
| #include <time.h> |
| |
| #include "tmux.h" |
| |
| static struct screen *window_buffer_init(struct window_mode_entry *, |
| struct cmd_find_state *, struct args *); |
| static void window_buffer_free(struct window_mode_entry *); |
| static void window_buffer_resize(struct window_mode_entry *, u_int, |
| u_int); |
| static void window_buffer_key(struct window_mode_entry *, |
| struct client *, struct session *, |
| struct winlink *, key_code, struct mouse_event *); |
| |
| #define WINDOW_BUFFER_DEFAULT_COMMAND "paste-buffer -b '%%'" |
| |
| #define WINDOW_BUFFER_DEFAULT_FORMAT \ |
| "#{buffer_size} bytes (#{t:buffer_created})" |
| |
| static const struct menu_item window_buffer_menu_items[] = { |
| { "Paste", 'p', NULL }, |
| { "Paste Tagged", 'P', NULL }, |
| { "", KEYC_NONE, NULL }, |
| { "Tag", 't', NULL }, |
| { "Tag All", '\024', NULL }, |
| { "Tag None", 'T', NULL }, |
| { "", KEYC_NONE, NULL }, |
| { "Delete", 'd', NULL }, |
| { "Delete Tagged", 'D', NULL }, |
| { "", KEYC_NONE, NULL }, |
| { "Cancel", 'q', NULL }, |
| |
| { NULL, KEYC_NONE, NULL } |
| }; |
| |
| const struct window_mode window_buffer_mode = { |
| .name = "buffer-mode", |
| .default_format = WINDOW_BUFFER_DEFAULT_FORMAT, |
| |
| .init = window_buffer_init, |
| .free = window_buffer_free, |
| .resize = window_buffer_resize, |
| .key = window_buffer_key, |
| }; |
| |
| enum window_buffer_sort_type { |
| WINDOW_BUFFER_BY_TIME, |
| WINDOW_BUFFER_BY_NAME, |
| WINDOW_BUFFER_BY_SIZE, |
| }; |
| static const char *window_buffer_sort_list[] = { |
| "time", |
| "name", |
| "size" |
| }; |
| static struct mode_tree_sort_criteria *window_buffer_sort; |
| |
| struct window_buffer_itemdata { |
| const char *name; |
| u_int order; |
| size_t size; |
| }; |
| |
| struct window_buffer_modedata { |
| struct window_pane *wp; |
| struct cmd_find_state fs; |
| |
| struct mode_tree_data *data; |
| char *command; |
| char *format; |
| |
| struct window_buffer_itemdata **item_list; |
| u_int item_size; |
| }; |
| |
| static struct window_buffer_itemdata * |
| window_buffer_add_item(struct window_buffer_modedata *data) |
| { |
| struct window_buffer_itemdata *item; |
| |
| data->item_list = xreallocarray(data->item_list, data->item_size + 1, |
| sizeof *data->item_list); |
| item = data->item_list[data->item_size++] = xcalloc(1, sizeof *item); |
| return (item); |
| } |
| |
| static void |
| window_buffer_free_item(struct window_buffer_itemdata *item) |
| { |
| free((void *)item->name); |
| free(item); |
| } |
| |
| static int |
| window_buffer_cmp(const void *a0, const void *b0) |
| { |
| const struct window_buffer_itemdata *const *a = a0; |
| const struct window_buffer_itemdata *const *b = b0; |
| int result = 0; |
| |
| if (window_buffer_sort->field == WINDOW_BUFFER_BY_TIME) |
| result = (*b)->order - (*a)->order; |
| else if (window_buffer_sort->field == WINDOW_BUFFER_BY_SIZE) |
| result = (*b)->size - (*a)->size; |
| |
| /* Use WINDOW_BUFFER_BY_NAME as default order and tie breaker. */ |
| if (result == 0) |
| result = strcmp((*a)->name, (*b)->name); |
| |
| if (window_buffer_sort->reversed) |
| result = -result; |
| return (result); |
| } |
| |
| static void |
| window_buffer_build(void *modedata, struct mode_tree_sort_criteria *sort_crit, |
| __unused uint64_t *tag, const char *filter) |
| { |
| struct window_buffer_modedata *data = modedata; |
| struct window_buffer_itemdata *item; |
| u_int i; |
| struct paste_buffer *pb; |
| char *text, *cp; |
| struct format_tree *ft; |
| struct session *s = NULL; |
| struct winlink *wl = NULL; |
| struct window_pane *wp = NULL; |
| |
| for (i = 0; i < data->item_size; i++) |
| window_buffer_free_item(data->item_list[i]); |
| free(data->item_list); |
| data->item_list = NULL; |
| data->item_size = 0; |
| |
| pb = NULL; |
| while ((pb = paste_walk(pb)) != NULL) { |
| item = window_buffer_add_item(data); |
| item->name = xstrdup(paste_buffer_name(pb)); |
| paste_buffer_data(pb, &item->size); |
| item->order = paste_buffer_order(pb); |
| } |
| |
| window_buffer_sort = sort_crit; |
| qsort(data->item_list, data->item_size, sizeof *data->item_list, |
| window_buffer_cmp); |
| |
| if (cmd_find_valid_state(&data->fs)) { |
| s = data->fs.s; |
| wl = data->fs.wl; |
| wp = data->fs.wp; |
| } |
| |
| for (i = 0; i < data->item_size; i++) { |
| item = data->item_list[i]; |
| |
| pb = paste_get_name(item->name); |
| if (pb == NULL) |
| continue; |
| ft = format_create(NULL, NULL, FORMAT_NONE, 0); |
| format_defaults(ft, NULL, s, wl, wp); |
| format_defaults_paste_buffer(ft, pb); |
| |
| if (filter != NULL) { |
| cp = format_expand(ft, filter); |
| if (!format_true(cp)) { |
| free(cp); |
| format_free(ft); |
| continue; |
| } |
| free(cp); |
| } |
| |
| text = format_expand(ft, data->format); |
| mode_tree_add(data->data, NULL, item, item->order, item->name, |
| text, -1); |
| free(text); |
| |
| format_free(ft); |
| } |
| |
| } |
| |
| static void |
| window_buffer_draw(__unused void *modedata, void *itemdata, |
| struct screen_write_ctx *ctx, u_int sx, u_int sy) |
| { |
| struct window_buffer_itemdata *item = itemdata; |
| struct paste_buffer *pb; |
| char line[1024]; |
| const char *pdata, *end, *cp; |
| size_t psize, at; |
| u_int i, cx = ctx->s->cx, cy = ctx->s->cy; |
| |
| pb = paste_get_name(item->name); |
| if (pb == NULL) |
| return; |
| |
| pdata = end = paste_buffer_data(pb, &psize); |
| for (i = 0; i < sy; i++) { |
| at = 0; |
| while (end != pdata + psize && *end != '\n') { |
| if ((sizeof line) - at > 5) { |
| cp = vis(line + at, *end, VIS_OCTAL|VIS_TAB, 0); |
| at = cp - line; |
| } |
| end++; |
| } |
| if (at > sx) |
| at = sx; |
| line[at] = '\0'; |
| |
| if (*line != '\0') { |
| screen_write_cursormove(ctx, cx, cy + i, 0); |
| screen_write_puts(ctx, &grid_default_cell, "%s", line); |
| } |
| |
| if (end == pdata + psize) |
| break; |
| end++; |
| } |
| } |
| |
| static int |
| window_buffer_search(__unused void *modedata, void *itemdata, const char *ss) |
| { |
| struct window_buffer_itemdata *item = itemdata; |
| struct paste_buffer *pb; |
| const char *bufdata; |
| size_t bufsize; |
| |
| if ((pb = paste_get_name(item->name)) == NULL) |
| return (0); |
| if (strstr(item->name, ss) != NULL) |
| return (1); |
| bufdata = paste_buffer_data(pb, &bufsize); |
| return (memmem(bufdata, bufsize, ss, strlen(ss)) != NULL); |
| } |
| |
| static void |
| window_buffer_menu(void *modedata, struct client *c, key_code key) |
| { |
| struct window_buffer_modedata *data = modedata; |
| struct window_pane *wp = data->wp; |
| struct window_mode_entry *wme; |
| |
| wme = TAILQ_FIRST(&wp->modes); |
| if (wme == NULL || wme->data != modedata) |
| return; |
| window_buffer_key(wme, c, NULL, NULL, key, NULL); |
| } |
| |
| static struct screen * |
| window_buffer_init(struct window_mode_entry *wme, struct cmd_find_state *fs, |
| struct args *args) |
| { |
| struct window_pane *wp = wme->wp; |
| struct window_buffer_modedata *data; |
| struct screen *s; |
| |
| wme->data = data = xcalloc(1, sizeof *data); |
| data->wp = wp; |
| cmd_find_copy_state(&data->fs, fs); |
| |
| if (args == NULL || !args_has(args, 'F')) |
| data->format = xstrdup(WINDOW_BUFFER_DEFAULT_FORMAT); |
| else |
| data->format = xstrdup(args_get(args, 'F')); |
| if (args == NULL || args->argc == 0) |
| data->command = xstrdup(WINDOW_BUFFER_DEFAULT_COMMAND); |
| else |
| data->command = xstrdup(args->argv[0]); |
| |
| data->data = mode_tree_start(wp, args, window_buffer_build, |
| window_buffer_draw, window_buffer_search, window_buffer_menu, data, |
| window_buffer_menu_items, window_buffer_sort_list, |
| nitems(window_buffer_sort_list), &s); |
| mode_tree_zoom(data->data, args); |
| |
| mode_tree_build(data->data); |
| mode_tree_draw(data->data); |
| |
| return (s); |
| } |
| |
| static void |
| window_buffer_free(struct window_mode_entry *wme) |
| { |
| struct window_buffer_modedata *data = wme->data; |
| u_int i; |
| |
| if (data == NULL) |
| return; |
| |
| mode_tree_free(data->data); |
| |
| for (i = 0; i < data->item_size; i++) |
| window_buffer_free_item(data->item_list[i]); |
| free(data->item_list); |
| |
| free(data->format); |
| free(data->command); |
| |
| free(data); |
| } |
| |
| static void |
| window_buffer_resize(struct window_mode_entry *wme, u_int sx, u_int sy) |
| { |
| struct window_buffer_modedata *data = wme->data; |
| |
| mode_tree_resize(data->data, sx, sy); |
| } |
| |
| static void |
| window_buffer_do_delete(void *modedata, void *itemdata, |
| __unused struct client *c, __unused key_code key) |
| { |
| struct window_buffer_modedata *data = modedata; |
| struct window_buffer_itemdata *item = itemdata; |
| struct paste_buffer *pb; |
| |
| if (item == mode_tree_get_current(data->data)) |
| mode_tree_down(data->data, 0); |
| if ((pb = paste_get_name(item->name)) != NULL) |
| paste_free(pb); |
| } |
| |
| static void |
| window_buffer_do_paste(void *modedata, void *itemdata, struct client *c, |
| __unused key_code key) |
| { |
| struct window_buffer_modedata *data = modedata; |
| struct window_buffer_itemdata *item = itemdata; |
| struct paste_buffer *pb; |
| |
| if ((pb = paste_get_name(item->name)) != NULL) |
| mode_tree_run_command(c, NULL, data->command, item->name); |
| } |
| |
| static void |
| window_buffer_key(struct window_mode_entry *wme, struct client *c, |
| __unused struct session *s, __unused struct winlink *wl, key_code key, |
| struct mouse_event *m) |
| { |
| struct window_pane *wp = wme->wp; |
| struct window_buffer_modedata *data = wme->data; |
| struct mode_tree_data *mtd = data->data; |
| struct window_buffer_itemdata *item; |
| int finished; |
| |
| finished = mode_tree_key(mtd, c, &key, m, NULL, NULL); |
| switch (key) { |
| case 'd': |
| item = mode_tree_get_current(mtd); |
| window_buffer_do_delete(data, item, c, key); |
| mode_tree_build(mtd); |
| break; |
| case 'D': |
| mode_tree_each_tagged(mtd, window_buffer_do_delete, c, key, 0); |
| mode_tree_build(mtd); |
| break; |
| case 'P': |
| mode_tree_each_tagged(mtd, window_buffer_do_paste, c, key, 0); |
| finished = 1; |
| break; |
| case 'p': |
| case '\r': |
| item = mode_tree_get_current(mtd); |
| window_buffer_do_paste(data, item, c, key); |
| finished = 1; |
| break; |
| } |
| if (finished || paste_get_top(NULL) == NULL) |
| window_pane_reset_mode(wp); |
| else { |
| mode_tree_draw(mtd); |
| wp->flags |= PANE_REDRAW; |
| } |
| } |