#!/usr/bin/php -q
<?php
/*
 +-------------------------------------------------------------------------+
 | Copyright (C) 2004-2016 The Cacti Group                                 |
 |                                                                         |
 | This program is free software; you can redistribute it and/or           |
 | modify it under the terms of the GNU General Public License             |
 | as published by the Free Software Foundation; either version 2          |
 | of the License, or (at your option) any later version.                  |
 |                                                                         |
 | This program is distributed in the hope that it will be useful,         |
 | but WITHOUT ANY WARRANTY; without even the implied warranty of          |
 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the           |
 | GNU General Public License for more details.                            |
 +-------------------------------------------------------------------------+
 | Cacti: The Complete RRDTool-based Graphing Solution                     |
 +-------------------------------------------------------------------------+
 | This code is designed, written, and maintained by the Cacti Group. See  |
 | about.php and/or the AUTHORS file for specific developer information.   |
 +-------------------------------------------------------------------------+
 | http://www.cacti.net/                                                   |
 +-------------------------------------------------------------------------+
*/

/* do NOT run this script through a web browser */
if (!isset($_SERVER['argv'][0]) || isset($_SERVER['REQUEST_METHOD'])  || isset($_SERVER['REMOTE_ADDR'])) {
	die('<br><strong>This script is only meant to run at the command line.</strong>');
}

$no_http_headers = true;

include(dirname(__FILE__) . '/../include/global.php');
include_once($config['base_path'] . '/lib/api_automation_tools.php');
include_once($config['base_path'] . '/lib/data_query.php');
include_once($config['base_path'] . '/lib/utility.php');
include_once($config['base_path'] . '/lib/sort.php');
include_once($config['base_path'] . '/lib/template.php');
include_once($config['base_path'] . '/lib/api_data_source.php');
include_once($config['base_path'] . '/lib/api_graph.php');
include_once($config['base_path'] . '/lib/snmp.php');
include_once($config['base_path'] . '/lib/data_query.php');
include_once($config['base_path'] . '/lib/api_device.php');

/* process calling arguments */
$parms = $_SERVER['argv'];
array_shift($parms);

