File Coverage

blib/lib/Geo/IPinfo.pm

Criterion	Covered	Total	%
statement	14	59	23.7
branch	0	14	0.0
condition			n/a
subroutine	5	11	45.4
pod	5	5	100.0
total	24	89	26.9

package Geo::IPinfo;

use 5.006;
use strict;
use warnings;

use LWP::UserAgent;
use JSON;

our $VERSION = '1.0';

my %valid_fields = (
                    ip => 1,
                    hostname => 1,
                    city => 1,
                    region => 1,
                    country => 1,
                    loc => 1,
                    org => 1,
                    postal => 1,
                    phone => 1
                );
my $base_url = 'https://ipinfo.io/';

#-------------------------------------------------------------------------------

sub new
{
  my ($pkg, $token) = @_;

  my $self = {};
  $self->{token} = defined $token ? "?token=$token" : "";

  $self->{base_url} = $base_url;
  $self->{ua} = LWP::UserAgent->new;
  $self->{ua}->agent("curl/Geo::IP $VERSION");
  $self->{message} = "";

  bless($self, $pkg);
  return $self;
}

#-------------------------------------------------------------------------------

sub info
{
  my ($self, $ip) = @_;

  return $self->_getinfo($ip, "");
}

#-------------------------------------------------------------------------------

sub geo
{
  my ($self, $ip) = @_;

  return $self->_getinfo($ip, "geo");
}

#-------------------------------------------------------------------------------

sub field
{
  my ($self, $ip, $field) = @_;

  if (not defined $ip)
  {
    $self->{message} = "IP is undefined";
    return undef;
  }

  if (not defined $field)
  {
    $self->{message} = "field() requires 2 arguments";
    return undef;
  }

  if (not defined $valid_fields{$field})
  {
    $self->{message} = "Invalid field: $field";
    return undef;
  }

  my $url = $self->{base_url} . $ip . "/" . $field . $self->{token};

  my $res = $self->{ua}->get($url);
  if ($res->is_success)
  {
    $self->{message} = "";
    my $value = $res->decoded_content;
    chomp $value;
    return $value;
  }
  else
  {
    $self->{message} = $res->status_line;
    return undef;
  }
}

#-------------------------------------------------------------------------------

sub error_msg
{
  my $self = shift;

  return $self->{message};
}

#-------------------------------------------------------------------------------
#-- private method(s) below , don't call them directly -------------------------

sub _getinfo
{
  my ($self, $ip, $type) = @_;

  if (not defined $ip)
  {
    $self->{message} = "IP is undefined";
    return undef;
  }

  my $url = $self->{base_url} . $ip . "/" . $type . $self->{token};

  my $res = $self->{ua}->get($url);
  if ($res->is_success)
  {
    $self->{message} = "";
    return from_json($res->decoded_content);
  }
  else
  {
    $self->{message} = $res->status_line;
    return undef;
  }
}

#-------------------------------------------------------------------------------

1;
__END__