#!/usr/bin/perl -w

use strict;
use Fierce::Parser;
use Data::Dumper;
my $fp = new Fierce::Parser;


if (defined($ARGV[0])){
    my $file = $ARGV[0];
    my $parser = $fp->parse_file($file);
    my @nodes    = $parser->get_all_nodes();

    foreach my $n (@nodes){
        my $node = $n;
        my $name_servers = $node->name_servers;
        my $zone_transfers = $node->zone_transfers;
        my $bruteforce = $node->bruteforce;
        my $vhost = $node->vhost;
        my $subdomain_bruteforce = $node->subdomain_bruteforce;
        my $ext_bruteforce = $node->ext_bruteforce;
        my $reverse_lookups = $node->reverse_lookups;
        my $wildcard = $node->wildcard;
        my $findmx = $node->findmx;
        print "===== " . $n->domain . " ====\n";
        if ( $name_servers ) {
            print "Nameservers:\n";
            foreach my $i ( @{ $name_servers->nodes } ) {
                print "\tHostname:" . "\t" . $i->hostname . "\n";
                print "\tIP:" . "\t\t" . $i->ip . "\n";
                print "\tZoneTransfer:" . "\t";
                ( scalar(grep($i->hostname eq $_->name_server, @{ $zone_transfers->result } )) == 1 ) ? print "Enabled\n" : print "Disabled\n";
            }   
            print "\n";
        }
        if ( $findmx ) {
            print "MX:\n";
            foreach ( @{ $findmx->result } ) {
                print "\tPreference:" . "\t" . $_->preference . "\n";
                print "\tExchange: " . "\t" . $_->exchange . "\n";
            }   
            print "\n";
        }
        
        if ($bruteforce) {
            print "Prefix Bruteforce:\n";
            foreach ( @{ $bruteforce->nodes } ) {
                print "\tHostname:" . "\t" . $_->hostname . "\n";
                print "\tIP: " . "\t\t" . $_->ip . "\n";
            }
            print "\n";
        }
        if ($vhost) {
            print "Virtual Hosts:\n";
            foreach ( @{ $vhost->nodes } ) {
                print "\tHostname:" . "\t" . $_->hostname . "\n";
                print "\tIP: " . "\t\t" . $_->ip . "\n";
            }
            print "\n";
        }
=begin
        if ($ext_bruteforce){
            print "Extension Bruteforce:\n";
            foreach ( @{ $ext_bruteforce->nodes } ) {
                print "\tHostname:" . "\t" . $_->hostname . "\n";
                print "\tIP: " . "\t\t" . $_->ip . "\n";
            }
            print "\n";
        }
=end
=cut

        if ($reverse_lookups){
            print "Reverse Lookup:\n";
            foreach ( @{ $reverse_lookups->nodes } ) {
                print "\tHostname:" . "\t" . $_->hostname . "\n";
                print "\tIP: " . "\t\t" . $_->ip . "\n";
            }
            print "\n";
        }

    }

    if ($ARGV[1]) {
        print "ARIN:\n";
        my $arin_search = $ARGV[1];

        use Net::Whois::ARIN;
        use Data::Dumper;
        my $w = Net::Whois::ARIN->new(
                    host    => 'whois.arin.net',
                    port    => 43,
                    timeout => 30,
                );
        if (!defined($ARGV[0])){
            print "Usage: $0 [arin query]\n";
            exit;
        }
        my @out = $w->query($arin_search);
        foreach(@out){
            print "\t$_\n";
        }
    }
}
else {
    print "Usage: $0 [fierce-xml] (arin search term)\n";
}