[gitmo/Mouse.git] / xs-src / MouseTypeConstraints.xs

/*
 *   full definition of built-in type constraints (ware in Moose::Util::TypeConstraints::OptimizedConstraints)
 */

#include "mouse.h"

#if PERL_BCDVERSION >= 0x5008005
#define LooksLikeNumber(sv) looks_like_number(sv)
#else
#define LooksLikeNumber(sv) ( SvPOKp(sv) ? looks_like_number(sv) : SvNIOKp(sv) )
#endif

#ifndef SvRXOK
#define SvRXOK(sv) (SvROK(sv) && SvMAGICAL(SvRV(sv)) && mg_find(SvRV(sv), PERL_MAGIC_qr))
#endif

typedef int (*check_fptr_t)(pTHX_ SV* const data, SV* const sv);

int
mouse_tc_check(pTHX_ SV* const tc_code, SV* const sv) {
    CV* const cv = (CV*)SvRV(tc_code);
    assert(SvTYPE(cv) == Svt_PVCV);

    if(CvISXSUB(cv)){ /* can be built-in tc */
        if(CvXSUB(cv) == XS_Mouse__Util__TypeConstraints_Item){
            assert(CvXSUBANY(cv).any_iv > 0);

            return mouse_builtin_tc_check(aTHX_ CvXSUBANY(cv).any_iv, sv);
        }
        else if(CvXSUB(cv) == XS_Mouse_parameterized_check){
            MAGIC* const mg = (MAGIC*)CvXSUBANY(cv).any_ptr;

            assert(CvXSUBANY(cv).any_ptr != NULL);

            /* call the check function directly, skipping call_sv() */
            return CALL_FPTR((check_fptr_t)mg->mg_ptr)(aTHX_ mg->mg_obj, sv);
        }
    }

    /* user-defined type constraints */
    {
        int ok;
        dSP;

        ENTER;
        SAVETMPS;

        PUSHMARK(SP);
        XPUSHs(sv);
        PUTBACK;

        call_sv(tc_code, G_SCALAR);

        SPAGAIN;
        ok = SvTRUEx(POPs);
        PUTBACK;

        FREETMPS;
        LEAVE;

        return ok;
    }
}

int
mouse_builtin_tc_check(pTHX_ mouse_tc const tc, SV* const sv) {
    switch(tc){
    case MOUSE_TC_ANY:        return mouse_tc_Any(aTHX_ sv);
    case MOUSE_TC_ITEM:       return mouse_tc_Any(aTHX_ sv);
    case MOUSE_TC_UNDEF:      return mouse_tc_Undef(aTHX_ sv);
    case MOUSE_TC_DEFINED:    return mouse_tc_Defined(aTHX_ sv);
    case MOUSE_TC_BOOL:       return mouse_tc_Bool(aTHX_ sv);
    case MOUSE_TC_VALUE:      return mouse_tc_Value(aTHX_ sv);
    case MOUSE_TC_REF:        return mouse_tc_Ref(aTHX_ sv);
    case MOUSE_TC_STR:        return mouse_tc_Str(aTHX_ sv);
    case MOUSE_TC_NUM:        return mouse_tc_Num(aTHX_ sv);
    case MOUSE_TC_INT:        return mouse_tc_Int(aTHX_ sv);
    case MOUSE_TC_SCALAR_REF: return mouse_tc_ScalarRef(aTHX_ sv);
    case MOUSE_TC_ARRAY_REF:  return mouse_tc_ArrayRef(aTHX_ sv);
    case MOUSE_TC_HASH_REF:   return mouse_tc_HashRef(aTHX_ sv);
    case MOUSE_TC_CODE_REF:   return mouse_tc_CodeRef(aTHX_ sv);
    case MOUSE_TC_GLOB_REF:   return mouse_tc_GlobRef(aTHX_ sv);
    case MOUSE_TC_FILEHANDLE: return mouse_tc_FileHandle(aTHX_ sv);
    case MOUSE_TC_REGEXP_REF: return mouse_tc_RegexpRef(aTHX_ sv);
    case MOUSE_TC_OBJECT:     return mouse_tc_Object(aTHX_ sv);
    case MOUSE_TC_CLASS_NAME: return mouse_tc_ClassName(aTHX_ sv);
    case MOUSE_TC_ROLE_NAME:  return mouse_tc_RoleName(aTHX_ sv);
    default:
        NOOP;
    }

    croak("Mouse-panic: unrecognized type constraint id: %d", (int)tc);
    return FALSE; /* not reached */
}


/*
    The following type check functions return an integer, not a bool, to keep them simple,
    so if you assign these return value to bool variable, you must use "expr ? TRUE : FALSE".
*/

int
mouse_tc_Any(pTHX_ SV* const sv PERL_UNUSED_DECL) {
    assert(sv);
    return TRUE;
}

int
mouse_tc_Bool(pTHX_ SV* const sv) {
    assert(sv);

    if(SvTRUE(sv)){
        if(SvIOKp(sv)){
            return SvIVX(sv) == 1;
        }
        else if(SvNOKp(sv)){
            return SvNVX(sv) == 1.0;
        }
        else if(SvPOKp(sv)){ /* "1" */
            return SvCUR(sv) == 1 && SvPVX(sv)[0] == '1';
        }
        else{
            return FALSE;
        }
    }
    else{
        /* false must be boolean */
        return TRUE;
    }
}

int
mouse_tc_Undef(pTHX_ SV* const sv) {
    assert(sv);
    return !SvOK(sv);
}

