[gitmo/Package-Stash-XS.git] / Stash.xs

#include "EXTERN.h"
#include "perl.h"
#include "XSUB.h"

typedef enum {
    VAR_NONE = 0,
    VAR_SCALAR,
    VAR_ARRAY,
    VAR_HASH,
    VAR_CODE,
    VAR_IO,
    VAR_GLOB,  /* TODO: unimplemented */
    VAR_FORMAT /* TODO: unimplemented */
} vartype_t;

typedef struct {
    vartype_t type;
    char sigil;
    char *name;
} varspec_t;

const char *vartype_to_string(vartype_t type)
{
    switch (type) {
    case VAR_SCALAR:
        return "SCALAR";
    case VAR_ARRAY:
        return "ARRAY";
    case VAR_HASH:
        return "HASH";
    case VAR_CODE:
        return "CODE";
    case VAR_IO:
        return "IO";
    default:
        return "unknown";
    }
}

I32 vartype_to_svtype(vartype_t type)
{
    switch (type) {
    case VAR_SCALAR:
        return SVt_PV; /* or whatever */
    case VAR_ARRAY:
        return SVt_PVAV;
    case VAR_HASH:
        return SVt_PVHV;
    case VAR_CODE:
        return SVt_PVCV;
    case VAR_IO:
        return SVt_PVIO;
    default:
        return SVt_NULL;
    }
}

vartype_t string_to_vartype(char *vartype)
{
    if (strEQ(vartype, "SCALAR")) {
        return VAR_SCALAR;
    }
    else if (strEQ(vartype, "ARRAY")) {
        return VAR_ARRAY;
    }
    else if (strEQ(vartype, "HASH")) {
        return VAR_HASH;
    }
    else if (strEQ(vartype, "CODE")) {
        return VAR_CODE;
    }
    else if (strEQ(vartype, "IO")) {
        return VAR_IO;
    }
    else {
        croak("Type must be one of 'SCALAR', 'ARRAY', 'HASH', 'CODE', or 'IO'");
    }
}

void _deconstruct_variable_name(char *variable, varspec_t *varspec)
{
    if (!variable || !variable[0])
        croak("You must pass a variable name");

    varspec->type = VAR_NONE;

    switch (variable[0]) {
    case '$':
        varspec->type = VAR_SCALAR;
        break;
    case '@':
        varspec->type = VAR_ARRAY;
        break;
    case '%':
        varspec->type = VAR_HASH;
        break;
    case '&':
        varspec->type = VAR_CODE;
        break;
    }

    if (varspec->type != VAR_NONE) {
        varspec->sigil = variable[0];
        varspec->name = &variable[1];
    }
    else {
        varspec->type = VAR_IO;
        varspec->sigil = '\0';
        varspec->name = variable;
    }
}

void _deconstruct_variable_hash(HV *variable, varspec_t *varspec)
{
    SV **val;

    val = hv_fetch(variable, "name", 4, 0);
    if (!val)
        croak("The 'name' key is required in variable specs");

    varspec->name = savesvpv(*val);
    SAVEFREEPV(varspec->name);

    val = hv_fetch(variable, "sigil", 5, 0);
    if (!val)
        croak("The 'sigil' key is required in variable specs");

    varspec->sigil = (SvPV_nolen(*val))[0];

    val = hv_fetch(variable, "type", 4, 0);
    if (!val)
        croak("The 'type' key is required in variable specs");

    varspec->type = string_to_vartype(SvPV_nolen(*val));
}

int _valid_for_type(SV *value, vartype_t type)
{
    svtype sv_type = SvROK(value) ? SvTYPE(SvRV(value)) : SVt_NULL;

    switch (type) {
    case VAR_SCALAR:
        return sv_type == SVt_NULL ||
               sv_type == SVt_IV   ||
               sv_type == SVt_NV   ||
               sv_type == SVt_PV   ||
               sv_type == SVt_RV;
    case VAR_ARRAY:
        return sv_type == SVt_PVAV;
    case VAR_HASH:
        return sv_type == SVt_PVHV;
    case VAR_CODE:
        return sv_type == SVt_PVCV;
    case VAR_IO:
        return sv_type == SVt_PVIO;
    default:
        return 0;
    }
}

HV *_get_namespace(SV *self)
{
    dSP;
    SV *ret;

    PUSHMARK(SP);
    XPUSHs(self);
    PUTBACK;

    call_method("namespace", G_SCALAR);

    SPAGAIN;
    ret = POPs;
    PUTBACK;

    return (HV*)SvRV(ret);
}

