<?/*
+-------------------------------------------------------------------------+
| Copyright (C) 2002 Ian Berry                                            |
|                                                                         |
| 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 rrdtool frontend [php-auth, php-tree, php-form]              |
+-------------------------------------------------------------------------+
| This code is currently maintained and debugged by Ian Berry, any        |
| questions or comments regarding this code should be directed to:        |
| - iberry@raxnet.net                                                     |
+-------------------------------------------------------------------------+
| - raXnet - http://www.raxnet.net/                                       |
+-------------------------------------------------------------------------+
*/?>
<?

function DuplicateGraph($graph_id) {
	include ("include/database.php");
	
	$sql_id = mysql_query("select * from rrd_graph where id=$graph_id", $cnn_id);
	$name = mysql_result($sql_id, 0, "title") . " (Copy)";
	
	/* duplicate graph defs */
	mysql_query("insert into rrd_graph (id,imageformatid,title,height,width,upperlimit,
	lowerlimit,verticallabel,autoscale,autopadding,autoscaleopts,rigid,basevalue,grouping,export) 
	values (0,
		" . mysql_result($sql_id, 0, "imageformatid") . ",\"$name\""
		. "," . mysql_result($sql_id, 0, "height") 
		. "," . mysql_result($sql_id, 0, "width")
		. "," . mysql_result($sql_id, 0, "upperlimit") 
		. "," . mysql_result($sql_id, 0, "lowerlimit")
		. ",\"" . mysql_result($sql_id, 0, "verticallabel") . "\""
		. ",\"" . mysql_result($sql_id, 0, "autoscale") . "\""
		. ",\"" . mysql_result($sql_id, 0, "autopadding") . "\""
		. ",\"" . mysql_result($sql_id, 0, "autoscaleopts") . "\""
		. ",\"" . mysql_result($sql_id, 0, "rigid") . "\""
		. ",\"" . mysql_result($sql_id, 0, "basevalue") . "\""
		. ",\"" . mysql_result($sql_id, 0, "grouping") . "\""
		. ",\"" . mysql_result($sql_id, 0, "export") . "\")", $cnn_id);
	
	$graph_grouping = mysql_result($sql_id, 0, "grouping");
	
	/* duplicate graph items */
	$sql_id = mysql_query("select LAST_INSERT_ID() as id", $cnn_id);
	$last_id = mysql_result($sql_id, 0, "id");
	
	$sql_id = mysql_query("select * from rrd_graph_item where graphid=$graph_id", $cnn_id);
	$rows = mysql_num_rows($sql_id); $i = 0;
	
	while ($i < $rows) {
		mysql_query("insert into rrd_graph_item (id,dsid,colorid,textformat,value,sequence,
			graphid,graphtypeid,consolidationfunction,hardreturn,cdefid,sequenceparent,parent) 
			values (0" 
			. ",\"" . mysql_result($sql_id, $i, "dsid") . "\""
			. ",\"" . mysql_result($sql_id, $i, "colorid") . "\""
			. ",\"" . mysql_result($sql_id, $i, "textformat") . "\""
			. ",\"" . mysql_result($sql_id, $i, "value") . "\""
			. ",\"" . mysql_result($sql_id, $i, "sequence") . "\""
			. ",\"" . $last_id . "\""
			. ",\"" . mysql_result($sql_id, $i, "graphtypeid") . "\""
			. ",\"" . mysql_result($sql_id, $i, "consolidationfunction") . "\""
			. ",\"" . mysql_result($sql_id, $i, "hardreturn") . "\""
			. ",\"" . mysql_result($sql_id, $i, "cdefid") . "\""
			. ",\"" . mysql_result($sql_id, $i, "sequenceparent") . "\""
			. ",\"" . mysql_result($sql_id, $i, "parent") . "\")", $cnn_id);
		$i++;
	}
	
	/* if graph grouping is turned on; we are going to "reset" it by undoing and redoing
	the graph grouping. This will force each item to get valid parent items instead of
	pointing to the old graph items */
	if ($graph_grouping == "on") {
		ungroup_graph_items($last_id);
		group_graph_items($last_id);
	}
}

