cmd-command-prompt.c

   1 /* $OpenBSD$ */
   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
  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
  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
 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
 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 }