fix symbol manipulators
[gitmo/Class-MOP.git] / xs / Package.xs
index ce8d390..6e907f3 100644 (file)
@@ -1,5 +1,106 @@
+
 #include "mop.h"
 
+#define GLOB_CREATE     0x01
+#define VARIABLE_CREATE 0x02
+
+
+static const char*
+mop_deconstruct_variable_name(pTHX_ SV* const variable, svtype* const type, const char** const type_name, I32* const flags) {
+       const char* name;
+
+       if(SvROK(variable) && SvTYPE(SvRV(variable)) == SVt_PVHV){
+               /* e.g. variable = { type => "SCALAR", name => "foo" } */
+               HV* const hv = (HV*)SvRV(variable);
+               SV** svp;
+               STRLEN len;
+               const char* pv;
+
+               svp = hv_fetchs(hv, "name", FALSE);
+               if(!(svp && SvOK(*svp))){
+                       croak("You must pass a variable name");
+               }
+               name = SvPV_const(*svp, len);
+               if(len < 1){
+                       croak("You must pass a variable name");
+               }
+
+               svp = hv_fetchs(hv, "type", FALSE);
+               if(!(svp && SvOK(*svp))) {
+                       croak("You must pass a variable type");
+               }
+               pv = SvPV_nolen_const(*svp);
+               if(strEQ(pv, "SCALAR")){
+                       *type = SVt_PV; /* for all the type of scalars */
+               }
+               else if(strEQ(pv, "ARRAY")){
+                       *type = SVt_PVAV;
+               }
+               else if(strEQ(pv, "HASH")){
+                       *type = SVt_PVHV;
+               }
+               else if(strEQ(pv, "CODE")){
+                       *type = SVt_PVCV;
+               }
+               else if(strEQ(pv, "GLOB")){
+                       *type = SVt_PVGV;
+               }
+               else if(strEQ(pv, "IO")){
+                       *type = SVt_PVIO;
+               }
+               else{
+                       croak("I do not recognize that type '%s'", pv);
+               }
+               *type_name = pv;
+
+               svp = hv_fetchs(hv, "create", FALSE);
+               if(svp && SvTRUE(*svp)){
+                       *flags = VARIABLE_CREATE | GLOB_CREATE;
+               }
+       }
+       else {
+               STRLEN len;
+               const char* pv;
+               /* e.g. variable = '$foo' */
+               if(!SvOK(variable)) {
+                       croak("You must pass a variable name");
+               }
+               pv = SvPV_const(variable, len);
+               if(len < 2){
+                       croak("You must pass a variable name including a sigil");
+               }
+
+               switch(pv[0]){
+               case '$':
+                       *type      = SVt_PV; /* for all the types of scalars */
+                       *type_name = "SCALAR";
+                       break;
+               case '@':
+                       *type      = SVt_PVAV;
+                       *type_name = "ARRAY";
+                       break;
+               case '%':
+                       *type      = SVt_PVHV;
+                       *type_name = "HASH";
+                       break;
+               case '&':
+                       *type      = SVt_PVCV;
+                       *type_name = "CODE";
+                       break;
+               case '*':
+                       *type      = SVt_PVGV;
+                       *type_name = "GLOB";
+                       break;
+               default:
+                       croak("I do not recognize that sigil '%c'", pv[0]);
+               }
+
+               name = pv + 1;
+       }
+
+       return name;
+}
+
 MODULE = Class::MOP::Package   PACKAGE = Class::MOP::Package
 
 PROTOTYPES: DISABLE
@@ -37,3 +138,114 @@ get_all_package_symbols(self, filter=TYPE_FILTER_NONE)
 
 BOOT:
     INSTALL_SIMPLE_READER_WITH_KEY(Package, name, package);
