1/* $OpenBSD: cmd-bind-key.c,v 1.46 2021/08/27 17:25:55 nicm Exp $ */
2
3/*
4 * Copyright (c) 2007 Nicholas Marriott <nicholas.marriott@gmail.com>
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.
28 */
29
30static enum args_parse_type	cmd_bind_key_args_parse(struct args *, u_int,
31				    char **);
32static enum cmd_retval		cmd_bind_key_exec(struct cmd *,
33				    struct cmdq_item *);
34
35const struct cmd_entry cmd_bind_key_entry = {
36	.name = "bind-key",
37	.alias = "bind",
38
39	.args = { "nrN:T:", 1, -1, cmd_bind_key_args_parse },
40	.usage = "[-nr] [-T key-table] [-N note] key "
41	         "[command [arguments]]",
42
43	.flags = CMD_AFTERHOOK,
44	.exec = cmd_bind_key_exec
45};
46
47static enum args_parse_type
48cmd_bind_key_args_parse(__unused struct args *args, __unused u_int idx,
49    __unused char **cause)
50{
51	return (ARGS_PARSE_COMMANDS_OR_STRING);
52}
53
54static enum cmd_retval
55cmd_bind_key_exec(struct cmd *self, struct cmdq_item *item)
56{
57	struct args		 *args = cmd_get_args(self);
58	key_code		  key;
59	const char		 *tablename, *note = args_get(args, 'N');
60	struct cmd_parse_result	 *pr;
61	int			  repeat;
62	struct args_value	 *value;
63	u_int			  count = args_count(args);
64
65	key = key_string_lookup_string(args_string(args, 0));
66	if (key == KEYC_NONE || key == KEYC_UNKNOWN) {
67		cmdq_error(item, "unknown key: %s", args_string(args, 0));
68		return (CMD_RETURN_ERROR);
69	}
70
71	if (args_has(args, 'T'))
72		tablename = args_get(args, 'T');
73	else if (args_has(args, 'n'))
74		tablename = "root";
75	else
76		tablename = "prefix";
77	repeat = args_has(args, 'r');
78
79	if (count == 1) {
80		key_bindings_add(tablename, key, note, repeat, NULL);
81		return (CMD_RETURN_NORMAL);
82	}
83
84	value = args_value(args, 1);
85	if (count == 2 && value->type == ARGS_COMMANDS) {
86		key_bindings_add(tablename, key, note, repeat, value->cmdlist);
87		value->cmdlist->references++;
88		return (CMD_RETURN_NORMAL);
89	}
90
91	if (count == 2)
92		pr = cmd_parse_from_string(args_string(args, 1), NULL);
93	else {
94		pr = cmd_parse_from_arguments(args_values(args) + 1, count - 1,
95		    NULL);
96	}
97	switch (pr->status) {
98	case CMD_PARSE_ERROR:
99		cmdq_error(item, "%s", pr->error);
100		free(pr->error);
101		return (CMD_RETURN_ERROR);
102	case CMD_PARSE_SUCCESS:
103		break;
104	}
105	key_bindings_add(tablename, key, note, repeat, pr->cmdlist);
106	return (CMD_RETURN_NORMAL);
107}
108