function DuplicateDataSource($datasource_id) {
	include ("include/database.php");
	
	$sql_id = mysql_query("select * from rrd_ds where id=$datasource_id", $cnn_id);
	
	$name = mysql_result($sql_id, 0, "name") . "_copy";
	/* duplicate data source defs */
	mysql_query("insert into rrd_ds (id,name,datasourcetypeid,heartbeat,
		minvalue,maxvalue,srcid,active,dsname,dspath,step) values (0"
		. ",\"" . $name . "\""
		. ",\"" . mysql_result($sql_id, $i, "datasourcetypeid") . "\""
		. ",\"" . mysql_result($sql_id, $i, "heartbeat") . "\""
		. ",\"" . mysql_result($sql_id, $i, "minvalue") . "\""
		. ",\"" . mysql_result($sql_id, $i, "maxvalue") . "\""
		. ",\"" . mysql_result($sql_id, $i, "srcid") . "\""
		. ",\"" . mysql_result($sql_id, $i, "active") . "\""
		. ",\"" . mysql_result($sql_id, $i, "dsname") . "\""
		. ",\"\""
		. ",\"" . mysql_result($sql_id, $i, "step") . "\""
		. ")", $cnn_id);
	
	/* duplicate data source data */
	$sql_id = mysql_query("select LAST_INSERT_ID() as id", $cnn_id);
	$last_id = mysql_result($sql_id, 0, "id");
	
	$sql_id = mysql_query("select * from src_data where dsid=$datasource_id", $cnn_id);
	$rows = mysql_num_rows($sql_id); $i = 0;
	
	while ($i < $rows) {
		mysql_query("insert into src_data (id,fieldid,dsid,value) values (0" 
			. ",\"" . mysql_result($sql_id, $i, "fieldid") . "\""
			. ",\"" . $last_id . "\""
			. ",\"" . mysql_result($sql_id, $i, "value") . "\""
			. ")", $cnn_id);
		$i++;
	}
	
	/* duplicate data source -> rra link data */
	$sql_id = mysql_query("select * from lnk_ds_rra where dsid=$datasource_id", $cnn_id);
	$rows = mysql_num_rows($sql_id); $i = 0;
	
	while ($i < $rows) {
		mysql_query("insert into lnk_ds_rra (dsid,rraid) values ($last_id"
			. ",\"" . mysql_result($sql_id, $i, "rraid") . "\""
			. ")", $cnn_id);
		$i++;
	}
	
	return $last_id;
}

