File Coverage

lib/PAGI/App/NotFound.pm

Criterion	Covered	Total	%
statement	21	21	100.0
branch	1	2	50.0
condition	4	6	66.6
subroutine	6	6	100.0
pod	0	2	0.0
total	32	37	86.4

package PAGI::App::NotFound;

use strict;
use warnings;
use Future::AsyncAwait;

=head1 NAME

PAGI::App::NotFound - Customizable 404 response

=head1 SYNOPSIS

    use PAGI::App::NotFound;

    my $app = PAGI::App::NotFound->new(
        body => 'Page not found',
        content_type => 'text/html',
    )->to_app;

=cut

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

    return bless {
        body => $args{body} // 'Not Found',
        content_type => $args{content_type} // 'text/plain',
        status => $args{status} // 404,
    }, $class;
}

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

    my $body = $self->{body};
    my $content_type = $self->{content_type};
    my $status = $self->{status};

    return async sub  {
        my ($scope, $receive, $send) = @_;
        my $response_body = ref $body eq 'CODE' ? $body->($scope) : $body;

        await $send->({
            type => 'http.response.start',
            status => $status,
            headers => [
                ['content-type', $content_type],
                ['content-length', length($response_body)],
            ],
        });
        await $send->({ type => 'http.response.body', body => $response_body, more => 0 });
    };
}

1;

__END__