File Coverage

blib/lib/Net/Peep/Host/Pool.pm

Criterion	Covered	Total	%
statement	15	71	21.1
branch	0	16	0.0
condition	0	24	0.0
subroutine	5	13	38.4
pod	0	8	0.0
total	20	132	15.1

package Net::Peep::Host::Pool;

require 5.00503;
use strict;
# use warnings; # commented out for 5.005 compatibility
use Carp;
use Socket;
require Exporter;
use Net::Peep::Log;

use vars qw{ @ISA %EXPORT_TAGS @EXPORT_OK @EXPORT $VERSION $AUTOLOAD @Attributes $LOGGER };

@ISA = qw(Exporter);
%EXPORT_TAGS = ( 'all' => [ qw( ) ] );
@EXPORT_OK = ( @{ $EXPORT_TAGS{'all'} } );
@EXPORT = qw( );
$VERSION = do { my @r = (q$Revision: 1.1 $ =~ /\d+/g); sprintf "%d."."%02d" x $#r, @r };

$LOGGER = new Net::Peep::Log;

sub new {

    my $self = shift;
    my $class = ref($self) || $self;
    my $this = {};
    bless $this, $class;
    $this->{'HOSTS'} = [];
    $this;

} # end sub new

sub addHost {

    my $self = shift;
    my $host = shift || confess "Cannot add host to host pool:  Host not found.";
    push @{$self->{'HOSTS'}}, $host;
    return 1;

} # end sub addHost

sub hosts {

    my $self = shift;

    return wantarray ? @{$self->{'HOSTS'}} : $self->{'HOSTS'};

} # end sub hosts

sub getHostByName {

    my $self = shift;
    my $name = shift || confess "Cannot get host by name:  No name found.";

    my $return = undef;
    for my $host ($self->hosts()) {
	$return = $host if $host->name() eq $name or $host->name() eq 'localhost';
    }
    return $return;

} # end sub getHostByName

sub getHostByAddr {

    my $self = shift;
    my $addr = shift || confess "Cannot get host by IP address:  No address found.";

    my $return = undef;
    for my $host ($self->hosts()) {
	$return = $host if $host->ip() eq $addr or $host->ip() eq '127.0.0.1';
    }
    return $return;

} # end sub getHostByName

sub addHosts {

    my $self = shift;
    my $hosts = shift || confess "Cannot add hosts:  No hosts found.";

    my @hosts = split /\s*,\s*/, $hosts;
    for my $id (@hosts) {
	my $host = new Net::Peep::Host;
	if ($host->isIP($id)) {
	    $host->ip($id);
	} else {
	    $host->name($id);
	}
	$self->addHost($host);
    }
    return 1;

} # end sub addHosts

sub isInHostPool {

    my $self = shift;
    my $nameOrIp = shift || confess "Cannot check whether host is in pool:  No host found.";

    my @hosts = $self->hosts();

    my $host = new Net::Peep::Host;

    my $got;
    if ($host->isIP($nameOrIp)) {
	$got = $self->getHostByAddr($nameOrIp);
    } else {
	$got = $self->getHostByName($nameOrIp);
    }

    return $got ? 1 : 0;

} # end sub isInHostPool

sub areInHostPool {

    my $self = shift;
    my @hosts = @_;

    my $host = new Net::Peep::Host;

    my $return = 0;
    for my $nameOrIp (@hosts) {
	my $got;
	if ($self->isIP($nameOrIp)) {
	    $got = $self->getHostByAddr($nameOrIp);
	} else {
	    $got = $self->getHostByName($nameOrIp);
	}
	$return++ if $got;
    }

    return $return;

} # end areInHostPool

1;

__END__