1 /* $OpenBSD: cmd-bind-key.c,v 1.16 2013/03/24 09:54:10 nicm Exp $ */ 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 } 137