int
mouse_tc_Defined(pTHX_ SV* const sv) {
    assert(sv);
    return SvOK(sv);
}

int
mouse_tc_Value(pTHX_ SV* const sv) {
    assert(sv);
    return SvOK(sv) && !SvROK(sv);
}

int
mouse_tc_Num(pTHX_ SV* const sv) {
    assert(sv);
    return LooksLikeNumber(sv);
}

int
mouse_tc_Int(pTHX_ SV* const sv) {
    assert(sv);
    if(SvIOKp(sv)){
        return TRUE;
    }
    else if(SvNOKp(sv)){
        NV const nv = SvNVX(sv);
        return nv > 0 ? (nv == (NV)(UV)nv) : (nv == (NV)(IV)nv);
    }
    else if(SvPOKp(sv)){
        int const num_type = grok_number(SvPVX(sv), SvCUR(sv), NULL);
        if(num_type){
            return !(num_type & IS_NUMBER_NOT_INT);
        }
    }
    return FALSE;
}

int
mouse_tc_Str(pTHX_ SV* const sv) {
    assert(sv);
    return SvOK(sv) && !SvROK(sv) && !isGV(sv);
}

int
mouse_tc_ClassName(pTHX_ SV* const sv){ 
    assert(sv);
    return is_class_loaded(sv);
}

int
mouse_tc_RoleName(pTHX_ SV* const sv) {
    assert(sv);
    if(is_class_loaded(sv)){
        int ok;
        SV* meta;
        dSP;

        ENTER;
        SAVETMPS;

        PUSHMARK(SP);
        XPUSHs(sv);
        PUTBACK;
        call_pv("Mouse::Util::get_metaclass_by_name", G_SCALAR);
        SPAGAIN;
        meta = POPs;
        PUTBACK;

        ok =  is_an_instance_of("Mouse::Meta::Role", meta);

        FREETMPS;
        LEAVE;

        return ok;
    }
    return FALSE;
}

int
mouse_tc_Ref(pTHX_ SV* const sv) {
    assert(sv);
    return SvROK(sv);
}

int
mouse_tc_ScalarRef(pTHX_ SV* const sv) {
    assert(sv);
    return SvROK(sv) && !SvOBJECT(SvRV(sv)) && (SvTYPE(SvRV(sv)) <= SVt_PVLV && !isGV(SvRV(sv)));
}

int
mouse_tc_ArrayRef(pTHX_ SV* const sv) {
    assert(sv);
    return SvROK(sv) && !SvOBJECT(SvRV(sv)) && SvTYPE(SvRV(sv)) == SVt_PVAV;
}

int
mouse_tc_HashRef(pTHX_ SV* const sv) {
    assert(sv);
    return SvROK(sv) && !SvOBJECT(SvRV(sv)) && SvTYPE(SvRV(sv)) == SVt_PVHV;
}

int
mouse_tc_CodeRef(pTHX_ SV* const sv) {
    assert(sv);
    return SvROK(sv)  && !SvOBJECT(SvRV(sv))&& SvTYPE(SvRV(sv)) == SVt_PVCV;
}

int
mouse_tc_RegexpRef(pTHX_ SV* const sv) {
    assert(sv);
    return SvRXOK(sv);
}

int
mouse_tc_GlobRef(pTHX_ SV* const sv) {
    assert(sv);
    return SvROK(sv) && !SvOBJECT(SvRV(sv)) && isGV(SvRV(sv));
}

int
mouse_tc_FileHandle(pTHX_ SV* const sv) {
    GV* gv;
    assert(sv);

    /* see pp_fileno() in pp_sys.c and Scalar::Util::openhandle() */

    gv = (GV*)(SvROK(sv) ? SvRV(sv) : sv);
    if(isGV(gv) || SvTYPE(gv) == SVt_PVIO){
        IO* const io = isGV(gv) ? GvIO(gv) : (IO*)gv;

        if(io && ( IoIFP(io) || SvTIED_mg((SV*)io, PERL_MAGIC_tiedscalar) )){
            return TRUE;
        }
    }

    return is_an_instance_of("IO::Handle", sv);
}

int
mouse_tc_Object(pTHX_ SV* const sv) {
    assert(sv);
    return SvROK(sv) && SvOBJECT(SvRV(sv)) && !SvRXOK(sv);
}

/* Parameterized type constraints */

int
mouse_parameterized_ArrayRef(pTHX_ SV* const param, SV* const sv) {
    if(mouse_tc_ArrayRef(aTHX_ sv)){
        AV* const av  = (AV*)SvRV(sv);
        I32 const len = av_len(av) + 1;
        I32 i;
        for(i = 0; i < len; i++){
            SV* const value = *av_fetch(av, i, TRUE);
            SvGETMAGIC(value);
            if(!mouse_tc_check(aTHX_ param, value)){
                return FALSE;
            }
        }
        return TRUE;
    }
    return FALSE;
}

int
mouse_parameterized_HashRef(pTHX_ SV* const param, SV* const sv) {
    if(mouse_tc_HashRef(aTHX_ sv)){
        HV* const hv  = (HV*)SvRV(sv);
        HE* he;

        hv_iterinit(hv);
        while((he = hv_iternext(hv))){
            SV* const value = hv_iterval(hv, he);
            SvGETMAGIC(value);
            if(!mouse_tc_check(aTHX_ param, value)){
                return FALSE;
            }
        }
        return TRUE;
    }
    return FALSE;
}