SV *_get_name(SV *self)
{
    dSP;
    SV *ret;

    PUSHMARK(SP);
    XPUSHs(self);
    PUTBACK;

    call_method("name", G_SCALAR);

    SPAGAIN;
    ret = POPs;
    PUTBACK;

    return ret;
}

SV *_get_package_symbol(SV *self, varspec_t *variable, int vivify)
{
    HV *namespace;
    SV **entry;
    GV *glob;

    namespace = _get_namespace(self);
    entry = hv_fetch(namespace, variable->name, strlen(variable->name), vivify);
    if (!entry)
        return NULL;

    glob = (GV*)(*entry);
    if (!isGV(glob)) {
        SV *namesv;

        namesv = newSVsv(_get_name(self));
        sv_catpvs(namesv, "::");
        sv_catpv(namesv, variable->name);

        /* can't use gv_init here, because it screws up @ISA in a way that I
         * can't reproduce, but that CMOP triggers */
        gv_fetchsv(namesv, GV_ADD, vartype_to_svtype(variable->type));
    }

    if (vivify) {
        switch (variable->type) {
        case VAR_SCALAR:
            if (!GvSV(glob))
                GvSV(glob) = newSV(0);
            break;
        case VAR_ARRAY:
            if (!GvAV(glob))
                GvAV(glob) = newAV();
            break;
        case VAR_HASH:
            if (!GvHV(glob))
                GvHV(glob) = newHV();
            break;
        case VAR_CODE:
            croak("Don't know how to vivify CODE variables");
        case VAR_IO:
            if (!GvIO(glob))
                GvIOp(glob) = newIO();
            break;
        default:
            croak("Unknown type in vivication");
        }
    }

    switch (variable->type) {
    case VAR_SCALAR:
        return GvSV(glob);
    case VAR_ARRAY:
        return (SV*)GvAV(glob);
    case VAR_HASH:
        return (SV*)GvHV(glob);
    case VAR_CODE:
        return (SV*)GvCV(glob);
    case VAR_IO:
        return (SV*)GvIO(glob);
    default:
        return NULL;
    }
}

MODULE = Package::Stash  PACKAGE = Package::Stash

PROTOTYPES: DISABLE

SV*
new(class, package_name)
    char *class
    SV *package_name
  PREINIT:
    HV *instance;
    HV *namespace;
  CODE:
    if (!SvPOK(package_name))
        croak("The constructor argument must be the name of a package");

    instance = newHV();

    hv_store(instance, "name", 4, SvREFCNT_inc(package_name), 0);
    namespace = gv_stashpv(SvPV_nolen(package_name), GV_ADD);
    hv_store(instance, "namespace", 9, newRV_inc((SV*)namespace), 0);

    RETVAL = sv_bless(newRV_noinc((SV*)instance), gv_stashpv(class, 0));
  OUTPUT:
    RETVAL

SV*
name(self)
    SV *self
  PREINIT:
    SV **slot;
  CODE:
    if (!sv_isobject(self))
        croak("Can't call name as a class method");
    slot = hv_fetch((HV*)SvRV(self), "name", 4, 0);
    RETVAL = slot ? SvREFCNT_inc(*slot) : &PL_sv_undef;
  OUTPUT:
    RETVAL

SV*
namespace(self)
    SV *self
  PREINIT:
    SV **slot;
  CODE:
    if (!sv_isobject(self))
        croak("Can't call namespace as a class method");
    slot = hv_fetch((HV*)SvRV(self), "namespace", 9, 0);
    RETVAL = slot ? SvREFCNT_inc(*slot) : &PL_sv_undef;
  OUTPUT:
    RETVAL

void
add_package_symbol(self, variable, initial=NULL, ...)
    SV *self
    varspec_t variable
    SV *initial
  PREINIT:
    SV *name;
    GV *glob;
  CODE:
    if (initial && !_valid_for_type(initial, variable.type))
        croak("%s is not of type %s",
              SvPV_nolen(initial), vartype_to_string(variable.type));

    name = newSVsv(_get_name(self));
    sv_catpvs(name, "::");
    sv_catpv(name, variable.name);

    /* XXX: come back to this when i feel like reimplementing caller() */
