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 <stdlib.h>
  22 #include <string.h>
  23
  24 #include "tmux.h"
  25
  26 /*
  27  * Bind a key to a command, this recurses through cmd_*.
  28  */
  29
  30 enum cmd_retval  cmd_bind_key_check(struct args *);
  31 enum cmd_retval  cmd_bind_key_exec(struct cmd *, struct cmd_q *);
  32
  33 enum cmd_retval  cmd_bind_key_table(struct cmd *, struct cmd_q *, int);
  34
  35 const struct cmd_entry cmd_bind_key_entry = {
  36         "bind-key", "bind",
  37         "cnrt:", 1, -1,
  38         "[-cnr] [-t key-table] key command [arguments]",
  39         0,
  40         NULL,
  41         cmd_bind_key_check,
  42         cmd_bind_key_exec
  43 };
  44
  45 enum cmd_retval
  46 cmd_bind_key_check(struct args *args)
  47 {
  48         if (args_has(args, 't')) {
  49                 if (args->argc != 2 && args->argc != 3)
  50                         return (CMD_RETURN_ERROR);
  51         } else {
  52                 if (args->argc < 2)
  53                         return (CMD_RETURN_ERROR);
  54         }
  55         return (CMD_RETURN_NORMAL);
  56 }
  57
  58 enum cmd_retval
  59 cmd_bind_key_exec(struct cmd *self, struct cmd_q *cmdq)
  60 {
  61         struct args     *args = self->args;
  62         char            *cause;
  63         struct cmd_list *cmdlist;
  64         int              key;
  65
  66         key = key_string_lookup_string(args->argv[0]);
  67         if (key == KEYC_NONE) {
  68                 cmdq_error(cmdq, "unknown key: %s", args->argv[0]);
  69                 return (CMD_RETURN_ERROR);
  70         }
  71
  72         if (args_has(args, 't'))
  73                 return (cmd_bind_key_table(self, cmdq, key));
  74
  75         cmdlist = cmd_list_parse(args->argc - 1, args->argv + 1, NULL, 0,
  76             &cause);
  77         if (cmdlist == NULL) {
  78                 cmdq_error(cmdq, "%s", cause);
  79                 free(cause);
  80                 return (CMD_RETURN_ERROR);
  81         }
  82
  83         if (!args_has(args, 'n'))
  84             key |= KEYC_PREFIX;
  85         key_bindings_add(key, args_has(args, 'r'), cmdlist);
  86         return (CMD_RETURN_NORMAL);
  87 }
  88
  89 enum cmd_retval
  90 cmd_bind_key_table(struct cmd *self, struct cmd_q *cmdq, int key)
  91 {
  92         struct args                     *args = self->args;
  93         const char                      *tablename;
  94         const struct mode_key_table     *mtab;
  95         struct mode_key_binding         *mbind, mtmp;
  96         enum mode_key_cmd                cmd;
  97         const char                      *arg;
  98
  99         tablename = args_get(args, 't');
 100         if ((mtab = mode_key_findtable(tablename)) == NULL) {
 101                 cmdq_error(cmdq, "unknown key table: %s", tablename);
 102                 return (CMD_RETURN_ERROR);
 103         }
 104
 105         cmd = mode_key_fromstring(mtab->cmdstr, args->argv[1]);
 106         if (cmd == MODEKEY_NONE) {
 107                 cmdq_error(cmdq, "unknown command: %s", args->argv[1]);
 108                 return (CMD_RETURN_ERROR);
 109         }
 110
 111         if (cmd != MODEKEYCOPY_COPYPIPE) {
 112                 if (args->argc != 2) {
 113                         cmdq_error(cmdq, "no argument allowed");
 114                         return (CMD_RETURN_ERROR);
 115                 }
 116                 arg = NULL;
 117         } else {
 118                 if (args->argc != 3) {
 119                         cmdq_error(cmdq, "no argument given");
 120                         return (CMD_RETURN_ERROR);
 121                 }
 122                 arg = args->argv[2];
 123         }
 124
 125         mtmp.key = key;
 126         mtmp.mode = !!args_has(args, 'c');
 127         if ((mbind = RB_FIND(mode_key_tree, mtab->tree, &mtmp)) == NULL) {
 128                 mbind = xmalloc(sizeof *mbind);
 129                 mbind->key = mtmp.key;
 130                 mbind->mode = mtmp.mode;
 131                 RB_INSERT(mode_key_tree, mtab->tree, mbind);
 132         }
 133         mbind->cmd = cmd;
 134         mbind->arg = arg != NULL ? xstrdup(arg) : NULL;
 135         return (CMD_RETURN_NORMAL);
 136 }