#!/bin/bash

STALE_DIR="${HOME}/var/stale/"

get_names() {
    /usr/sbin/bcfg2-reports "$@" --fields=name | sed 's/ *$//'
}

get_stale() {
    get_names --stale | sed 's/ *$//'
}

get_all() {
    get_names -a | sed 's/ *$//'
}

get_bad() {
    get_names -d | sed 's/ *$//'
}

get_clean() {
    get_names -c | sed 's/ *$//'
}

get_count() {
    count=$(/usr/sbin/bcfg2-reports "$@" | wc -l)

    if [ "$count" = "0" ]; then
        echo 0
    else
        echo $((count - 1))
    fi
}

get_nagios() {
    if [ -z "$1" ]; then
        clean=$(get_clean | wc -l)
        bad=$(get_bad | wc -l)
        stale=$(get_stale | wc -l)

        status="OK"
        exit=0
        if [ "$bad" != "0" -o "$stale" != "0" ]; then
            status="CRITICAL"
            exit=2
        fi

        echo "$status | clean=$clean; bad=$bad; stale=$stale;"
        exit $exit
    else
        if ! get_names -a | grep -q "^$1$"; then
            echo "CRITICAL $1 not known to bcfg2"
            exit 2
        fi

        if get_names --stale | grep -q "^$1$"; then
            echo "CRITICAL $1 is stale"
            exit 2
        fi

        bad=$(get_count -b "$1")
        if [ "$bad" != "0" ]; then
            echo "CRITICAL $bad bad entries"
            exit 2
        fi

        extra=$(get_count -e "$1")
        total=$(bcfg2-reports -t "$1" | sed "s/good: [0-9]*)/extra: $extra)/")
        echo "OK $total"
    fi
}

get_stale_once() {
    get_stale | while read host; do
	STALE_FILE="${STALE_DIR}/${host}"
        test -f "${STALE_FILE}" || (
            touch "${STALE_FILE}"
            echo $host
        )
    done

    comm -13 <(get_stale | sort) <(get_all | sort) | while read host; do
        STALE_FILE="${STALE_DIR}/${host}"
        test -f "${STALE_FILE}" && rm -f "${STALE_FILE}"
    done
}

cron() {
    stale=$(get_stale_once)
    if [ -n "$stale" ]; then
        sendmail -t <<EOM
To: bcfg2-admins@spline.inf.fu-berlin.de
From: cron@bcfg2.spline.inf.fu-berlin.de
Subject: New stale hosts

Hallo,

die folgenden Hosts haben sich seit 24h nicht mehr bei bcfg2 gemeldet:

$(echo "$stale" | sed 's/^/    /')

Gruß,
bcfg2
EOM
    fi
}

show_usage() {
     cat <<EOU
Usage:
	$0 [mode]

mode should be one of the following:
  -a|--all	display all hosts
  -b|--bad	display bad hosts
  -s|--stale	display hosts, that haven't run in the last 24 hours
  -e|--extra	display hosts with extra entries

  -c|--cron	run in cron mode (report new stale hosts)
  -n|--nagios	run in nagios mode (report count hosts in groups)
EOU
}

ARGS="$(getopt -o 'absecn::' --long 'all,bad,stale,extra,cron,nagios::' -- "$@")"

if [ $? -ne 0 ]; then
    show_usage
    exit 1
fi

eval set -- "$ARGS"

while true; do
    case "$1" in
        -a|--all)
            get_all
            exit 0
            ;;
        -b|--bad)
            get_bad
            exit 0
            ;;
        -s|--stale)
            get_stale
            exit 0
            ;;
        -e|--extra)
            get_extra
            exit 0
            ;;

        -c|--cron)
            cron
            exit 0
            ;;
        -n|--nagios)
            get_nagios "$2"
            exit 0
            ;;
        *)
            show_usage
            exit 1
            ;;
    esac
    shift
done

exit 0