|  | /* $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" | 
|  | }; | 
|  |  | 
|  | 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_name(const void *a0, const void *b0) | 
|  | { | 
|  | const struct window_buffer_itemdata *const *a = a0; | 
|  | const struct window_buffer_itemdata *const *b = b0; | 
|  |  | 
|  | return (strcmp((*a)->name, (*b)->name)); | 
|  | } | 
|  |  | 
|  | static int | 
|  | window_buffer_cmp_time(const void *a0, const void *b0) | 
|  | { | 
|  | const struct window_buffer_itemdata *const *a = a0; | 
|  | const struct window_buffer_itemdata *const *b = b0; | 
|  |  | 
|  | if ((*a)->order > (*b)->order) | 
|  | return (-1); | 
|  | if ((*a)->order < (*b)->order) | 
|  | return (1); | 
|  | return (strcmp((*a)->name, (*b)->name)); | 
|  | } | 
|  |  | 
|  | static int | 
|  | window_buffer_cmp_size(const void *a0, const void *b0) | 
|  | { | 
|  | const struct window_buffer_itemdata *const *a = a0; | 
|  | const struct window_buffer_itemdata *const *b = b0; | 
|  |  | 
|  | if ((*a)->size > (*b)->size) | 
|  | return (-1); | 
|  | if ((*a)->size < (*b)->size) | 
|  | return (1); | 
|  | return (strcmp((*a)->name, (*b)->name)); | 
|  | } | 
|  |  | 
|  | static void | 
|  | window_buffer_build(void *modedata, u_int sort_type, __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); | 
|  | } | 
|  |  | 
|  | switch (sort_type) { | 
|  | case WINDOW_BUFFER_BY_NAME: | 
|  | qsort(data->item_list, data->item_size, sizeof *data->item_list, | 
|  | window_buffer_cmp_name); | 
|  | break; | 
|  | case WINDOW_BUFFER_BY_TIME: | 
|  | qsort(data->item_list, data->item_size, sizeof *data->item_list, | 
|  | window_buffer_cmp_time); | 
|  | break; | 
|  | case WINDOW_BUFFER_BY_SIZE: | 
|  | qsort(data->item_list, data->item_size, sizeof *data->item_list, | 
|  | window_buffer_cmp_size); | 
|  | break; | 
|  | } | 
|  |  | 
|  | 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; | 
|  | } | 
|  | } |