int
mouse_parameterized_Maybe(pTHX_ SV* const param, SV* const sv) {
    if(SvOK(sv)){
        return mouse_tc_check(aTHX_ param, sv);
    }
    return TRUE;
}

/*
 *  This class_type generator is taken from Scalar::Util::Instance
 */

#define MY_CXT_KEY "Mouse::Util::TypeConstraints::_guts" XS_VERSION
typedef struct sui_cxt{
    GV* universal_isa;
} my_cxt_t;
START_MY_CXT

#define MG_klass_stash(mg) ((HV*)(mg)->mg_obj)
#define MG_klass_pv(mg)    ((mg)->mg_ptr)
#define MG_klass_len(mg)   ((mg)->mg_len)

static const char*
mouse_canonicalize_package_name(const char* name){

    /* "::Foo" -> "Foo" */
    if(name[0] == ':' && name[1] == ':'){
        name += 2;
    }

    /* "main::main::main::Foo" -> "Foo" */
    while(strnEQ(name, "main::", sizeof("main::")-1)){
        name += sizeof("main::")-1;
    }

    return name;
}

static int
mouse_lookup_isa(pTHX_ HV* const instance_stash, const char* const klass_pv){
    AV*  const linearized_isa = mro_get_linear_isa(instance_stash);
    SV**       svp            = AvARRAY(linearized_isa);
    SV** const end            = svp + AvFILLp(linearized_isa) + 1;

    while(svp != end){
        assert(SvPVX(*svp));
        if(strEQ(klass_pv, mouse_canonicalize_package_name(SvPVX(*svp)))){
            return TRUE;
        }
        svp++;
    }
    return FALSE;
}

int
mouse_is_an_instance_of(pTHX_ HV* const stash, SV* const instance){
    assert(stash);
    assert(SvTYPE(stash) == SVt_PVHV);

    if(IsObject(instance)){
        dMY_CXT;
        HV* const instance_stash = SvSTASH(SvRV(instance));
        GV* const instance_isa   = gv_fetchmeth_autoload(instance_stash, "isa", sizeof("isa")-1, 0);

        /* the instance has no own isa method */
        if(instance_isa == NULL || GvCV(instance_isa) == GvCV(MY_CXT.universal_isa)){
            return stash == instance_stash
                || mouse_lookup_isa(aTHX_ instance_stash, HvNAME_get(stash));
        }
        /* the instance has its own isa method */
        else {
            int retval;
            dSP;

            ENTER;
            SAVETMPS;

            PUSHMARK(SP);
            EXTEND(SP, 2);
            PUSHs(instance);
            mPUSHp(HvNAME_get(stash), HvNAMELEN_get(stash));
            PUTBACK;

            call_sv((SV*)instance_isa, G_SCALAR);

            SPAGAIN;

            retval = SvTRUEx(POPs);

            PUTBACK;

            FREETMPS;
            LEAVE;

            return retval;
        }
    }
    return FALSE;
}

static int
mouse_is_an_instance_of_universal(pTHX_ SV* const data, SV* const sv){
    PERL_UNUSED_ARG(data);
    return SvROK(sv) && SvOBJECT(SvRV(sv));
}

static MGVTBL mouse_util_type_constraints_vtbl; /* not used, only for identity */

static CV*
mouse_tc_parameterize(pTHX_ const char* const name, check_fptr_t const fptr, SV* const param) {
    CV* xsub;

    xsub = newXS(name, XS_Mouse_parameterized_check, __FILE__);
    CvXSUBANY(xsub).any_ptr = sv_magicext(
        (SV*)xsub,
        param,       /* mg_obj: refcnt will be increased */
        PERL_MAGIC_ext,
        &mouse_util_type_constraints_vtbl,
        (void*)fptr, /* mg_ptr */
        0            /* mg_len: 0 for static data */
    );

    if(!name){
        sv_2mortal((SV*)xsub);
    }

    return xsub;
}

CV*
mouse_generate_isa_predicate_for(pTHX_ SV* const klass, const char* const predicate_name){
    STRLEN klass_len;
    const char* klass_pv = SvPV_const(klass, klass_len);
    SV*   param;
    void* fptr;

    klass_pv = mouse_canonicalize_package_name(klass_pv);

    if(strNE(klass_pv, "UNIVERSAL")){
        param = (SV*)gv_stashpvn(klass_pv, klass_len, GV_ADD);
        fptr = (void*)mouse_is_an_instance_of;

    }
    else{
        param = NULL;
        fptr = (void*)mouse_is_an_instance_of_universal;
    }

    return mouse_tc_parameterize(aTHX_ predicate_name, fptr, param);
}

XS(XS_Mouse_parameterized_check) {
    dVAR;
    dXSARGS;
    MAGIC* const mg = (MAGIC*)XSANY.any_ptr;

    if(items < 1){
        croak("Too few arguments for parameterized check functions");
    }

    SvGETMAGIC( ST(0) );
    ST(0) = boolSV( CALL_FPTR((check_fptr_t)mg->mg_ptr)(aTHX_ mg->mg_obj, ST(0)) );
    XSRETURN(1);
}

static void
setup_my_cxt(pTHX_ pMY_CXT){
    MY_CXT.universal_isa = gv_fetchpvs("UNIVERSAL::isa", GV_ADD, SVt_PVCV);
    SvREFCNT_inc_simple_void_NN(MY_CXT.universal_isa);
}

MODULE = Mouse::Util::TypeConstraints    PACKAGE = Mouse::Util::TypeConstraints

