Implement install_subroutines in XS
[gitmo/Mouse.git] / xs-src / MouseAttribute.xs
1 #include "mouse.h"
2
3 static MGVTBL mouse_xa_vtbl; /* identity */
4
5 static AV*
6 mouse_build_xa(pTHX_ SV* const attr) {
7     AV*    xa;
8     MAGIC* mg;
9
10     SV* slot;
11     STRLEN len;
12     const char* pv;
13     U16 flags = 0x00;
14
15     ENTER;
16     SAVETMPS;
17
18     xa    = newAV();
19
20     mg = sv_magicext(SvRV(attr), (SV*)xa, PERL_MAGIC_ext, &mouse_xa_vtbl, NULL, 0);
21     SvREFCNT_dec(xa); /* refcnt++ in sv_magicext */
22
23     av_extend(xa, MOUSE_XA_last - 1);
24
25     slot = mcall0(attr, mouse_name);
26     pv = SvPV_const(slot, len);
27     av_store(xa, MOUSE_XA_SLOT, newSVpvn_share(pv, len, 0U));
28
29     av_store(xa, MOUSE_XA_ATTRIBUTE, newSVsv(attr));
30
31     av_store(xa, MOUSE_XA_INIT_ARG, newSVsv(mcall0s(attr, "init_arg")));
32
33     if(predicate_calls(attr, "has_type_constraint")){
34         SV* tc;
35         flags |= MOUSEf_ATTR_HAS_TC;
36
37         tc = mcall0s(attr, "type_constraint");
38         av_store(xa, MOUSE_XA_TC, newSVsv(tc));
39
40         if(predicate_calls(attr, "should_auto_deref")){
41             SV* const is_a_type_of = sv_2mortal(newSVpvs_share("is_a_type_of"));
42
43             flags |= MOUSEf_ATTR_SHOULD_AUTO_DEREF;
44             if( sv_true(mcall1(tc, is_a_type_of, newSVpvs_flags("ArrayRef", SVs_TEMP))) ){
45                 flags |= MOUSEf_TC_IS_ARRAYREF;
46             }
47             else if( sv_true(mcall1(tc, is_a_type_of, newSVpvs_flags("HashRef", SVs_TEMP))) ){
48                 flags |= MOUSEf_TC_IS_HASHREF;
49             }
50             else{
51                 mouse_throw_error(attr, tc,
52                     "Can not auto de-reference the type constraint '%"SVf"'",
53                         mcall0(tc, mouse_name));
54             }
55         }
56
57         if(predicate_calls(attr, "should_coerce") && predicate_calls(tc, "has_coercion")){
58             flags |= MOUSEf_ATTR_SHOULD_COERCE;
59         }
60
61     }
62
63     if(predicate_calls(attr, "has_trigger")){
64         flags |= MOUSEf_ATTR_HAS_TRIGGER;
65     }
66
67     if(predicate_calls(attr, "is_lazy")){
68         flags |= MOUSEf_ATTR_IS_LAZY;
69     }
70     if(predicate_calls(attr, "has_builder")){
71         flags |= MOUSEf_ATTR_HAS_BUILDER;
72     }
73     else if(predicate_calls(attr, "has_default")){
74         flags |= MOUSEf_ATTR_HAS_DEFAULT;
75     }
76
77     if(predicate_calls(attr, "is_weak_ref")){
78         flags |= MOUSEf_ATTR_IS_WEAK_REF;
79     }
80
81     if(predicate_calls(attr, "is_required")){
82         flags |= MOUSEf_ATTR_IS_REQUIRED;
83     }
84
85     av_store(xa, MOUSE_XA_FLAGS, newSVuv(flags));
86     MOUSE_mg_flags(mg) = flags;
87
88     FREETMPS;
89     LEAVE;
90
91     return xa;
92 }
93
94 AV*
95 mouse_get_xa(pTHX_ SV* const attr) {
96     AV*    xa;
97     MAGIC* mg;
98
99     if(!IsObject(attr)){
100         croak("Not a Mouse meta attribute");
101     }
102
103     mg = mouse_mg_find(aTHX_ SvRV(attr), &mouse_xa_vtbl, 0x00);
104     if(!mg){
105         xa = mouse_build_xa(aTHX_ attr);
106     }
107     else{
108         xa = (AV*)MOUSE_mg_obj(mg);
109
110         assert(xa);
111         assert(SvTYPE(xa) == SVt_PVAV);
112     }
113
114     return xa;
115 }
116
117 SV*
118 mouse_xa_apply_type_constraint(pTHX_ AV* const xa, SV* value, U16 const flags){
119     SV* const tc = MOUSE_xa_tc(xa);
120     SV* tc_code;
121
122     if(flags & MOUSEf_ATTR_SHOULD_COERCE){
123           value = mcall1(tc, mouse_coerce, value);
124     }
125
126     if(!SvOK(MOUSE_xa_tc_code(xa))){
127         tc_code = mcall0s(tc, "_compiled_type_constraint");
128         av_store(xa, MOUSE_XA_TC_CODE, newSVsv(tc_code));
129
130         if(!IsCodeRef(tc_code)){
131             mouse_throw_error(MOUSE_xa_attribute(xa), tc, "Not a CODE reference");
132         }
133     }
134     else{
135         tc_code = MOUSE_xa_tc_code(xa);
136     }
137
138     if(!mouse_tc_check(aTHX_ tc_code, value)){
139         mouse_throw_error(MOUSE_xa_attribute(xa), value,
140             "Attribute (%"SVf") does not pass the type constraint because: %"SVf,
141                 mcall0(MOUSE_xa_attribute(xa), mouse_name),
142                 mcall1s(tc, "get_message", value));
143     }
144
145     return value;
146 }
147
148
149 SV*
150 mouse_xa_set_default(pTHX_ AV* const xa, SV* const object) {
151     U16 const flags = (U16)MOUSE_xa_flags(xa);
152     SV* value;
153
154     ENTER;
155     SAVETMPS;
156
157     /* get default value by $attr->builder or $attr->default */
158     if(flags & MOUSEf_ATTR_HAS_BUILDER){
159         SV* const builder = mcall0s(MOUSE_xa_attribute(xa), "builder");
160         value = mcall0(object, builder); /* $object->$builder() */
161     }
162     else {
163         value = mcall0s(MOUSE_xa_attribute(xa), "default");
164
165         if(IsCodeRef(value)){
166             value = mcall0(object, value);
167         }
168     }
169
170     /* apply coerce and type constraint */
171     if(flags & MOUSEf_ATTR_HAS_TC){
172         value = mouse_xa_apply_type_constraint(aTHX_ xa, value, flags);
173     }
174
175     /* store value to slot */
176     value = set_slot(object, MOUSE_xa_slot(xa), value);
177     if(flags & MOUSEf_ATTR_IS_WEAK_REF && SvROK(value)){
178         weaken_slot(object, MOUSE_xa_slot(xa));
179     }
180
181     FREETMPS;
182     LEAVE;
183
184     return value;
185 }
186
187 static void
188 mouse_check_isa_does_does(pTHX_ SV* const klass, SV* const name, SV* const isa, SV* const does){
189     STRLEN len;
190     const char* const pv = SvPV_const(isa, len); /* need strigify */
191     bool does_ok;
192     dSP;
193
194     ENTER;
195     SAVETMPS;
196
197     SAVESPTR(ERRSV);
198     ERRSV = sv_newmortal();
199
200     PUSHMARK(SP);
201     EXTEND(SP, 2);
202     mPUSHp(pv, len);
203     PUSHs(does);
204     PUTBACK;
205
206     call_method("does", G_EVAL | G_SCALAR);
207
208     SPAGAIN;
209     does_ok = sv_true(POPs);
210     PUTBACK;
211
212     FREETMPS;
213     LEAVE;
214
215     if(!does_ok){
216         mouse_throw_error(klass, NULL,
217             "Cannot have both an isa option and a does option"
218             "because '%"SVf"' does not do '%"SVf"' on attribute (%"SVf")",
219             isa, does, name
220         );
221     }
222 }
223
224 MODULE = Mouse::Meta::Attribute  PACKAGE = Mouse::Meta::Attribute
225
226 PROTOTYPES: DISABLE
227
228 BOOT:
229     /* readers */
230     INSTALL_SIMPLE_READER(Attribute, name);
231     INSTALL_SIMPLE_READER(Attribute, associated_class);
232     INSTALL_SIMPLE_READER(Attribute, accessor);
233     INSTALL_SIMPLE_READER(Attribute, reader);
234     INSTALL_SIMPLE_READER(Attribute, writer);
235     INSTALL_SIMPLE_READER(Attribute, predicate);
236     INSTALL_SIMPLE_READER(Attribute, clearer);
237     INSTALL_SIMPLE_READER(Attribute, handles);
238
239     INSTALL_SIMPLE_READER_WITH_KEY(Attribute, _is_metadata, is);
240     INSTALL_SIMPLE_READER_WITH_KEY(Attribute, is_required, required);
241     INSTALL_SIMPLE_READER(Attribute, default);
242     INSTALL_SIMPLE_READER_WITH_KEY(Attribute, is_lazy, lazy);
243     INSTALL_SIMPLE_READER_WITH_KEY(Attribute, is_lazy_build, lazy_build);
244     INSTALL_SIMPLE_READER_WITH_KEY(Attribute, is_weak_ref, weak_ref);
245     INSTALL_SIMPLE_READER(Attribute, init_arg);
246     INSTALL_SIMPLE_READER(Attribute, type_constraint);
247     INSTALL_SIMPLE_READER(Attribute, trigger);
248     INSTALL_SIMPLE_READER(Attribute, builder);
249     INSTALL_SIMPLE_READER_WITH_KEY(Attribute, should_auto_deref, auto_deref);
250     INSTALL_SIMPLE_READER_WITH_KEY(Attribute, should_coerce, coerce);
251     INSTALL_SIMPLE_READER(Attribute, documentation);
252
253     /* predicates */
254     INSTALL_SIMPLE_PREDICATE_WITH_KEY(Attribute, has_accessor, accessor);
255     INSTALL_SIMPLE_PREDICATE_WITH_KEY(Attribute, has_reader, reader);
256     INSTALL_SIMPLE_PREDICATE_WITH_KEY(Attribute, has_writer, writer);
257     INSTALL_SIMPLE_PREDICATE_WITH_KEY(Attribute, has_predicate, predicate);
258     INSTALL_SIMPLE_PREDICATE_WITH_KEY(Attribute, has_clearer, clearer);
259     INSTALL_SIMPLE_PREDICATE_WITH_KEY(Attribute, has_handles, handles);
260
261     INSTALL_SIMPLE_PREDICATE_WITH_KEY(Attribute, has_default, default);
262     INSTALL_SIMPLE_PREDICATE_WITH_KEY(Attribute, has_type_constraint, type_constraint);
263     INSTALL_SIMPLE_PREDICATE_WITH_KEY(Attribute, has_trigger, trigger);
264     INSTALL_SIMPLE_PREDICATE_WITH_KEY(Attribute, has_builder, builder);
265     INSTALL_SIMPLE_PREDICATE_WITH_KEY(Attribute, has_documentation, documentation);
266
267     INSTALL_CLASS_HOLDER(Attribute, accessor_metaclass, "Mouse::Meta::Method::Accessor::XS");
268
269 void
270 _process_options(SV* klass, SV* name, HV* args)
271 CODE:
272 {
273     SV** svp;
274     SV* tc = NULL;
275
276     /* 'required' requires eigher 'init_arg', 'builder', or 'default' */
277     bool can_be_required = FALSE;
278     bool has_default     = FALSE;
279     bool has_builder     = FALSE;
280
281     /* taken from Class::MOP::Attribute::new */
282
283     if(!SvOK(name)){
284         mouse_throw_error(klass, NULL,
285             "You must provide a name for the attribute");
286     }
287
288     svp = hv_fetchs(args, "init_arg", FALSE);
289     if(!svp){
290         (void)hv_stores(args, "init_arg", newSVsv(name));
291         can_be_required = TRUE;
292     }
293     else{
294         can_be_required = SvOK(*svp) ? TRUE : FALSE;
295     }
296
297     svp = hv_fetchs(args, "builder", FALSE);
298     if(svp){
299         if(!SvOK(*svp)){
300             mouse_throw_error(klass, *svp,
301                 "builder must be a defined scalar value which is a method name");
302         }
303         can_be_required = TRUE;
304         has_builder     = TRUE;
305     }
306     else if((svp = hv_fetchs(args, "default", FALSE))){
307         if(SvROK(*svp) && SvTYPE(SvRV(*svp)) != SVt_PVCV) {
308             mouse_throw_error(klass, *svp,
309                "References are not allowed as default values, you must "
310                 "wrap the default of '%"SVf"' in a CODE reference "
311                 "(ex: sub { [] } and not [])", name);
312         }
313         can_be_required = TRUE;
314         has_default     = TRUE;
315     }
316
317     svp = hv_fetchs(args, "required", FALSE);
318     if( (svp && sv_true(*svp)) && !can_be_required){
319         mouse_throw_error(klass, NULL,
320             "You cannot have a required attribute (%"SVf") "
321             "without a default, builder, or an init_arg", name);
322     }
323
324      /* taken from Mouse::Meta::Attribute->new and ->_process_args */
325
326     svp = hv_fetchs(args, "is", FALSE);
327     if(svp){
328         const char* const is = SvOK(*svp) ? SvPV_nolen_const(*svp) : "undef";
329         if(strEQ(is, "ro")){
330             svp = hv_fetchs(args, "reader", TRUE);
331             if(!sv_true(*svp)){
332                 sv_setsv(*svp, name);
333             }
334         }
335         else if(strEQ(is, "rw")){
336             if(hv_fetchs(args, "writer", FALSE)){
337                 svp = hv_fetchs(args, "reader", TRUE);
338             }
339             else{
340                 svp = hv_fetchs(args, "accessor", TRUE);
341             }
342             sv_setsv(*svp, name);
343         }
344         else if(strEQ(is, "bare")){
345             /* do nothing, but don't complain (later) about missing methods */
346         }
347         else{
348             mouse_throw_error(klass, NULL,
349                 "I do not understand this option (is => %s) on attribute (%"SVf")",
350                 is, name);
351         }
352     }
353
354     svp = hv_fetchs(args, "isa", FALSE);
355     if(svp){
356         SPAGAIN;
357         PUSHMARK(SP);
358         XPUSHs(*svp);
359         PUTBACK;
360
361         call_pv("Mouse::Util::TypeConstraints::find_or_create_isa_type_constraint",
362             G_SCALAR);
363         SPAGAIN;
364         tc = newSVsv(POPs);
365         PUTBACK;
366     }
367
368     if((svp = hv_fetchs(args, "does", FALSE))){
369         /* check 'isa' does 'does' */
370         if(tc){
371             mouse_check_isa_does_does(aTHX_ klass, name, tc, *svp);
372             /* nothing to do */
373         }
374         else{
375             SPAGAIN;
376             PUSHMARK(SP);
377             XPUSHs(*svp);
378             PUTBACK;
379
380             call_pv("Mouse::Util::TypeConstraints::find_or_create_does_type_constraint",
381                 G_SCALAR);
382             SPAGAIN;
383             tc = newSVsv(POPs);
384             PUTBACK;
385         }
386     }
387     if(tc){
388         (void)hv_stores(args, "type_constraint", tc);
389     }
390
391     svp = hv_fetchs(args, "coerce", FALSE);
392     if(svp){
393         if(!tc){
394             mouse_throw_error(klass, NULL,
395                 "You cannot have coercion without specifying a type constraint "
396                 "on attribute (%"SVf")", name);
397         }
398         svp = hv_fetchs(args, "weak_ref", FALSE);
399         if(svp && sv_true(*svp)){
400             mouse_throw_error(klass, NULL,
401                 "You cannot have a weak reference to a coerced value on "
402                 "attribute (%"SVf")", name);
403         }
404     }
405
406     svp = hv_fetchs(args, "lazy_build", FALSE);
407     if(svp){
408         SV* clearer;
409         SV* predicate;
410         if(has_default){
411             mouse_throw_error(klass, NULL,
412                 "You can not use lazy_build and default for the same "
413                 "attribute (%"SVf")", name);
414         }
415
416         svp = hv_fetchs(args, "lazy", TRUE);
417         sv_setiv(*svp, TRUE);
418
419         svp = hv_fetchs(args, "builder", TRUE);
420         if(!sv_true(*svp)){
421             sv_setpvf(*svp, "_build_%"SVf, name);
422         }
423         has_builder = TRUE;
424
425         clearer   = *hv_fetchs(args, "clearer",   TRUE);
426         predicate = *hv_fetchs(args, "predicate", TRUE);
427
428         if(SvPV_nolen_const(name)[0] == '_'){
429             if(!sv_true(clearer)){
430                 sv_setpvf(clearer, "_clear%"SVf, name);
431             }
432             if(!sv_true(predicate)){
433                 sv_setpvf(predicate, "_has%"SVf, name);
434             }
435         }
436         else{
437             if(!sv_true(clearer)){
438                 sv_setpvf(clearer, "clear_%"SVf, name);
439             }
440             if(!sv_true(predicate)){
441                 sv_setpvf(predicate, "has_%"SVf, name);
442             }
443         }
444     }
445
446     svp = hv_fetchs(args, "auto_deref", FALSE);
447     if(svp && sv_true(*svp)){
448         SV* const meth = sv_2mortal(newSVpvs_share("is_a_type_of"));
449         if(!tc){
450             mouse_throw_error(klass, NULL,
451                 "You cannot auto-dereference without specifying a type "
452                 "constraint on attribute (%"SVf")", name);
453         }
454
455         if(!(sv_true(mcall1(tc, meth, newSVpvs_flags("ArrayRef", SVs_TEMP)))
456             || sv_true(mcall1(tc, meth, newSVpvs_flags("HashRef", SVs_TEMP))) )){
457             mouse_throw_error(klass, NULL,
458                 "You cannot auto-dereference anything other than a ArrayRef "
459                 "or HashRef on attribute (%"SVf")", name);
460         }
461     }
462
463     svp = hv_fetchs(args, "trigger", FALSE);
464     if(svp){
465         if(!IsCodeRef(*svp)){
466             mouse_throw_error(klass, NULL,
467                 "Trigger must be a CODE ref on attribute (%"SVf")",
468                 name);
469         }
470     }
471
472
473     svp = hv_fetchs(args, "lazy", FALSE);
474     if(svp && sv_true(*svp)){
475         if(!(has_default || has_builder)){
476             mouse_throw_error(klass, NULL,
477                 "You cannot have lazy attribute (%"SVf") without specifying "
478                 "a default value for it", name);
479         }
480     }
481 }