File Coverage

blib/lib/DWIM/Block.pm

Criterion	Covered	Total	%
statement	14	19	73.6
branch	0	2	0.0
condition	0	10	0.0
subroutine	5	6	83.3
pod			n/a
total	19	37	51.3

package DWIM::Block;

use 5.022;
use warnings;
use Filter::Simple;
use PPR::X;
use AI::Chat;

# Look for DWIM {...} blocks in any source that loads this module...
FILTER {
    my %dwim;

    # This is what code with embedded DWIM blocks looks like...
    my $PERL_WITH_DWIM_BLOCKS = qr{
        (?&PerlEntireDocument)

        (?(DEFINE)
            (?
                DWIM \b  (?{ pos() - 4 })  (?>(?&PerlOWS))
                (?  \{ (?>(?&PPR_X_balanced_curlies_interpolated)) \}  )
                (?{ $dwim{ $^R } = { block => 1, len => pos() - $^R, request => $+{REQUEST} } })
            |
                (?>(?&PerlStdControlBlock))
            )

            (?
                DWIM \b  (?{ pos() - 4 }) (?>(?&PerlOWS))
                (?  \{ (?>(?&PPR_X_balanced_curlies_interpolated)) \}  )
                (?{ $dwim{ $^R } = { len => pos() - $^R, request => $+{REQUEST} } })
            |
                (?>(?&PerlStdCall))
            )
        )

        $PPR::X::GRAMMAR
    }xms;

    # If we find DWIM blocks...
    if (/$PERL_WITH_DWIM_BLOCKS/) {
        # Work backwards from the end of the source (so the replacements don't mess up pos info)...
        for my $pos (sort {$b <=> $a} keys %dwim) {

            # Generate the replacement code...
            my $code = "do { DWIM::Block::_DWIM( qq$dwim{$pos}{request}) }";

            # Substitute the replacement code into the source...
            substr($_, $pos, $dwim{$pos}{len}) = $dwim{$pos}{block} ? "{$code}" : $code;
        }
    }
}

our $VERSION = '0.000002';

# This sub relays each request to the API and returns the result...
sub _DWIM {
    my ($request) = @_;

    # Discover and remember the API key...
    state $API_KEY = $ENV{'DWIM_API_KEY'}
                  // $ENV{'OPENAI_API_KEY'}
                  // die sprintf qq{Can't find API key at %s line %d\n}, (caller 0)[1,2];

    # Discover and remember the requested model...
    state $MODEL   = $ENV{'DWIM_MODEL'}
                  // $ENV{'OPENAI_MODEL'}
                  // q{};

    # Build and retain an API object...
    state $GPT = AI::Chat->new( key => $API_KEY, $MODEL ? ( model => $MODEL ) : () );

    # Send the request and return the response...
    return $GPT->prompt($request);
}


1; # Magic true value required at end of module
__END__