PROTOTYPES:   DISABLE
VERSIONCHECK: DISABLE

BOOT:
{
    MY_CXT_INIT;
    setup_my_cxt(aTHX_ aMY_CXT);
}

#ifdef USE_ITHREADS

void
CLONE(...)
CODE:
{
    MY_CXT_CLONE;
    setup_my_cxt(aTHX_ aMY_CXT);
    PERL_UNUSED_VAR(items);
}

#endif /* !USE_ITHREADS */

void
Item(SV* sv = &PL_sv_undef)
ALIAS:
    Any        = MOUSE_TC_ANY
    Item       = MOUSE_TC_ITEM
    Undef      = MOUSE_TC_UNDEF
    Defined    = MOUSE_TC_DEFINED
    Bool       = MOUSE_TC_BOOL
    Value      = MOUSE_TC_VALUE
    Ref        = MOUSE_TC_REF
    Str        = MOUSE_TC_STR
    Num        = MOUSE_TC_NUM
    Int        = MOUSE_TC_INT
    ScalarRef  = MOUSE_TC_SCALAR_REF
    ArrayRef   = MOUSE_TC_ARRAY_REF
    HashRef    = MOUSE_TC_HASH_REF
    CodeRef    = MOUSE_TC_CODE_REF
    GlobRef    = MOUSE_TC_GLOB_REF
    FileHandle = MOUSE_TC_FILEHANDLE
    RegexpRef  = MOUSE_TC_REGEXP_REF
    Object     = MOUSE_TC_OBJECT
    ClassName  = MOUSE_TC_CLASS_NAME
    RoleName   = MOUSE_TC_ROLE_NAME
CODE:
    SvGETMAGIC(sv);
    ST(0) = boolSV( mouse_builtin_tc_check(aTHX_ ix, sv) );
    XSRETURN(1);


CV*
_parameterize_ArrayRef_for(SV* param)
ALIAS:
    _parameterize_ArrayRef_for = MOUSE_TC_ARRAY_REF
    _parameterize_HashRef_for  = MOUSE_TC_HASH_REF
    _parameterize_Maybe_for    = MOUSE_TC_MAYBE
CODE:
{
    check_fptr_t fptr;
    SV* const tc_code = mcall0s(param, "_compiled_type_constraint");
    if(!(SvROK(tc_code) && SvTYPE(SvRV(tc_code)) == SVt_PVCV)){
        croak("_compiled_type_constraint didn't return a CODE reference");
    }

    switch(ix){
    case MOUSE_TC_ARRAY_REF:
        fptr = mouse_parameterized_ArrayRef;
        break;
    case MOUSE_TC_HASH_REF:
        fptr = mouse_parameterized_HashRef;
        break;
    default: /* Maybe type */
        fptr = mouse_parameterized_Maybe;
    }
    RETVAL = mouse_tc_parameterize(aTHX_ NULL, fptr, tc_code);
}
OUTPUT:
    RETVAL
Commit	Line	Data
7d96ae4d	1	/*
	2	* full definition of built-in type constraints (ware in Moose::Util::TypeConstraints::OptimizedConstraints)
	3	*/
	4
	5	#include "mouse.h"
	6
	7	#if PERL_BCDVERSION >= 0x5008005
	8	#define LooksLikeNumber(sv) looks_like_number(sv)
	9	#else
	10	#define LooksLikeNumber(sv) ( SvPOKp(sv) ? looks_like_number(sv) : SvNIOKp(sv) )
	11	#endif
	12
	13	#ifndef SvRXOK
	14	#define SvRXOK(sv) (SvROK(sv) && SvMAGICAL(SvRV(sv)) && mg_find(SvRV(sv), PERL_MAGIC_qr))
	15	#endif
	16
d33d8840	17	typedef int (check_fptr_t)(pTHX_ SV const data, SV* const sv);
d33d8840	18
1d5ecd5f	19	int
1d5ecd5f	20	mouse_tc_check(pTHX_ SV* const tc_code, SV* const sv) {
28ba5a26	21	CV* const cv = (CV*)SvRV(tc_code);
28ba5a26	22	assert(SvTYPE(cv) == Svt_PVCV);
d33d8840	23
	24	if(CvISXSUB(cv)){ /* can be built-in tc */
	25	if(CvXSUB(cv) == XS_Mouse__Util__TypeConstraints_Item){
	26	assert(CvXSUBANY(cv).any_iv > 0);
	27
	28	return mouse_builtin_tc_check(aTHX_ CvXSUBANY(cv).any_iv, sv);
	29	}
	30	else if(CvXSUB(cv) == XS_Mouse_parameterized_check){
	31	MAGIC* const mg = (MAGIC*)CvXSUBANY(cv).any_ptr;
	32
	33	assert(CvXSUBANY(cv).any_ptr != NULL);
28ba5a26	34
	35	/* call the check function directly, skipping call_sv() */
	36	return CALL_FPTR((check_fptr_t)mg->mg_ptr)(aTHX_ mg->mg_obj, sv);
d33d8840	37	}
1d5ecd5f	38	}
d33d8840	39
	40	/* user-defined type constraints */
	41	{
1d5ecd5f	42	int ok;
	43	dSP;
	44
	45	ENTER;
	46	SAVETMPS;
	47
	48	PUSHMARK(SP);
	49	XPUSHs(sv);
	50	PUTBACK;
	51
	52	call_sv(tc_code, G_SCALAR);
	53
	54	SPAGAIN;
	55	ok = SvTRUEx(POPs);
	56	PUTBACK;
	57
	58	FREETMPS;
	59	LEAVE;
	60
	61	return ok;
	62	}
	63	}
