cmd-paste-buffer.c

   1 /* $OpenBSD$ */
   2
   3 /*
   4  * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net>
   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 /*
  28  * Paste paste buffer if present.
  29  */
  30
  31 struct cmd_paste_buffer_data {
  32         char    *target;
  33         int      buffer;
  34
  35         int      flag_delete;
  36         char    *sepstr;
  37 };
  38
  39 void    cmd_paste_buffer_init(struct cmd *, int);
  40 int     cmd_paste_buffer_parse(struct cmd *, int, char **, char **);
  41 int     cmd_paste_buffer_exec(struct cmd *, struct cmd_ctx *);
  42 void    cmd_paste_buffer_filter(
  43             struct window_pane *, const char *, size_t, char *);
  44 void    cmd_paste_buffer_free(struct cmd *);
  45 size_t  cmd_paste_buffer_print(struct cmd *, char *, size_t);
  46
  47 const struct cmd_entry cmd_paste_buffer_entry = {
  48         "paste-buffer", "pasteb",
  49         "[-dr] [-s separator] [-b buffer-index] [-t target-pane]",
  50         0, "",
  51         cmd_paste_buffer_init,
  52         cmd_paste_buffer_parse,
  53         cmd_paste_buffer_exec,
  54         cmd_paste_buffer_free,
  55         cmd_paste_buffer_print
  56 };
  57
  58 /* ARGSUSED */
  59 void
  60 cmd_paste_buffer_init(struct cmd *self, unused int arg)
  61 {
  62         struct cmd_paste_buffer_data    *data;
  63
  64         self->data = data = xmalloc(sizeof *data);
  65         data->target = NULL;
  66         data->buffer = -1;
  67         data->flag_delete = 0;
  68         data->sepstr = xstrdup("\r");
  69 }
  70
  71 int
  72 cmd_paste_buffer_parse(struct cmd *self, int argc, char **argv, char **cause)
  73 {
  74         struct cmd_paste_buffer_data    *data;
  75         int                      opt, n;
  76         const char              *errstr;
  77
  78         cmd_paste_buffer_init(self, 0);
  79         data = self->data;
  80
  81         while ((opt = getopt(argc, argv, "b:ds:t:r")) != -1) {
  82                 switch (opt) {
  83                 case 'b':
  84                         if (data->buffer == -1) {
  85                                 n = strtonum(optarg, 0, INT_MAX, &errstr);
  86                                 if (errstr != NULL) {
  87                                         xasprintf(cause, "buffer %s", errstr);
  88                                         goto error;
  89                                 }
  90                                 data->buffer = n;
  91                         }
  92                         break;
  93                 case 'd':
  94                         data->flag_delete = 1;
  95                         break;
  96                 case 's':
  97                         if (data->sepstr != NULL)
  98                                 xfree(data->sepstr);
  99                         data->sepstr = xstrdup(optarg);
 100                         break;
 101                 case 't':
 102                         if (data->target == NULL)
 103                                 data->target = xstrdup(optarg);
 104                         break;
 105                 case 'r':
 106                         if (data->sepstr != NULL)
 107                                 xfree(data->sepstr);
 108                         data->sepstr = xstrdup("\n");
 109                         break;
 110                 default:
 111                         goto usage;
 112                 }
 113         }
 114         argc -= optind;
 115         argv += optind;
 116
 117         return (0);
 118
 119 usage:
 120         xasprintf(cause, "usage: %s %s", self->entry->name, self->entry->usage);
 121
 122 error:
 123         self->entry->free(self);
 124         return (-1);
 125 }
 126
 127 int
 128 cmd_paste_buffer_exec(struct cmd *self, struct cmd_ctx *ctx)
 129 {
 130         struct cmd_paste_buffer_data    *data = self->data;
 131         struct window_pane              *wp;
 132         struct session                  *s;
 133         struct paste_buffer             *pb;
 134
 135         if (cmd_find_pane(ctx, data->target, &s, &wp) == NULL)
 136                 return (-1);
 137
 138         if (data->buffer == -1)
 139                 pb = paste_get_top(&s->buffers);
 140         else {
 141                 if ((pb = paste_get_index(&s->buffers, data->buffer)) == NULL) {
 142                         ctx->error(ctx, "no buffer %d", data->buffer);
 143                         return (-1);
 144                 }
 145         }
 146
 147         if (pb != NULL)
 148                 cmd_paste_buffer_filter(wp, pb->data, pb->size, data->sepstr);
 149
 150         /* Delete the buffer if -d. */
 151         if (data->flag_delete) {
 152                 if (data->buffer == -1)
 153                         paste_free_top(&s->buffers);
 154                 else
 155                         paste_free_index(&s->buffers, data->buffer);
 156         }
 157
 158         return (0);
 159 }
 160
 161 /* Add bytes to a buffer and filter '\n' according to separator. */
 162 void
 163 cmd_paste_buffer_filter(
 164     struct window_pane *wp, const char *data, size_t size, char *sep)
 165 {
 166         const char      *end = data + size;
 167         const char      *lf;
 168         size_t           seplen;
 169
 170         seplen = strlen(sep);
 171         while ((lf = memchr(data, '\n', end - data)) != NULL) {
 172                 if (lf != data)
 173                         bufferevent_write(wp->event, data, lf - data);
 174                 bufferevent_write(wp->event, sep, seplen);
 175                 data = lf + 1;
 176         }
 177
 178         if (end != data)
 179                 bufferevent_write(wp->event, data, end - data);
 180 }
 181
 182 void
 183 cmd_paste_buffer_free(struct cmd *self)
 184 {
 185         struct cmd_paste_buffer_data    *data = self->data;
 186
 187         if (data->target != NULL)
 188                 xfree(data->target);
 189         if (data->sepstr != NULL)
 190                 xfree(data->sepstr);
 191         xfree(data);
 192 }
 193
 194 size_t
 195 cmd_paste_buffer_print(struct cmd *self, char *buf, size_t len)
 196 {
 197         struct cmd_paste_buffer_data    *data = self->data;
 198         size_t                           off = 0;
 199         char                             tmp[BUFSIZ];
 200         int                              r_flag;
 201
 202         r_flag = 0;
 203         if (data->sepstr != NULL)
 204                 r_flag = (data->sepstr[0] == '\n' && data->sepstr[1] == '\0');
 205
 206         off += xsnprintf(buf, len, "%s", self->entry->name);
 207         if (data == NULL)
 208                 return (off);
 209         if (off < len && data->flag_delete)
 210                 off += xsnprintf(buf + off, len - off, " -d");
 211         if (off < len && r_flag)
 212                 off += xsnprintf(buf + off, len - off, " -r");
 213         if (off < len && data->buffer != -1)
 214                 off += xsnprintf(buf + off, len - off, " -b %d", data->buffer);
 215         if (off < len && data->sepstr != NULL && !r_flag) {
 216                 strnvis(
 217                     tmp, data->sepstr, sizeof tmp, VIS_OCTAL|VIS_TAB|VIS_NL);
 218                 off += cmd_prarg(buf + off, len - off, " -s ", tmp);
 219         }
 220         if (off < len && data->target != NULL)
 221                 off += cmd_prarg(buf + off, len - off, " -t ", data->target);
 222         return (off);
 223 }