#!/usr/bin/perl -w
# (c) 2012 Joerg Mann / 1&1 Internet AG
#
# $Id: 89bfc1b46c3f5907b173cb59379663c9b9f06049 $
# last update at Tue Apr 10 17:20:32 CEST 2012 by joerg.mann@1und1.de

# TODO:
# check deutsch/englich
# add todo-global delete-logfiles
# anzeige bandbreite / i/o-load wenn dieser genutzt wird
# bugix fuer nachtaegliches join (log-v-4 ...)

###
use warnings;
use strict;
use English;
use Getopt::Long;
use Term::ANSIColor;
use Date::Language;
use POSIX qw(strftime);

### defaults
my $version       = "0.067v";
my $alife_timeout = "99";	# sec
my $is_tty 	  = 0;
my $mars_dir      = '/mars';
my $himself       = `uname -n` or die "cannot determine my network node name\n";
my $clearscreen	  = `clear`;
chomp $himself;


### ARGV
# Optionen in Hash-Ref parsen
my $params = {};
GetOptions( $params, 'help', 'h', 'version', 'v', 'monitor', 'm', 'resource=s', 'interval=i', 'long', 'history', 'debug' );

if($params->{help} || $params->{h}) {
    print "Usage: mars-status [--help]\n";
    print "Usage: mars-status [--version]\n";
    print "Usage: mars-status (without specification of parameters, an abstract of all the informations spent)\n";
    print "Usage: mars-status [--resource <RESNAME>] [--interval <seconds>] [--long [--history] [--debug]]\n";
    print "       --resource : limits the display to the specified resource\n";
    print "       --interval : refreshes the display every second xxx\n";
    print "       --long     : displays more detailed information\n";
    print "       --history  : shows information about the log files, version numbers and their status\n";
    print "       --debug    : additional display error messages and warnings to internal\n";
    print "       --monitor  : indicator to use for monitoring (not available)\n";
    print "\nAdvanced information are also available here: http://http://wiki.intranet.1and1.com/bin/view/PO/ProjektTEC1603\n";
    exit;
}

# Farbe zuruecksetzen
$SIG{INT} = sub {
    print color 'reset';
    print $clearscreen;
    exit;	
};


### figure out TTY
my $tty = readlink '/dev/stdout';
while ( my $temp = readlink $tty ) {
    $tty = $temp;
}
if ( $tty =~ /^\/dev\/pts\// ) {
    $is_tty = 1;
} elsif ( $tty =~ /^\/dev\/tty/ ) {
    $is_tty = 1;
}

#########################################################################################
### subs
sub check_link {
    my $dir = shift;
    my $result = readlink $dir;
    if ( !$result ) {
	return 0;
    } else {
	return $result;
    }
}

### print color
sub print_screen {
    my $text  = shift;
    my $color = shift;
    print color "$color" if ( $is_tty );
    print "$text";
    print color 'reset' if ( $is_tty );
}

### read links
sub convert_link {
    my $link = shift;
    $link = check_link "$link";
    if ( ( !$link ) || ( $link eq 0 ) ) {
        print_screen "off", 'red';
    } else {
	print_screen "on", 'green';
    }
    return $link;
}

