add_attribute tweaks
[gitmo/Class-MOP.git] / xs / Class.xs
1 #include "mop.h"
2
3 static void
4 mop_update_method_map(pTHX_ SV *const self, SV *const class_name, HV *const stash, HV *const map)
5 {
6     const char *const class_name_pv = HvNAME(stash); /* must be HvNAME(stash), not SvPV_nolen_const(class_name) */
7     SV   *method_metaclass_name;
8     char *method_name;
9     I32   method_name_len;
10     SV   *coderef;
11     HV   *symbols;
12     dSP;
13
14     symbols = mop_get_all_package_symbols(stash, TYPE_FILTER_CODE);
15     sv_2mortal((SV*)symbols);
16     (void)hv_iterinit(symbols);
17     while ( (coderef = hv_iternextsv(symbols, &method_name, &method_name_len)) ) {
18         CV *cv = (CV *)SvRV(coderef);
19         char *cvpkg_name;
20         char *cv_name;
21         SV *method_slot;
22         SV *method_object;
23
24         if (!mop_get_code_info(coderef, &cvpkg_name, &cv_name)) {
25             continue;
26         }
27
28         /* this checks to see that the subroutine is actually from our package  */
29         if ( !(strEQ(cvpkg_name, "constant") && strEQ(cv_name, "__ANON__")) ) {
30             if ( strNE(cvpkg_name, class_name_pv) ) {
31                 continue;
32             }
33         }
34
35         method_slot = *hv_fetch(map, method_name, method_name_len, TRUE);
36         if ( SvOK(method_slot) ) {
37             SV *const body = mop_call0(aTHX_ method_slot, KEY_FOR(body)); /* $method_object->body() */
38             if ( SvROK(body) && ((CV *) SvRV(body)) == cv ) {
39                 continue;
40             }
41         }
42
43         method_metaclass_name = mop_call0(aTHX_ self, mop_method_metaclass); /* $self->method_metaclass() */
44
45         /*
46             $method_object = $method_metaclass->wrap(
47                 $cv,
48                 associated_metaclass => $self,
49                 package_name         => $class_name,
50                 name                 => $method_name
51             );
52         */
53         ENTER;
54         SAVETMPS;
55
56         PUSHMARK(SP);
57         EXTEND(SP, 8);
58         PUSHs(method_metaclass_name); /* invocant */
59         mPUSHs(newRV_inc((SV *)cv));
60         PUSHs(mop_associated_metaclass);
61         PUSHs(self);
62         PUSHs(KEY_FOR(package_name));
63         PUSHs(class_name);
64         PUSHs(KEY_FOR(name));
65         mPUSHs(newSVpv(method_name, method_name_len));
66         PUTBACK;
67
68         call_sv(mop_wrap, G_SCALAR | G_METHOD);
69         SPAGAIN;
70         method_object = POPs;
71         PUTBACK;
72         /* $map->{$method_name} = $method_object */
73         sv_setsv(method_slot, method_object);
74
75         FREETMPS;
76         LEAVE;
77     }
78 }
79
80 MODULE = Class::MOP::Class    PACKAGE = Class::MOP::Class
81
82 PROTOTYPES: DISABLE
83
84 void
85 get_method_map(self)
86     SV *self
87     PREINIT:
88         HV *const obj        = (HV *)SvRV(self);
89         SV *const class_name = HeVAL( hv_fetch_ent(obj, KEY_FOR(package), 0, HASH_FOR(package)) );
90         HV *const stash      = gv_stashsv(class_name, 0);
91         UV current;
92         SV *cache_flag;
93         SV *map_ref;
94     PPCODE:
95         if (!stash) {
96              mXPUSHs(newRV_noinc((SV *)newHV()));
97              return;
98         }
99
100         current    = mop_check_package_cache_flag(aTHX_ stash);
101         cache_flag = HeVAL( hv_fetch_ent(obj, KEY_FOR(package_cache_flag), TRUE, HASH_FOR(package_cache_flag)));
102         map_ref    = HeVAL( hv_fetch_ent(obj, KEY_FOR(methods), TRUE, HASH_FOR(methods)));
103
104         /* $self->{methods} does not yet exist (or got deleted) */
105         if ( !SvROK(map_ref) || SvTYPE(SvRV(map_ref)) != SVt_PVHV ) {
106             SV *new_map_ref = newRV_noinc((SV *)newHV());
107             sv_2mortal(new_map_ref);
108             sv_setsv(map_ref, new_map_ref);
109         }
110
111         if ( !SvOK(cache_flag) || SvUV(cache_flag) != current ) {
112             mop_update_method_map(aTHX_ self, class_name, stash, (HV *)SvRV(map_ref));
113             sv_setuv(cache_flag, mop_check_package_cache_flag(aTHX_ stash)); /* update_cache_flag() */
114         }
115
116         XPUSHs(map_ref);