From f2cb6288bc6f1d5e693841734ce5eb04ff41c2a9 Mon Sep 17 00:00:00 2001 From: Nobuyoshi Nakada Date: Thu, 16 Sep 2021 19:50:29 +0900 Subject: [Feature #18172] Add MatchData#match_length The method to return the length of the matched substring corresponding to the given argument. --- re.c | 37 +++++++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) (limited to 're.c') diff --git a/re.c b/re.c index 5d5f6fc65c..8e2df1f82c 100644 --- a/re.c +++ b/re.c @@ -1335,6 +1335,42 @@ match_nth(VALUE match, VALUE n) return rb_str_subseq(RMATCH(match)->str, start, end - start); } +/* + * call-seq: + * mtch.match_length(n) -> array + * + * Returns the length of the captured substring corresponding to the argument. + * n can be a string or symbol to reference a named capture. + * + * m = /(.)(.)(\d+)(\d)(\w)?/.match("THX1138.") + * m.match_length(0) #=> 6 + * m.match_length(4) #=> 1 + * m.match_length(5) #=> nil + * + * m = /(?.)(.)(?.+)/.match("hoge") + * m.match_length(:foo) #=> 1 + * m.match_length(:bar) #=> 2 + * + */ + +static VALUE +match_nth_length(VALUE match, VALUE n) +{ + int i = match_backref_number(match, n); + struct re_registers *regs = RMATCH_REGS(match); + + match_check(match); + backref_number_check(regs, i); + + if (BEG(i) < 0) + return Qnil; + + update_char_offset(match); + const struct rmatch_offset *const ofs = + &RMATCH(match)->rmatch->char_offset[i]; + return LONG2NUM(ofs->end - ofs->beg); +} + #define MATCH_BUSY FL_USER2 void @@ -4136,6 +4172,7 @@ Init_Regexp(void) rb_define_method(rb_cMatch, "begin", match_begin, 1); rb_define_method(rb_cMatch, "end", match_end, 1); rb_define_method(rb_cMatch, "match", match_nth, 1); + rb_define_method(rb_cMatch, "match_length", match_nth_length, 1); rb_define_method(rb_cMatch, "to_a", match_to_a, 0); rb_define_method(rb_cMatch, "[]", match_aref, -1); rb_define_method(rb_cMatch, "captures", match_captures, 0); -- cgit v1.2.3