cmd-bind-key.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 <string.h>
  22
  23 #include "tmux.h"
  24
  25 /*
  26  * Bind a key to a command, this recurses through cmd_*.
  27  */
  28
  29 int     cmd_bind_key_parse(struct cmd *, int, char **, char **);
  30 int     cmd_bind_key_exec(struct cmd *, struct cmd_ctx *);
  31 void    cmd_bind_key_free(struct cmd *);
  32 size_t  cmd_bind_key_print(struct cmd *, char *, size_t);
  33
  34 int     cmd_bind_key_table(struct cmd *, struct cmd_ctx *);
  35
  36 struct cmd_bind_key_data {
  37         int              key;
  38         int              can_repeat;
  39         struct cmd_list *cmdlist;
  40
  41         int              command_key;
  42         char            *tablename;
  43         char            *modecmd;
  44 };
  45
  46 const struct cmd_entry cmd_bind_key_entry = {
  47         "bind-key", "bind",
  48         "[-cnr] [-t key-table] key command [arguments]",
  49         0, "",
  50         NULL,
  51         cmd_bind_key_parse,
  52         cmd_bind_key_exec,
  53         cmd_bind_key_free,
  54         cmd_bind_key_print
  55 };
  56
  57 int
  58 cmd_bind_key_parse(struct cmd *self, int argc, char **argv, char **cause)
  59 {
  60         struct cmd_bind_key_data        *data;
  61         int                              opt, no_prefix = 0;
  62
  63         self->data = data = xmalloc(sizeof *data);
  64         data->can_repeat = 0;
  65         data->cmdlist = NULL;
  66         data->command_key = 0;
  67         data->tablename = NULL;
  68         data->modecmd = NULL;
  69
  70         while ((opt = getopt(argc, argv, "cnrt:")) != -1) {
  71                 switch (opt) {
  72                 case 'c':
  73                         data->command_key = 1;
  74                         break;
  75                 case 'n':
  76                         no_prefix = 1;
  77                         break;
  78                 case 'r':
  79                         data->can_repeat = 1;
  80                         break;
  81                 case 't':
  82                         if (data->tablename == NULL)
  83                                 data->tablename = xstrdup(optarg);
  84                         break;
  85                 default:
  86                         goto usage;
  87                 }
  88         }
  89         argc -= optind;
  90         argv += optind;
  91         if (argc < 1)
  92                 goto usage;
  93
  94         if ((data->key = key_string_lookup_string(argv[0])) == KEYC_NONE) {
  95                 xasprintf(cause, "unknown key: %s", argv[0]);
  96                 goto error;
  97         }
  98         if (!no_prefix)
  99                 data->key |= KEYC_PREFIX;
 100
 101         argc--;
 102         argv++;
 103         if (data->tablename != NULL) {
 104                 if (argc != 1)
 105                         goto usage;
 106                 data->modecmd = xstrdup(argv[0]);
 107         } else {
 108                 if ((data->cmdlist = cmd_list_parse(argc, argv, cause)) == NULL)
 109                         goto error;
 110         }
 111
 112         return (0);
 113
 114 usage:
 115         xasprintf(cause, "usage: %s %s", self->entry->name, self->entry->usage);
 116
 117 error:
 118         self->entry->free(self);
 119         return (-1);
 120 }
 121
 122 int
 123 cmd_bind_key_exec(struct cmd *self, unused struct cmd_ctx *ctx)
 124 {
 125         struct cmd_bind_key_data        *data = self->data;
 126
 127         if (data == NULL)
 128                 return (0);
 129         if (data->tablename != NULL)
 130                 return (cmd_bind_key_table(self, ctx));
 131
 132         key_bindings_add(data->key, data->can_repeat, data->cmdlist);
 133         data->cmdlist->references++;
 134
 135         return (0);
 136 }
 137
 138 int
 139 cmd_bind_key_table(struct cmd *self, struct cmd_ctx *ctx)
 140 {
 141         struct cmd_bind_key_data        *data = self->data;
 142         const struct mode_key_table     *mtab;
 143         struct mode_key_binding         *mbind, mtmp;
 144         enum mode_key_cmd                cmd;
 145
 146         if ((mtab = mode_key_findtable(data->tablename)) == NULL) {
 147                 ctx->error(ctx, "unknown key table: %s", data->tablename);
 148                 return (-1);
 149         }
 150
 151         cmd = mode_key_fromstring(mtab->cmdstr, data->modecmd);
 152         if (cmd == MODEKEY_NONE) {
 153                 ctx->error(ctx, "unknown command: %s", data->modecmd);
 154                 return (-1);
 155         }
 156
 157         mtmp.key = data->key & ~KEYC_PREFIX;
 158         mtmp.mode = data->command_key ? 1 : 0;
 159         if ((mbind = SPLAY_FIND(mode_key_tree, mtab->tree, &mtmp)) != NULL) {
 160                 mbind->cmd = cmd;
 161                 return (0);
 162         }
 163         mbind = xmalloc(sizeof *mbind);
 164         mbind->key = mtmp.key;
 165         mbind->mode = mtmp.mode;
 166         mbind->cmd = cmd;
 167         SPLAY_INSERT(mode_key_tree, mtab->tree, mbind);
 168         return (0);
 169 }
 170
 171 void
 172 cmd_bind_key_free(struct cmd *self)
 173 {
 174         struct cmd_bind_key_data        *data = self->data;
 175
 176         if (data->cmdlist != NULL)
 177                 cmd_list_free(data->cmdlist);
 178         if (data->tablename != NULL)
 179                 xfree(data->tablename);
 180         if (data->modecmd != NULL)
 181                 xfree(data->modecmd);
 182         xfree(data);
 183 }
 184
 185 size_t
 186 cmd_bind_key_print(struct cmd *self, char *buf, size_t len)
 187 {
 188         struct cmd_bind_key_data        *data = self->data;
 189         size_t                           off = 0;
 190         const char                      *skey;
 191
 192         off += xsnprintf(buf, len, "%s", self->entry->name);
 193         if (data == NULL)
 194                 return (off);
 195
 196         if (off < len && data->command_key)
 197                 off += xsnprintf(buf + off, len - off, " -c");
 198         if (off < len && !(data->key & KEYC_PREFIX))
 199                 off += xsnprintf(buf + off, len - off, " -n");
 200         if (off < len && data->can_repeat)
 201                 off += xsnprintf(buf + off, len - off, " -r");
 202         if (off < len && data->tablename != NULL)
 203                 off += cmd_prarg(buf + off, len - off, " -t ", data->tablename);
 204         if (off < len) {
 205                 skey = key_string_lookup_key(data->key & ~KEYC_PREFIX);
 206                 off += xsnprintf(buf + off, len - off, " %s ", skey);
 207         }
 208         if (off < len)
 209                 off += cmd_list_print(data->cmdlist, buf + off, len - off);
 210         return (off);
 211 }