File Coverage

blib/lib/Pickles/Plugin/Encode.pm

Criterion	Covered	Total	%
statement	38	38	100.0
branch	3	4	75.0
condition	2	4	50.0
subroutine	7	7	100.0
pod	0	1	0.0
total	50	54	92.5

package Pickles::Plugin::Encode;
use strict;
use Encode ();

sub _decode {
    my ($hmv, $ie) = @_;
    for my $key( $hmv->keys ) {
        my @values = map { Encode::decode($ie, $_) } $hmv->get_all( $key );
        $hmv->remove( $key );
        $hmv->add( $key => @values );
    }
}

sub _decode_args {
    my($h, $ie) = @_;
    for my $key( keys %{$h} ) {
        if ( ref $h->{$key} eq 'ARRAY' ) {
            my @values = map { Encode::decode($ie, $_) } @{$h->{$key}};
            $h->{$key} = \@values;
        }
        else {
            $h->{$key} = Encode::decode($ie, $h->{$key});
        }
    }
}

sub install {
    my( $class, $pkg ) = @_;
    $pkg->add_trigger( init => sub {
        my $c = shift;
        my $config = $c->config->{'Plugin::Encode'};
        my $ie = $config->{input_encoding} || 'utf-8';
        _decode( $c->req->query_parameters, $ie );
        _decode( $c->req->body_parameters, $ie );
        _decode_args($c->args, $ie);
        delete $c->req->env->{'plack.request.merged'}; # make sure
    });
    $pkg->add_trigger( pre_finalize => sub {
        my( $c ) = @_;
        if ( $c->res->content_type =~ m{^text/} ) {
            my $body = $c->res->body;
            my $config = $c->config->{'Plugin::Encode'};
            my $oe = $config->{output_encoding} || 'utf-8';
            $c->res->content_type( $c->res->content_type. '; charset='. $oe );
            $c->res->body( Encode::encode( $oe, $body ) );
        }
    });
}

1;

__END__