1*c224fd4bSnicm /* $OpenBSD: window-tree.c,v 1.7 2017/06/30 22:36:11 nicm Exp $ */ 2a42faf7dSnicm 3a42faf7dSnicm /* 4a42faf7dSnicm * Copyright (c) 2017 Nicholas Marriott <nicholas.marriott@gmail.com> 5a42faf7dSnicm * 6a42faf7dSnicm * Permission to use, copy, modify, and distribute this software for any 7a42faf7dSnicm * purpose with or without fee is hereby granted, provided that the above 8a42faf7dSnicm * copyright notice and this permission notice appear in all copies. 9a42faf7dSnicm * 10a42faf7dSnicm * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES 11a42faf7dSnicm * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF 12a42faf7dSnicm * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR 13a42faf7dSnicm * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 14a42faf7dSnicm * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER 15a42faf7dSnicm * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING 16a42faf7dSnicm * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 17a42faf7dSnicm */ 18a42faf7dSnicm 19a42faf7dSnicm #include <sys/types.h> 20a42faf7dSnicm 21a42faf7dSnicm #include <stdlib.h> 22a42faf7dSnicm #include <string.h> 23a42faf7dSnicm 24a42faf7dSnicm #include "tmux.h" 25a42faf7dSnicm 26a42faf7dSnicm static struct screen *window_tree_init(struct window_pane *, 27a42faf7dSnicm struct cmd_find_state *, struct args *); 28a42faf7dSnicm static void window_tree_free(struct window_pane *); 29a42faf7dSnicm static void window_tree_resize(struct window_pane *, u_int, u_int); 30a42faf7dSnicm static void window_tree_key(struct window_pane *, 31a42faf7dSnicm struct client *, struct session *, key_code, 32a42faf7dSnicm struct mouse_event *); 33a42faf7dSnicm 34a42faf7dSnicm #define WINDOW_TREE_DEFAULT_COMMAND "switch-client -t '%%'" 35a42faf7dSnicm 36a42faf7dSnicm const struct window_mode window_tree_mode = { 37a42faf7dSnicm .name = "tree-mode", 38a42faf7dSnicm 39a42faf7dSnicm .init = window_tree_init, 40a42faf7dSnicm .free = window_tree_free, 41a42faf7dSnicm .resize = window_tree_resize, 42a42faf7dSnicm .key = window_tree_key, 43a42faf7dSnicm }; 44a42faf7dSnicm 45a42faf7dSnicm enum window_tree_sort_type { 46a42faf7dSnicm WINDOW_TREE_BY_INDEX, 47a42faf7dSnicm WINDOW_TREE_BY_NAME, 48a42faf7dSnicm WINDOW_TREE_BY_TIME, 49a42faf7dSnicm }; 50a42faf7dSnicm static const char *window_tree_sort_list[] = { 51a42faf7dSnicm "index", 52a42faf7dSnicm "name", 53a42faf7dSnicm "time" 54a42faf7dSnicm }; 55a42faf7dSnicm 56a42faf7dSnicm enum window_tree_type { 57a42faf7dSnicm WINDOW_TREE_NONE, 58a42faf7dSnicm WINDOW_TREE_SESSION, 59a42faf7dSnicm WINDOW_TREE_WINDOW, 60a42faf7dSnicm WINDOW_TREE_PANE, 61a42faf7dSnicm }; 62a42faf7dSnicm 63a42faf7dSnicm struct window_tree_itemdata { 64a42faf7dSnicm enum window_tree_type type; 65a42faf7dSnicm int session; 66a42faf7dSnicm int winlink; 67a42faf7dSnicm int pane; 68a42faf7dSnicm }; 69a42faf7dSnicm 70a42faf7dSnicm struct window_tree_modedata { 71a42faf7dSnicm struct window_pane *wp; 72a42faf7dSnicm int dead; 73a42faf7dSnicm int references; 74a42faf7dSnicm 75a42faf7dSnicm struct mode_tree_data *data; 76a42faf7dSnicm char *command; 77a42faf7dSnicm 78a42faf7dSnicm struct window_tree_itemdata **item_list; 79a42faf7dSnicm u_int item_size; 80a42faf7dSnicm 81a42faf7dSnicm struct client *client; 82a42faf7dSnicm const char *entered; 83a42faf7dSnicm 84a42faf7dSnicm struct cmd_find_state fs; 85a42faf7dSnicm enum window_tree_type type; 86a42faf7dSnicm }; 87a42faf7dSnicm 88a42faf7dSnicm static void 89a42faf7dSnicm window_tree_pull_item(struct window_tree_itemdata *item, struct session **sp, 90a42faf7dSnicm struct winlink **wlp, struct window_pane **wp) 91a42faf7dSnicm { 92a42faf7dSnicm *wp = NULL; 93a42faf7dSnicm *wlp = NULL; 94a42faf7dSnicm *sp = session_find_by_id(item->session); 95a42faf7dSnicm if (*sp == NULL) 96a42faf7dSnicm return; 97a42faf7dSnicm if (item->type == WINDOW_TREE_SESSION) { 98a42faf7dSnicm *wlp = (*sp)->curw; 99a42faf7dSnicm *wp = (*wlp)->window->active; 100a42faf7dSnicm return; 101a42faf7dSnicm } 102a42faf7dSnicm 103a42faf7dSnicm *wlp = winlink_find_by_index(&(*sp)->windows, item->winlink); 104a42faf7dSnicm if (*wlp == NULL) { 105a42faf7dSnicm *sp = NULL; 106a42faf7dSnicm return; 107a42faf7dSnicm } 108a42faf7dSnicm if (item->type == WINDOW_TREE_WINDOW) { 109a42faf7dSnicm *wp = (*wlp)->window->active; 110a42faf7dSnicm return; 111a42faf7dSnicm } 112a42faf7dSnicm 113a42faf7dSnicm *wp = window_pane_find_by_id(item->pane); 114a42faf7dSnicm if (!window_has_pane((*wlp)->window, *wp)) 115a42faf7dSnicm *wp = NULL; 116a42faf7dSnicm if (*wp == NULL) { 117a42faf7dSnicm *sp = NULL; 118a42faf7dSnicm *wlp = NULL; 119a42faf7dSnicm return; 120a42faf7dSnicm } 121a42faf7dSnicm } 122a42faf7dSnicm 123a42faf7dSnicm static struct window_tree_itemdata * 124a42faf7dSnicm window_tree_add_item(struct window_tree_modedata *data) 125a42faf7dSnicm { 126a42faf7dSnicm struct window_tree_itemdata *item; 127a42faf7dSnicm 128a42faf7dSnicm data->item_list = xreallocarray(data->item_list, data->item_size + 1, 129a42faf7dSnicm sizeof *data->item_list); 130a42faf7dSnicm item = data->item_list[data->item_size++] = xcalloc(1, sizeof *item); 131a42faf7dSnicm return (item); 132a42faf7dSnicm } 133a42faf7dSnicm 134a42faf7dSnicm static void 135a42faf7dSnicm window_tree_free_item(struct window_tree_itemdata *item) 136a42faf7dSnicm { 137a42faf7dSnicm free(item); 138a42faf7dSnicm } 139a42faf7dSnicm 140a42faf7dSnicm static int 141a42faf7dSnicm window_tree_cmp_session_name(const void *a0, const void *b0) 142a42faf7dSnicm { 143a42faf7dSnicm const struct session *const *a = a0; 144a42faf7dSnicm const struct session *const *b = b0; 145a42faf7dSnicm 146a42faf7dSnicm return (strcmp((*a)->name, (*b)->name)); 147a42faf7dSnicm } 148a42faf7dSnicm 149a42faf7dSnicm static int 150a42faf7dSnicm window_tree_cmp_session_time(const void *a0, const void *b0) 151a42faf7dSnicm { 152a42faf7dSnicm const struct session *const *a = a0; 153a42faf7dSnicm const struct session *const *b = b0; 154a42faf7dSnicm 155a42faf7dSnicm if (timercmp(&(*a)->activity_time, &(*b)->activity_time, >)) 156a42faf7dSnicm return (-1); 157a42faf7dSnicm if (timercmp(&(*a)->activity_time, &(*b)->activity_time, <)) 158a42faf7dSnicm return (1); 159a42faf7dSnicm return (strcmp((*a)->name, (*b)->name)); 160a42faf7dSnicm } 161a42faf7dSnicm 162a42faf7dSnicm static int 163a42faf7dSnicm window_tree_cmp_window_name(const void *a0, const void *b0) 164a42faf7dSnicm { 165a42faf7dSnicm const struct winlink *const *a = a0; 166a42faf7dSnicm const struct winlink *const *b = b0; 167a42faf7dSnicm 168a42faf7dSnicm return (strcmp((*a)->window->name, (*b)->window->name)); 169a42faf7dSnicm } 170a42faf7dSnicm 171a42faf7dSnicm static int 172a42faf7dSnicm window_tree_cmp_window_time(const void *a0, const void *b0) 173a42faf7dSnicm { 174a42faf7dSnicm const struct winlink *const *a = a0; 175a42faf7dSnicm const struct winlink *const *b = b0; 176a42faf7dSnicm 177a42faf7dSnicm if (timercmp(&(*a)->window->activity_time, 178a42faf7dSnicm &(*b)->window->activity_time, >)) 179a42faf7dSnicm return (-1); 180a42faf7dSnicm if (timercmp(&(*a)->window->activity_time, 181a42faf7dSnicm &(*b)->window->activity_time, <)) 182a42faf7dSnicm return (1); 183a42faf7dSnicm return (strcmp((*a)->window->name, (*b)->window->name)); 184a42faf7dSnicm } 185a42faf7dSnicm 186a42faf7dSnicm static int 187a42faf7dSnicm window_tree_cmp_pane_time(const void *a0, const void *b0) 188a42faf7dSnicm { 189a42faf7dSnicm const struct window_pane *const *a = a0; 190a42faf7dSnicm const struct window_pane *const *b = b0; 191a42faf7dSnicm 192a42faf7dSnicm if ((*a)->active_point < (*b)->active_point) 193a42faf7dSnicm return (-1); 194a42faf7dSnicm if ((*a)->active_point > (*b)->active_point) 195a42faf7dSnicm return (1); 196a42faf7dSnicm return (0); 197a42faf7dSnicm } 198a42faf7dSnicm 199a42faf7dSnicm static void 200a42faf7dSnicm window_tree_build_pane(struct session *s, struct winlink *wl, 201a42faf7dSnicm struct window_pane *wp, void *modedata, struct mode_tree_item *parent) 202a42faf7dSnicm { 203a42faf7dSnicm struct window_tree_modedata *data = modedata; 204a42faf7dSnicm struct window_tree_itemdata *item; 205a42faf7dSnicm char *name, *text; 206a42faf7dSnicm u_int idx; 207a42faf7dSnicm 208a42faf7dSnicm window_pane_index(wp, &idx); 209a42faf7dSnicm 210a42faf7dSnicm item = window_tree_add_item(data); 211a42faf7dSnicm item->type = WINDOW_TREE_PANE; 212a42faf7dSnicm item->session = s->id; 213a42faf7dSnicm item->winlink = wl->idx; 214a42faf7dSnicm item->pane = wp->id; 215a42faf7dSnicm 216a42faf7dSnicm text = format_single(NULL, 217a42faf7dSnicm "#{pane_current_command} \"#{pane_title}\"", 218a42faf7dSnicm NULL, s, wl, wp); 219a42faf7dSnicm xasprintf(&name, "%u", idx); 220a42faf7dSnicm 221a42faf7dSnicm mode_tree_add(data->data, parent, item, (uint64_t)wp, name, text, -1); 222a42faf7dSnicm free(text); 223a42faf7dSnicm free(name); 224a42faf7dSnicm } 225a42faf7dSnicm 226a42faf7dSnicm static int 227a42faf7dSnicm window_tree_build_window(struct session *s, struct winlink *wl, void* modedata, 228024c311aSnicm u_int sort_type, struct mode_tree_item *parent, const char *filter) 229a42faf7dSnicm { 230a42faf7dSnicm struct window_tree_modedata *data = modedata; 231a42faf7dSnicm struct window_tree_itemdata *item; 232a42faf7dSnicm struct mode_tree_item *mti; 233a42faf7dSnicm char *name, *text, *cp; 234a42faf7dSnicm struct window_pane *wp, **l; 235a42faf7dSnicm u_int n, i; 236a42faf7dSnicm int expanded; 237a42faf7dSnicm 238a42faf7dSnicm item = window_tree_add_item(data); 239a42faf7dSnicm item->type = WINDOW_TREE_WINDOW; 240a42faf7dSnicm item->session = s->id; 241a42faf7dSnicm item->winlink = wl->idx; 242a42faf7dSnicm item->pane = -1; 243a42faf7dSnicm 244a42faf7dSnicm text = format_single(NULL, 245a42faf7dSnicm "#{window_name}#{window_flags} (#{window_panes} panes)", 246a42faf7dSnicm NULL, s, wl, NULL); 247a42faf7dSnicm xasprintf(&name, "%u", wl->idx); 248a42faf7dSnicm 249a42faf7dSnicm if (data->type == WINDOW_TREE_SESSION || 250a42faf7dSnicm data->type == WINDOW_TREE_WINDOW) 251a42faf7dSnicm expanded = 0; 252a42faf7dSnicm else 253a42faf7dSnicm expanded = 1; 254a42faf7dSnicm mti = mode_tree_add(data->data, parent, item, (uint64_t)wl, name, text, 255a42faf7dSnicm expanded); 256a42faf7dSnicm free(text); 257a42faf7dSnicm free(name); 258a42faf7dSnicm 259a42faf7dSnicm l = NULL; 260a42faf7dSnicm n = 0; 261a42faf7dSnicm TAILQ_FOREACH(wp, &wl->window->panes, entry) { 262024c311aSnicm if (filter != NULL) { 263024c311aSnicm cp = format_single(NULL, filter, NULL, s, wl, wp); 264a42faf7dSnicm if (!format_true(cp)) { 265a42faf7dSnicm free(cp); 266a42faf7dSnicm continue; 267a42faf7dSnicm } 268a42faf7dSnicm free(cp); 269a42faf7dSnicm } 270a42faf7dSnicm l = xreallocarray(l, n + 1, sizeof *l); 271a42faf7dSnicm l[n++] = wp; 272a42faf7dSnicm } 273a42faf7dSnicm if (n == 0) { 274a42faf7dSnicm window_tree_free_item(item); 275a42faf7dSnicm data->item_size--; 276a42faf7dSnicm mode_tree_remove(data->data, mti); 277a42faf7dSnicm return (0); 278a42faf7dSnicm } 279a42faf7dSnicm 280a42faf7dSnicm switch (sort_type) { 281a42faf7dSnicm case WINDOW_TREE_BY_INDEX: 282a42faf7dSnicm break; 283a42faf7dSnicm case WINDOW_TREE_BY_NAME: 284a42faf7dSnicm /* Panes don't have names, so leave in number order. */ 285a42faf7dSnicm break; 286a42faf7dSnicm case WINDOW_TREE_BY_TIME: 287a42faf7dSnicm qsort(l, n, sizeof *l, window_tree_cmp_pane_time); 288a42faf7dSnicm break; 289a42faf7dSnicm } 290a42faf7dSnicm 291a42faf7dSnicm for (i = 0; i < n; i++) 292a42faf7dSnicm window_tree_build_pane(s, wl, l[i], modedata, mti); 293a42faf7dSnicm free(l); 294a42faf7dSnicm return (1); 295a42faf7dSnicm } 296a42faf7dSnicm 297a42faf7dSnicm static void 298a42faf7dSnicm window_tree_build_session(struct session *s, void* modedata, 299024c311aSnicm u_int sort_type, const char *filter) 300a42faf7dSnicm { 301a42faf7dSnicm struct window_tree_modedata *data = modedata; 302a42faf7dSnicm struct window_tree_itemdata *item; 303a42faf7dSnicm struct mode_tree_item *mti; 304a42faf7dSnicm char *text; 305a42faf7dSnicm struct winlink *wl, **l; 306a42faf7dSnicm u_int n, i, empty; 307a42faf7dSnicm int expanded; 308a42faf7dSnicm 309a42faf7dSnicm item = window_tree_add_item(data); 310a42faf7dSnicm item->type = WINDOW_TREE_SESSION; 311a42faf7dSnicm item->session = s->id; 312a42faf7dSnicm item->winlink = -1; 313a42faf7dSnicm item->pane = -1; 314a42faf7dSnicm 315a42faf7dSnicm text = format_single(NULL, 316a42faf7dSnicm "#{session_windows} windows" 317a42faf7dSnicm "#{?session_grouped, (group ,}" 318a42faf7dSnicm "#{session_group}#{?session_grouped,),}" 319a42faf7dSnicm "#{?session_attached, (attached),}", 320a42faf7dSnicm NULL, s, NULL, NULL); 321a42faf7dSnicm 322a42faf7dSnicm if (data->type == WINDOW_TREE_SESSION) 323a42faf7dSnicm expanded = 0; 324a42faf7dSnicm else 325a42faf7dSnicm expanded = 1; 326a42faf7dSnicm mti = mode_tree_add(data->data, NULL, item, (uint64_t)s, s->name, text, 327a42faf7dSnicm expanded); 328a42faf7dSnicm free(text); 329a42faf7dSnicm 330a42faf7dSnicm l = NULL; 331a42faf7dSnicm n = 0; 332a42faf7dSnicm RB_FOREACH(wl, winlinks, &s->windows) { 333a42faf7dSnicm l = xreallocarray(l, n + 1, sizeof *l); 334a42faf7dSnicm l[n++] = wl; 335a42faf7dSnicm } 336a42faf7dSnicm switch (sort_type) { 337a42faf7dSnicm case WINDOW_TREE_BY_INDEX: 338a42faf7dSnicm break; 339a42faf7dSnicm case WINDOW_TREE_BY_NAME: 340a42faf7dSnicm qsort(l, n, sizeof *l, window_tree_cmp_window_name); 341a42faf7dSnicm break; 342a42faf7dSnicm case WINDOW_TREE_BY_TIME: 343a42faf7dSnicm qsort(l, n, sizeof *l, window_tree_cmp_window_time); 344a42faf7dSnicm break; 345a42faf7dSnicm } 346a42faf7dSnicm 347a42faf7dSnicm empty = 0; 348a42faf7dSnicm for (i = 0; i < n; i++) { 349a42faf7dSnicm if (!window_tree_build_window(s, l[i], modedata, sort_type, mti, 350024c311aSnicm filter)) 351a42faf7dSnicm empty++; 352a42faf7dSnicm } 353a42faf7dSnicm if (empty == n) { 354a42faf7dSnicm window_tree_free_item(item); 355a42faf7dSnicm data->item_size--; 356a42faf7dSnicm mode_tree_remove(data->data, mti); 357a42faf7dSnicm } 358a42faf7dSnicm free(l); 359a42faf7dSnicm } 360a42faf7dSnicm 361a42faf7dSnicm static void 362024c311aSnicm window_tree_build(void *modedata, u_int sort_type, uint64_t *tag, 363024c311aSnicm const char *filter) 364a42faf7dSnicm { 365a42faf7dSnicm struct window_tree_modedata *data = modedata; 366a42faf7dSnicm struct session *s, **l; 367a42faf7dSnicm u_int n, i; 368a42faf7dSnicm 369a42faf7dSnicm for (i = 0; i < data->item_size; i++) 370a42faf7dSnicm window_tree_free_item(data->item_list[i]); 371a42faf7dSnicm free(data->item_list); 372a42faf7dSnicm data->item_list = NULL; 373a42faf7dSnicm data->item_size = 0; 374a42faf7dSnicm 375a42faf7dSnicm l = NULL; 376a42faf7dSnicm n = 0; 377a42faf7dSnicm RB_FOREACH(s, sessions, &sessions) { 378a42faf7dSnicm l = xreallocarray(l, n + 1, sizeof *l); 379a42faf7dSnicm l[n++] = s; 380a42faf7dSnicm } 381a42faf7dSnicm switch (sort_type) { 382a42faf7dSnicm case WINDOW_TREE_BY_INDEX: 383a42faf7dSnicm break; 384a42faf7dSnicm case WINDOW_TREE_BY_NAME: 385a42faf7dSnicm qsort(l, n, sizeof *l, window_tree_cmp_session_name); 386a42faf7dSnicm break; 387a42faf7dSnicm case WINDOW_TREE_BY_TIME: 388a42faf7dSnicm qsort(l, n, sizeof *l, window_tree_cmp_session_time); 389a42faf7dSnicm break; 390a42faf7dSnicm } 391a42faf7dSnicm 392a42faf7dSnicm for (i = 0; i < n; i++) 393024c311aSnicm window_tree_build_session(l[i], modedata, sort_type, filter); 394a42faf7dSnicm free(l); 395a42faf7dSnicm 396a42faf7dSnicm switch (data->type) { 397a42faf7dSnicm case WINDOW_TREE_NONE: 398a42faf7dSnicm break; 399a42faf7dSnicm case WINDOW_TREE_SESSION: 400a42faf7dSnicm *tag = (uint64_t)data->fs.s; 401a42faf7dSnicm break; 402a42faf7dSnicm case WINDOW_TREE_WINDOW: 403a42faf7dSnicm *tag = (uint64_t)data->fs.wl; 404a42faf7dSnicm break; 405a42faf7dSnicm case WINDOW_TREE_PANE: 406a42faf7dSnicm *tag = (uint64_t)data->fs.wp; 407a42faf7dSnicm break; 408a42faf7dSnicm } 409a42faf7dSnicm } 410a42faf7dSnicm 411367b65a4Snicm static void 412367b65a4Snicm window_tree_draw_session(struct session *s, struct screen_write_ctx *ctx, 413367b65a4Snicm u_int sx, u_int sy) 414367b65a4Snicm { 415367b65a4Snicm struct options *oo = s->options; 416367b65a4Snicm struct winlink *wl; 417367b65a4Snicm struct window *w; 418367b65a4Snicm u_int i, n, each, width, left; 419367b65a4Snicm struct grid_cell gc; 420367b65a4Snicm int colour, active_colour; 421367b65a4Snicm char *label; 422367b65a4Snicm size_t len; 423367b65a4Snicm 424367b65a4Snicm if (sx < 6) 425367b65a4Snicm return; 426367b65a4Snicm n = winlink_count(&s->windows); 427367b65a4Snicm 428367b65a4Snicm memcpy(&gc, &grid_default_cell, sizeof gc); 429367b65a4Snicm colour = options_get_number(oo, "display-panes-colour"); 430367b65a4Snicm active_colour = options_get_number(oo, "display-panes-active-colour"); 431367b65a4Snicm 432367b65a4Snicm each = sx / n; 433367b65a4Snicm if (each < 24) { 434367b65a4Snicm n = (sx - 6) / 24; 435*c224fd4bSnicm if (n == 0) 436*c224fd4bSnicm n = 1; 437367b65a4Snicm each = (sx - 6) / n; 438367b65a4Snicm left = sx - (n * each); 439367b65a4Snicm 440367b65a4Snicm screen_write_cursormove(ctx, sx - left, 0); 441367b65a4Snicm screen_write_vline(ctx, sy, 0, 0); 442367b65a4Snicm screen_write_cursormove(ctx, sx - left + left / 2, sy / 2); 443367b65a4Snicm screen_write_puts(ctx, &grid_default_cell, "..."); 444367b65a4Snicm 445*c224fd4bSnicm if (each == 0) 446367b65a4Snicm return; 447367b65a4Snicm left = 0; 448367b65a4Snicm } else 449367b65a4Snicm left = sx - (n * each); 450367b65a4Snicm 451367b65a4Snicm wl = RB_MIN(winlinks, &s->windows); 452367b65a4Snicm for (i = 0; i < n; i++) { 453367b65a4Snicm if (wl == s->curw) 454367b65a4Snicm gc.fg = active_colour; 455367b65a4Snicm else 456367b65a4Snicm gc.fg = colour; 457367b65a4Snicm if (i == n - 1) 458367b65a4Snicm width = each + left; 459367b65a4Snicm else 460367b65a4Snicm width = each - 1; 461367b65a4Snicm w = wl->window; 462367b65a4Snicm 463367b65a4Snicm screen_write_cursormove(ctx, i * each, 0); 464367b65a4Snicm screen_write_preview(ctx, &w->active->base, width, sy); 465367b65a4Snicm 466367b65a4Snicm xasprintf(&label, " %u:%s ", wl->idx, w->name); 467367b65a4Snicm if (strlen(label) > width) 468367b65a4Snicm xasprintf(&label, " %u ", wl->idx); 469367b65a4Snicm len = strlen(label) / 2; 470367b65a4Snicm screen_write_cursormove(ctx, i * each + each / 2 - len, sy / 2); 471*c224fd4bSnicm if (len <= width) 472367b65a4Snicm screen_write_puts(ctx, &gc, "%s", label); 473367b65a4Snicm free(label); 474367b65a4Snicm 475367b65a4Snicm if (i != n - 1) { 476367b65a4Snicm screen_write_cursormove(ctx, i * each + width, 0); 477367b65a4Snicm screen_write_vline(ctx, sy, 0, 0); 478367b65a4Snicm } 479367b65a4Snicm wl = RB_NEXT(winlinks, &s->windows, wl); 480367b65a4Snicm } 481367b65a4Snicm } 482367b65a4Snicm 483367b65a4Snicm static void 484367b65a4Snicm window_tree_draw_window(struct session *s, struct window *w, 485367b65a4Snicm struct screen_write_ctx *ctx, u_int sx, u_int sy) 486367b65a4Snicm { 487367b65a4Snicm struct options *oo = s->options; 488367b65a4Snicm struct window_pane *wp; 489367b65a4Snicm u_int i, n, each, width, left; 490367b65a4Snicm struct grid_cell gc; 491367b65a4Snicm int colour, active_colour; 492367b65a4Snicm char *label; 493367b65a4Snicm size_t len; 494367b65a4Snicm 495367b65a4Snicm if (sx < 6) 496367b65a4Snicm return; 497367b65a4Snicm n = window_count_panes(w); 498367b65a4Snicm 499367b65a4Snicm memcpy(&gc, &grid_default_cell, sizeof gc); 500367b65a4Snicm colour = options_get_number(oo, "display-panes-colour"); 501367b65a4Snicm active_colour = options_get_number(oo, "display-panes-active-colour"); 502367b65a4Snicm 503367b65a4Snicm each = sx / n; 504367b65a4Snicm if (each < 24) { 505367b65a4Snicm n = (sx - 6) / 24; 506*c224fd4bSnicm if (n == 0) 507*c224fd4bSnicm n = 1; 508367b65a4Snicm each = (sx - 6) / n; 509367b65a4Snicm left = sx - (n * each); 510367b65a4Snicm 511367b65a4Snicm screen_write_cursormove(ctx, sx - left, 0); 512367b65a4Snicm screen_write_vline(ctx, sy, 0, 0); 513367b65a4Snicm screen_write_cursormove(ctx, sx - left + left / 2, sy / 2); 514367b65a4Snicm screen_write_puts(ctx, &grid_default_cell, "..."); 515367b65a4Snicm 516*c224fd4bSnicm if (each == 0) 517367b65a4Snicm return; 518367b65a4Snicm left = 0; 519367b65a4Snicm } else 520367b65a4Snicm left = sx - (n * each); 521367b65a4Snicm 522367b65a4Snicm wp = TAILQ_FIRST(&w->panes); 523367b65a4Snicm for (i = 0; i < n; i++) { 524367b65a4Snicm if (wp == w->active) 525367b65a4Snicm gc.fg = active_colour; 526367b65a4Snicm else 527367b65a4Snicm gc.fg = colour; 528367b65a4Snicm if (i == n - 1) 529367b65a4Snicm width = each + left; 530367b65a4Snicm else 531367b65a4Snicm width = each - 1; 532367b65a4Snicm 533367b65a4Snicm screen_write_cursormove(ctx, i * each, 0); 534367b65a4Snicm screen_write_preview(ctx, &wp->base, width, sy); 535367b65a4Snicm 536367b65a4Snicm xasprintf(&label, " %u ", i); 537367b65a4Snicm len = strlen(label) / 2; 538367b65a4Snicm screen_write_cursormove(ctx, i * each + each / 2 - len, sy / 2); 539*c224fd4bSnicm if (len <= width) 540367b65a4Snicm screen_write_puts(ctx, &gc, "%s", label); 541367b65a4Snicm free(label); 542367b65a4Snicm 543367b65a4Snicm if (i != n - 1) { 544367b65a4Snicm screen_write_cursormove(ctx, i * each + width, 0); 545367b65a4Snicm screen_write_vline(ctx, sy, 0, 0); 546367b65a4Snicm } 547367b65a4Snicm wp = TAILQ_NEXT(wp, entry); 548367b65a4Snicm } 549367b65a4Snicm } 550367b65a4Snicm 551a42faf7dSnicm static struct screen * 552a42faf7dSnicm window_tree_draw(__unused void *modedata, void *itemdata, u_int sx, u_int sy) 553a42faf7dSnicm { 554a42faf7dSnicm struct window_tree_itemdata *item = itemdata; 555a42faf7dSnicm struct session *sp; 556a42faf7dSnicm struct winlink *wlp; 557a42faf7dSnicm struct window_pane *wp; 558a42faf7dSnicm static struct screen s; 559a42faf7dSnicm struct screen_write_ctx ctx; 560a42faf7dSnicm 561a42faf7dSnicm window_tree_pull_item(item, &sp, &wlp, &wp); 562a42faf7dSnicm if (wp == NULL) 563a42faf7dSnicm return (NULL); 564a42faf7dSnicm 565a42faf7dSnicm screen_init(&s, sx, sy, 0); 566a42faf7dSnicm screen_write_start(&ctx, NULL, &s); 567a42faf7dSnicm 568367b65a4Snicm switch (item->type) { 569367b65a4Snicm case WINDOW_TREE_NONE: 570367b65a4Snicm return (0); 571367b65a4Snicm case WINDOW_TREE_SESSION: 572367b65a4Snicm window_tree_draw_session(sp, &ctx, sx, sy); 573367b65a4Snicm break; 574367b65a4Snicm case WINDOW_TREE_WINDOW: 575367b65a4Snicm window_tree_draw_window(sp, wlp->window, &ctx, sx, sy); 576367b65a4Snicm break; 577367b65a4Snicm case WINDOW_TREE_PANE: 578a42faf7dSnicm screen_write_preview(&ctx, &wp->base, sx, sy); 579367b65a4Snicm break; 580367b65a4Snicm } 581a42faf7dSnicm 582a42faf7dSnicm screen_write_stop(&ctx); 583a42faf7dSnicm return (&s); 584a42faf7dSnicm } 585a42faf7dSnicm 586943a08b1Snicm static int 587943a08b1Snicm window_tree_search(__unused void *modedata, void *itemdata, const char *ss) 588943a08b1Snicm { 589943a08b1Snicm struct window_tree_itemdata *item = itemdata; 590943a08b1Snicm struct session *s; 591943a08b1Snicm struct winlink *wl; 592943a08b1Snicm struct window_pane *wp; 593943a08b1Snicm const char *cmd; 594943a08b1Snicm 595943a08b1Snicm window_tree_pull_item(item, &s, &wl, &wp); 596943a08b1Snicm 597943a08b1Snicm switch (item->type) { 598943a08b1Snicm case WINDOW_TREE_NONE: 599943a08b1Snicm return (0); 600943a08b1Snicm case WINDOW_TREE_SESSION: 601943a08b1Snicm if (s == NULL) 602943a08b1Snicm return (0); 603943a08b1Snicm return (strstr(s->name, ss) != NULL); 604943a08b1Snicm case WINDOW_TREE_WINDOW: 605943a08b1Snicm if (s == NULL || wl == NULL) 606943a08b1Snicm return (0); 607943a08b1Snicm return (strstr(wl->window->name, ss) != NULL); 608943a08b1Snicm case WINDOW_TREE_PANE: 609943a08b1Snicm if (s == NULL || wl == NULL || wp == NULL) 610943a08b1Snicm break; 611943a08b1Snicm cmd = get_proc_name(wp->fd, wp->tty); 612943a08b1Snicm if (cmd == NULL || *cmd == '\0') 613943a08b1Snicm return (0); 614943a08b1Snicm return (strstr(cmd, ss) != NULL); 615943a08b1Snicm } 616943a08b1Snicm return (0); 617943a08b1Snicm } 618943a08b1Snicm 619a42faf7dSnicm static struct screen * 620a42faf7dSnicm window_tree_init(struct window_pane *wp, struct cmd_find_state *fs, 621a42faf7dSnicm struct args *args) 622a42faf7dSnicm { 623a42faf7dSnicm struct window_tree_modedata *data; 624a42faf7dSnicm struct screen *s; 625a42faf7dSnicm 626a42faf7dSnicm wp->modedata = data = xcalloc(1, sizeof *data); 627a42faf7dSnicm 628a42faf7dSnicm if (args_has(args, 's')) 629a42faf7dSnicm data->type = WINDOW_TREE_SESSION; 630a42faf7dSnicm else if (args_has(args, 'w')) 631a42faf7dSnicm data->type = WINDOW_TREE_WINDOW; 632a42faf7dSnicm else 633a42faf7dSnicm data->type = WINDOW_TREE_PANE; 634a42faf7dSnicm memcpy(&data->fs, fs, sizeof data->fs); 635a42faf7dSnicm 636a42faf7dSnicm data->wp = wp; 637a42faf7dSnicm data->references = 1; 638a42faf7dSnicm 639a42faf7dSnicm if (args == NULL || args->argc == 0) 640a42faf7dSnicm data->command = xstrdup(WINDOW_TREE_DEFAULT_COMMAND); 641a42faf7dSnicm else 642a42faf7dSnicm data->command = xstrdup(args->argv[0]); 643a42faf7dSnicm 644b38aa712Snicm data->data = mode_tree_start(wp, args, window_tree_build, 645b38aa712Snicm window_tree_draw, window_tree_search, data, window_tree_sort_list, 646a42faf7dSnicm nitems(window_tree_sort_list), &s); 647a42faf7dSnicm 648a42faf7dSnicm mode_tree_build(data->data); 649a42faf7dSnicm mode_tree_draw(data->data); 650a42faf7dSnicm 651a42faf7dSnicm data->type = WINDOW_TREE_NONE; 652a42faf7dSnicm 653a42faf7dSnicm return (s); 654a42faf7dSnicm } 655a42faf7dSnicm 656a42faf7dSnicm static void 657a42faf7dSnicm window_tree_destroy(struct window_tree_modedata *data) 658a42faf7dSnicm { 659a42faf7dSnicm u_int i; 660a42faf7dSnicm 661a42faf7dSnicm if (--data->references != 0) 662a42faf7dSnicm return; 663a42faf7dSnicm 664a42faf7dSnicm mode_tree_free(data->data); 665a42faf7dSnicm 666a42faf7dSnicm for (i = 0; i < data->item_size; i++) 667a42faf7dSnicm window_tree_free_item(data->item_list[i]); 668a42faf7dSnicm free(data->item_list); 669a42faf7dSnicm 670a42faf7dSnicm free(data->command); 671a42faf7dSnicm free(data); 672a42faf7dSnicm } 673a42faf7dSnicm 674a42faf7dSnicm static void 675a42faf7dSnicm window_tree_free(struct window_pane *wp) 676a42faf7dSnicm { 677a42faf7dSnicm struct window_tree_modedata *data = wp->modedata; 678a42faf7dSnicm 679a42faf7dSnicm if (data == NULL) 680a42faf7dSnicm return; 681a42faf7dSnicm 682a42faf7dSnicm data->dead = 1; 683a42faf7dSnicm window_tree_destroy(data); 684a42faf7dSnicm } 685a42faf7dSnicm 686a42faf7dSnicm static void 687a42faf7dSnicm window_tree_resize(struct window_pane *wp, u_int sx, u_int sy) 688a42faf7dSnicm { 689a42faf7dSnicm struct window_tree_modedata *data = wp->modedata; 690a42faf7dSnicm 691a42faf7dSnicm mode_tree_resize(data->data, sx, sy); 692a42faf7dSnicm } 693a42faf7dSnicm 694a42faf7dSnicm static char * 695a42faf7dSnicm window_tree_get_target(struct window_tree_itemdata *item, 696a42faf7dSnicm struct cmd_find_state *fs) 697a42faf7dSnicm { 698a42faf7dSnicm struct session *s; 699a42faf7dSnicm struct winlink *wl; 700a42faf7dSnicm struct window_pane *wp; 701a42faf7dSnicm char *target; 702a42faf7dSnicm 703a42faf7dSnicm window_tree_pull_item(item, &s, &wl, &wp); 704a42faf7dSnicm 705a42faf7dSnicm target = NULL; 706a42faf7dSnicm switch (item->type) { 707a42faf7dSnicm case WINDOW_TREE_NONE: 708a42faf7dSnicm break; 709a42faf7dSnicm case WINDOW_TREE_SESSION: 710a42faf7dSnicm if (s == NULL) 711a42faf7dSnicm break; 712a42faf7dSnicm xasprintf(&target, "=%s:", s->name); 713a42faf7dSnicm break; 714a42faf7dSnicm case WINDOW_TREE_WINDOW: 715a42faf7dSnicm if (s == NULL || wl == NULL) 716a42faf7dSnicm break; 717a42faf7dSnicm xasprintf(&target, "=%s:%u.", s->name, wl->idx); 718a42faf7dSnicm break; 719a42faf7dSnicm case WINDOW_TREE_PANE: 720a42faf7dSnicm if (s == NULL || wl == NULL || wp == NULL) 721a42faf7dSnicm break; 722a42faf7dSnicm xasprintf(&target, "=%s:%u.%%%u", s->name, wl->idx, wp->id); 723a42faf7dSnicm break; 724a42faf7dSnicm } 725a42faf7dSnicm if (target == NULL) 726a42faf7dSnicm cmd_find_clear_state(fs, 0); 727a42faf7dSnicm else 728a42faf7dSnicm cmd_find_from_winlink_pane(fs, wl, wp); 729a42faf7dSnicm return (target); 730a42faf7dSnicm } 731a42faf7dSnicm 732a42faf7dSnicm static void 733a42faf7dSnicm window_tree_command_each(void* modedata, void* itemdata, __unused key_code key) 734a42faf7dSnicm { 735a42faf7dSnicm struct window_tree_modedata *data = modedata; 736a42faf7dSnicm struct window_tree_itemdata *item = itemdata; 737a42faf7dSnicm char *name; 738a42faf7dSnicm struct cmd_find_state fs; 739a42faf7dSnicm 740a42faf7dSnicm name = window_tree_get_target(item, &fs); 741a42faf7dSnicm if (name != NULL) 742a42faf7dSnicm mode_tree_run_command(data->client, &fs, data->entered, name); 743a42faf7dSnicm free(name); 744a42faf7dSnicm } 745a42faf7dSnicm 746a42faf7dSnicm static enum cmd_retval 747a42faf7dSnicm window_tree_command_done(__unused struct cmdq_item *item, void *modedata) 748a42faf7dSnicm { 749a42faf7dSnicm struct window_tree_modedata *data = modedata; 750a42faf7dSnicm 751a42faf7dSnicm if (!data->dead) { 752a42faf7dSnicm mode_tree_build(data->data); 753a42faf7dSnicm mode_tree_draw(data->data); 754a42faf7dSnicm data->wp->flags |= PANE_REDRAW; 755a42faf7dSnicm } 756a42faf7dSnicm window_tree_destroy(data); 757a42faf7dSnicm return (CMD_RETURN_NORMAL); 758a42faf7dSnicm } 759a42faf7dSnicm 760a42faf7dSnicm static int 761a42faf7dSnicm window_tree_command_callback(struct client *c, void *modedata, const char *s, 762a42faf7dSnicm __unused int done) 763a42faf7dSnicm { 764a42faf7dSnicm struct window_tree_modedata *data = modedata; 765a42faf7dSnicm 766a42faf7dSnicm if (data->dead) 767a42faf7dSnicm return (0); 768a42faf7dSnicm 769a42faf7dSnicm data->client = c; 770a42faf7dSnicm data->entered = s; 771a42faf7dSnicm 772a42faf7dSnicm mode_tree_each_tagged(data->data, window_tree_command_each, KEYC_NONE, 773a42faf7dSnicm 1); 774a42faf7dSnicm 775a42faf7dSnicm data->client = NULL; 776a42faf7dSnicm data->entered = NULL; 777a42faf7dSnicm 778a42faf7dSnicm data->references++; 779a42faf7dSnicm cmdq_append(c, cmdq_get_callback(window_tree_command_done, data)); 780a42faf7dSnicm 781a42faf7dSnicm return (0); 782a42faf7dSnicm } 783a42faf7dSnicm 784a42faf7dSnicm static void 785a42faf7dSnicm window_tree_command_free(void *modedata) 786a42faf7dSnicm { 787a42faf7dSnicm struct window_tree_modedata *data = modedata; 788a42faf7dSnicm 789a42faf7dSnicm window_tree_destroy(data); 790a42faf7dSnicm } 791a42faf7dSnicm 792a42faf7dSnicm static void 793a42faf7dSnicm window_tree_key(struct window_pane *wp, struct client *c, 794a42faf7dSnicm __unused struct session *s, key_code key, struct mouse_event *m) 795a42faf7dSnicm { 796a42faf7dSnicm struct window_tree_modedata *data = wp->modedata; 797a42faf7dSnicm struct window_tree_itemdata *item; 798a42faf7dSnicm char *command, *name, *prompt; 799a42faf7dSnicm struct cmd_find_state fs; 800a42faf7dSnicm int finished; 801a42faf7dSnicm u_int tagged; 802a42faf7dSnicm 803943a08b1Snicm finished = mode_tree_key(data->data, c, &key, m); 804a42faf7dSnicm switch (key) { 805a42faf7dSnicm case ':': 806a42faf7dSnicm tagged = mode_tree_count_tagged(data->data); 807a42faf7dSnicm if (tagged != 0) 808a42faf7dSnicm xasprintf(&prompt, "(%u tagged) ", tagged); 809a42faf7dSnicm else 810a42faf7dSnicm xasprintf(&prompt, "(current) "); 811a42faf7dSnicm data->references++; 812a42faf7dSnicm status_prompt_set(c, prompt, "", window_tree_command_callback, 813a42faf7dSnicm window_tree_command_free, data, PROMPT_NOFORMAT); 814a42faf7dSnicm free(prompt); 815a42faf7dSnicm break; 816a42faf7dSnicm case '\r': 817a42faf7dSnicm item = mode_tree_get_current(data->data); 818a42faf7dSnicm command = xstrdup(data->command); 819a42faf7dSnicm name = window_tree_get_target(item, &fs); 820a42faf7dSnicm window_pane_reset_mode(wp); 821a42faf7dSnicm if (name != NULL) 8228af98d93Snicm mode_tree_run_command(c, NULL, command, name); 823a42faf7dSnicm free(name); 824a42faf7dSnicm free(command); 825a42faf7dSnicm return; 826a42faf7dSnicm } 827a42faf7dSnicm if (finished) 828a42faf7dSnicm window_pane_reset_mode(wp); 829a42faf7dSnicm else { 830a42faf7dSnicm mode_tree_draw(data->data); 831a42faf7dSnicm wp->flags |= PANE_REDRAW; 832a42faf7dSnicm } 833a42faf7dSnicm } 834