File Coverage

blib/lib/Plack/Middleware/Reproxy.pm

Criterion	Covered	Total	%
statement	17	25	68.0
branch	5	8	62.5
condition	0	3	0.0
subroutine	4	5	80.0
pod	1	2	50.0
total	27	43	62.7

package Plack::Middleware::Reproxy;
use strict;
use parent qw(Plack::Middleware);
our $VERSION = '0.00004';

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

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

    # Otherwise, wait for headers
    return $self->response_cb( $res, sub {
        my $res = shift;

        if ($res->[0] ne '200') {
            # Not success, then just pass through
            return;
        }

        # Now check header
        my $reproxy_url = Plack::Util::header_get( $res->[1], 'X-Reproxy-URL' );
        if ($reproxy_url) {
            # it's a reproxy!
            $reproxy_url = URI->new($reproxy_url);
            if (! $reproxy_url->scheme) {
                $reproxy_url->scheme( $ENV{'psgi.url_scheme'} );
            }
            if (! $reproxy_url->host) {
                $reproxy_url->host( $ENV{HTTP_HOST} || $ENV{SERVER_NAME} );
            }
            return $self->reproxy_to( $res, $env, $reproxy_url );
        }
    } );
}

sub extract_headers {
    my ($self, $env) = @_;
    return map {
        (my $field = $_) =~ s/^HTTPS?_//;
        ( $field => $env->{$_} );
    } grep { /^(?:HTTP|CONTENT(?!_TYPE$)|COOKIE)/i } keys %$env;
}

1;

__END__