1 /* Id */
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 <ctype.h>
20 #include <stdlib.h>
21 #include <string.h>
22
23 #include "tmux.h"
24
25 /*
26 * Asks for confirmation before executing a command.
27 */
28
29 void cmd_confirm_before_key_binding(struct cmd *, int);
30 enum cmd_retval cmd_confirm_before_exec(struct cmd *, struct cmd_q *);
31
32 int cmd_confirm_before_callback(void *, const char *);
33 void cmd_confirm_before_free(void *);
34
35 const struct cmd_entry cmd_confirm_before_entry = {
36 "confirm-before", "confirm",
37 "p:t:", 1, 1,
38 "[-p prompt] " CMD_TARGET_CLIENT_USAGE " command",
39 0,
40 cmd_confirm_before_key_binding,
41 cmd_confirm_before_exec
42 };
43
44 struct cmd_confirm_before_data {
45 char *cmd;
46 struct client *client;
47 };
48
49 void
cmd_confirm_before_key_binding(struct cmd * self,int key)50 cmd_confirm_before_key_binding(struct cmd *self, int key)
51 {
52 switch (key) {
53 case '&':
54 self->args = args_create(1, "kill-window");
55 args_set(self->args, 'p', "kill-window #W? (y/n)");
56 break;
57 case 'x':
58 self->args = args_create(1, "kill-pane");
59 args_set(self->args, 'p', "kill-pane #P? (y/n)");
60 break;
61 default:
62 self->args = args_create(0);
63 break;
64 }
65 }
66
67 enum cmd_retval
cmd_confirm_before_exec(struct cmd * self,struct cmd_q * cmdq)68 cmd_confirm_before_exec(struct cmd *self, struct cmd_q *cmdq)
69 {
70 struct args *args = self->args;
71 struct cmd_confirm_before_data *cdata;
72 struct client *c;
73 char *cmd, *copy, *new_prompt, *ptr;
74 const char *prompt;
75
76 if ((c = cmd_find_client(cmdq, args_get(args, 't'), 0)) == NULL)
77 return (CMD_RETURN_ERROR);
78
79 if ((prompt = args_get(args, 'p')) != NULL)
80 xasprintf(&new_prompt, "%s ", prompt);
81 else {
82 ptr = copy = xstrdup(args->argv[0]);
83 cmd = strsep(&ptr, " \t");
84 xasprintf(&new_prompt, "Confirm '%s'? (y/n) ", cmd);
85 free(copy);
86 }
87
88 cdata = xmalloc(sizeof *cdata);
89 cdata->cmd = xstrdup(args->argv[0]);
90
91 cdata->client = c;
92 cdata->client->references++;
93
94 status_prompt_set(c, new_prompt, NULL,
95 cmd_confirm_before_callback, cmd_confirm_before_free, cdata,
96 PROMPT_SINGLE);
97
98 free(new_prompt);
99 return (CMD_RETURN_NORMAL);
100 }
101
102 int
cmd_confirm_before_callback(void * data,const char * s)103 cmd_confirm_before_callback(void *data, const char *s)
104 {
105 struct cmd_confirm_before_data *cdata = data;
106 struct client *c = cdata->client;
107 struct cmd_list *cmdlist;
108 char *cause;
109
110 if (c->flags & CLIENT_DEAD)
111 return (0);
112
113 if (s == NULL || *s == '\0')
114 return (0);
115 if (tolower((u_char) s[0]) != 'y' || s[1] != '\0')
116 return (0);
117
118 if (cmd_string_parse(cdata->cmd, &cmdlist, NULL, 0, &cause) != 0) {
119 if (cause != NULL) {
120 cmdq_error(c->cmdq, "%s", cause);
121 free(cause);
122 }
123 return (0);
124 }
125
126 cmdq_run(c->cmdq, cmdlist);
127 cmd_list_free(cmdlist);
128
129 return (0);
130 }
131
132 void
cmd_confirm_before_free(void * data)133 cmd_confirm_before_free(void *data)
134 {
135 struct cmd_confirm_before_data *cdata = data;
136 struct client *c = cdata->client;
137
138 c->references--;
139
140 free(cdata->cmd);
141 free(cdata);
142 }
143