File Coverage

blib/lib/Plack/Middleware/DevFilter.pm

Criterion	Covered	Total	%
statement	18	48	37.5
branch	0	16	0.0
condition	0	3	0.0
subroutine	6	12	50.0
pod	2	2	100.0
total	26	81	32.1

package Plack::Middleware::DevFilter;
use strict;
use warnings;
use Carp qw/croak/;
use parent 'Plack::Middleware';
use Plack::Util::Accessor qw/force_enable filters image_type/;
use Plack::Util;

our $VERSION = '0.01';

sub prepare_app {
    my $self = shift;

    croak "'filters' option must be ARRAY." unless ref $self->filters eq 'ARRAY';

    unless ($self->image_type) {
        $self->image_type(
            sub {
                my ($env, $res) = @_;

                my $ct = Plack::Util::header_get($res->[1], 'content-type');
                my $image_type = ($ct =~ m!(png|gif|jpeg|ico)!) ? $1 : undef;
                return $image_type;
            },
        );
    }
}

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

    my $res = $self->app->($env);

    return $res if $self->_no_filter;

    my $body = '';
    Plack::Util::foreach($res->[2], sub {
        my($buf) = @_;
        $body .= $buf;
    });

    my ($imager, $image_type);
    if ( $image_type = $self->image_type->($env, $res) ) {
        $imager = $self->_imager_obj(\$body, $image_type);
    }

    my $filtered = 0;
    for my $filter (@{$self->filters}) {
        if ( $filter->{match}->($self, $env) ) {
            $filter->{proc}->($self, $env, $res,
                                    \$body, $imager, $image_type);
            $filtered = 1;
        }
    }

    return $filtered ? $res : [ $res->[0], $res->[1], [$body] ];
}

sub _no_filter {
    my $self = shift;

    return !$self->force_enable
                && ( ! $ENV{PLACK_ENV}
                            || $ENV{PLACK_ENV} !~ m!^(?:development|test)$! )
}

sub _imager_obj {
    my ($self, $body_ref, $image_type) = @_;

    require Imager;
    return Imager->new(
        data => $$body_ref,
        type => $image_type,
    ) or croak Imager->errstr;
}

1;

__END__