more cleanup of XS version of get_all_package_symbols
[gitmo/Class-MOP.git] / MOP.xs
1
2 #include "EXTERN.h"
3 #include "perl.h"
4 #include "XSUB.h"
5
6 #define NEED_sv_2pv_nolen
7 #include "ppport.h"
8
9 SV *key_name;
10 U32 hash_name;
11
12 SV *key_package;
13 U32 hash_package;
14
15 SV *key_body;
16 U32 hash_body;
17
18 /*
19 get_code_info:
20   Pass in a coderef, returns:
21   [ $pkg_name, $coderef_name ] ie:
22   [ 'Foo::Bar', 'new' ]
23 */
24
25 MODULE = Class::MOP   PACKAGE = Class::MOP
26
27 BOOT:
28     key_name = newSVpvs("name");
29     key_body = newSVpvs("body");
30     key_package = newSVpvs("package");
31
32     PERL_HASH(hash_name, "name", 4);
33     PERL_HASH(hash_body, "body", 4);
34     PERL_HASH(hash_package, "package", 7);
35
36
37 PROTOTYPES: ENABLE
38
39
40 void
41 get_code_info(coderef)
42   SV* coderef
43   PREINIT:
44     char* name;
45     char* pkg;
46   PPCODE:
47     if( SvOK(coderef) && SvROK(coderef) && SvTYPE(SvRV(coderef)) == SVt_PVCV){
48       coderef = SvRV(coderef);
49       /* I think this only gets triggered with a mangled coderef, but if
50          we hit it without the guard, we segfault. The slightly odd return
51          value strikes me as an improvement (mst)
52       */
53 #ifdef isGV_with_GP
54       if ( isGV_with_GP(CvGV(coderef))) {
55 #endif
56         pkg     = HvNAME( GvSTASH(CvGV(coderef)) );
57         name    = GvNAME( CvGV(coderef) );
58 #ifdef isGV_with_GP
59       } else {
60         pkg     = "__UNKNOWN__";
61         name    = "__ANON__";
62       }
63 #endif
64
65       EXTEND(SP, 2);
66       PUSHs(newSVpvn(pkg, strlen(pkg)));
67       PUSHs(newSVpvn(name, strlen(name)));
68     }
69
70
71 MODULE = Class::MOP   PACKAGE = Class::MOP::Package
72
73 void
74 get_all_package_symbols(self, ...)
75     SV *self
76     PROTOTYPE: $;$
77     PREINIT:
78         HV *stash = NULL;
79         SV *type_filter = NULL;
80         register HE *he;
81     PPCODE:
82
83         switch ( GIMME_V ) {
84             case G_VOID: return; break;
85             case G_SCALAR: ST(0) = &PL_sv_undef; return; break;
86         }
87
88         if ( items > 1 ) type_filter = ST(1);
89
90         PUTBACK;
91
92         if (SvROK(self) && (he = hv_fetch_ent((HV *)SvRV(self), key_package, 0, hash_package)))
93             stash = gv_stashsv(HeVAL(he),0);
94
95         if ( stash ) {
96
97             (void)hv_iterinit(stash);
98
99             if ( type_filter && SvPOK(type_filter) ) {
100                 const char *const type = SvPV_nolen(type_filter);
101
102
103                 while ((he = hv_iternext(stash))) {
104                     SV *const gv = HeVAL(he);
105                     SV *sv;
106                     char *package = HvNAME(stash);
107                     STRLEN pkglen = strlen(package);
108                     char *key;
109                     STRLEN keylen;
110                     char *fq;
111                     STRLEN fqlen;
112
113                     switch( SvTYPE(gv) ) {
114                         case SVt_PVGV:
115                             switch (*type) {
116                                 case 'C': sv = (SV *)GvCVu(gv); break; /* CODE */
117                                 case 'A': sv = (SV *)GvAV(gv); break; /* ARRAY */
118                                 case 'I': sv = (SV *)GvIO(gv); break; /* IO */
119                                 case 'H': sv = (SV *)GvHV(gv); break; /* HASH */
120                                 case 'S': sv = (SV *)GvSV(gv); break; /* SCALAR */
121                                 default:
122                                           croak("Unknown type %s\n", type);
123                             }
124                             break;
125                         case SVt_RV:
126                             /* BAH! constants are horrible */
127                             key = HePV(he, keylen);
128                             fqlen = pkglen + keylen + 3;
129                             fq = (char *)alloca(fqlen);
130                             snprintf(fq, fqlen, "%s::%s", package, key);
131                             sv = (SV*)get_cv(fq, 0);
132                             sv_2mortal(sv);
133                             break;
134                         default:
135                             continue;
136                     }
137
138                     if ( sv ) {
139                         SV *key = hv_iterkeysv(he);
140                         SPAGAIN;
141                         EXTEND(SP, 2);
142                         PUSHs(key);
143                         PUSHs(sv_2mortal(newRV_inc(sv)));
144                         PUTBACK;
145                     }
146                 }
147             } else {
148                 EXTEND(SP, HvKEYS(stash) * 2);
149
150                 while ((he = hv_iternext(stash))) {
151                     SV *key = hv_iterkeysv(he);
152                     SV *sv = hv_iterval(stash, he);
153                     SPAGAIN;
154                     PUSHs(key);
155                     PUSHs(sv);
156                     PUTBACK;
157                 }
158             }
159
160         }
161
162 SV *
163 name(self)
164     SV *self
165     PREINIT:
166         register HE *he;
167     PPCODE:
168         if (SvROK(self) && (he = hv_fetch_ent((HV *)SvRV(self), key_package, 0, hash_package)))
169             XPUSHs(HeVAL(he));
170         else
171             ST(0) = &PL_sv_undef;
172
173 MODULE = Class::MOP   PACKAGE = Class::Attribute
174
175 SV *
176 name(self)
177     SV *self
178     PREINIT:
179         register HE *he;
180     PPCODE:
181         if (SvROK(self) && (he = hv_fetch_ent((HV *)SvRV(self), key_name, 0, hash_name)))
182             XPUSHs(HeVAL(he));
183         else
184             ST(0) = &PL_sv_undef;
185
186 MODULE = Class::MOP   PACKAGE = Class::Method
187
188 SV *
189 name(self)
190     SV *self
191     PREINIT:
192         register HE *he;
193     PPCODE:
194         if (SvROK(self) && (he = hv_fetch_ent((HV *)SvRV(self), key_name, 0, hash_name)))
195             XPUSHs(HeVAL(he));
196         else
197             ST(0) = &PL_sv_undef;
198
199 SV *
200 body(self)
201     SV *self
202     PREINIT:
203         register HE *he;
204     PPCODE:
205         if (SvROK(self) && (he = hv_fetch_ent((HV *)SvRV(self), key_body, 0, hash_body)))
206             XPUSHs(HeVAL(he));
207         else
208             ST(0) = &PL_sv_undef;