cmd-choose-tree.c

   1 /* $OpenBSD$ */
   2
   3 /*
   4  * Copyright (c) 2012 Thomas Adam <thomas@xteddy.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 <sys/types.h>
  20
  21 #include <ctype.h>
  22 #include <stdlib.h>
  23
  24 #include <string.h>
  25
  26 #include "tmux.h"
  27
  28 #define CMD_CHOOSE_TREE_WINDOW_ACTION "select-window -t '%%'"
  29 #define CMD_CHOOSE_TREE_SESSION_ACTION "switch-client -t '%%'"
  30
  31 /*
  32  * Enter choice mode to choose a session and/or window.
  33  */
  34
  35 enum cmd_retval cmd_choose_tree_exec(struct cmd *, struct cmd_q *);
  36
  37 const struct cmd_entry cmd_choose_tree_entry = {
  38         "choose-tree", NULL,
  39         "S:W:swub:c:t:", 0, 1,
  40         "[-suw] [-b session-template] [-c window template] [-S format] " \
  41         "[-W format] " CMD_TARGET_WINDOW_USAGE,
  42         0,
  43         NULL,
  44         NULL,
  45         cmd_choose_tree_exec
  46 };
  47
  48 const struct cmd_entry cmd_choose_session_entry = {
  49         "choose-session", NULL,
  50         "F:t:", 0, 1,
  51         CMD_TARGET_WINDOW_USAGE " [-F format] [template]",
  52         0,
  53         NULL,
  54         NULL,
  55         cmd_choose_tree_exec
  56 };
  57
  58 const struct cmd_entry cmd_choose_window_entry = {
  59         "choose-window", NULL,
  60         "F:t:", 0, 1,
  61         CMD_TARGET_WINDOW_USAGE "[-F format] [template]",
  62         0,
  63         NULL,
  64         NULL,
  65         cmd_choose_tree_exec
  66 };
  67
  68 enum cmd_retval
  69 cmd_choose_tree_exec(struct cmd *self, struct cmd_q *cmdq)
  70 {
  71         struct args                     *args = self->args;
  72         struct winlink                  *wl, *wm;
  73         struct session                  *s, *s2;
  74         struct client                   *c;
  75         struct window_choose_data       *wcd = NULL;
  76         const char                      *ses_template, *win_template;
  77         char                            *final_win_action, *cur_win_template;
  78         char                            *final_win_template_middle;
  79         char                            *final_win_template_last;
  80         const char                      *ses_action, *win_action;
  81         u_int                            cur_win, idx_ses, win_ses, win_max;
  82         u_int                            wflag, sflag;
  83
  84         ses_template = win_template = NULL;
  85         ses_action = win_action = NULL;
  86
  87         if ((c = cmd_current_client(cmdq)) == NULL) {
  88                 cmdq_error(cmdq, "no client available");
  89                 return (CMD_RETURN_ERROR);
  90         }
  91
  92         if ((s = c->session) == NULL)
  93                 return (CMD_RETURN_ERROR);
  94
  95         if ((wl = cmd_find_window(cmdq, args_get(args, 't'), NULL)) == NULL)
  96                 return (CMD_RETURN_ERROR);
  97
  98         if (window_pane_set_mode(wl->window->active, &window_choose_mode) != 0)
  99                 return (CMD_RETURN_NORMAL);
 100
 101         /* Sort out which command this is. */
 102         wflag = sflag = 0;
 103         if (self->entry == &cmd_choose_session_entry) {
 104                 sflag = 1;
 105                 if ((ses_template = args_get(args, 'F')) == NULL)
 106                         ses_template = CHOOSE_TREE_SESSION_TEMPLATE;
 107
 108                 if (args->argc != 0)
 109                         ses_action = args->argv[0];
 110                 else
 111                         ses_action = CMD_CHOOSE_TREE_SESSION_ACTION;
 112         } else if (self->entry == &cmd_choose_window_entry) {
 113                 wflag = 1;
 114                 if ((win_template = args_get(args, 'F')) == NULL)
 115                         win_template = CHOOSE_TREE_WINDOW_TEMPLATE;
 116
 117                 if (args->argc != 0)
 118                         win_action = args->argv[0];
 119                 else
 120                         win_action = CMD_CHOOSE_TREE_WINDOW_ACTION;
 121         } else {
 122                 wflag = args_has(args, 'w');
 123                 sflag = args_has(args, 's');
 124
 125                 if ((ses_action = args_get(args, 'b')) == NULL)
 126                         ses_action = CMD_CHOOSE_TREE_SESSION_ACTION;
 127
 128                 if ((win_action = args_get(args, 'c')) == NULL)
 129                         win_action = CMD_CHOOSE_TREE_WINDOW_ACTION;
 130
 131                 if ((ses_template = args_get(args, 'S')) == NULL)
 132                         ses_template = CHOOSE_TREE_SESSION_TEMPLATE;
 133
 134                 if ((win_template = args_get(args, 'W')) == NULL)
 135                         win_template = CHOOSE_TREE_WINDOW_TEMPLATE;
 136         }
 137
 138         /*
 139          * If not asking for windows and sessions, assume no "-ws" given and
 140          * hence display the entire tree outright.
 141          */
 142         if (!wflag && !sflag)
 143                 wflag = sflag = 1;
 144
 145         /*
 146          * If we're drawing in tree mode, including sessions, then pad the
 147          * window template, otherwise just render the windows as a flat list
 148          * without any padding.
 149          */
 150         if (wflag && sflag) {
 151                 xasprintf(&final_win_template_middle,
 152                     " \001tq\001> %s", win_template);
 153                 xasprintf(&final_win_template_last,
 154                     " \001mq\001> %s", win_template);
 155         } else if (wflag) {
 156                 final_win_template_middle = xstrdup(win_template);
 157                 final_win_template_last = xstrdup(win_template);
 158         } else
 159                 final_win_template_middle = final_win_template_last = NULL;
 160
 161         idx_ses = cur_win = -1;
 162         RB_FOREACH(s2, sessions, &sessions) {
 163                 idx_ses++;
 164
 165                 /*
 166                  * If we're just choosing windows, jump straight there. Note
 167                  * that this implies the current session, so only choose
 168                  * windows when the session matches this one.
 169                  */
 170                 if (wflag && !sflag) {
 171                         if (s != s2)
 172                                 continue;
 173                         goto windows_only;
 174                 }
 175
 176                 wcd = window_choose_add_session(wl->window->active,
 177                     c, s2, ses_template, ses_action, idx_ses);
 178
 179                 /* If we're just choosing sessions, skip choosing windows. */
 180                 if (sflag && !wflag) {
 181                         if (s == s2)
 182                                 cur_win = idx_ses;
 183                         continue;
 184                 }
 185 windows_only:
 186                 win_ses = win_max = -1;
 187                 RB_FOREACH(wm, winlinks, &s2->windows)
 188                         win_max++;
 189                 RB_FOREACH(wm, winlinks, &s2->windows) {
 190                         win_ses++;
 191                         if (sflag && wflag)
 192                                 idx_ses++;
 193
 194                         if (wm == s2->curw && s == s2) {
 195                                 if (wflag && !sflag) {
 196                                         /*
 197                                          * Then we're only counting windows.
 198                                          * So remember which is the current
 199                                          * window in the list.
 200                                          */
 201                                         cur_win = win_ses;
 202                                 } else
 203                                         cur_win = idx_ses;
 204                         }
 205
 206                         xasprintf(&final_win_action, "%s %s %s",
 207                             wcd != NULL ? wcd->command : "",
 208                             wcd != NULL ? ";" : "", win_action);
 209
 210                         if (win_ses != win_max)
 211                                 cur_win_template = final_win_template_middle;
 212                         else
 213                                 cur_win_template = final_win_template_last;
 214
 215                         window_choose_add_window(wl->window->active,
 216                             c, s2, wm, cur_win_template,
 217                             final_win_action,
 218                             (wflag && !sflag) ? win_ses : idx_ses);
 219
 220                         free(final_win_action);
 221                 }
 222
 223                 /*
 224                  * If we're just drawing windows, don't consider moving on to
 225                  * other sessions as we only list windows in this session.
 226                  */
 227                 if (wflag && !sflag)
 228                         break;
 229         }
 230         free(final_win_template_middle);
 231         free(final_win_template_last);
 232
 233         window_choose_ready(wl->window->active, cur_win, NULL);
 234
 235         if (args_has(args, 'u'))
 236                 window_choose_expand_all(wl->window->active);
 237
 238         return (CMD_RETURN_NORMAL);
 239 }