#!/bin/bash
#
# Nagios plugin to check Postgresql streamin replication state
#
# Could be use on Master or on standby node
#
# Requirement :
#
#     On master node : Slaves must be able to connect with user PG_USER
#                      to database postgres as trust
#
#     On standby node : PG_USER must be able to connect localy as trust
#
# Author : Benjamin Renard <brenard@easter-eggs.com>
# Date : Wed, 14 Mar 2012 14:45:55 +0000
# Source : http://git.zionetrix.net/check_pg_streaming_replication
#

GIT_ROOT=/srv/common

if [ "$1" == "-h" ]
then
	echo "Usage : $0 [directory] [-d]
	[directory]	Git root directory (default : $GIT_ROOT)
	[-d]		Enable debug mode"
	exit 0
fi

[ -n "$1" -a "$1" != "-d" ] && GIT_ROOT="$1"
[ ! -d "$GIT_ROOT" ] && echo "UNKNOWN : Git root directory does not exists !" && exit 3
[ ! -d "$GIT_ROOT/.git" ] && echo "UNKNOWN : Git root directory seem to not being a git repository." && exit 3

cd $GIT_ROOT

DEBUG=0
[ "$1" == "-d" -o "$2" == "-d" ] && DEBUG=1

STATUS=$( git status -s )

[ $DEBUG -eq 1 ] && echo -e "Status : $STATUS"

if [ -n "$STATUS" ]
then
	echo "WARNING : Git config repo on $( hostname ) not clean"
	exit 1
else
	[ $DEBUG -eq 1 ] && echo -n "Fecth : "
	git fetch > /dev/null 2>&1
	res=$?
	[ $DEBUG -eq 1 ] && echo "done. (Return $?)"

	if [ $res -ne 0 ]
	then
		echo "UNKNOWN : Error fetching remote"
		exit 3
	fi

	HEAD="$( git show HEAD|grep ^commit )"
	[ $DEBUG -eq 1 ] && echo "Local : $HEAD"

	ORIGIN="$( git show origin|grep ^commit )"
	[ $DEBUG -eq 1 ] && echo "Remote : $ORIGIN"
	
	if [ "$HEAD" != "$ORIGIN" ]
	then
		echo "CRITICAL : Git config not uptodate"
		exit 2
	else
		echo "OK"
		exit 0
	fi
fi