#!/usr/bin/perl

use strict;
use warnings;

use SiteSummary;
use Getopt::Std;

my %sites;
my %sitegroups;
my %hostmap;
my %sitegroupmap;
my %opts;

sub usage {
    my $retval = shift;
    print <<EOF;
Usage: $0 [-l]
   -l   list hosts with the given site/sitegroup
EOF
    exit $retval;
}

getopt("l", \%opts) || usage(1);

for_all_hosts(\&handle_host);

print_summary();

sub handle_host {
    my $hostid = shift;
    #print "$hostid\n";
    for my $site (get_site($hostid)) {
        $site = "SiteMissing" unless defined $site;
        $sites{$site}++;
        $hostmap{$site} = [] unless exists $hostmap{$site};
        my $sitegroup = get_sitegroup($hostid);
        if ($sitegroup) {
            $sitegroups{$site}{$sitegroup}++;
            if (exists $sitegroupmap{$site} && exists $sitegroupmap{$site}{$sitegroup}) {
                push @{$sitegroupmap{$site}{$sitegroup}}, $hostid ;
            } else {
                $sitegroupmap{$site}{$sitegroup} = [$hostid];
            }
        } else {
            push @{$hostmap{$site}}, $hostid ;
        }
    }
}

sub print_summary {
    printf("  %-20s %5s\n", "site", "count");
    for my $site (sort keys %sites) {
        printf("  %-20s %5d\n", $site, $sites{$site});
        if (exists $opts{l}) {
            for my $hostid (sort @{$hostmap{$site}}) {
                my $hostname = get_hostname($hostid);
                my $site = get_site($hostid) || "";
                my $sitegroup = get_sitegroup($hostid) || "";
                printf "    %s %s/%s %s\n", $hostname, $site, $sitegroup, $hostid;
            }
        }
        for my $sitegroup (sort keys %{$sitegroups{$site}}) {
            printf("    %-18s %5d\n", $sitegroup, $sitegroups{$site}{$sitegroup});
            if (exists $opts{l}) {
                for my $hostid (sort @{$sitegroupmap{$site}{$sitegroup}}) {
                    my $hostname = get_hostname($hostid);
                    my $site = get_site($hostid) || "";
                    my $sitegroup = get_sitegroup($hostid) || "";
                    printf "      %s %s/%s %s\n", $hostname, $site, $sitegroup, $hostid;
                }
            }
        }
    }
}