###########################################################################
## Copyright (c) 2012 Pat Riehecky
## Covered under the included MIT/X-Consortium License:
##    http://www.opensource.org/licenses/mit-license.php
## Permission is hereby granted, free of charge, to any person obtaining a
## copy of this software and associated documentation files (the "Software"),
## to deal in the Software without restriction, including without limitation
## the rights to use, copy, modify, merge, publish, distribute, sublicense,
## and/or sell copies of the Software, and to permit persons to whom the
## Software is furnished to do so, subject to the following conditions:
##
## The above copyright notice and this permission notice shall be included
## in all copies or substantial portions of the Software.
##
## THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
## OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
## MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
## IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
## CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
## TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
## SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
############################################################################

use strict;

my $Detail = $ENV{'LOGWATCH_DETAIL_LEVEL'} || 0;
my $enable_scan = $ENV{'mdadm_enable_scan'} || 0;

my @devices = ();
if ( -f "/etc/mdadm.conf" ) {
  open(MDADM,"< /etc/mdadm.conf");
} elsif ( -f "/etc/mdadm/mdadm.conf" ) {
  open(MDADM,"< /etc/mdadm/mdadm.conf");
} elsif ($enable_scan) {
  open(MDADM,"mdadm --detail --scan 2>/dev/null|");
}
while (<MDADM>) {
  if (/^ARRAY/) {
    push(@devices,(split())[1]);
  }
}
close(MDADM);

DEV: foreach my $dev (@devices) {
  my %mdhash;

  open(MDADM,"mdadm --misc --detail $dev 2>&1 |");
  while (<MDADM>) {
    if ($_ =~ /cannot open .*: No such file or directory/) {
      print $_;
      close(MDADM);
      next DEV;
    }

    $mdhash{'level'} = $1 if ($_ =~ /Raid Level ?: ?(.*)$/);
    $mdhash{'active'} = $1 if ($_ =~ /Active Devices ?: ?(.*)$/);
    $mdhash{'working'} = $1 if ($_ =~ /Working Devices ?: ?(.*)$/);
    $mdhash{'failed'} = $1 if ($_ =~ /Failed Devices ?: ?(.*)$/);
    $mdhash{'spare'} = $1 if ($_ =~ /Spare Devices ?: ?(.*)$/);
    $mdhash{'state'} = $1 if ($_ =~ /State ?: ?(.*)$/);
    $mdhash{'rebuild'} = $1 if ($_ =~ /Rebuild Status ?: ?(.*)$/);
    push(@{$mdhash{'good devices'}},$1) if ($_ =~ /sync .*(\/dev\/[\w\d\-\_]*)/);
    push(@{$mdhash{'middle devices'}},$1) if ($_ =~ /rebuilding .*(\/dev\/[\w\d\-\_]*)/);
    push(@{$mdhash{'bad devices'}},$1) if ($_ =~ /faulty .*(\/dev\/[\w\d\-\_]*)/);
  }
  close(MDADM);

  if ($Detail <= 4) {
    if (lc($mdhash{'state'}) =~ /clean|active/) {
      print "$dev : $mdhash{'state'}\n" if $Detail;
    } else {
      print "$dev : $mdhash{'state'}\n";
      if (defined($mdhash{'middle devices'})) {
        if (defined($mdhash{'rebuild'}) and ($mdhash{'rebuild'} ne '')) {
          print "\tRebuilding status: $mdhash{'rebuild'}\n";
        }
        print "\tRebuilding @{$mdhash{'middle devices'}}\n";
      }
      if (defined($mdhash{'bad devices'})) {
        print "\tFailed @{$mdhash{'bad devices'}}\n";
      }
    }
  }
  elsif($Detail <= 9) {
    if (lc($mdhash{'state'}) =~ /clean|active/) {
      print "$dev : $mdhash{'state'} - @{$mdhash{'good devices'}}\n";
    } else {
      print "$dev : $mdhash{'state'}\n";
      if (@{$mdhash{'middle devices'}}) {
        if (defined($mdhash{'rebuild'}) and ($mdhash{'rebuild'} ne '')) {
          print "\tRebuilding status: $mdhash{'rebuild'}\n";
        }
        print "\t  Rebuilding : @{$mdhash{'middle devices'}}\n";
      }
      if (@{$mdhash{'bad devices'}}) {
        print "\t      Failed : @{$mdhash{'bad devices'}}\n";
      }
      print "\t        Good : @{$mdhash{'good devices'}}\n";
    }
  } else {
    print "$dev : $mdhash{'state'}\n";
    print "\t  Raid Level : $mdhash{'level'}\n";
    print "\tGood Devices : @{$mdhash{'good devices'}}\n";
    if (defined ($mdhash{'middle devices'}) and @{$mdhash{'middle devices'}}) {
      if (defined($mdhash{'rebuild'}) and ($mdhash{'rebuild'} ne '')) {
        print "\tRebuilding status: $mdhash{'rebuild'}\n";
      }
      print "\t  Rebuilding : @{$mdhash{'middle devices'}}\n";
    }
    if (defined($mdhash{'bad devices'}) and @{$mdhash{'bad devices'}}) {
      print "\t      Failed : @{$mdhash{'bad devices'}}\n";
    }
    if ($mdhash{'spare'} ne 0) {
      print "\t      Spares : $mdhash{'spare'}\n";
    }
    print "\n";
  }
}