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