7d96ae4d	64
7d96ae4d	65	int
1d5ecd5f	66	mouse_builtin_tc_check(pTHX_ mouse_tc const tc, SV* const sv) {
7d96ae4d	67	switch(tc){
	68	case MOUSE_TC_ANY: return mouse_tc_Any(aTHX_ sv);
	69	case MOUSE_TC_ITEM: return mouse_tc_Any(aTHX_ sv);
	70	case MOUSE_TC_UNDEF: return mouse_tc_Undef(aTHX_ sv);
	71	case MOUSE_TC_DEFINED: return mouse_tc_Defined(aTHX_ sv);
	72	case MOUSE_TC_BOOL: return mouse_tc_Bool(aTHX_ sv);
	73	case MOUSE_TC_VALUE: return mouse_tc_Value(aTHX_ sv);
	74	case MOUSE_TC_REF: return mouse_tc_Ref(aTHX_ sv);
	75	case MOUSE_TC_STR: return mouse_tc_Str(aTHX_ sv);
	76	case MOUSE_TC_NUM: return mouse_tc_Num(aTHX_ sv);
	77	case MOUSE_TC_INT: return mouse_tc_Int(aTHX_ sv);
	78	case MOUSE_TC_SCALAR_REF: return mouse_tc_ScalarRef(aTHX_ sv);
	79	case MOUSE_TC_ARRAY_REF: return mouse_tc_ArrayRef(aTHX_ sv);
	80	case MOUSE_TC_HASH_REF: return mouse_tc_HashRef(aTHX_ sv);
	81	case MOUSE_TC_CODE_REF: return mouse_tc_CodeRef(aTHX_ sv);
	82	case MOUSE_TC_GLOB_REF: return mouse_tc_GlobRef(aTHX_ sv);
	83	case MOUSE_TC_FILEHANDLE: return mouse_tc_FileHandle(aTHX_ sv);
	84	case MOUSE_TC_REGEXP_REF: return mouse_tc_RegexpRef(aTHX_ sv);
	85	case MOUSE_TC_OBJECT: return mouse_tc_Object(aTHX_ sv);
	86	case MOUSE_TC_CLASS_NAME: return mouse_tc_ClassName(aTHX_ sv);
	87	case MOUSE_TC_ROLE_NAME: return mouse_tc_RoleName(aTHX_ sv);
	88	default:
7d96ae4d	89	NOOP;
	90	}
	91
f6e3b8a9	92	croak("Mouse-panic: unrecognized type constraint id: %d", (int)tc);
7d96ae4d	93	return FALSE; /* not reached */
	94	}
	95
	96
	97	/*
	98	The following type check functions return an integer, not a bool, to keep them simple,
	99	so if you assign these return value to bool variable, you must use "expr ? TRUE : FALSE".
	100	*/
	101
	102	int
	103	mouse_tc_Any(pTHX_ SV* const sv PERL_UNUSED_DECL) {
	104	assert(sv);
	105	return TRUE;
	106	}
	107
	108	int
	109	mouse_tc_Bool(pTHX_ SV* const sv) {
	110	assert(sv);
f6e3b8a9	111
f6e3b8a9	112	if(SvTRUE(sv)){
7d96ae4d	113	if(SvIOKp(sv)){
f6e3b8a9	114	return SvIVX(sv) == 1;
7d96ae4d	115	}
7d96ae4d	116	else if(SvNOKp(sv)){
f6e3b8a9	117	return SvNVX(sv) == 1.0;
7d96ae4d	118	}
f6e3b8a9	119	else if(SvPOKp(sv)){ /* "1" */
f6e3b8a9	120	return SvCUR(sv) == 1 && SvPVX(sv)[0] == '1';
7d96ae4d	121	}
	122	else{
	123	return FALSE;
	124	}
	125	}
	126	else{
f6e3b8a9	127	/* false must be boolean */
7d96ae4d	128	return TRUE;
	129	}
	130	}
	131
	132	int
	133	mouse_tc_Undef(pTHX_ SV* const sv) {
	134	assert(sv);
	135	return !SvOK(sv);
	136	}
	137
	138	int
	139	mouse_tc_Defined(pTHX_ SV* const sv) {
	140	assert(sv);
	141	return SvOK(sv);
	142	}
	143
	144	int
	145	mouse_tc_Value(pTHX_ SV* const sv) {
	146	assert(sv);
	147	return SvOK(sv) && !SvROK(sv);
	148	}
	149
	150	int
	151	mouse_tc_Num(pTHX_ SV* const sv) {
	152	assert(sv);
	153	return LooksLikeNumber(sv);
	154	}
	155
	156	int
	157	mouse_tc_Int(pTHX_ SV* const sv) {
	158	assert(sv);
	159	if(SvIOKp(sv)){
	160	return TRUE;
	161	}
	162	else if(SvNOKp(sv)){
	163	NV const nv = SvNVX(sv);
	164	return nv > 0 ? (nv == (NV)(UV)nv) : (nv == (NV)(IV)nv);
	165	}
	166	else if(SvPOKp(sv)){
	167	int const num_type = grok_number(SvPVX(sv), SvCUR(sv), NULL);
	168	if(num_type){
	169	return !(num_type & IS_NUMBER_NOT_INT);
	170	}
	171	}
	172	return FALSE;
	173	}
	174
	175	int
	176	mouse_tc_Str(pTHX_ SV* const sv) {
	177	assert(sv);
	178	return SvOK(sv) && !SvROK(sv) && !isGV(sv);
	179	}
	180
	181	int
	182	mouse_tc_ClassName(pTHX_ SV* const sv){
	183	assert(sv);
	184	return is_class_loaded(sv);
	185	}
	186
	187	int
	188	mouse_tc_RoleName(pTHX_ SV* const sv) {
	189	assert(sv);
	190	if(is_class_loaded(sv)){
	191	int ok;
192	SV* meta;
193	dSP;
194
195	ENTER;
196	SAVETMPS;
197
198	PUSHMARK(SP);
199	XPUSHs(sv);
200	PUTBACK;
201	call_pv("Mouse::Util::get_metaclass_by_name", G_SCALAR);
202	SPAGAIN;
203	meta = POPs;
204	PUTBACK;
205
016e4681	206	ok = is_an_instance_of("Mouse::Meta::Role", meta);
7d96ae4d	207
	208	FREETMPS;
	209	LEAVE;
	210
	211	return ok;
	212	}
	213	return FALSE;
	214	}
	215
	216	int
	217	mouse_tc_Ref(pTHX_ SV* const sv) {
	218	assert(sv);
	219	return SvROK(sv);
	220	}
	221
	222	int
	223	mouse_tc_ScalarRef(pTHX_ SV* const sv) {
	224	assert(sv);
	225	return SvROK(sv) && !SvOBJECT(SvRV(sv)) && (SvTYPE(SvRV(sv)) <= SVt_PVLV && !isGV(SvRV(sv)));
	226	}
	227
	228	int
	229	mouse_tc_ArrayRef(pTHX_ SV* const sv) {
	230	assert(sv);
	231	return SvROK(sv) && !SvOBJECT(SvRV(sv)) && SvTYPE(SvRV(sv)) == SVt_PVAV;
	232	}
	233
	234	int
	235	mouse_tc_HashRef(pTHX_ SV* const sv) {
	236	assert(sv);
	237	return SvROK(sv) && !SvOBJECT(SvRV(sv)) && SvTYPE(SvRV(sv)) == SVt_PVHV;
	238	}
	239
	240	int
	241	mouse_tc_CodeRef(pTHX_ SV* const sv) {
	242	assert(sv);
	243	return SvROK(sv) && !SvOBJECT(SvRV(sv))&& SvTYPE(SvRV(sv)) == SVt_PVCV;
	244	}
	245
	246	int
	247	mouse_tc_RegexpRef(pTHX_ SV* const sv) {
	248	assert(sv);
	249	return SvRXOK(sv);
	250	}
	251
	252	int
	253	mouse_tc_GlobRef(pTHX_ SV* const sv) {
	254	assert(sv);
	255	return SvROK(sv) && !SvOBJECT(SvRV(sv)) && isGV(SvRV(sv));
	256	}
	257
	258	int
	259	mouse_tc_FileHandle(pTHX_ SV* const sv) {
	260	GV* gv;
	261	assert(sv);
	262
	263	/* see pp_fileno() in pp_sys.c and Scalar::Util::openhandle() */
	264
	265	gv = (GV*)(SvROK(sv) ? SvRV(sv) : sv);
	266	if(isGV(gv) \|\| SvTYPE(gv) == SVt_PVIO){
	267	IO* const io = isGV(gv) ? GvIO(gv) : (IO*)gv;
	268
	269	if(io && ( IoIFP(io) \|\| SvTIED_mg((SV*)io, PERL_MAGIC_tiedscalar) )){
	270	return TRUE;
271	}
272	}
273
016e4681	274	return is_an_instance_of("IO::Handle", sv);
7d96ae4d	275	}
	276
	277	int
	278	mouse_tc_Object(pTHX_ SV* const sv) {
	279	assert(sv);
	280	return SvROK(sv) && SvOBJECT(SvRV(sv)) && !SvRXOK(sv);
	281	}
	282
