File Coverage

blib/lib/Plack/Middleware/Proxy/ByHeader.pm

Criterion	Covered	Total	%
statement	45	45	100.0
branch	18	18	100.0
condition			n/a
subroutine	8	8	100.0
pod	2	2	100.0
total	73	73	100.0

package Plack::Middleware::Proxy::ByHeader;
use strict;
use warnings;

use parent 'Plack::Middleware';

use Plack::Util::Accessor qw(header allowed);
use Plack::Request;
use Carp qw(croak);

our $VERSION = '0.01';

sub prepare_app {
    my $self = shift;

    $self->header( ['Host'] ) if !defined $self->header;
    $self->allowed( [] ) if !defined $self->allowed;

    croak("argument to 'allowed' must be an array reference")
      if not( ref( $self->allowed ) eq 'ARRAY' );

    croak("argument to 'header' must be an array reference")
      if not( ref( $self->header ) eq 'ARRAY' );

    return;
}

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

    my $req  = Plack::Request->new($env);
    my $uri  = $req->uri;

    my $host;
    for my $header ( @{$self->header} ) {
	$host = $req->header($header);
	last if $host;
    }

    if ($host) {
	my $host = (split(/\s*,\s*/,$host))[-1];

        my $allow = 1;
        if ( @{ $self->allowed } ) {
            my %allowed = map { $_ => 1 } @{ $self->allowed };
            $allow = 0 if not $allowed{$host};
        }

        if ($allow) {
            $uri->host($host);
            $env->{'plack.proxy.url'} = $uri;
        }
    }
    return $self->app->($env);
}

1;

__END__