/*
    my $filename = $opts{filename};
    my $first_line_num = $opts{first_line_num};

    (undef, $filename, $first_line_num) = caller
        if not defined $filename;

    my $last_line_num = $opts{last_line_num} || ($first_line_num ||= 0);

    # http://perldoc.perl.org/perldebguts.html#Debugger-Internals
    $DB::sub{$pkg . '::' . $name} = "$filename:$first_line_num-$last_line_num";
*/
/*
    if (items > 2 && (PL_perldb & 0x10) && variable.type == VAR_CODE) {
        int i;
        char *filename = NULL, *name;
        I32 first_line_num, last_line_num;

        if ((items - 3) % 2)
            croak("add_package_symbol: Odd number of elements in %%opts");

        for (i = 3; i < items; i += 2) {
            char *key;
            key = SvPV_nolen(ST(i));
            if (strEQ(key, "filename")) {
                if (!SvPOK(ST(i + 1)))
                    croak("add_package_symbol: filename must be a string");
                filename = SvPV_nolen(ST(i + 1));
            }
            else if (strEQ(key, "first_line_num")) {
                if (!SvIOK(ST(i + 1)))
                    croak("add_package_symbol: first_line_num must be an integer");
                first_line_num = SvIV(ST(i + 1));
            }
            else if (strEQ(key, "last_line_num")) {
                if (!SvIOK(ST(i + 1)))
                    croak("add_package_symbol: last_line_num must be an integer");
                last_line_num = SvIV(ST(i + 1));
            }
        }

        if (!filename) {
        }
    }
*/

    glob = gv_fetchsv(name, GV_ADDMULTI, vartype_to_svtype(variable.type));

    if (initial) {
        SV *val;

        if (SvROK(initial)) {
            val = SvRV(initial);
            SvREFCNT_inc(val);
        }
        else {
            val = newSVsv(initial);
        }

        switch (variable.type) {
        case VAR_SCALAR:
            GvSV(glob) = val;
            break;
        case VAR_ARRAY:
            GvAV(glob) = (AV*)val;
            break;
        case VAR_HASH:
            GvHV(glob) = (HV*)val;
            break;
        case VAR_CODE:
            GvCV(glob) = (CV*)val;
            break;
        case VAR_IO:
            GvIOp(glob) = (IO*)val;
            break;
        }
    }

    SvREFCNT_dec(name);

void
remove_package_glob(self, name)
    SV *self
    char *name
  CODE:
    hv_delete(_get_namespace(self), name, strlen(name), G_DISCARD);

int
has_package_symbol(self, variable)
    SV *self
    varspec_t variable
  PREINIT:
    HV *namespace;
    SV **entry;
  CODE:
    namespace = _get_namespace(self);
    entry = hv_fetch(namespace, variable.name, strlen(variable.name), 0);
    if (!entry)
        XSRETURN_UNDEF;

    if (isGV(*entry)) {
        GV *glob = (GV*)(*entry);
        switch (variable.type) {
        case VAR_SCALAR:
            RETVAL = GvSV(glob) ? 1 : 0;
            break;
        case VAR_ARRAY:
            RETVAL = GvAV(glob) ? 1 : 0;
            break;
        case VAR_HASH:
            RETVAL = GvHV(glob) ? 1 : 0;
            break;
        case VAR_CODE:
            RETVAL = GvCV(glob) ? 1 : 0;
            break;
        case VAR_IO:
            RETVAL = GvIO(glob) ? 1 : 0;
            break;
        }
    }
    else {
        RETVAL = (variable.type == VAR_CODE);
    }
  OUTPUT:
    RETVAL

SV*
get_package_symbol(self, variable)
    SV *self
    varspec_t variable
  PREINIT:
    SV *val;
  CODE:
    val = _get_package_symbol(self, &variable, 0);
    if (!val)
        XSRETURN_UNDEF;
    RETVAL = newRV_inc(val);
  OUTPUT:
    RETVAL

SV*
get_or_add_package_symbol(self, variable)
    SV *self
    varspec_t variable
  PREINIT:
    SV *val;
  CODE:
    val = _get_package_symbol(self, &variable, 1);
    if (!val)
        XSRETURN_UNDEF;
    RETVAL = newRV_inc(val);
  OUTPUT:
    RETVAL

