<?php
// +----------------------------------------------------------------------+
// | Copyright 2013  Madpixels  (email : visualizer@madpixels.net)        |
// +----------------------------------------------------------------------+
// | This program is free software; you can redistribute it and/or modify |
// | it under the terms of the GNU General Public License, version 2, as  |
// | published by the Free Software Foundation.                           |
// |                                                                      |
// | 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.                         |
// |                                                                      |
// | You should have received a copy of the GNU General Public License    |
// | along with this program; if not, write to the Free Software          |
// | Foundation, Inc., 51 Franklin St, Fifth Floor, Boston,               |
// | MA 02110-1301 USA                                                    |
// +----------------------------------------------------------------------+
// | Author: Eugene Manuilov <eugene@manuilov.org>                        |
// +----------------------------------------------------------------------+
/**
 * Sources module class.
 *
 * @category Visualizer
 * @package Module
 *
 * @since 1.1.0
 */
class Visualizer_Module_Sources extends Visualizer_Module {

	const NAME = __CLASS__;

	/**
	 * The array of fetched sources.
	 *
	 * @since 1.1.0
	 *
	 * @access private
	 * @var array
	 */
	private $_sources = array();

	/**
	 * Constructor.
	 *
	 * @since 1.1.0
	 *
	 * @access public
	 *
	 * @param Visualizer_Plugin $plugin The instance of the plugin.
	 */
	public function __construct( Visualizer_Plugin $plugin ) {
		parent::__construct( $plugin );
		$this->_addFilter( Visualizer_Plugin::FILTER_GET_CHART_SERIES, 'filterChartSeries', 1, 2 );
		$this->_addFilter( Visualizer_Plugin::FILTER_GET_CHART_DATA, 'filterChartData', 1, 2 );
		$this->_addFilter( 'visualizer_pro_upsell', 'addProUpsell', 10, 2 );
	}

	/**
	 * Filters chart sereis.
	 *
	 * @since 1.1.0
	 *
	 * @access public
	 *
	 * @param array $series The array of chart series.
	 * @param int   $chart_id The chart id.
	 *
	 * @return array The array of filtered series.
	 */
	public function filterChartSeries( $series, $chart_id ) {
		$source = $this->_getSource( $chart_id );
		if ( ! $source ) {
			return $series;
		}

		return $source->repopulateSeries( $series, $chart_id );
	}

	/**
	 * Returns appropriate source object for a chart.
	 *
	 * @since 1.1.0
	 *
	 * @access private
	 *
	 * @param int $chart_id The chart id.
	 *
	 * @return Visualizer_Source The source object if source exists, otherwise FALSE.
	 */
	private function _getSource( $chart_id ) {
		if ( ! isset( $this->_sources[ $chart_id ] ) ) {
			$class = get_post_meta( $chart_id, Visualizer_Plugin::CF_SOURCE, true );
			if ( ! class_exists( $class, true ) ) {
				return false;
			}
			$this->_sources[ $chart_id ] = new $class();
		}

		return $this->_sources[ $chart_id ];
	}

	/**
	 * Filters chart data.
	 *
	 * @since 1.1.0
	 *
	 * @access public
	 *
	 * @param array $data The array of chart data.
	 * @param int   $chart_id The chart id.
	 *
	 * @return array The array of filtered data.
	 */
	public function filterChartData( $data, $chart_id ) {
		$source = $this->_getSource( $chart_id );
		if ( ! $source ) {
			return $data;
		}

		return $source->repopulateData( $data, $chart_id );
	}

	/**
	 * Add the pro upsell html.
	 *
	 * @param string $old The previous html string.
	 * @param string $feature What feature is this filter running for.
	 *
	 * @return string The new html code.
	 */
	public function addProUpsell( $old, $feature = null ) {
		$pro_features   = Visualizer_Module::get_features_for_license( 1 );
		$biz_features   = Visualizer_Module::get_features_for_license( 2 );
		$return  = '';
		$feature = strval( $feature );
		if ( empty( $feature ) ||
			( in_array( $feature, $biz_features, true ) && ! apply_filters( 'visualizer_is_business', false ) ) ||
			( in_array( $feature, $pro_features, true ) && ! Visualizer_Module::is_pro() )
		) {
			// translators: $s - the name of the plan (PRO).
			$msg = sprintf( __( 'Upgrade to %s to activate this feature!', 'visualizer' ), 'PRO' );
			// translators: $s - the name of the plan (Plus).
			$plus_msg = sprintf( __( 'Upgrade to %s plan to activate this feature!', 'visualizer' ), 'Plus' );
			if ( in_array( $feature, $biz_features, true ) && Visualizer_Module::is_pro() ) {
				$msg = $plus_msg;
			}
			if ( in_array( $feature, array( 'db-query', 'chart-permissions' ), true ) ) {
				$msg = $plus_msg;
			}

			$return = '<div class="only-pro-content">';
			$return .= '	<div class="only-pro-container">';
			$return .= '		<div class="only-pro-inner">';
			$return .= '			<p>' . $msg . '</p>';
			$return .= '            <a target="_blank" href="' . tsdk_utmify( Visualizer_Plugin::PRO_TEASER_URL, esc_attr( $feature ) ) . '" title="' . __( 'Upgrade Now', 'visualizer' ) . '">' . __( 'Upgrade Now', 'visualizer' ) . '</a>';
			$return .= ' 		</div>';
			$return .= ' 	</div>';
			$return .= '</div>';
		}
		if ( empty( $feature ) && Visualizer_Module::is_pro() ) {
			remove_filter( 'visualizer_pro_upsell', 'addProUpsell', 10, 1 );
			$return = '';
		}

		return $return;
	}
}