#########################################################################################
### sub display resource-partner
sub display_partner {
    my %p 			= @_;
    my $PRes  		= $p{ressource};
    my $PName 		= $p{nodename};
    my $PSize 		= $p{ressource_size};
    my $ref_ResPartner  	= $p{res_partner};
    my $ref_ResInReplay 	= $p{res_inreplay};
    my $ref_ResInSync   	= $p{res_insync};
    my $ref_AULogfile	= $p{res_AULogfile};

    my $PStatus  = check_link "$mars_dir/$PRes/primary";
    my $PDevice  = check_link "$mars_dir/$PRes/device-$PName";  
    my $Ljoined  = check_link "$mars_dir/$PRes/device-$himself";
    
    ### status
    if ( $PStatus eq $PName ) {   
        print_screen "as Primary, ",'blue';
    } else {
        if ( $PDevice eq 0 ) {
            print_screen "not joined, ",'red';
        } else {
            print_screen "as Secondary, ",'blue';
        }
    }

    ### alive
    my @PAlive = lstat("$mars_dir/alive-$PName");
    if ( !$PAlive[9] ) { $PAlive[9]=0 };
    my $PAlive =  time()- $PAlive[9] - $alife_timeout;
    if ( $PAlive > 1  ) {
        print_screen "Status: unknown (last message before $PAlive sec) !!!\n", 'red';
    } else {
        print_screen "Status: connected\n",'blue'; 
    }

    ### device
    # joined  ?
    if ( $PDevice eq 0 ) {
        if ( $params->{'long'} ) { print_screen "   -> Resource is not joined to this node\n", 'red'; }
        return; 
    }
    if ( $params->{'long'} ) {
        print "\tDevice  : ".check_link "$mars_dir/$PRes/data-$PName";
        print ", used as $PDevice";

        # check mountpint
        if ( $himself eq $PName ) {
            my $PUDevice = "/dev/mars/$PDevice";
            #print " and ";
            if ( stat( $PUDevice) ) {
                open my $fh, '<', '/proc/mounts' or die $!;
                $PUDevice = ( grep { /^$PUDevice / } <$fh> )[0];
                if ( $PUDevice ) {
                    $PUDevice = ( split / /, $PUDevice )[1];
                    print_screen " and mountet as $PUDevice\n",'blue';
                } else {
                    print_screen "\n\t\t---> TODO: enable to mount\n",'green';
                }
            } else {
                print_screen "\n\t\t---> HINT: unable to mount, Device is Secondary or mars is starting ...\n",'blue';
            }
        } else {
            print "\n";
        }
    }
    $$ref_ResPartner++;


    ### logfile
    my @PLogFile  = split (',', check_link "$mars_dir/$PRes/replay-$PName" );
    my @PLogLink  = split ("-", $PLogFile[0]);
    # TODO: kein Logfile vorhanden ...
    my $PLogName  = "$PLogLink[0]-$PLogLink[1]";
    my $PLogSize  = -s "$mars_dir/$PRes/$PLogFile[0]";
    if ( !$PLogFile[1] ) { $PLogFile[1] = 0; $PLogFile[2] = 0; }
    if (( !$PLogSize ) || ( $PLogSize eq 0 )) { $PLogSize = 0.0001; }
    if ( $params->{'long'} ) {
        printf "\tLogfile : %s with %s bytes (%.3fGB) received\n", $PLogName, $PLogSize, ( $PLogSize/1024/1024/1024 );
        if ( $Ljoined eq "0" || $PLogSize eq "0.0001" ) {
            print_screen "\t\t---> TODO: Logfile inactive or empty (Size: $PLogSize)\n", 'red';
        }
        if ( ( $ref_AULogfile ) && !($PLogName eq $ref_AULogfile) ) {
            print_screen "\t\t---> TODO: Logfile Version not actual = ($PLogName ! $ref_AULogfile)\n", 'red';
        }
    }
	

    ### replay
    my $RStatus = sprintf("%.2f", ( $PLogFile[1] / $PLogSize * 100));
    if ( $Ljoined eq "0" || $PLogSize eq "1" ) { $RStatus = 0; }
    $$ref_ResInReplay = $RStatus;
    if ( $params->{'long'} ) {
        printf "\tReplayed: %s bytes (%.3fGB) replayed, Todo %d (%.3fGB) = ", 
            $PLogFile[1], ( $PLogFile[1]/1024/1024/1024 ), 
            $PLogFile[2], ( $PLogFile[2]/1024/1024/1024 );
        if (( $RStatus < 1 ) && ( $PLogSize != 0.0001 )) {
            print_screen "$RStatus%\n\t\t---> TODO: Replay not started, Logfile inactive or empty (Size: $PLogSize)\n", 'red';
        } elsif (( $RStatus < 100 ) && ( $PLogSize != 0.0001 )) {
            print_screen "$RStatus%\n\t\t---> WORK: Replay in progress = ($RStatus% < 100.00%)\n", 'red';
        } elsif ( $PLogFile[2] > 0 ) {
            $RStatus = sprintf("%.2f", ($PLogFile[1]-$PLogFile[2])/$PLogFile[1] * 100);
            print_screen "$RStatus%\n", 'red';
        } elsif ( $PLogSize = 0.0001 ) {
            $RStatus = "100.00";
            print_screen "$RStatus%\n", 'green';
        } else {
            print_screen "$RStatus%\n", 'green';
        }
    }	


    ### sync
    my $PSyncsize = check_link "$mars_dir/$PRes/syncstatus-$PName";
    my $SStatus = sprintf("%.2f", ($PSyncsize / $PSize * 100));
    $$ref_ResInSync = $SStatus;
    if ( $params->{'long'} ) {
        printf "\tSync    : %s bytes (%.3fTB) synced = ", $PSyncsize, ( $PSyncsize/1024/1024/1024/1024);
        if ( $SStatus < 100) {
            print_screen "$SStatus%\n\t\t---> WORK: Sync in progress = ($SStatus% < 100.00%)\n", 'red';
        } else {
            print_screen "$SStatus%\n", 'green';
        }
    }


    if ( $params->{'long'} ) {
        ### actual
        my $ActStatus  = check_link "$mars_dir/$PRes/actual-$PName/is-primary";
        if ( $ActStatus eq 1 ) {
            print "\tActual  : Status Primary, used Device="; convert_link "$mars_dir/$PRes/actual-$PName/device-$PDevice";
            print "\n";
        } else {
            print "\tActual  : Status Secondary, Syncstatus="; convert_link "$mars_dir/$PRes/actual-$PName/copy-syncstatus-$PName";
            print ", Logfileupdate="; convert_link "$mars_dir/$PRes/actual-$PName/logfile-update";
            print "\n";
        }

        ### switch
        print "\tSwitch  : Attach="; convert_link "$mars_dir/$PRes/todo-$PName/attach";
        print ", Connect="; convert_link "$mars_dir/$PRes/todo-$PName/connect";
        print ", Sync="; convert_link "$mars_dir/$PRes/todo-$PName/sync";
        print ", AllowReplay="; convert_link "$mars_dir/$PRes/todo-$PName/allow-replay";
        print "\n";
    }
    return $PLogName;
}


