admin/mruby-mruby
1
0
Fork 0
mirror of https://github.com/mruby/mruby synced 2026-06-08 16:11:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Extended Module#const_get to support class paths

Browse Source
This commit is contained in:
Christopher Aue
2017-07-28 22:46:26 +02:00
parent c7018ea962
commit f937af5745
3 changed files with 42 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/class.c
+34 -4
View File
@@ -2161,15 +2161,45 @@ mrb_mod_const_defined(mrb_state *mrb, mrb_value mod)
}
static mrb_value
mrb_mod_const_get(mrb_state *mrb, mrb_value mod)
mrb_const_get_sym(mrb_state *mrb, mrb_value mod, mrb_sym id)
{
mrb_sym id;
mrb_get_args(mrb, "n", &id);
check_const_name_sym(mrb, id);
return mrb_const_get(mrb, mod, id);
}
static mrb_value
mrb_mod_const_get(mrb_state *mrb, mrb_value mod)
{
mrb_value path;
mrb_sym id;
char *ptr;
mrb_int off, end, len;
mrb_get_args(mrb, "o", &path);
if (mrb_symbol_p(path)) {
/* const get with symbol */
id = mrb_symbol(path);
return mrb_const_get_sym(mrb, mod, id);
}
/* const get with class path string */
path = mrb_string_type(mrb, path);
ptr = RSTRING_PTR(path);
len = RSTRING_LEN(path);
off = 0;
while (off < len) {
end = mrb_str_index_lit(mrb, path, "::", off);
end = (end == -1) ? len : end;
id = mrb_intern(mrb, ptr+off, end-off);
mod = mrb_const_get_sym(mrb, mod, id);
off = (end == len) ? end : end+2;
}
return mod;
}
static mrb_value
mrb_mod_const_set(mrb_state *mrb, mrb_value mod)
{