<?php
/**
 * Stats Base
 *
 * @package     RPRESS
 * @subpackage  Classes/Stats
 * @copyright   Copyright (c) 2018, Magnigenie
 * @license     http://opensource.org/licenses/gpl-2.0.php GNU Public License
 * @since 1.0
*/
// Exit if accessed directly
if ( ! defined( 'ABSPATH' ) ) exit;
/**
 * RPRESS_Stats Class
 *
 * Base class for other stats classes
 *
 * Primarily for setting up dates and ranges
 *
 * @since 1.0
 */
class RPRESS_Stats {
	/**
	 * The start date for the period we're getting stats for
	 *
	 * Can be a timestamp, formatted date, date string (such as August 3, 2013),
	 * or a predefined date string, such as last_week or this_month
	 *
	 * Predefined date options are: today, yesterday, this_week, last_week, this_month, last_month
	 * this_quarter, last_quarter, this_year, last_year
	 *
	 * @since 1.0
	 */
	public $start_date;
	/**
	 * The end date for the period we're getting stats for
	 *
	 * Can be a timestamp, formatted date, date string (such as August 3, 2013),
	 * or a predefined date string, such as last_week or this_month
	 *
	 * Predefined date options are: today, yesterday, this_week, last_week, this_month, last_month
	 * this_quarter, last_quarter, this_year, last_year
	 *
	 * The end date is optional
	 *
	 * @since 1.0
	 */
	public $end_date;
	/**
	 * Flag to determine if current query is based on timestamps
	 *
	 * @since  1.0.0
	 */
	public $timestamp;
	/**
	 *
	 * @since 1.0
	 * @return void
	 */
	public function __construct() { /* nothing here. Call get_sales() and get_earnings() directly */ }
	/**
	 * Get the predefined date periods permitted
	 *
	 * @since 1.0
	 * @return array
	 */
	public function get_predefined_dates() {
		$predefined = array(
			'today'        => __( 'Today',        'restropress' ),
			'yesterday'    => __( 'Yesterday',    'restropress' ),
			'this_week'    => __( 'This Week',    'restropress' ),
			'last_week'    => __( 'Last Week',    'restropress' ),
			'this_month'   => __( 'This Month',   'restropress' ),
			'last_month'   => __( 'Last Month',   'restropress' ),
			'this_quarter' => __( 'This Quarter', 'restropress' ),
			'last_quarter' => __( 'Last Quarter',  'restropress' ),
			'this_year'    => __( 'This Year',    'restropress' ),
			'last_year'    => __( 'Last Year',    'restropress' )
		);
		return apply_filters( 'rpress_stats_predefined_dates', $predefined );
	}
	/**
	 * Setup the dates passed to our constructor.
	 *
	 * This calls the convert_date() member function to ensure the dates are formatted correctly
	 *
	 * @since 1.0
	 * @return void
	 */
	public function setup_dates( $_start_date = 'this_month', $_end_date = false ) {
		if( empty( $_start_date ) ) {
			$_start_date = 'this_month';
		}
		if( empty( $_end_date ) ) {
			$_end_date = $_start_date;
		}
		$this->start_date = $this->convert_date( $_start_date );
		$this->end_date   = $this->convert_date( $_end_date, true );
	}
	/**
	 * Converts a date to a timestamp
	 *
	 * @since 1.0
	 * @return array|WP_Error If the date is invalid, a WP_Error object will be returned
	 */
	public function convert_date( $date, $end_date = false ) {
		$this->timestamp = false;
		$second          = $end_date ? 59 : 0;
		$minute          = $end_date ? 59 : 0;
		$hour            = $end_date ? 23 : 0;
		$day             = 1;
		$month           = gmdate( 'n', current_time( 'timestamp' ) );
		$year            = gmdate( 'Y', current_time( 'timestamp' ) );
		if ( ( is_string( $date ) || is_int( $date ) ) && array_key_exists( $date, $this->get_predefined_dates() ) ) {
			// This is a predefined date rate, such as last_week
			switch( $date ) {
				case 'this_month' :
					if( $end_date ) {
						$day    = cal_days_in_month( CAL_GREGORIAN, $month, $year );
						$hour   = 23;
						$minute = 59;
						$second = 59;
					}
					break;
				case 'last_month' :
					if( $month == 1 ) {
						$month = 12;
						$year--;
					} else {
						$month--;
					}
					if( $end_date ) {
						$day = cal_days_in_month( CAL_GREGORIAN, $month, $year );
					}
					break;
				case 'today' :
					$day = gmdate( 'd', current_time( 'timestamp' ) );
					if( $end_date ) {
						$hour   = 23;
						$minute = 59;
						$second = 59;
					}
					break;
				case 'yesterday' :
					$day = gmdate( 'd', current_time( 'timestamp' ) ) - 1;
					// Check if Today is the first day of the month (meaning subtracting one will get us 0)
					if( $day < 1 ) {
						// If current month is 1
						if( 1 == $month ) {
							$year -= 1; // Today is January 1, so skip back to last day of December
							$month = 12;
							$day   = cal_days_in_month( CAL_GREGORIAN, $month, $year );
						} else {
							// Go back one month and get the last day of the month
							$month -= 1;
							$day    = cal_days_in_month( CAL_GREGORIAN, $month, $year );
						}
					}
					break;
				case 'this_week' :
					$days_to_week_start = ( gmdate( 'w', current_time( 'timestamp' ) ) - 1 ) *60*60*24;
				 	$today = gmdate( 'd', current_time( 'timestamp' ) ) *60*60*24;
				 	if( $today < $days_to_week_start ) {
				 		if( $month > 1 ) {
					 		$month -= 1;
					 	} else {
					 		$month = 12;
					 	}
				 	}
					if( ! $end_date ) {
					 	// Getting the start day
						$day = gmdate( 'd', current_time( 'timestamp' ) - $days_to_week_start ) - 1;
						$day += get_option( 'start_of_week' );
					} else {
						// Getting the end day
						$day = gmdate( 'd', current_time( 'timestamp' ) - $days_to_week_start ) - 1;
						$day += get_option( 'start_of_week' ) + 6;
					}
					break;
				case 'last_week' :
					$days_to_week_start = ( gmdate( 'w', current_time( 'timestamp' ) ) - 1 ) *60*60*24;
				 	$today = gmdate( 'd', current_time( 'timestamp' ) ) *60*60*24;
				 	if( $today < $days_to_week_start ) {
				 		if( $month > 1 ) {
					 		$month -= 1;
					 	} else {
					 		$month = 12;
					 	}
				 	}
					if( ! $end_date ) {
					 	// Getting the start day
						$day = gmdate( 'd', current_time( 'timestamp' ) - $days_to_week_start ) - 8;
						$day += get_option( 'start_of_week' );
					} else {
						// Getting the end day
						$day = gmdate( 'd', current_time( 'timestamp' ) - $days_to_week_start ) - 8;
						$day += get_option( 'start_of_week' ) + 6;
					}
					break;
				case 'this_quarter' :
					$month_now = gmdate( 'n', current_time( 'timestamp' ) );
					if ( $month_now <= 3 ) {
						if( ! $end_date ) {
							$month = 1;
						} else {
							$month = 3;
							$day    = cal_days_in_month( CAL_GREGORIAN, $month, $year );
							$hour   = 23;
							$minute = 59;
							$second = 59;
						}
					} else if ( $month_now <= 6 ) {
						if( ! $end_date ) {
							$month = 4;
						} else {
							$month = 6;
							$day    = cal_days_in_month( CAL_GREGORIAN, $month, $year );
							$hour   = 23;
							$minute = 59;
							$second = 59;
						}
					} else if ( $month_now <= 9 ) {
						if( ! $end_date ) {
							$month = 7;
						} else {
							$month = 9;
							$day    = cal_days_in_month( CAL_GREGORIAN, $month, $year );
							$hour   = 23;
							$minute = 59;
							$second = 59;
						}
					} else {
						if( ! $end_date ) {
							$month = 10;
						} else {
							$month = 12;
							$day    = cal_days_in_month( CAL_GREGORIAN, $month, $year );
							$hour   = 23;
							$minute = 59;
							$second = 59;
						}
					}
					break;
				case 'last_quarter' :
					$month_now = gmdate( 'n', current_time( 'timestamp' ) );
					if ( $month_now <= 3 ) {
						if( ! $end_date ) {
							$month = 10;
						} else {
							$year -= 1;
							$month = 12;
							$day    = cal_days_in_month( CAL_GREGORIAN, $month, $year );
							$hour   = 23;
							$minute = 59;
							$second = 59;
						}
					} else if ( $month_now <= 6 ) {
						if( ! $end_date ) {
							$month = 1;
						} else {
							$month = 3;
							$day    = cal_days_in_month( CAL_GREGORIAN, $month, $year );
							$hour   = 23;
							$minute = 59;
							$second = 59;
						}
					} else if ( $month_now <= 9 ) {
						if( ! $end_date ) {
							$month = 4;
						} else {
							$month = 6;
							$day    = cal_days_in_month( CAL_GREGORIAN, $month, $year );
							$hour   = 23;
							$minute = 59;
							$second = 59;
						}
					} else {
						if( ! $end_date ) {
							$month = 7;
						} else {
							$month = 9;
							$day    = cal_days_in_month( CAL_GREGORIAN, $month, $year );
							$hour   = 23;
							$minute = 59;
							$second = 59;
						}
					}
					break;
				case 'this_year' :
					if( ! $end_date ) {
						$month  = 1;
					} else {
						$month  = 12;
						$day    = cal_days_in_month( CAL_GREGORIAN, $month, $year );
						$hour   = 23;
						$minute = 59;
						$second = 59;
					}
					break;
				case 'last_year' :
					$year -= 1;
					if( ! $end_date ) {
						$month = 1;
						$day   = 1;
					} else {
						$month  = 12;
						$day    = cal_days_in_month( CAL_GREGORIAN, $month, $year );
						$hour   = 23;
						$minute = 59;
						$second = 59;
					}
				break;
			}
		} else if( is_numeric( $date ) ) {
			// return $date unchanged since it is a timestamp
			$this->timestamp = true;
		} else if( false !== strtotime( $date ) ) {
			$date  = strtotime( $date, current_time( 'timestamp' ) );
			$year  = gmdate( 'Y', $date );
			$month = gmdate( 'm', $date );
			$day   = gmdate( 'd', $date );
		} else {
			return new WP_Error( 'invalid_date', __( 'Improper date provided.', 'restropress' ) );
		}
		if( false === $this->timestamp ) {
			// Create an exact timestamp
			$date = mktime( $hour, $minute, $second, $month, $day, $year );
		}
		return apply_filters( 'rpress_stats_date', $date, $end_date, $this );
	}
	/**
	 * Modifies the WHERE flag for payment counts
	 *
	 * @since 1.0
	 * @return string
	 */
	public function count_where( $where = '' ) {
		// Only get payments in our date range
		$start_where = '';
		$end_where   = '';
		if( $this->start_date ) {
			if( $this->timestamp ) {
				$format = 'Y-m-d H:i:s';
			} else {
				$format = 'Y-m-d 00:00:00';
			}
			$start_date  = gmdate( $format, $this->start_date );
			$start_where = " AND p.post_date >= '{$start_date}'";
		}
		if( $this->end_date ) {
			if( $this->timestamp ) {
				$format = 'Y-m-d H:i:s';
			} else {
				$format = 'Y-m-d 23:59:59';
			}
			$end_date  = gmdate( $format, $this->end_date );
			$end_where = " AND p.post_date <= '{$end_date}'";
		}
		$where .= "{$start_where}{$end_where}";
		return $where;
	}
	/**
	 * Modifies the WHERE flag for payment queries
	 *
	 * @since 1.0
	 * @return string
	 */
	public function payments_where( $where = '' ) {
		global $wpdb;
		$start_where = '';
		$end_where   = '';
		if( ! is_wp_error( $this->start_date ) ) {
			if( $this->timestamp ) {
				$format = 'Y-m-d H:i:s';
			} else {
				$format = 'Y-m-d 00:00:00';
			}
			$start_date  = gmdate( $format, $this->start_date );
			$start_where = " AND $wpdb->posts.post_date >= '{$start_date}'";
		}
		if( ! is_wp_error( $this->end_date ) ) {
			if( $this->timestamp ) {
				$format = 'Y-m-d H:i:s';
			} else {
				$format = 'Y-m-d 23:59:59';
			}
			$end_date  = gmdate( $format, $this->end_date );
			$end_where = " AND $wpdb->posts.post_date <= '{$end_date}'";
		}
		$where .= "{$start_where}{$end_where}";
		return $where;
	}
}
