1 /* Id */
2
3 /*
4 * Copyright (c) 2008 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 <ctype.h>
22 #include <stdlib.h>
23 #include <string.h>
24 #include <time.h>
25
26 #include "tmux.h"
27
28 /*
29 * Prompt for command in client.
30 */
31
32 void cmd_command_prompt_key_binding(struct cmd *, int);
33 enum cmd_retval cmd_command_prompt_exec(struct cmd *, struct cmd_q *);
34
35 int cmd_command_prompt_callback(void *, const char *);
36 void cmd_command_prompt_free(void *);
37
38 const struct cmd_entry cmd_command_prompt_entry = {
39 "command-prompt", NULL,
40 "I:p:t:", 0, 1,
41 "[-I inputs] [-p prompts] " CMD_TARGET_CLIENT_USAGE " [template]",
42 0,
43 cmd_command_prompt_key_binding,
44 cmd_command_prompt_exec
45 };
46
47 struct cmd_command_prompt_cdata {
48 struct client *c;
49 char *inputs;
50 char *next_input;
51 char *next_prompt;
52 char *prompts;
53 char *template;
54 int idx;
55 };
56
57 void
cmd_command_prompt_key_binding(struct cmd * self,int key)58 cmd_command_prompt_key_binding(struct cmd *self, int key)
59 {
60 switch (key) {
61 case '$':
62 self->args = args_create(1, "rename-session '%%'");
63 args_set(self->args, 'I', "#S");
64 break;
65 case ',':
66 self->args = args_create(1, "rename-window '%%'");
67 args_set(self->args, 'I', "#W");
68 break;
69 case '.':
70 self->args = args_create(1, "move-window -t '%%'");
71 break;
72 case 'f':
73 self->args = args_create(1, "find-window '%%'");
74 break;
75 case '\'':
76 self->args = args_create(1, "select-window -t ':%%'");
77 args_set(self->args, 'p', "index");
78 break;
79 default:
80 self->args = args_create(0);
81 break;
82 }
83 }
84
85 enum cmd_retval
cmd_command_prompt_exec(struct cmd * self,struct cmd_q * cmdq)86 cmd_command_prompt_exec(struct cmd *self, struct cmd_q *cmdq)
87 {
88 struct args *args = self->args;
89 const char *inputs, *prompts;
90 struct cmd_command_prompt_cdata *cdata;
91 struct client *c;
92 char *prompt, *ptr, *input = NULL;
93 size_t n;
94
95 if ((c = cmd_find_client(cmdq, args_get(args, 't'), 0)) == NULL)
96 return (CMD_RETURN_ERROR);
97
98 if (c->prompt_string != NULL)
99 return (CMD_RETURN_NORMAL);
100
101 cdata = xmalloc(sizeof *cdata);
102 cdata->c = c;
103 cdata->idx = 1;
104 cdata->inputs = NULL;
105 cdata->next_input = NULL;
106 cdata->next_prompt = NULL;
107 cdata->prompts = NULL;
108 cdata->template = NULL;
109
110 if (args->argc != 0)
111 cdata->template = xstrdup(args->argv[0]);
112 else
113 cdata->template = xstrdup("%1");
114
115 if ((prompts = args_get(args, 'p')) != NULL)
116 cdata->prompts = xstrdup(prompts);
117 else if (args->argc != 0) {
118 n = strcspn(cdata->template, " ,");
119 xasprintf(&cdata->prompts, "(%.*s) ", (int) n, cdata->template);
120 } else
121 cdata->prompts = xstrdup(":");
122
123 /* Get first prompt. */
124 cdata->next_prompt = cdata->prompts;
125 ptr = strsep(&cdata->next_prompt, ",");
126 if (prompts == NULL)
127 prompt = xstrdup(ptr);
128 else
129 xasprintf(&prompt, "%s ", ptr);
130
131 /* Get initial prompt input. */
132 if ((inputs = args_get(args, 'I')) != NULL) {
133 cdata->inputs = xstrdup(inputs);
134 cdata->next_input = cdata->inputs;
135 input = strsep(&cdata->next_input, ",");
136 }
137
138 status_prompt_set(c, prompt, input, cmd_command_prompt_callback,
139 cmd_command_prompt_free, cdata, 0);
140 free(prompt);
141
142 return (CMD_RETURN_NORMAL);
143 }
144
145 int
cmd_command_prompt_callback(void * data,const char * s)146 cmd_command_prompt_callback(void *data, const char *s)
147 {
148 struct cmd_command_prompt_cdata *cdata = data;
149 struct client *c = cdata->c;
150 struct cmd_list *cmdlist;
151 char *cause, *new_template, *prompt, *ptr;
152 char *input = NULL;
153
154 if (s == NULL)
155 return (0);
156
157 new_template = cmd_template_replace(cdata->template, s, cdata->idx);
158 free(cdata->template);
159 cdata->template = new_template;
160
161 /*
162 * Check if there are more prompts; if so, get its respective input
163 * and update the prompt data.
164 */
165 if ((ptr = strsep(&cdata->next_prompt, ",")) != NULL) {
166 xasprintf(&prompt, "%s ", ptr);
167 input = strsep(&cdata->next_input, ",");
168 status_prompt_update(c, prompt, input);
169
170 free(prompt);
171 cdata->idx++;
172 return (1);
173 }
174
175 if (cmd_string_parse(new_template, &cmdlist, NULL, 0, &cause) != 0) {
176 if (cause != NULL) {
177 *cause = toupper((u_char) *cause);
178 status_message_set(c, "%s", cause);
179 free(cause);
180 }
181 return (0);
182 }
183
184 cmdq_run(c->cmdq, cmdlist);
185 cmd_list_free(cmdlist);
186
187 if (c->prompt_callbackfn != (void *) &cmd_command_prompt_callback)
188 return (1);
189 return (0);
190 }
191
192 void
cmd_command_prompt_free(void * data)193 cmd_command_prompt_free(void *data)
194 {
195 struct cmd_command_prompt_cdata *cdata = data;
196
197 free(cdata->inputs);
198 free(cdata->prompts);
199 free(cdata->template);
200 free(cdata);
201 }
202