X-Git-Url: http://git.vpit.fr/?a=blobdiff_plain;f=lib%2FSub%2FPrototype%2FUtil.pm;h=c06a913290c7fcac0f4e4d576195b017d3940299;hb=dc8381d3afc8122719bdacce75ee5c2d25c27d1d;hp=d2564d85e33dd66c758302d3798343d97b999367;hpb=c7c6c66b6fb5da113036318ca7e196ac422b32b0;p=perl%2Fmodules%2FSub-Prototype-Util.git diff --git a/lib/Sub/Prototype/Util.pm b/lib/Sub/Prototype/Util.pm index d2564d8..c06a913 100644 --- a/lib/Sub/Prototype/Util.pm +++ b/lib/Sub/Prototype/Util.pm @@ -207,8 +207,9 @@ sub _wrap { } sub _check_name { - my $name = $_[0]; + my ($name) = @_; croak 'No subroutine specified' unless $name; + my $proto; my $r = ref $name; if (!$r) { @@ -219,8 +220,10 @@ sub _check_name { } else { croak 'Unhandled ' . $r . ' reference as first argument'; } + $name =~ s/^\s+//; $name =~ s/[\s\$\@\%\*\&;].*//; + return $name, $proto; } @@ -228,28 +231,39 @@ sub wrap { my ($name, $proto) = _check_name shift; croak 'Optional arguments must be passed as key => value pairs' if @_ % 2; my %opts = @_; + $opts{ref} ||= 'ref'; - $opts{sub} = 1 if not defined $opts{sub}; - $opts{compile} = 1 if not defined $opts{compile} and $opts{sub}; - $opts{wrong_ref} = 'undef' if not defined $opts{wrong_ref}; - my @cr; + $opts{sub} = 1 unless defined $opts{sub}; + $opts{compile} = 1 if not defined $opts{compile} and $opts{sub}; + $opts{wrong_ref} = 'undef' unless defined $opts{wrong_ref}; + + my @coderefs; my $call; if (defined $proto) { - $call = _wrap $name, $proto, 0, '', \@cr, \%opts; + $call = _wrap $name, $proto, 0, '', \@coderefs, \%opts; } else { $call = _wrap $name, '', 0, '@_'; } - if (@cr) { - $call = 'my @c; ' - . join('', map { 'push @c, $_[' . $_ . ']; ' } @cr) - . $call + + if (@coderefs) { + my $decls = @coderefs > 1 ? 'my @c = @_[' . join(', ', @coderefs) . ']; ' + : 'my @c = ($_[' . $coderefs[0] . ']); '; + $call = $decls . $call; } - $call = '{ ' . $call . ' }'; - $call = 'sub ' . $call if $opts{sub}; + + $call = "{ $call }"; + $call = "sub $call" if $opts{sub}; + if ($opts{compile}) { - $call = eval $call; - croak _clean_msg $@ if $@; + my $err; + { + local $@; + $call = eval $call; + $err = $@; + } + croak _clean_msg $err if $err; } + return $call; }