package Sub::Prototype::Util;
+use 5.006;
+
use strict;
use warnings;
=head1 VERSION
-Version 0.08
+Version 0.09
=cut
use vars qw/$VERSION/;
-$VERSION = '0.08';
+$VERSION = '0.09';
=head1 SYNOPSIS
Flattens the array C<@args> according to the prototype C<$proto>.
When C<@args> is what C<@_> is after calling a subroutine with prototype C<$proto>, C<flatten> returns the list of what C<@_> would have been if there were no prototype.
+It croaks if the arguments can't possibly match the required prototype, e.g. when a reference type is wrong or when not enough elements were provided.
=cut
sub flatten {
my $proto = shift;
return @_ unless defined $proto;
- my @args;
+ my @args;
while ($proto =~ /(\\?)(\[[^\]]+\]|[^\];])/g) {
my $p = $2;
if ($1) {
Specifies the function used in the generated code to test the reference type of scalars.
Defaults to C<'ref'>.
-You may also want to use C<Scalar::Util::reftype>.
+You may also want to use L<Scalar::Util/reftype>.
=item C<< wrong_ref => $code >>
=cut
sub _wrap {
- my ($name, $proto, $i, $args, $cr, $opts) = @_;
+ my ($name, $proto, $i, $args, $coderefs, $opts) = @_;
+
while ($proto =~ s/(\\?)(\[[^\]]+\]|[^\];])//) {
- my ($ref, $p) = ($1, $2);
- $p = $1 if $p =~ /^\[([^\]]+)\]/;
- my $cur = '$_[' . $i . ']';
+ my ($ref, $sigil) = ($1, $2);
+ $sigil = $1 if $sigil =~ /^\[([^\]]+)\]/;
+
+ my $cur = "\$_[$i]";
+
if ($ref) {
- if (length $p > 1) {
- return 'my $r = ' . $opts->{ref} . '(' . $cur . '); '
- . join ' els',
- map( {
- "if (\$r eq '" . $reftypes{$_} ."') { "
- . _wrap($name, $proto, ($i + 1),
- $args . $_ . '{' . $cur . '}, ',
- $cr, $opts)
- . ' }'
- } split //, $p),
- 'e { ' . $opts->{wrong_ref} . ' }'
+ if (length $sigil > 1) {
+ my $code = "my \$r = $opts->{ref}($cur); ";
+ my @branches = map {
+ my $subcall = _wrap(
+ $name, $proto, ($i + 1), $args . "$_\{$cur}, ", $coderefs, $opts
+ );
+ "if (\$r eq '$reftypes{$_}') { $subcall }";
+ } split //, $sigil;
+ $code .= join ' els', @branches, "e { $opts->{wrong_ref} }";
+ return $code;
} else {
- $args .= $p . '{' . $cur . '}, ';
+ $args .= "$sigil\{$cur}, ";
}
- } elsif ($p =~ /[\@\%]/) {
+ } elsif ($sigil =~ /[\@\%]/) {
$args .= '@_[' . $i . '..$#_]';
- } elsif ($p =~ /\&/) {
- my %h = do { my $c; map { $_ => $c++ } @$cr };
+ } elsif ($sigil =~ /\&/) {
+ my %h = do { my $c; map { $_ => $c++ } @$coderefs };
my $j;
- if (not exists $h{$i}) {
- push @$cr, $i;
- $j = $#{$cr};
- } else {
+ if (exists $h{$i}) {
$j = int $h{$i};
+ } else {
+ push @$coderefs, $i;
+ $j = $#{$coderefs};
}
- $args .= 'sub{&{$c[' . $j . ']}}, ';
- } elsif ($p eq '_') {
- $args .= '((@_ > ' . $i . ') ? ' . $cur . ' : $_), ';
+ $args .= "sub{&{\$c[$j]}}, ";
+ } elsif ($sigil eq '_') {
+ $args .= "((\@_ > $i) ? $cur : \$_), ";
} else {
- $args .= $cur . ', ';
+ $args .= "$cur, ";
}
+ } continue {
++$i;
}
+
$args =~ s/,\s*$//;
- return $name . '(' . $args . ')';
+
+ return "$name($args)";
}
sub _check_name {
- my $name = $_[0];
+ my ($name) = @_;
croak 'No subroutine specified' unless $name;
+
my $proto;
my $r = ref $name;
if (!$r) {
} else {
croak 'Unhandled ' . $r . ' reference as first argument';
}
+
$name =~ s/^\s+//;
$name =~ s/[\s\$\@\%\*\&;].*//;
+
return $name, $proto;
}
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;
}
=cut
sub recall {
- my $wrap = eval { wrap shift };
- croak _clean_msg $@ if $@;
- return $wrap->(@_);
+ my $name = shift;
+
+ my ($wrap, $err);
+ {
+ local $@;
+ $wrap = eval { wrap $name };
+ $err = $@;
+ }
+ croak _clean_msg $err if $err;
+
+ goto $wrap;
}
=head1 EXPORT
=head1 COPYRIGHT & LICENSE
-Copyright 2008 Vincent Pit, all rights reserved.
+Copyright 2008,2009,2010,2011 Vincent Pit, all rights reserved.
This program is free software; you can redistribute it and/or modify it
under the same terms as Perl itself.