aboutsummaryrefslogtreecommitdiff
path: root/format.c
diff options
context:
space:
mode:
Diffstat (limited to 'format.c')
-rw-r--r--format.c56
1 files changed, 41 insertions, 15 deletions
diff --git a/format.c b/format.c
index 251af7df..e09ce423 100644
--- a/format.c
+++ b/format.c
@@ -1315,7 +1315,7 @@ format_build_modifiers(struct format_tree *ft, const char **s, u_int *count)
}
/* Now try single character with arguments. */
- if (strchr("mCs=", cp[0]) == NULL)
+ if (strchr("mCs=p", cp[0]) == NULL)
break;
c = cp[0];
@@ -1576,15 +1576,15 @@ static int
format_replace(struct format_tree *ft, const char *key, size_t keylen,
char **buf, size_t *len, size_t *off)
{
- struct window_pane *wp = ft->wp;
- const char *errptr, *copy, *cp, *marker = NULL;
- char *copy0, *condition, *found, *new;
- char *value, *left, *right;
- size_t valuelen;
- int modifiers = 0, limit = 0, j;
- struct format_modifier *list, *fm, *cmp = NULL, *search = NULL;
- struct format_modifier *sub = NULL;
- u_int i, count;
+ struct window_pane *wp = ft->wp;
+ const char *errptr, *copy, *cp, *marker = NULL;
+ char *copy0, *condition, *found, *new;
+ char *value, *left, *right;
+ size_t valuelen;
+ int modifiers = 0, limit = 0, width = 0, j;
+ struct format_modifier *list, *fm, *cmp = NULL, *search = NULL;
+ struct format_modifier **sub = NULL;
+ u_int i, count, nsub = 0;
/* Make a copy of the key. */
copy = copy0 = xstrndup(key, keylen);
@@ -1613,7 +1613,9 @@ format_replace(struct format_tree *ft, const char *key, size_t keylen,
case 's':
if (fm->argc < 2)
break;
- sub = fm;
+ sub = xreallocarray (sub, nsub + 1,
+ sizeof *sub);
+ sub[nsub++] = fm;
break;
case '=':
if (fm->argc < 1)
@@ -1625,6 +1627,14 @@ format_replace(struct format_tree *ft, const char *key, size_t keylen,
if (fm->argc >= 2 && fm->argv[1] != NULL)
marker = fm->argv[1];
break;
+ case 'p':
+ if (fm->argc < 1)
+ break;
+ width = strtonum(fm->argv[0], INT_MIN, INT_MAX,
+ &errptr);
+ if (errptr != NULL)
+ width = 0;
+ break;
case 'l':
modifiers |= FORMAT_LITERAL;
break;
@@ -1825,10 +1835,10 @@ done:
}
/* Perform substitution if any. */
- if (sub != NULL) {
- left = format_expand(ft, sub->argv[0]);
- right = format_expand(ft, sub->argv[1]);
- new = format_sub(sub, value, left, right);
+ for (i = 0; i < nsub; i++) {
+ left = format_expand(ft, sub[i]->argv[0]);
+ right = format_expand(ft, sub[i]->argv[1]);
+ new = format_sub(sub[i], value, left, right);
format_log(ft, "substitute '%s' to '%s': %s", left, right, new);
free(value);
value = new;
@@ -1859,6 +1869,19 @@ done:
format_log(ft, "applied length limit %d: %s", limit, value);
}
+ /* Pad the value if needed. */
+ if (width > 0) {
+ new = utf8_padcstr(value, width);
+ free(value);
+ value = new;
+ format_log(ft, "applied padding width %d: %s", width, value);
+ } else if (width < 0) {
+ new = utf8_rpadcstr(value, -width);
+ free(value);
+ value = new;
+ format_log(ft, "applied padding width %d: %s", width, value);
+ }
+
/* Expand the buffer and copy in the value. */
valuelen = strlen(value);
while (*len - *off < valuelen + 1) {
@@ -1871,12 +1894,15 @@ done:
format_log(ft, "replaced '%s' with '%s'", copy0, value);
free(value);
+ free(sub);
format_free_modifiers(list, count);
free(copy0);
return (0);
fail:
format_log(ft, "failed %s", copy0);
+
+ free(sub);
format_free_modifiers(list, count);
free(copy0);
return (-1);