File Coverage

blib/lib/Plack/Middleware/LogFilter.pm

Criterion	Covered	Total	%
statement	22	23	95.6
branch	2	2	100.0
condition	4	8	50.0
subroutine	7	8	87.5
pod	1	1	100.0
total	36	42	85.7

package Plack::Middleware::LogFilter;
use strict;
use warnings;
use parent qw(Plack::Middleware);

use Plack::Util::Accessor qw(filter);

our $VERSION = "0.01";

sub call {
    my ($self, $env) = @_;

    $env->{'psgi.errors'} =
        Plack::Middleware::LogFilter::_wrap->new(
            $env,
            $self->filter,
        );
    my $res = $self->app->($env);
    return $res;
}


package Plack::Middleware::LogFilter::_wrap;

sub new {
    my ($class, $env, $filter) = @_;
    return bless {
        env => $env,
        '_psgi.errors' => $env->{'psgi.errors'},
        filter => $filter || sub { return 1; },
    }, $class;
}

sub print {
    my ($self, $output) = @_;

    my $filter = $self->{filter};
    if (!$filter || (ref $filter eq 'CODE' && $filter->($self->{env}, $output))) {
        return $self->{'_psgi.errors'}->print($output);
    }
}

1;
__END__