summaryrefslogtreecommitdiff
path: root/xs_regex.h
diff options
context:
space:
mode:
Diffstat (limited to 'xs_regex.h')
-rw-r--r--xs_regex.h49
1 files changed, 49 insertions, 0 deletions
diff --git a/xs_regex.h b/xs_regex.h
index b86949e..6fb6cca 100644
--- a/xs_regex.h
+++ b/xs_regex.h
@@ -8,6 +8,8 @@ xs_list *xs_regex_split_n(const char *str, const char *rx, int count);
#define xs_regex_split(str, rx) xs_regex_split_n(str, rx, XS_ALL)
xs_list *xs_regex_match_n(const char *str, const char *rx, int count);
#define xs_regex_match(str, rx) xs_regex_match_n(str, rx, XS_ALL)
+xs_list *xs_regex_replace_n(const char *str, const char *rx, const char *rep, int count);
+#define xs_regex_replace(str, rx, rep) xs_regex_replace_n(str, rx, rep, XS_ALL)
#ifdef XS_IMPLEMENTATION
@@ -75,6 +77,53 @@ xs_list *xs_regex_match_n(const char *str, const char *rx, int count)
return list;
}
+
+xs_list *xs_regex_replace_n(const char *str, const char *rx, const char *rep, int count)
+/* replaces all matches with the rep string. If it contains unescaped &,
+ they are replaced with the match */
+{
+ xs_str *s = xs_str_new(NULL);
+ xs *split = xs_regex_split_n(str, rx, count);
+ xs_list *p;
+ xs_val *v;
+ int n = 0;
+ int pholder = !!strchr(rep, '&');
+
+ p = split;
+ while (xs_list_iter(&p, &v)) {
+ if (n & 0x1) {
+ if (pholder) {
+ /* rep has a placeholder; process char by char */
+ const char *p = rep;
+
+ while (*p) {
+ if (*p == '&')
+ s = xs_str_cat(s, v);
+ else {
+ if (*p == '\\')
+ p++;
+
+ if (!*p)
+ break;
+
+ s = xs_append_m(s, p, 1);
+ }
+
+ p++;
+ }
+ }
+ else
+ s = xs_str_cat(s, rep);
+ }
+ else
+ s = xs_str_cat(s, v);
+
+ n++;
+ }
+
+ return s;
+}
+
#endif /* XS_IMPLEMENTATION */
#endif /* XS_REGEX_H */