#########################################################################################
###
sub check_logfile {
    my $LResource 	= shift;
    my $LPartner  	= shift;
    my $oldEqual 	= 0;
    my $LogCount	= 0;
    my $LogFailed   = 0;
    print_screen "   -> History Replay/Status\n",'blue';

    my @logfile 	= <$mars_dir/$LResource/log*>;
    foreach my $logfile (@logfile) {
        my $LVersion    = $logfile;
        $LVersion       =~ s/^.*log-([0-9]+)-.*$/$1/;
        my $LogStatus   = check_link "$logfile";
        my $allEqual    = 1;
        if ( $LogStatus eq 0 ) { 
#	    # info to old logfiles (old loop) ...
#	    if (( $oldEqual eq 1 ) && ( $LogFailed eq 0 )) {
#           	print_screen "\t\t---> TODO: logfiles has all equal Sizes and Checksums, can be deleted?\n",'green';
#               # TODO delete Links
#	     } elsif (( $oldEqual eq 1 ) && ( $LogFailed ne 0 )) {
#               print_screen "\t\t---> TODO: logfiles has same other errors - Please check History of Logfiles\n",'blue';
#	     }

            # found logfile
            my $OldCheck;
            my $OldSize;
            my $LogSize = -s "$logfile";
            print "\tLogfile Version: $LVersion - Size: $LogSize\n";

            # check other ...
            my @LVersion = <$mars_dir/$LResource/version-$LVersion*>;
            foreach my $LVersion (@LVersion) {
                my @LogDetail = split (',', check_link "$LVersion" );
                my $LogServer = $LVersion;
                $LogServer    =~ s/.*[0-9]-//;
                $LogCount++;
                print "\t\tSource: $LogServer, Check: $LogDetail[0], ReplayPosition: $LogDetail[2], Todo: $LogDetail[3] blocks\n";
                # Initial Values
                if ( !defined $OldCheck ) {
                      # new
                      $OldCheck = $LogDetail[0];
                      $OldSize  = $LogDetail[2];
                      $allEqual = 1; 
                } elsif (!(( $LogDetail[0] eq $OldCheck ) and ( $LogDetail[2] eq $OldSize ))) { 
                      # not same
                      $allEqual = 0;
                      if ( !($LogDetail[0] eq $OldCheck) && ($LogDetail[2] eq $OldSize) ) {
                            print_screen "\t\t---> TODO: check logfiles has not equal Checksums and same size !!!\n",'red';
                            $LogFailed = 1;
                      } elsif ( $LogFailed eq 0 )  { 
                            print_screen "\t\t---> TODO: check logfiles has not equal Checksums and different size ???\n",'red';
                            $LogFailed = 1;
                      }
                } else {
                      # same
                      $allEqual = 1;
                }

                # check bad values
                if ( $LogDetail[3] < 0 ) {
                        print_screen "\t\t---> TODO: Found bad values ($LogDetail[3])it's ok ???\n", 'red';
                        $LogFailed = 1;
                }
            } # end foreach

            if ( $allEqual eq 1 ) {
                $oldEqual = 1;
            } else {
                $oldEqual = 0;
            }
            # check Count Logfiles
            if ( !($LogCount eq $LPartner) ) {
                print_screen "\t\t---> TODO: Count of Logfiles different (have:$LPartner found:$LogCount)\n", 'red';
                $LogFailed = 1;
                $oldEqual = 0;
            }
            $LogCount=0;

            if (( $oldEqual eq 1 ) && ( $LogFailed eq 0 )) {
                    print_screen "\t\t*---> TODO: logfiles has all equal Sizes and Checksums, can be deleted?\n",'green';
                    # TODO check aktuell logfile
                    # TODO delete links !
            } elsif (( $oldEqual eq 1 ) && ( $LogFailed ne 0 )) {
                    print_screen "\t\t*---> TODO: logfiles has same other errors - Please check History of Logfiles\n",'red';
            }

        } # end logstatus
    } # end foreach
}

