#!/usr/bin/perl
#
# Based on script from https://www.zabbix.com/forum/showthread.php?t=17273
#
# Backuppc
use lib "/usr/share/backuppc/lib";
use BackupPC::Lib;
use BackupPC::CGI::Lib;

#Other
use Data::Dumper;

# Globals
$bpc = BackupPC::Lib->new();
$hosts = $bpc->HostInfoRead();

# Setup
&var_config();
&zabbix_config();

# %Info = Pool Info
# %QueueLen = Command Queues
# %Jobs = jobs

# Collect Data
GetStatusInfo("jobs queueLen info");
#print Dumper(%Info);
&jobs_info();
&pool_info();
&queue_info();
&general_info();
&hosts_info();

# Push Data to Zabbix
foreach my $key (keys %Val) {
	if ($Var{$key})
	{
		zabbix_post($Var{$key},$Val{$key});
		#print "$Var{$key} | $Val{$key}\n";
	}
}
print "1";


# Functions
sub hosts_info
{
	# Variables
	$fullSizeTot = 0;
	$fullCnt = 0;
	$fullCnt2 = 0;
	$incrSizeTot = 0;
	$incrCnt = 0;
	$incrCnt2 = 0;
	$total_speed_full = 0;
	$total_speed_incr = 0;
	
	$no_backups_2 = "NONE";
        $no_backups_3 = "NONE";
        $no_backups_4 = "NONE";
        $no_backups_5 = "NONE";

	while ( my ($host, $value) = each(%$hosts) )
	{

	        my @Backups = $bpc->BackupInfoRead($host);
        	$bpc->ConfigRead($host);
		my %Conf = $bpc->Conf();

		# Variables
		my $fullAge;
		my $fullSize;
		my $fullDur;
		my $incrAge;
		my $incrSize;
		my $incrDur;
                my $vfullAge="fullAge".$host;
                my $vfullSize="fullSize".$host;
                my $vfullDur="fullDur".$host;
                my $vincrAge="incrAge".$host;
                my $vincrSize="incrSize".$host;
                my $vincrDur="incrDur".$host;
                my $vfullErrs="fullErrs".$host;
                my $vfullBadShare="fullBadShare".$host;
                my $vfullBadFile="fullBadFile".$host;
                my $vfulltarErrs="fulltarErrs".$host;
                my $vincrErrs="incrErrs".$host;
                my $vincrBadShare="incrBadShare".$host;
                my $vincrBadFile="incrBadFile".$host;
                my $vincrtarErrs="incrtarErrs".$host;

                $Var{$vfullAge}="backuppc.fullage[".$host."]";
                $Var{$vfullSize}="backuppc.fullsize[".$host."]";
                $Var{$vfullDur}="backuppc.fulldur[".$host."]";
                $Var{$vincrAge}="backuppc.incrage[".$host."]";
                $Var{$vincrSize}="backuppc.incrsize[".$host."]";
                $Var{$vincrDur}="backuppc.incrdur[".$host."]";
                $Var{$vfullErrs}="backuppc.fullerrs[".$host."]";
                $Var{$vfullBadShare}="backuppc.fullbadshare[".$host."]";
                $Var{$vfullBadFile}="backuppc.fullbadfile[".$host."]";
                $Var{$vfulltarErrs}="backuppc.fulltarerrs[".$host."]";
                $Var{$vincrErrs}="backuppc.increrrs[".$host."]";
                $Var{$vincrBadShare}="backuppc.incrbadshare[".$host."]";
                $Var{$vincrBadFile}="backuppc.incrbadfile[".$host."]";
                $Var{$vincrtarErrs}="backuppc.incrtarerrs[".$host."]";
 	       	for ( my $i = 0 ; $i < @Backups ; $i++ ) 
		{
                        if ( $Backups[$i]{type} eq "full" ) 
			{
                		$fullCnt++;
                		if ( $fullAge < 0 || $Backups[$i]{startTime} > $fullAge ) 
				{
                			$fullAge  = $Backups[$i]{startTime};
                    			$fullSize = $Backups[$i]{size};
                    			$fullDur  = $Backups[$i]{endTime} - $Backups[$i]{startTime};
            		                $Val{$vfullErrs}=$Backups[$i]{xferErrs};
                                        $Val{$vfullBadShare}=$Backups[$i]{xferBadShare};
                                        $Val{$vfullBadFile}=$Backups[$i]{xferBadFile};
                                        $Val{$vfulltarErrs}=$Backups[$i]{tarErrs};
                                        $Val{$vfullAge}=(time() - $fullAge)/86400;
                                        $Val{$vfullSize}=$fullSize;
                                        $Val{$vfullDur}=$fullDur;
                		}
                	$fullSizeTot += $Backups[$i]{size};
            		} 
			else 
			{
                		$incrCnt++;
                		if ( $incrAge < 0 || $Backups[$i]{startTime} > $incrAge ) 
				{
                    			$incrAge  = $Backups[$i]{startTime};
                                        $incrSize = $Backups[$i]{size};
                                        $incrDur  = $Backups[$i]{endTime} - $Backups[$i]{startTime};
            		                $Val{$vincrErrs}=$Backups[$i]{xferErrs};
                                        $Val{$vincrBadShare}=$Backups[$i]{xferBadShare};
                                        $Val{$vincrBadFile}=$Backups[$i]{xferBadFile};
                                        $Val{$vincrtarErrs}=$Backups[$i]{tarErrs};
                                        $Val{$vincrAge}=(time() - $incrAge)/86400;
                                        $Val{$vincrSize}=$incrSize;
                                        $Val{$vincrDur}=$incrDur;
				}
				$incrSizeTot += $Backups[$i]{size};
                	}
            	}
		# Sum the Last Full Backup Speed
		if ($fullSize > 0 && $fullDur >0)
		{
			$total_speed_full += ($fullSize / $fullDur);
			$fullCnt2++;
		}
		# Sum the Last Full Incr Speed
                if ($incrSize > 0 && $incrDur >0)
                {
                        $total_speed_incr += ($incrSize / $incrDur);
			$incrCnt2++;
                }

		if ($host eq "haulingaz")
		{ 
			print "Full Age: $fullAge\n";
			print "Full Period: $Conf{FullPeriod}\n";
			print "Incr Age: $incrAge\n";
			print "Incr Period: $Conf{IncrPeriod}\n";
		}

                # Check for Hosts that don't have backups (Full)         
		my $skip_inc = 0;

		# If we have a full that less than 2 days old we can skip
                if ( (time() - $fullAge) < ( (86400 * 2) ) && $Conf{BackupsDisable} == 0 )
                { $skip_inc = 1;}

		# If we don't have any fulls, we are going to set the no_backups so we can skip inc
                if ( (time() - $fullAge) > ( ($Conf{FullPeriod} * 86400) + (86400 * 2) ) && $Conf{BackupsDisable} == 0 )
                { if ($no_backups_2 eq "NONE") { $no_backups_2 = "$host"} else { $no_backups_2 .= "\n$host" } $skip_inc = 1;}

                if ( (time() - $fullAge) > ( ($Conf{FullPeriod} * 86400) + (86400 * 3) ) && $Conf{BackupsDisable} == 0 )
                { if ($no_backups_3 eq "NONE") { $no_backups_3 = "$host"} else { $no_backups_3 .= "\n$host" } $skip_inc = 1;}

                if ( (time() - $fullAge) > ( ($Conf{FullPeriod} * 86400) + (86400 * 4) ) && $Conf{BackupsDisable} == 0 )
                { if ($no_backups_4 eq "NONE") { $no_backups_4 = "$host"} else { $no_backups_4 .= "\n$host" } $skip_inc = 1;}

                if ( (time() - $fullAge) > ( ($Conf{FullPeriod} * 86400) + (86400 * 5) ) && $Conf{BackupsDisable} == 0 )
                { if ($no_backups_5 eq "NONE") { $no_backups_5 = "$host"} else { $no_backups_5 .= "\n$host" } $skip_inc = 1;}

		# Check for Hosts that don't have backups (Incremental)		
		if ( (time() - $incrAge) > ( ($Conf{IncrPeriod} * 86400) + (86400 * 2) ) && $Conf{BackupsDisable} == 0 && $Conf{IncrKeepCnt} > 0 && $skip_inc == 0)
		{ if ($no_backups_2 eq "NONE") { $no_backups_2 = "$host"} else { $no_backups_2 .= "\n$host" } }
		
		if ( (time() - $incrAge) > ( ($Conf{IncrPeriod} * 86400) + (86400 * 3) ) && $Conf{BackupsDisable} == 0 && $Conf{IncrKeepCnt} > 0 && $skip_inc == 0)
                { if ($no_backups_3 eq "NONE") { $no_backups_3 = "$host"} else { $no_backups_3 .= "\n$host" } }

		if ( (time() - $incrAge) > ( ($Conf{IncrPeriod} * 86400) + (86400 * 4) ) && $Conf{BackupsDisable} == 0 && $Conf{IncrKeepCnt} > 0 && $skip_inc == 0)
                { if ($no_backups_4 eq "NONE") { $no_backups_4 = "$host"} else { $no_backups_4 .= "\n$host" } }

		if ( (time() - $incrAge) > ( ($Conf{IncrPeriod} * 86400) + (86400 * 5) ) && $Conf{BackupsDisable} == 0 && $Conf{IncrKeepCnt} > 0 && $skip_inc == 0)
                { if ($no_backups_5 eq "NONE") { $no_backups_5 = "$host"} else { $no_backups_5 .= "\n$host" } }


       	}
	if ($fullCnt2 > 0 )
	{ $Val{hostsAvgFullSpeed} = ($total_speed_full / $fullCnt2); }
	if ($incrCnt2 > 0 )
	{ $Val{hostsAvgIncrSpeed} = ($total_speed_incr / $incrCnt2); }
	$Val{hostsFullSize}  = $fullSizeTot;
	$Val{hostsFullCount} = $fullCnt;
        $Val{hostsIncrSize}  = $incrSizeTot;
        $Val{hostsIncrCount} = $incrCnt;
	$Val{hostsNoBackups2} = $no_backups_2;
	$Val{hostsNoBackups3} = $no_backups_3;
	$Val{hostsNoBackups4} = $no_backups_4;
	$Val{hostsNoBackups5} = $no_backups_5;	
}

