Title / Description
Code /* remix.c */ /* (C) John Mair 2010 * This program is distributed under the terms of the MIT License * */ #include <ruby.h> #include "compat.h" VALUE rb_swap_modules(VALUE self, VALUE mod1, VALUE mod2); #define Enforce_Classmod(klass) \ do { \ if (!RTEST(rb_obj_is_kind_of(klass, rb_cModule))) \ rb_raise(rb_eTypeError, "Must be a Module or Class type."); \ } while(0) #define Validate_Type(klass) \ do { \ if (TYPE(klass) != T_OBJECT && TYPE(klass) != T_CLASS && TYPE(klass) != T_MODULE && TYPE(klass) != T_ICLASS && \ TYPE(klass) != T_FALSE) \ rb_raise(rb_eTypeError, "Must be a T_MODULE, T_CLASS, T_ICLASS, T_OBJECT, or T_FALSE type."); \ } while(0) /* Tiny utility method to return an object attached to a singleton */ static VALUE rb_singleton_attached(VALUE self) { if(FL_TEST(self, FL_SINGLETON)) return rb_iv_get(self, "__attached__"); else return Qnil; } /* a modified version of include_class_new from class.c */ static VALUE j_class_new(VALUE module, VALUE sup) { VALUE klass = create_class(T_ICLASS, rb_cClass); if (!RCLASS_IV_TBL(module)) { RCLASS_IV_TBL(module) = (struct st_table *)st_init_numtable(); } /* assign iv_tbl, m_tbl and super */ RCLASS_IV_TBL(klass) = RCLASS_IV_TBL(module); if (TYPE(module) == T_ICLASS) { if (!RTEST(rb_iv_get(module, "__module__"))) rb_iv_set(klass, "__module__", KLASS_OF(module)); } else if (TYPE(module) == T_MODULE || TYPE(module) == T_CLASS) rb_iv_set(klass, "__module__", module); RCLASS_SUPER(klass) = sup; if(TYPE(module) != T_OBJECT) { RCLASS_M_TBL(klass) = RCLASS_M_TBL(module); } else { RCLASS_M_TBL(klass) = RCLASS_M_TBL(CLASS_OF(module)); } if (TYPE(module) == T_ICLASS) { KLASS_OF(klass) = rb_iv_get(klass, "__module__"); } else { KLASS_OF(klass) = module; } if(TYPE(module) != T_OBJECT) { OBJ_INFECT(klass, module); OBJ_INFECT(klass, sup); } return (VALUE)klass; } static VALUE set_supers(VALUE c) { if (RCLASS_SUPER(c) == rb_cObject || RCLASS_SUPER(c) == 0) { return RCLASS_SUPER(c); } else { return j_class_new(RCLASS_SUPER(c), set_supers(RCLASS_SUPER(c))); } } VALUE rb_prepare_for_remix(VALUE klass) { Enforce_Classmod(klass); /* class chain is already prepared for remixing */ if (RTEST(rb_iv_get(klass, "__remix_ready__"))) return klass; RCLASS_SUPER(klass) = set_supers(klass); rb_iv_set(klass, "__remix_ready__", Qtrue); rb_clear_cache(); return klass; } inline static VALUE get_source_module(VALUE mod) { switch (TYPE(mod)) { case T_FALSE: return Qfalse; break; case T_ICLASS: if (RTEST(rb_iv_get(mod, "__module__"))) return rb_iv_get(mod, "__module__"); else return KLASS_OF(mod); break; case T_CLASS: case T_MODULE: return mod; break; case T_OBJECT: return rb_singleton_class(mod); break; default: Validate_Type(mod); } /* never reached */ return Qnil; } static VALUE retrieve_imod_before_mod(VALUE m, VALUE before) { Validate_Type(before); VALUE k = get_source_module(RCLASS_SUPER(m)); while(k != before && m != 0 && m != rb_cObject) { m = RCLASS_SUPER(m); k = get_source_module(RCLASS_SUPER(m)); } if (k != before) rb_raise(rb_eRuntimeError, "'before' module not found"); return m; } static VALUE retrieve_imod_for_mod(VALUE m, VALUE after) { Validate_Type(after); VALUE k = get_source_module(m); while(k != after && m != 0 && m != rb_cObject) { m = RCLASS_SUPER(m); k = get_source_module(m); } if (k != after) rb_raise(rb_eRuntimeError, "'after' module not found"); return m; } VALUE rb_module_move_up(VALUE self, VALUE mod) { rb_prepare_for_remix(self); if (self == mod) return self; VALUE included_mod = retrieve_imod_for_mod(self, mod); if (RCLASS_SUPER(included_mod) == rb_cObject || RCLASS_SUPER(included_mod) == 0) return self; rb_swap_modules(self, mod, get_source_module(RCLASS_SUPER(included_mod))); return self; } VALUE rb_module_move_down(VALUE self, VALUE mod) { rb_prepare_for_remix(self); if (self == mod) return self; VALUE before_included_mod = retrieve_imod_before_mod(self, mod); if (before_included_mod == self) return self; rb_swap_modules(self, mod, get_source_module(before_included_mod)); return self; } VALUE rb_include_at_top(VALUE self, VALUE mod) { rb_prepare_for_remix(self); if (TYPE(self) == T_MODULE) rb_include_module(retrieve_imod_before_mod(self, Qfalse), mod); else rb_include_module(retrieve_imod_before_mod(self, rb_cObject), mod); return self; } VALUE rb_include_after(VALUE self, VALUE after, VALUE mod) { rb_prepare_for_remix(self); rb_include_module(retrieve_imod_for_mod(self, after), mod); return self; } VALUE rb_include_before(VALUE self, VALUE before, VALUE mod) { rb_prepare_for_remix(self); if (before == self) rb_raise(rb_eRuntimeError, "Prepend not supported yet!"); rb_include_module(retrieve_imod_before_mod(self, before), mod); return self; } VALUE rb_include_at(VALUE self, VALUE rb_index, VALUE mod) { rb_prepare_for_remix(self); Check_Type(rb_index, T_FIXNUM); int index = FIX2INT(rb_index); VALUE m = self; int i = 1; while(i++ < index && RCLASS_SUPER(m) != 0 && RCLASS_SUPER(m) != rb_cObject) m = RCLASS_SUPER(m); rb_include_module(m, mod); return self; } #define SWAP(X, Y) {(X) ^= (Y); (Y) ^= (X); (X) ^= (Y);} VALUE rb_swap_modules(VALUE self, VALUE mod1, VALUE mod2) { rb_prepare_for_remix(self); VALUE before_mod1, before_mod2; VALUE included_mod1, included_mod2; if (mod1 == rb_cObject || mod2 == rb_cObject) rb_raise(rb_eRuntimeError, "can't swap Object"); if (mod1 == mod2) return self; included_mod1 = retrieve_imod_for_mod(self, mod1); included_mod2 = retrieve_imod_for_mod(self, mod2); before_mod1 = retrieve_imod_before_mod(self, mod1); before_mod2 = retrieve_imod_before_mod(self, mod2); SWAP(RCLASS_SUPER(before_mod1), RCLASS_SUPER(before_mod2)); SWAP(RCLASS_SUPER(included_mod1), RCLASS_SUPER(included_mod2)); rb_clear_cache(); return self; } static void remove_nested_module(VALUE included_mod, VALUE module) { VALUE source_mod = get_source_module(RCLASS_SUPER(included_mod)); if (source_mod == rb_cObject || source_mod == Qfalse) return; else if (source_mod == get_source_module(RCLASS_SUPER(module))) { remove_nested_module(RCLASS_SUPER(included_mod), RCLASS_SUPER(module)); RCLASS_SUPER(included_mod) = RCLASS_SUPER(RCLASS_SUPER(included_mod)); } } static VALUE rb_classmod_include_p(VALUE mod, VALUE mod2) { VALUE p; Enforce_Classmod(mod); for (p = RCLASS_SUPER(mod); p; p = RCLASS_SUPER(p)) { if (BUILTIN_TYPE(p) == T_ICLASS) { if (RTEST(rb_iv_get(p, "__module__"))) { if (rb_iv_get(p, "__module__") == mod2) return Qtrue; } if (RBASIC(p)->klass == mod2) return Qtrue; } } return Qfalse; } VALUE rb_uninclude(int argc, VALUE * argv, VALUE self) { rb_prepare_for_remix(self); VALUE mod1, recurse = Qfalse; rb_scan_args(argc, argv, "11", &mod1, &recurse); if (TYPE(mod1) == T_OBJECT) mod1 = rb_singleton_class(mod1); if (!RTEST(rb_classmod_include_p(self, mod1))) rb_raise(rb_eArgError, "Module not found"); VALUE before = retrieve_imod_before_mod(self, mod1); VALUE included_mod = retrieve_imod_for_mod(self, mod1); if (mod1 == rb_cObject) rb_raise(rb_eRuntimeError, "can't delete Object"); if (RTEST(recurse)) remove_nested_module(included_mod, mod1); RCLASS_SUPER(before) = RCLASS_SUPER(included_mod); rb_clear_cache(); return self; } VALUE rb_replace_module(VALUE self, VALUE mod1, VALUE mod2) { rb_prepare_for_remix(self); if (mod1 == self) rb_raise(rb_eRuntimeError, "can't replace root module."); if (rb_classmod_include_p(self, mod2)) return rb_swap_modules(self, mod1, mod2); if (mod1 == mod2) return; VALUE before = retrieve_imod_before_mod(self, mod1); rb_uninclude(1, &mod1, self); rb_include_module(before, mod2); return self; } void Init_remix() { VALUE mRemix = rb_define_module("Remix"); VALUE mModuleExtensions = rb_define_module_under(mRemix, "ModuleExtensions"); rb_define_method(mModuleExtensions, "ready_remix", rb_prepare_for_remix, 0); rb_define_method(mModuleExtensions, "module_move_up", rb_module_move_up, 1); rb_define_method(mModuleExtensions, "module_move_down", rb_module_move_down, 1); rb_define_method(mModuleExtensions, "include_at", rb_include_at, 2); rb_define_method(mModuleExtensions, "include_below", rb_include_before, 2); rb_define_alias(mModuleExtensions, "include_before", "include_below"); rb_define_method(mModuleExtensions, "include_above", rb_include_after, 2); rb_define_alias(mModuleExtensions, "include_after", "include_above"); rb_define_method(mModuleExtensions, "include_at_top", rb_include_at_top, 1); rb_define_method(mModuleExtensions, "swap_modules", rb_swap_modules, 2); rb_define_method(mModuleExtensions, "uninclude", rb_uninclude, -1); rb_define_alias(mModuleExtensions, "remove_module", "uninclude"); rb_define_method(mModuleExtensions, "replace_module", rb_replace_module, 2); rb_define_method(mModuleExtensions, "__attached__", rb_singleton_attached, 0); }
Author
Highlight as C C++ CSS Clojure Delphi ERb Groovy (beta) HAML HTML JSON Java JavaScript PHP Plain text Python Ruby SQL XML YAML diff code
1.1.2
1.1.2