#########################################################################################
### avg_limit
sub check_avg_limit {
    if ( open (MARS_LOADAVG, "< /proc/sys/mars/loadavg_limit") ) {
        my $mars_avg_limit;
        while (<MARS_LOADAVG>) {
            $mars_avg_limit = $_;
        }
        close MARS_LOADAVG;
        print_screen "-> Node AVG-Speed-Limit is ", 'bold';
        if (( !$mars_avg_limit ) || ( $mars_avg_limit < "1" )) {
            print_screen "unset, used full speed\n", 'green';
        } else { 
            print_screen "is $mars_avg_limit", 'red';
        }
    }
}


#########################################################################################
### diskfull
sub check_disk_is_full {
    my @diskfull = glob("$mars_dir/rest-space-*");
    my $diskfull_mars = "";
    print_screen "-> Diskspace on Cluster:", 'bold';
    if ( @diskfull ) { 
        foreach ( @diskfull) {
            my $diskfull_space     = check_link "$_";                        
            my $diskfull_system = $_;                
            $diskfull_system    =~ s!/mars/rest-space-!!;
            if ( $diskfull_space < 1 ) {
                $diskfull_space = sprintf ("%.2f", $diskfull_space / 1024 );
                if ( $diskfull_system eq $himself ) {
                    print_screen "\n\t-> ERROR ! Local Partition $mars_dir full ($diskfull_space kb Limit) !!! mars is stopping !!!\n\n", "red";
                    $diskfull_mars = "$diskfull_mars,$diskfull_system";
                } else {
                    print_screen "\n\t-> WARNING ! Remotesystem $diskfull_system have mars-disk full ($diskfull_space kb Limit) !!!\n\n", "red";
                    $diskfull_mars = "$diskfull_mars,$diskfull_system";
                }
            }
       }
    }
    # TODO /0
    if ( !$diskfull_mars ) {
        print_screen " ok\n", 'green';
    }
}

#########################################################################################
### check /proc/sys/mars/warnings
sub check_mars_warn {
    if ( open  (MARS_WARN, "< /proc/sys/mars/warnings") ) {
        my $mars_warn = "";
        while ( <MARS_WARN> ) {
            if ( m/^(\d+\.\d+)/ ) {
                s/^(\d+\.\d+)/strftime("%a %b %e %H:%M:%S %Y:", localtime($1))/e;
            } else {
                $_ = localtime(0) . ': ' . $_;
            }
            s/MARS_WARN\s+//;
            $mars_warn = "$mars_warn\t$_";
        }
        close MARS_WARN;
        if ( $mars_warn ne "" ) { print_screen "-> MARS WARNINGS:\n", 'red'; print "$mars_warn" }
    } 
}       

