<?php
/**
 * @brief		Dynamic Database Chart Helper
 * @author		<a href='https://www.invisioncommunity.com'>Invision Power Services, Inc.</a>
 * @copyright	(c) Invision Power Services, Inc.
 * @license		https://www.invisioncommunity.com/legal/standards/
 * @package		Invision Community
 * @since		27 Aug 2013
 */

namespace IPS\Helpers\Chart;

/* To prevent PHP errors (extending class does not exist) revealing path */
if ( !defined( '\IPS\SUITE_UNIQUE_KEY' ) )
{
	header( ( isset( $_SERVER['SERVER_PROTOCOL'] ) ? $_SERVER['SERVER_PROTOCOL'] : 'HTTP/1.0' ) . ' 403 Forbidden' );
	exit;
}

/**
 * Dynamic Database Chart Helper
 */
class _Database extends \IPS\Helpers\Chart\Dynamic
{
	/**
	 * @brief	Database Table
	 */
	protected $table;
	
	/**
	 * @brief	Database column that contains date
	 */
	protected $dateField;
	
	/**
	 * @brief	Where clauses
	 */
	public $where	= array();
	
	/**
	 * @brief	Extra column to group by (useful for multi-data line charts)
	 */
	public $groupBy	= '';

	/**
	 * @brief	Group by keys for the series
	 */
	protected $groupByKeys = array();

	/**
	 * @brief	Table Columns
	 */
	public $tableInclude = array();
	
	/**
	 * @brief	Table Column Formatters
	 */
	public $tableParsers = array();

	/**
	 * Constructor
	 *
	 * @param	\IPS\Http\Url	$url			The URL the chart will be displayed on
	 * @param	string			$table			Database Table
	 * @param	string			$dateField		Database column that contains date
	 * @param	string			$title			Title
	 * @param	array			$options		Options
	 * @param	string			$defaultType	The default chart type
	 * @param	string			$defaultTimescale	The default timescale to use
	 * @param	array			$defaultTimes	The default start/end times to use
	 * @param	array 			$tableInclude	Table columns to include in results
	 * @param	string			$identifier		If there will be more than one chart per page, provide a unique identifier
	 * @param	\IPS\DateTime|NULL	$minimumDate	The earliest available date for this chart
	 * @see		<a href='https://google-developers.appspot.com/chart/interactive/docs/gallery'>Charts Gallery - Google Charts - Google Developers</a>
	 * @return	void
	 */
	public function __construct( \IPS\Http\Url $url, $table, $dateField, $title='', $options=array(), $defaultType='AreaChart', $defaultTimescale='monthly', $defaultTimes=array( 'start' => 0, 'end' => 0 ), $tableInclude=array(), $identifier='', $minimumDate=NULL )
	{
		$this->table		= $table;
		$this->dateField	= $dateField;
		$this->identifier	= \substr( md5( $table . $dateField ), 0, 6 ) . $identifier;

		if ( !empty( $tableInclude ) )
		{
			$this->tableInclude = $tableInclude;
			$this->availableTypes[] = 'Table';
		}

		parent::__construct( $url, $title, $options, $defaultType, $defaultTimescale, $defaultTimes, $identifier, $minimumDate );
	}
	
	/**
	 * Add Series
	 *
	 * @param	string	$name		Name
	 * @param	string	$type		Type of value
	 *	@li	string
	 *	@li	number
	 *	@li	boolean
	 *	@li	date
	 *	@li	datetime
	 *	@li	timeofday
	 * @param	string	$sql		SQL expression to get value
	 * @param	bool	$filterable	If TRUE, will show as a filter option to be toggled on/off
	 * @param	string	$groupByKey	If $this->groupBy is set, the raw key value
	 * @return	void
	 */
	public function addSeries( $name, $type, $sql, $filterable=TRUE, $groupByKey=NULL )
	{
		if ( $groupByKey !== NULL )
		{
			$filterKey = $groupByKey;
		}
		else
		{
			\IPS\Member::loggedIn()->language()->parseOutputForDisplay( $name );
			$filterKey = $name;
		}
		
		if ( !$filterable or !isset( \IPS\Request::i()->filters[ $this->identifier ] ) or in_array( $filterKey, \IPS\Request::i()->filters[ $this->identifier ] ) )
		{
			if ( $this->type !== 'PieChart' and $this->type !== 'GeoChart' )
			{
				$this->addHeader( $name, $type );
			}

			if( $this->groupBy )
			{
				$this->groupByKeys[]	= $groupByKey;
			}

			$this->series[ $filterKey ] = $sql;
			
			if ( $filterable )
			{
				$this->currentFilters[] = $filterKey;
				
				if ( isset( \IPS\Request::i()->filters[ $this->identifier ] ) )
				{
					$this->url = $this->url->setQueryString( 'filters', array( $this->identifier => $this->currentFilters ) );
				}
			}
		}
		
		if ( $filterable )
		{
			$this->availableFilters[ $filterKey ] = $name;
		}
	}
	
