File Coverage

blib/lib/Plack/Middleware/IPAddressFilter.pm

Criterion	Covered	Total	%
statement	32	34	94.1
branch	5	8	62.5
condition	1	3	33.3
subroutine	8	8	100.0
pod	2	3	66.6
total	48	56	85.7

package Plack::Middleware::IPAddressFilter;
use parent qw(Plack::Middleware);

use strict;
use warnings;
our $VERSION = '0.02';

use Net::IP::AddrRanges;

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

sub prepare_app {
    my $self = shift;

    my $rules = $self->rules;
    if(ref $rules ne 'ARRAY') {
        die 'rules should be an array reference';
    }
    my $ranges = Net::IP::AddrRanges->new();
    for(@$rules) {
        s/^([\+\-])? *//;
        if(not defined $1 or $1 eq '+') {
            $ranges->add($_);
        }
        else {
            $ranges->subtract($_);
        }
    }
    $self->rules($ranges);
}

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

    my $remote_address = $env->{REMOTE_ADDR}
        or return $self->forbidden;

    if($self->rules->find($remote_address)) {
        return $self->app->($env);
    }
    return $self->forbidden;
}

sub forbidden {
    my $self = shift;
    my $body = "You don't have permission to access this resouce.";
    return [
        403,
        [ 'Content-Type' => 'text/plain',
          'Content-Length' => length $body ],
        [ $body ],
    ];
}

1;
__END__