X-Git-Url: http://git.vpit.fr/?a=blobdiff_plain;f=lib%2FSub%2FPrototype%2FUtil.pm;h=68ac767034d309c575945a2fb85a893e8a91d53a;hb=8c3d123a078fe87b9b1b20fce287044d3f7f62db;hp=73a1a60115f6d384ace73083584543d99bf60cc0;hpb=29d3a214c70390edd665985eddfb42ff7e29a02a;p=perl%2Fmodules%2FSub-Prototype-Util.git diff --git a/lib/Sub/Prototype/Util.pm b/lib/Sub/Prototype/Util.pm index 73a1a60..68ac767 100644 --- a/lib/Sub/Prototype/Util.pm +++ b/lib/Sub/Prototype/Util.pm @@ -12,23 +12,25 @@ Sub::Prototype::Util - Prototype-related utility routines. =head1 VERSION -Version 0.05 +Version 0.08 =cut use vars qw/$VERSION/; -$VERSION = '0.05'; +$VERSION = '0.08'; =head1 SYNOPSIS - use Sub::Prototype::Util qw/flatten recall/; + use Sub::Prototype::Util qw/flatten recall wrap/; my @a = qw/a b c/; my @args = ( \@a, 1, { d => 2 }, undef, 3 ); my @flat = flatten '\@$;$', @args; # ('a', 'b', 'c', 1, { d => 2 }) recall 'CORE::push', @args; # @a contains 'a', 'b', 'c', 1, { d => 2 }, undef, 3 + my $splice = wrap 'CORE::splice', compile => 1; + my @b = $splice->(\@a, 4, 2); # @a is now ('a', 'b', 'c', 1, 3) and @b is ({ d => 2 }, undef) =head1 DESCRIPTION @@ -41,6 +43,7 @@ They all handle C<5.10>'s C<_> prototype. =cut my %sigils = qw/SCALAR $ ARRAY @ HASH % GLOB * CODE &/; +my %reftypes = reverse %sigils; sub _check_ref { my ($a, $p) = @_; @@ -108,20 +111,26 @@ This allows you to recall into C and C by using the C<\&@ =cut -sub recall { - my $name = shift; +sub _check_name { + my $name = $_[0]; croak 'No subroutine specified' unless $name; my $proto; my $r = ref $name; - if ($r eq 'HASH') { + if (!$r) { + $proto = prototype $name; + } elsif ($r eq 'HASH') { croak 'Forced prototype hash reference must contain exactly one key/value pair' unless keys %$name == 1; ($name, $proto) = %$name; - } elsif (length $r) { + } else { croak 'Unhandled ' . $r . ' reference as first argument'; } $name =~ s/^\s+//; $name =~ s/[\s\$\@\%\*\&;].*//; - $proto = prototype $name unless $proto; + return $name, $proto; +} + +sub recall { + my ($name, $proto) = _check_name shift; my $call = $name . '('; my @cr; if (defined $proto) { @@ -154,9 +163,115 @@ sub recall { return @ret; } +=head2 C + +Generates a wrapper that does the same thing as L, but specialized for a given function. This wrapper can be compiled once for all to avoid calling C at each run (like L does). You can still force the prototype by passing C<< { $name => $proto } >> as the first argument. Others arguments are seen as key / value pairs and tune the code generated by L. Valid keys are : + +=over 4 + +=item C<< ref => $func >> + +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. + +=item C<< wrong_ref => $code >> + +The code executed when a reference of incorrect type is encountered. The result of this snippet is also the result of the generated code, hence it defaults to C<'undef'>. It's a good place to C or C too. + +=item C<< sub => $bool >> + +Encloses the code into a C block. Default is true. + +=item C<< compile => $bool >> + +Makes L compile the code generated and return the resulting code reference. Implies C<< sub => 1 >>. Be careful that in this case C must be a fully qualified function name. Defaults to false. + +=back + +This is how you make your own C that pushes into array references : + + my @a = (0 .. 2); + my $push = wrap 'CORE::push', compile => 1; + $push->(\@a, 3 .. 7); # returns 3 + 5 = 8, and @a now contains 0 .. 7 + +=cut + +sub _wrap { + my ($name, $proto, $i, $args, $cr, $opts) = @_; + if ($proto =~ /(\\?)(\[[^\]]+\]|[^\];])(.*)/g) { + my ($ref, $p) = ($1, $2); + $proto = $3; + $p = $1 if $p =~ /^\[([^\]]+)\]/; + 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} . ' }' + } else { + $args .= $p . '{' . $cur . '}, '; + } + } elsif ($p =~ /[\@\%]/) { + $args .= '@_[' . $i . '..$#_]'; + } elsif ($p =~ /\&/) { + my %h = do { my $c; map { $_ => $c++ } @$cr }; + my $j; + if (not exists $h{$i}) { + push @$cr, $i; + $j = $#{$cr}; + } else { + $j = int $h{$i}; + } + $args .= 'sub{&{$c[' . $j . ']}}, '; + } elsif ($p eq '_') { + $args .= '((@_ > ' . $i . ') ? ' . $cur . ' : $_), '; + } else { + $args .= $cur . ', '; + } + return _wrap($name, $proto, ($i + 1), $args, $cr, $opts); + } else { + $args =~ s/,\s*$//; + return $name . '(' . $args . ')'; + } +} + +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 exists $opts{sub} or $opts{compile}; + $opts{wrong_ref} = 'undef' if not defined $opts{wrong_ref}; + my @cr; + my $call; + if (defined $proto) { + $call = _wrap $name, $proto, 0, '', \@cr, \%opts; + } else { + $call = _wrap $name, '', 0, '@_'; + } + if (@cr) { + $call = 'my @c; ' + . join('', map { 'push @c, $_[' . $_ . ']; ' } @cr) + . $call + } + $call = '{ ' . $call . ' }'; + $call = 'sub ' . $call if $opts{sub}; + if ($opts{compile}) { + $call = eval $call; + croak $@ if $@; + } + return $call; +} + =head1 EXPORT -The functions L and L are only exported on request, either by providing their name or by the C<':funcs'> and C<':all'> tags. +The functions L, L and L are only exported on request, either by providing their name or by the C<':funcs'> and C<':all'> tags. =cut @@ -166,7 +281,7 @@ use vars qw/@EXPORT @EXPORT_OK %EXPORT_TAGS/; @EXPORT = (); %EXPORT_TAGS = ( - 'funcs' => [ qw/flatten recall/ ] + 'funcs' => [ qw/flatten recall wrap/ ] ); @EXPORT_OK = map { @$_ } values %EXPORT_TAGS; $EXPORT_TAGS{'all'} = [ @EXPORT_OK ]; @@ -179,7 +294,7 @@ L, L (core modules since perl 5), L (since 5.7.3). Vincent Pit, C<< >>, L. -You can contact me by mail or on #perl @ FreeNode (vincent or Prof_Vince). +You can contact me by mail or on C (vincent). =head1 BUGS