<?php

require_once dirname(__FILE__) . '/../ds-common.php';

class WeatherMapDataSource_dsstats extends WeatherMapDataSource
{
    function Init(&$map)
    {
        global $config;

        if ($map->context === 'cacti') {
            if (false === function_exists('db_fetch_row')) {
                wm_debug(
                    "ReadData DSStats: Cacti database library not found. [DSSTATS001]\n");
                return (false);
            }

            if (true === function_exists('api_plugin_is_enabled')) {
                if (false === api_plugin_is_enabled('dsstats')) {
                    wm_debug(
                        "ReadData DSStats: DSStats plugin not enabled (new-style). [DSSTATS002B]\n");
                    return (false);
                }
            } else {
                if ( (false === isset($plugins)) || (false === in_array('dsstats', $plugins))) {
                    wm_debug(
                        "ReadData DSStats: DSStats plugin not enabled (old-style). [DSSTATS002A]\n");
                    return (false);
                }
            }

            $sql = 'show tables';
            $result = db_fetch_assoc($sql) or die(mysql_error());
            $tables = array ();

            foreach ($result as $index => $arr) {
                foreach ($arr as $t) {
                    $tables[] = $t;
                }
            }

            if (false === in_array('data_source_stats_hourly_last', $tables)) {
                wm_debug(
                    'ReadData DSStats: data_source_stats_hourly_last database table not found. [DSSTATS003]\n');
                return (false);
            }

            return (true);
        }

        return (false);
    }

    function Recognise($targetstring)
    {
        if (1 === preg_match('/^dsstats:([a-z]+):(\d+):([\-a-zA-Z0-9_]+):([\-a-zA-Z0-9_]+)$/',
            $targetstring, $matches)) {
            return true;
        } elseif (1 === preg_match('/^dsstats:(\d+):([\-a-zA-Z0-9_]+):([\-a-zA-Z0-9_]+)$/',
            $targetstring, $matches)) {
            return true;
        } else {
            return false;
        }
    }


    // Actually read data from a data source, and return it
    // returns a 3-part array (invalue, outvalue and datavalid time_t)
    // invalue and outvalue should be -1,-1 if there is no valid data
    // data_time is intended to allow more informed graphing in the future
    function ReadData($targetstring, &$map, &$item)
    {
        global $config;

        $dsnames[IN] = 'traffic_in';
        $dsnames[OUT] = 'traffic_out';
        $data[IN] = null;
        $data[OUT] = null;

        $inbw = null;
        $outbw = null;
        $data_time = 0;

        $table = '';
        $keyfield = 'rrd_name';
        $datatype = '';
        $field = '';

        if (1 === preg_match('/^dsstats:(\d+):([\-a-zA-Z0-9_]+):([\-a-zA-Z0-9_]+)$/',
            $targetstring, $matches)) {
            $local_data_id = $matches[1];
            $dsnames[IN] = $matches[2];
            $dsnames[OUT] = $matches[3];

            $datatype = 'last';

            if ($map->get_hint('dsstats_default_type') !== null) {
                $datatype = $map->get_hint('dsstats_default_type');
                wm_debug("Default datatype changed to %s.\n", $datatype);
            }
        } elseif (preg_match(
            '/^dsstats:([a-z]+):(\d+):([\-a-zA-Z0-9_]+):([\-a-zA-Z0-9_]+)$/',
            $targetstring, $matches)) {
            $dsnames[IN] = $matches[3];
            $dsnames[OUT] = $matches[4];
            $datatype = $matches[1];
            $local_data_id = $matches[2];
        }

        if (substr($datatype, 0, 5) === 'daily')
        {            $table = 'data_source_stats_daily';
        }
        if (substr($datatype, 0, 6) === 'weekly')
        {   $table = 'data_source_stats_weekly';
        }
        if (substr($datatype, 0, 7) === 'monthly')
        {   $table = 'data_source_stats_monthly';
        }
        if (substr($datatype, 0, 6) === 'hourly')
        {   $table = 'data_source_stats_hourly';
        }
        if (substr($datatype, 0, 6) === 'yearly')
        {   $table = 'data_source_stats_yearly';
        }
        if (substr($datatype, -7) === 'average')
        {    $field = 'average';
        }
        if (substr($datatype, -4) === 'peak') {
            $field = 'peak';
        }

        if ($datatype === 'last') {
            $field = 'calculated';
            $table = 'data_source_stats_hourly_last';
        }

        if ($datatype === 'wm') {
            $field = 'last_calc';
            $table = 'weathermap_data';
            $keyfield = 'data_source_name';
        }

        if ($table !== '' and $field !== '') {
            $SQL =
                sprintf(
                    "select %s as name, %s as result from %s where local_data_id=%d and (%s='%s' or %s='%s')",
                    $keyfield, $field, $table, $local_data_id, $keyfield,
                    mysql_escape_string($dsnames[IN]), $keyfield,
                    mysql_escape_string($dsnames[OUT]));

            $results = db_fetch_assoc($SQL);

            if (count($results) > 0) {
                foreach ($results as $result) {
                    foreach (array (
                        IN,
                        OUT
                    ) as $dir) {
                        if (($dsnames[$dir] === $result['name'])
                            && ($result['result'] !== '-90909090909')
                                && ($result['result'] !== 'U')) {
                            $data[$dir] = $result['result'];
                        }
                    }
                }
            }

            if ($datatype === 'wm' && ($data[IN] === null || $data[OUT] === null)) {
                wm_debug("Didn't get data for 'wm' source. Inserting new tasks.");
// insert the required details into weathermap_data, so it will be picked up next time
                $SQL =
                    sprintf(
                        'select data_template_data.data_source_path as path from data_template_data,data_template_rrd where data_template_data.local_data_id=data_template_rrd.local_data_id and data_template_rrd.local_data_id=%d',
                        $local_data_id);
                $result = db_fetch_row($SQL);

                if (count($result) > 0) {
                    $db_rrdname = $result['path'];
                    wm_debug("Filename is %s\n",$db_rrdname);

                    foreach (array (
                        IN,
                        OUT
                    ) as $dir) {
                        if ($data[$dir] === null) {
                            $SQLins =
                                "insert into weathermap_data (rrdfile, data_source_name, sequence, local_data_id) values ('"
                                . mysql_real_escape_string($db_rrdname) . "','"
                                . mysql_real_escape_string($dsnames[$dir]) . "', 0,"
                                . $local_data_id . ')';

                            db_execute($SQLins);
                        }
                    }
                } else {
                    wm_warn(
                        "DSStats ReadData: Failed to find a filename for DS id %s [WMDSTATS01]\n", $local_data_id);
                }
            }
        }
	
		// fill all that other information (ifSpeed, etc)
		if( ($map->get_hint("dsstats_no_cacti_extras") === null) && ($local_data_id > 0) ) {
            UpdateCactiData($item, $local_data_id);
        }

        wm_debug( "DSStats ReadData: Returning (%s, %s, %s)\n",
		        string_or_null($data[IN]),
		        string_or_null($data[OUT]),
		        $data_time
        	);

        return (array (
            $data[IN],
            $data[OUT],
            $data_time
        ));
    }
}

// vim:ts=4:sw=4:
?>