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