void
remove_package_symbol(self, variable)
    SV *self
    varspec_t variable
  PREINIT:
    HV *namespace;
    SV **entry;
  CODE:
    namespace = _get_namespace(self);
    entry = hv_fetch(namespace, variable.name, strlen(variable.name), 0);
    if (!entry)
        XSRETURN_EMPTY;

    if (isGV(*entry)) {
        GV *glob = (GV*)(*entry);
        switch (variable.type) {
        case VAR_SCALAR:
            GvSV(glob) = (SV *)NULL;
            break;
        case VAR_ARRAY:
            GvAV(glob) = (AV *)NULL;
            break;
        case VAR_HASH:
            GvHV(glob) = (HV *)NULL;
            break;
        case VAR_CODE:
            GvCV(glob) = (CV *)NULL;
            break;
        case VAR_IO:
            GvIOp(glob) = (IO *)NULL;
            break;
        }
    }
    else {
        if (variable.type == VAR_CODE) {
            hv_delete(namespace, variable.name, strlen(variable.name), G_DISCARD);
        }
    }

void
list_all_package_symbols(self, vartype=VAR_NONE)
    SV *self
    vartype_t vartype
  PPCODE:
    if (vartype == VAR_NONE) {
        HV *namespace;
        HE *entry;
        int keys;

        namespace = _get_namespace(self);
        keys = hv_iterinit(namespace);
        EXTEND(SP, keys);
        while ((entry = hv_iternext(namespace))) {
            mPUSHs(newSVhek(HeKEY_hek(entry)));
        }
    }
    else {
        HV *namespace;
        SV *val;
        char *key;
        int len;

        namespace = _get_namespace(self);
        hv_iterinit(namespace);
        while ((val = hv_iternextsv(namespace, &key, &len))) {
            GV *gv = (GV*)val;
            if (isGV(gv)) {
                switch (vartype) {
                case VAR_SCALAR:
                    if (GvSV(val))
                        mXPUSHp(key, len);
                    break;
                case VAR_ARRAY:
                    if (GvAV(val))
                        mXPUSHp(key, len);
                    break;
                case VAR_HASH:
                    if (GvHV(val))
                        mXPUSHp(key, len);
                    break;
                case VAR_CODE:
                    if (GvCVu(val))
                        mXPUSHp(key, len);
                    break;
                case VAR_IO:
                    if (GvIO(val))
                        mXPUSHp(key, len);
                    break;
                }
            }
            else if (vartype == VAR_CODE) {
                mXPUSHp(key, len);
            }
        }
    }
Commit	Line	Data
59017825	1	#include "EXTERN.h"
	2	#include "perl.h"
	3	#include "XSUB.h"
	4
a382a84b	5	typedef enum {
	6	VAR_NONE = 0,
	7	VAR_SCALAR,
	8	VAR_ARRAY,
	9	VAR_HASH,
	10	VAR_CODE,
	11	VAR_IO,
	12	VAR_GLOB, /* TODO: unimplemented */
	13	VAR_FORMAT /* TODO: unimplemented */
	14	} vartype_t;
	15
	16	typedef struct {
	17	vartype_t type;
	18	char sigil;
	19	char *name;
	20	} varspec_t;
	21
76f7c306	22	const char *vartype_to_string(vartype_t type)
	23	{
	24	switch (type) {
	25	case VAR_SCALAR:
	26	return "SCALAR";
	27	case VAR_ARRAY:
	28	return "ARRAY";
	29	case VAR_HASH:
	30	return "HASH";
	31	case VAR_CODE:
	32	return "CODE";
	33	case VAR_IO:
	34	return "IO";
	35	default:
	36	return "unknown";
	37	}
	38	}
	39
	40	I32 vartype_to_svtype(vartype_t type)
	41	{
	42	switch (type) {
	43	case VAR_SCALAR:
	44	return SVt_PV; /* or whatever */
	45	case VAR_ARRAY:
	46	return SVt_PVAV;
	47	case VAR_HASH:
	48	return SVt_PVHV;
	49	case VAR_CODE:
	50	return SVt_PVCV;
	51	case VAR_IO:
	52	return SVt_PVIO;
	53	default:
	54	return SVt_NULL;
	55	}
	56	}
	57
