From 7bc4d08f71950f673d788653c50703263edb61df Mon Sep 17 00:00:00 2001 From: take_cheeze Date: Wed, 28 May 2014 23:14:28 +0900 Subject: Support string key in `Struct#[]=`. --- mrbgems/mruby-struct/src/struct.c | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'mrbgems/mruby-struct/src') diff --git a/mrbgems/mruby-struct/src/struct.c b/mrbgems/mruby-struct/src/struct.c index a15655dbb..adac60202 100644 --- a/mrbgems/mruby-struct/src/struct.c +++ b/mrbgems/mruby-struct/src/struct.c @@ -660,6 +660,14 @@ mrb_struct_aset(mrb_state *mrb, mrb_value s) mrb_get_args(mrb, "oo", &idx, &val); + if (mrb_string_p(idx)) { + mrb_value sym = mrb_check_intern_str(mrb, idx); + + if (mrb_nil_p(sym)) { + mrb_raisef(mrb, E_INDEX_ERROR, "no member '%S' in struct", idx); + } + idx = sym; + } if (mrb_symbol_p(idx)) { return mrb_struct_aset_sym(mrb, s, mrb_symbol(idx), val); } -- cgit v1.2.3