Annotation of src/usr.bin/tmux/format.c, Revision 1.95
1.95 ! nicm 1: /* $OpenBSD: format.c,v 1.94 2015/11/12 11:09:11 nicm Exp $ */
1.1 nicm 2:
3: /*
4: * Copyright (c) 2011 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>
1.55 nicm 20: #include <sys/wait.h>
1.1 nicm 21:
1.30 nicm 22: #include <ctype.h>
23: #include <errno.h>
1.87 nicm 24: #include <libgen.h>
1.1 nicm 25: #include <netdb.h>
26: #include <stdarg.h>
1.9 nicm 27: #include <stdlib.h>
1.1 nicm 28: #include <string.h>
29: #include <time.h>
30: #include <unistd.h>
31:
32: #include "tmux.h"
33:
34: /*
35: * Build a list of key-value pairs and use them to expand #{key} entries in a
36: * string.
37: */
38:
1.79 nicm 39: struct format_entry;
40: typedef void (*format_cb)(struct format_tree *, struct format_entry *);
41:
1.68 nicm 42: void format_job_callback(struct job *);
1.86 nicm 43: char *format_job_get(struct format_tree *, const char *);
1.77 nicm 44: void format_job_timer(int, short, void *);
1.68 nicm 45:
1.79 nicm 46: void format_cb_host(struct format_tree *, struct format_entry *);
47: void format_cb_host_short(struct format_tree *, struct format_entry *);
48: void format_cb_pid(struct format_tree *, struct format_entry *);
1.80 nicm 49: void format_cb_session_alerts(struct format_tree *, struct format_entry *);
50: void format_cb_window_layout(struct format_tree *, struct format_entry *);
1.79 nicm 51: void format_cb_start_command(struct format_tree *, struct format_entry *);
52: void format_cb_current_command(struct format_tree *, struct format_entry *);
1.80 nicm 53: void format_cb_history_bytes(struct format_tree *, struct format_entry *);
54: void format_cb_pane_tabs(struct format_tree *, struct format_entry *);
1.79 nicm 55:
1.87 nicm 56: char *format_find(struct format_tree *, const char *, int);
1.79 nicm 57: void format_add_cb(struct format_tree *, const char *, format_cb);
1.87 nicm 58: void format_add_tv(struct format_tree *, const char *, struct timeval *);
1.32 nicm 59: int format_replace(struct format_tree *, const char *, size_t, char **,
60: size_t *, size_t *);
1.60 nicm 61: char *format_time_string(time_t);
1.57 nicm 62:
63: void format_defaults_pane_tabs(struct format_tree *, struct window_pane *);
64: void format_defaults_session(struct format_tree *, struct session *);
65: void format_defaults_client(struct format_tree *, struct client *);
66: void format_defaults_winlink(struct format_tree *, struct session *,
67: struct winlink *);
1.1 nicm 68:
1.68 nicm 69: /* Entry in format job tree. */
70: struct format_job {
71: const char *cmd;
72:
73: time_t last;
74: char *out;
75:
76: struct job *job;
77: int status;
78:
79: RB_ENTRY(format_job) entry;
80: };
81:
82: /* Format job tree. */
1.77 nicm 83: struct event format_job_event;
1.68 nicm 84: int format_job_cmp(struct format_job *, struct format_job *);
85: RB_HEAD(format_job_tree, format_job) format_jobs = RB_INITIALIZER();
86: RB_PROTOTYPE(format_job_tree, format_job, entry, format_job_cmp);
87: RB_GENERATE(format_job_tree, format_job, entry, format_job_cmp);
88:
89: /* Format job tree comparison function. */
90: int
91: format_job_cmp(struct format_job *fj1, struct format_job *fj2)
92: {
93: return (strcmp(fj1->cmd, fj2->cmd));
94: }
95:
1.87 nicm 96: /* Format modifiers. */
97: #define FORMAT_TIMESTRING 0x1
98: #define FORMAT_BASENAME 0x2
99: #define FORMAT_DIRNAME 0x4
100:
1.54 nicm 101: /* Entry in format tree. */
102: struct format_entry {
1.79 nicm 103: char *key;
104: char *value;
1.87 nicm 105: time_t t;
1.79 nicm 106: format_cb cb;
107: RB_ENTRY(format_entry) entry;
1.54 nicm 108: };
109:
1.68 nicm 110: /* Format entry tree. */
1.54 nicm 111: struct format_tree {
1.79 nicm 112: struct window *w;
113: struct session *s;
114: struct window_pane *wp;
1.54 nicm 115:
1.84 nicm 116: int flags;
1.68 nicm 117:
118: RB_HEAD(format_entry_tree, format_entry) tree;
1.54 nicm 119: };
1.68 nicm 120: int format_entry_cmp(struct format_entry *, struct format_entry *);
121: RB_PROTOTYPE(format_entry_tree, format_entry, entry, format_entry_cmp);
122: RB_GENERATE(format_entry_tree, format_entry, entry, format_entry_cmp);
1.54 nicm 123:
1.68 nicm 124: /* Format entry tree comparison function. */
1.1 nicm 125: int
1.68 nicm 126: format_entry_cmp(struct format_entry *fe1, struct format_entry *fe2)
1.1 nicm 127: {
128: return (strcmp(fe1->key, fe2->key));
129: }
130:
1.25 nicm 131: /* Single-character uppercase aliases. */
132: const char *format_upper[] = {
1.1 nicm 133: NULL, /* A */
134: NULL, /* B */
135: NULL, /* C */
136: "pane_id", /* D */
137: NULL, /* E */
138: "window_flags", /* F */
139: NULL, /* G */
140: "host", /* H */
141: "window_index", /* I */
142: NULL, /* J */
143: NULL, /* K */
144: NULL, /* L */
145: NULL, /* M */
146: NULL, /* N */
147: NULL, /* O */
148: "pane_index", /* P */
149: NULL, /* Q */
150: NULL, /* R */
151: "session_name", /* S */
152: "pane_title", /* T */
153: NULL, /* U */
154: NULL, /* V */
155: "window_name", /* W */
156: NULL, /* X */
157: NULL, /* Y */
158: NULL /* Z */
159: };
160:
1.25 nicm 161: /* Single-character lowercase aliases. */
162: const char *format_lower[] = {
163: NULL, /* a */
164: NULL, /* b */
165: NULL, /* c */
166: NULL, /* d */
167: NULL, /* e */
168: NULL, /* f */
169: NULL, /* g */
170: "host_short", /* h */
171: NULL, /* i */
172: NULL, /* j */
173: NULL, /* k */
174: NULL, /* l */
175: NULL, /* m */
176: NULL, /* n */
177: NULL, /* o */
178: NULL, /* p */
179: NULL, /* q */
180: NULL, /* r */
181: NULL, /* s */
182: NULL, /* t */
183: NULL, /* u */
184: NULL, /* v */
185: NULL, /* w */
186: NULL, /* x */
187: NULL, /* y */
188: NULL /* z */
189: };
190:
1.68 nicm 191: /* Format job callback. */
192: void
193: format_job_callback(struct job *job)
194: {
195: struct format_job *fj = job->data;
196: char *line, *buf;
197: size_t len;
198: struct client *c;
199:
200: fj->job = NULL;
201: free(fj->out);
202:
203: buf = NULL;
204: if ((line = evbuffer_readline(job->event->input)) == NULL) {
205: len = EVBUFFER_LENGTH(job->event->input);
206: buf = xmalloc(len + 1);
207: if (len != 0)
208: memcpy(buf, EVBUFFER_DATA(job->event->input), len);
209: buf[len] = '\0';
210: } else
211: buf = line;
212: fj->out = buf;
213:
214: if (fj->status) {
215: TAILQ_FOREACH(c, &clients, entry)
216: server_status_client(c);
217: fj->status = 0;
218: }
1.77 nicm 219:
220: log_debug("%s: %s: %s", __func__, fj->cmd, fj->out);
1.68 nicm 221: }
222:
223: /* Find a job. */
1.86 nicm 224: char *
1.68 nicm 225: format_job_get(struct format_tree *ft, const char *cmd)
226: {
227: struct format_job fj0, *fj;
1.84 nicm 228: time_t t;
1.68 nicm 229:
230: fj0.cmd = cmd;
1.82 nicm 231: if ((fj = RB_FIND(format_job_tree, &format_jobs, &fj0)) == NULL) {
1.68 nicm 232: fj = xcalloc(1, sizeof *fj);
233: fj->cmd = xstrdup(cmd);
234:
235: xasprintf(&fj->out, "<'%s' not ready>", fj->cmd);
236:
237: RB_INSERT(format_job_tree, &format_jobs, fj);
238: }
239:
1.84 nicm 240: t = time(NULL);
241: if (fj->job == NULL && ((ft->flags & FORMAT_FORCE) || fj->last != t)) {
1.92 nicm 242: fj->job = job_run(fj->cmd, NULL, NULL, format_job_callback,
1.68 nicm 243: NULL, fj);
244: if (fj->job == NULL) {
245: free(fj->out);
246: xasprintf(&fj->out, "<'%s' didn't start>", fj->cmd);
247: }
1.84 nicm 248: fj->last = t;
1.68 nicm 249: }
1.84 nicm 250:
251: if (ft->flags & FORMAT_STATUS)
252: fj->status = 1;
1.68 nicm 253:
1.86 nicm 254: return (format_expand(ft, fj->out));
1.68 nicm 255: }
256:
257: /* Remove old jobs. */
258: void
1.77 nicm 259: format_job_timer(unused int fd, unused short events, unused void *arg)
1.68 nicm 260: {
261: struct format_job *fj, *fj1;
262: time_t now;
1.77 nicm 263: struct timeval tv = { .tv_sec = 60 };
1.68 nicm 264:
265: now = time(NULL);
266: RB_FOREACH_SAFE(fj, format_job_tree, &format_jobs, fj1) {
267: if (fj->last > now || now - fj->last < 3600)
268: continue;
269: RB_REMOVE(format_job_tree, &format_jobs, fj);
270:
1.77 nicm 271: log_debug("%s: %s", __func__, fj->cmd);
272:
1.68 nicm 273: if (fj->job != NULL)
274: job_free(fj->job);
275:
1.78 nicm 276: free((void *)fj->cmd);
1.68 nicm 277: free(fj->out);
278:
279: free(fj);
280: }
1.77 nicm 281:
282: evtimer_del(&format_job_event);
283: evtimer_add(&format_job_event, &tv);
1.68 nicm 284: }
285:
1.79 nicm 286: /* Callback for host. */
287: void
288: format_cb_host(unused struct format_tree *ft, struct format_entry *fe)
289: {
290: char host[HOST_NAME_MAX + 1];
291:
292: if (gethostname(host, sizeof host) != 0)
293: fe->value = xstrdup("");
294: else
295: fe->value = xstrdup(host);
296: }
297:
298: /* Callback for host_short. */
299: void
300: format_cb_host_short(unused struct format_tree *ft, struct format_entry *fe)
301: {
302: char host[HOST_NAME_MAX + 1], *cp;
303:
304: if (gethostname(host, sizeof host) != 0)
305: fe->value = xstrdup("");
306: else {
307: if ((cp = strchr(host, '.')) != NULL)
308: *cp = '\0';
309: fe->value = xstrdup(host);
310: }
311: }
312:
313: /* Callback for pid. */
314: void
315: format_cb_pid(unused struct format_tree *ft, struct format_entry *fe)
316: {
317: xasprintf(&fe->value, "%ld", (long)getpid());
318: }
319:
1.80 nicm 320: /* Callback for session_alerts. */
321: void
322: format_cb_session_alerts(struct format_tree *ft, struct format_entry *fe)
323: {
324: struct session *s = ft->s;
325: struct winlink *wl;
326: char alerts[256], tmp[16];
327:
328: if (s == NULL)
329: return;
330:
331: *alerts = '\0';
332: RB_FOREACH(wl, winlinks, &s->windows) {
333: if ((wl->flags & WINLINK_ALERTFLAGS) == 0)
334: continue;
335: xsnprintf(tmp, sizeof tmp, "%u", wl->idx);
336:
337: if (*alerts != '\0')
338: strlcat(alerts, ",", sizeof alerts);
339: strlcat(alerts, tmp, sizeof alerts);
340: if (wl->flags & WINLINK_ACTIVITY)
341: strlcat(alerts, "#", sizeof alerts);
342: if (wl->flags & WINLINK_BELL)
343: strlcat(alerts, "!", sizeof alerts);
344: if (wl->flags & WINLINK_SILENCE)
345: strlcat(alerts, "~", sizeof alerts);
346: }
347: fe->value = xstrdup(alerts);
348: }
349:
350: /* Callback for window_layout. */
351: void
352: format_cb_window_layout(struct format_tree *ft, struct format_entry *fe)
353: {
354: struct window *w = ft->w;
355:
356: if (w == NULL)
357: return;
358:
359: if (w->saved_layout_root != NULL)
360: fe->value = layout_dump(w->saved_layout_root);
361: else
362: fe->value = layout_dump(w->layout_root);
363: }
364:
1.79 nicm 365: /* Callback for pane_start_command. */
366: void
367: format_cb_start_command(struct format_tree *ft, struct format_entry *fe)
368: {
369: struct window_pane *wp = ft->wp;
370:
371: if (wp == NULL)
372: return;
373:
374: fe->value = cmd_stringify_argv(wp->argc, wp->argv);
375: }
376:
377: /* Callback for pane_current_command. */
378: void
379: format_cb_current_command(struct format_tree *ft, struct format_entry *fe)
380: {
381: struct window_pane *wp = ft->wp;
382: char *cmd;
383:
384: if (wp == NULL)
385: return;
386:
387: cmd = get_proc_name(wp->fd, wp->tty);
388: if (cmd == NULL || *cmd == '\0') {
389: free(cmd);
390: cmd = cmd_stringify_argv(wp->argc, wp->argv);
391: if (cmd == NULL || *cmd == '\0') {
392: free(cmd);
393: cmd = xstrdup(wp->shell);
394: }
395: }
396: fe->value = parse_window_name(cmd);
397: free(cmd);
398: }
399:
1.80 nicm 400: /* Callback for history_bytes. */
401: void
402: format_cb_history_bytes(struct format_tree *ft, struct format_entry *fe)
403: {
404: struct window_pane *wp = ft->wp;
405: struct grid *gd;
406: struct grid_line *gl;
407: unsigned long long size;
408: u_int i;
409:
410: if (wp == NULL)
411: return;
412: gd = wp->base.grid;
413:
414: size = 0;
415: for (i = 0; i < gd->hsize; i++) {
416: gl = &gd->linedata[i];
417: size += gl->cellsize * sizeof *gl->celldata;
1.95 ! nicm 418: size += gl->extdsize * sizeof *gl->extddata;
1.80 nicm 419: }
420: size += gd->hsize * sizeof *gd->linedata;
421:
422: xasprintf(&fe->value, "%llu", size);
423: }
424:
425: /* Callback for pane_tabs. */
426: void
427: format_cb_pane_tabs(struct format_tree *ft, struct format_entry *fe)
428: {
429: struct window_pane *wp = ft->wp;
430: struct evbuffer *buffer;
431: u_int i;
432: int size;
433:
434: if (wp == NULL)
435: return;
436:
437: buffer = evbuffer_new();
438: for (i = 0; i < wp->base.grid->sx; i++) {
439: if (!bit_test(wp->base.tabs, i))
440: continue;
441:
442: if (EVBUFFER_LENGTH(buffer) > 0)
443: evbuffer_add(buffer, ",", 1);
444: evbuffer_add_printf(buffer, "%u", i);
445: }
446: size = EVBUFFER_LENGTH(buffer);
447: xasprintf(&fe->value, "%.*s", size, EVBUFFER_DATA(buffer));
448: evbuffer_free(buffer);
449: }
450:
1.1 nicm 451: /* Create a new tree. */
452: struct format_tree *
453: format_create(void)
454: {
1.84 nicm 455: return (format_create_flags(0));
1.68 nicm 456: }
457:
458: /* Create a new tree for the status line. */
459: struct format_tree *
1.84 nicm 460: format_create_flags(int flags)
1.68 nicm 461: {
1.1 nicm 462: struct format_tree *ft;
1.77 nicm 463:
464: if (!event_initialized(&format_job_event)) {
465: evtimer_set(&format_job_event, format_job_timer, NULL);
466: format_job_timer(-1, 0, NULL);
467: }
1.1 nicm 468:
1.54 nicm 469: ft = xcalloc(1, sizeof *ft);
470: RB_INIT(&ft->tree);
1.84 nicm 471: ft->flags = flags;
1.1 nicm 472:
1.79 nicm 473: format_add_cb(ft, "host", format_cb_host);
474: format_add_cb(ft, "host_short", format_cb_host_short);
475: format_add_cb(ft, "pid", format_cb_pid);
1.1 nicm 476:
477: return (ft);
478: }
479:
480: /* Free a tree. */
481: void
482: format_free(struct format_tree *ft)
483: {
1.54 nicm 484: struct format_entry *fe, *fe1;
1.1 nicm 485:
1.68 nicm 486: RB_FOREACH_SAFE(fe, format_entry_tree, &ft->tree, fe1) {
487: RB_REMOVE(format_entry_tree, &ft->tree, fe);
1.9 nicm 488: free(fe->value);
489: free(fe->key);
490: free(fe);
1.1 nicm 491: }
492:
1.25 nicm 493: free(ft);
1.1 nicm 494: }
495:
496: /* Add a key-value pair. */
497: void
498: format_add(struct format_tree *ft, const char *key, const char *fmt, ...)
499: {
500: struct format_entry *fe;
1.28 nicm 501: struct format_entry *fe_now;
1.1 nicm 502: va_list ap;
503:
504: fe = xmalloc(sizeof *fe);
505: fe->key = xstrdup(key);
506:
1.79 nicm 507: fe_now = RB_INSERT(format_entry_tree, &ft->tree, fe);
508: if (fe_now != NULL) {
509: free(fe->key);
510: free(fe);
511: free(fe_now->value);
512: fe = fe_now;
513: }
514:
515: fe->cb = NULL;
1.87 nicm 516: fe->t = 0;
1.79 nicm 517:
1.1 nicm 518: va_start(ap, fmt);
519: xvasprintf(&fe->value, fmt, ap);
520: va_end(ap);
1.79 nicm 521: }
522:
1.87 nicm 523: /* Add a key and time. */
524: void
525: format_add_tv(struct format_tree *ft, const char *key, struct timeval *tv)
526: {
527: struct format_entry *fe;
528: struct format_entry *fe_now;
529:
530: fe = xmalloc(sizeof *fe);
531: fe->key = xstrdup(key);
532:
533: fe_now = RB_INSERT(format_entry_tree, &ft->tree, fe);
534: if (fe_now != NULL) {
535: free(fe->key);
536: free(fe);
537: free(fe_now->value);
538: fe = fe_now;
539: }
540:
541: fe->cb = NULL;
542: fe->t = tv->tv_sec;
543:
544: fe->value = NULL;
545: }
546:
1.79 nicm 547: /* Add a key and function. */
548: void
549: format_add_cb(struct format_tree *ft, const char *key, format_cb cb)
550: {
551: struct format_entry *fe;
552: struct format_entry *fe_now;
553:
554: fe = xmalloc(sizeof *fe);
555: fe->key = xstrdup(key);
1.1 nicm 556:
1.68 nicm 557: fe_now = RB_INSERT(format_entry_tree, &ft->tree, fe);
1.28 nicm 558: if (fe_now != NULL) {
559: free(fe->key);
560: free(fe);
1.79 nicm 561: free(fe_now->value);
562: fe = fe_now;
1.28 nicm 563: }
1.79 nicm 564:
565: fe->cb = cb;
1.87 nicm 566: fe->t = 0;
1.79 nicm 567:
568: fe->value = NULL;
1.1 nicm 569: }
570:
571: /* Find a format entry. */
1.87 nicm 572: char *
573: format_find(struct format_tree *ft, const char *key, int modifiers)
1.1 nicm 574: {
575: struct format_entry *fe, fe_find;
1.54 nicm 576: struct options_entry *o;
1.76 nicm 577: struct environ_entry *envent;
1.87 nicm 578: static char s[64];
579: const char *found;
580: char *copy, *saved;
581:
582: found = NULL;
583:
584: if (~modifiers & FORMAT_TIMESTRING) {
1.90 nicm 585: o = options_find(global_options, key);
1.87 nicm 586: if (o == NULL && ft->w != NULL)
1.90 nicm 587: o = options_find(ft->w->options, key);
1.87 nicm 588: if (o == NULL)
1.90 nicm 589: o = options_find(global_w_options, key);
1.87 nicm 590: if (o == NULL && ft->s != NULL)
1.90 nicm 591: o = options_find(ft->s->options, key);
1.87 nicm 592: if (o == NULL)
1.90 nicm 593: o = options_find(global_s_options, key);
1.87 nicm 594: if (o != NULL) {
595: switch (o->type) {
596: case OPTIONS_STRING:
597: found = o->str;
598: goto found;
599: case OPTIONS_NUMBER:
600: xsnprintf(s, sizeof s, "%lld", o->num);
601: found = s;
602: goto found;
603: case OPTIONS_STYLE:
604: found = style_tostring(&o->style);
605: goto found;
606: }
1.54 nicm 607: }
608: }
1.1 nicm 609:
610: fe_find.key = (char *) key;
1.68 nicm 611: fe = RB_FIND(format_entry_tree, &ft->tree, &fe_find);
1.79 nicm 612: if (fe != NULL) {
1.87 nicm 613: if (modifiers & FORMAT_TIMESTRING) {
614: if (fe->t == 0)
615: return (NULL);
616: ctime_r(&fe->t, s);
617: s[strcspn(s, "\n")] = '\0';
618: found = s;
619: goto found;
620: }
621: if (fe->t != 0) {
622: xsnprintf(s, sizeof s, "%lld", (long long)fe->t);
623: found = s;
624: goto found;
625: }
1.79 nicm 626: if (fe->value == NULL && fe->cb != NULL)
627: fe->cb(ft, fe);
1.87 nicm 628: found = fe->value;
629: goto found;
1.79 nicm 630: }
1.76 nicm 631:
1.87 nicm 632: if (~modifiers & FORMAT_TIMESTRING) {
633: envent = NULL;
634: if (ft->s != NULL)
1.91 nicm 635: envent = environ_find(ft->s->environ, key);
1.87 nicm 636: if (envent == NULL)
1.91 nicm 637: envent = environ_find(global_environ, key);
1.87 nicm 638: if (envent != NULL) {
639: found = envent->value;
640: goto found;
641: }
642: }
1.76 nicm 643:
644: return (NULL);
1.87 nicm 645:
646: found:
1.88 nicm 647: if (found == NULL)
648: return (NULL);
1.87 nicm 649: copy = xstrdup(found);
650: if (modifiers & FORMAT_BASENAME) {
651: saved = copy;
652: copy = xstrdup(basename(saved));
653: free(saved);
654: }
655: if (modifiers & FORMAT_DIRNAME) {
656: saved = copy;
657: copy = xstrdup(dirname(saved));
658: free(saved);
659: }
660: return (copy);
1.1 nicm 661: }
662:
663: /*
664: * Replace a key/value pair in buffer. #{blah} is expanded directly,
665: * #{?blah,a,b} is replace with a if blah exists and is nonzero else b.
666: */
667: int
1.30 nicm 668: format_replace(struct format_tree *ft, const char *key, size_t keylen,
669: char **buf, size_t *len, size_t *off)
1.1 nicm 670: {
1.87 nicm 671: char *copy, *copy0, *endptr, *ptr, *saved, *trimmed, *value;
1.1 nicm 672: size_t valuelen;
1.44 nicm 673: u_long limit = 0;
1.89 nicm 674: int modifiers = 0, brackets;
1.1 nicm 675:
676: /* Make a copy of the key. */
1.30 nicm 677: copy0 = copy = xmalloc(keylen + 1);
1.1 nicm 678: memcpy(copy, key, keylen);
679: copy[keylen] = '\0';
680:
1.30 nicm 681: /* Is there a length limit or whatnot? */
1.87 nicm 682: switch (copy[0]) {
683: case '=':
684: errno = 0;
685: limit = strtoul(copy + 1, &endptr, 10);
686: if (errno == ERANGE && limit == ULONG_MAX)
687: break;
688: if (*endptr != ':')
689: break;
690: copy = endptr + 1;
691: break;
692: case 'b':
693: if (copy[1] != ':')
694: break;
695: modifiers |= FORMAT_BASENAME;
696: copy += 2;
697: break;
698: case 'd':
699: if (copy[1] != ':')
700: break;
701: modifiers |= FORMAT_DIRNAME;
702: copy += 2;
703: break;
704: case 't':
705: if (copy[1] != ':')
706: break;
707: modifiers |= FORMAT_TIMESTRING;
708: copy += 2;
709: break;
1.30 nicm 710: }
711:
1.1 nicm 712: /*
713: * Is this a conditional? If so, check it exists and extract either the
714: * first or second element. If not, look up the key directly.
715: */
716: if (*copy == '?') {
717: ptr = strchr(copy, ',');
718: if (ptr == NULL)
719: goto fail;
720: *ptr = '\0';
721:
1.89 nicm 722: value = ptr + 1;
723: saved = format_find(ft, copy + 1, modifiers);
724:
725: brackets = 0;
726: for (ptr = ptr + 1; *ptr != '\0'; ptr++) {
727: if (*ptr == '{')
728: brackets++;
729: if (*ptr == '}')
730: brackets--;
731: if (*ptr == ',' && brackets == 0)
732: break;
733: }
734: if (*ptr == '\0')
735: goto fail;
736:
737: if (saved != NULL && *saved != '\0' &&
738: (saved[0] != '0' || saved[1] != '\0')) {
1.1 nicm 739: *ptr = '\0';
1.89 nicm 740: } else
1.1 nicm 741: value = ptr + 1;
1.87 nicm 742: value = format_expand(ft, value);
743: free(saved);
744: saved = value;
1.1 nicm 745: } else {
1.87 nicm 746: saved = value = format_find(ft, copy, modifiers);
1.1 nicm 747: if (value == NULL)
1.87 nicm 748: saved = value = xstrdup("");
1.1 nicm 749: }
750:
1.30 nicm 751: /* Truncate the value if needed. */
1.44 nicm 752: if (limit != 0) {
753: value = trimmed = utf8_trimcstr(value, limit);
754: free(saved);
755: saved = trimmed;
756: }
757: valuelen = strlen(value);
1.30 nicm 758:
1.1 nicm 759: /* Expand the buffer and copy in the value. */
760: while (*len - *off < valuelen + 1) {
1.50 nicm 761: *buf = xreallocarray(*buf, 2, *len);
1.1 nicm 762: *len *= 2;
763: }
764: memcpy(*buf + *off, value, valuelen);
765: *off += valuelen;
766:
1.31 nicm 767: free(saved);
1.30 nicm 768: free(copy0);
1.1 nicm 769: return (0);
770:
771: fail:
1.30 nicm 772: free(copy0);
1.1 nicm 773: return (-1);
774: }
775:
1.58 nicm 776: /* Expand keys in a template, passing through strftime first. */
777: char *
778: format_expand_time(struct format_tree *ft, const char *fmt, time_t t)
779: {
780: char *tmp, *expanded;
781: size_t tmplen;
782: struct tm *tm;
783:
1.67 nicm 784: if (fmt == NULL || *fmt == '\0')
1.58 nicm 785: return (xstrdup(""));
786:
787: tm = localtime(&t);
788:
789: tmp = NULL;
790: tmplen = strlen(fmt);
791:
792: do {
793: tmp = xreallocarray(tmp, 2, tmplen);
794: tmplen *= 2;
795: } while (strftime(tmp, tmplen, fmt, tm) == 0);
796:
797: expanded = format_expand(ft, tmp);
798: free(tmp);
799:
800: return (expanded);
801: }
802:
1.1 nicm 803: /* Expand keys in a template. */
804: char *
805: format_expand(struct format_tree *ft, const char *fmt)
806: {
1.86 nicm 807: char *buf, *tmp, *cmd, *out;
1.79 nicm 808: const char *ptr, *s, *saved = fmt;
1.86 nicm 809: size_t off, len, n, outlen;
1.31 nicm 810: int ch, brackets;
1.58 nicm 811:
812: if (fmt == NULL)
813: return (xstrdup(""));
1.1 nicm 814:
815: len = 64;
816: buf = xmalloc(len);
817: off = 0;
818:
819: while (*fmt != '\0') {
820: if (*fmt != '#') {
821: while (len - off < 2) {
1.50 nicm 822: buf = xreallocarray(buf, 2, len);
1.1 nicm 823: len *= 2;
824: }
825: buf[off++] = *fmt++;
826: continue;
827: }
828: fmt++;
829:
830: ch = (u_char) *fmt++;
831: switch (ch) {
1.68 nicm 832: case '(':
833: brackets = 1;
834: for (ptr = fmt; *ptr != '\0'; ptr++) {
835: if (*ptr == '(')
836: brackets++;
837: if (*ptr == ')' && --brackets == 0)
838: break;
839: }
840: if (*ptr != ')' || brackets != 0)
841: break;
842: n = ptr - fmt;
843:
844: tmp = xmalloc(n + 1);
845: memcpy(tmp, fmt, n);
846: tmp[n] = '\0';
1.69 nicm 847: cmd = format_expand(ft, tmp);
1.68 nicm 848:
1.86 nicm 849: out = format_job_get(ft, cmd);
850: outlen = strlen(out);
1.69 nicm 851:
852: free(cmd);
853: free(tmp);
1.68 nicm 854:
1.86 nicm 855: while (len - off < outlen + 1) {
1.68 nicm 856: buf = xreallocarray(buf, 2, len);
857: len *= 2;
858: }
1.86 nicm 859: memcpy(buf + off, out, outlen);
860: off += outlen;
861:
862: free(out);
1.68 nicm 863:
864: fmt += n + 1;
865: continue;
1.1 nicm 866: case '{':
1.31 nicm 867: brackets = 1;
868: for (ptr = fmt; *ptr != '\0'; ptr++) {
869: if (*ptr == '{')
870: brackets++;
871: if (*ptr == '}' && --brackets == 0)
872: break;
873: }
874: if (*ptr != '}' || brackets != 0)
1.1 nicm 875: break;
876: n = ptr - fmt;
877:
878: if (format_replace(ft, fmt, n, &buf, &len, &off) != 0)
879: break;
880: fmt += n + 1;
1.40 nicm 881: continue;
882: case '#':
883: while (len - off < 2) {
1.50 nicm 884: buf = xreallocarray(buf, 2, len);
1.40 nicm 885: len *= 2;
886: }
887: buf[off++] = '#';
1.1 nicm 888: continue;
889: default:
1.25 nicm 890: s = NULL;
891: if (ch >= 'A' && ch <= 'Z')
892: s = format_upper[ch - 'A'];
893: else if (ch >= 'a' && ch <= 'z')
894: s = format_lower[ch - 'a'];
895: if (s == NULL) {
896: while (len - off < 3) {
1.50 nicm 897: buf = xreallocarray(buf, 2, len);
1.25 nicm 898: len *= 2;
1.1 nicm 899: }
1.25 nicm 900: buf[off++] = '#';
901: buf[off++] = ch;
902: continue;
1.1 nicm 903: }
1.25 nicm 904: n = strlen(s);
905: if (format_replace(ft, s, n, &buf, &len, &off) != 0)
906: break;
1.1 nicm 907: continue;
908: }
909:
910: break;
911: }
912: buf[off] = '\0';
913:
1.79 nicm 914: log_debug("format '%s' -> '%s'", saved, buf);
1.1 nicm 915: return (buf);
916: }
917:
1.57 nicm 918: /* Set defaults for any of arguments that are not NULL. */
919: void
920: format_defaults(struct format_tree *ft, struct client *c, struct session *s,
921: struct winlink *wl, struct window_pane *wp)
922: {
923: if (s == NULL && c != NULL)
924: s = c->session;
925: if (wl == NULL && s != NULL)
926: wl = s->curw;
927: if (wp == NULL && wl != NULL)
928: wp = wl->window->active;
929:
930: if (c != NULL)
931: format_defaults_client(ft, c);
932: if (s != NULL)
933: format_defaults_session(ft, s);
934: if (s != NULL && wl != NULL)
935: format_defaults_winlink(ft, s, wl);
936: if (wp != NULL)
937: format_defaults_pane(ft, wp);
938: }
939:
1.1 nicm 940: /* Set default format keys for a session. */
941: void
1.57 nicm 942: format_defaults_session(struct format_tree *ft, struct session *s)
1.1 nicm 943: {
944: struct session_group *sg;
945:
1.54 nicm 946: ft->s = s;
947:
1.1 nicm 948: format_add(ft, "session_name", "%s", s->name);
949: format_add(ft, "session_windows", "%u", winlink_count(&s->windows));
950: format_add(ft, "session_width", "%u", s->sx);
951: format_add(ft, "session_height", "%u", s->sy);
1.23 nicm 952: format_add(ft, "session_id", "$%u", s->id);
1.1 nicm 953:
954: sg = session_group_find(s);
955: format_add(ft, "session_grouped", "%d", sg != NULL);
956: if (sg != NULL)
957: format_add(ft, "session_group", "%u", session_group_index(sg));
958:
1.87 nicm 959: format_add_tv(ft, "session_created", &s->creation_time);
960: format_add_tv(ft, "session_last_attached", &s->last_attached_time);
961: format_add_tv(ft, "session_activity", &s->activity_time);
1.1 nicm 962:
1.41 nicm 963: format_add(ft, "session_attached", "%u", s->attached);
1.59 nicm 964: format_add(ft, "session_many_attached", "%d", s->attached > 1);
1.66 nicm 965:
1.80 nicm 966: format_add_cb(ft, "session_alerts", format_cb_session_alerts);
1.3 nicm 967: }
968:
969: /* Set default format keys for a client. */
970: void
1.57 nicm 971: format_defaults_client(struct format_tree *ft, struct client *c)
1.3 nicm 972: {
1.60 nicm 973: struct session *s;
1.3 nicm 974:
1.54 nicm 975: if (ft->s == NULL)
976: ft->s = c->session;
977:
1.72 nicm 978: format_add(ft, "client_pid", "%ld", (long) c->pid);
1.6 nicm 979: format_add(ft, "client_height", "%u", c->tty.sy);
980: format_add(ft, "client_width", "%u", c->tty.sx);
1.27 nicm 981: if (c->tty.path != NULL)
982: format_add(ft, "client_tty", "%s", c->tty.path);
983: if (c->tty.termname != NULL)
984: format_add(ft, "client_termname", "%s", c->tty.termname);
1.75 nicm 985: format_add(ft, "client_control_mode", "%d",
986: !!(c->flags & CLIENT_CONTROL));
1.3 nicm 987:
1.87 nicm 988: format_add_tv(ft, "client_created", &c->creation_time);
989: format_add_tv(ft, "client_activity", &c->activity_time);
1.14 nicm 990:
1.61 nicm 991: if (strcmp(c->keytable->name, "root") == 0)
992: format_add(ft, "client_prefix", "%d", 0);
993: else
994: format_add(ft, "client_prefix", "%d", 1);
995: format_add(ft, "client_key_table", "%s", c->keytable->name);
1.3 nicm 996:
997: if (c->tty.flags & TTY_UTF8)
998: format_add(ft, "client_utf8", "%d", 1);
999: else
1000: format_add(ft, "client_utf8", "%d", 0);
1001:
1002: if (c->flags & CLIENT_READONLY)
1003: format_add(ft, "client_readonly", "%d", 1);
1004: else
1005: format_add(ft, "client_readonly", "%d", 0);
1.15 nicm 1006:
1007: s = c->session;
1008: if (s != NULL)
1009: format_add(ft, "client_session", "%s", s->name);
1010: s = c->last_session;
1011: if (s != NULL && session_alive(s))
1012: format_add(ft, "client_last_session", "%s", s->name);
1.1 nicm 1013: }
1014:
1.32 nicm 1015: /* Set default format keys for a window. */
1016: void
1.57 nicm 1017: format_defaults_window(struct format_tree *ft, struct window *w)
1.32 nicm 1018: {
1.54 nicm 1019: ft->w = w;
1020:
1.87 nicm 1021: format_add_tv(ft, "window_activity", &w->activity_time);
1.32 nicm 1022: format_add(ft, "window_id", "@%u", w->id);
1023: format_add(ft, "window_name", "%s", w->name);
1024: format_add(ft, "window_width", "%u", w->sx);
1025: format_add(ft, "window_height", "%u", w->sy);
1.80 nicm 1026: format_add_cb(ft, "window_layout", format_cb_window_layout);
1.32 nicm 1027: format_add(ft, "window_panes", "%u", window_count_panes(w));
1.59 nicm 1028: format_add(ft, "window_zoomed_flag", "%d",
1.53 nicm 1029: !!(w->flags & WINDOW_ZOOMED));
1.32 nicm 1030: }
1031:
1.1 nicm 1032: /* Set default format keys for a winlink. */
1033: void
1.57 nicm 1034: format_defaults_winlink(struct format_tree *ft, struct session *s,
1035: struct winlink *wl)
1.1 nicm 1036: {
1037: struct window *w = wl->window;
1.32 nicm 1038: char *flags;
1.1 nicm 1039:
1.54 nicm 1040: if (ft->w == NULL)
1041: ft->w = wl->window;
1042:
1.1 nicm 1043: flags = window_printable_flags(s, wl);
1044:
1.57 nicm 1045: format_defaults_window(ft, w);
1.32 nicm 1046:
1.1 nicm 1047: format_add(ft, "window_index", "%d", wl->idx);
1048: format_add(ft, "window_flags", "%s", flags);
1049: format_add(ft, "window_active", "%d", wl == s->curw);
1.29 nicm 1050:
1.59 nicm 1051: format_add(ft, "window_bell_flag", "%d",
1.29 nicm 1052: !!(wl->flags & WINLINK_BELL));
1.59 nicm 1053: format_add(ft, "window_activity_flag", "%d",
1.29 nicm 1054: !!(wl->flags & WINLINK_ACTIVITY));
1.59 nicm 1055: format_add(ft, "window_silence_flag", "%d",
1.29 nicm 1056: !!(wl->flags & WINLINK_SILENCE));
1.59 nicm 1057: format_add(ft, "window_last_flag", "%d",
1.49 nicm 1058: !!(wl == TAILQ_FIRST(&s->lastw)));
1.64 nicm 1059: format_add(ft, "window_linked", "%d", session_is_linked(s, wl->window));
1.32 nicm 1060:
1.9 nicm 1061: free(flags);
1.1 nicm 1062: }
1063:
1064: /* Set default format keys for a window pane. */
1065: void
1.57 nicm 1066: format_defaults_pane(struct format_tree *ft, struct window_pane *wp)
1.1 nicm 1067: {
1.80 nicm 1068: struct grid *gd = wp->base.grid;
1069: u_int idx;
1.85 nicm 1070: int status, scroll_position;
1.54 nicm 1071:
1072: if (ft->w == NULL)
1073: ft->w = wp->window;
1.79 nicm 1074: ft->wp = wp;
1.1 nicm 1075:
1.16 nicm 1076: format_add(ft, "history_size", "%u", gd->hsize);
1077: format_add(ft, "history_limit", "%u", gd->hlimit);
1.80 nicm 1078: format_add_cb(ft, "history_bytes", format_cb_history_bytes);
1.1 nicm 1079:
1.4 nicm 1080: if (window_pane_index(wp, &idx) != 0)
1081: fatalx("index not found");
1.16 nicm 1082: format_add(ft, "pane_index", "%u", idx);
1.4 nicm 1083:
1.1 nicm 1084: format_add(ft, "pane_width", "%u", wp->sx);
1085: format_add(ft, "pane_height", "%u", wp->sy);
1086: format_add(ft, "pane_title", "%s", wp->base.title);
1087: format_add(ft, "pane_id", "%%%u", wp->id);
1088: format_add(ft, "pane_active", "%d", wp == wp->window->active);
1.55 nicm 1089: format_add(ft, "pane_input_off", "%d", !!(wp->flags & PANE_INPUTOFF));
1090:
1091: status = wp->status;
1092: if (wp->fd == -1 && WIFEXITED(status))
1093: format_add(ft, "pane_dead_status", "%d", WEXITSTATUS(status));
1.1 nicm 1094: format_add(ft, "pane_dead", "%d", wp->fd == -1);
1.47 nicm 1095:
1096: if (window_pane_visible(wp)) {
1097: format_add(ft, "pane_left", "%u", wp->xoff);
1098: format_add(ft, "pane_top", "%u", wp->yoff);
1099: format_add(ft, "pane_right", "%u", wp->xoff + wp->sx - 1);
1100: format_add(ft, "pane_bottom", "%u", wp->yoff + wp->sy - 1);
1101: }
1.16 nicm 1102:
1103: format_add(ft, "pane_in_mode", "%d", wp->screen != &wp->base);
1.26 nicm 1104: format_add(ft, "pane_synchronized", "%d",
1.90 nicm 1105: !!options_get_number(wp->window->options, "synchronize-panes"));
1.16 nicm 1106:
1.71 nicm 1107: format_add(ft, "pane_tty", "%s", wp->tty);
1.16 nicm 1108: format_add(ft, "pane_pid", "%ld", (long) wp->pid);
1.79 nicm 1109: format_add_cb(ft, "pane_start_command", format_cb_start_command);
1110: format_add_cb(ft, "pane_current_command", format_cb_current_command);
1.16 nicm 1111:
1.59 nicm 1112: format_add(ft, "cursor_x", "%u", wp->base.cx);
1113: format_add(ft, "cursor_y", "%u", wp->base.cy);
1114: format_add(ft, "scroll_region_upper", "%u", wp->base.rupper);
1115: format_add(ft, "scroll_region_lower", "%u", wp->base.rlower);
1.85 nicm 1116:
1117: scroll_position = window_copy_scroll_position(wp);
1118: if (scroll_position != -1)
1119: format_add(ft, "scroll_position", "%d", scroll_position);
1.16 nicm 1120:
1121: format_add(ft, "alternate_on", "%d", wp->saved_grid ? 1 : 0);
1.59 nicm 1122: format_add(ft, "alternate_saved_x", "%u", wp->saved_cx);
1123: format_add(ft, "alternate_saved_y", "%u", wp->saved_cy);
1.16 nicm 1124:
1125: format_add(ft, "cursor_flag", "%d",
1126: !!(wp->base.mode & MODE_CURSOR));
1127: format_add(ft, "insert_flag", "%d",
1128: !!(wp->base.mode & MODE_INSERT));
1129: format_add(ft, "keypad_cursor_flag", "%d",
1130: !!(wp->base.mode & MODE_KCURSOR));
1131: format_add(ft, "keypad_flag", "%d",
1132: !!(wp->base.mode & MODE_KKEYPAD));
1133: format_add(ft, "wrap_flag", "%d",
1134: !!(wp->base.mode & MODE_WRAP));
1135:
1.62 nicm 1136: format_add(ft, "mouse_any_flag", "%d",
1137: !!(wp->base.mode & (MODE_MOUSE_STANDARD|MODE_MOUSE_BUTTON)));
1.16 nicm 1138: format_add(ft, "mouse_standard_flag", "%d",
1139: !!(wp->base.mode & MODE_MOUSE_STANDARD));
1140: format_add(ft, "mouse_button_flag", "%d",
1141: !!(wp->base.mode & MODE_MOUSE_BUTTON));
1.19 nicm 1142:
1.80 nicm 1143: format_add_cb(ft, "pane_tabs", format_cb_pane_tabs);
1.8 nicm 1144: }
1145:
1.19 nicm 1146: /* Set default format keys for paste buffer. */
1.8 nicm 1147: void
1.94 nicm 1148: format_defaults_paste_buffer(struct format_tree *ft, struct paste_buffer *pb)
1.8 nicm 1149: {
1.81 nicm 1150: size_t bufsize;
1.42 nicm 1151: char *s;
1.8 nicm 1152:
1.81 nicm 1153: paste_buffer_data(pb, &bufsize);
1154: format_add(ft, "buffer_size", "%zu", bufsize);
1155: format_add(ft, "buffer_name", "%s", paste_buffer_name(pb));
1.8 nicm 1156:
1.94 nicm 1157: s = paste_make_sample(pb);
1.42 nicm 1158: format_add(ft, "buffer_sample", "%s", s);
1159: free(s);
1.1 nicm 1160: }