#########################################################################################
### check /proc/sys/mars/errors
sub check_mars_error {
    if ( open (MARS_ERROR, "< /proc/sys/mars/errors") ) {
        my $mars_error = "";
        while ( <MARS_ERROR> ) {
            if ( m/^(\d+\.\d+)/ ) {
                s/^(\d+\.\d+)/strftime("%a %b %e %H:%M:%S %Y:", localtime($1))/e;
            } else {
                $_ = localtime(0) . ': ' . $_;
            }
            s/MARS_ERROR\s+//;
            $mars_error = "$mars_error\t$_";
        }
        close MARS_ERROR;
        if ( $mars_error ne "" ) { print_screen "-> MARS ERRORS:\n", 'red'; print "$mars_error" }
    }
}

#########################################################################################
### info version
sub info_version {
    ### module
    my %mars_info;
    open ( my $lsmod_handle,'-|','lsmod | grep mars' ) || die "blub ... $!";
    if (!<$lsmod_handle>) {
        print_screen "Module Mars not running\n",'red';
        sleep(10);
        next;
    }
    open ( my $modinfo_handle, '-|', 'modinfo mars' ) || die "cannot run modinfo mars: $!";
    while ( my $line = <$modinfo_handle> ) {
        chomp $line;
        my ( $key, $value) = split /: +/, $line;
        if ( $value) {
            $mars_info{$key} = $value;
        }
    }
    if ( $mars_info{author} eq "") {
        print_screen "Module Mars not running\n",'red';
        next;
    }
    
    ### status
    print_screen "MARS Status - $himself, $version",'blue';
    if ( $params->{'resource'}  ) { print_screen ", Ressource: $params->{'resource'}",'blue'; }
    print "\n";
    
    ### marsadm
    my $MAVersion = qx"marsadm version";
    print_screen "MARS Admin  - $MAVersion",'blue';
    
    ### module
    print_screen "MARS Module - $mars_info{version}\n",'blue';

    ### kernel
    my $KVersion = '/proc/version';
    open my $Kfh, '<', "$KVersion" or die $!;
    $KVersion = ( grep { /^Linux/ } <$Kfh> )[0];
    $KVersion = ( split / /, $KVersion )[2];
    print_screen "MARS Kernel - $KVersion\n",'blue';
    
    print "-------------------------------------------------------------------------------\n";
}

