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

Merge pull request #3224 from ksss/define_method

Browse Source 
Module#define_method supports proc argument
This commit is contained in:
Yukihiro "Matz" Matsumoto
2016-11-11 22:57:52 +09:00
committed by GitHub
parent 752ca514f6 efe4f30b39
commit feb9b3debd
2 changed files with 25 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/class.c
+13 -1
View File
@@ -1783,9 +1783,21 @@ mod_define_method(mrb_state *mrb, mrb_value self)
struct RClass *c = mrb_class_ptr(self);
struct RProc *p;
mrb_sym mid;
mrb_value proc = mrb_undef_value();
mrb_value blk;
mrb_get_args(mrb, "n&", &mid, &blk);
mrb_get_args(mrb, "n|o&", &mid, &proc, &blk);
switch (mrb_type(proc)) {
case MRB_TT_PROC:
blk = proc;
break;
case MRB_TT_UNDEF:
/* ignored */
break;
default:
mrb_raisef(mrb, E_TYPE_ERROR, "wrong argument type %S (expected Proc)", mrb_obj_value(mrb_obj_class(mrb, proc)));
break;
}
if (mrb_nil_p(blk)) {
mrb_raise(mrb, E_ARGUMENT_ERROR, "no block given");
}