a382a84b	58	vartype_t string_to_vartype(char *vartype)
	59	{
	60	if (strEQ(vartype, "SCALAR")) {
	61	return VAR_SCALAR;
	62	}
	63	else if (strEQ(vartype, "ARRAY")) {
	64	return VAR_ARRAY;
	65	}
	66	else if (strEQ(vartype, "HASH")) {
	67	return VAR_HASH;
	68	}
	69	else if (strEQ(vartype, "CODE")) {
	70	return VAR_CODE;
	71	}
	72	else if (strEQ(vartype, "IO")) {
	73	return VAR_IO;
	74	}
	75	else {
	76	croak("Type must be one of 'SCALAR', 'ARRAY', 'HASH', 'CODE', or 'IO'");
	77	}
	78	}
	79
	80	void _deconstruct_variable_name(char variable, varspec_t varspec)
	81	{
	82	if (!variable \|\| !variable[0])
	83	croak("You must pass a variable name");
	84
	85	varspec->type = VAR_NONE;
	86
	87	switch (variable[0]) {
	88	case '$':
	89	varspec->type = VAR_SCALAR;
	90	break;
	91	case '@':
	92	varspec->type = VAR_ARRAY;
	93	break;
	94	case '%':
	95	varspec->type = VAR_HASH;
	96	break;
	97	case '&':
	98	varspec->type = VAR_CODE;
	99	break;
	100	}
	101
	102	if (varspec->type != VAR_NONE) {
	103	varspec->sigil = variable[0];
	104	varspec->name = &variable[1];
	105	}
	106	else {
	107	varspec->type = VAR_IO;
	108	varspec->sigil = '\0';
	109	varspec->name = variable;
	110	}
	111	}
	112
	113	void _deconstruct_variable_hash(HV variable, varspec_t varspec)
	114	{
	115	SV **val;
a382a84b	116
	117	val = hv_fetch(variable, "name", 4, 0);
	118	if (!val)
	119	croak("The 'name' key is required in variable specs");
	120
	121	varspec->name = savesvpv(*val);
301f570b	122	SAVEFREEPV(varspec->name);
a382a84b	123
	124	val = hv_fetch(variable, "sigil", 5, 0);
	125	if (!val)
	126	croak("The 'sigil' key is required in variable specs");
	127
	128	varspec->sigil = (SvPV_nolen(*val))[0];
	129
	130	val = hv_fetch(variable, "type", 4, 0);
	131	if (!val)
	132	croak("The 'type' key is required in variable specs");
	133
	134	varspec->type = string_to_vartype(SvPV_nolen(*val));
	135	}
	136
	137	int _valid_for_type(SV *value, vartype_t type)
	138	{
	139	svtype sv_type = SvROK(value) ? SvTYPE(SvRV(value)) : SVt_NULL;
	140
	141	switch (type) {
	142	case VAR_SCALAR:
	143	return sv_type == SVt_NULL \|\|
	144	sv_type == SVt_IV \|\|
	145	sv_type == SVt_NV \|\|
	146	sv_type == SVt_PV \|\|
	147	sv_type == SVt_RV;
	148	case VAR_ARRAY:
	149	return sv_type == SVt_PVAV;
	150	case VAR_HASH:
	151	return sv_type == SVt_PVHV;
	152	case VAR_CODE:
	153	return sv_type == SVt_PVCV;
	154	case VAR_IO:
76f7c306	155	return sv_type == SVt_PVIO;
a382a84b	156	default:
	157	return 0;
	158	}
	159	}
	160
3fd56b4d	161	HV _get_namespace(SV self)
	162	{
	163	dSP;
	164	SV *ret;
	165
	166	PUSHMARK(SP);
	167	XPUSHs(self);
	168	PUTBACK;
	169
612dcf3b	170	call_method("namespace", G_SCALAR);
3fd56b4d	171
	172	SPAGAIN;
	173	ret = POPs;
	174	PUTBACK;
	175
	176	return (HV*)SvRV(ret);
	177	}
	178
76f7c306	179	SV _get_name(SV self)
	180	{
	181	dSP;
	182	SV *ret;
	183
	184	PUSHMARK(SP);
	185	XPUSHs(self);
	186	PUTBACK;
	187
612dcf3b	188	call_method("name", G_SCALAR);
76f7c306	189
	190	SPAGAIN;
	191	ret = POPs;
	192	PUTBACK;
	193
	194	return ret;
	195	}
	196
