#!/bin/bash

# check_cups - nagios plugin
#
# Copyright (C) 2008-2010, Novell, Inc.
# Copyright (C) 2011-2013, SUSE Linux Products GmbH
# Author: Martin Caj <mcaj@suse.cz>
#
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice, this
#   list of conditions and the following disclaimer.
#
# * Redistributions in binary form must reproduce the above copyright notice,
#   this list of conditions and the following disclaimer in the documentation
#   and/or other materials provided with the distribution.
#
# * Neither the name of the Novell nor the names of its contributors may be
#   used to endorse or promote products derived from this software without
#   specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#
# $Id$
#

# Autor notes:	I`d like to thank to John E. Vincent (nagios-plugs@lusis.org)
#		I learn a lof from his check CUPS print queue plugin.
#		Then I`d like to thank to Mark Shirley for his check_cups_printer.sh
#		script, which was also inspiration for me.
#		Both of them you can find on http://exchange.nagios.org/ web site.
#		Martin Caj 31/01/2013 <mcaj@suse.cz>

#		version 0.3 has no awk any more, cut can do it as well.
#		the appamor profile was fix as well.
# 		Martin Caj 01/11/2013 

# check_cups - nagios plugin for checking cups service
# Description:
#
# This plugin will check the status of a remote CUPS
# print service for the printer status, then if status is ok 
# it will check the queue. It can check all avaible printes
# on the cups, or just one of them.
# Then the plugin will check the queue:
# it will provide the size of the queue
# and optionally the age of the queue
#
# Version : 0.3

#searchning the lpstat:
LPSTAT="$(which lpstat)"

# debug the script:
#set -x

# Nagios return codes
STATE_OK=0
STATE_WARNING=1
STATE_CRITICAL=2
STATE_UNKNOWN=3
STATE_DEPENDENT=4


# check it lpstat is missing.
if [ ! -x "$LPSTAT" ]
then
	echo "UNKNOWN: "$LPSTAT" not found or is not executable by the nagios user"
	exitstatus="$STATE_UNKNOWN"
	exit "$exitstatus"
fi

PROGNAME=$(basename $0)


print_usage() {
# Name: print_usage
# Desc: It just print the usage out. 

	echo "Usage: "$PROGNAME" -H <hostname> -P -p<The CUPS printer name> | -Q <s|b>  -w <size warning level> -c <size critical level> -a <max age>"
	echo 
	echo "Notes:"
	echo "-H: Hostname - Can be a hostname or IP address."
	echo "-P: Check only the printers status."
	echo "-p: It will check only one specific printer."
	echo "-Q: Type of check - Can be queue size (s) or both queu size and queue age (b)."
	echo "-w: WARNING level for queue size."
	echo "-c: CRITICAL level for queue size."
	echo "-a: Max age of queue. Returns CRITICAL if jobs exists older than <max age> days."
	echo 
}

print_help() {
# Name: print_help
# Desc: It just print the usage and the help out. Then it end with exit code 0

	print_usage
	echo 
	echo "This plugin will check the CUPS print service for the printer status"
	echo "it can check the queue on a remote (or local with -H localhost) CUPS server."
	echo "It can check both the size of the queue and the age of the oldest print job in the queue."
	echo "-w and -c are for reporting warning and critical levels of the queue size."
	echo "-a is optional for specifying the max age of a job in the print queue. Anything older thatn <max age>"
	echo "will return a CRITICAL"
	echo "For more details have look into README file. "
	echo 
	exit 0
}


check_queue_size() {
# Name: check_queue_size
# Desc: It check the status of the CUPS queue size.
# $exitstatus= might be ok|warn"crittical deppends on -w and -c 
                        
if [ "$JOBCOUNT" -ge "$critlevel" ]
then
	MESSAGE="CRITICAL: CUPS queue size - "$JOBCOUNT"| "$PERFDATA""
        exitstatus="$STATE_CRITICAL"
elif [ "$JOBCOUNT" -ge "$warnlevel" ]
then
	MESSAGE="WARNING: CUPS queue size - "$JOBCOUNT"| "$PERFDATA""
        exitstatus="$STATE_WARNING"
else
	MESSAGE="OK: CUPS queue size - "$JOBCOUNT"| "$PERFDATA""
        exitstatus="$STATE_OK"
fi 

}

check_printer_status() {
# Name:check_printer_status
# Desc: It check status of all printers or one specific printer 
# output is store in $OUTPUT and $exitstatus

if [ -z $printername ]
	then
	echo "Checking all printers..."
	RESULT=$("$LPSTAT" -h "$hostname" -p )
	if [ $? != 0 ]
        then
                echo "ERROR:  Probably wrong host name: "$hostname", or CUPS is not running there." 
                exit "$STATE_UNKNOWN"
        fi

else
	echo "Checking only the printer "$printername"."
	RESULT=$("$LPSTAT" -h "$hostname" -p "$printername")
	if [ $? != 0 ] 
	then
		echo "ERROR: the printer "$printername" doesn't exist on the cups server "$hostname"." 
		echo "please check command: '"$LPSTAT" -h "$hostname" -p' without printer name."
		exit "$STATE_UNKNOWN"
	fi
fi

case "$RESULT" in
        *Rejecting*)
		messages=$(echo "$RESULT"|grep -i rejecting )
                OUTPUT="CRITICAL - CUPS printer is rejecting jobs for: "$messages"."
                exitstatus="$STATE_CRITICAL"
        ;;
        *Unable*)
		messages=$(echo "$RESULT"|grep -i unable )
                OUTPUT="CRITICAL - CUPS Unable to connect: "$messages"."
                exitstatus="$STATE_CRITICAL"
        ;;
        *disabled*)
		messages=$(echo "$RESULT"|grep -i disabled)
                OUTPUT="CRITICAL - CUPS printer: "$messages"."
                exitstatus="$STATE_CRITICAL"
	;;
	*Paused*)
		messages=$(echo "$RESULT"|grep -i paused)
		OUTPUT="WARNING: - CUPS printer is: "$messages"."
                exitstatus="$STATE_WARNING"
        ;;
	*printing*)
                OUTPUT="OK - CUPS printer is printing now."
                exitstatus="$STATE_OK"
        ;;
	*idle*)
                OUTPUT="OK - CUPS printer "$printername" is idle."
                exitstatus="$STATE_OK"
        ;;
        *)
                OUTPUT="CRITICAL - Unknown error occured while checking: "$RESULT"."
                exitstatus="$STATE_CRITICAL"
        ;;