	/**
	 * Get the chart output
	 *
	 * @return	string
	 */
	public function getOutput()
	{
		/* Init data */
		$data = $this->initData();

		/* Work out where clause */
		$where = $this->where;
		$where[] = array( "{$this->dateField}>?", 0 );
		if ( $this->start )
		{
			$where[] = array( "{$this->dateField}>?", $this->start->getTimestamp() );
		}
		if ( $this->end )
		{
			$where[] = array( "{$this->dateField}<?", $this->end->getTimestamp() );
		}
		
		/* What's our SQL time? */
		switch ( $this->timescale )
		{
			case 'daily':
				$timescale = '%Y-%c-%e';
				break;
			
			case 'weekly':
				$timescale = '%Y-%u';
				break;
				
			case 'monthly':
				$timescale = '%Y-%c';
				break;
		}
		
		/* Table... */
		if ( $this->type === 'Table' )
		{
			$table = new \IPS\Helpers\Table\Db( $this->table, $this->url, $where );
			$table->include = $this->tableInclude;
			$table->parsers = $this->tableParsers;
			$table->sortBy = $table->sortBy ?: $this->dateField;
			$output = (string) $table;
		}
		
		/* Pie Chart */
		if ( $this->type === 'PieChart' or $this->type === 'GeoChart' )
		{						
			$keys = array_unique( $this->series );
			$key = array_pop( $keys );
			
			$stmt = \IPS\Db::i()->select(
				"{$key}" . ( $this->groupBy ? ", {$this->groupBy}" : '' ),
				$this->table,
				$where,
				NULL,
				NULL,
				$this->groupBy
			)->setKeyField( $this->groupBy )->setValueField( $key );
								
			foreach ( $stmt as $k => $v )
			{
				if( count( $this->availableFilters ) and !in_array( $k, $this->currentFilters ) )
				{
					continue;
				}
				$this->addRow( array( 'key' => $this->availableFilters[ $k ], 'value' => $v ) );
			}
			
			$output = $this->render( $this->type, $this->options, $this->format );
		}
		
		/* Graph */
		else
		{
			/* Fetch */
			$fromUnixTime = "FROM_UNIXTIME( IFNULL( {$this->dateField}, 0 ) )";
			if ( !$this->timezoneError and \IPS\Member::loggedIn()->timezone and in_array( \IPS\Member::loggedIn()->timezone, \DateTimeZone::listIdentifiers() ) )
			{
				$fromUnixTime = "CONVERT_TZ( {$fromUnixTime}, @@session.time_zone, '" . \IPS\Db::i()->escape_string( \IPS\Member::loggedIn()->timezone ) . "' )";
			}
			$stmt = \IPS\Db::i()->select(
				"DATE_FORMAT( {$fromUnixTime}, '{$timescale}' ) AS time_as_alias, " . implode( ', ', array_unique( $this->series ) ) . ( $this->groupBy ? ", " . $this->groupBy : '' ),
				$this->table,
				$where,
				'time_as_alias ASC',
				NULL,
				$this->groupBy ? array( 'time_as_alias', $this->groupBy ) : 'time_as_alias'
			);

			foreach ( $stmt as $row )
			{
				$result	= array();
	
				if( $this->groupBy )
				{
					if( count( $this->availableFilters ) AND !in_array( $row[ $this->groupBy ], $this->currentFilters ) )
					{
						continue;
					}
				}
	
				foreach( $this->series as $column )
				{
					if( $this->groupBy )
					{
						if( empty( $data[ $row['time_as_alias'] ] ) )
						{
							$result	= array( $row[ $this->groupBy ] => $row[ $column ] );
						}
						else
						{
							$result	= $data[ $row['time_as_alias'] ];
							$result[ $row[ $this->groupBy ] ] = $row[ $column ];
						}
					}
					else
					{
						$result[ $column ]	= $row[ $column ];
					}
				}
	
				$data[ $row['time_as_alias'] ] = $result;
			}

			ksort( $data, SORT_NATURAL );
			
			/* Add to graph */
			$min = NULL;
			$max = NULL;
			foreach ( $data as $time => $d )
			{
				$datetime = new \IPS\DateTime;

				if ( \IPS\Member::loggedIn()->timezone )
				{
					try
					{
						$datetime->setTimezone( new \DateTimeZone( \IPS\Member::loggedIn()->timezone ) );
					}
					catch ( \Exception $e )
					{
						\IPS\Member::loggedIn()->timezone	= null;
						\IPS\Member::loggedIn()->save();
					}
				}

				$datetime->setTime( 0, 0, 0 );
				$exploded = explode( '-', $time );
				switch ( $this->timescale )
				{
					case 'none':
						$datetime = \IPS\DateTime::ts( $time );
						break;

					case 'daily':
						$datetime->setDate( (float) $exploded[0], $exploded[1], $exploded[2] );
						//$datetime = $datetime->localeDate();
						break;
						
					case 'weekly':
						$datetime->setISODate( (float) $exploded[0], $exploded[1] );
						//$datetime = $datetime->localeDate();
						break;
						
					case 'monthly':
						$datetime->setDate( (float) $exploded[0], $exploded[1], 1 );
						//$datetime = $datetime->format( 'F Y' );
						break;
				}
							
				if ( empty( $d ) )
				{
					if ( empty( $this->series ) )
					{
						$this->addRow( array( $datetime ) );
					}
					else
					{
						$this->addRow( array_merge( array( $datetime ), array_fill( 0, count( $this->series ), 0 ) ) );
					}
				}
				else
				{
					if( $this->groupBy )
					{
						$_values	= array();
						foreach ( $this->series as $id => $col )
						{
							$_values[ $id ] = ( isset( $d[ $id ] ) ) ? $d[ $id ] : ( $this->plotZeros ? 0 : NULL );
						}
						
						$this->addRow( array_merge( array( $datetime ), $_values ) );
					}
					else
					{
						if( count($d) < count($this->series) )
						{
							$this->addRow( array_merge( array( $datetime ), $d, array_fill( 0, count($this->series) - count($d), 0 ) ) );
						}
						else
						{
							$this->addRow( array_merge( array( $datetime ), $d ) );
						}
					}
				}
			}
			
			if ( count( $data ) === 1 )
			{
				$this->options['domainAxis']['type'] = 'category';
			}

			$output = $this->render( $this->type, $this->options, $this->format );
		}

		return $output;
	}
}