<?php

$c			= $_POST["c"];
$clustername		= $c;
$cluster		= $c;

// Supplied by ExtJS when DataStore has remoteSort: true
//
$sortfield		= isset($_POST['sort'] ) ? $_POST["sort"] : "jid";
$sortorder		= isset($_POST['dir'] ) ? $_POST["dir"] : "ASC"; // ASC or DESC

// Search query from ext.SearchField
//
$query			= isset($_POST['query']) ? $_POST['query'] : null;

global $c, $clustername, $cluster;

// Grid Paging stuff
//
//$pstart	= (int) (isset($_POST['start']) ? $_POST['start'] : $_GET['pstart']);
$pstart	= (int) $_POST['start'];
//$pend	= (int) (isset($_POST['limit']) ? $_POST['limit'] : $_GET['plimit']);
$pend	= (int) $_POST['limit'];

//echo $pend.'p ';
// Need to fool Ganglia here: or it won't parse XML for our cluster
//
$HTTP_POST_VARS["c"]	= $c;
$_GET["c"]		= $c;

global $c, $clustername, $cluster;

include_once "./libtoga.php";

$ds             = new DataSource();
$myxml_data     = &$ds->getData();

global $jobs;

$data_gatherer  = new DataGatherer( $clustername );
$data_gatherer->parseXML( &$myxml_data );

$heartbeat      = &$data_gatherer->getHeartbeat();
$jobs           = &$data_gatherer->getJobs();
//$gnodes         = $data_gatherer->getNodes();
$cpus           = &$data_gatherer->getCpus();
$use_fqdn       = &$data_gatherer->getUsingFQDN();

// The ext grid script will send  a task field which will specify what it wants to do
//$task = '';

if( isset($_POST['task']) )
{
	$task = $_POST['task'];
}
if( isset( $HTTP_POST_VARS['task' ] ) )
{
	$task = $HTTP_POST_VARS['task'];
}

switch($task)
{
    case "LISTING":
        getList();
        break;		
    default:
        echo "{failure:true}";
        break;
}

function quickSearchJobs( $jobs, $query )
{
	$searchresults	= array();

	foreach( $jobs as $jobid => $jobattrs )
	{
		if( $query != null )
		{
			if( strpos( $jobattrs['jid'], $query ) !== false )
			{
				$searchresults[$jobid]	= $jobattrs;
			}
			if( strpos( $jobattrs['owner'], $query ) !== false )
			{
				$searchresults[$jobid]	= $jobattrs;
			}
			if( strpos( $jobattrs['queue'], $query ) !== false )
			{
				$searchresults[$jobid]	= $jobattrs;
			}
			if( strpos( $jobattrs['name'], $query ) !== false )
			{
				$searchresults[$jobid]	= $jobattrs;
			}
		}
	}

	return $searchresults;
}

function sortJobs( $jobs, $sortby, $sortorder )
{
        $sorted = array();

        $cmp    = create_function( '$a, $b',
                "global \$sortby, \$sortorder;".

                "if( \$a == \$b ) return 0;".

                "if (\$sortorder==\"DESC\")".
                        "return ( \$a < \$b ) ? 1 : -1;".
                "else if (\$sortorder==\"ASC\")".
                        "return ( \$a > \$b ) ? 1 : -1;" );

        if( isset( $jobs ) && count( $jobs ) > 0 )
        {
                foreach( $jobs as $jobid => $jobattrs )
                {
                                $state          = $jobattrs[status];
                                $user           = $jobattrs[owner];
                                $queue          = $jobattrs[queue];
                                $name           = $jobattrs[name];
                                $req_cpu        = $jobattrs[requested_time];
                                $req_memory     = $jobattrs[requested_memory];

                                if( $state == 'R' )
                                {
                                        $nodes = count( $jobattrs[nodes] );
                                }
                                else
                                {
                                        $nodes = $jobattrs[nodes];
                                }

                                $ppn            = (int) $jobattrs[ppn] ? $jobattrs[ppn] : 1;
                                $cpus           = $nodes * $ppn;
                                $queued_time    = (int) $jobattrs[queued_timestamp];
                                $start_time     = (int) $jobattrs[start_timestamp];
                                $runningtime    = $report_time - $start_time;

                                switch( $sortby )
                                {
                                        case "jid":
                                                $sorted[$jobid] = $jobid;
                                                break;

                                        case "status":
                                                $sorted[$jobid] = $state;
                                                break;

                                        case "owner":
                                                $sorted[$jobid] = $user;
                                                break;

                                        case "queue":
                                                $sorted[$jobid] = $queue;
                                                break;

                                        case "name":
                                                $sorted[$jobid] = $name;
                                                break;

                                        case "requested_time":
                                                $sorted[$jobid] = timeToEpoch( $req_cpu );
                                                break;

                                        case "requested_memory":
                                                $sorted[$jobid] = $req_memory;
                                                break;

                                        case "ppn":
                                                $sorted[$jobid] = $ppn;
                                                break;
                                        case "nodesct":
                                                $sorted[$jobid] = $nodes;
                                                break;
                                        case "cpus":
                                                $sorted[$jobid] = $cpus;
                                                break;

                                        case "queued_timestamp":
                                                $sorted[$jobid] = $queued_time;
                                                break;

                                        case "start_timestamp":
                                                $sorted[$jobid] = $start_time;
                                                break;

                                        case "runningtime":
                                                $sorted[$jobid] = $runningtime;
                                                break;

                                        default:
                                                break;
                                }
                }
        }

        if( $sortorder == "ASC" )
        {
                asort( $sorted );
        }
        else if( $sortorder == "DESC" )
        {
                arsort( $sorted );
        }

        return $sorted;
}


