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