619338ac	283	/* Parameterized type constraints */
	284
	285	int
	286	mouse_parameterized_ArrayRef(pTHX_ SV* const param, SV* const sv) {
	287	if(mouse_tc_ArrayRef(aTHX_ sv)){
	288	AV* const av = (AV*)SvRV(sv);
	289	I32 const len = av_len(av) + 1;
	290	I32 i;
	291	for(i = 0; i < len; i++){
	292	SV* const value = *av_fetch(av, i, TRUE);
	293	SvGETMAGIC(value);
	294	if(!mouse_tc_check(aTHX_ param, value)){
	295	return FALSE;
	296	}
	297	}
	298	return TRUE;
	299	}
	300	return FALSE;
	301	}
	302
	303	int
	304	mouse_parameterized_HashRef(pTHX_ SV* const param, SV* const sv) {
	305	if(mouse_tc_HashRef(aTHX_ sv)){
	306	HV* const hv = (HV*)SvRV(sv);
	307	HE* he;
	308
	309	hv_iterinit(hv);
	310	while((he = hv_iternext(hv))){
	311	SV* const value = hv_iterval(hv, he);
	312	SvGETMAGIC(value);
	313	if(!mouse_tc_check(aTHX_ param, value)){
	314	return FALSE;
	315	}
	316	}
	317	return TRUE;
	318	}
	319	return FALSE;
	320	}
	321
	322	int
	323	mouse_parameterized_Maybe(pTHX_ SV* const param, SV* const sv) {
	324	if(SvOK(sv)){
	325	return mouse_tc_check(aTHX_ param, sv);
	326	}
	327	return TRUE;
	328	}
	329
