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)); } } }