Ruby » Ruby trunk

static int

match_named_captures_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;

    int i;

    int found = 0;

    for (i = 0; i < back_num; i++) {

	value = rb_reg_nth_match(back_refs[i], match);

	if (RTEST(value)) {

	    rb_hash_aset(hash, key, value);

	    found = 1;

	}

    }

    if (found == 0) {

	rb_hash_aset(hash, key, Qnil);

    }

    return 0;

}

/*

 *  call-seq:

 *     mtch.named_captures -> 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 successful capture of corresponding

 *  group.

 *

 *     m = /(?<a>.)(?<b>.)/.match("01")

 *     m.named_captures #=> {"a" => "0", "b" => "1"}

 *

 *     m = /(?<a>.)(?<b>.)?/.match("0")

 *     m.named_captures #=> {"a" => "0", "b" => nil}

 *

 *     m = /(?<a>.)(?<a>.)/.match("01")

 *     m.named_captures #=> {"a" => "1"}

 *

 *     m = /(?<a>x)|(?<a>y)/.match("x")

 *     m.named_captures #=> {"a" => "x"}

 */

static VALUE

match_named_captures(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_named_captures_iter, (void*)memo);

    return hash;

}

    rb_define_method(rb_cMatch, "named_captures", match_named_captures, 0);

  def test_match_data_named_captures

    assert_equal({'a' => '1', 'b' => '2', 'c' => nil}, /^(?<a>.)(?<b>.)(?<c>.)?/.match('12').named_captures)

    assert_equal({'a' => '1', 'b' => '2', 'c' => '3'}, /^(?<a>.)(?<b>.)(?<c>.)?/.match('123').named_captures)

    assert_equal({'a' => '1', 'b' => '2', 'c' => ''}, /^(?<a>.)(?<b>.)(?<c>.?)/.match('12').named_captures)

    assert_equal({'a' => 'x'}, /(?<a>x)|(?<a>y)/.match('x').named_captures)

    assert_equal({'a' => 'y'}, /(?<a>x)|(?<a>y)/.match('y').named_captures)

    assert_equal({'a' => '1', 'b' => '2'}, /^(.)(?<a>.)(?<b>.)/.match('012').named_captures)

    assert_equal({'a' => '2'}, /^(?<a>.)(?<a>.)/.match('12').named_captures)

    assert_equal({}, /^(.)/.match('123').named_captures)

  end