<?php
#
# Copyright (C) 2008 Red Hat, Inc.
# Author: Andreas Thienemann <athienem@redhat.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU Library General Public License as published by
# the Free Software Foundation; version 2 only
#
# 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 Library General Public License for more details.
#
# You should have received a copy of the GNU Library General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
# Copyright 2004, 2005 Red Hat, Inc.
#
# AUTHOR: Andreas Thienemann <athienem@redhat.com>
#

require_once('stacked-bars.inc.php');

// Get the data from the table, format it and write it out to a file
function chart_prepare($flowsrc, $starttime, $endtime, $timeframe = False) {
	global $date_fmt_str;
	global $dbh;

	// Define the plot properties
	$style = 'Top Sources';
	$chart_title = $style.' '.$flowsrc.' - '.strftime($date_fmt_str, $starttime).' to '.strftime($date_fmt_str, $endtime);
	$ytitle = 'Bytes';

	// Grab the data from the database
	// get top-talkers for the timespan: top 9 hosts plus rest-traffic
	$q = $dbh->prepare('SELECT IPDst, SUM(Bytes) AS TrafficSum
			FROM Flows
			WHERE TimeStop BETWEEN ? AND ?
			AND FlowSrc = ?
			GROUP BY IPDst
			ORDER BY TrafficSum DESC
			LIMIT 0, 9');
	$q->bindParam(1, $starttime, PDO::PARAM_INT);
	$q->bindParam(2, $endtime, PDO::PARAM_INT);
	$q->bindParam(3, sprintf('%u', ip2long($flowsrc)), PDO::PARAM_INT);
	$q->execute();

	// Did this return anything at all?
	if ($q->rowCount() == 0) {
		// No
		print('No data for query-period');
		return False;
	} else {
		// Yes? Carry on!

		// Fill the legend build the WHERE limit
		$legend = $q->fetchAll(PDO::FETCH_COLUMN, 0);
		$limit = '';
		for ($i = 0; $i < count($legend); $i++) {
			$limit .= @$legend[$i].', ';
			if (array_key_exists($i, $legend)) {
				$legend[$i] = long2ip($legend[$i]);
			} else {
				$legend[$i] = '';
			}
		}
		$legend[] = 'rest';
		$limit = substr($limit, 0, (strlen($limit) - 2));

		// divide the timespan into 5min intervals
		// query the traffic per talker in 5min intervals
		// Initialize the data_table
		$data = array();
		$steps = 0;
		$rest_table = array();
		for ($time1 = $starttime; $time1 < $endtime; $time1 += (5 * 60)) {
			$time2 = ($time1 + (5 * 60));
			$steps++;
			// Add the column header (time) to the data array
			$data[] = array(strftime('%H:%M', $time1));
			$q = $dbh->prepare('SELECT IPDst, SUM(Bytes) AS TrafficSum
					FROM Flows
					WHERE TimeStop BETWEEN ? AND ?
					AND IPDst IN ('.$limit.')
					AND FlowSrc = ?
					GROUP BY IPDst
					ORDER BY TrafficSum DESC
					LIMIT 0, 9');
			$q->bindParam(1, $time1, PDO::PARAM_INT);
			$q->bindParam(2, $time2, PDO::PARAM_INT);
			$q->bindParam(3, sprintf('%u', ip2long($flowsrc)), PDO::PARAM_INT);
			$q->execute();
			$r = $q->fetchAll();
			$i = (count($data) - 1);
			foreach ($r as $row) {
				$tmpdata[$i][$row[0]] = $row[1];
			}

			// Get the rest of the aggregated traffic to fill up the bars
			$q = $dbh->prepare('SELECT SUM(Bytes) AS TrafficSum
					FROM Flows
					WHERE TimeStop BETWEEN ? AND ?
					AND IPDst NOT IN ('.$limit.')
					AND FlowSrc = ?');
			$q->bindParam(1, $time1, PDO::PARAM_INT);
			$q->bindParam(2, $time2, PDO::PARAM_INT);
			$q->bindParam(3, sprintf('%u', ip2long($flowsrc)), PDO::PARAM_INT);
			$q->execute();
			$rest_table[] = $q->fetch();
		}

		// Iterate through the legend and fill the data array from the temporary data array
		// This function makes sure that each top talker is always at the same position
		// in the multidimensional data table.
		// This is necessary for phpplot to assign the same color to each host on
		// sequential columns
		// $ip is the key for the temp_data lookup
		foreach ($legend as $ip) {
			for($i = 0; $i < $steps; $i++) {
				if (array_key_exists($i, $tmpdata)) {
				// The "column" exists, let's sort the rows
					$key = sprintf('%u', ip2long($ip));
					if (array_key_exists($key, $tmpdata[$i])) {
						$data[$i][] = $tmpdata[$i][$key];
					} else {
						$data[$i][] = '';
					}
				} else {
					$data[$i][] = '';
				}
			}
		}


		// Append the data from the rest_table to the data array
		for($i = 0; $i < $steps; $i++) {
			$data[$i][(count($data[$i]) - 1)] = $rest_table[$i][0];
		}

		return draw_stackedbar($chart_title, $legend, $ytitle, $data);
	}
}