<?php

/* 
	program to import an existing rrd into cacti; this is very messy... 

	Aaron S. Bush (01/31/02)

	TODO: 
	: ability to import files not on the server via file upload.
	: ability to customize graphs...etc
	: a common API to work with the cacti DB would help prevent broken data.
*/

$section = "Add/Edit Data Sources"; 
include ('auth/include/auth.php');
header("Cache-control: no-cache");
include ('include/database.php');
include ('include/config.php');
include_once ('include/form.php');

include_once ("include/cmd_functions.php");
include_once ('include/top_header.php');


/* start element handler for import of RRD XML data */

Function start_element($parser, $name, $attribs) {
	GLOBAL $element_status;

	switch($name) {
		case 'step':
			$element_status['in_step'] = TRUE;
			break;
		case 'ds':
			$element_status['in_ds'] = TRUE;
			break;
		case 'name':
			if ($element_status['in_ds']) {
				$element_status['in_ds_name'] = TRUE;
			}
			break;
		case 'type':
			if ($element_status['in_ds']) {
				$element_status['in_ds_type'] = TRUE;
			}
			break;
		case 'minimal_heartbeat':
			if ($element_status['in_ds']) {
				$element_status['in_ds_heart'] = TRUE;
			}
			break;
		case 'min':
			if ($element_status['in_ds']) {
				$element_status['in_ds_min'] = TRUE;
			}
			break;
		case 'max':
			if ($element_status['in_ds']) {
				$element_status['in_ds_max'] = TRUE;
			}
			break;
		default:
			break;
	}
} /* start_element */


/* end element handler for import of RRD XML data */

Function end_element($parser, $name) {
	GLOBAL $element_status, $ds_import_offset;

	switch($name) {
		case 'step':
			$element_status['in_step'] = FALSE;
			break;
		case 'ds':
			$element_status['in_ds'] = FALSE;
			$ds_import_offset++;
			break;
		case 'name':
			if ($element_status['in_ds']) {
				$element_status['in_ds_name'] = FALSE;
			}
			break;
		case 'type':
			if ($element_status['in_ds']) {
				$element_status['in_ds_type'] = FALSE;
			}
			break;
		case 'minimal_heartbeat':
			if ($element_status['in_ds']) {
				$element_status['in_ds_heart'] = FALSE;
			}
			break;
		case 'min':
			if ($element_status['in_ds']) {
				$element_status['in_ds_min'] = FALSE;
			}
			break;
		case 'max':
			if ($element_status['in_ds']) {
				$element_status['in_ds_max'] = FALSE;
			}
			break;
		default:
			break;
	}
} /* end_element */


/* data element handler for import of RRD XML data */

Function data_handler($parser, $data) {
	GLOBAL $element_status, $step, $ds_import_query, $ds_import_offset, $ds_types;

	if (strlen($data = trim($data)) <= 0 ) {
		return;
	}

	if ($element_status['in_step']) {
		$step = $data;
	} elseif ($element_status['in_ds']) {
		if ($element_status['in_ds_name']) {
			$ds_import_query[$ds_import_offset] .= "DSName = '$data', ";
		} elseif ($element_status['in_ds_type']) {
			$ds_import_query[$ds_import_offset] .= "DataSourceTypeID = ".$ds_types["$data"].", ";
		} elseif ($element_status['in_ds_heart']) {
			$ds_import_query[$ds_import_offset] .= "Heartbeat = $data, ";
		} elseif ($element_status['in_ds_min']) {
			$sci_split = explode('e', $data, 2);
			if (count($sci_split) == 2) {
				$data = $sci_split[0] * (pow(10,$sci_split[1]));
			} else {
				$data = 0;
			}
			$ds_import_query[$ds_import_offset] .= "MinValue = ".$data.", ";
		} elseif ($element_status['in_ds_max']) {
			$sci_split = explode('e', $data, 2);
			if (count($sci_split) == 2) {
				$data = $sci_split[0] * (pow(10,$sci_split[1]));
			} else {
				$data = 0;
			}
			$ds_import_query[$ds_import_offset] .= "MaxValue = ".$data;
		} else {
			return;
		}
	} else {
		return;
	}
} /* data_handler */


/* == MAIN == */