4871b1a0	197	SV _get_package_symbol(SV self, varspec_t *variable, int vivify)
	198	{
	199	HV *namespace;
	200	SV **entry;
	201	GV *glob;
	202
	203	namespace = _get_namespace(self);
	204	entry = hv_fetch(namespace, variable->name, strlen(variable->name), vivify);
	205	if (!entry)
	206	return NULL;
	207
	208	glob = (GV)(entry);
	209	if (!isGV(glob)) {
	210	SV *namesv;
4871b1a0	211
	212	namesv = newSVsv(_get_name(self));
	213	sv_catpvs(namesv, "::");
	214	sv_catpv(namesv, variable->name);
	215
b55c97c2	216	/* can't use gv_init here, because it screws up @ISA in a way that I
	217	* can't reproduce, but that CMOP triggers */
	218	gv_fetchsv(namesv, GV_ADD, vartype_to_svtype(variable->type));
4871b1a0	219	}
	220
	221	if (vivify) {
	222	switch (variable->type) {
	223	case VAR_SCALAR:
	224	if (!GvSV(glob))
	225	GvSV(glob) = newSV(0);
	226	break;
	227	case VAR_ARRAY:
	228	if (!GvAV(glob))
	229	GvAV(glob) = newAV();
	230	break;
	231	case VAR_HASH:
	232	if (!GvHV(glob))
	233	GvHV(glob) = newHV();
	234	break;
	235	case VAR_CODE:
	236	croak("Don't know how to vivify CODE variables");
	237	case VAR_IO:
	238	if (!GvIO(glob))
	239	GvIOp(glob) = newIO();
	240	break;
	241	default:
	242	croak("Unknown type in vivication");
	243	}
	244	}
	245
	246	switch (variable->type) {
	247	case VAR_SCALAR:
	248	return GvSV(glob);
	249	case VAR_ARRAY:
	250	return (SV*)GvAV(glob);
	251	case VAR_HASH:
	252	return (SV*)GvHV(glob);
	253	case VAR_CODE:
	254	return (SV*)GvCV(glob);
	255	case VAR_IO:
	256	return (SV*)GvIO(glob);
	257	default:
	258	return NULL;
	259	}
	260	}
	261
59017825	262	MODULE = Package::Stash PACKAGE = Package::Stash
59017825	263
be2a7e99	264	PROTOTYPES: DISABLE
be2a7e99	265
59017825	266	SV*
	267	new(class, package_name)
	268	char *class
	269	SV *package_name
3496f1e8	270	PREINIT:
59017825	271	HV *instance;
	272	HV *namespace;
	273	CODE:
	274	if (!SvPOK(package_name))
	275	croak("The constructor argument must be the name of a package");
	276
6c3e69c6	277	instance = newHV();
59017825	278
ce38493c	279	hv_store(instance, "name", 4, SvREFCNT_inc(package_name), 0);
59017825	280	namespace = gv_stashpv(SvPV_nolen(package_name), GV_ADD);
ce38493c	281	hv_store(instance, "namespace", 9, newRV_inc((SV*)namespace), 0);
59017825	282
6c3e69c6	283	RETVAL = sv_bless(newRV_noinc((SV*)instance), gv_stashpv(class, 0));
59017825	284	OUTPUT:
59017825	285	RETVAL
194acf47	286
	287	SV*
	288	name(self)
	289	SV *self
3496f1e8	290	PREINIT:
194acf47	291	SV **slot;
	292	CODE:
	293	if (!sv_isobject(self))
	294	croak("Can't call name as a class method");
	295	slot = hv_fetch((HV*)SvRV(self), "name", 4, 0);
	296	RETVAL = slot ? SvREFCNT_inc(*slot) : &PL_sv_undef;
	297	OUTPUT:
	298	RETVAL
	299
	300	SV*
	301	namespace(self)
	302	SV *self
3496f1e8	303	PREINIT:
194acf47	304	SV **slot;
	305	CODE:
	306	if (!sv_isobject(self))
	307	croak("Can't call namespace as a class method");
	308	slot = hv_fetch((HV*)SvRV(self), "namespace", 9, 0);
	309	RETVAL = slot ? SvREFCNT_inc(*slot) : &PL_sv_undef;
	310	OUTPUT:
	311	RETVAL
