X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FClass%2FMOP%2FMethod.pm;h=c07895d3ed617671cd21151c700f02293eb58041;hb=195f5bf802be0d32838ddb6077866a5951c1d918;hp=75205b2715e1c3676a0e7a7afc97534da5641580;hpb=a2e85e6c752e6dd43555a7eb5623696a86afa858;p=gitmo%2FClass-MOP.git diff --git a/lib/Class/MOP/Method.pm b/lib/Class/MOP/Method.pm index 75205b2..c07895d 100644 --- a/lib/Class/MOP/Method.pm +++ b/lib/Class/MOP/Method.pm @@ -5,25 +5,207 @@ use strict; use warnings; use Carp 'confess'; -use Scalar::Util 'reftype'; +use Scalar::Util 'reftype', 'blessed'; +use B 'svref_2object'; -our $VERSION = '0.01'; +our $VERSION = '0.02'; + +# introspection sub meta { require Class::MOP::Class; - Class::MOP::Class->initialize($_[0]) + Class::MOP::Class->initialize(blessed($_[0]) || $_[0]); } +# construction + sub wrap { my $class = shift; my $code = shift; - - (reftype($code) && reftype($code) eq 'CODE') - || confess "You must supply a CODE reference to wrap"; - - bless $code => $class; + ('CODE' eq (reftype($code) || '')) + || confess "You must supply a CODE reference to bless"; + bless $code => blessed($class) || $class; +} + +# informational + +sub package_name { + my $code = shift; + (blessed($code)) + || confess "Can only ask the package name of a blessed CODE"; + svref_2object($code)->GV->STASH->NAME; +} + +sub name { + my $code = shift; + (blessed($code)) + || confess "Can only ask the package name of a blessed CODE"; + svref_2object($code)->GV->NAME; +} + +sub fully_qualified_name { + my $code = shift; + (blessed($code)) + || confess "Can only ask the package name of a blessed CODE"; + $code->package_name . '::' . $code->name; +} + +package Class::MOP::Method::Wrapped; + +use strict; +use warnings; + +use Carp 'confess'; +use Scalar::Util 'reftype', 'blessed'; +use Sub::Name 'subname'; + +our $VERSION = '0.01'; + +our @ISA = ('Class::MOP::Method'); + +# NOTE: +# this ugly beast is the result of trying +# to micro optimize this as much as possible +# while not completely loosing maintainability. +# At this point it's "fast enough", after all +# you can't get something for nothing :) +my $_build_wrapped_method = sub { + my $modifier_table = shift; + my ($before, $after, $around) = ( + $modifier_table->{before}, + $modifier_table->{after}, + $modifier_table->{around}, + ); + if (@$before && @$after) { + $modifier_table->{cache} = sub { + $_->(@_) for @{$before}; + my @rval; + ((defined wantarray) ? + ((wantarray) ? + (@rval = $around->{cache}->(@_)) + : + ($rval[0] = $around->{cache}->(@_))) + : + $around->{cache}->(@_)); + $_->(@_) for @{$after}; + return unless defined wantarray; + return wantarray ? @rval : $rval[0]; + } + } + elsif (@$before && !@$after) { + $modifier_table->{cache} = sub { + $_->(@_) for @{$before}; + return $around->{cache}->(@_); + } + } + elsif (@$after && !@$before) { + $modifier_table->{cache} = sub { + my @rval; + ((defined wantarray) ? + ((wantarray) ? + (@rval = $around->{cache}->(@_)) + : + ($rval[0] = $around->{cache}->(@_))) + : + $around->{cache}->(@_)); + $_->(@_) for @{$after}; + return unless defined wantarray; + return wantarray ? @rval : $rval[0]; + } + } + else { + $modifier_table->{cache} = $around->{cache}; + } +}; + +my %MODIFIERS; + +sub wrap { + my $class = shift; + my $code = shift; + (blessed($code) && $code->isa('Class::MOP::Method')) + || confess "Can only wrap blessed CODE"; + my $modifier_table = { + cache => undef, + orig => $code, + before => [], + after => [], + around => { + cache => $code, + methods => [], + }, + }; + $_build_wrapped_method->($modifier_table); + my $method = $class->SUPER::wrap(sub { $modifier_table->{cache}->(@_) }); + $MODIFIERS{$method} = $modifier_table; + $method; +} + +sub get_original_method { + my $code = shift; + $MODIFIERS{$code}->{orig} + if exists $MODIFIERS{$code}; +} + +sub add_before_modifier { + my $code = shift; + my $modifier = shift; + (exists $MODIFIERS{$code}) + || confess "You must first wrap your method before adding a modifier"; + (blessed($code)) + || confess "Can only ask the package name of a blessed CODE"; + ('CODE' eq (reftype($code) || '')) + || confess "You must supply a CODE reference for a modifier"; + unshift @{$MODIFIERS{$code}->{before}} => $modifier; + $_build_wrapped_method->($MODIFIERS{$code}); } - + +sub add_after_modifier { + my $code = shift; + my $modifier = shift; + (exists $MODIFIERS{$code}) + || confess "You must first wrap your method before adding a modifier"; + (blessed($code)) + || confess "Can only ask the package name of a blessed CODE"; + ('CODE' eq (reftype($code) || '')) + || confess "You must supply a CODE reference for a modifier"; + push @{$MODIFIERS{$code}->{after}} => $modifier; + $_build_wrapped_method->($MODIFIERS{$code}); +} + +{ + # NOTE: + # this is another possible canidate for + # optimization as well. There is an overhead + # associated with the currying that, if + # eliminated might make around modifiers + # more manageable. + my $compile_around_method = sub {{ + my $f1 = pop; + return $f1 unless @_; + my $f2 = pop; + push @_, sub { $f2->( $f1, @_ ) }; + redo; + }}; + + sub add_around_modifier { + my $code = shift; + my $modifier = shift; + (exists $MODIFIERS{$code}) + || confess "You must first wrap your method before adding a modifier"; + (blessed($code)) + || confess "Can only ask the package name of a blessed CODE"; + ('CODE' eq (reftype($code) || '')) + || confess "You must supply a CODE reference for a modifier"; + unshift @{$MODIFIERS{$code}->{around}->{methods}} => $modifier; + $MODIFIERS{$code}->{around}->{cache} = $compile_around_method->( + @{$MODIFIERS{$code}->{around}->{methods}}, + $MODIFIERS{$code}->{orig} + ); + $_build_wrapped_method->($MODIFIERS{$code}); + } +} + 1; __END__ @@ -50,16 +232,62 @@ Suggestions for this are welcome. =head1 METHODS +=head2 Introspection + +=over 4 + +=item B + +This will return a B instance which is related +to this class. + +=back + +=head2 Construction + =over 4 =item B This simply blesses the C<&code> reference passed to it. -=item B +=back -This will return a B instance which is related -to this class. +=head2 Informational + +=over 4 + +=item B + +=item B + +=item B + +=back + +=head1 Class::MOP::Method::Wrapped METHODS + +=head2 Construction + +=over 4 + +=item B + +This simply blesses the C<&code> reference passed to it. + +=item B + +=back + +=head2 Modifiers + +=over 4 + +=item B + +=item B + +=item B =back