File Coverage

lib/PAGI/App/Cascade.pm

Criterion	Covered	Total	%
statement	49	49	100.0
branch	6	6	100.0
condition	5	7	71.4
subroutine	8	8	100.0
pod	2	3	66.6
total	70	73	95.8

package PAGI::App::Cascade;
$PAGI::App::Cascade::VERSION = '0.002001';
use strict;
use warnings;
use Future::AsyncAwait;
use PAGI::Utils ();

=head1 NAME

PAGI::App::Cascade - Try apps in sequence until success

=head1 SYNOPSIS

    use PAGI::App::Cascade;

    my $app = PAGI::App::Cascade->new(
        apps => [$static_app, PAGI::App::NotFound->new(body => 'nope')],
        catch => [404, 405],
    )->to_app;

=cut

sub new {
    my ($class, %args) = @_;

    return bless {
        apps  => [map { PAGI::Utils::to_app($_) } @{$args{apps} // []}],
        catch => { map { $_ => 1 } @{$args{catch} // [404, 405]} },
    }, $class;
}

sub add {
    my ($self, $app) = @_;

    push @{$self->{apps}}, PAGI::Utils::to_app($app);
    return $self;
}

sub to_app {
    my ($self) = @_;

    my @apps = @{$self->{apps}};
    my %catch = %{$self->{catch}};

    return async sub  {
        my ($scope, $receive, $send) = @_;
        for my $i (0 .. $#apps) {
            my $app = $apps[$i];
            my $is_last = ($i == $#apps);

            # For non-last apps, we need to capture the response
            if (!$is_last) {
                my @captured_events;
                my $captured_status;

                my $capture_send = async sub  {
        my ($event) = @_;
                    push @captured_events, $event;
                    if ($event->{type} eq 'http.response.start') {
                        $captured_status = $event->{status};
                    }
                };

                await $app->($scope, $receive, $capture_send);

                # Check if we should try next app
                if ($captured_status && $catch{$captured_status}) {
                    next;  # Try next app
                }

                # Send captured events
                for my $event (@captured_events) {
                    await $send->($event);
                }
                return;
            }

            # Last app - send directly
            await $app->($scope, $receive, $send);
            return;
        }
    };
}

1;

__END__