mirror of
https://github.com/tmux/tmux.git
synced 2025-09-01 20:57:00 +00:00
Add regular expression support for the format search, match and
substitute modifiers.
This commit is contained in:
108
format.c
108
format.c
@ -23,6 +23,7 @@
|
||||
#include <errno.h>
|
||||
#include <fnmatch.h>
|
||||
#include <libgen.h>
|
||||
#include <regex.h>
|
||||
#include <stdarg.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
@ -1247,7 +1248,7 @@ format_build_modifiers(struct format_tree *ft, const char **s, u_int *count)
|
||||
cp++;
|
||||
|
||||
/* Check single character modifiers with no arguments. */
|
||||
if (strchr("lmCbdtqETSWP<>", cp[0]) != NULL &&
|
||||
if (strchr("lbdtqETSWP<>", cp[0]) != NULL &&
|
||||
format_is_end(cp[1])) {
|
||||
format_add_modifier(&list, count, cp, 1, NULL, 0);
|
||||
cp++;
|
||||
@ -1268,7 +1269,7 @@ format_build_modifiers(struct format_tree *ft, const char **s, u_int *count)
|
||||
}
|
||||
|
||||
/* Now try single character with arguments. */
|
||||
if (strchr("s=", cp[0]) == NULL)
|
||||
if (strchr("mCs=", cp[0]) == NULL)
|
||||
break;
|
||||
c = cp[0];
|
||||
|
||||
@ -1329,39 +1330,67 @@ format_build_modifiers(struct format_tree *ft, const char **s, u_int *count)
|
||||
return list;
|
||||
}
|
||||
|
||||
/* Match against an fnmatch(3) pattern or regular expression. */
|
||||
static char *
|
||||
format_match(struct format_modifier *fm, const char *pattern, const char *text)
|
||||
{
|
||||
const char *s = "";
|
||||
regex_t r;
|
||||
int flags = 0;
|
||||
|
||||
if (fm->argc >= 1)
|
||||
s = fm->argv[0];
|
||||
if (strchr(s, 'r') == NULL) {
|
||||
if (strchr(s, 'i') != NULL)
|
||||
flags |= FNM_CASEFOLD;
|
||||
if (fnmatch(pattern, text, flags) != 0)
|
||||
return (xstrdup("0"));
|
||||
} else {
|
||||
flags = REG_EXTENDED|REG_NOSUB;
|
||||
if (strchr(s, 'i') != NULL)
|
||||
flags |= REG_ICASE;
|
||||
if (regcomp(&r, pattern, flags) != 0)
|
||||
return (xstrdup("0"));
|
||||
if (regexec(&r, text, 0, NULL, 0) != 0) {
|
||||
regfree(&r);
|
||||
return (xstrdup("0"));
|
||||
}
|
||||
regfree(&r);
|
||||
}
|
||||
return (xstrdup("1"));
|
||||
}
|
||||
|
||||
/* Perform substitution in string. */
|
||||
static char *
|
||||
format_substitute(const char *source, const char *from, const char *to)
|
||||
format_sub(struct format_modifier *fm, const char *text, const char *pattern,
|
||||
const char *with)
|
||||
{
|
||||
char *copy, *new;
|
||||
const char *cp;
|
||||
size_t fromlen, tolen, newlen, used;
|
||||
char *value;
|
||||
int flags = REG_EXTENDED;
|
||||
|
||||
fromlen = strlen(from);
|
||||
tolen = strlen(to);
|
||||
if (fm->argc >= 3 && strchr(fm->argv[2], 'i') != NULL)
|
||||
flags |= REG_ICASE;
|
||||
value = regsub(pattern, with, text, flags);
|
||||
if (value == NULL)
|
||||
return (xstrdup(text));
|
||||
return (value);
|
||||
}
|
||||
|
||||
newlen = strlen(source) + 1;
|
||||
copy = new = xmalloc(newlen);
|
||||
/* Search inside pane. */
|
||||
static char *
|
||||
format_search(struct format_modifier *fm, struct window_pane *wp, const char *s)
|
||||
{
|
||||
int ignore = 0, regex = 0;
|
||||
char *value;
|
||||
|
||||
for (cp = source; *cp != '\0'; /* nothing */) {
|
||||
if (strncmp(cp, from, fromlen) != 0) {
|
||||
*new++ = *cp++;
|
||||
continue;
|
||||
}
|
||||
used = new - copy;
|
||||
|
||||
newlen += tolen;
|
||||
copy = xrealloc(copy, newlen);
|
||||
|
||||
new = copy + used;
|
||||
memcpy(new, to, tolen);
|
||||
|
||||
new += tolen;
|
||||
cp += fromlen;
|
||||
if (fm->argc >= 1) {
|
||||
if (strchr(fm->argv[0], 'i') != NULL)
|
||||
ignore = 1;
|
||||
if (strchr(fm->argv[0], 'r') != NULL)
|
||||
regex = 1;
|
||||
}
|
||||
|
||||
*new = '\0';
|
||||
return (copy);
|
||||
xasprintf(&value, "%u", window_pane_search(wp, s, regex, ignore));
|
||||
return (value);
|
||||
}
|
||||
|
||||
/* Loop over sessions. */
|
||||
@ -1506,11 +1535,10 @@ format_replace(struct format_tree *ft, const char *key, size_t keylen,
|
||||
char *copy0, *condition, *found, *new;
|
||||
char *value, *left, *right;
|
||||
size_t valuelen;
|
||||
int modifiers = 0, limit = 0;
|
||||
int modifiers = 0, limit = 0, j;
|
||||
struct format_modifier *list, *fm, *cmp = NULL, *search = NULL;
|
||||
struct format_modifier *sub = NULL;
|
||||
u_int i, count;
|
||||
int j;
|
||||
|
||||
/* Make a copy of the key. */
|
||||
copy = copy0 = xstrndup(key, keylen);
|
||||
@ -1537,18 +1565,18 @@ format_replace(struct format_tree *ft, const char *key, size_t keylen,
|
||||
search = fm;
|
||||
break;
|
||||
case 's':
|
||||
if (fm->argc != 2)
|
||||
if (fm->argc < 2)
|
||||
break;
|
||||
sub = fm;
|
||||
break;
|
||||
case '=':
|
||||
if (fm->argc != 1 && fm->argc != 2)
|
||||
if (fm->argc < 1)
|
||||
break;
|
||||
limit = strtonum(fm->argv[0], INT_MIN, INT_MAX,
|
||||
&errptr);
|
||||
if (errptr != NULL)
|
||||
limit = 0;
|
||||
if (fm->argc == 2 && fm->argv[1] != NULL)
|
||||
if (fm->argc >= 2 && fm->argv[1] != NULL)
|
||||
marker = fm->argv[1];
|
||||
break;
|
||||
case 'l':
|
||||
@ -1619,7 +1647,7 @@ format_replace(struct format_tree *ft, const char *key, size_t keylen,
|
||||
value = xstrdup("0");
|
||||
} else {
|
||||
format_log(ft, "search '%s' pane %%%u", copy, wp->id);
|
||||
xasprintf(&value, "%u", window_pane_search(wp, copy));
|
||||
value = format_search(fm, wp, copy);
|
||||
}
|
||||
} else if (cmp != NULL) {
|
||||
/* Comparison of left and right. */
|
||||
@ -1671,12 +1699,8 @@ format_replace(struct format_tree *ft, const char *key, size_t keylen,
|
||||
value = xstrdup("1");
|
||||
else
|
||||
value = xstrdup("0");
|
||||
} else if (strcmp(cmp->modifier, "m") == 0) {
|
||||
if (fnmatch(left, right, 0) == 0)
|
||||
value = xstrdup("1");
|
||||
else
|
||||
value = xstrdup("0");
|
||||
}
|
||||
} else if (strcmp(cmp->modifier, "m") == 0)
|
||||
value = format_match(fm, left, right);
|
||||
|
||||
free(right);
|
||||
free(left);
|
||||
@ -1754,8 +1778,8 @@ done:
|
||||
|
||||
/* Perform substitution if any. */
|
||||
if (sub != NULL) {
|
||||
new = format_substitute(value, sub->argv[0], sub->argv[1]);
|
||||
format_log(ft, "substituted '%s' to '%s: %s", sub->argv[0],
|
||||
new = format_sub(sub, value, sub->argv[0], sub->argv[1]);
|
||||
format_log(ft, "substituted '%s' to '%s': %s", sub->argv[0],
|
||||
sub->argv[1], new);
|
||||
free(value);
|
||||
value = new;
|
||||
|
Reference in New Issue
Block a user