File Coverage

blib/lib/WWW/Correios/CEP.pm

Criterion	Covered	Total	%
statement	67	69	97.1
branch	19	28	67.8
condition	2	5	40.0
subroutine	10	11	90.9
pod	2	2	100.0
total	100	115	86.9

package WWW::Correios::CEP;
use strict;
use warnings;

use LWP::UserAgent;
use JSON;

our $VERSION = 1.044;

use Encode;
use utf8;

sub new {
    my ( $class, $params ) = @_;

    my $this = {
        _user_agent => defined $params->{user_agent}
        ? $params->{user_agent}
        : 'Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1)',
        _lwp_ua      => undef,
        _lwp_options => $params->{lwp_options} || { timeout => 30 },

        _post_url => defined $params->{post_url}
        ? $params->{post_url}
        : 'https://buscacepinter.correios.com.br/app/endereco/carrega-cep-endereco.php',

        _post_content => defined $params->{post_content}
        ? $params->{post_content}
        : 'pagina=%2Fapp%2Fendereco%2Findex.php&cepaux=&mensagem_alerta=&tipoCEP=ALL&endereco='
    };

    $this->{_lwp_options}{timeout} = $params->{timeout}
      if defined $params->{timeout};

    return bless $this, $class;
}

sub find {
    my ( $this, $cep, $as_html_tree ) = @_;

    my @list_address = $this->_extractAddress( $cep, $as_html_tree );
    $list_address[0]{address_count} = @list_address unless wantarray;

    return wantarray ? @list_address : $list_address[0];
}

sub _extractAddress {
    my ( $this, $cep, $as_html_tree ) = @_;

    my @result = ();

    $cep =~ s/[^\d]//go;
    $cep = sprintf( '%08d', $cep );

    if ( $cep =~ /^00/o || $cep =~ /(\d)\1{7}/ ) {
        $result[0]->{status} = "Error: Invalid CEP number ($cep)";
    }
    else {
        if ( !defined $this->{_lwp_ua} ) {

            my $ua = LWP::UserAgent->new( %{ $this->{_lwp_options} } );
            $ua->agent( $this->{_user_agent} );
            $ua->timeout( $this->{_lwp_options}{timeout} );
            $this->{_lwp_ua} = $ua;
        }
        my $ua = $this->{_lwp_ua};

        my $url = $this->{_post_url};
        my $req = HTTP::Request->new(
            POST => $url,
            [
                'Content-Type' => 'application/x-www-form-urlencoded',
                'Origin'       => 'https://buscacepinter.correios.com.br',
                'Referer' =>
'https://buscacepinter.correios.com.br/app/endereco/index.php?t',
            ],
            $this->{_post_content} . $cep
        );

        eval {
            local $SIG{ALRM} =
              sub { die "Can't connect to server [alarm timeout]\n" };
            alarm( $this->{_lwp_options}{timeout} + 1 );

            # Pass request to the user agent and get a response back
            my $res = $ua->request($req);

            # Check the outcome of the response

            if ( $res->is_success ) {
                $this->_parseJSON( \@result, $res->content, $as_html_tree );
            }
            else {
                $result[0]->{status} = "Error: " . $res->status_line;
            }
        };
        alarm(0);
        die $@ if ($@);
    }

    return wantarray ? @result : $result[0];
}

sub _parseJSON {
    my ( $this, $address_ref, $json, $as_html_tree ) = @_;

    my $obj = from_json($json);

    for my $p ( @{ $obj->{dados} || [] } ) {

        if ($as_html_tree) {
            push( @$address_ref, $p );
        }
        else {
            my $address = {};

            $address->{street}       = $p->{logradouroDNEC};
            $address->{neighborhood} = $p->{bairro};
            $address->{cep} =
              substr( $p->{cep}, 0, 5 ) . '-' . substr( $p->{cep}, 5, 3 );

            $address->{location} = $p->{localidade};
            $address->{uf}       = $p->{uf};

            $address->{status} = $p->{situacao};

            $address->{raw} = $p;

            push( @$address_ref, $address );
        }
    }

    $address_ref->[0]->{status} = 'Error: Address not found'
      if ( !@$address_ref );

    return 1;
}

1;
__END__