esac

}

# It testing how many variable user add on command line
# The minimu for test printes only is 3

if [ $# -lt 3 ]; then
	print_usage
	exit "$STATE_UNKNOWN"
fi

# this set default exit status to:
exitstatus="$STATE_UNKNOWN"

# by default is test pritner disabled, you must allow it with -p $printer or -P all printers
testprinter="0"


# testing arguments:
while test -n "$1"; do
	case "$1" in
		--help)
			print_help
			exit "$STATE_OK"
			;;
		-h)
			print_help
			exit "$STATE_OK"
			;;
		-P)
                        testprinter="1"
                        ;;
		-p)
                        testprinter="2"
			printername="$2"
                        shift
                        ;;
		-H)
			hostname=$2
			shift
			;;
		-Q)
			testtype=$2
			shift
			;;
		-w)
			warnlevel=$2
			shift
			;;
		-c)	
			critlevel=$2
			shift
			;;
		-a)	
			maxage=$2
			shift
			;;
	esac
	shift
done

# Check arguments for validity:
if [ -z $hostname ]
then
	echo "You must specify a hostname (or localhost to test the local system)"
	print_usage
	exitstatus="$STATE_UNKNOWN"
	exit "$exitstatus"
fi

# testing priner(s) 


if [ $testprinter -eq 2 ] #Check specifics printer and continue with the script
then
        check_printer_status "$printername" 
	
	if [ -z $testtype ] # exits if there is no -Q checks
	then
		echo "$OUTPUT"
	        exit "$exitstatus"
	fi
elif	[ $testprinter -eq 1 ]; then # check all printers 
       	check_printer_status
	if [ -z $testtype ]; then # exits if there is no -Q checks
		echo "$OUTPUT"
       		exit "$exitstatus"
	fi
else # no cuos check is need 
	echo "No printer check is require. Checking the queue ..."
fi

# testing arguments for the queue checks:
if [[ -z $critlevel || -z $warnlevel ]] # Did we get warn and crit values?
then    
        echo "You must specify a warning and critical level"
	print_usage
        exitstatus="$STATE_UNKNOWN"
        exit "$exitstatus" 
elif [ $critlevel -lt $warnlevel ] # Do the warn/crit values make sense?
then
	echo "CRITICAL value of "$critlevel" is less than WARNING level of "$warnlevel""
	print_usage
	exitstatus="$STATE_UNKNOWN"
	exit "$exitstatus"
fi

# what kind of queue test will be run:
if [ -z $testtype ]
then

	echo "You must specify a test type"
	print_usage
	exitstatus="$STATE_UNKNOWN"
	exit "$exitstatus"
# this is a very nice elif, it match if -a X is missing
elif [[ "$testtype" = [b]* && -z $maxage ]] 
then
	echo "You must specify <max age> when using a test type of 'b'"
	print_usage
	exitstatus="$STATE_UNKNOWN"
	exit "$exitstatus"
else
	echo "Testing queue on the CUPS..."	
	JOBTMP=$(mktemp -t lpstat.XXXXXX) # Create a tmpfile to store the lpstat results
	STALEJOBCOUNT=0 # default number of old jobs
	CURDATETS=$(date +%s) # Get the current date as unixtime
	"$LPSTAT" -h "$hostname" -o > "$JOBTMP" # run the lpstat command against the host.

	if [ $? -ne 0 ]
	then
	    rm -rf "$JOBTMP"
		echo "UNKNOWN: lpstat command returned an error. Please test this script manually."
		exitstatus="$STATE_UNKNOWN"
		exit "$exitstatus"
	fi
	JOBCOUNT=$(wc -l < $JOBTMP) # populate the jobcount
	PERFDATA="print_jobs="$JOBCOUNT";"$warnlevel";"$critlevel";0"
	if [[ "$JOBCOUNT" -gt 0 && $maxage ]]
	then
		MAXAGETS=$(echo "86400 * $maxage" | bc) # 86400 seconds in a day * maxage
		exec<"$JOBTMP" # read the file to determine job age
		while read PRINTJOB
		do
			# Grab the job date from the job listing
			JOBDATE=$(echo $PRINTJOB | cut -c50-73) 
			# Convert the job date to unixtime
			JOBDATETS=$(date --date="$JOBDATE" +%s) 
			DATEDIFF=$(echo "($CURDATETS - $JOBDATETS)" | bc)
			if [ $DATEDIFF -gt $MAXAGETS ]
			then
				MESSAGE="CRITICAL: Some CUPS jobs are older than "$maxage" days| "$PERFDATA""
				exitstatus="$STATE_CRITICAL"
			else
				check_queue_size
			fi
		done
	else
		check_queue_size
	fi
	rm -rf "$JOBTMP"
fi


# end: print the results and end with exit code for Nagios
echo "$OUTPUT"
echo "$MESSAGE"
exit "$exitstatus"