function getList() 
{
	global $jobs, $hearbeat, $pstart, $pend;
	global $sortfield, $sortorder, $query;

	$job_count		= count( $jobs );

	if( $job_count == 0 )
	{
		echo '({"total":"0", "results":""})';
		return 0;
	}

	$jobresults		= array();

	$cur_job		= 0;

	$sorted_jobs            = sortJobs( $jobs, $sortfield, $sortorder );

	if( $query != null )
	{
		$jobs			= quickSearchJobs( $jobs, $query );
	}
	$result_count		= count( $jobs );

	foreach( $sorted_jobs as $jobid => $jobattrs )
	{
		//if( $jobattrs['reported'] != $heartbeat )
		//{
		//	continue;
		//}

		if( ! array_key_exists( $jobid, $jobs ) )
		{
			continue;
		}

		$jr			= array();
		$jr['jid']		= strval( $jobid );
		$jr['status']		= $jobs[$jobid]['status'];
		$jr['owner']		= $jobs[$jobid]['owner'];
		$jr['queue']		= $jobs[$jobid]['queue'];
		$jr['name']		= $jobs[$jobid]['name'];
		$jr['requested_time']	= makeTime( timeToEpoch( $jobs[$jobid]['requested_time'] ) );

		if( $jr['status'] == 'R' )
		{
			$nodes 		= count( $jobs[$jobid][nodes] );
		}
		else
		{
			$nodes 		= (int) $jobs[$jobid][nodes];
		}

		$jr['ppn']		= strval( $jobs[$jobid][ppn] ? $jobs[$jobid][ppn] : 1 );
		$jr['nodect']		= strval( $nodes );

		if( $jr['status'] == 'R' )
		{
			$jr['nodes']	= implode( ",", $jobs[$jobid]['nodes'] );
		}
		else
		{
			$jr['nodes']	= "";
		}

		$jr['queued_timestamp']	= makeDate( $jobs[$jobid]['queued_timestamp'] );
		$jr['start_timestamp']	= ($jobs[$jobid]['start_timestamp'] ? makeDate( $jobs[$jobid]['start_timestamp'] ) : "");

		if( $jr['status'] == 'R' )
		{
			$runningtime		= (int) $jobs[$jobid]['reported'] - (int) $jobs[$jobid]['start_timestamp'];
			$jr['runningtime']	= makeTime( $runningtime );
		}
		else
		{
			$jr['runningtime']	= "";
		}

		if( ( $cur_job < $pstart ) || ( ($cur_job - $pstart) >= $pend ) )
		{
			$cur_job	= $cur_job + 1;
			continue;
		}
		else
		{
			$cur_job	= $cur_job + 1;
		}

		$jobresults[]		= $jr;
	}

	$jsonresults	= JEncode( $jobresults );

	echo '{"total":"'. $result_count .'","results":'. $jsonresults .'}';

	return 0;
}

// Encodes a SQL array into a JSON formated string: so that Javascript may understand it
//
function JEncode( $arr )
{
	if( version_compare( PHP_VERSION, "5.2", "<" ) )
	{    
		require_once( "./JSON.php" );		//if php<5.2 need JSON class

		$json	= new Services_JSON();		//instantiate new json object
		$data	= $json->encode( $arr );	//encode the data in json format
	} 
	else
	{
		$data	= json_encode( $arr );		//encode the data in json format
	}

	return $data;
}

?>