#########################################################################################
### check ressources
sub check_ressource {
    opendir my $dirhandle, $mars_dir or die "Cannot open $mars_dir: $!";
    my @resources = grep { /^res/ && -d "$mars_dir/$_" } readdir $dirhandle;
    if ( !@resources ) {
        print_screen "---> no resources found\n", 'red';
        exit;
    }
	
    foreach my $res (@resources) {
        my $ResPartner     = 0;
        my $ResInReplay    = 0;
        my $ResInReplaySum = 0;
        my $ResInReplayPar = 0;
        my $ResInSync      = 0;
        my $ResInSyncSum   = 0;
        my $ResInSyncPar   = 0;
        my $res_name       = $res;
        $res_name          =~ s/^resource-//;
        if ( $params->{'resource'} ) {
            if (!( $params->{'resource'} eq $res_name)) {
                next;
            }
        }
        my $res_size     = check_link "$mars_dir/$res/size";
        if ( $res_size eq 0 ) { $res_size = 1 };
        my $res_tbsize   = ( $res_size) / 1024 / 1024 /1024 / 1024;
        my $res_master   = check_link "$mars_dir/$res/primary";
        if ( $res_master eq 0 ) { $res_master = "unknown" };
        print color 'bold' if ( $is_tty );
        printf  "-> check resource %s, with %d bytes (%.3fTB), Primary Node is %s\n", $res_name, $res_size, $res_tbsize, $res_master;
        print color 'reset' if ( $is_tty );
        
        ### him self
        print_screen "   -> local node ($himself) ",'bold';
        my $ActualUsedLogfile = display_partner(
            ressource	=> $res,
            nodename	=> $himself,
            ressource_size	=> $res_size,
            res_partner	=> \$ResPartner,
            res_inreplay    => \$ResInReplay,
            res_insync      => \$ResInSync,
            res_AULogfile   => "",
        );
        if ($ResInReplay ne "100.00") { 
            $ResInReplaySum = $ResInReplaySum + $ResInReplay;
            $ResInReplayPar = $ResInReplayPar + 1; 
        }
        if ($ResInSync ne "100.00") { 
            $ResInSyncSum   = $ResInSyncSum + $ResInSync;
            $ResInSyncPar = $ResInSyncPar + 1; 
        } # end him self

        ### joined ...
        if ( $ResPartner eq 1) {
            ### partners
            opendir my $server_dh, "$mars_dir/$res" or die "Cannot open $mars_dir/$res: $!";
            my @servers = grep { /^data/ && readlink "$mars_dir/$res/$_" } readdir $server_dh;
            @servers    = sort (@servers);
            foreach my $partner (@servers) {
                $partner  =~ s/^data-//;
                if ( $partner eq $himself ) { next; }
                print_screen "   -> remote node ($partner) ", 'bold';
                display_partner(
                    ressource	=> $res,
                    nodename	=> $partner,
                    ressource_size	=> $res_size,
                    res_partner	=> \$ResPartner,
                    res_inreplay    => \$ResInReplay,
                    res_insync      => \$ResInSync,
                    res_AULogfile   => $ActualUsedLogfile,
                    );
            }

            if ($ResInReplay ne "100.00") { 
                $ResInReplaySum = $ResInReplaySum + $ResInReplay;
                $ResInReplayPar = $ResInReplayPar + 1; 
            }
            if ($ResInSync ne "100.00") { 
                $ResInSyncSum   = $ResInSyncSum + $ResInSync;
                $ResInSyncPar = $ResInSyncPar + 1; 
            }
        } # end joined
	
        ### modus
        if ( $ResPartner eq 0) { 
            if ( $params->{'long'} ) { print_screen "   -> modus for $res_name is remote ($ResPartner nodes)\n",'bold'; }
        } elsif ( $ResPartner eq 1 ) { 
            if ( $params->{'long'} ) { print_screen "   -> modus for $res_name is standalone ($ResPartner node)\n",'bold'; }
        } else {
            print_screen "   -> modus for $res_name is cluster ($ResPartner nodes), ",'bold';
            $ResInReplaySum = sprintf("%.2f", $ResInReplaySum / $ResInReplayPar );
            $ResInSyncSum   = sprintf("%.2f", $ResInSyncSum / $ResInSyncPar );
            print_screen "ClusterSummary: ", 'black';
            if ( $ResInReplaySum eq "100.00" ) {
                print_screen "in replay ($ResInReplaySum%),", 'green';
            } elsif ( $ResInReplaySum eq "0.00" ) {
                print_screen "inaktiv ($ResInReplaySum%),", 'red';
            } else {
                print_screen "not in replay ($ResInReplaySum%),", 'red';
            }
            if ( $ResInSyncSum eq "100.00" ) {
                print_screen " in sync ($ResInSyncSum%)\n", 'green';
            } else {
                print_screen " not in sync ($ResInSyncSum%)\n", "red";
            }
        } # end modus
	
        ### debug output
        if ( $params->{'long'} ) { 
                ### history
                if ( $params->{'history'} ) {
                        check_logfile( $res, $ResPartner );
                }
        } # end debug

    } # end foreach

}

#########################################################################################
### main loop ...
while(1) {
    my $dateFormat = Date::Language->new('English');

    ### version only
    if ( $params->{version} || $params->{v}) {
        info_version;
        exit 0;
    }
        
    ### monitor only
    if ( $params->{monitor} || $params->{m}) {
        print "Funktion nicht vorhanden\n";
        exit 0;
    }
        
    #########################################################################################
    ### main run

    print $clearscreen;
    ### read mars infos
    if ( $params->{'long'} ) { 
        # read mars info
        info_version;

        # check load-limit
        check_avg_limit;
    }

    ### check resources
    check_ressource;
	
    ### debug output
    if ( $params->{'long'} ) { 
        ### mars-warn/error
        check_disk_is_full;
        if  ( $params->{'debug'} ) {
            check_mars_warn;
            check_mars_error;
        }
    }

    ### end
    print color 'reset';
    exit if (not $params->{'interval'});
    sleep($params->{'interval'});
}
exit;