fix coderef vivification
[gitmo/Package-Stash-PP.git] / lib / Package / Stash.pm
CommitLineData
e94260da 1package Package::Stash;
f10f6217 2use strict;
3use warnings;
988beb41 4# ABSTRACT: routines for manipulating stashes
f10f6217 5
6use Carp qw(confess);
7use Scalar::Util qw(reftype);
5d3589c8 8use Symbol;
f4979588 9
f4979588 10=head1 SYNOPSIS
11
e94260da 12 my $stash = Package::Stash->new('Foo');
683542f5 13 $stash->add_package_symbol('%foo', {bar => 1});
14 # $Foo::foo{bar} == 1
15 $stash->has_package_symbol('$foo') # false
16 my $namespace = $stash->namespace;
17 *{ $namespace->{foo} }{HASH} # {bar => 1}
f4979588 18
19=head1 DESCRIPTION
20
683542f5 21Manipulating stashes (Perl's symbol tables) is occasionally necessary, but
22incredibly messy, and easy to get wrong. This module hides all of that behind a
23simple API.
24
56a29840 25NOTE: Most methods in this class require a variable specification that includes
26a sigil. If this sigil is absent, it is assumed to represent the IO slot.
27
988beb41 28=method new $package_name
683542f5 29
e94260da 30Creates a new C<Package::Stash> object, for the package given as the only
683542f5 31argument.
f4979588 32
33=cut
34
f10f6217 35sub new {
36 my $class = shift;
e55970bc 37 my ($package) = @_;
38 my $namespace;
39 {
40 no strict 'refs';
41 # supposedly this caused a bug in earlier perls, but I can't reproduce
42 # it, so re-enabling the caching
43 $namespace = \%{$package . '::'};
44 }
45 return bless {
46 'package' => $package,
47 'namespace' => $namespace,
48 }, $class;
f10f6217 49}
50
988beb41 51=method name
683542f5 52
53Returns the name of the package that this object represents.
54
55=cut
56
f10f6217 57sub name {
58 return $_[0]->{package};
59}
60
988beb41 61=method namespace
683542f5 62
63Returns the raw stash itself.
64
65=cut
66
f10f6217 67sub namespace {
e55970bc 68 return $_[0]->{namespace};
f10f6217 69}
70
71{
72 my %SIGIL_MAP = (
73 '$' => 'SCALAR',
74 '@' => 'ARRAY',
75 '%' => 'HASH',
76 '&' => 'CODE',
56a29840 77 '' => 'IO',
f10f6217 78 );
79
80 sub _deconstruct_variable_name {
81 my ($self, $variable) = @_;
82
56a29840 83 (defined $variable && length $variable)
f10f6217 84 || confess "You must pass a variable name";
85
86 my $sigil = substr($variable, 0, 1, '');
87
56a29840 88 if (exists $SIGIL_MAP{$sigil}) {
89 return ($variable, $sigil, $SIGIL_MAP{$sigil});
90 }
91 else {
92 return ("${sigil}${variable}", '', $SIGIL_MAP{''});
93 }
f10f6217 94 }
95}
96
988beb41 97=method add_package_symbol $variable $value %opts
683542f5 98
99Adds a new package symbol, for the symbol given as C<$variable>, and optionally
100gives it an initial value of C<$value>. C<$variable> should be the name of
101variable including the sigil, so
102
e94260da 103 Package::Stash->new('Foo')->add_package_symbol('%foo')
683542f5 104
105will create C<%Foo::foo>.
106
c61010aa 107Valid options (all optional) are C<filename>, C<first_line_num>, and
108C<last_line_num>.
109
110C<$opts{filename}>, C<$opts{first_line_num}>, and C<$opts{last_line_num}> can
111be used to indicate where the symbol should be regarded as having been defined.
4ada57e0 112Currently these values are only used if the symbol is a subroutine ('C<&>'
c61010aa 113sigil) and only if C<$^P & 0x10> is true, in which case the special C<%DB::sub>
114hash is updated to record the values of C<filename>, C<first_line_num>, and
115C<last_line_num> for the subroutine. If these are not passed, their values are
116inferred (as much as possible) from C<caller> information.
4ada57e0 117
118This is especially useful for debuggers and profilers, which use C<%DB::sub> to
119determine where the source code for a subroutine can be found. See
120L<http://perldoc.perl.org/perldebguts.html#Debugger-Internals> for more
121information about C<%DB::sub>.
122
683542f5 123=cut
124
3634ce60 125sub _valid_for_type {
126 my $self = shift;
127 my ($value, $type) = @_;
128 if ($type eq 'HASH' || $type eq 'ARRAY'
129 || $type eq 'IO' || $type eq 'CODE') {
130 return reftype($value) eq $type;
131 }
132 else {
133 my $ref = reftype($value);
134 return !defined($ref) || $ref eq 'SCALAR' || $ref eq 'REF' || $ref eq 'LVALUE';
135 }
136}
137
f10f6217 138sub add_package_symbol {
640de369 139 my ($self, $variable, $initial_value, %opts) = @_;
f10f6217 140
141 my ($name, $sigil, $type) = ref $variable eq 'HASH'
142 ? @{$variable}{qw[name sigil type]}
143 : $self->_deconstruct_variable_name($variable);
144
4ada57e0 145 my $pkg = $self->name;
146
3634ce60 147 if (@_ > 2) {
148 $self->_valid_for_type($initial_value, $type)
149 || confess "$initial_value is not of type $type";
3634ce60 150
4ada57e0 151 # cheap fail-fast check for PERLDBf_SUBLINE and '&'
152 if ($^P and $^P & 0x10 && $sigil eq '&') {
640de369 153 my $filename = $opts{filename};
154 my $first_line_num = $opts{first_line_num};
4ada57e0 155
640de369 156 (undef, $filename, $first_line_num) = caller
4ada57e0 157 if not defined $filename;
640de369 158
159 my $last_line_num = $opts{last_line_num} || ($first_line_num ||= 0);
4ada57e0 160
161 # http://perldoc.perl.org/perldebguts.html#Debugger-Internals
640de369 162 $DB::sub{$pkg . '::' . $name} = "$filename:$first_line_num-$last_line_num";
4ada57e0 163 }
164 }
f10f6217 165
166 no strict 'refs';
167 no warnings 'redefine', 'misc', 'prototype';
168 *{$pkg . '::' . $name} = ref $initial_value ? $initial_value : \$initial_value;
169}
170
988beb41 171=method remove_package_glob $name
683542f5 172
173Removes all package variables with the given name, regardless of sigil.
174
175=cut
176
f10f6217 177sub remove_package_glob {
178 my ($self, $name) = @_;
179 no strict 'refs';
180 delete ${$self->name . '::'}{$name};
181}
182
183# ... these functions deal with stuff on the namespace level
184
988beb41 185=method has_package_symbol $variable
683542f5 186
187Returns whether or not the given package variable (including sigil) exists.
188
189=cut
190
f10f6217 191sub has_package_symbol {
192 my ($self, $variable) = @_;
193
194 my ($name, $sigil, $type) = ref $variable eq 'HASH'
195 ? @{$variable}{qw[name sigil type]}
196 : $self->_deconstruct_variable_name($variable);
197
198 my $namespace = $self->namespace;
199
200 return unless exists $namespace->{$name};
201
202 my $entry_ref = \$namespace->{$name};
203 if (reftype($entry_ref) eq 'GLOB') {
204 if ( $type eq 'SCALAR' ) {
205 return defined ${ *{$entry_ref}{SCALAR} };
206 }
207 else {
208 return defined *{$entry_ref}{$type};
209 }
210 }
211 else {
212 # a symbol table entry can be -1 (stub), string (stub with prototype),
213 # or reference (constant)
214 return $type eq 'CODE';
215 }
216}
217
988beb41 218=method get_package_symbol $variable
683542f5 219
220Returns the value of the given package variable (including sigil).
221
222=cut
223
f10f6217 224sub get_package_symbol {
e55803fc 225 my ($self, $variable, %opts) = @_;
f10f6217 226
227 my ($name, $sigil, $type) = ref $variable eq 'HASH'
228 ? @{$variable}{qw[name sigil type]}
229 : $self->_deconstruct_variable_name($variable);
230
231 my $namespace = $self->namespace;
232
7486ccf3 233 if (!exists $namespace->{$name}) {
234 if ($opts{vivify}) {
235 if ($type eq 'ARRAY') {
236 $self->add_package_symbol(
237 $variable,
238 # setting our own arrayref manually loses the magicalness
239 # or something
240 $name eq 'ISA' ? () : ([])
241 );
242 }
243 elsif ($type eq 'HASH') {
244 $self->add_package_symbol($variable, {});
245 }
246 elsif ($type eq 'SCALAR') {
247 $self->add_package_symbol($variable);
248 }
249 elsif ($type eq 'IO') {
250 $self->add_package_symbol($variable, Symbol::geniosym);
251 }
252 elsif ($type eq 'CODE') {
253 confess "Don't know how to vivify CODE variables";
254 }
255 else {
256 confess "Unknown type $type in vivication";
257 }
5d3589c8 258 }
e55803fc 259 else {
7486ccf3 260 if ($type eq 'CODE') {
261 # this effectively "de-vivifies" the code slot. if we don't do
262 # this, referencing the coderef at the end of this function
263 # will cause perl to auto-vivify a stub coderef in the slot,
264 # which isn't what we want
265 $self->add_package_symbol($variable);
266 }
e55803fc 267 }
30d1a098 268 }
f10f6217 269
270 my $entry_ref = \$namespace->{$name};
271
272 if (ref($entry_ref) eq 'GLOB') {
273 return *{$entry_ref}{$type};
274 }
275 else {
276 if ($type eq 'CODE') {
277 no strict 'refs';
278 return \&{ $self->name . '::' . $name };
279 }
280 else {
281 return undef;
282 }
283 }
284}
285
988beb41 286=method get_or_add_package_symbol $variable
e55803fc 287
288Like C<get_package_symbol>, except that it will return an empty hashref or
289arrayref if the variable doesn't exist.
290
291=cut
292
293sub get_or_add_package_symbol {
294 my $self = shift;
295 $self->get_package_symbol(@_, vivify => 1);
296}
297
988beb41 298=method remove_package_symbol $variable
683542f5 299
300Removes the package variable described by C<$variable> (which includes the
301sigil); other variables with the same name but different sigils will be
302untouched.
303
304=cut
305
f10f6217 306sub remove_package_symbol {
307 my ($self, $variable) = @_;
308
309 my ($name, $sigil, $type) = ref $variable eq 'HASH'
310 ? @{$variable}{qw[name sigil type]}
311 : $self->_deconstruct_variable_name($variable);
312
313 # FIXME:
9a3d1390 314 # no doubt this is grossly inefficient and
f10f6217 315 # could be done much easier and faster in XS
316
b1a00d0e 317 my ($scalar_desc, $array_desc, $hash_desc, $code_desc, $io_desc) = (
f10f6217 318 { sigil => '$', type => 'SCALAR', name => $name },
319 { sigil => '@', type => 'ARRAY', name => $name },
320 { sigil => '%', type => 'HASH', name => $name },
321 { sigil => '&', type => 'CODE', name => $name },
b1a00d0e 322 { sigil => '', type => 'IO', name => $name },
f10f6217 323 );
324
b1a00d0e 325 my ($scalar, $array, $hash, $code, $io);
f10f6217 326 if ($type eq 'SCALAR') {
327 $array = $self->get_package_symbol($array_desc) if $self->has_package_symbol($array_desc);
328 $hash = $self->get_package_symbol($hash_desc) if $self->has_package_symbol($hash_desc);
329 $code = $self->get_package_symbol($code_desc) if $self->has_package_symbol($code_desc);
b1a00d0e 330 $io = $self->get_package_symbol($io_desc) if $self->has_package_symbol($io_desc);
f10f6217 331 }
332 elsif ($type eq 'ARRAY') {
42fa5cfc 333 $scalar = $self->get_package_symbol($scalar_desc);
f10f6217 334 $hash = $self->get_package_symbol($hash_desc) if $self->has_package_symbol($hash_desc);
335 $code = $self->get_package_symbol($code_desc) if $self->has_package_symbol($code_desc);
b1a00d0e 336 $io = $self->get_package_symbol($io_desc) if $self->has_package_symbol($io_desc);
f10f6217 337 }
338 elsif ($type eq 'HASH') {
42fa5cfc 339 $scalar = $self->get_package_symbol($scalar_desc);
f10f6217 340 $array = $self->get_package_symbol($array_desc) if $self->has_package_symbol($array_desc);
341 $code = $self->get_package_symbol($code_desc) if $self->has_package_symbol($code_desc);
b1a00d0e 342 $io = $self->get_package_symbol($io_desc) if $self->has_package_symbol($io_desc);
f10f6217 343 }
344 elsif ($type eq 'CODE') {
42fa5cfc 345 $scalar = $self->get_package_symbol($scalar_desc);
f10f6217 346 $array = $self->get_package_symbol($array_desc) if $self->has_package_symbol($array_desc);
347 $hash = $self->get_package_symbol($hash_desc) if $self->has_package_symbol($hash_desc);
b1a00d0e 348 $io = $self->get_package_symbol($io_desc) if $self->has_package_symbol($io_desc);
349 }
350 elsif ($type eq 'IO') {
42fa5cfc 351 $scalar = $self->get_package_symbol($scalar_desc);
b1a00d0e 352 $array = $self->get_package_symbol($array_desc) if $self->has_package_symbol($array_desc);
353 $hash = $self->get_package_symbol($hash_desc) if $self->has_package_symbol($hash_desc);
354 $code = $self->get_package_symbol($code_desc) if $self->has_package_symbol($code_desc);
f10f6217 355 }
356 else {
357 confess "This should never ever ever happen";
358 }
359
360 $self->remove_package_glob($name);
361
42fa5cfc 362 $self->add_package_symbol($scalar_desc => $scalar);
f10f6217 363 $self->add_package_symbol($array_desc => $array) if defined $array;
364 $self->add_package_symbol($hash_desc => $hash) if defined $hash;
365 $self->add_package_symbol($code_desc => $code) if defined $code;
b1a00d0e 366 $self->add_package_symbol($io_desc => $io) if defined $io;
f10f6217 367}
368
988beb41 369=method list_all_package_symbols $type_filter
683542f5 370
371Returns a list of package variable names in the package, without sigils. If a
372C<type_filter> is passed, it is used to select package variables of a given
373type, where valid types are the slots of a typeglob ('SCALAR', 'CODE', 'HASH',
374etc).
375
376=cut
377
f10f6217 378sub list_all_package_symbols {
379 my ($self, $type_filter) = @_;
380
381 my $namespace = $self->namespace;
382 return keys %{$namespace} unless defined $type_filter;
383
384 # NOTE:
9a3d1390 385 # or we can filter based on
f10f6217 386 # type (SCALAR|ARRAY|HASH|CODE)
387 if ($type_filter eq 'CODE') {
388 return grep {
389 (ref($namespace->{$_})
390 ? (ref($namespace->{$_}) eq 'SCALAR')
391 : (ref(\$namespace->{$_}) eq 'GLOB'
392 && defined(*{$namespace->{$_}}{CODE})));
393 } keys %{$namespace};
394 } else {
395 return grep { *{$namespace->{$_}}{$type_filter} } keys %{$namespace};
396 }
397}
f4979588 398
0992f4ec 399=head1 BUGS
400
401No known bugs.
402
403Please report any bugs through RT: email
404C<bug-package-stash at rt.cpan.org>, or browse to
405L<http://rt.cpan.org/NoAuth/ReportBug.html?Queue=Package-Stash>.
406
f4979588 407=head1 SEE ALSO
408
f4979588 409=over 4
410
988beb41 411=item * L<Class::MOP::Package>
f4979588 412
988beb41 413This module is a factoring out of code that used to live here
f4979588 414
415=back
416
0992f4ec 417=head1 SUPPORT
418
419You can find this documentation for this module with the perldoc command.
420
421 perldoc Package::Stash
422
423You can also look for information at:
424
425=over 4
426
427=item * AnnoCPAN: Annotated CPAN documentation
428
429L<http://annocpan.org/dist/Package-Stash>
430
431=item * CPAN Ratings
432
433L<http://cpanratings.perl.org/d/Package-Stash>
434
435=item * RT: CPAN's request tracker
436
437L<http://rt.cpan.org/NoAuth/Bugs.html?Dist=Package-Stash>
438
439=item * Search CPAN
440
441L<http://search.cpan.org/dist/Package-Stash>
442
443=back
444
445=head1 AUTHOR
446
447Jesse Luehrs <doy at tozt dot net>
448
449Mostly copied from code from L<Class::MOP::Package>, by Stevan Little and the
450Moose Cabal.
451
f4979588 452=cut
453
4541;