sub pool_info
{
        while ( my ($key, $value) = each(%Info) )
        {
                if ($key =~ /pool/)
                {$Val{$key} = int($Info{$key}); }
        }
}


sub general_info
{
        $Val{startTime}                         = time() - $Info{startTime};
        $Val{Version}                           = $Info{Version};
	$Val{ConfigLTime}			= time() - $Info{ConfigLTime};

}

sub queue_info
{
	while ( my ($key, $value) = each(%QueueLen) ) 
	{
		if ($key =~ /Queue/)
		{$Val{$key} = $QueueLen{$key};}
	}
}

sub jobs_info 
{
        $Val{JobsIncr}  = 0;
        $Val{JobsFull}  = 0;
        $Val{JobsOther} = 0;

	# Jobs
	while ( my ($key, $value) = each(%Jobs) ) {
        	#print "$key => $value\n";
        	#print Dumper($value);
		if (!($key =~ /trashClean/i))
		{
			# Count Incrementail Jobs
			if( $value->{'type'} eq 'incr')
			{ $Val{JobsIncr}++; }

                        # Count Full Jobs
                        elsif( $value->{'type'} eq 'full')
                        { $Val{JobsFull}++; }

			# Everything Else
			else { $Val{JobsOther}++; }
		}
	}

}

