File Coverage

blib/lib/GraphViz/Zone.pm

Criterion	Covered	Total	%
statement	7	9	77.7
branch			n/a
condition			n/a
subroutine	3	3	100.0
pod			n/a
total	10	12	83.3

package GraphViz::Zone;

use strict;
use vars qw($VERSION);

$VERSION = '0.01';

use GraphViz;

sub new {
	my ($this, @args) = @_;
	my $class = shift;
	my $self = bless { @_ }, $class;
	return $self->_init(@_);
}


sub _init {
	my $g = GraphViz->new(directed => 0, width => 7, height => 7);

	my ($self, %params) = @_;;
	my ($host, $ip, $text, %host_info);

	# Caller can set:
	#  \- zonefile:		(mandatory)
	#  \- output: 		(default to 'zone.png')

	return undef unless exists($params{zonefile});
	$self->{output} 	||= 'zone.png';

	# Open the zone file for reading.
	open(ZONEFILE, $self->{zonefile});

	while() {
		# If we have an A record.
		if (m/^(\w*)\s+IN\s+A\s+(.*)$/) {
			if (length($2) > 0) {
				# Create a hash entry for the host, add the IP.
				$host_info{$1}{ip} = $2;
			}
		}

		# If we have a TXT record.
		if (m/^(\w*)\s+IN\s+TXT\s+"(.*)"$/) {
			# Add the TXT entry to the host's hash key.
			my $text = $2;
			if (length($text) > 0 and exists($host_info{$1})) {
				$host_info{$1}{text} = $text;
			}
		}
	}

	$g->add_node($self->{zonefile});
	for(keys %host_info) {
		my $string = "$_\n";
		$string .= "IP: $host_info{$_}{ip}\n" if (defined ($host_info{$_}{ip}));
		$string .= "Text: $host_info{$_}{text}" if (defined ($host_info{$_}{text}));
		$g->add_node($_, label => $string, minlen => 5, weight => 0.5);
		$g->add_edge($self->{zonefile} => $_);
	}

	$g->as_png($self->{output});

}


1;
__END__