cmd-confirm-before.c

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