+
+#define S_HAS GV_NOADD_NOINIT
+#define S_GET 0
+#define S_ADD GV_ADDMULTI
+
+SV*
+add_package_symbol(SV* self, SV* variable, SV* ref = &PL_sv_undef)
+ALIAS:
+       has_package_symbol = S_HAS
+       get_package_symbol = S_GET
+       add_package_symbol = S_ADD
+PREINIT:
+       svtype type;
+       const char* type_name;
+       const char* var_name;
+       SV* package_name;
+       const char* fq_name;
+       I32 flags = 0; /* not used */
+CODE:
+       var_name = mop_deconstruct_variable_name(aTHX_ variable, &type, &type_name, &flags);
+
+       package_name = mop_call0(aTHX_ self, KEY_FOR(name));
+       if(!SvOK(package_name)){
+               croak("name() did not return a defined value");
+       }
+       fq_name = Perl_form(aTHX_ "%"SVf"::%s", package_name, var_name);
+
+       if(items == 3 && ix != S_ADD){
+               croak("Too many arguments for %s", GvNAME(CvGV(cv)));
+       }
+
+       if(SvOK(ref)){ /* add_package_symbol with a value */
+               GV* gv;
+
+               if(type == SVt_PV){
+                       if(!SvROK(ref)){
+                               ref = newRV_noinc(newSVsv(ref));
+                               sv_2mortal(ref);
+                       }
+               }
+               else if(!(SvROK(ref) && SvTYPE(SvRV(ref)) == type)){
+                       croak("You must pass a reference of %s for the value of %s", type_name, GvNAME(CvGV(cv)));
+               }
+               gv = gv_fetchpv(fq_name, GV_ADDMULTI, type);
+
+               if(type == SVt_PVCV && GvCV(gv)){
+                       /* XXX: should introduce an option { redefine => 1 } ? */
+                       SvREFCNT_dec(GvCV(gv));
+                       GvCV(gv) = NULL;
+               }
+               sv_setsv_mg((SV*)gv, ref); /* *glob = $ref */
+               RETVAL = ref;
+       }
+       else { /* no values */
+               GV* const gv = gv_fetchpv(fq_name, ix | (flags & GLOB_CREATE ? GV_ADDMULTI : 0), type);
+               SV* sv;
+
+               if(!gv){
+                       if(ix == S_HAS){
+                               XSRETURN_NO;
+                       }
+                       else{
+                               XSRETURN_UNDEF;
+                       }
+               }
+
+               if(!isGV(gv)){ /* In has_package_symbol, the stash entry is a stub or constant */
+                       assert(ix == S_HAS);
+                       if(type == SVt_PVCV){
+                               XSRETURN_YES;
+                       }
+                       else{
+                               XSRETURN_NO;
+                       }
+               }
+
+               switch(type){
+               case SVt_PVAV:
+                       sv = (SV*)((flags & VARIABLE_CREATE) ? GvAVn(gv) : GvAV(gv));
+                       break;
+               case SVt_PVHV:
+                       sv = (SV*)((flags & VARIABLE_CREATE) ? GvHVn(gv) : GvHV(gv));
+                       break;
+               case SVt_PVCV:
+                       sv = (SV*)GvCV(gv);
+                       break;
+               case SVt_PVIO:
+                       sv = (SV*)((flags & VARIABLE_CREATE) ? GvIOn(gv) : GvIO(gv));
+                       break;
+               case SVt_PVGV:
+                       sv = (SV*)gv;
+                       break;
+               default: /* SCALAR */
+                       sv =       (flags & VARIABLE_CREATE) ? GvSVn(gv) : GvSV(gv);
+                       break;
+               }
+
+               if(ix == S_HAS){
+                       RETVAL = boolSV(sv);
+               }
+               else{
+                       if(sv){
+                               RETVAL = sv_2mortal(newRV_inc(sv));
+                       }
+                       else{
+                               RETVAL = &PL_sv_undef;
+                       }
+               }
+       }
+       ST(0) = RETVAL;
+