#!/usr/bin/php
<?php

/*
 * README
 * This plugin makes use of the GameQ library to be able to query many different
 * types of game servers and display the results in munin. You can see a list of
 * supported games at http://gameq.sourceforge.net/#games
 *
 * This plugin has 2 environment variables:
 * game_config_file - Defaults to /etc/munin/munin-game.ini. The location of the config file
 * game_state_file - Defaults to /var/lib/munin/plugin-state/game. The location of the munin state directory
 *
 * (Probably) FAQ
 * Q: Why PHP?
 * A: The GameQ library is written in PHP, so it was either that or rewrite the entire library.
 *
 * Q: Why have you used a config file instead of environment variables?
 * A: Way too much to fit data inside an environment variable unfortunately. Imagine
 *    trying to fit information about 50 servers all inside one variable.
 *
 * Q: I want to ask you a question!
 * A: I'm on most IRC networks under the nick Azelphur, or email me. support@azelphur.com
 *
 * Q: It's not working!
 * A: Try running ./game autoconf, it'll probably tell you what's up.
 */

// Check environment variables
$var = getenv('game_config_file');
$config = ($var) ? $var : "/etc/munin/munin-game.ini";
$var = getenv('game_state_file');
$state = ($var) ? $var : (getenv('MUNIN_PLUGSTATE') . "/game");

function p($str) {
    // Quick function to print a string with an EOL on the end
    echo $str . PHP_EOL;
}

function printMultigraph($ini_array, $machine_name, $title, $info, $max) {
    // Print out a standard graph config.
    p("multigraph $machine_name");
    p("graph_title $title");
    p("graph_vlabel players");
    p("graph_category games");
    p("graph_info $info");
    p("graph_printf %6.0lf");

    if (isset($ini_array['settings'][$machine_name . '_colour']))
        p("players.colour " . $ini_array['settings'][$machine_name . '_colour']);

    if (isset($ini_array['settings'][$machine_name . '_draw']))
        p("players.draw " . $ini_array['settings'][$machine_name . '_draw']);

    p("players.label players");
    p("players.info Number of players");
    p("players.min 0");
    p("players.max $max");
}

function printValue($machine_name, $value) {
    // Print a value
    p("multigraph $machine_name");
    p("players.value " . $value);
}

function queryServers($ini_array) {
    // Query all game servers and return the results

    require "gameq/GameQ.php";

    // Populate the $servers array from the ini file, ready to pass to GameQ
    $servers = array();
    foreach ($ini_array as $section => $value) {
        if ($section != 'settings')
            $servers[$section] = array($value['game'], $value['address'], $value['port']);
    }

    // Create a new GameQ object and pass it a list of servers
    $gq = new GameQ();
    $gq->addServers($servers);

    // Set timeout from the config file
    $gq->setOption('timeout', $ini_array['settings']['timeout']);
    $gq->setOption('sock_count', $ini_array['settings']['sock_count']);
    $gq->setOption('sock_start', $ini_array['settings']['sock_start']);

    // This filter makes sure a subset of data is always available, next to the normal data
    $gq->setFilter('normalise');

    // Send request(s)
    $results = $gq->requestData();

    return $results;
}

// Parse command line arguments if required.
if (isset($_SERVER['argv'][1])) {
    if ($_SERVER['argv'][1] == 'config') {
        // Load our config.ini
        $ini_array = parse_ini_file($config, true);

        // Load games.ini so we can show pretty game names
        $games = parse_ini_file('gameq/GameQ/games.ini', true);

        // Query the game servers
        $results = queryServers($ini_array);
        // Cache the query in the state file
        $fp = fopen($state, 'w+') or die("I could not open state file.");
        fwrite($fp, serialize($results));
        fclose($fp);


        // Loop through each server, printing graphs.
        foreach ($results as $name => $server) {
            // If sub graphs and the game total graphs are enabled, make this be a sub-graph.
            if ($ini_array['settings']['show_game_total'] && $ini_array['settings']['enable_sub_graphs'])
                $machine_name = "gameserver_" . $ini_array[$name]['game'] . ".$name";
            else
                $machine_name = "gameserver_" . $ini_array[$name]['game'] . "_$name";
            $title = $ini_array[$name]['name'] . " players";
            $info = "The number of players connected to the " . $games[$ini_array[$name]['game']]['name'] . " server";
            printMultigraph($ini_array, $machine_name, $title, $info, $server['gq_maxplayers']);
        }

        // Game total graphs.
        if ($ini_array['settings']['show_game_total']) {
            $game_total_max = array();

            // Count players connected to each game
            foreach ($results as $name => $server) {
                if (!isset($game_total_max[$ini_array[$name]['game']]))
                    $game_total_max[$ini_array[$name]['game']] = $server['gq_maxplayers'];
                else
                    $game_total_max[$ini_array[$name]['game']] += $server['gq_maxplayers'];
            }

            // Print all the game total graphs.
            foreach ($game_total_max as $game => $total)
            {
                $machine_name = "gameserver_" . $game;
                $title = "Total " . $games[$game]['name'] . " players";
                $info = "Total players connected to all " . $games[$game]['name'] . " servers";
                printMultigraph($ini_array, $machine_name, $title, $info, $total);
            }
        }

        // Global total graph
        if ($ini_array['settings']['show_global_total']) {
            $total_max = 0;
            // Add up all the players connected to all the servers
            foreach ($results as $name => $server) {
                $total_max += $server['gq_maxplayers'];
            }

            printMultigraph($ini_array, "gameserver", "Total Players", "Total players connected to all servers", $total_max);
        }
    }
    if ($_SERVER['argv'][1] == 'autoconf') {
        if (!@include("gameq/GameQ.php"))
            p("no (GameQ Library not found)");
        elseif (!file_exists($config))
            p("no ($config not found)");
        else
            p("yes");
    }
    die();
}

// No command line arguments, print values.

// Load our config.ini
$ini_array = parse_ini_file($config, true);

// Load games.ini so we can show pretty game names
$games = parse_ini_file('gameq/GameQ/games.ini', true);

$results = unserialize(file_get_contents($state));

// Print individual game values
foreach ($results as $name => $server){
    if ($ini_array['settings']['show_game_total'] && $ini_array['settings']['enable_sub_graphs'])
        $machine_name = "gameserver_" . $ini_array[$name]['game'] . ".$name";
    else
        $machine_name = "gameserver_" . $ini_array[$name]['game'] . "_$name";
    printValue($machine_name, $server['gq_numplayers']);
}

// Print game total values
if ($ini_array['settings']['show_game_total']) {
    $game_total = array();
    foreach ($results as $name => $server) {
        // Add up counts for the total graph
        if (!isset($game_total_max[$ini_array[$name]['game']]))
            $game_total[$ini_array[$name]['game']] = $server['gq_numplayers'];
        else
            $game_total[$ini_array[$name]['game']] += $server['gq_numplayers'];
    }
    foreach ($game_total as $game => $total)
    {
        $machine_name = "gameserver_" . $game;
        printValue($machine_name, $total);
    }
}

// Are global totals enabled?
if ($ini_array['settings']['show_global_total']) {
    $total = 0;
    foreach ($results as $name => $server) {
        $total += $server['gq_numplayers'];
    }
    printValue("gameserver", $total);
}


?>