#!/usr/bin/php
<?php

# Copyright (c) 2012-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';

if ($argc == 2 && strcmp($argv[1], "autoconf") == 0)
{
	print "yes\n";
	exit(0);
}

if ($argc == 2 && strcmp($argv[1], "config") == 0)
{
	echo "graph_title YK-VAL YubiKey stats\n";
	echo "graph_vlabel Known YubiKeys\n";
	echo "graph_category ykval\n";

	echo "yubikeys_enabled.label Enabled YubiKeys\n";
	echo "yubikeys_enabled.draw AREA\n";

	echo "yubikeys_disabled.label Disabled YubiKeys\n";
	echo "yubikeys_disabled.draw STACK\n";

	echo "yubikeys_1month.label YubiKeys seen last month\n";
	echo "yubikeys_1month.draw LINE2\n";

	echo "clients_enabled.label Enabled validation clients\n";
	echo "clients_enabled.draw LINE2\n";

	echo "clients_disabled.label Disabled validation clients\n";
	echo "clients_disabled.draw LINE2\n";

	exit(0);
}

$db = Db::GetDatabaseHandle($baseParams, 'ykval-munin-yubikeystats');
if (!$db->connect())
	logdie($myLog, 'ERROR Database connect error (1)');

function get_count($db, $table, $conditions)
{
	$res = $db->customQuery("SELECT count(1) as count FROM $table WHERE $conditions");

	if ($res)
	{
		$r = $res->fetch(PDO::FETCH_ASSOC);
		return $r['count'];
	}

	return Null;
}

if ($count = get_count($db, 'yubikeys', 'active=true'))
	echo "yubikeys_enabled.value $count\n";

if ($count = get_count($db, 'yubikeys', 'active=false'))
	echo "yubikeys_disabled.value $count\n";

if ($count = get_count($db, 'yubikeys', 'modified >= ' . (time() - (31 * 86400))))
	echo "yubikeys_1month.value $count\n";

if ($count = get_count($db, 'clients', 'active=true'))
	echo "clients_enabled.value $count\n";

if ($count = get_count($db, 'clients', 'active=false'))
	echo "clients_disabled.value $count\n";