findall.c

Go to the documentation of this file.

/*
 * This file is part of the Fun programming language.
 * https://fun-lang.xyz/
 *
 * Copyright 2025 Johannes Findeisen <you@hanez.org>
 * Licensed under the terms of the Apache-2.0 license.
 * https://opensource.org/license/apache-2-0
 */



/* PCRE2_FINDALL */
case OP_PCRE2_FINDALL: {
#ifdef FUN_WITH_PCRE2
 Value vflags = pop_value(vm);
 Value vtext = pop_value(vm);
 Value vpat = pop_value(vm);
 int flags = 0;
 if (vflags.type == VAL_INT || vflags.type == VAL_BOOL) flags = (int)vflags.i;
 char *pattern = value_to_string_alloc(&vpat);
 char *subject = value_to_string_alloc(&vtext);
 free_value(vflags);
 free_value(vtext);
 free_value(vpat);
 if (!pattern || !subject) {
 if (pattern) free(pattern);
 if (subject) free(subject);
 push_value(vm, make_array_from_values(NULL, 0));
 break;
 }
#ifndef PCRE2_CODE_UNIT_WIDTH
#define PCRE2_CODE_UNIT_WIDTH 8
#endif
#include <pcre2.h>
 int errorcode;
 PCRE2_SIZE erroff;
 uint32_t opt = 0;
 if (flags & 1) opt |= PCRE2_CASELESS; /* I */
 if (flags & 2) opt |= PCRE2_MULTILINE; /* M */
 if (flags & 4) opt |= PCRE2_DOTALL; /* S */
 if (flags & 8) opt |= PCRE2_UTF; /* U */
 if (flags & 16) opt |= PCRE2_EXTENDED; /* X */
 pcre2_code *re = pcre2_compile((PCRE2_SPTR)pattern, PCRE2_ZERO_TERMINATED, opt, &errorcode, &erroff, NULL);
 if (!re) {
 free(pattern);
 free(subject);
 push_value(vm, make_array_from_values(NULL, 0));
 break;
 }
 pcre2_match_data *mdata = pcre2_match_data_create_from_pattern(re, NULL);
 Value out = make_array_from_values(NULL, 0);
 size_t subj_len = strlen(subject);
 size_t start_off = 0;
 int gcount = 0;
 while (1) {
 int rc = pcre2_match(re, (PCRE2_SPTR)subject, (PCRE2_SIZE)subj_len, start_off, 0, mdata, NULL);
 if (rc <= 0) break;
 PCRE2_SIZE *ov = pcre2_get_ovector_pointer(mdata);
 int s0 = (int)ov[0];
 int e0 = (int)ov[1];
 /* result map for this match */
 Value res = make_map_empty();
 char *full = string_substr(subject, s0, e0 - s0);
 (void)map_set(&res, "full", make_string(full ? full : ""));
 if (full) free(full);
 (void)map_set(&res, "start", make_int(s0));
 (void)map_set(&res, "end", make_int(e0));
 Value groups = make_array_from_values(NULL, 0);
 for (int i = 1; i < rc; ++i) {
 int s = (int)ov[2 * i];
 int e = (int)ov[2 * i + 1];
 char *gstr = (s >= 0 && e >= s) ? string_substr(subject, s, e - s) : NULL;
 Value gv = make_string(gstr ? gstr : "");
 if (gstr) free(gstr);
 (void)array_push(&groups, gv);
 }
 (void)map_set(&res, "groups", groups);
 (void)array_push(&out, res);
 /* advance start offset; guard against empty match */
 if (e0 == s0) {
 if ((size_t)e0 < subj_len) {
 start_off = e0 + 1;
 } else {
 break;
 }
 } else {
 start_off = e0;
 }
 gcount = rc;
 }
 pcre2_match_data_free(mdata);
 pcre2_code_free(re);
 free(pattern);
 free(subject);
 push_value(vm, out);
#else
 Value a = pop_value(vm);
 free_value(a);
 Value b = pop_value(vm);
 free_value(b);
 Value c = pop_value(vm);
 free_value(c);
 push_value(vm, make_array_from_values(NULL, 0));
#endif
 break;
}