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