$type_constraint->check() accepts extra arguments
[gitmo/Mouse.git] / xs-src / MouseTypeConstraints.xs
CommitLineData
7d96ae4d 1/*
6e647cac 2 * TypeConstraint stuff
3 * - Mouse::Util::TypeConstraints (including OptimizedConstraionts)
4 * - Mouse::Meta::TypeConstraint
7d96ae4d 5 */
6
7#include "mouse.h"
8
7d96ae4d 9#ifndef SvRXOK
10#define SvRXOK(sv) (SvROK(sv) && SvMAGICAL(SvRV(sv)) && mg_find(SvRV(sv), PERL_MAGIC_qr))
11#endif
12
d33d8840 13typedef int (*check_fptr_t)(pTHX_ SV* const data, SV* const sv);
14
91ee66cb 15static
16XSPROTO(XS_Mouse_constraint_check);
17
85a74fcf 18/*
19 NOTE: mouse_tc_check() handles GETMAGIC
20*/
1d5ecd5f 21int
22mouse_tc_check(pTHX_ SV* const tc_code, SV* const sv) {
28ba5a26 23 CV* const cv = (CV*)SvRV(tc_code);
7ca3334f 24 assert(SvTYPE(cv) == SVt_PVCV);
d33d8840 25
19462c57 26 if(CvXSUB(cv) == XS_Mouse_constraint_check){ /* built-in type constraints */
7ca3334f 27 MAGIC* const mg = (MAGIC*)CvXSUBANY(cv).any_ptr;
d33d8840 28
7ca3334f 29 assert(CvXSUBANY(cv).any_ptr != NULL);
19462c57 30 assert(mg->mg_ptr != NULL);
28ba5a26 31
a901dc2e 32 SvGETMAGIC(sv);
7ca3334f 33 /* call the check function directly, skipping call_sv() */
34 return CALL_FPTR((check_fptr_t)mg->mg_ptr)(aTHX_ mg->mg_obj, sv);
1d5ecd5f 35 }
7ca3334f 36 else { /* custom */
1d5ecd5f 37 int ok;
38 dSP;
39
40 ENTER;
41 SAVETMPS;
42
43 PUSHMARK(SP);
44 XPUSHs(sv);
45 PUTBACK;
46
47 call_sv(tc_code, G_SCALAR);
48
49 SPAGAIN;
6ad77996 50 ok = sv_true(POPs);
1d5ecd5f 51 PUTBACK;
52
53 FREETMPS;
54 LEAVE;
55
56 return ok;
57 }
58}
7d96ae4d 59
7d96ae4d 60/*
3f1bd57f 61 The following type check functions return an integer, not a bool, to keep
62 the code simple,
63 so if you assign these return value to a bool variable, you must use
64 "expr ? TRUE : FALSE".
7d96ae4d 65*/
66
67int
19462c57 68mouse_tc_Any(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv PERL_UNUSED_DECL) {
7d96ae4d 69 assert(sv);
70 return TRUE;
71}
72
73int
19462c57 74mouse_tc_Bool(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 75 assert(sv);
f6e3b8a9 76
6ad77996 77 if(sv_true(sv)){
7f881031 78 if(SvPOKp(sv)){ /* "1" */
79 return SvCUR(sv) == 1 && SvPVX(sv)[0] == '1';
80 }
81 else if(SvIOKp(sv)){
f6e3b8a9 82 return SvIVX(sv) == 1;
7d96ae4d 83 }
84 else if(SvNOKp(sv)){
f6e3b8a9 85 return SvNVX(sv) == 1.0;
7d96ae4d 86 }
7d96ae4d 87 else{
88 return FALSE;
89 }
90 }
91 else{
7f881031 92 /* any false value is a boolean */
7d96ae4d 93 return TRUE;
94 }
95}
96
97int
19462c57 98mouse_tc_Undef(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 99 assert(sv);
100 return !SvOK(sv);
101}
102
103int
19462c57 104mouse_tc_Defined(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 105 assert(sv);
106 return SvOK(sv);
107}
108
109int
19462c57 110mouse_tc_Value(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 111 assert(sv);
112 return SvOK(sv) && !SvROK(sv);
113}
114
115int
19462c57 116mouse_tc_Num(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 117 assert(sv);
118 return LooksLikeNumber(sv);
119}
120
678d1902 121static int
0cfe08a4 122S_nv_is_integer(pTHX_ NV const nv) {
123 if(nv == (NV)(IV)nv){
124 return TRUE;
125 }
126 else {
127 char buf[64]; /* Must fit sprintf/Gconvert of longest NV */
3f1bd57f 128 const char* p;
f3e4c7eb 129 (void)Gconvert(nv, NV_DIG, 0, buf);
0cfe08a4 130 p = &buf[0];
131
132 /* -?[0-9]+ */
133 if(*p == '-') p++;
134
135 while(*p){
136 if(!isDIGIT(*p)){
137 return FALSE;
138 }
139 p++;
140 }
141 return TRUE;
142 }
143}
144
145int
19462c57 146mouse_tc_Int(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 147 assert(sv);
7f881031 148 if(SvPOKp(sv)){
149 int const num_type = grok_number(SvPVX(sv), SvCUR(sv), NULL);
150 return num_type && !(num_type & IS_NUMBER_NOT_INT);
151 }
152 else if(SvIOKp(sv)){
7d96ae4d 153 return TRUE;
154 }
9ab5eba8 155 else if(SvNOKp(sv)) {
0cfe08a4 156 return S_nv_is_integer(aTHX_ SvNVX(sv));
7d96ae4d 157 }
7d96ae4d 158 return FALSE;
159}
160
161int
19462c57 162mouse_tc_Str(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 163 assert(sv);
164 return SvOK(sv) && !SvROK(sv) && !isGV(sv);
165}
166
167int
7f881031 168mouse_tc_ClassName(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv){
7d96ae4d 169 assert(sv);
170 return is_class_loaded(sv);
171}
172
173int
19462c57 174mouse_tc_RoleName(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 175 assert(sv);
176 if(is_class_loaded(sv)){
177 int ok;
7d96ae4d 178
179 ENTER;
180 SAVETMPS;
181
3f1bd57f 182 ok = is_an_instance_of("Mouse::Meta::Role", get_metaclass(sv));
7d96ae4d 183
184 FREETMPS;
185 LEAVE;
186
187 return ok;
188 }
189 return FALSE;
190}
191
192int
19462c57 193mouse_tc_Ref(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 194 assert(sv);
195 return SvROK(sv);
196}
197
198int
bcf612a8 199mouse_tc_ScalarRef(pTHX_ SV* const data PERL_UNUSED_DECL, SV* sv) {
7d96ae4d 200 assert(sv);
bcf612a8 201 if(SvROK(sv)){
202 sv = SvRV(sv);
203 return !SvOBJECT(sv) && (SvTYPE(sv) <= SVt_PVLV && !isGV(sv));
204 }
205 return FALSE;
7d96ae4d 206}
207
208int
19462c57 209mouse_tc_ArrayRef(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 210 assert(sv);
80aa5731 211 return IsArrayRef(sv);
7d96ae4d 212}
213
214int
19462c57 215mouse_tc_HashRef(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 216 assert(sv);
80aa5731 217 return IsHashRef(sv);
7d96ae4d 218}
219
220int
19462c57 221mouse_tc_CodeRef(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 222 assert(sv);
80aa5731 223 return IsCodeRef(sv);
7d96ae4d 224}
225
226int
19462c57 227mouse_tc_RegexpRef(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 228 assert(sv);
229 return SvRXOK(sv);
230}
231
232int
19462c57 233mouse_tc_GlobRef(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 234 assert(sv);
235 return SvROK(sv) && !SvOBJECT(SvRV(sv)) && isGV(SvRV(sv));
236}
237
238int
19462c57 239mouse_tc_FileHandle(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 240 GV* gv;
241 assert(sv);
242
243 /* see pp_fileno() in pp_sys.c and Scalar::Util::openhandle() */
244
245 gv = (GV*)(SvROK(sv) ? SvRV(sv) : sv);
246 if(isGV(gv) || SvTYPE(gv) == SVt_PVIO){
247 IO* const io = isGV(gv) ? GvIO(gv) : (IO*)gv;
248
249 if(io && ( IoIFP(io) || SvTIED_mg((SV*)io, PERL_MAGIC_tiedscalar) )){
250 return TRUE;
251 }
252 }
253
016e4681 254 return is_an_instance_of("IO::Handle", sv);
7d96ae4d 255}
256
257int
19462c57 258mouse_tc_Object(pTHX_ SV* const data PERL_UNUSED_DECL, SV* const sv) {
7d96ae4d 259 assert(sv);
260 return SvROK(sv) && SvOBJECT(SvRV(sv)) && !SvRXOK(sv);
261}
262
619338ac 263/* Parameterized type constraints */
264
f790c46b 265static int
619338ac 266mouse_parameterized_ArrayRef(pTHX_ SV* const param, SV* const sv) {
80aa5731 267 if(IsArrayRef(sv)){
619338ac 268 AV* const av = (AV*)SvRV(sv);
269 I32 const len = av_len(av) + 1;
270 I32 i;
271 for(i = 0; i < len; i++){
272 SV* const value = *av_fetch(av, i, TRUE);
619338ac 273 if(!mouse_tc_check(aTHX_ param, value)){
274 return FALSE;
275 }
276 }
277 return TRUE;
278 }
279 return FALSE;
280}
281
f790c46b 282static int
619338ac 283mouse_parameterized_HashRef(pTHX_ SV* const param, SV* const sv) {
3f1bd57f 284 if(IsHashRef(sv)){
619338ac 285 HV* const hv = (HV*)SvRV(sv);
286 HE* he;
287
288 hv_iterinit(hv);
289 while((he = hv_iternext(hv))){
290 SV* const value = hv_iterval(hv, he);
619338ac 291 if(!mouse_tc_check(aTHX_ param, value)){
a1cf0531 292 hv_iterinit(hv); /* reset */
619338ac 293 return FALSE;
294 }
295 }
296 return TRUE;
297 }
298 return FALSE;
299}
300
f790c46b 301static int
619338ac 302mouse_parameterized_Maybe(pTHX_ SV* const param, SV* const sv) {
303 if(SvOK(sv)){
304 return mouse_tc_check(aTHX_ param, sv);
305 }
306 return TRUE;
307}
308
f790c46b 309static int
310mouse_types_union_check(pTHX_ AV* const types, SV* const sv) {
311 I32 const len = AvFILLp(types) + 1;
312 I32 i;
313
314 for(i = 0; i < len; i++){
315 if(mouse_tc_check(aTHX_ AvARRAY(types)[i], sv)){
316 return TRUE;
317 }
318 }
319
320 return FALSE;
321}
322
323static int
324mouse_types_check(pTHX_ AV* const types, SV* const sv) {
325 I32 const len = AvFILLp(types) + 1;
326 I32 i;
327
328 ENTER;
329 SAVE_DEFSV;
330 DEFSV_set(sv);
331
332 for(i = 0; i < len; i++){
333 if(!mouse_tc_check(aTHX_ AvARRAY(types)[i], sv)){
334 LEAVE;
335 return FALSE;
336 }
337 }
338
339 LEAVE;
340
341 return TRUE;
342}
343
1d5ecd5f 344/*
345 * This class_type generator is taken from Scalar::Util::Instance
346 */
347
348#define MY_CXT_KEY "Mouse::Util::TypeConstraints::_guts" XS_VERSION
349typedef struct sui_cxt{
350 GV* universal_isa;
ebe91068 351 GV* universal_can;
1d5ecd5f 352} my_cxt_t;
353START_MY_CXT
354
355#define MG_klass_stash(mg) ((HV*)(mg)->mg_obj)
356#define MG_klass_pv(mg) ((mg)->mg_ptr)
357#define MG_klass_len(mg) ((mg)->mg_len)
358
441964ce 359static const char*
f48920c1 360mouse_canonicalize_package_name(const char* name){
1d5ecd5f 361
362 /* "::Foo" -> "Foo" */
363 if(name[0] == ':' && name[1] == ':'){
364 name += 2;
365 }
366
367 /* "main::main::main::Foo" -> "Foo" */
368 while(strnEQ(name, "main::", sizeof("main::")-1)){
369 name += sizeof("main::")-1;
370 }
371
372 return name;
373}
374
375static int
441964ce 376mouse_lookup_isa(pTHX_ HV* const instance_stash, const char* const klass_pv){
1d5ecd5f 377 AV* const linearized_isa = mro_get_linear_isa(instance_stash);
378 SV** svp = AvARRAY(linearized_isa);
379 SV** const end = svp + AvFILLp(linearized_isa) + 1;
380
381 while(svp != end){
382 assert(SvPVX(*svp));
f48920c1 383 if(strEQ(klass_pv, mouse_canonicalize_package_name(SvPVX(*svp)))){
1d5ecd5f 384 return TRUE;
385 }
386 svp++;
387 }
388 return FALSE;
389}
390
ce89cb68 391#define find_method_pvn(a, b, c) mouse_stash_find_method(aTHX_ a, b, c)
392#define find_method_pvs(a, b) mouse_stash_find_method(aTHX_ a, STR_WITH_LEN(b))
393
fc650569 394STATIC_INLINE GV*
ce89cb68 395mouse_stash_find_method(pTHX_ HV* const stash, const char* const name, I32 const namelen){
396 GV** const gvp = (GV**)hv_fetch(stash, name, namelen, FALSE);
397 if(gvp && isGV(*gvp) && GvCV(*gvp)){ /* shortcut */
398 return *gvp;
399 }
400
401 return gv_fetchmeth_autoload(stash, name, namelen, 0);
402}
403
441964ce 404int
405mouse_is_an_instance_of(pTHX_ HV* const stash, SV* const instance){
406 assert(stash);
407 assert(SvTYPE(stash) == SVt_PVHV);
408
016e4681 409 if(IsObject(instance)){
441964ce 410 dMY_CXT;
411 HV* const instance_stash = SvSTASH(SvRV(instance));
28e3abdf 412 GV* const myisa = find_method_pvs(instance_stash, "isa");
441964ce 413
414 /* the instance has no own isa method */
28e3abdf 415 if(myisa == NULL || GvCV(myisa) == GvCV(MY_CXT.universal_isa)){
441964ce 416 return stash == instance_stash
417 || mouse_lookup_isa(aTHX_ instance_stash, HvNAME_get(stash));
418 }
419 /* the instance has its own isa method */
420 else {
28e3abdf 421 SV* package;
422 int ok;
1d5ecd5f 423
441964ce 424 ENTER;
425 SAVETMPS;
1d5ecd5f 426
28e3abdf 427 package = newSVpvn_share(HvNAME_get(stash), HvNAMELEN_get(stash), 0U);
428 ok = sv_true(mcall1s(instance, "isa", sv_2mortal(package)));
1d5ecd5f 429
441964ce 430 FREETMPS;
431 LEAVE;
1d5ecd5f 432
28e3abdf 433 return ok;
441964ce 434 }
1d5ecd5f 435 }
441964ce 436 return FALSE;
437}
438
439static int
440mouse_is_an_instance_of_universal(pTHX_ SV* const data, SV* const sv){
441 PERL_UNUSED_ARG(data);
442 return SvROK(sv) && SvOBJECT(SvRV(sv));
1d5ecd5f 443}
444
ebe91068 445static int
446mouse_can_methods(pTHX_ AV* const methods, SV* const instance){
447 if(IsObject(instance)){
448 dMY_CXT;
449 HV* const mystash = SvSTASH(SvRV(instance));
ce89cb68 450 GV* const mycan = find_method_pvs(mystash, "can");
451 bool const use_builtin = (mycan == NULL || GvCV(mycan) == GvCV(MY_CXT.universal_can)) ? TRUE : FALSE;
ebe91068 452 I32 const len = AvFILLp(methods) + 1;
453 I32 i;
454 for(i = 0; i < len; i++){
455 SV* const name = MOUSE_av_at(methods, i);
456
457 if(use_builtin){
ce89cb68 458 if(!find_method_pvn(mystash, SvPVX(name), SvCUR(name))){
ebe91068 459 return FALSE;
460 }
461 }
462 else{
463 bool ok;
ebe91068 464
465 ENTER;
466 SAVETMPS;
467
28e3abdf 468 ok = sv_true(mcall1s(instance, "can", sv_mortalcopy(name)));
ebe91068 469
470 FREETMPS;
471 LEAVE;
472
473 if(!ok){
474 return FALSE;
475 }
476 }
477 }
478 return TRUE;
479 }
480 return FALSE;
481}
482
441964ce 483static MGVTBL mouse_util_type_constraints_vtbl; /* not used, only for identity */
f48920c1 484
619338ac 485static CV*
19462c57 486mouse_tc_generate(pTHX_ const char* const name, check_fptr_t const fptr, SV* const param) {
619338ac 487 CV* xsub;
916eb82f 488
19462c57 489 xsub = newXS(name, XS_Mouse_constraint_check, __FILE__);
916eb82f 490 CvXSUBANY(xsub).any_ptr = sv_magicext(
491 (SV*)xsub,
492 param, /* mg_obj: refcnt will be increased */
493 PERL_MAGIC_ext,
494 &mouse_util_type_constraints_vtbl,
1b364415 495 (char*)fptr, /* mg_ptr */
916eb82f 496 0 /* mg_len: 0 for static data */
497 );
498
499 if(!name){
500 sv_2mortal((SV*)xsub);
501 }
502
503 return xsub;
504}
505
506CV*
672bde7c 507mouse_generate_isa_predicate_for(pTHX_ SV* const klass, const char* const predicate_name){
441964ce 508 STRLEN klass_len;
509 const char* klass_pv = SvPV_const(klass, klass_len);
916eb82f 510 SV* param;
1b364415 511 check_fptr_t fptr;
1d5ecd5f 512
441964ce 513 klass_pv = mouse_canonicalize_package_name(klass_pv);
1d5ecd5f 514
441964ce 515 if(strNE(klass_pv, "UNIVERSAL")){
916eb82f 516 param = (SV*)gv_stashpvn(klass_pv, klass_len, GV_ADD);
1b364415 517 fptr = (check_fptr_t)mouse_is_an_instance_of;
441964ce 518
519 }
520 else{
916eb82f 521 param = NULL;
1b364415 522 fptr = (check_fptr_t)mouse_is_an_instance_of_universal;
441964ce 523 }
524
19462c57 525 return mouse_tc_generate(aTHX_ predicate_name, fptr, param);
441964ce 526}
f48920c1 527
ebe91068 528CV*
529mouse_generate_can_predicate_for(pTHX_ SV* const methods, const char* const predicate_name){
530 AV* av;
531 AV* const param = newAV_mortal();
532 I32 len;
533 I32 i;
534
d06d9266 535 must_ref(methods, "an ARRAY ref for method names", SVt_PVAV);
ebe91068 536 av = (AV*)SvRV(methods);
537
538 len = av_len(av) + 1;
539 for(i = 0; i < len; i++){
540 SV* const name = *av_fetch(av, i, TRUE);
541 STRLEN pvlen;
542 const char* const pv = SvPV_const(name, pvlen);
543
544 av_push(param, newSVpvn_share(pv, pvlen, 0U));
545 }
546
547 return mouse_tc_generate(aTHX_ predicate_name, (check_fptr_t)mouse_can_methods, (SV*)param);
548}
549
91ee66cb 550static
551XSPROTO(XS_Mouse_constraint_check) {
1d5ecd5f 552 dVAR;
553 dXSARGS;
441964ce 554 MAGIC* const mg = (MAGIC*)XSANY.any_ptr;
6e1e4889 555 SV* sv;
1d5ecd5f 556
441964ce 557 if(items < 1){
19462c57 558 croak("Too few arguments for type constraint check functions");
1d5ecd5f 559 }
560
6e1e4889 561 sv = ST(0);
562 SvGETMAGIC(sv);
563 ST(0) = boolSV( CALL_FPTR((check_fptr_t)mg->mg_ptr)(aTHX_ mg->mg_obj, sv) );
1d5ecd5f 564 XSRETURN(1);
565}
566
91ee66cb 567static
568XSPROTO(XS_Mouse_TypeConstraint_fallback) {
6e647cac 569 dXSARGS;
570 PERL_UNUSED_VAR(cv);
571 PERL_UNUSED_VAR(items);
572 XSRETURN_EMPTY;
573}
574
1d5ecd5f 575static void
576setup_my_cxt(pTHX_ pMY_CXT){
577 MY_CXT.universal_isa = gv_fetchpvs("UNIVERSAL::isa", GV_ADD, SVt_PVCV);
578 SvREFCNT_inc_simple_void_NN(MY_CXT.universal_isa);
ebe91068 579
580 MY_CXT.universal_can = gv_fetchpvs("UNIVERSAL::can", GV_ADD, SVt_PVCV);
581 SvREFCNT_inc_simple_void_NN(MY_CXT.universal_can);
1d5ecd5f 582}
583
19462c57 584#define DEFINE_TC(name) mouse_tc_generate(aTHX_ "Mouse::Util::TypeConstraints::" STRINGIFY(name), CAT2(mouse_tc_, name), NULL)
585
6e647cac 586#define MTC_CLASS "Mouse::Meta::TypeConstraint"
587
1d5ecd5f 588MODULE = Mouse::Util::TypeConstraints PACKAGE = Mouse::Util::TypeConstraints
589
646c0371 590PROTOTYPES: DISABLE
591VERSIONCHECK: DISABLE
1d5ecd5f 592
593BOOT:
594{
595 MY_CXT_INIT;
596 setup_my_cxt(aTHX_ aMY_CXT);
19462c57 597
598 /* setup built-in type constraints */
599 DEFINE_TC(Any);
600 DEFINE_TC(Undef);
601 DEFINE_TC(Defined);
602 DEFINE_TC(Bool);
603 DEFINE_TC(Value);
604 DEFINE_TC(Ref);
605 DEFINE_TC(Str);
606 DEFINE_TC(Num);
607 DEFINE_TC(Int);
608 DEFINE_TC(ScalarRef);
609 DEFINE_TC(ArrayRef);
610 DEFINE_TC(HashRef);
611 DEFINE_TC(CodeRef);
612 DEFINE_TC(GlobRef);
613 DEFINE_TC(FileHandle);
614 DEFINE_TC(RegexpRef);
615 DEFINE_TC(Object);
616 DEFINE_TC(ClassName);
617 DEFINE_TC(RoleName);
1d5ecd5f 618}
619
620#ifdef USE_ITHREADS
621
622void
623CLONE(...)
624CODE:
625{
626 MY_CXT_CLONE;
627 setup_my_cxt(aTHX_ aMY_CXT);
628 PERL_UNUSED_VAR(items);
629}
630
631#endif /* !USE_ITHREADS */
632
19462c57 633#define MOUSE_TC_MAYBE 0
634#define MOUSE_TC_ARRAY_REF 1
635#define MOUSE_TC_HASH_REF 2
1d5ecd5f 636
619338ac 637CV*
638_parameterize_ArrayRef_for(SV* param)
639ALIAS:
640 _parameterize_ArrayRef_for = MOUSE_TC_ARRAY_REF
641 _parameterize_HashRef_for = MOUSE_TC_HASH_REF
642 _parameterize_Maybe_for = MOUSE_TC_MAYBE
643CODE:
644{
645 check_fptr_t fptr;
646 SV* const tc_code = mcall0s(param, "_compiled_type_constraint");
80aa5731 647 if(!IsCodeRef(tc_code)){
619338ac 648 croak("_compiled_type_constraint didn't return a CODE reference");
649 }
650
651 switch(ix){
652 case MOUSE_TC_ARRAY_REF:
653 fptr = mouse_parameterized_ArrayRef;
654 break;
655 case MOUSE_TC_HASH_REF:
656 fptr = mouse_parameterized_HashRef;
657 break;
658 default: /* Maybe type */
659 fptr = mouse_parameterized_Maybe;
660 }
19462c57 661 RETVAL = mouse_tc_generate(aTHX_ NULL, fptr, tc_code);
619338ac 662}
663OUTPUT:
664 RETVAL
665
f790c46b 666MODULE = Mouse::Util::TypeConstraints PACKAGE = Mouse::Meta::TypeConstraint
667
a4b901ac 668BOOT:
669 INSTALL_SIMPLE_READER(TypeConstraint, name);
670 INSTALL_SIMPLE_READER(TypeConstraint, parent);
671 INSTALL_SIMPLE_READER(TypeConstraint, message);
672
fc83f4cf 673 INSTALL_SIMPLE_READER(TypeConstraint, type_parameter);
674
a4b901ac 675 INSTALL_SIMPLE_READER_WITH_KEY(TypeConstraint, _compiled_type_constraint, compiled_type_constraint);
a4b901ac 676
677 INSTALL_SIMPLE_PREDICATE_WITH_KEY(TypeConstraint, has_coercion, _compiled_type_coercion);
fc83f4cf 678 INSTALL_SIMPLE_PREDICATE_WITH_KEY(TypeConstraint, __is_parameterized, type_parameter); /* Mouse specific */
a4b901ac 679
6e647cac 680 /* overload stuff */
681 PL_amagic_generation++;
682 (void)newXS( MTC_CLASS "::()",
683 XS_Mouse_TypeConstraint_fallback, file);
684
685 /* fallback => 1 */
686 sv_setsv(
687 get_sv( MTC_CLASS "::()", GV_ADD ),
688 &PL_sv_yes
689 );
690
691 /* '""' => '_as_string' */
692 {
693 SV* const code_ref = sv_2mortal(newRV_inc(
694 (SV*)get_cv( MTC_CLASS "::_as_string", GV_ADD )));
695 sv_setsv_mg(
696 (SV*)gv_fetchpvs( MTC_CLASS "::(\"\"", GV_ADDMULTI, SVt_PVCV ),
697 code_ref );
698 }
7f881031 699
6e647cac 700 /* '0+' => '_identity' */
701 {
702 SV* const code_ref = sv_2mortal(newRV_inc(
703 (SV*)get_cv( MTC_CLASS "::_identity", GV_ADD )));
704 sv_setsv_mg(
705 (SV*)gv_fetchpvs( MTC_CLASS "::(0+", GV_ADDMULTI, SVt_PVCV ),
706 code_ref );
707 }
708
709 /* '|' => '_unite' */
710 {
711 SV* const code_ref = sv_2mortal(newRV_inc(
712 (SV*)get_cv( MTC_CLASS "::_unite", GV_ADD )));
713 sv_setsv_mg(
714 (SV*)gv_fetchpvs( MTC_CLASS "::(|", GV_ADDMULTI, SVt_PVCV ),
715 code_ref );
716 }
717
f6c81f00 718UV
719_identity(SV* self, ...)
720CODE:
721{
722 if(!SvROK(self)) {
723 croak("Invalid object instance: '%"SVf"'", self);
724 }
725 RETVAL = PTR2UV(SvRV(self));
726}
727OUTPUT:
728 RETVAL
6e647cac 729
f790c46b 730void
731compile_type_constraint(SV* self)
732CODE:
733{
6fab29c6 734 AV* const checks = newAV_mortal();
f790c46b 735 SV* check; /* check function */
736 SV* parent;
737 SV* types_ref;
738
f790c46b 739 for(parent = get_slots(self, "parent"); parent; parent = get_slots(parent, "parent")){
740 check = get_slots(parent, "hand_optimized_type_constraint");
741 if(check && SvOK(check)){
3f1bd57f 742 if(!IsCodeRef(check)){
f790c46b 743 croak("Not a CODE reference");
744 }
745 av_unshift(checks, 1);
746 av_store(checks, 0, newSVsv(check));
747 break; /* a hand optimized constraint must include all the parent */
748 }
749
750 check = get_slots(parent, "constraint");
751 if(check && SvOK(check)){
19462c57 752 if(!mouse_tc_CodeRef(aTHX_ NULL, check)){
f790c46b 753 croak("Not a CODE reference");
754 }
755 av_unshift(checks, 1);
756 av_store(checks, 0, newSVsv(check));
757 }
758 }
759
760 check = get_slots(self, "constraint");
761 if(check && SvOK(check)){
19462c57 762 if(!mouse_tc_CodeRef(aTHX_ NULL, check)){
f790c46b 763 croak("Not a CODE reference");
764 }
765 av_push(checks, newSVsv(check));
766 }
767
768 types_ref = get_slots(self, "type_constraints");
769 if(types_ref && SvOK(types_ref)){ /* union type */
770 AV* types;
771 AV* union_checks;
772 CV* union_check;
773 I32 len;
774 I32 i;
775
80aa5731 776 if(!IsArrayRef(types_ref)){
f790c46b 777 croak("Not an ARRAY reference");
778 }
779 types = (AV*)SvRV(types_ref);
780 len = av_len(types) + 1;
781
6fab29c6 782 union_checks = newAV_mortal();
f790c46b 783
784 for(i = 0; i < len; i++){
785 SV* const tc = *av_fetch(types, i, TRUE);
786 SV* const c = get_slots(tc, "compiled_type_constraint");
19462c57 787 if(!(c && mouse_tc_CodeRef(aTHX_ NULL, c))){
c7576321 788 mouse_throw_error(self, c, "'%"SVf"' has no compiled type constraint", self);
f790c46b 789 }
790 av_push(union_checks, newSVsv(c));
791 }
792
19462c57 793 union_check = mouse_tc_generate(aTHX_ NULL, (check_fptr_t)mouse_types_union_check, (SV*)union_checks);
f790c46b 794 av_push(checks, newRV_inc((SV*)union_check));
795 }
796
797 if(AvFILLp(checks) < 0){
798 check = newRV_inc((SV*)get_cv("Mouse::Util::TypeConstraints::Any", TRUE));
799 }
800 else{
19462c57 801 check = newRV_inc((SV*)mouse_tc_generate(aTHX_ NULL, (check_fptr_t)mouse_types_check, (SV*)checks));
f790c46b 802 }
ca8e67d6 803 (void)set_slots(self, "compiled_type_constraint", check);
f790c46b 804}
805
c7576321 806bool
31a596ed 807check(SV* self, SV* sv, ...)
c7576321 808CODE:
809{
810 SV* const check = get_slots(self, "compiled_type_constraint");
3f1bd57f 811 if(!(check && IsCodeRef(check))){
812 mouse_throw_error(self, check,
813 "'%"SVf"' has no compiled type constraint", self);
c7576321 814 }
815 RETVAL = mouse_tc_check(aTHX_ check, sv) ? TRUE : FALSE;
816}
817OUTPUT:
818 RETVAL
819