X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2Fnamespace%2Fclean.pm;h=5d41c9de630c8e0b164cfce5abd0748398c1909e;hb=39dbc69b27dbf7800719b3e60a9042b9a1ae58f8;hp=93ecb9f8671aed10bd25ed1531464043bc0506c3;hpb=fcfe7810e5ba4f72dadf41d7d7cd92621bbcad4f;p=p5sagit%2Fnamespace-clean.git diff --git a/lib/namespace/clean.pm b/lib/namespace/clean.pm index 93ecb9f..5d41c9d 100644 --- a/lib/namespace/clean.pm +++ b/lib/namespace/clean.pm @@ -1,26 +1,238 @@ package namespace::clean; -=head1 NAME +use warnings; +use strict; -namespace::clean - Keep imports and functions out of your namespace +our $VERSION = '0.25'; +our $STORAGE_VAR = '__NAMESPACE_CLEAN_STORAGE'; -=cut +use B::Hooks::EndOfScope 'on_scope_end'; -use warnings; -use strict; +# FIXME This is a crock of shit, needs to go away +# currently here to work around https://rt.cpan.org/Ticket/Display.html?id=74151 +# kill with fire when PS::XS is *finally* fixed +BEGIN { + my $provider; + + if ( $] < 5.008007 ) { + require Package::Stash::PP; + $provider = 'Package::Stash::PP'; + } + else { + require Package::Stash; + $provider = 'Package::Stash'; + } + eval <<"EOS" or die $@; + +sub stash_for (\$) { + $provider->new(\$_[0]); +} + +1; + +EOS +} + +use namespace::clean::_Util qw( DEBUGGER_NEEDS_CV_RENAME DEBUGGER_NEEDS_CV_PIVOT ); + +# Debugger fixup necessary before perl 5.15.5 +# +# In perl 5.8.9-5.12, it assumes that sub_fullname($sub) can +# always be used to find the CV again. +# In perl 5.8.8 and 5.14, it assumes that the name of the glob +# passed to entersub can be used to find the CV. +# since we are deleting the glob where the subroutine was originally +# defined, those assumptions no longer hold. +# +# So in 5.8.9-5.12 we need to move it elsewhere and point the +# CV's name to the new glob. +# +# In 5.8.8 and 5.14 we move it elsewhere and rename the +# original glob by assigning the new glob back to it. +my $sub_utils_loaded; +my $DebuggerFixup = sub { + my ($f, $sub, $cleanee_stash, $deleted_stash) = @_; + + if (DEBUGGER_NEEDS_CV_RENAME) { + # + # Note - both get_subname and set_subname are only compiled when CV_RENAME + # is true ( the 5.8.9 ~ 5.12 range ). On other perls this entire block is + # constant folded away, and so are the definitions in ::_Util + # + # Do not be surprised that they are missing without DEBUGGER_NEEDS_CV_RENAME + # + namespace::clean::_Util::get_subname( $sub ) eq ( $cleanee_stash->name . "::$f" ) + and + $deleted_stash->add_symbol( + "&$f", + namespace::clean::_Util::set_subname( $deleted_stash->name . "::$f", $sub ), + ); + } + else { + $deleted_stash->add_symbol("&$f", $sub); + } +}; + +my $RemoveSubs = sub { + my $cleanee = shift; + my $store = shift; + my $cleanee_stash = stash_for($cleanee); + my $deleted_stash; + + SYMBOL: + for my $f (@_) { + + # ignore already removed symbols + next SYMBOL if $store->{exclude}{ $f }; + + my $sub = $cleanee_stash->get_symbol("&$f") + or next SYMBOL; + + my $need_debugger_fixup = + ( DEBUGGER_NEEDS_CV_RENAME or DEBUGGER_NEEDS_CV_PIVOT ) + && + $^P + && + ref(my $globref = \$cleanee_stash->namespace->{$f}) eq 'GLOB' + ; + + if ($need_debugger_fixup) { + # convince the Perl debugger to work + # see the comment on top of $DebuggerFixup + $DebuggerFixup->( + $f, + $sub, + $cleanee_stash, + $deleted_stash ||= stash_for("namespace::clean::deleted::$cleanee"), + ); + } + + my @symbols = map { + my $name = $_ . $f; + my $def = $cleanee_stash->get_symbol($name); + defined($def) ? [$name, $def] : () + } '$', '@', '%', ''; + + $cleanee_stash->remove_glob($f); -use vars qw( $VERSION $STORAGE_VAR $SCOPE_HOOK_KEY $SCOPE_EXPLICIT ); -use Symbol qw( qualify_to_ref ); -use B::Hooks::EndOfScope; + # if this perl needs no renaming trick we need to + # rename the original glob after the fact + # (see commend of $DebuggerFixup + if (DEBUGGER_NEEDS_CV_PIVOT && $need_debugger_fixup) { + *$globref = $deleted_stash->namespace->{$f}; + } + + $cleanee_stash->add_symbol(@$_) for @symbols; + } +}; + +sub clean_subroutines { + my ($nc, $cleanee, @subs) = @_; + $RemoveSubs->($cleanee, {}, @subs); +} + +sub import { + my ($pragma, @args) = @_; + + my (%args, $is_explicit); + + ARG: + while (@args) { + + if ($args[0] =~ /^\-/) { + my $key = shift @args; + my $value = shift @args; + $args{ $key } = $value; + } + else { + $is_explicit++; + last ARG; + } + } -=head1 VERSION + my $cleanee = exists $args{ -cleanee } ? $args{ -cleanee } : scalar caller; + if ($is_explicit) { + on_scope_end { + $RemoveSubs->($cleanee, {}, @args); + }; + } + else { + + # calling class, all current functions and our storage + my $functions = $pragma->get_functions($cleanee); + my $store = $pragma->get_class_store($cleanee); + my $stash = stash_for($cleanee); + + # except parameter can be array ref or single value + my %except = map {( $_ => 1 )} ( + $args{ -except } + ? ( ref $args{ -except } eq 'ARRAY' ? @{ $args{ -except } } : $args{ -except } ) + : () + ); + + # register symbols for removal, if they have a CODE entry + for my $f (keys %$functions) { + next if $except{ $f }; + next unless $stash->has_symbol("&$f"); + $store->{remove}{ $f } = 1; + } + + # register EOF handler on first call to import + unless ($store->{handler_is_installed}) { + on_scope_end { + $RemoveSubs->($cleanee, $store, keys %{ $store->{remove} }); + }; + $store->{handler_is_installed} = 1; + } + + return 1; + } +} + +sub unimport { + my ($pragma, %args) = @_; + + # the calling class, the current functions and our storage + my $cleanee = exists $args{ -cleanee } ? $args{ -cleanee } : scalar caller; + my $functions = $pragma->get_functions($cleanee); + my $store = $pragma->get_class_store($cleanee); + + # register all unknown previous functions as excluded + for my $f (keys %$functions) { + next if $store->{remove}{ $f } + or $store->{exclude}{ $f }; + $store->{exclude}{ $f } = 1; + } + + return 1; +} + +sub get_class_store { + my ($pragma, $class) = @_; + my $stash = stash_for($class); + my $var = "%$STORAGE_VAR"; + $stash->add_symbol($var, {}) + unless $stash->has_symbol($var); + return $stash->get_symbol($var); +} + +sub get_functions { + my ($pragma, $class) = @_; + + my $stash = stash_for($class); + return { + map { $_ => $stash->get_symbol("&$_") } + $stash->list_all_symbols('CODE') + }; +} + +'Danger! Laws of Thermodynamics may not apply.' -0.11 +__END__ -=cut +=head1 NAME -$VERSION = '0.11'; -$STORAGE_VAR = '__NAMESPACE_CLEAN_STORAGE'; +namespace::clean - Keep imports and functions out of your namespace =head1 SYNOPSIS @@ -60,7 +272,7 @@ $STORAGE_VAR = '__NAMESPACE_CLEAN_STORAGE'; When you define a function, or import one, into a Perl package, it will naturally also be available as a method. This does not per se cause problems, but it can complicate subclassing and, for example, plugin -classes that are included via multiple inheritance by loading them as +classes that are included via multiple inheritance by loading them as base classes. The C pragma will remove all previously declared or @@ -81,9 +293,9 @@ be a module exporting an C method along with some functions: If you just want to C<-except> a single sub, you can pass it directly. For more than one value you have to use an array reference. -=head2 Explicitely removing functions when your scope is compiled +=head2 Explicitly removing functions when your scope is compiled -It is also possible to explicitely tell C what packages +It is also possible to explicitly tell C what packages to remove when the surrounding scope has finished compiling. Here is an example: @@ -137,11 +349,6 @@ just want to remove subroutines, try L. =head1 METHODS -You shouldn't need to call any of these. Just C the package at the -appropriate place. - -=cut - =head2 clean_subroutines This exposes the actual subroutine-removal logic. @@ -153,103 +360,11 @@ subroutines B and not wait for scope end. If you want to have this effect at a specific time (e.g. C acts on scope compile end) it is your responsibility to make sure it runs at that time. -=cut - -my $RemoveSubs = sub { - - my $cleanee = shift; - my $store = shift; - SYMBOL: - for my $f (@_) { - - # ignore already removed symbols - next SYMBOL if $store->{exclude}{ $f }; - no strict 'refs'; - - # keep original value to restore non-code slots - { no warnings 'uninitialized'; # fix possible unimports - local *__tmp = *{ ${ "${cleanee}::" }{ $f } }; - delete ${ "${cleanee}::" }{ $f }; - } - - SLOT: - # restore non-code slots to symbol - for my $t (qw( SCALAR ARRAY HASH IO FORMAT )) { - next SLOT unless defined *__tmp{ $t }; - *{ "${cleanee}::$f" } = *__tmp{ $t }; - } - } -}; - -sub clean_subroutines { - my ($nc, $cleanee, @subs) = @_; - $RemoveSubs->($cleanee, {}, @subs); -} - =head2 import Makes a snapshot of the current defined functions and installs a L hook in the current scope to invoke the cleanups. -=cut - -sub import { - my ($pragma, @args) = @_; - - my (%args, $is_explicit); - - ARG: - while (@args) { - - if ($args[0] =~ /^\-/) { - my $key = shift @args; - my $value = shift @args; - $args{ $key } = $value; - } - else { - $is_explicit++; - last ARG; - } - } - - my $cleanee = exists $args{ -cleanee } ? $args{ -cleanee } : scalar caller; - if ($is_explicit) { - on_scope_end { - $RemoveSubs->($cleanee, {}, @args); - }; - } - else { - - # calling class, all current functions and our storage - my $functions = $pragma->get_functions($cleanee); - my $store = $pragma->get_class_store($cleanee); - - # except parameter can be array ref or single value - my %except = map {( $_ => 1 )} ( - $args{ -except } - ? ( ref $args{ -except } eq 'ARRAY' ? @{ $args{ -except } } : $args{ -except } ) - : () - ); - - # register symbols for removal, if they have a CODE entry - for my $f (keys %$functions) { - next if $except{ $f }; - next unless $functions->{ $f } - and *{ $functions->{ $f } }{CODE}; - $store->{remove}{ $f } = 1; - } - - # register EOF handler on first call to import - unless ($store->{handler_is_installed}) { - on_scope_end { - $RemoveSubs->($cleanee, $store, keys %{ $store->{remove} }); - }; - $store->{handler_is_installed} = 1; - } - - return 1; - } -} =head2 unimport @@ -259,62 +374,20 @@ This method will be called when you do a It will start a new section of code that defines functions to clean up. -=cut - -sub unimport { - my ($pragma, %args) = @_; - - # the calling class, the current functions and our storage - my $cleanee = exists $args{ -cleanee } ? $args{ -cleanee } : scalar caller; - my $functions = $pragma->get_functions($cleanee); - my $store = $pragma->get_class_store($cleanee); - - # register all unknown previous functions as excluded - for my $f (keys %$functions) { - next if $store->{remove}{ $f } - or $store->{exclude}{ $f }; - $store->{exclude}{ $f } = 1; - } - - return 1; -} - =head2 get_class_store -This returns a reference to a hash in a passed package containing +This returns a reference to a hash in a passed package containing information about function names included and excluded from removal. -=cut - -sub get_class_store { - my ($pragma, $class) = @_; - no strict 'refs'; - return \%{ "${class}::${STORAGE_VAR}" }; -} - =head2 get_functions Takes a class as argument and returns all currently defined functions in it as a hash reference with the function name as key and a typeglob reference to the symbol as value. -=cut - -sub get_functions { - my ($pragma, $class) = @_; - - return { - map { @$_ } # key => value - grep { *{ $_->[1] }{CODE} } # only functions - map { [$_, qualify_to_ref( $_, $class )] } # get globref - grep { $_ !~ /::$/ } # no packages - do { no strict 'refs'; keys %{ "${class}::" } } # symbol entries - }; -} - =head1 IMPLEMENTATION DETAILS -This module works through the effect that a +This module works through the effect that a delete $SomePackage::{foo}; @@ -333,17 +406,38 @@ use C instead. L -=head1 AUTHOR AND COPYRIGHT +=head1 THANKS -Robert 'phaylon' Sedlacek Crs@474.atE>, with many thanks to -Matt S Trout for the inspiration on the whole idea. +Many thanks to Matt S Trout for the inspiration on the whole idea. -=head1 LICENSE +=head1 AUTHORS -This program is free software; you can redistribute it and/or modify -it under the same terms as perl itself. +=over -=cut +=item * -no warnings; -'Danger! Laws of Thermodynamics may not apply.' +Robert 'phaylon' Sedlacek + +=item * + +Florian Ragwitz + +=item * + +Jesse Luehrs + +=item * + +Peter Rabbitson + +=item * + +Father Chrysostomos + +=back + +=head1 COPYRIGHT AND LICENSE + +This software is copyright (c) 2011 by L + +This is free software; you can redistribute it and/or modify it under the same terms as the Perl 5 programming language system itself.