if (sizeof($parms)) {
	/* setup defaults */
	$graph_type    = '';
	$templateGraph = array();
	$dsGraph       = array();
	$dsGraph['snmpFieldSpec']  = '';
	$dsGraph['snmpQueryId']    = '';
	$dsGraph['snmpQueryType']  = '';
	$dsGraph['snmpField']      = array();
	$dsGraph['snmpValue']      = array();
	$dsGraph['snmpValueRegex'] = array();
	$dsGraph['reindex_method'] = DATA_QUERY_AUTOINDEX_BACKWARDS_UPTIME;

	$input_fields  = array();
	$values['cg']  = array();

	$hosts          = getHosts();
	$graphTemplates = getGraphTemplates();

	$graphTitle = '';
	$cgInputFields = '';

	$host_id     	= 0;
	$template_id 	= 0;
	$hostTemplateId = 0;
	$force      	= 0;

	$listHosts       		= FALSE;
	$listGraphTemplates 	= FALSE;
	$listSNMPFields  		= FALSE;
	$listSNMPValues  		= FALSE;
	$listQueryTypes  		= FALSE;
	$listSNMPQueries 		= FALSE;
	$listInputFields 		= FALSE;

	$quietMode       = FALSE;

	foreach($parms as $parameter) {
		if (strpos($parameter, '=')) {
			list($arg, $value) = explode('=', $parameter);
		} else {
			$arg = $parameter;
			$value = '';
		}

		switch($arg) {
		case '--graph-type':
			$graph_type = $value;

			break;
		case '--graph-title':
			$graphTitle = $value;

			break;
		case '--graph-template-id':
			$template_id = $value;

			break;
		case '--host-template-id':
			$hostTemplateId = $value;

			break;
		case '--host-id':
			$host_id = $value;

			break;
		case '--input-fields':
			$cgInputFields = $value;

			break;
		case '--snmp-query-id':
			$dsGraph['snmpQueryId'] = $value;

			break;
		case '--snmp-query-type-id':
			$dsGraph['snmpQueryType'] = $value;

			break;
		case '--snmp-field':
			$dsGraph['snmpField'][] = $value;

			break;
		case '--snmp-value-regex':
			$dsGraph['snmpValueRegex'][] = $value;

			break;
		case '--snmp-value':
			$dsGraph['snmpValue'][] = $value;

			break;
		case '--reindex-method':
			if (is_numeric($value) &&
				($value >= DATA_QUERY_AUTOINDEX_NONE) &&
				($value <= DATA_QUERY_AUTOINDEX_FIELD_VERIFICATION)) {
				$dsGraph['reindex_method'] = $value;
			} else {
				switch (strtolower($value)) {
					case 'none':
						$dsGraph['reindex_method'] = DATA_QUERY_AUTOINDEX_NONE;
						break;
					case 'uptime':
						$dsGraph['reindex_method'] = DATA_QUERY_AUTOINDEX_BACKWARDS_UPTIME;
						break;
					case 'index':
						$dsGraph['reindex_method'] = DATA_QUERY_AUTOINDEX_INDEX_NUM_CHANGE;
						break;
					case 'fields':
						$dsGraph['reindex_method'] = DATA_QUERY_AUTOINDEX_FIELD_VERIFICATION;
						break;
					default:
						echo "ERROR: You must supply a valid reindex method for this graph!\n";
						exit(1);
				}
			}

			break;
		case '--list-hosts':
			$listHosts = TRUE;

			break;
		case '--list-snmp-fields':
			$listSNMPFields = TRUE;

			break;
		case '--list-snmp-values':
			$listSNMPValues = TRUE;

			break;
		case '--list-query-types':
			$listQueryTypes = TRUE;

			break;
		case '--list-snmp-queries':
			$listSNMPQueries = TRUE;

			break;
		case '--force':
			$force = TRUE;

			break;
		case '--quiet':
			$quietMode = TRUE;

			break;
		case '--list-input-fields':
			$listInputFields = TRUE;

			break;
		case '--list-graph-templates':
			$listGraphTemplates = TRUE;

			break;
		case '--version':
		case '-V':
		case '-v':
			display_version();
			exit;
		case '--help':
		case '-H':
		case '-h':
			display_help();
			exit(0);
		default:
			echo "ERROR: Invalid Argument: ($arg)\n\n";
			display_help();
			exit(1);
		}
	}

	if ($listGraphTemplates) {
		/* is a Host Template Id is given, print the related Graph Templates */
		if ($hostTemplateId > 0) {
			$graphTemplates = getGraphTemplatesByHostTemplate($hostTemplateId);
			if (!sizeof($graphTemplates)) {
				echo "ERROR: You must supply a valid --host-template-id before you can list its graph templates\n";
				echo "Try --list-graph-template-id --host-template-id=[ID]\n";
				exit(1);
			}
		}

		displayGraphTemplates($graphTemplates, $quietMode);

		exit(0);
	}


	if ($listInputFields) {
		if ($template_id > 0) {
			$input_fields = getInputFields($template_id, $quietMode);
			displayInputFields($input_fields, $quietMode);
		} else {
			echo "ERROR: You must supply an graph-template-id before you can list its input fields\n";
			echo "Try --graph-template-id=[ID] --list-input-fields\n";
			exit(1);
		}

		exit(0);
	}

	if ($listHosts) {
		displayHosts($hosts, $quietMode);
		exit(0);
	}

	/* get the existing snmp queries */
	$snmpQueries = getSNMPQueries();

	if ($listSNMPQueries) {
		displaySNMPQueries($snmpQueries, $quietMode);
		exit(0);
	}

	/* Some sanity checking... */
	if ($dsGraph['snmpQueryId'] != '') {
		if (!isset($snmpQueries[$dsGraph['snmpQueryId']])) {
			echo 'ERROR: Unknown snmp-query-id (' . $dsGraph['snmpQueryId'] . ")\n";
			echo "Try --list-snmp-queries\n";
			exit(1);
		}

		/* get the snmp query types for comparison */
		$snmp_query_types = getSNMPQueryTypes($dsGraph['snmpQueryId']);

		if ($listQueryTypes) {
			displayQueryTypes($snmp_query_types, $quietMode);
			exit(0);
		}

		if ($dsGraph['snmpQueryType'] != '') {
			if (!isset($snmp_query_types[$dsGraph['snmpQueryType']])) {
				echo 'ERROR: Unknown snmp-query-type-id (' . $dsGraph['snmpQueryType'] . ")\n";
				echo 'Try --snmp-query-id=' . $dsGraph['snmpQueryId'] . " --list-query-types\n";
				exit(1);
			}
		}

		if (!($listHosts ||			# you really want to create a new graph
			$listSNMPFields || 		# add this check to avoid reindexing on any list option
			$listSNMPValues ||
			$listQueryTypes ||
			$listSNMPQueries ||
			$listInputFields)) {

			/* if data query is not yet associated,
			 * add it and run it once to get the cache filled */

			/* is this data query already associated (independent of the reindex method)? */
			$exists_already = db_fetch_cell("SELECT COUNT(host_id) FROM host_snmp_query WHERE host_id=$host_id AND snmp_query_id=" . $dsGraph['snmpQueryId']);
			if ((isset($exists_already)) &&
				($exists_already > 0)) {
				/* yes: do nothing, everything's fine */
			}else{
				db_execute('REPLACE INTO host_snmp_query 
					(host_id,snmp_query_id,reindex_method) 
					VALUES ('. 
						$host_id                    . ',' . 
						$dsGraph['snmpQueryId']    . ',' . 
						$dsGraph['reindex_method'] .  ')');

				/* recache snmp data, this is time consuming,
				 * but should happen only once even if multiple graphs
				 * are added for the same data query
				 * because we checked above, if dq was already associated */
				run_data_query($host_id, $dsGraph['snmpQueryId']);
			}
		}
	}

	/* Verify the host's existance */
	if (!isset($hosts[$host_id]) || $host_id == 0) {
		echo "ERROR: Unknown Host ID ($host_id)\n";
		echo "Try --list-hosts\n";
		exit(1);
	}

	/* process the snmp fields */
	$snmpFields = getSNMPFields($host_id, $dsGraph['snmpQueryId']);

	if ($listSNMPFields) {
		displaySNMPFields($snmpFields, $host_id, $quietMode);
		exit(0);
	}

	$snmpValues = array();

	/* More sanity checking */
	/* Testing SnmpValues and snmpFields args */
	if ($dsGraph['snmpValue'] and $dsGraph['snmpValueRegex'] ) {
		echo "ERROR: You can't supply --snmp-value and --snmp-value-regex at the same time\n";
		exit(1);
	}

	$nbSnmpFields      = sizeof($dsGraph['snmpField']);
	$nbSnmpValues      = sizeof($dsGraph['snmpValue']);
	$nbSnmpValuesRegex = sizeof($dsGraph['snmpValueRegex']);

	if ($nbSnmpValues) {
		if ($nbSnmpFields != $nbSnmpValues) {
			echo "ERROR: number of --snmp-field and --snmp-value does not match\n";
			exit(1);
		}
	} elseif ($nbSnmpValuesRegex) {
		if ($nbSnmpFields != $nbSnmpValuesRegex) {
			echo "ERROR: number of --snmp-field ($nbSnmpFields) and --snmp-value-regex ($nbSnmpValuesRegex) does not match\n";
			exit(1);
		}
	} else {
		echo "ERROR: You must supply a --snmp-value or --snmp-value-regex option with --snmp-field\n";
		exit(1);
	}

	$index_filter = 0;
	foreach($dsGraph['snmpField'] as $snmpField) {
		if ($snmpField != '') {
			if (!isset($snmpFields[$snmpField] )) {
				echo 'ERROR: Unknown snmp-field ' . $dsGraph['snmpField'] . " for host $host_id\n";
				echo "Try --list-snmp-fields\n";
				exit(1);
			}
		}

		$snmpValues = getSNMPValues($host_id, $snmpField, $dsGraph['snmpQueryId']);

		$snmpValue      = '';
		$snmpValueRegex = '';

		if ($dsGraph['snmpValue']) {
			$snmpValue 	= $dsGraph['snmpValue'][$index_filter];
		} else {
			$snmpValueRegex = $dsGraph['snmpValueRegex'][$index_filter];
		}

		if ($snmpValue) {
			$ok = 0;

			foreach ($snmpValues as $snmpValueKnown => $snmpValueSet) {
				if ($snmpValue == $snmpValueKnown) {
					$ok = 1;
				}
			}

			if (! $ok) {
				echo "ERROR: Unknown snmp-value for field $snmpField - $snmpValue\n";
				echo "Try --snmp-field=$snmpField --list-snmp-values\n";
				exit(1);
			}
		} elseif ($snmpValueRegex) {
			$ok = 0;

			foreach ($snmpValues as $snmpValueKnown => $snmpValueSet) {
				if (preg_match("/$snmpValueRegex/", $snmpValueKnown)) {
					$ok = 1;
				}
			}

			if (! $ok) {
				echo "ERROR: Unknown snmp-value for field $snmpField - $snmpValue\n";
				echo "Try --snmp-field=$snmpField --list-snmp-values\n";
				exit(1);
			}
		}

		$index_filter++;
	}

	if ($listSNMPValues)  {
		if (!$dsGraph['snmpField']) {
			echo "ERROR: You must supply an snmp-field before you can list its values\n";
			echo "Try --list-snmp-fields\n";
			exit(1);
		}

		if (sizeof($dsGraph['snmpField'])) {
			foreach($dsGraph['snmpField'] as $snmpField) {
				if ($snmpField = "") {
					echo "ERROR: You must supply a valid snmp-field before you can list its values\n";
					echo "Try --list-snmp-fields\n";
					exit(1);
				}
		
				displaySNMPValues($snmpValues, $host_id, $snmpField, $quietMode);
			}
		}

		exit(0);
	}

	if (!isset($graphTemplates[$template_id])) {
		echo 'ERROR: Unknown graph-template-id (' . $template_id . ")\n";
		echo "Try --list-graph-templates\n";
		exit(1);
	}

	if ((!isset($template_id)) || (!isset($host_id))) {
		echo "ERROR: Must have at least a host-id and a graph-template-id\n\n";
		display_help();
		exit(1);
	}

	if (strlen($cgInputFields)) {
		$fields = explode(' ', $cgInputFields);
		if ($template_id > 0) {
			$input_fields = getInputFields($template_id, $quietMode);
		}

		if (sizeof($fields)) {
			foreach ($fields as $option) {
				$data_template_id = 0;
				$option_value = explode('=', $option);

				if (substr_count($option_value[0], ':')) {
					$compound = explode(':', $option_value[0]);
					$data_template_id = $compound[0];
					$field_name       = $compound[1];
				}else{
					$field_name       = $option_value[0];
				}

				/* check for the input fields existance */
				$field_found = FALSE;
				if (sizeof($input_fields)) {
					foreach ($input_fields as $key => $row) {
						if (substr_count($key, $field_name)) {
							if ($data_template_id == 0) {
								$data_template_id = $row['data_template_id'];
							}

							$field_found = TRUE;

							break;
						}
					}
				}

				if (!$field_found) {
					echo 'ERROR: Unknown input-field (' . $field_name . ")\n";
					echo "Try --list-input-fields\n";
					exit(1);
				}

				$value = $option_value[1];

				$values['cg'][$template_id]['custom_data'][$data_template_id][$input_fields[$data_template_id . ':' . $field_name]['data_input_field_id']] = $value;
			}
		}
	}

	$returnArray = array();

	if ($graph_type == 'cg') {
		$existsAlready = db_fetch_cell("SELECT id FROM graph_local WHERE graph_template_id=$template_id AND host_id=$host_id");

		if ((isset($existsAlready)) &&
			($existsAlready > 0) &&
			(!$force)) {
			$dataSourceId  = db_fetch_cell("SELECT
				data_template_rrd.local_data_id
				FROM graph_templates_item, data_template_rrd
				WHERE graph_templates_item.local_graph_id = " . $existsAlready . "
				AND graph_templates_item.task_item_id = data_template_rrd.id
				LIMIT 1");

			echo "NOTE: Not Adding Graph - this graph already exists - graph-id: ($existsAlready) - data-source-id: ($dataSourceId)\n";
			exit(1);
		}else{
			$returnArray = create_complete_graph_from_template($template_id, $host_id, '', $values['cg']);
			$dataSourceId = '';
		}

		if ($graphTitle != '') {
			db_execute("UPDATE graph_templates_graph
				SET title=\"$graphTitle\"
				WHERE local_graph_id=" . $returnArray['local_graph_id']);

			update_graph_title_cache($returnArray['local_graph_id']);
		}

		if (sizeof($returnArray['local_data_id'])) {
			foreach($returnArray['local_data_id'] as $item) {
				push_out_host($host_id, $item);

				if (strlen($dataSourceId)) {
					$dataSourceId .= ', ' . $item;
				}else{
					$dataSourceId = $item;
				}
			}
		}

		/* add this graph template to the list of associated graph templates for this host */
		db_execute('REPLACE INTO host_graph (host_id,graph_template_id) VALUES (' . $host_id . ',' . $template_id . ')');

		echo 'Graph Added - graph-id: (' . $returnArray['local_graph_id'] . ") - data-source-ids: ($dataSourceId)\n";
	}elseif ($graph_type == 'ds') {
		if (($dsGraph['snmpQueryId'] == '') || ($dsGraph['snmpQueryType'] == '') || (sizeof($dsGraph['snmpField']) == 0) ) {
			echo "ERROR: For graph-type of 'ds' you must supply more options\n";
			display_help();
			exit(1);
		}

		$snmp_query_array = array();
		$snmp_query_array['snmp_query_id']       = $dsGraph['snmpQueryId'];
		$snmp_query_array['snmp_index_on']       = get_best_data_query_index_type($host_id, $dsGraph['snmpQueryId']);
		$snmp_query_array['snmp_query_graph_id'] = $dsGraph['snmpQueryType'];

		$req = 'SELECT distinct snmp_index
			FROM host_snmp_cache
			WHERE host_id=' . $host_id . '
			AND snmp_query_id=' . $dsGraph['snmpQueryId'];

		$index_snmp_filter = 0;
		if (sizeof($dsGraph['snmpField'])) {
			foreach ($dsGraph['snmpField'] as $snmpField) {
				$req  .= ' AND snmp_index in (
					SELECT DISTINCT snmp_index FROM host_snmp_cache WHERE host_id=' . $host_id . ' AND field_name = ' . db_qstr($snmpField);

				if (sizeof($dsGraph['snmpValue'])) {
					$req .= ' AND field_value = ' . db_qstr($dsGraph['snmpValue'][$index_snmp_filter]). ')';
				} else {
					$req .= ' AND field_value LIKE "%' . addslashes($dsGraph['snmpValueRegex'][$index_snmp_filter]) . '%")';
				}

				$index_snmp_filter++;
			}
		}

		$snmp_indexes = db_fetch_assoc($req);

		if (sizeof($snmp_indexes)) {
			foreach ($snmp_indexes as $snmp_index) {
				$snmp_query_array['snmp_index'] = $snmp_index['snmp_index'];

				$existsAlready = db_fetch_cell("SELECT id
					FROM graph_local
					WHERE graph_template_id=$template_id
					AND host_id=$host_id
					AND snmp_query_id=" . $dsGraph['snmpQueryId'] . "
					AND snmp_index='" . $snmp_query_array['snmp_index'] . "'");

				if (isset($existsAlready) && $existsAlready > 0) {
					if ($graphTitle != '') {
						db_execute("UPDATE graph_templates_graph
							SET title = \"$graphTitle\"
							WHERE local_graph_id = $existsAlready");

						update_graph_title_cache($existsAlready);
					}

					$dataSourceId = db_fetch_cell("SELECT
						data_template_rrd.local_data_id
						FROM graph_templates_item, data_template_rrd
						WHERE graph_templates_item.local_graph_id = " . $existsAlready . "
						AND graph_templates_item.task_item_id = data_template_rrd.id
						LIMIT 1");

					echo "NOTE: Not Adding Graph - this graph already exists - graph-id: ($existsAlready) - data-source-id: ($dataSourceId)\n";

					continue;
				}

				$empty = array(); /* Suggested Values are not been implemented */

				$returnArray = create_complete_graph_from_template($template_id, $host_id, $snmp_query_array, $empty);

				if ($graphTitle != '') {
					db_execute("UPDATE graph_templates_graph
						SET title=\"$graphTitle\"
						WHERE local_graph_id=" . $returnArray['local_graph_id']);

					update_graph_title_cache($returnArray['local_graph_id']);
				}

				$dataSourceId = db_fetch_cell("SELECT
					data_template_rrd.local_data_id
					FROM graph_templates_item, data_template_rrd
					WHERE graph_templates_item.local_graph_id = " . $returnArray['local_graph_id'] . "
					AND graph_templates_item.task_item_id = data_template_rrd.id
					LIMIT 1");

				foreach($returnArray['local_data_id'] as $item) {
					push_out_host($host_id, $item);

					if (strlen($dataSourceId)) {
						$dataSourceId .= ', ' . $item;
					}else{
						$dataSourceId = $item;
					}
				}

				echo 'Graph Added - graph-id: (' . $returnArray['local_graph_id'] . ") - data-source-ids: ($dataSourceId)\n";
			}
		}else{
			$err_msg = 'ERROR: Could not find snmp-field ' . implode(',', $dsGraph['snmpField']) . ' (';

			if (sizeof($dsGraph['snmpValue'])) {
				$err_msg .= implode($dsGraph['snmpValue']);
			} else {
				$err_msg .= implode($dsGraph['snmpValueRegex']);
			}
			$err_msg .= ') for host-id ' . $host_id . ' (' . $hosts[$host_id]['hostname'] . ")\n";
			
			echo $err_msg;
			echo 'Try --host-id=' . $host_id . " --list-snmp-fields\n";
			exit(1);
		}
	}else{
		echo "ERROR: Graph Types must be either 'cg' or 'ds'\n";
		exit(1);
	}

	exit(0);
}else{
	display_help();
	exit(1);
}

/*  display_version - displays version information */
function display_version() {
	$version = db_fetch_cell('SELECT cacti FROM version');
	echo "Cacti Add Graphs Utility, Version $version, " . COPYRIGHT_YEARS . "\n";
}

function display_help() {
	display_version();

	echo "\nusage: add_graphs.php --graph-type=[cg|ds] --graph-template-id=[ID]\n";
	echo "    --host-id=[ID] [--graph-title=title] [graph options] [--force] [--quiet]\n\n";
	echo "Cacti utility for creating graphs via a command line interface.  This utility can\n";
	echo "create both Data Query (ds) type Graphs as well as Graph Template (cg) type graphs.\n\n";
	echo "For Non Data Query (cg) Graphs:\n";
	echo "    [--input-fields=\"[data-template-id:]field-name=value ...\"] [--force]\n\n";
	echo "    --input-fields  If your data template allows for custom input data, you may specify that\n";
	echo "                    here.  The data template id is optional and applies where two input fields\n";
	echo "                    have the same name.\n";
	echo "    --force         If you set this flag, then new cg graphs will be created, even though they\n";
	echo "                    may already exist\n\n";
	echo "For Data Query (ds) Graphs:\n";
	echo "    --snmp-query-id=[ID] --snmp-query-type-id=[ID] --snmp-field=[SNMP Field] \n";
	echo "                         --snmp-value=[SNMP Value] | --snmp-value-regex=[REGEX]\n";
	echo "    [--graph-title=S]       Defaults to what ever is in the Graph Template/Data Template.\n";
	echo "    [--reindex-method=N]    The reindex method to be used for that data query.\n";
	echo "                            NOTE: If Data Query is already associated, the reindex method will NOT be changed.\n\n";
	echo "    Valid --reindex-methos include\n";
	echo "        0|None   = No reindexing\n";
	echo "        1|Uptime = Uptime goes Backwards (Default)\n";
	echo "        2|Index  = Index Count Changed\n";
	echo "        3|Fields = Verify all Fields\n\n";
	echo "    NOTE: You may supply multiples of the --snmp-field and --snmp-value | --snmp-value-regex arguments.\n\n";
	echo "List Options:\n";
	echo "    --list-hosts\n";
	echo "    --list-graph-templates [--host-template-id=[ID]]\n";
	echo "    --list-input-fields --graph-template-id=[ID]\n";
	echo "    --list-snmp-queries\n";
	echo "    --list-query-types  --snmp-query-id [ID]\n";
	echo "    --list-snmp-fields  --host-id=[ID] [--snmp-query-id=[ID]]\n";
	echo "    --list-snmp-values  --host-id=[ID] [--snmp-query-id=[ID]] --snmp-field=[Field]\n\n";
}