#!/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_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
        /usr/sbin/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)
EOU
}

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

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
            ;;
        *)
            show_usage
            exit 1
            ;;
    esac
    shift
done

exit 0