3fd56b4d	312
3fd56b4d	313	void
76f7c306	314	add_package_symbol(self, variable, initial=NULL, ...)
	315	SV *self
	316	varspec_t variable
	317	SV *initial
	318	PREINIT:
	319	SV *name;
	320	GV *glob;
	321	CODE:
	322	if (initial && !_valid_for_type(initial, variable.type))
	323	croak("%s is not of type %s",
	324	SvPV_nolen(initial), vartype_to_string(variable.type));
	325
	326	name = newSVsv(_get_name(self));
	327	sv_catpvs(name, "::");
	328	sv_catpv(name, variable.name);
	329
	330	/* XXX: come back to this when i feel like reimplementing caller() */
	331	/*
	332	my $filename = $opts{filename};
	333	my $first_line_num = $opts{first_line_num};
	334
	335	(undef, $filename, $first_line_num) = caller
	336	if not defined $filename;
	337
	338	my $last_line_num = $opts{last_line_num} \|\| ($first_line_num \|\|= 0);
	339
	340	# http://perldoc.perl.org/perldebguts.html#Debugger-Internals
	341	$DB::sub{$pkg . '::' . $name} = "$filename:$first_line_num-$last_line_num";
	342	*/
	343	/*
	344	if (items > 2 && (PL_perldb & 0x10) && variable.type == VAR_CODE) {
	345	int i;
	346	char filename = NULL, name;
	347	I32 first_line_num, last_line_num;
	348
	349	if ((items - 3) % 2)
	350	croak("add_package_symbol: Odd number of elements in %%opts");
	351
	352	for (i = 3; i < items; i += 2) {
	353	char *key;
	354	key = SvPV_nolen(ST(i));
	355	if (strEQ(key, "filename")) {
	356	if (!SvPOK(ST(i + 1)))
	357	croak("add_package_symbol: filename must be a string");
	358	filename = SvPV_nolen(ST(i + 1));
	359	}
	360	else if (strEQ(key, "first_line_num")) {
	361	if (!SvIOK(ST(i + 1)))
	362	croak("add_package_symbol: first_line_num must be an integer");
	363	first_line_num = SvIV(ST(i + 1));
	364	}
	365	else if (strEQ(key, "last_line_num")) {
	366	if (!SvIOK(ST(i + 1)))
	367	croak("add_package_symbol: last_line_num must be an integer");
	368	last_line_num = SvIV(ST(i + 1));
	369	}
	370	}
	371
	372	if (!filename) {
	373	}
	374	}
	375	*/
	376
fedea7f9	377	glob = gv_fetchsv(name, GV_ADDMULTI, vartype_to_svtype(variable.type));
76f7c306	378
	379	if (initial) {
	380	SV *val;
	381
	382	if (SvROK(initial)) {
	383	val = SvRV(initial);
	384	SvREFCNT_inc(val);
	385	}
	386	else {
	387	val = newSVsv(initial);
	388	}
	389
	390	switch (variable.type) {
	391	case VAR_SCALAR:
	392	GvSV(glob) = val;
	393	break;
	394	case VAR_ARRAY:
	395	GvAV(glob) = (AV*)val;
	396	break;
	397	case VAR_HASH:
	398	GvHV(glob) = (HV*)val;
	399	break;
	400	case VAR_CODE:
	401	GvCV(glob) = (CV*)val;
	402	break;
	403	case VAR_IO:
	404	GvIOp(glob) = (IO*)val;
	405	break;
	406	}
	407	}
	408
5bb2b07b	409	SvREFCNT_dec(name);
5bb2b07b	410
76f7c306	411	void
3fd56b4d	412	remove_package_glob(self, name)
	413	SV *self
	414	char *name
3fd56b4d	415	CODE:
3fd56b4d	416	hv_delete(_get_namespace(self), name, strlen(name), G_DISCARD);
9b608466	417
34805376	418	int
	419	has_package_symbol(self, variable)
	420	SV *self
	421	varspec_t variable
	422	PREINIT:
	423	HV *namespace;
	424	SV **entry;
	425	CODE:
	426	namespace = _get_namespace(self);
	427	entry = hv_fetch(namespace, variable.name, strlen(variable.name), 0);
	428	if (!entry)
	429	XSRETURN_UNDEF;
	430
	431	if (isGV(*entry)) {
	432	GV glob = (GV)(*entry);
	433	switch (variable.type) {
	434	case VAR_SCALAR:
	435	RETVAL = GvSV(glob) ? 1 : 0;
	436	break;
	437	case VAR_ARRAY:
	438	RETVAL = GvAV(glob) ? 1 : 0;
	439	break;
	440	case VAR_HASH:
	441	RETVAL = GvHV(glob) ? 1 : 0;
	442	break;
	443	case VAR_CODE:
	444	RETVAL = GvCV(glob) ? 1 : 0;
	445	break;
	446	case VAR_IO:
	447	RETVAL = GvIO(glob) ? 1 : 0;
	448	break;
	449	}
	450	}
	451	else {
	452	RETVAL = (variable.type == VAR_CODE);
	453	}
	454	OUTPUT:
	455	RETVAL
	456
