#!/usr/bin/php
<?php

# Copyright (c) 2010-2015 Yubico AB
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
#
#   * Redistributions of source code must retain the above copyright
#     notice, this list of conditions and the following disclaimer.
#
#   * Redistributions in binary form must reproduce the above
#     copyright notice, this list of conditions and the following
#     disclaimer in the documentation and/or other materials provided
#     with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

set_include_path(implode(PATH_SEPARATOR, array(
	get_include_path(),
	'/usr/share/yubikey-val',
	'/etc/yubico/val',
)));

require_once 'ykval-config.php';
require_once 'ykval-db.php';
require_once 'ykval-common.php';

function get_db_counters()
{
	$db = Db::GetDatabaseHandle($baseParams, 'ykval-munin-queuelength');

	if (!$db->connect())
		return false;

	$result = $db->customQuery('
				SELECT server,
				COUNT(server) as count
				FROM queue
				GROUP BY server');

	if ($result)
		$result = $result->fetchAll(PDO::FETCH_ASSOC);
	else
		return false;

	$counters = array();

	foreach ($result as $row)
	{
		$counters[$row['server']] = $row['count'];
	}

	return $counters;
}

$urls = $baseParams['__YKVAL_SYNC_POOL__'];

if ($argc == 2 && strcmp($argv[1], 'autoconf') == 0)
{
	if (is_array($urls) && count($urls) > 0)
	{
		echo "yes\n";
		exit(0);
	}

	echo "no (sync pool not configured)\n";
	exit(0);
}

if (($endpoints = endpoints($urls)) === FALSE)
{
	echo "Cannot parse URLs from sync pool list\n";
	exit(1);
}

if ($argc == 2 && strcmp($argv[1], 'config') == 0)
{
	echo "graph_title YK-VAL queue size\n";
	echo "graph_vlabel sync requests in queue\n";
	echo "graph_category ykval\n";

	foreach ($endpoints as $endpoint)
	{
		list ($internal, $label, $url) = $endpoint;

		echo "${internal}_queuelength.label sync ${label}\n";
		echo "${internal}_queuelength.draw AREASTACK\n";
		echo "${internal}_queuelength.type GAUGE\n";
	}

	exit(0);
}

if (($counters = get_db_counters()) === FALSE)
{
	echo "Error fetching data from database\n";
	exit(1);
}

foreach ($endpoints as $endpoint)
{
	list ($internal, $label, $url) = $endpoint;

	$count = 0;

	if (array_key_exists($url, $counters))
		$count = $counters[$url];

	echo "${internal}_queuelength.value $count\n";
}

exit(0);