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