static VALUE
rb_mod_modfunc(VALUE module, SEL sel, int argc, VALUE *argv)
{
int i;
if (TYPE(module) != T_MODULE) {
rb_raise(rb_eTypeError, "module_function must be called for modules");
}
secure_visibility(module);
if (argc == 0) {
// TODO change scope!
return module;
}
set_method_visibility(module, argc, argv, NOEX_PRIVATE);
for (i = 0; i < argc; i++) {
ID id = rb_to_id(argv[i]);
IMP imp;
rb_vm_method_node_t *node;
SEL sel;
if (!rb_vm_lookup_method2((Class)module, id, &sel, &imp, &node)) {
// Methods are checked in set_method_visibility().
rb_bug("undefined method `%s'; can't happen", rb_id2name(id));
}
rb_vm_define_method2(*(Class *)module, sel, node, false);
}
return module;
}
static VALUE
rb_mod_modfunc(VALUE module, SEL sel, int argc, VALUE *argv)
{
if (TYPE(module) != T_MODULE) {
rb_raise(rb_eTypeError, "module_function must be called for modules");
}
secure_visibility(module);
if (argc == 0) {
rb_vm_set_current_scope(module, SCOPE_MODULE_FUNC);
return module;
}
set_method_visibility(module, argc, argv, NOEX_PRIVATE);
for (int i = 0; i < argc; i++) {
ID id = rb_to_id(argv[i]);
IMP imp = NULL;
rb_vm_method_node_t *node = NULL;
SEL sel = 0;
if (rb_vm_lookup_method2((Class)module, id, &sel, &imp, &node)
|| (TYPE(module) == T_MODULE
&& rb_vm_lookup_method2((Class)rb_cObject, id, &sel,
&imp, &node))) {
rb_vm_define_method2(*(Class *)module, sel, node, -1, false);
}
else {
rb_bug("undefined method `%s'; can't happen", rb_id2name(id));
}
}
return module;
}
static void
rb_export_method(VALUE klass, ID name, ID noex)
{
rb_vm_method_node_t *node;
SEL sel;
if (klass == rb_cObject) {
rb_secure(4);
}
if (!rb_vm_lookup_method2((Class)klass, name, &sel, NULL, &node)) {
if (TYPE(klass) != T_MODULE
|| !rb_vm_lookup_method2((Class)rb_cObject, name, &sel, NULL,
&node)) {
rb_print_undef(klass, name, 0);
}
}
if (node == NULL) {
rb_raise(rb_eRuntimeError,
"can't change visibility of non Ruby method `%s'",
sel_getName(sel));
}
long flags = (node->flags & ~VM_METHOD_PRIVATE) & ~VM_METHOD_PROTECTED;
switch (noex) {
case NOEX_PRIVATE:
flags |= VM_METHOD_PRIVATE;
break;
case NOEX_PROTECTED:
flags |= VM_METHOD_PROTECTED;
break;
default:
break;
}
if (node->flags != flags) {
if (node->klass == (Class)klass) {
node->flags = flags;
}
else {
rb_vm_define_method2((Class)klass, sel, node, flags, false);
}
}
}
static VALUE
rb_mod_define_method(VALUE mod, SEL sel, int argc, VALUE *argv)
{
#if MACRUBY_STATIC
not_implemented_in_static(sel);
#else
ID id;
VALUE body;
if (argc == 1) {
id = rb_to_id(argv[0]);
body = rb_block_lambda();
}
else if (argc == 2) {
id = rb_to_id(argv[0]);
body = argv[1];
if (!rb_obj_is_method(body) && !rb_obj_is_proc(body)) {
rb_raise(rb_eTypeError,
"wrong argument type %s (expected Proc/Method)",
rb_obj_classname(body));
}
}
else {
rb_raise(rb_eArgError, "wrong number of arguments (%d for 1)", argc);
}
if (rb_obj_is_method(body)) {
rb_vm_method_t *data;
Data_Get_Struct(body, rb_vm_method_t, data);
if (data->node == NULL) {
rb_raise(rb_eArgError, "cannot use Method object of pure Objective-C method");
}
SEL msel = rb_vm_id_to_sel(id, data->arity);
rb_vm_define_method2((Class)mod, msel, data->node, data->node->flags, false);
}
else {
rb_vm_block_t *proc;
GetProcPtr(body, proc);
rb_vm_define_method3((Class)mod, id, proc);
}
return body;
#endif
}
