diff options
Diffstat (limited to 'lib/Package/Stash/PP.pm')
-rw-r--r-- | lib/Package/Stash/PP.pm | 530 |
1 files changed, 530 insertions, 0 deletions
diff --git a/lib/Package/Stash/PP.pm b/lib/Package/Stash/PP.pm new file mode 100644 index 0000000..1e17d4b --- /dev/null +++ b/lib/Package/Stash/PP.pm @@ -0,0 +1,530 @@ +package Package::Stash::PP; +BEGIN { + $Package::Stash::PP::AUTHORITY = 'cpan:DOY'; +} +$Package::Stash::PP::VERSION = '0.37'; +use strict; +use warnings; +# ABSTRACT: pure perl implementation of the Package::Stash API + +use B; +use Carp qw(confess); +use Scalar::Util qw(blessed reftype weaken); +use Symbol; +# before 5.12, assigning to the ISA glob would make it lose its magical ->isa +# powers +use constant BROKEN_ISA_ASSIGNMENT => ($] < 5.012); +# before 5.10, stashes don't ever seem to drop to a refcount of zero, so +# weakening them isn't helpful +use constant BROKEN_WEAK_STASH => ($] < 5.010); +# before 5.10, the scalar slot was always treated as existing if the +# glob existed +use constant BROKEN_SCALAR_INITIALIZATION => ($] < 5.010); +# add_method on anon stashes triggers rt.perl #1804 otherwise +# fixed in perl commit v5.13.3-70-g0fe688f +use constant BROKEN_GLOB_ASSIGNMENT => ($] < 5.013004); +# pre-5.10, ->isa lookups were cached in the ::ISA::CACHE:: slot +use constant HAS_ISA_CACHE => ($] < 5.010); + + +sub new { + my $class = shift; + my ($package) = @_; + + if (!defined($package) || (ref($package) && reftype($package) ne 'HASH')) { + confess "Package::Stash->new must be passed the name of the " + . "package to access"; + } + elsif (ref($package) && reftype($package) eq 'HASH') { + confess "The PP implementation of Package::Stash does not support " + . "anonymous stashes before perl 5.14" + if BROKEN_GLOB_ASSIGNMENT; + + return bless { + 'namespace' => $package, + }, $class; + } + elsif ($package =~ /\A[0-9A-Z_a-z]+(?:::[0-9A-Z_a-z]+)*\z/) { + return bless { + 'package' => $package, + }, $class; + } + else { + confess "$package is not a module name"; + } + +} + +sub name { + confess "Can't call name as a class method" + unless blessed($_[0]); + confess "Can't get the name of an anonymous package" + unless defined($_[0]->{package}); + return $_[0]->{package}; +} + +sub namespace { + confess "Can't call namespace as a class method" + unless blessed($_[0]); + + if (BROKEN_WEAK_STASH) { + no strict 'refs'; + return \%{$_[0]->name . '::'}; + } + else { + return $_[0]->{namespace} if defined $_[0]->{namespace}; + + { + no strict 'refs'; + $_[0]->{namespace} = \%{$_[0]->name . '::'}; + } + + weaken($_[0]->{namespace}); + + return $_[0]->{namespace}; + } +} + +{ + my %SIGIL_MAP = ( + '$' => 'SCALAR', + '@' => 'ARRAY', + '%' => 'HASH', + '&' => 'CODE', + '' => 'IO', + ); + + sub _deconstruct_variable_name { + my ($variable) = @_; + + my @ret; + if (ref($variable) eq 'HASH') { + @ret = @{$variable}{qw[name sigil type]}; + } + else { + (defined $variable && length $variable) + || confess "You must pass a variable name"; + + my $sigil = substr($variable, 0, 1, ''); + + if (exists $SIGIL_MAP{$sigil}) { + @ret = ($variable, $sigil, $SIGIL_MAP{$sigil}); + } + else { + @ret = ("${sigil}${variable}", '', $SIGIL_MAP{''}); + } + } + + # XXX in pure perl, this will access things in inner packages, + # in xs, this will segfault - probably look more into this at + # some point + ($ret[0] !~ /::/) + || confess "Variable names may not contain ::"; + + return @ret; + } +} + +sub _valid_for_type { + my ($value, $type) = @_; + if ($type eq 'HASH' || $type eq 'ARRAY' + || $type eq 'IO' || $type eq 'CODE') { + return reftype($value) eq $type; + } + else { + my $ref = reftype($value); + return !defined($ref) || $ref eq 'SCALAR' || $ref eq 'REF' || $ref eq 'LVALUE' || $ref eq 'REGEXP' || $ref eq 'VSTRING'; + } +} + +sub add_symbol { + my ($self, $variable, $initial_value, %opts) = @_; + + my ($name, $sigil, $type) = _deconstruct_variable_name($variable); + + if (@_ > 2) { + _valid_for_type($initial_value, $type) + || confess "$initial_value is not of type $type"; + + # cheap fail-fast check for PERLDBf_SUBLINE and '&' + if ($^P and $^P & 0x10 && $sigil eq '&') { + 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{$self->name . '::' . $name} = "$filename:$first_line_num-$last_line_num"; + } + } + + if (BROKEN_GLOB_ASSIGNMENT) { + if (@_ > 2) { + no strict 'refs'; + no warnings 'redefine'; + *{ $self->name . '::' . $name } = ref $initial_value + ? $initial_value : \$initial_value; + } + else { + no strict 'refs'; + if (BROKEN_ISA_ASSIGNMENT && $name eq 'ISA') { + *{ $self->name . '::' . $name }; + } + else { + my $undef = _undef_ref_for_type($type); + *{ $self->name . '::' . $name } = $undef; + } + } + } + else { + my $namespace = $self->namespace; + { + # using glob aliasing instead of Symbol::gensym, because otherwise, + # magic doesn't get applied properly. + # see <20120710063744.19360.qmail@lists-nntp.develooper.com> on p5p + local *__ANON__:: = $namespace; + no strict 'refs'; + no warnings 'void'; + no warnings 'once'; + *{"__ANON__::$name"}; + } + + if (@_ > 2) { + no warnings 'redefine'; + *{ $namespace->{$name} } = ref $initial_value + ? $initial_value : \$initial_value; + } + else { + return if BROKEN_ISA_ASSIGNMENT && $name eq 'ISA'; + *{ $namespace->{$name} } = _undef_ref_for_type($type); + } + } +} + +sub _undef_ref_for_type { + my ($type) = @_; + + if ($type eq 'ARRAY') { + return []; + } + elsif ($type eq 'HASH') { + return {}; + } + elsif ($type eq 'SCALAR') { + return \undef; + } + elsif ($type eq 'IO') { + return Symbol::geniosym; + } + elsif ($type eq 'CODE') { + confess "Don't know how to vivify CODE variables"; + } + else { + confess "Unknown type $type in vivication"; + } +} + +sub remove_glob { + my ($self, $name) = @_; + delete $self->namespace->{$name}; +} + +sub has_symbol { + my ($self, $variable) = @_; + + my ($name, $sigil, $type) = _deconstruct_variable_name($variable); + + my $namespace = $self->namespace; + + return unless exists $namespace->{$name}; + + my $entry_ref = \$namespace->{$name}; + if (reftype($entry_ref) eq 'GLOB') { + if ($type eq 'SCALAR') { + if (BROKEN_SCALAR_INITIALIZATION) { + return defined ${ *{$entry_ref}{$type} }; + } + else { + my $sv = B::svref_2object($entry_ref)->SV; + return $sv->isa('B::SV') + || ($sv->isa('B::SPECIAL') + && $B::specialsv_name[$$sv] ne 'Nullsv'); + } + } + else { + return defined *{$entry_ref}{$type}; + } + } + else { + # a symbol table entry can be -1 (stub), string (stub with prototype), + # or reference (constant) + return $type eq 'CODE'; + } +} + +sub get_symbol { + my ($self, $variable, %opts) = @_; + + my ($name, $sigil, $type) = _deconstruct_variable_name($variable); + + my $namespace = $self->namespace; + + if (!exists $namespace->{$name}) { + if ($opts{vivify}) { + $self->add_symbol($variable); + } + else { + return undef; + } + } + + my $entry_ref = \$namespace->{$name}; + + if (ref($entry_ref) eq 'GLOB') { + return *{$entry_ref}{$type}; + } + else { + if ($type eq 'CODE') { + if (BROKEN_GLOB_ASSIGNMENT || defined($self->{package})) { + no strict 'refs'; + return \&{ $self->name . '::' . $name }; + } + + # XXX we should really be able to support arbitrary anonymous + # stashes here... (not just via Package::Anon) + if (blessed($namespace) && $namespace->isa('Package::Anon')) { + # ->can will call gv_init for us, which inflates the glob + # don't know how to do this in general + $namespace->bless(\(my $foo))->can($name); + } + else { + confess "Don't know how to inflate a " . ref($entry_ref) + . " into a full coderef (perhaps you could use" + . " Package::Anon instead of a bare stash?)" + } + + return *{ $namespace->{$name} }{CODE}; + } + else { + return undef; + } + } +} + +sub get_or_add_symbol { + my $self = shift; + $self->get_symbol(@_, vivify => 1); +} + +sub remove_symbol { + my ($self, $variable) = @_; + + my ($name, $sigil, $type) = _deconstruct_variable_name($variable); + + # FIXME: + # no doubt this is grossly inefficient and + # could be done much easier and faster in XS + + my %desc = ( + SCALAR => { sigil => '$', type => 'SCALAR', name => $name }, + ARRAY => { sigil => '@', type => 'ARRAY', name => $name }, + HASH => { sigil => '%', type => 'HASH', name => $name }, + CODE => { sigil => '&', type => 'CODE', name => $name }, + IO => { sigil => '', type => 'IO', name => $name }, + ); + confess "This should never ever ever happen" if !$desc{$type}; + + my @types_to_store = grep { $type ne $_ && $self->has_symbol($desc{$_}) } + keys %desc; + my %values = map { $_, $self->get_symbol($desc{$_}) } @types_to_store; + + $values{SCALAR} = $self->get_symbol($desc{SCALAR}) + if !defined $values{SCALAR} + && $type ne 'SCALAR' + && BROKEN_SCALAR_INITIALIZATION; + + $self->remove_glob($name); + + $self->add_symbol($desc{$_} => $values{$_}) + for grep { defined $values{$_} } keys %values; +} + +sub list_all_symbols { + my ($self, $type_filter) = @_; + + my $namespace = $self->namespace; + if (HAS_ISA_CACHE) { + return grep { $_ ne '::ISA::CACHE::' } keys %{$namespace} + unless defined $type_filter; + } + else { + return keys %{$namespace} + unless defined $type_filter; + } + + # NOTE: + # or we can filter based on + # type (SCALAR|ARRAY|HASH|CODE) + if ($type_filter eq 'CODE') { + return grep { + # any non-typeglob in the symbol table is a constant or stub + ref(\$namespace->{$_}) ne 'GLOB' + # regular subs are stored in the CODE slot of the typeglob + || defined(*{$namespace->{$_}}{CODE}) + } keys %{$namespace}; + } + elsif ($type_filter eq 'SCALAR') { + return grep { + !(HAS_ISA_CACHE && $_ eq '::ISA::CACHE::') && + (BROKEN_SCALAR_INITIALIZATION + ? (ref(\$namespace->{$_}) eq 'GLOB' + && defined(${*{$namespace->{$_}}{'SCALAR'}})) + : (do { + my $entry = \$namespace->{$_}; + ref($entry) eq 'GLOB' + && B::svref_2object($entry)->SV->isa('B::SV') + })) + } keys %{$namespace}; + } + else { + return grep { + ref(\$namespace->{$_}) eq 'GLOB' + && defined(*{$namespace->{$_}}{$type_filter}) + } keys %{$namespace}; + } +} + +sub get_all_symbols { + my ($self, $type_filter) = @_; + + my $namespace = $self->namespace; + return { %{$namespace} } unless defined $type_filter; + + return { + map { $_ => $self->get_symbol({name => $_, type => $type_filter}) } + $self->list_all_symbols($type_filter) + } +} + + +1; + +__END__ + +=pod + +=encoding UTF-8 + +=head1 NAME + +Package::Stash::PP - pure perl implementation of the Package::Stash API + +=head1 VERSION + +version 0.37 + +=head1 SYNOPSIS + + use Package::Stash; + +=head1 DESCRIPTION + +This is a backend for L<Package::Stash> implemented in pure perl, for those without a compiler or who would like to use this inline in scripts. + +=head1 BUGS + +=over 4 + +=item * remove_symbol also replaces the associated typeglob + +This can cause unexpected behavior when doing manipulation at compile time - +removing subroutines will still allow them to be called from within the package +as subroutines (although they will not be available as methods). This can be +considered a feature in some cases (this is how L<namespace::clean> works, for +instance), but should not be relied upon - use C<remove_glob> directly if you +want this behavior. + +=item * Some minor memory leaks + +The pure perl implementation has a couple minor memory leaks (see the TODO +tests in t/20-leaks.t) that I'm having a hard time tracking down - these may be +core perl bugs, it's hard to tell. + +=back + +Please report any bugs through RT: email +C<bug-package-stash at rt.cpan.org>, or browse to +L<http://rt.cpan.org/NoAuth/ReportBug.html?Queue=Package-Stash>. + +=head1 SEE ALSO + +=over 4 + +=item * L<Class::MOP::Package> + +This module is a factoring out of code that used to live here + +=back + +=head1 SUPPORT + +You can find this documentation for this module with the perldoc command. + + perldoc Package::Stash + +You can also look for information at: + +=over 4 + +=item * AnnoCPAN: Annotated CPAN documentation + +L<http://annocpan.org/dist/Package-Stash> + +=item * CPAN Ratings + +L<http://cpanratings.perl.org/d/Package-Stash> + +=item * RT: CPAN's request tracker + +L<http://rt.cpan.org/NoAuth/Bugs.html?Dist=Package-Stash> + +=item * Search CPAN + +L<http://search.cpan.org/dist/Package-Stash> + +=back + +=head1 AUTHOR + +Jesse Luehrs <doy at tozt dot net> + +Mostly copied from code from L<Class::MOP::Package>, by Stevan Little and the +Moose Cabal. + +=for Pod::Coverage BROKEN_ISA_ASSIGNMENT +add_symbol +get_all_symbols +get_or_add_symbol +get_symbol +has_symbol +list_all_symbols +name +namespace +new +remove_glob + +=head1 AUTHOR + +Jesse Luehrs <doy@tozt.net> + +=head1 COPYRIGHT AND LICENSE + +This software is copyright (c) 2014 by Jesse Luehrs. + +This is free software; you can redistribute it and/or modify it under +the same terms as the Perl 5 programming language system itself. + +=cut |