XS version of get_all_package_symbols

[gitmo/Class-MOP.git] / MOP.xs

diff --git a/MOP.xs b/MOP.xs
index f1071dd..b7dfd03 100644 (file)
--- a/MOP.xs
+++ b/MOP.xs
@@ -2,6 +2,8 @@
 #include "EXTERN.h"
 #include "perl.h"
 #include "XSUB.h"
+
+#define NEED_sv_2pv_nolen
 #include "ppport.h"
 
 /*
@@ -45,3 +47,107 @@ get_code_info(coderef)
       PUSHs(newSVpvn(name, strlen(name)));
     }
 
+
+MODULE = Class::MOP   PACKAGE = Class::MOP::Package
+
+void
+get_all_package_symbols(package, ...)
+    SV *package
+    PROTOTYPE: $;$
+    PREINIT:
+        HV *stash;
+        SV *type_filter = NULL;
+    PPCODE:
+
+        switch ( GIMME_V ) {
+            case G_VOID: return; break;
+            case G_SCALAR: ST(0) = &PL_sv_undef; return; break;
+        }
+
+        if ( items > 1 ) type_filter = ST(1);
+
+        PUTBACK;
+
+        ENTER;
+        SAVETMPS;
+        PUSHMARK(SP);
+        XPUSHs(package);
+        PUTBACK;
+        call_method("name", 0);
+        SPAGAIN;
+        stash = gv_stashsv(POPs, 0);
+        FREETMPS;
+        LEAVE;
+
+        PUTBACK;
+
+        if ( stash ) {
+            register HE *entry;
+
+            (void)hv_iterinit(stash);
+
+            if ( type_filter && SvPOK(type_filter) ) {
+                const char *const type = SvPV_nolen(type_filter);
+
+
+                while ((entry = hv_iternext(stash))) {
+                    SV *const gv = hv_iterval(stash, entry);
+                    SV *const key = hv_iterkeysv(entry);
+                    SV *sv;
+                    char *package = HvNAME(stash);
+                    STRLEN pkglen = strlen(package);
+                    char *fq;
+                    STRLEN fqlen;
+
+                    SPAGAIN;
+
+                    switch( SvTYPE(gv) ) {
+                        case SVt_PVGV:
+                            switch (*type) {
+                                case 'C': sv = (SV *)GvCVu(gv); break; /* CODE */
+                                case 'A': sv = (SV *)GvAV(gv); break; /* ARRAY */
+                                case 'I': sv = (SV *)GvIO(gv); break; /* IO */
+                                case 'H': sv = (SV *)GvHV(gv); break; /* HASH */
+                                case 'S': sv = (SV *)GvSV(gv); break; /* SCALAR */
+                                default:
+                                          croak("Unknown type %s\n", type);
+                            }
+                            break;
+                        case SVt_RV:
+                            /* BAH! constants are horrible */
+                            fqlen = pkglen + SvCUR(key) + 3;
+                            fq = (char *)alloca(fqlen);
+                            snprintf(fq, fqlen, "%s::%s", package, SvPV_nolen(key));
+                            sv = get_cv(fq, 0);
+                            break;
+                        default:
+                            continue;
+                    }
+
+                    if ( sv ) {
+                        SPAGAIN;
+                        EXTEND(SP, 2);
+                        PUSHs(key);
+                        PUSHs(newRV_noinc(sv));
+                        PUTBACK;
+                    }
+                }
+            } else {
+                EXTEND(SP, HvKEYS(stash) * 2);
+
+                while ((entry = hv_iternext(stash))) {
+                    SV *sv;
+                    SPAGAIN;
+                    sv = hv_iterkeysv(entry);
+                    SPAGAIN;
+                    PUSHs(sv);
+                    PUTBACK;
+                    sv = hv_iterval(stash, entry);
+                    SPAGAIN;
+                    PUSHs(sv);
+                    PUTBACK;
+                }
+            }
+
+        }
+