function group_graph_items($graph_id) {
	include ("include/database.php");
	
	$sql_id = mysql_query("select 
		i.consolidationfunction,i.sequence,i.id,i.dsid,
		t.name 
		from rrd_graph_item i left join 
		def_graph_type t on 
		i.graphtypeid=t.id 
		where i.graphid=$graph_id 
		order by i.sequenceparent, i.sequence", $cnn_id);
	$rows = mysql_num_rows($sql_id); $i = 0;
	
	while ($i < $rows) {
		$child_counter[mysql_result($sql_id, $i, "dsid")]++;
		
		if ((mysql_result($sql_id, $i, "name") == "AREA") || (mysql_result($sql_id, $i, "name") == "STACK") || (mysql_result($sql_id, $i, "name") == "LINE1") || (mysql_result($sql_id, $i, "name") == "LINE2") || (mysql_result($sql_id, $i, "name") == "LINE3")) {
			if (isset($has_been_used[mysql_result($sql_id, $i, "dsid")][mysql_result($sql_id, $i, "consolidationfunction")]) == false) {
				$parent_counter++;
			}
			
			$current_parent_id = mysql_result($sql_id, $i, "id");
			$current_sequence = 0;
			
			$has_been_used[mysql_result($sql_id, $i, "dsid")][mysql_result($sql_id, $i, "consolidationfunction")] = true;
		}else{
			$current_sequence = $child_counter[mysql_result($sql_id, $i, "dsid")];
		}
		
		mysql_query("update rrd_graph_item set sequence=$current_sequence
			, sequenceparent=$parent_counter, parent=$current_parent_id
			  where id=" . mysql_result($sql_id, $i, "id"), $cnn_id);
		$i++;
	}
}

function ungroup_graph_items($graph_id) {
	include ("include/database.php");
	
	$sql_id = mysql_query("select sequence,id,dsid from rrd_graph_item where graphid=$graph_id
		order by sequenceparent,sequence", $cnn_id);
	$rows = mysql_num_rows($sql_id); $i = 0;
	
	while ($i < $rows) {
		mysql_query("update rrd_graph_item set sequence=" . ($i+1) . " where id=" . mysql_result($sql_id, $i, "id"), $cnn_id);
		$i++;
	}
	
	mysql_query("update rrd_graph_item set sequenceparent=0 where graphid=$graph_id", $cnn_id);
	mysql_query("update rrd_graph_item set parent=0 where graphid=$graph_id", $cnn_id);
	
	$i++;
}

function CreateGraphDataFromSNMPData($graph_parameters) {
	include ("include/database.php");
	
	$sql_id = mysql_query("select * from snmp_hosts_interfaces where id=" . $graph_parameters["snmp_interface_id"], $cnn_id);
	
	/* ---- Data Source ---- */
	
	/* defaults */
	$ds_datasourcetype = "COUNTER";
	$ds_heartbeat = "600";
	$ds_minvalue = "0";
	$ds_maxvalue = mysql_result($sql_id, 0, "speed");
	$ds_srctype = "snmp_net";
	$snmp_interface_index = mysql_result($sql_id, 0, "interfacenumber");
	$snmp_interface_description = mysql_result($sql_id, 0, "description");
	
	/* host info */
	$sql_id = mysql_query("select * from snmp_hosts where id=" . mysql_result($sql_id, 0, "hostid"), $cnn_id);
	$snmp_host_hostname = mysql_result($sql_id, 0, "hostname");
	$snmp_host_community = mysql_result($sql_id, 0, "community");
	
	$sql_id = mysql_query("select id from def_ds where name=\"$ds_datasourcetype\"", $cnn_id);
	$ds_datasourcetype = mysql_result($sql_id, 0, "id");
	
	/* get the src_id given the src_type */
	$sql_id = mysql_query("select id from src where type=\"$ds_srctype\"", $cnn_id);
	
	if (mysql_num_rows($sql_id) != 0) {
		$ds_srcid = mysql_result($sql_id, 0, "id");
	}
	
	$o = 0;
	while ($o <= 1) {
		switch ($o) {
			case 0:
				$ds_direction = "in";
				break;
			case 1:
				$ds_direction = "out";
				break;
		}
		
		/* set a path for this new data source [RRDPATH/FullDSName_[in|out].rrd] */
		$data_source_path = "<path_rra>/" . strtolower(CleanUpName($graph_parameters["data_source_name"])) . "_" . "$ds_direction.rrd";
		
		$sql_id = mysql_query("insert into rrd_ds (name,datasourcetypeid,heartbeat,minvalue,
			maxvalue,srcid,active,dsname,dspath) values (\"" . $graph_parameters["data_source_name"] . "_" . "$ds_direction\",
			$ds_datasourcetype,$ds_heartbeat,$ds_minvalue,$ds_maxvalue,$ds_srcid,\"on\",\"\",\"$data_source_path\")"
			,$cnn_id);
		
		/* get newly saved id */
		$sql_id = mysql_query("select LAST_INSERT_ID()",$cnn_id);
		$datasource_id[$o] = mysql_result($sql_id,"",0); /* store both dsid's */
		
		/* this will do any of the cleanup that is required on the data source name to make
		sure rrdtool is ok with it. */
		SyncDataSourceName($datasource_id[$o], "", $data_source_path);
		
		/* save rra data: use all rra's */
		$sql_id = mysql_query("select id from rrd_rra", $cnn_id);
		$rows = mysql_num_rows($sql_id); $i = 0;
		
		while ($i < $rows) {
			mysql_query("insert into lnk_ds_rra (dsid,rraid) values ($datasource_id[$o]," . mysql_result($sql_id, $i, "id") . ")",$cnn_id);
			
			$i++;
		}
		
		/* now we have to write the acual data for these ds's to start working, this will
		only work correctly if the user did not tamper with the data names of the
		"Gather SNMP Network Data" data input source! */
		$sql_id = mysql_query("select id,dataname from src_fields where srcid=$ds_srcid and 
			inputoutput=\"in\"", $cnn_id);
		$rows = mysql_num_rows($sql_id); $i = 0;
		
		while ($i < $rows) {
			$ds_input_value = "";
			/* get values for the input fields that need them */
			switch (mysql_result($sql_id, $i, "dataname")) {
				case 'inout':
					$ds_input_value = $ds_direction;
					break;
				case 'ip':
					$ds_input_value = $snmp_host_hostname;
					break;
				case 'community':
					$ds_input_value = $snmp_host_community;
					break;
				case 'ifnum':
					$ds_input_value = $snmp_interface_index;
					break;
			}
			
			mysql_query("insert into src_data (fieldid,dsid,value) values (" . mysql_result($sql_id, $i, "id") . ",$datasource_id[$o],\"$ds_input_value\")",$cnn_id);
			
			$i++;
		}
		
		$o++;
	}
	
	/* ---- Graph Data ---- */
	
	/* Graph Title */
	$graph_title = BuildGraphTitleFromSNMP($graph_parameters);
	
	if ($graph_title == "") {
		$graph_title = "Traffic Analysis for $snmp_interface_description";
	}
	
	/* defaults */
	$graph_image_format = "PNG";
	$graph_height = "120";
	$graph_width = "500";
	$graph_vertical_label = "Bytes Per Second";
	$graph_color_in = "00CF00";
	$graph_color_out = "002A97";
	
	$sql_id = mysql_query("select id from def_colors where hex=\"$graph_color_in\"", $cnn_id);
	$graph_color_in = mysql_result($sql_id, 0, "id");
	
	$sql_id = mysql_query("select id from def_colors where hex=\"$graph_color_out\"", $cnn_id);
	$graph_color_out = mysql_result($sql_id, 0, "id");
	
	$sql_id = mysql_query("select id from def_image_type where name=\"$graph_image_format\"", $cnn_id);
	$graph_image_format = mysql_result($sql_id, 0, "id");
	
	$sql_id = mysql_query("insert into rrd_graph (imageformatid,title,height,width,upperlimit,
		lowerlimit,verticallabel,autoscale,autopadding,autoscaleopts,rigid,basevalue,grouping,export) 
		values ($graph_image_format,\"$graph_title\",$graph_height,$graph_width,0,0,
		\"$graph_vertical_label\",\"on\",\"on\",2,\"on\",1000,\"on\",\"on\")",$cnn_id);
	
	/* get newly saved id */
	$sql_id = mysql_query("select LAST_INSERT_ID()",$cnn_id);
	$graph_id = mysql_result($sql_id,"",0);
	
	/* this would be a great place to put some code for graph templates, however it is not
	written yet, so its all going to be manual for now */
	
	$graph_hard_return_array[0] = "";
	$graph_hard_return_array[1] = "";
	$graph_hard_return_array[2] = "";
	
	$graph_color_array[1] = "0";
	$graph_color_array[2] = "0";
	$graph_color_array[3] = "0";
	
	$graph_text_format_array[1] = "Current:";
	$graph_text_format_array[2] = "Average:";
	$graph_text_format_array[3] = "Maximum:";
	
	$graph_graph_type_array[1] = "GPRINT";
	$graph_graph_type_array[2] = "GPRINT";
	$graph_graph_type_array[3] = "GPRINT";
	
	$graph_cf_function_array[0] = "AVERAGE";
	$graph_cf_function_array[1] = "LAST";
	$graph_cf_function_array[2] = "AVERAGE";
	$graph_cf_function_array[3] = "MAX";
	
	$o = 0;
	while ($o <= 1) {
		switch ($o) {
			case 0:
				$graph_text_format_array[0] = "Inbound";
				$graph_graph_type_array[0] = "AREA";
				$graph_color_array[0] = $graph_color_in;
				$graph_hard_return_array[3] = "on";
				break;
			case 1:
				$graph_text_format_array[0] = "Outbound";
				$graph_graph_type_array[0] = "LINE1";
				$graph_color_array[0] = $graph_color_out;
				$graph_hard_return_array[3] = "";
				break;
		}
		
		$s = 0;
		while ($s < count($graph_text_format_array)) {
			$sql_id = mysql_query("select id from def_graph_type where name=\"$graph_graph_type_array[$s]\"", $cnn_id);
			$graph_type = mysql_result($sql_id, 0, "id");
			
			$sql_id = mysql_query("select id from def_cf where name=\"$graph_cf_function_array[$s]\"", $cnn_id);
			$graph_cf_function = mysql_result($sql_id, 0, "id");
			
			$sql_id = mysql_query("insert into rrd_graph_item (dsid,colorid,textformat,
				sequence,graphid,graphtypeid,consolidationfunction,hardreturn) values ($datasource_id[$o],
				$graph_color_array[$s],\"$graph_text_format_array[$s]\",((($o*4)+$s)+1),$graph_id,$graph_type,$graph_cf_function,
				\"$graph_hard_return_array[$s]\")",$cnn_id);
			
			$s++;
		}
		
		$o++;
	}
	
	/* turn on graph grouping */
	group_graph_items($graph_id);
}

?>