1 /* $OpenBSD: cmd-bind-key.c,v 1.22 2015/11/12 11:05:34 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_exec(struct cmd *, struct cmd_q *); 31 32 enum cmd_retval cmd_bind_key_mode_table(struct cmd *, struct cmd_q *, 33 key_code); 34 35 const struct cmd_entry cmd_bind_key_entry = { 36 "bind-key", "bind", 37 "cnrt:T:", 1, -1, 38 "[-cnr] [-t mode-table] [-T key-table] key command [arguments]", 39 0, 40 cmd_bind_key_exec 41 }; 42 43 enum cmd_retval 44 cmd_bind_key_exec(struct cmd *self, struct cmd_q *cmdq) 45 { 46 struct args *args = self->args; 47 char *cause; 48 struct cmd_list *cmdlist; 49 key_code key; 50 const char *tablename; 51 52 if (args_has(args, 't')) { 53 if (args->argc != 2 && args->argc != 3) { 54 cmdq_error(cmdq, "not enough arguments"); 55 return (CMD_RETURN_ERROR); 56 } 57 } else { 58 if (args->argc < 2) { 59 cmdq_error(cmdq, "not enough arguments"); 60 return (CMD_RETURN_ERROR); 61 } 62 } 63 64 key = key_string_lookup_string(args->argv[0]); 65 if (key == KEYC_NONE) { 66 cmdq_error(cmdq, "unknown key: %s", args->argv[0]); 67 return (CMD_RETURN_ERROR); 68 } 69 70 if (args_has(args, 't')) 71 return (cmd_bind_key_mode_table(self, cmdq, key)); 72 73 if (args_has(args, 'T')) 74 tablename = args_get(args, 'T'); 75 else if (args_has(args, 'n')) 76 tablename = "root"; 77 else 78 tablename = "prefix"; 79 80 cmdlist = cmd_list_parse(args->argc - 1, args->argv + 1, NULL, 0, 81 &cause); 82 if (cmdlist == NULL) { 83 cmdq_error(cmdq, "%s", cause); 84 free(cause); 85 return (CMD_RETURN_ERROR); 86 } 87 88 key_bindings_add(tablename, key, args_has(args, 'r'), cmdlist); 89 return (CMD_RETURN_NORMAL); 90 } 91 92 enum cmd_retval 93 cmd_bind_key_mode_table(struct cmd *self, struct cmd_q *cmdq, key_code key) 94 { 95 struct args *args = self->args; 96 const char *tablename; 97 const struct mode_key_table *mtab; 98 struct mode_key_binding *mbind, mtmp; 99 enum mode_key_cmd cmd; 100 const char *arg; 101 102 tablename = args_get(args, 't'); 103 if ((mtab = mode_key_findtable(tablename)) == NULL) { 104 cmdq_error(cmdq, "unknown key table: %s", tablename); 105 return (CMD_RETURN_ERROR); 106 } 107 108 cmd = mode_key_fromstring(mtab->cmdstr, args->argv[1]); 109 if (cmd == MODEKEY_NONE) { 110 cmdq_error(cmdq, "unknown command: %s", args->argv[1]); 111 return (CMD_RETURN_ERROR); 112 } 113 114 switch (cmd) { 115 case MODEKEYCOPY_APPENDSELECTION: 116 case MODEKEYCOPY_COPYSELECTION: 117 case MODEKEYCOPY_STARTNAMEDBUFFER: 118 if (args->argc == 2) 119 arg = NULL; 120 else { 121 arg = args->argv[2]; 122 if (strcmp(arg, "-x") != 0) { 123 cmdq_error(cmdq, "unknown argument"); 124 return (CMD_RETURN_ERROR); 125 } 126 } 127 break; 128 case MODEKEYCOPY_COPYPIPE: 129 if (args->argc != 3) { 130 cmdq_error(cmdq, "no argument given"); 131 return (CMD_RETURN_ERROR); 132 } 133 arg = args->argv[2]; 134 break; 135 default: 136 if (args->argc != 2) { 137 cmdq_error(cmdq, "no argument allowed"); 138 return (CMD_RETURN_ERROR); 139 } 140 arg = NULL; 141 break; 142 } 143 144 mtmp.key = key; 145 mtmp.mode = !!args_has(args, 'c'); 146 if ((mbind = RB_FIND(mode_key_tree, mtab->tree, &mtmp)) == NULL) { 147 mbind = xmalloc(sizeof *mbind); 148 mbind->key = mtmp.key; 149 mbind->mode = mtmp.mode; 150 RB_INSERT(mode_key_tree, mtab->tree, mbind); 151 } 152 mbind->cmd = cmd; 153 mbind->arg = arg != NULL ? xstrdup(arg) : NULL; 154 return (CMD_RETURN_NORMAL); 155 } 156