File Coverage

blib/lib/Compile/Generators.pm

Criterion	Covered	Total	%
statement	10	12	83.3
branch			n/a
condition			n/a
subroutine	4	4	100.0
pod			n/a
total	14	16	87.5

package Compile::Generators;
use strict;
use 5.006001;
use warnings;
our $VERSION = '0.11';

use Module::Compile -base;

my $label = 'GENAAAA';

sub pmc_compile {
    my ($class, $input) = @_;
    my $output;
    while ($output = $class->compile($input)) {
        last if $output eq $input;
        $input = $output;
    }
    return $output;
}

sub compile {
    my ($class, $source) = @_;
    return $source unless $source =~
      /^(sub .*):generator (.*\n)((?s:.*?))(^\}.*\n)/m;
    my $start = "$1$2";
    my $end = $4;
    my ($static, $dynamic) = split /^\s*\n/m, $3, 2;
    if (not defined $dynamic) {
        $dynamic = $static;
        $static = '';
    }
    while ($dynamic =~ s/^(\s*yield\s*(.*))/#$1\n        \$__GEN_STATE__ = '$label'; return $2; ${label}:/m) {
        $label++;
    }
#     $dynamic =~ s/^/    /mg;

    my $code = <<"_";
${start}    my \$__GEN_STATE__ = undef;
${static}
return sub {
    goto \$__GEN_STATE__ if defined \$__GEN_STATE__;
${dynamic}
    \$__GEN_STATE__ = '$label';
    ${label}:
    return;
}
${end}
_
    $label++;
    $source =~ s/^sub .*:generator .*\n(?s:.*?)^\}.*\n/$code/m;
    return $source;
}

1;

__DATA__