#!/bin/bash
# Script to check number of items in a directory and alert appropriately
# 7/8/2011
# Usage: check_dir_filecount.sh [directory] [warn value] [critical value]
#

# Return codes:
STATE_OK=0
STATE_WARNING=1
STATE_CRITICAL=2
STATE_UNKNOWN=3

# Arguments:
DIRECTORY=$1
WARNLEVEL=$2
CRITLEVEL=$3

# Right number of args?

if [ $# -ne 3 ]
then
   echo "Usage: dir_filecount [directory] [warnlevel] [critlevel]"
   exitstatus=$STATE_UNKNOWN
   exit $exitstatus
fi

# Does $DIRECTORY exist?

if [ ! -d $DIRECTORY ]; then
   echo "Error, $DIRECTORY does not exist."
   exitstatus=$STATE_UNKNOWN
   exit exitstatus
fi

# Is the crit level less than the warn level?

if [ $CRITLEVEL -lt $WARNLEVEL -o $CRITLEVEL -eq $WARNLEVEL ]; then
   echo "Error, warning value should be less than critical value."
   exitstatus=$STATE_UNKNOWN
   exit $exitstatus
fi

# Do we have grep/wc installed?
grepproc=`which grep`
wcproc=`which wc`

if [ $grepproc == "" ]; then
    echo "Error, /bin/grep doesn't exist."
    exitstatus=$STATE_UNKNOWN
    exit $exitstatus
fi

if [ $wcproc == "" ]; then
    echo "Error, /usr/bin/wc doesn't exist."
    exitstatus=$STATE_UNKNOWN
    exit $exitstatus
fi

# Get current file count:
FCOUNT=`ls -l $DIRECTORY |grep -v tot* |wc -l`

if [ $FCOUNT -lt $WARNLEVEL ]
then
   echo "Ok, $FCOUNT items in $DIRECTORY"
   exitstatus=$STATE_OK
   exit $exitstatus
elif [ $FCOUNT -lt $CRITLEVEL -a $FCOUNT -gt $WARNLEVEL ]
then
    echo "WARNING: $FCOUNT items in $DIRECTORY"
    exitstatus=$STATE_WARNING
    exit $exitstatus
elif [ $FCOUNT -gt $CRITLEVEL ]
then
    echo "CRITICAL: $FCOUNT items in $DIRECTORY"
    exitstatus=$STATE_CRITICAL
    exit $exitstatus
fi

define service{
   host_name                 client-233
   use                             generic-service
   check_command          check_nrpe!check_dir_count
   service_description      check_curr_dir_filecount
   notifications_enabled  1
   event_handler_enabled  1
   notification_period    t247
   check_period           t247
   max_check_attempts     3
   check_interval         5
   contact_groups         tech-admins
   retry_check_interval   2
   notification_options   w,u,c
}