e3ad44fd	457	SV*
4871b1a0	458	get_package_symbol(self, variable)
e3ad44fd	459	SV *self
	460	varspec_t variable
	461	PREINIT:
fca4ed0c	462	SV *val;
e3ad44fd	463	CODE:
4871b1a0	464	val = _get_package_symbol(self, &variable, 0);
4871b1a0	465	if (!val)
e3ad44fd	466	XSRETURN_UNDEF;
93c36417	467	RETVAL = newRV_inc(val);
4871b1a0	468	OUTPUT:
4871b1a0	469	RETVAL
e3ad44fd	470
4871b1a0	471	SV*
	472	get_or_add_package_symbol(self, variable)
	473	SV *self
	474	varspec_t variable
	475	PREINIT:
	476	SV *val;
	477	CODE:
	478	val = _get_package_symbol(self, &variable, 1);
fca4ed0c	479	if (!val)
fca4ed0c	480	XSRETURN_UNDEF;
93c36417	481	RETVAL = newRV_inc(val);
e3ad44fd	482	OUTPUT:
	483	RETVAL
	484
9b608466	485	void
215f49f8	486	remove_package_symbol(self, variable)
	487	SV *self
	488	varspec_t variable
	489	PREINIT:
	490	HV *namespace;
	491	SV **entry;
	492	CODE:
	493	namespace = _get_namespace(self);
	494	entry = hv_fetch(namespace, variable.name, strlen(variable.name), 0);
	495	if (!entry)
	496	XSRETURN_EMPTY;
	497
	498	if (isGV(*entry)) {
	499	GV glob = (GV)(*entry);
	500	switch (variable.type) {
	501	case VAR_SCALAR:
4e3b237e	502	GvSV(glob) = (SV *)NULL;
215f49f8	503	break;
215f49f8	504	case VAR_ARRAY:
4e3b237e	505	GvAV(glob) = (AV *)NULL;
215f49f8	506	break;
215f49f8	507	case VAR_HASH:
4e3b237e	508	GvHV(glob) = (HV *)NULL;
215f49f8	509	break;
215f49f8	510	case VAR_CODE:
4e3b237e	511	GvCV(glob) = (CV *)NULL;
215f49f8	512	break;
215f49f8	513	case VAR_IO:
4e3b237e	514	GvIOp(glob) = (IO *)NULL;
215f49f8	515	break;
	516	}
	517	}
	518	else {
	519	if (variable.type == VAR_CODE) {
	520	hv_delete(namespace, variable.name, strlen(variable.name), G_DISCARD);
	521	}
	522	}
	523
	524	void
9b608466	525	list_all_package_symbols(self, vartype=VAR_NONE)
	526	SV *self
	527	vartype_t vartype
	528	PPCODE:
	529	if (vartype == VAR_NONE) {
	530	HV *namespace;
	531	HE *entry;
	532	int keys;
	533
	534	namespace = _get_namespace(self);
	535	keys = hv_iterinit(namespace);
	536	EXTEND(SP, keys);
ebcc8ba8	537	while ((entry = hv_iternext(namespace))) {
9b608466	538	mPUSHs(newSVhek(HeKEY_hek(entry)));
	539	}
	540	}
	541	else {
	542	HV *namespace;
9b608466	543	SV *val;
	544	char *key;
	545	int len;
	546
	547	namespace = _get_namespace(self);
	548	hv_iterinit(namespace);
ebcc8ba8	549	while ((val = hv_iternextsv(namespace, &key, &len))) {
9b608466	550	GV gv = (GV)val;
	551	if (isGV(gv)) {
	552	switch (vartype) {
	553	case VAR_SCALAR:
	554	if (GvSV(val))
	555	mXPUSHp(key, len);
	556	break;
	557	case VAR_ARRAY:
	558	if (GvAV(val))
	559	mXPUSHp(key, len);
	560	break;
	561	case VAR_HASH:
	562	if (GvHV(val))
	563	mXPUSHp(key, len);
	564	break;
	565	case VAR_CODE:
	566	if (GvCVu(val))
	567	mXPUSHp(key, len);
	568	break;
	569	case VAR_IO:
	570	if (GvIO(val))
	571	mXPUSHp(key, len);
	572	break;
	573	}
	574	}
	575	else if (vartype == VAR_CODE) {
	576	mXPUSHp(key, len);
	577	}
	578	}
	579	}