Implement get_all_attributes in XS
[gitmo/Mouse.git] / xs-src / Mouse.xs
index 59070e4..9886ec8 100644 (file)
@@ -1,3 +1,4 @@
+#define  NEED_newSVpvn_flags_GLOBAL
 #include "mouse.h"
 
 SV* mouse_package;
@@ -5,127 +6,115 @@ SV* mouse_namespace;
 SV* mouse_methods;
 SV* mouse_name;
 
-MODULE = Mouse  PACKAGE = Mouse::Util
+static SV* mouse_all_attrs_cache;
+static SV* mouse_all_attrs_cache_gen;
 
-PROTOTYPES: DISABLE
+AV*
+mouse_get_all_attributes(pTHX_ SV* const metaclass){
+    SV* const package = get_slot(metaclass, mouse_package);
+    HV* const stash   = gv_stashsv(package, TRUE);
+    UV const pkg_gen  = mro_get_pkg_gen(stash);
+    SV* cache_gen     = get_slot(metaclass, mouse_all_attrs_cache_gen);
 
-BOOT:
-    mouse_package   = newSVpvs_share("package");
-    mouse_namespace = newSVpvs_share("namespace");
-    mouse_methods   = newSVpvs_share("methods");
-    mouse_name      = newSVpvs_share("name");
+    if(!(cache_gen && pkg_gen == SvUV(cache_gen))){ /* update */
+        CV* const get_metaclass  = get_cvs("Mouse::Util::get_metaclass_by_name", TRUE);
+        AV* const linearized_isa = mro_get_linear_isa(stash);
+        I32 const len            = AvFILLp(linearized_isa);
+        I32 i;
+        HV* seen;
+        AV* const all_attrs = newAV();
 
-    MOUSE_CALL_BOOT(Mouse__Util__TypeConstraints);
+        /* warn("Update all_attrs_cache (cache_gen %d != pkg_gen %d)", (cache_gen ? (int)SvIV(cache_gen) : 0), (int)pkg_gen); //*/
 
+        ENTER;
+        SAVETMPS;
 
-bool
-is_class_loaded(SV* sv)
+        set_slot(metaclass, mouse_all_attrs_cache, sv_2mortal(newRV_inc((SV*)all_attrs)));
 
-void
-get_code_info(CV* code)
-PREINIT:
-    GV* gv;
-    HV* stash;
-PPCODE:
-    if((gv = CvGV(code)) && isGV(gv) && (stash = GvSTASH(gv))){
-        EXTEND(SP, 2);
-        mPUSHs(newSVpvn_share(HvNAME_get(stash), HvNAMELEN_get(stash), 0U));
-        mPUSHs(newSVpvn_share(GvNAME_get(gv), GvNAMELEN_get(gv), 0U));
-    }
+        seen = newHV();
+        sv_2mortal((SV*)seen);
 
-SV*
-get_code_package(CV* code)
-PREINIT:
-    HV* stash;
-CODE:
-    if(CvGV(code) && isGV(CvGV(code)) && (stash = GvSTASH(CvGV(code)))){
-        RETVAL = newSVpvn_share(HvNAME_get(stash), HvNAMELEN_get(stash), 0U);
-    }
-    else{
-        RETVAL = &PL_sv_no;
-    }
-OUTPUT:
-    RETVAL
+        for(i = 0; i < len; i++){
+            SV* const klass = MOUSE_av_at(linearized_isa, i);
+            SV* meta;
+            I32 n;
+            dSP;
 
-CV*
-get_code_ref(SV* package, SV* name)
-CODE:
-{
-    HV* stash;
-    HE* he;
+            PUSHMARK(SP);
+            XPUSHs(klass);
+            PUTBACK;
 
-    if(!SvOK(package)){
-        croak("You must define a package name");
-    }
-    if(!SvOK(name)){
-        croak("You must define a subroutine name");
-    }
+            call_sv((SV*)get_metaclass, G_SCALAR);
 
-    stash = gv_stashsv(package, FALSE);
-    if(!stash){
-        XSRETURN_UNDEF;
-    }
-    he = hv_fetch_ent(stash, name, FALSE, 0U);
-    if(he){
-        GV* const gv = (GV*)hv_iterval(stash, he);
-        if(!isGV(gv)){ /* special constant or stub */
-            STRLEN len;
-            const char* const pv = SvPV_const(name, len);
-            gv_init(gv, stash, pv, len, GV_ADDMULTI);
-        }
-        RETVAL = GvCVu(gv);
-    }
-    else{
-        RETVAL = NULL;
-    }
+            SPAGAIN;
+            meta = POPs;
+            PUTBACK;
 
-    if(!RETVAL){
-        XSRETURN_UNDEF;
-    }
-}
-OUTPUT:
-    RETVAL
+            if(!SvOK(meta)){
+                continue; /* skip non-Mouse classes */
+            }
 
-void
-_generate_isa_predicate_for(SV* klass, const char* predicate_name = NULL)
-PPCODE:
-{
-    STRLEN klass_len;
-    const char* klass_pv;
-    HV* stash;
-    CV* xsub;
+            /* $meta->get_attribute_list */
+            PUSHMARK(SP);
+            XPUSHs(meta);
+            PUTBACK;
 
-    if(!SvOK(klass)){
-        croak("You must define a class name for generate_for");
-    }
-    klass_pv = SvPV_const(klass, klass_len);
-    klass_pv = mouse_canonicalize_package_name(klass_pv);
+            n = call_method("get_attribute_list", G_ARRAY);
+            for(NOOP; n > 0; n--){
+                SV* name;
 
-    if(strNE(klass_pv, "UNIVERSAL")){
-        static MGVTBL mouse_util_type_constraints_vtbl; /* not used, only for identity */
+                SPAGAIN;
+                name = POPs;
+                PUTBACK;
 
-        xsub = newXS(predicate_name, XS_isa_check, __FILE__);
+                if(hv_exists_ent(seen, name, 0U)){
+                    continue;
+                }
+                (void)hv_store_ent(seen, name, &PL_sv_undef, 0U);
 
-        stash = gv_stashpvn(klass_pv, klass_len, GV_ADD);
+                av_push(all_attrs, newSVsv( mcall1s(meta, "get_attribute", name) ));
+            }
+        }
 
-        CvXSUBANY(xsub).any_ptr = sv_magicext(
-            (SV*)xsub,
-            (SV*)stash, /* mg_obj */
-            PERL_MAGIC_ext,
-            &mouse_util_type_constraints_vtbl,
-            klass_pv,   /* mg_ptr */
-            klass_len   /* mg_len */
-        );
-    }
-    else{
-        xsub = newXS(predicate_name, XS_isa_check_for_universal, __FILE__);
+        if(!cache_gen){
+            cache_gen = sv_newmortal();
+        }
+        sv_setuv(cache_gen, mro_get_pkg_gen(stash));
+        set_slot(metaclass, mouse_all_attrs_cache_gen, cache_gen);
+
+        FREETMPS;
+        LEAVE;
+
+        return all_attrs;
     }
+    else {
+        SV* const all_attrs_ref = get_slot(metaclass, mouse_all_attrs_cache);
 
-    if(predicate_name == NULL){ /* anonymous predicate */
-        XPUSHs( newRV_noinc((SV*)xsub) );
+        if(!IsArrayRef(all_attrs_ref)){
+            croak("Not an ARRAY reference");
+        }
+
+        return (AV*)SvRV(all_attrs_ref);
     }
 }
 
+MODULE = Mouse  PACKAGE = Mouse
+
+PROTOTYPES: DISABLE
+
+BOOT:
+    mouse_package   = newSVpvs_share("package");
+    mouse_namespace = newSVpvs_share("namespace");
+    mouse_methods   = newSVpvs_share("methods");
+    mouse_name      = newSVpvs_share("name");
+
+    mouse_all_attrs_cache      = newSVpvs_share("__all_attrs_cache");
+    mouse_all_attrs_cache_gen  = newSVpvs_share("__all_attrs_cache_gen");
+
+    MOUSE_CALL_BOOT(Mouse__Util);
+    MOUSE_CALL_BOOT(Mouse__Util__TypeConstraints);
+    MOUSE_CALL_BOOT(Mouse__Meta__Method__Accessor__XS);
+
 
 MODULE = Mouse  PACKAGE = Mouse::Meta::Module
 
@@ -138,7 +127,7 @@ HV*
 namespace(SV* self)
 CODE:
 {
-    SV* const package = mouse_instance_get_slot(aTHX_ self, mouse_package);
+    SV* const package = get_slot(self, mouse_package);
     if(!(package && SvOK(package))){
         croak("No package name defined");
     }
@@ -152,8 +141,8 @@ void
 add_method(SV* self, SV* name, SV* code, ...)
 CODE:
 {
-    SV* const package = mouse_instance_get_slot(aTHX_ self, mouse_package); /* $self->{package} */
-    SV* const methods = mouse_instance_get_slot(aTHX_ self, mouse_methods); /* $self->{methods} */
+    SV* const package = get_slot(self, mouse_package); /* $self->{package} */
+    SV* const methods = get_slot(self, mouse_methods); /* $self->{methods} */
     GV* gv;
     SV* code_ref;
 
@@ -190,7 +179,7 @@ CODE:
     }
     sv_setsv_mg((SV*)gv, code_ref); /* *gv = $code_ref */
 
-    mouse_instance_set_slot(aTHX_ methods, name, code); /* $self->{methods}{$name} = $code */
+    set_slot(methods, name, code); /* $self->{methods}{$name} = $code */
 
     /* TODO: name the CODE ref if it's anonymous */
     //code_entity = (CV*)SvRV(code_ref);
@@ -227,6 +216,20 @@ PPCODE:
     }
 }
 
+void
+get_all_attributes(SV* self)
+PPCODE:
+{
+    AV* const all_attrs = mouse_get_all_attributes(aTHX_ self);
+    I32 const len       = AvFILLp(all_attrs) + 1;
+    I32 i;
+
+    EXTEND(SP, len);
+    for(i = 0; i < len; i++){
+        PUSHs( MOUSE_av_at(all_attrs, i) );
+    }
+}
+
 MODULE = Mouse  PACKAGE = Mouse::Meta::Role
 
 BOOT:
@@ -277,69 +280,3 @@ BOOT:
     newCONSTSUB(gv_stashpvs("Mouse::Meta::Attribute", TRUE), "accessor_metaclass",
         newSVpvs("Mouse::Meta::Method::Accessor::XS"));
 
-MODULE = Mouse  PACKAGE = Mouse::Meta::TypeConstraint
-
-BOOT:
-    INSTALL_SIMPLE_READER(TypeConstraint, name);
-    INSTALL_SIMPLE_READER(TypeConstraint, parent);
-    INSTALL_SIMPLE_READER(TypeConstraint, message);
-
-    INSTALL_SIMPLE_READER_WITH_KEY(TypeConstraint, _compiled_type_constraint, compiled_type_constraint);
-    INSTALL_SIMPLE_READER(TypeConstraint, _compiled_type_coercion); /* Mouse specific */
-
-    INSTALL_SIMPLE_PREDICATE_WITH_KEY(TypeConstraint, has_coercion, _compiled_type_coercion);
-
-
-MODULE = Mouse  PACKAGE = Mouse::Meta::Method::Accessor::XS
-
-CV*
-_generate_accessor(klass, SV* attr, metaclass)
-CODE:
-{
-    RETVAL = mouse_instantiate_xs_accessor(aTHX_ attr, mouse_xs_accessor);
-}
-OUTPUT:
-    RETVAL
-
-CV*
-_generate_reader(klass, SV* attr, metaclass)
-CODE:
-{
-    RETVAL = mouse_instantiate_xs_accessor(aTHX_ attr, mouse_xs_reader);
-}
-OUTPUT:
-    RETVAL
-
-CV*
-_generate_writer(klass, SV* attr, metaclass)
-CODE:
-{
-    RETVAL = mouse_instantiate_xs_accessor(aTHX_ attr, mouse_xs_writer);
-}
-OUTPUT:
-    RETVAL
-
-CV*
-_generate_clearer(klass, SV* attr, metaclass)
-CODE:
-{
-    SV* const slot = mcall0s(attr, "name");
-    STRLEN len;
-    const char* const pv = SvPV_const(slot, len);
-    RETVAL = mouse_install_simple_accessor(aTHX_ NULL, pv, len, mouse_xs_simple_clearer);
-}
-OUTPUT:
-    RETVAL
-
-CV*
-_generate_predicate(klass, SV* attr, metaclass)
-CODE:
-{
-    SV* const slot = mcall0s(attr, "name");
-    STRLEN len;
-    const char* const pv = SvPV_const(slot, len);
-    RETVAL = mouse_install_simple_accessor(aTHX_ NULL, pv, len, mouse_xs_simple_predicate);
-}
-OUTPUT:
-    RETVAL
-