cmd-confirm-before.c revision 1.42 
1/* $OpenBSD: cmd-confirm-before.c,v 1.42 2020/05/16 16:16:07 nicm Exp $ */
2
3/*
4 * Copyright (c) 2009 Tiago Cunha <me@tiagocunha.org>
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 <ctype.h>
22#include <stdlib.h>
23#include <string.h>
24
25#include "tmux.h"
26
27/*
28 * Asks for confirmation before executing a command.
29 */
30
31static enum cmd_retval	cmd_confirm_before_exec(struct cmd *,
32			    struct cmdq_item *);
33
34static int	cmd_confirm_before_callback(struct client *, void *,
35		    const char *, int);
36static void	cmd_confirm_before_free(void *);
37
38const struct cmd_entry cmd_confirm_before_entry = {
39	.name = "confirm-before",
40	.alias = "confirm",
41
42	.args = { "p:t:", 1, 1 },
43	.usage = "[-p prompt] " CMD_TARGET_CLIENT_USAGE " command",
44
45	.flags = CMD_CLIENT_TFLAG,
46	.exec = cmd_confirm_before_exec
47};
48
49struct cmd_confirm_before_data {
50	char	*cmd;
51};
52
53static enum cmd_retval
54cmd_confirm_before_exec(struct cmd *self, struct cmdq_item *item)
55{
56	struct args			*args = cmd_get_args(self);
57	struct cmd_confirm_before_data	*cdata;
58	struct client			*tc = cmdq_get_target_client(item);
59	struct cmd_find_state		*target = cmdq_get_target(item);
60	char				*cmd, *copy, *new_prompt, *ptr;
61	const char			*prompt;
62
63	if ((prompt = args_get(args, 'p')) != NULL)
64		xasprintf(&new_prompt, "%s ", prompt);
65	else {
66		ptr = copy = xstrdup(args->argv[0]);
67		cmd = strsep(&ptr, " \t");
68		xasprintf(&new_prompt, "Confirm '%s'? (y/n) ", cmd);
69		free(copy);
70	}
71
72	cdata = xmalloc(sizeof *cdata);
73	cdata->cmd = xstrdup(args->argv[0]);
74
75	status_prompt_set(tc, target, new_prompt, NULL,
76	    cmd_confirm_before_callback, cmd_confirm_before_free, cdata,
77	    PROMPT_SINGLE);
78
79	free(new_prompt);
80	return (CMD_RETURN_NORMAL);
81}
82
83static int
84cmd_confirm_before_callback(struct client *c, void *data, const char *s,
85    __unused int done)
86{
87	struct cmd_confirm_before_data	*cdata = data;
88	char				*error;
89	enum cmd_parse_status		 status;
90
91	if (c->flags & CLIENT_DEAD)
92		return (0);
93
94	if (s == NULL || *s == '\0')
95		return (0);
96	if (tolower((u_char)s[0]) != 'y' || s[1] != '\0')
97		return (0);
98
99	status = cmd_parse_and_append(cdata->cmd, NULL, c, NULL, &error);
100	if (status == CMD_PARSE_ERROR) {
101		cmdq_append(c, cmdq_get_error(error));
102		free(error);
103	}
104
105	return (0);
106}
107
108static void
109cmd_confirm_before_free(void *data)
110{
111	struct cmd_confirm_before_data	*cdata = data;
112
113	free(cdata->cmd);
114	free(cdata);
115}
116