static pic_value pic_regexp_regexp_match(pic_state *pic) { pic_value reg; const char *input; regmatch_t match[100]; pic_value matches, positions; pic_str *str; int i, offset; pic_get_args(pic, "oz", ®, &input); pic_assert_type(pic, reg, regexp); matches = pic_nil_value(); positions = pic_nil_value(); if (strchr(pic_regexp_data_ptr(reg)->flags, 'g') != NULL) { /* global search */ offset = 0; while (regexec(&pic_regexp_data_ptr(reg)->reg, input, 1, match, 0) != REG_NOMATCH) { pic_push(pic, pic_obj_value(pic_str_new(pic, input, match[0].rm_eo - match[0].rm_so)), matches); pic_push(pic, pic_int_value(offset), positions); offset += match[0].rm_eo; input += match[0].rm_eo; } } else { /* local search */ if (regexec(&pic_regexp_data_ptr(reg)->reg, input, 100, match, 0) == 0) { for (i = 0; i < 100; ++i) { if (match[i].rm_so == -1) { break; } str = pic_str_new(pic, input + match[i].rm_so, match[i].rm_eo - match[i].rm_so); pic_push(pic, pic_obj_value(str), matches); pic_push(pic, pic_int_value(match[i].rm_so), positions); } } } if (pic_nil_p(matches)) { matches = pic_false_value(); positions = pic_false_value(); } else { matches = pic_reverse(pic, matches); positions = pic_reverse(pic, positions); } return pic_values2(pic, matches, positions); }
pic_value pic_str_new_cstr(pic_state *pic, const char *cstr) { size_t len; len = strlen(cstr); return pic_str_new(pic, cstr, len); }
static pic_value pic_regexp_regexp_replace(pic_state *pic) { pic_value reg; const char *input; regmatch_t match; pic_str *txt, *output = pic_str_new(pic, NULL, 0); pic_get_args(pic, "ozs", ®, &input, &txt); pic_assert_type(pic, reg, regexp); while (regexec(&pic_regexp_data_ptr(reg)->reg, input, 1, &match, 0) != REG_NOMATCH) { output = pic_strcat(pic, output, pic_str_new(pic, input, match.rm_so)); output = pic_strcat(pic, output, txt); input += match.rm_eo; } output = pic_strcat(pic, output, pic_str_new(pic, input, strlen(input))); return pic_obj_value(output); }
static pic_value pic_regexp_regexp_split(pic_state *pic) { pic_value reg; const char *input; regmatch_t match; pic_value output = pic_nil_value(); pic_get_args(pic, "oz", ®, &input); pic_assert_type(pic, reg, regexp); while (regexec(&pic_regexp_data_ptr(reg)->reg, input, 1, &match, 0) != REG_NOMATCH) { pic_push(pic, pic_obj_value(pic_str_new(pic, input, match.rm_so)), output); input += match.rm_eo; } pic_push(pic, pic_obj_value(pic_str_new_cstr(pic, input)), output); return pic_reverse(pic, output); }