switch($paction) {
	case 'import':

		$ds_import_query = array();
		$ds_import_offset = 0;
		
		/* collect existing def_ds records */

		if (! ($result = mysql_query("SELECT * FROM def_ds")) ) {
			DrawMatrixTableBegin("97%");
			DrawMatrixRowBegin();
				DrawMatrixHeaderTop("Error: Selecting def_ds",$color_dark_bar,"FF0000","4");
			DrawMatrixRowEnd();
			DrawMatrixTableEnd();
			break;
		}
		
		while ($row = mysql_fetch_array($result)) {
			$ds_types[$row['Name']] = $row['ID'];
		}
		mysql_free_result($result);

		/* if file is already on the server */

		if (! file_exists($path_rra.'/'.$s_fname)) {
			DrawMatrixTableBegin("97%");
			DrawMatrixRowBegin();
			DrawMatrixHeaderTop("Error: RRD (".$path_rra."/".$s_fname.") Does NOT Exist; Please upload the file first.",$color_dark_bar,"FF0000","4");
			DrawMatrixRowEnd();
			DrawMatrixTableEnd();
			break;
		} else {
			$process_files[] = $path_rra."/".$s_fname;
		} 

		/* prep XML handlers */

		$xml_parser = xml_parser_create();
	
		xml_parser_set_option($xml_parser, XML_OPTION_CASE_FOLDING, 0);
		xml_set_element_handler($xml_parser, "start_element", "end_element");
		xml_set_character_data_handler($xml_parser, "data_handler");

		while (list(,$file) = each($process_files)) {
			
			/* initialize xml element status flags */

			$element_status['in_step'] = FALSE;
			$element_status['in_ds'] = FALSE;
			$element_status['in_ds_name'] = FALSE;
			$element_status['in_ds_type'] = FALSE;
			$element_status['in_ds_heart'] = FALSE;
			$element_status['in_ds_min'] = FALSE;
			$element_status['in_ds_max'] = FALSE;

			list($base_name) = explode('.', basename($file), 2);
			if (!($fp = popen($config['path_rrdtool']['value']." dump $file", "r"))) {
				DrawMatrixTableBegin("97%");
				DrawMatrixRowBegin();
				DrawMatrixHeaderTop("Error: Opening RRD (".$path_rra."/".$s_fname.").",$color_dark_bar,"FF0000","4");
				DrawMatrixRowEnd();
				DrawMatrixTableEnd();
				break;
			}
		
			/* extract rrd info from XML dump */

			while ($data = fread($fp, 4096)) {
				if (!xml_parse($xml_parser, $data, feof($fp))) {
					DrawMatrixTableBegin("97%");
					DrawMatrixRowBegin();
					DrawMatrixHeaderTop("Error: ".
						sprintf("XML error: %s at line %d",
						xml_error_string(xml_get_error_code($xml_parser)),
						xml_get_current_line_number($xml_parser))
						,$color_dark_bar,"FF0000","4"
					);
					DrawMatrixRowEnd();
					DrawMatrixTableEnd();
					pclose($fp);
					exit;
				}
			}
			pclose($fp);

			/* prep import queries */

			while (list($x, $y) = each($ds_import_query)) {
				$ds_import_query[$x] = "INSERT INTO rrd_ds SET Step = $step, Name = '$base_name', Active = '', $y";
				$graph_import_query[$x] = "INSERT INTO rrd_graph SET Title = '$base_name ($x)', ImageFormatID = 1, Height = 120, Width = 500, AutoScale = 'on', AutoScaleOpts = 2";
			}


			/* execute import queries */

			reset($ds_import_query);
			while (list($x, $y) = each($ds_import_query)) {
				print "Setup Data Source: $y<BR>\n";
				if ($preview != 'on') {
					if ( ! mysql_query($y) ) {
						print "ERROR on DS import";
						exit;
					} else {
						if ( ! ($result = mysql_query("SELECT LAST_INSERT_ID()")) ) {
							print "ERROR on last id";
							exit;
						} else {
							$rrd_ds_id = mysql_result($result,'',0);
						}
					}
				} else {
					$rrd_ds_id = 'x';
				}

				print "Setup Graph: $graph_import_query[$x]<BR>\n";
				if ($preview != 'on') {
					if ( ! mysql_query($graph_import_query[$x]) ) {
						print "ERROR on Graph import";
						exit;
					} else {
						if ( ! ($result = mysql_query("SELECT LAST_INSERT_ID()")) ) {
							print "ERROR on last id";
							exit;
						} else {
							$graph_id = mysql_result($result,'',0);
							$graph_item_import = "INSERT INTO rrd_graph_item SET DSID = $rrd_ds_id, ColorID = 1, GraphTypeID = 7, GraphID = $graph_id, ConsolidationFunction = 1";
						}
					}
				} else {
					$graph_id = 'y';
				}
				
				print "Setup Graph Items: $graph_item_import<BR>\n";
				if ($preview != 'on') {
					if ( ! mysql_query($graph_item_import) ) {
						print "ERROR on Graph item import";
						exit;
					} 
				}
			}
		}
		xml_parser_free($xml_parser);

		break;

	default:
		DrawFormHeader("rrdtool Data Source Import",true,false);
			
		DrawFormItem("Server File Name","The name for the RRD file to import; this is for files that are already on the server in the $path_rra/ directory.");
		DrawFormItemTextBox("s_fname","$s_fname","","19");

		DrawFormItemCheckBox("preview","$preview","Preview only","off");
		
		DrawFormItemHiddenTextBox("paction","$paction","import");

		DrawFormSaveButton();
		DrawFormFooter();
		break;
	
}
include_once ("include/bottom_footer.php");
?>