#!/bin/bash

HOURS=12

CHRBIN=/usr/sbin/chroot
CHRDIR=/chroot/admin


if [ -z $SDDIR ]; then
    SDDIR=/etc/statusd
fi

if [ -z $SPOOLDIR ]; then
    SPOOLDIR=/var/spool/statusd
fi

STF=$SPOOLDIR/policyd.status


function sqlget() {
	if [ -S $CHRDIR/run/mysqld/mysqld.sock ]; then
		echo "SELECT IFNULL(GROUP_CONCAT(qt.TrackKey SEPARATOR ', '),'')
			FROM quotas_tracking qt LEFT JOIN quotas_limits ql ON qt.QuotasLimitsID=ql.id
			WHERE qt.Counter>(ql.CounterLimit*0.8)
			AND from_unixtime(qt.LastUpdate) > DATE_ADD(NOW(), INTERVAL -${HOURS} HOUR);" | \
		$CHRBIN $CHRDIR mysql \
		--defaults-file=/etc/mysql/debian.cnf \
		--socket=/run/mysqld/mysqld.sock \
		-ss -n sys_cluebringer
	fi
}

function quotacheck() {

	EXCEED=`sqlget`

	if [ "$EXCEED" != "" ]; then
		echo "$EXCEED"
	else
		echo "OK"
	fi
}

if [ "$1" == "cron" ]; then
	RES=`quotacheck`
	echo "$RES" >$STF
else
	if [ -f $STF ]; then
		RES=`cat $STF`
		echo "$RES"
		if [ "$RES" != "OK" ]; then
			exit 1
		else
			exit 0
		fi
	else
		echo "OK"
		exit 0
	fi
fi