sub GetStatusInfo
{
    my($status) = @_;
    ServerConnect();
    %Status = ()     if ( $status =~ /\bhosts\b/ );
    %StatusHost = () if ( $status =~ /\bhost\(/ );
    my $reply = $bpc->ServerMesg("status $status");
    $reply = $1 if ( $reply =~ /(.*)/s );
    eval($reply);
    # ignore status related to admin and trashClean jobs
    if ( $status =~ /\bhosts\b/ ) {
        foreach my $host ( grep(/admin/, keys(%Status)) ) {
            delete($Status{$host}) if ( $bpc->isAdminJob($host) );
        }
        delete($Status{$bpc->trashJob});
    }
}

#
# Returns the list of hosts that should appear in the navigation bar
# for this user.  If $getAll is set, the admin gets all the hosts.
# Otherwise, regular users get hosts for which they are the user or
# are listed in the moreUsers column in the hosts file.
#
sub GetUserHosts
{
    my($getAll) = @_;
    my @hosts;

    if ( $getAll ) {
        @hosts = sort keys %$Hosts;
    } else {
        @hosts = sort grep { $Hosts->{$_}{user} eq $User ||
                       defined($Hosts->{$_}{moreUsers}{$User}) } keys(%$Hosts);
    }
    return @hosts;
}


sub ServerConnect
{
    #
    # Verify that the server connection is ok
    #
    return if ( $bpc->ServerOK() );
    $bpc->ServerDisconnect();
    if ( my $err = $bpc->ServerConnect($Conf{ServerHost}, $Conf{ServerPort}) ) {
        if ( CheckPermission()
          && -f $Conf{ServerInitdPath}
          && $Conf{ServerInitdStartCmd} ne "" ) {
            my $content = eval("qq{$Lang->{Admin_Start_Server}}");
            Header(eval("qq{$Lang->{Unable_to_connect_to_BackupPC_server}}"), $content);
            Trailer();
            exit(1);
        } else {
            ErrorExit(eval("qq{$Lang->{Unable_to_connect_to_BackupPC_server}}"));
        }
    }
}


sub zabbix_config {

	open(CONFIG,"/etc/zabbix/zabbix_agentd.conf");
	foreach(<CONFIG>)
	{
		$zabbix_host = $1 if (/Hostname\s*=\s*(.*)/); 
		$zabbix_server = $1 if (/Server\s*=\s*(.*)/);
	}
	close CONFIG;
}

sub zabbix_post {
	my $key = $_[0];
	my $val = $_[1];
        my @servers = split(',', $zabbix_server);

        foreach my $server (@servers) {
          my $cmd = "zabbix_sender -z $server -p 10051 -s $zabbix_host -k $key -o '$val'";
          system("$cmd >/dev/null");
        }
}


sub var_config { 
	
	# Pool Info
	$Var{poolFileCnt}     			= "backuppc.pool_file_count";
	$Var{poolDirCnt}      			= "backuppc.pool_dir_count";
	#$Var{poolFileCntRm}                	= "backuppc.pool_file_removed";
        $Var{poolFileCntRep}                  	= "backuppc.pool_file_repeat";
        $Var{poolFileRepMax}                	= "backuppc.pool_file_repeat_max";
	$Var{poolFileLinkMax}                   = "backuppc.pool_file_link_max";
        $Var{poolKb}            		= "backuppc.pool_size";
        $Var{cpoolFileCnt}     			= "backuppc.cpool_file_count";
        $Var{cpoolDirCnt}      			= "backuppc.cpool_dir_count";
        #$Var{cpoolFileCntRm}                	= "backuppc.cpool_file_removed";
        $Var{cpoolFileCntRep}                  	= "backuppc.cpool_file_repeat";
        $Var{cpoolFileRepMax}                	= "backuppc.cpool_file_repeat_max";
        $Var{cpoolFileLinkMax}                  = "backuppc.cpool_file_link_max";
        $Var{cpoolKb}            		= "backuppc.cpool_size";

	# General Stats
        $Var{startTime}                         = "backuppc.uptime";
        $Var{Version}                           = "backuppc.version";
	$Var{ConfigLTime}			= "backuppc.config_load_time";

	# Queues
        $Var{CmdQueue}                          = "backuppc.queue_command";
        $Var{UserQueue}                         = "backuppc.queue_user";
        $Var{BgQueue}                           = "backuppc.queue_background";

	# Jobs
        $Var{JobsIncr}                          = "backuppc.jobs_incr";
        $Var{JobsFull}                          = "backuppc.jobs_full";
        $Var{JobsOther}                         = "backuppc.jobs_other";

	# Hosts
        $Var{hostsAvgFullSpeed}                 = "backuppc.hosts_full_speed";
        $Var{hostsAvgIncrSpeed}                 = "backuppc.hosts_incr_speed";
        $Var{hostsFullSize}                     = "backuppc.hosts_full_size";
        $Var{hostsFullCount}                    = "backuppc.hosts_full_count";
        $Var{hostsIncrSize}                     = "backuppc.hosts_incr_size";
        $Var{hostsIncrCount}                    = "backuppc.hosts_incr_count";
	$Var{hostsNoBackups2}			= "backuppc.hosts_nobackup_2";
	$Var{hostsNoBackups3}			= "backuppc.hosts_nobackup_3";
	$Var{hostsNoBackups4}			= "backuppc.hosts_nobackup_4";
	$Var{hostsNoBackups5}			= "backuppc.hosts_nobackup_5";
}