1d5ecd5f	330	/*
	331	* This class_type generator is taken from Scalar::Util::Instance
	332	*/
	333
	334	#define MY_CXT_KEY "Mouse::Util::TypeConstraints::_guts" XS_VERSION
	335	typedef struct sui_cxt{
	336	GV* universal_isa;
	337	} my_cxt_t;
	338	START_MY_CXT
	339
	340	#define MG_klass_stash(mg) ((HV*)(mg)->mg_obj)
	341	#define MG_klass_pv(mg) ((mg)->mg_ptr)
	342	#define MG_klass_len(mg) ((mg)->mg_len)
	343
441964ce	344	static const char*
f48920c1	345	mouse_canonicalize_package_name(const char* name){
1d5ecd5f	346
	347	/* "::Foo" -> "Foo" */
	348	if(name[0] == ':' && name[1] == ':'){
	349	name += 2;
	350	}
	351
	352	/* "main::main::main::Foo" -> "Foo" */
	353	while(strnEQ(name, "main::", sizeof("main::")-1)){
	354	name += sizeof("main::")-1;
	355	}
	356
	357	return name;
	358	}
	359
	360	static int
441964ce	361	mouse_lookup_isa(pTHX_ HV* const instance_stash, const char* const klass_pv){
1d5ecd5f	362	AV* const linearized_isa = mro_get_linear_isa(instance_stash);
	363	SV** svp = AvARRAY(linearized_isa);
	364	SV** const end = svp + AvFILLp(linearized_isa) + 1;
	365
	366	while(svp != end){
	367	assert(SvPVX(*svp));
f48920c1	368	if(strEQ(klass_pv, mouse_canonicalize_package_name(SvPVX(*svp)))){
1d5ecd5f	369	return TRUE;
	370	}
	371	svp++;
	372	}
	373	return FALSE;
	374	}
	375
441964ce	376	int
	377	mouse_is_an_instance_of(pTHX_ HV* const stash, SV* const instance){
	378	assert(stash);
	379	assert(SvTYPE(stash) == SVt_PVHV);
	380
016e4681	381	if(IsObject(instance)){
441964ce	382	dMY_CXT;
	383	HV* const instance_stash = SvSTASH(SvRV(instance));
	384	GV* const instance_isa = gv_fetchmeth_autoload(instance_stash, "isa", sizeof("isa")-1, 0);
	385
	386	/* the instance has no own isa method */
	387	if(instance_isa == NULL \|\| GvCV(instance_isa) == GvCV(MY_CXT.universal_isa)){
	388	return stash == instance_stash
	389	\|\| mouse_lookup_isa(aTHX_ instance_stash, HvNAME_get(stash));
	390	}
	391	/* the instance has its own isa method */
	392	else {
	393	int retval;
	394	dSP;
1d5ecd5f	395
441964ce	396	ENTER;
441964ce	397	SAVETMPS;
1d5ecd5f	398
441964ce	399	PUSHMARK(SP);
	400	EXTEND(SP, 2);
	401	PUSHs(instance);
	402	mPUSHp(HvNAME_get(stash), HvNAMELEN_get(stash));
	403	PUTBACK;
1d5ecd5f	404
441964ce	405	call_sv((SV*)instance_isa, G_SCALAR);
1d5ecd5f	406
441964ce	407	SPAGAIN;
1d5ecd5f	408
441964ce	409	retval = SvTRUEx(POPs);
1d5ecd5f	410
441964ce	411	PUTBACK;
1d5ecd5f	412
441964ce	413	FREETMPS;
441964ce	414	LEAVE;
1d5ecd5f	415
441964ce	416	return retval;
441964ce	417	}
1d5ecd5f	418	}
441964ce	419	return FALSE;
	420	}
	421
	422	static int
	423	mouse_is_an_instance_of_universal(pTHX_ SV* const data, SV* const sv){
	424	PERL_UNUSED_ARG(data);
	425	return SvROK(sv) && SvOBJECT(SvRV(sv));
1d5ecd5f	426	}
1d5ecd5f	427
441964ce	428	static MGVTBL mouse_util_type_constraints_vtbl; /* not used, only for identity */
f48920c1	429
619338ac	430	static CV*
916eb82f	431	mouse_tc_parameterize(pTHX_ const char* const name, check_fptr_t const fptr, SV* const param) {
619338ac	432	CV* xsub;
916eb82f	433
619338ac	434	xsub = newXS(name, XS_Mouse_parameterized_check, __FILE__);
916eb82f	435	CvXSUBANY(xsub).any_ptr = sv_magicext(
	436	(SV*)xsub,
	437	param, /* mg_obj: refcnt will be increased */
	438	PERL_MAGIC_ext,
	439	&mouse_util_type_constraints_vtbl,
	440	(void)fptr, / mg_ptr */
	441	0 /* mg_len: 0 for static data */
	442	);
	443
	444	if(!name){
	445	sv_2mortal((SV*)xsub);
	446	}
	447
	448	return xsub;
	449	}
	450
	451	CV*
