aboutsummaryrefslogtreecommitdiff
path: root/nscaw
blob: 591b171a8d2320188510bc8d933bab520562e043 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#!/usr/bin/env bash

#
# Command wrapper that sends an appropriate passive check to an NSCA server
# using send_nsca, depending on the command's outcome.
#
# Make sure send_nsca is in $PATH and that NSCAW_SERVER is set.
#
#   $ nscaw IMPORTANT_JOB -- important-job -o options args ...
#
# Author: Tom Ryder <tom@sanctum.geek.nz>
# License: MIT
#

# Name ourself
self=nscaw

# If there's a defaults file with environment variables for us, source it
if [[ -r /etc/default/"$self" ]] ; then
    source /etc/default/"$self"
fi

# Define a function to explain how to use this script
usage() {
    printf 'Usage: %s SERVICE -- COMMAND...\n' \
        "$self"
}

# Respond to requests for help with usage (exit success)
case $1 in
    -h|--help)
        usage
        exit 0
        ;;
esac

# Check that at least three arguments are present and in the correct form
if (( $# < 3 )) || [[ $2 != -- ]] ; then
    usage >&2
    exit 1
fi

# Pull the service from the first argument and shift the first two arguments
# off the arguments array; the rest of the array should be the command to run
service=$1
shift 2

# Figure out our hostname; most of the time `hostname -s` will be fine, but
# it can be overridden with the value of NSCAW_HOSTNAME if defined
hostname=${NSCAW_HOSTNAME:-$(hostname -s)}

# Attempt to run command within a time wrapper
# <http://mywiki.wooledge.org/BashFAQ/032>
exec 3>&1 4>&2
time=$( { time "$@" 1>&3 2>&4 ; } 2>&1) ; ret=$?
exec 3>&- 4>&-

# Munge the time to make it palatable as a single line
time=${time#$'\n'}      # Strip leading newline
time=${time//$'\n'/, }  # Change all newlines to comma-space
time=${time//$'\t'/ }   # Change all tabs to spaces

# Decide return code and message based on command exit value
case $ret in
    0)
        code=0 # OK
        message=$(printf '%s: `%s` ran with no errors as user %s: %s' \
            "$self" "$*" "$USER" "$time")
        ;;
    127)
        code=3 # UNKNOWN
        message=$(printf '%s: `%s` could not be found as user %s with PATH=%s' \
            "$self" "$1" "$USER" "$PATH")
        ;;
    *)
        code=2 # CRITICAL
        message=$(printf '%s: `%s` ran with errors as user %s: %s' \
            "$self" "$*" "$USER" "$time")
        ;;
esac
printf '%s\n' "$message"

# Format the passive check and pipe it into send_nsca; note that we ignore the
# stdout of send_nsca as it's just a diagnostic message
printf '%s\t%s\t%u\t%s\n' \
    "$hostname" "$service" "$code" "$message" |
        send_nsca -H "${NSCAW_SERVER:?}" >/dev/null