Project

General

Profile

Feature #11999 » 11999.diff

sorah (Sorah Fukumori), 01/18/2016 08:55 AM

View differences:

re.c
return str;
}
static int
match_to_h_iter(const OnigUChar *name, const OnigUChar *name_end,
int back_num, int *back_refs, OnigRegex regex, void *arg) {
struct MEMO *memo = MEMO_CAST(arg);
VALUE hash = memo->v1;
VALUE match = memo->v2;
VALUE key = rb_enc_str_new((const char *)name, name_end-name, regex->enc);
VALUE value = rb_reg_nth_match(back_refs[back_num-1], match);
rb_hash_aset(hash, key, value);
return 0;
}
/*
* call-seq:
* mtch.to_h -> hash
*
* Returns a Hash using named capture.
*
* A key of the hash is a name of the named captures.
* A value of the hash is a string of last capture of corresponding group.
*
* m = /(?<a>.)(?<b>.)/.match("01")
* m.to_h #=> {"a" => "0", "b" => "1"}
*
* m = /(?<a>.)(?<b>.)?/.match("0")
* m.to_h #=> {"a" => "0", "b" => nil}
*
* m = /(?<a>.)(?<a>.)/.match("01")
* m.to_h #=> {"a" => "1"}
*/
static VALUE
match_to_h(VALUE match)
{
VALUE hash;
struct MEMO *memo;
match_check(match);
hash = rb_hash_new();
memo = MEMO_NEW(hash, match, 0);
onig_foreach_name(RREGEXP(RMATCH(match)->regexp)->ptr, match_to_h_iter, (void*)memo);
return hash;
}
/*
* call-seq:
......
rb_define_method(rb_cMatch, "pre_match", rb_reg_match_pre, 0);
rb_define_method(rb_cMatch, "post_match", rb_reg_match_post, 0);
rb_define_method(rb_cMatch, "to_s", match_to_s, 0);
rb_define_method(rb_cMatch, "to_h", match_to_h, 0);
rb_define_method(rb_cMatch, "inspect", match_inspect, 0);
rb_define_method(rb_cMatch, "string", match_string, 0);
rb_define_method(rb_cMatch, "hash", match_hash, 0);
test/ruby/test_regexp.rb
assert_raise(IndexError, bug9903) {m[key.dup.force_encoding(Encoding::Shift_JIS)]}
end
def test_named_capture_to_h
assert_equal({'a' => '1', 'b' => '2', 'c' => nil}, /^(?<a>.)(?<b>.)(?<c>.)?/.match('12').to_h)
assert_equal({'a' => '1', 'b' => '2', 'c' => '3'}, /^(?<a>.)(?<b>.)(?<c>.)?/.match('123').to_h)
assert_equal({'a' => '1', 'b' => '2', 'c' => ''}, /^(?<a>.)(?<b>.)(?<c>.?)/.match('12').to_h)
assert_equal({'a' => '1', 'b' => '2'}, /^(.)(?<a>.)(?<b>.)/.match('012').to_h)
assert_equal({'a' => '2'}, /^(?<a>.)(?<a>.)/.match('12').to_h)
end
def test_assign_named_capture
assert_equal("a", eval('/(?<foo>.)/ =~ "a"; foo'))
assert_equal("a", eval('foo = 1; /(?<foo>.)/ =~ "a"; foo'))
(1-1/2)