672bde7c	452	mouse_generate_isa_predicate_for(pTHX_ SV* const klass, const char* const predicate_name){
441964ce	453	STRLEN klass_len;
441964ce	454	const char* klass_pv = SvPV_const(klass, klass_len);
916eb82f	455	SV* param;
916eb82f	456	void* fptr;
1d5ecd5f	457
441964ce	458	klass_pv = mouse_canonicalize_package_name(klass_pv);
1d5ecd5f	459
441964ce	460	if(strNE(klass_pv, "UNIVERSAL")){
916eb82f	461	param = (SV*)gv_stashpvn(klass_pv, klass_len, GV_ADD);
916eb82f	462	fptr = (void*)mouse_is_an_instance_of;
441964ce	463
	464	}
	465	else{
916eb82f	466	param = NULL;
916eb82f	467	fptr = (void*)mouse_is_an_instance_of_universal;
441964ce	468	}
441964ce	469
916eb82f	470	return mouse_tc_parameterize(aTHX_ predicate_name, fptr, param);
441964ce	471	}
f48920c1	472
441964ce	473	XS(XS_Mouse_parameterized_check) {
1d5ecd5f	474	dVAR;
1d5ecd5f	475	dXSARGS;
441964ce	476	MAGIC* const mg = (MAGIC*)XSANY.any_ptr;
1d5ecd5f	477
441964ce	478	if(items < 1){
441964ce	479	croak("Too few arguments for parameterized check functions");
1d5ecd5f	480	}
1d5ecd5f	481
441964ce	482	SvGETMAGIC( ST(0) );
441964ce	483	ST(0) = boolSV( CALL_FPTR((check_fptr_t)mg->mg_ptr)(aTHX_ mg->mg_obj, ST(0)) );
1d5ecd5f	484	XSRETURN(1);
	485	}
	486
	487	static void
	488	setup_my_cxt(pTHX_ pMY_CXT){
	489	MY_CXT.universal_isa = gv_fetchpvs("UNIVERSAL::isa", GV_ADD, SVt_PVCV);
	490	SvREFCNT_inc_simple_void_NN(MY_CXT.universal_isa);
	491	}
	492
	493	MODULE = Mouse::Util::TypeConstraints PACKAGE = Mouse::Util::TypeConstraints
	494
646c0371	495	PROTOTYPES: DISABLE
646c0371	496	VERSIONCHECK: DISABLE
1d5ecd5f	497
	498	BOOT:
	499	{
	500	MY_CXT_INIT;
	501	setup_my_cxt(aTHX_ aMY_CXT);
	502	}
	503
	504	#ifdef USE_ITHREADS
	505
	506	void
	507	CLONE(...)
	508	CODE:
	509	{
	510	MY_CXT_CLONE;
	511	setup_my_cxt(aTHX_ aMY_CXT);
	512	PERL_UNUSED_VAR(items);
	513	}
	514
	515	#endif /* !USE_ITHREADS */
	516
	517	void
1d5ecd5f	518	Item(SV* sv = &PL_sv_undef)
	519	ALIAS:
	520	Any = MOUSE_TC_ANY
	521	Item = MOUSE_TC_ITEM
	522	Undef = MOUSE_TC_UNDEF
	523	Defined = MOUSE_TC_DEFINED
	524	Bool = MOUSE_TC_BOOL
	525	Value = MOUSE_TC_VALUE
	526	Ref = MOUSE_TC_REF
	527	Str = MOUSE_TC_STR
	528	Num = MOUSE_TC_NUM
	529	Int = MOUSE_TC_INT
	530	ScalarRef = MOUSE_TC_SCALAR_REF
	531	ArrayRef = MOUSE_TC_ARRAY_REF
	532	HashRef = MOUSE_TC_HASH_REF
	533	CodeRef = MOUSE_TC_CODE_REF
	534	GlobRef = MOUSE_TC_GLOB_REF
	535	FileHandle = MOUSE_TC_FILEHANDLE
	536	RegexpRef = MOUSE_TC_REGEXP_REF
	537	Object = MOUSE_TC_OBJECT
	538	ClassName = MOUSE_TC_CLASS_NAME
	539	RoleName = MOUSE_TC_ROLE_NAME
	540	CODE:
	541	SvGETMAGIC(sv);
	542	ST(0) = boolSV( mouse_builtin_tc_check(aTHX_ ix, sv) );
	543	XSRETURN(1);
	544
	545
619338ac	546	CV*
	547	_parameterize_ArrayRef_for(SV* param)
	548	ALIAS:
	549	_parameterize_ArrayRef_for = MOUSE_TC_ARRAY_REF
	550	_parameterize_HashRef_for = MOUSE_TC_HASH_REF
	551	_parameterize_Maybe_for = MOUSE_TC_MAYBE
	552	CODE:
	553	{
	554	check_fptr_t fptr;
	555	SV* const tc_code = mcall0s(param, "_compiled_type_constraint");
	556	if(!(SvROK(tc_code) && SvTYPE(SvRV(tc_code)) == SVt_PVCV)){
	557	croak("_compiled_type_constraint didn't return a CODE reference");
	558	}
	559
	560	switch(ix){
	561	case MOUSE_TC_ARRAY_REF:
	562	fptr = mouse_parameterized_ArrayRef;
	563	break;
	564	case MOUSE_TC_HASH_REF:
	565	fptr = mouse_parameterized_HashRef;
	566	break;
	567	default: /* Maybe type */
	568	fptr = mouse_parameterized_Maybe;
	569	}
	570	RETVAL = mouse_tc_parameterize(aTHX_ NULL, fptr, tc_code);
	571	}
	572	OUTPUT:
	573	RETVAL
	574