V7_PRIVATE enum v7_err Str_split(struct v7_c_func_arg *cfa) {
#define v7 (cfa->v7) /* Needed for TRY() macro below */
struct v7_val *arg = cfa->args[0], *arr = v7_push_new_object(v7);
struct Resub sub, sub1;
int limit = 1000000, elem = 0, i, len;
unsigned long shift = 0;
v7_set_class(arr, V7_CLASS_ARRAY);
if (cfa->num_args > 0) {
if (cfa->num_args > 1 && cfa->args[1]->type == V7_TYPE_NUM)
limit = cfa->args[1]->v.num;
TRY(check_str_re_conv(v7, &arg, 1));
TRY(regex_check_prog(arg));
for (; elem < limit && shift < cfa->this_obj->v.str.len; elem++) {
if (re_exec(arg->v.str.prog, arg->fl.fl, cfa->this_obj->v.str.buf + shift,
&sub))
break;
v7_append(v7, arr,
v7_mkv(v7, V7_TYPE_STR, cfa->this_obj->v.str.buf + shift,
sub.sub[0].start - cfa->this_obj->v.str.buf - shift, 1));
for (i = 1; i < sub.subexpr_num; i++)
v7_append(v7, arr, v7_mkv(v7, V7_TYPE_STR, sub.sub[i].start,
sub.sub[i].end - sub.sub[i].start, 1));
shift = sub.sub[0].end - cfa->this_obj->v.str.buf;
sub1 = sub;
}
}
len = cfa->this_obj->v.str.len - shift;
if (elem < limit && len > 0)
v7_append(v7, arr, v7_mkv(v7, V7_TYPE_STR, cfa->this_obj->v.str.buf + shift,
len, 1));
return V7_OK;
#undef v7
}
V7_PRIVATE enum v7_err Str_match(struct v7_c_func_arg *cfa) {
#define v7 (cfa->v7) /* Needed for TRY() macro below */
struct v7_val *arg = cfa->args[0];
struct Resub sub;
struct v7_val *arr = NULL;
unsigned long shift = 0;
if (cfa->num_args > 0) {
TRY(check_str_re_conv(v7, &arg, 1));
TRY(regex_check_prog(arg));
do {
if (!re_exec(arg->v.str.prog, arg->fl.fl,
cfa->this_obj->v.str.buf + shift, &sub)) {
if (NULL == arr) {
arr = v7_push_new_object(v7);
v7_set_class(arr, V7_CLASS_ARRAY);
}
shift = sub.sub[0].end - cfa->this_obj->v.str.buf;
v7_append(v7, arr, v7_mkv(v7, V7_TYPE_STR, sub.sub[0].start,
sub.sub[0].end - sub.sub[0].start, 1));
}
} while (arg->fl.fl.re_g && shift < cfa->this_obj->v.str.len);
}
if (0 == shift) TRY(v7_make_and_push(v7, V7_TYPE_NULL));
return V7_OK;
#undef v7
}
static void Str_match(struct v7_c_func_arg *cfa) {
struct slre_cap caps[100];
const struct v7_string *s = &cfa->this_obj->v.str;
int i, n;
cfa->result->type = V7_TYPE_NULL;
memset(caps, 0, sizeof(caps));
if (cfa->num_args == 1 &&
v7_is_class(cfa->args[0], V7_CLASS_REGEXP) &&
(n = slre_match(cfa->args[0]->v.regex, s->buf, s->len,
caps, ARRAY_SIZE(caps) - 1, 0)) > 0) {
v7_set_class(cfa->result, V7_CLASS_ARRAY);
v7_append(cfa->v7, cfa->result,
v7_mkv(cfa->v7, V7_TYPE_STR, s->buf, (long) n, 1));
for (i = 0; i < (int) ARRAY_SIZE(caps); i++) {
if (caps[i].len == 0) break;
v7_append(cfa->v7, cfa->result,
v7_mkv(cfa->v7, V7_TYPE_STR, caps[i].ptr, (long) caps[i].len, 1));
}
}
}
static void Str_split(struct v7_c_func_arg *cfa) {
const struct v7_string *s = &cfa->this_obj->v.str;
const char *p1, *p2, *e = s->buf + s->len;
int limit = cfa->num_args == 2 && cfa->args[1]->type == V7_TYPE_NUM ?
cfa->args[1]->v.num : -1;
int num_elems = 0;
v7_set_class(cfa->result, V7_CLASS_ARRAY);
if (cfa->num_args == 0) {
v7_append(cfa->v7, cfa->result,
v7_mkv(cfa->v7, V7_TYPE_STR, s->buf, s->len, 1));
} else if (cfa->args[0]->type == V7_TYPE_STR) {
const struct v7_string *sep = &cfa->args[0]->v.str;
if (sep->len == 0) {
// Separator is empty. Split string by characters.
for (p1 = s->buf; p1 < e; p1++) {
if (limit >= 0 && limit <= num_elems) return;
v7_append(cfa->v7, cfa->result, v7_mkv(cfa->v7, V7_TYPE_STR, p1, 1, 1));
num_elems++;
}
} else {
p1 = s->buf;
while ((p2 = memstr(p1, e - p1, sep->buf, sep->len)) != NULL) {
if (limit >= 0 && limit <= num_elems) return;
v7_append(cfa->v7, cfa->result,
v7_mkv(cfa->v7, V7_TYPE_STR, p1, p2 - p1, 1));
p1 = p2 + sep->len;
num_elems++;
}
if (limit < 0 || limit > num_elems) {
v7_append(cfa->v7, cfa->result,
v7_mkv(cfa->v7, V7_TYPE_STR, p1, e - p1, 1));
}
}
} else if (instanceof(cfa->args[0], &s_constructors[V7_CLASS_REGEXP])) {
char regex[200];
struct slre_cap caps[20];
int n = 0;
snprintf(regex, sizeof(regex), "(%s)", cfa->args[0]->v.regex);
p1 = s->buf;
while ((n = slre_match(regex, p1, e - p1, caps, ARRAY_SIZE(caps), 0)) > 0) {
if (limit >= 0 && limit <= num_elems) return;
v7_append(cfa->v7, cfa->result,
v7_mkv(cfa->v7, V7_TYPE_STR, p1, caps[0].ptr - p1, 1));
p1 += n;
num_elems++;
}
if (limit < 0 || limit > num_elems) {
v7_append(cfa->v7, cfa->result,
v7_mkv(cfa->v7, V7_TYPE_STR, p1, e - p1, 1));
}
}
}
