shellscriptLib/notifyWhenUp.sh at master · rohegit/shellscriptLib · GitHub

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
#!/bin/sh
# =========================
# Notifies a given user via pushover (https://pushover.net) when a specified host goes online
# -------------------
# THIS SCRIPT DEPENDS ON checkHoststatus.sh
# You should call it by creating a cronjob
# =========================

# We do not want to notify if there is nothing to be notified of
TONOTIFY="0"
# Real uptime of host
UPTIME="0"
# How long has the host to be online before we send a notification (minutes)
UPTIMEBEFORENOTIFICATION="10"
# List of hosts which you like to watch. Separated by spaces
HOSTSTOWATCH=( host1 host2 host3 )
# File the status is logged to
HOSTSFILE="/www/hosts.xml"
# Pusover parameters
PUSHOVERTOKEN="abc123"
PUSHOVERUSER="user123"

# Has the host been online for long enough?
shouldNotificationBeSent() {
	HOST=$1
	TIMEONLINE=`xmlstarlet sel -t -v "//host[@id='$HOST']/status/uptime" -v . -n <$HOSTSFILE`
	ALREADYNOTIFIED=`xmlstarlet sel -t -v "//host[@id='$HOST']/status/notified" -v . -n <$HOSTSFILE`
	NOW=`date "+%Y-%m-%d %H:%M"`;
	TIMENOW=`date -d "$NOW" +%s`
	TIMEUPSEC=`expr $TIMENOW - $TIMEONLINE`;
	UPTIME=`expr $TIMEUPSEC / 60`

	# If uptime is higher than the set UPTIMEBEFORENOTIFICATION
	if [ "$UPTIME" -gt "$UPTIMEBEFORENOTIFICATION" ] && [ "$ALREADYNOTIFIED" -eq "0" ]; then
		TONOTIFY="1"
	fi

	echo $TONOTIFY
}

# Method to send a notification.
# For this to work you need to have pushover installed (see https://pushover.net)
sendNotification() {
	HOST=$1
	curl -s -F "token=$PUSHOVERTOKEN" -F "user=$PUSHOVERUSER" -F "message=$HOST has been  up for $UPTIME minutes" https://api.pushover.net/1/messages.json
}

# We need to know which notifications we have already sent
updateNotificationStatus() {
	HOST=$1
	TIMEONLINE=`xmlstarlet sel -t -v "//host[@id='$HOST']/status/uptime" -v . -n <$HOSTSFILE`

	# We sent a notification so set the status
	xmlstarlet ed -u "//host[@id='$HOST']/status/uptime" -v "$TIMEONLINE" "$HOSTSFILE"
	xmlstarlet ed -u "//host[@id='$HOST']/status/power" -v "1" "$HOSTSFILE"
	xmlstarlet ed -u "//host[@id='$HOST']/status/notified" -v "1" "$HOSTSFILE"
}

# Should a notification be sent
calcUptimeAndSendNotification() {
	HOSTS=($(xmlstarlet sel -t -m "//hosts/host/name" -v . -n < "$HOSTSFILE"))
	for HOST in "${HOSTS[@]}"; do
		SENDNOTIFICATION=`shouldNotificationBeSent $HOST`

		if [ "$SENDNOTIFICATION" -eq "1" ]; then
			sendNotification "$HOST"
			updateNotificationStatus "$HOST"
			echo "$HOST: Notification sent"
		fi
	done
}

calcUptimeAndSendNotification