<?php
/**
 * Earnings / Sales Stats
 *
 * @package     Give
 * @subpackage  Classes/Stats
 * @copyright   Copyright (c) 2016, GiveWP
 * @license     https://opensource.org/licenses/gpl-license GNU Public License
 * @since       1.0
 */

// Exit if accessed directly.
if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

/**
 * Give_Stats Class
 *
 * This class is for retrieving stats for earnings and sales.
 *
 * Stats can be retrieved for date ranges and pre-defined periods.
 *
 * @since 1.0
 */
class Give_Payment_Stats extends Give_Stats {

	/**
	 * Retrieve sale stats
	 *
	 * @since 4.18.0 Count donations with one SQL query instead of loading every ID
	 * @since  1.0
	 * @access public
	 *
	 * @param  $form_id    int          The donation form to retrieve stats for. If false, gets stats for all forms
	 * @param  $start_date string|bool  The starting date for which we'd like to filter our sale stats. If false, we'll use the default start date of `this_month`
	 * @param  $end_date   string|bool  The end date for which we'd like to filter our sale stats. If false, we'll use the default end date of `this_month`
	 * @param  $status     string|array The sale status(es) to count. Only valid when retrieving global stats
	 *
	 * @return float|int                Total amount of donations based on the passed arguments.
	 */
	public function get_sales( $form_id = 0, $start_date = false, $end_date = false, $status = 'publish' ) {

		$this->setup_dates( $start_date, $end_date );

		// Make sure start date is valid
		if ( is_wp_error( $this->start_date ) ) {
			return $this->start_date;
		}

		// Make sure end date is valid
		if ( is_wp_error( $this->end_date ) ) {
			return $this->end_date;
		}

		$args = [
			'status'     => 'publish',
			'start_date' => $this->start_date,
			'end_date'   => $this->end_date,
			'fields'     => 'ids',
			'number'     => - 1,
			'output'     => '',
		];

		if ( ! empty( $form_id ) ) {
			$args['give_forms'] = $form_id;
		}

		$count = $this->count_donations_in_sql( $args );

		if ( null !== $count ) {
			return $count;
		}

		/* @var Give_Payments_Query $payments */
		$payments = new Give_Payments_Query( $args );
		$payments = $payments->get_payments();

		return count( $payments );
	}


	/**
	 * Retrieve earning stats
	 *
	 * @since 4.18.0 Sum donation totals with one SQL query instead of loading every ID and formatting each amount
	 * @since  1.0
	 * @access public
	 *
	 * @param  $form_id     int         The donation form to retrieve stats for. If false, gets stats for all forms.
	 * @param  $start_date  string|bool The starting date for which we'd like to filter our donation earnings stats. If false, method will use the default start date of `this_month`.
	 * @param  $end_date    string|bool The end date for which we'd like to filter the donations stats. If false, method will use the default end date of `this_month`.
	 * @param  $gateway_id  string|bool The gateway to get earnings for such as 'paypal' or 'stripe'.
	 *
	 * @return float|int                Total amount of donations based on the passed arguments.
	 */
	public function get_earnings( $form_id = 0, $start_date = false, $end_date = false, $gateway_id = false ) {
		global $wpdb;
		$this->setup_dates( $start_date, $end_date );

		// Make sure start date is valid
		if ( is_wp_error( $this->start_date ) ) {
			return $this->start_date;
		}

		// Make sure end date is valid
		if ( is_wp_error( $this->end_date ) ) {
			return $this->end_date;
		}

		$args = [
			'status'     => 'publish',
			'start_date' => $this->start_date,
			'end_date'   => $this->end_date,
			'fields'     => 'ids',
			'number'     => - 1,
			'output'     => '',
		];

		// Filter by Gateway ID meta_key
		if ( $gateway_id ) {
			$args['meta_query'][] = [
				'key'   => '_give_payment_gateway',
				'value' => $gateway_id,
			];
		}

		// Filter by Gateway ID meta_key
		if ( $form_id ) {
			$args['meta_query'][] = [
				'key'   => '_give_payment_form_id',
				'value' => $form_id,
			];
		}

		if ( ! empty( $args['meta_query'] ) && 1 < count( $args['meta_query'] ) ) {
			$args['meta_query']['relation'] = 'AND';
		}

		$args = apply_filters( 'give_stats_earnings_args', $args );
		$key  = Give_Cache::get_key( 'give_stats', $args );

		// Set transient for faster stats.
		$earnings = Give_Cache::get( $key );

		if ( false === $earnings ) {

			$this->timestamp = false;
			$payments        = [];
			$earnings        = $this->sum_earnings_in_sql( $args );

			if ( null === $earnings ) {
				$payments = new Give_Payments_Query( $args );
				$payments = $payments->get_payments();
				$earnings = 0;
			}

			if ( ! empty( $payments ) ) {
				$donation_id_col = Give()->payment_meta->get_meta_type() . '_id';
				$query           = "SELECT {$donation_id_col} as id, meta_value as total
					FROM {$wpdb->donationmeta}
					WHERE meta_key='_give_payment_total'
					AND {$donation_id_col} IN ('" . implode( '\',\'', $payments ) . "')";

				$payments = $wpdb->get_results( $query, ARRAY_A );

				if ( ! empty( $payments ) ) {
					foreach ( $payments as $payment ) {
						$currency_code = give_get_payment_currency_code( $payment['id'] );

						/**
						 * Filter the donation amount
						 * Note: this filter documented in payments/functions.php:give_donation_amount()
						 *
						 * @since 2.1
						 */
						$formatted_amount = apply_filters(
							'give_donation_amount',
							give_format_amount( $payment['total'], [ 'donation_id' => $payment['id'] ] ),
							$payment['total'],
							$payment['id'],
							[
								'type'     => 'stats',
								'currency' => false,
								'amount'   => false,
							]
						);

						$earnings += (float) give_maybe_sanitize_amount( $formatted_amount, [ 'currency' => $currency_code ] );
					}
				}
			}

			// Cache the results for one hour.
			Give_Cache::set( $key, give_sanitize_amount_for_db( $earnings ), 60 * 60 );
		}

		/**
		 * Filter the earnings.
		 *
		 * @since 1.8.17
		 *
		 * @param  float       $earnings   Earning amount.
		 * @param  int         $form_id    Donation Form ID.
		 * @param  string|bool $start_date Earning start date.
		 * @param  string|bool $end_date   Earning end date.
		 * @param  string|bool $gateway_id Payment gateway id.
		 */
		$earnings = apply_filters( 'give_get_earnings', $earnings, $form_id, $start_date, $end_date, $gateway_id );

		// return earnings
		return round( $earnings, give_get_price_decimals( $form_id ) );

	}

	/**
	 * Retrieve earning stat transient key
	 *
	 * @since  1.0
	 * @access public
	 *
	 * @param  $form_id     int         The donation form to retrieve stats for. If false, gets stats for all forms
	 * @param  $start_date  string|bool The starting date for which we'd like to filter our donation earnings stats. If false, we'll use the default start date of `this_month`
	 * @param  $end_date    string|bool The end date for which we'd like to filter our sale stats. If false, we'll use the default end date of `this_month`
	 * @param  $gateway_id  string|bool The gateway to get earnings for such as 'paypal' or 'stripe'
	 *
	 * @return float|int                Total amount of donations based on the passed arguments.
	 */
	public function get_earnings_cache_key( $form_id = 0, $start_date = false, $end_date = false, $gateway_id = false ) {

		$this->setup_dates( $start_date, $end_date );

		// Make sure start date is valid
		if ( is_wp_error( $this->start_date ) ) {
			return $this->start_date;
		}

		// Make sure end date is valid
		if ( is_wp_error( $this->end_date ) ) {
			return $this->end_date;
		}

		$args = [
			'status'     => 'publish',
			'start_date' => $this->start_date,
			'end_date'   => $this->end_date,
			'fields'     => 'ids',
			'number'     => - 1,
		];

		// Filter by Gateway ID meta_key
		if ( $gateway_id ) {
			$args['meta_query'][] = [
				'key'   => '_give_payment_gateway',
				'value' => $gateway_id,
			];
		}

		// Filter by Gateway ID meta_key
		if ( $form_id ) {
			$args['meta_query'][] = [
				'key'   => '_give_payment_form_id',
				'value' => $form_id,
			];
		}

		if ( ! empty( $args['meta_query'] ) && 1 < count( $args['meta_query'] ) ) {
			$args['meta_query']['relation'] = 'AND';
		}

		$args = apply_filters( 'give_stats_earnings_args', $args );
		$key  = Give_Cache::get_key( 'give_stats', $args );

		// return earnings
		return $key;

	}

	/**
	 * Sum donation totals in one query instead of loading every matching donation ID into PHP
	 * and formatting each amount. Uses the Currency Switcher base amount when a donation has one,
	 * which is what its `give_donation_amount` filter returns for stats.
	 *
	 * Returns null when the query args contain something this method does not translate, or when an
	 * add-on filters `give_donation_amount`, so the caller falls back to the per-donation loop.
	 *
	 * @since 4.18.0
	 *
	 * @param array $args Give_Payments_Query arguments.
	 *
	 * @return float|null
	 */
	private function sum_earnings_in_sql( array $args ) {
		global $wpdb;

		if ( $this->has_donation_amount_filter() ) {
			return null;
		}

		$where = $this->stats_where_sql( $args );

		if ( null === $where ) {
			return null;
		}

		$donation_id_col = Give()->payment_meta->get_meta_type() . '_id';

		$sql = "SELECT SUM( COALESCE( NULLIF( base.meta_value, '' ), total.meta_value ) + 0 )
			FROM {$wpdb->posts} AS p
			INNER JOIN {$wpdb->donationmeta} AS total ON total.{$donation_id_col} = p.ID AND total.meta_key = '_give_payment_total'
			LEFT JOIN {$wpdb->donationmeta} AS base ON base.{$donation_id_col} = p.ID AND base.meta_key = '_give_cs_base_amount'
			WHERE p.post_type = 'give_payment' {$where}";

		return (float) $wpdb->get_var( $sql );
	}

	/**
	 * Count matching donations in one query instead of loading every ID into PHP.
	 *
	 * @since 4.18.0
	 *
	 * @param array $args Give_Payments_Query arguments.
	 *
	 * @return int|null
	 */
	private function count_donations_in_sql( array $args ) {
		global $wpdb;

		$where = $this->stats_where_sql( $args );

		if ( null === $where ) {
			return null;
		}

		return (int) $wpdb->get_var( "SELECT COUNT(*) FROM {$wpdb->posts} AS p WHERE p.post_type = 'give_payment' {$where}" );
	}

	/**
	 * Add-ons that change amounts through `give_donation_amount` (Fee Recovery, Currency Switcher) need
	 * the per-donation loop for earnings so their callbacks run. Counts are unaffected and stay in SQL. Core itself always hooks the deprecated filter
	 * mapping there, so that one callback does not count.
	 *
	 * @since 4.18.0
	 *
	 * @return bool
	 */
	private function has_donation_amount_filter() {
		global $wp_filter;

		if ( has_filter( 'give_payment_amount' ) ) {
			return true;
		}

		foreach ( $wp_filter['give_donation_amount']->callbacks ?? [] as $callbacks ) {
			foreach ( $callbacks as $callback ) {
				if ( 'give_deprecated_filter_mapping' !== $callback['function'] ) {
					return true;
				}
			}
		}

		return false;
	}

	/**
	 * Translate the subset of Give_Payments_Query arguments the stats methods build (status, date
	 * range, form, parent, and simple meta equality) into a WHERE fragment. Anything else returns null.
	 *
	 * @since 4.18.0
	 *
	 * @param array $args
	 *
	 * @return string|null
	 */
	private function stats_where_sql( array $args ) {
		global $wpdb;

		/**
		 * Allow add-ons that alter stats amounts some other way to keep the per-donation code path.
		 *
		 * @since 4.18.0
		 *
		 * @param bool  $aggregate_in_sql
		 * @param array $args
		 */
		if ( ! apply_filters( 'givewp_payment_stats_aggregate_in_sql', true, $args ) ) {
			return null;
		}

		$known = [ 'status', 'post_status', 'start_date', 'end_date', 'fields', 'number', 'output', 'meta_query', 'give_forms' ];

		if ( array_diff( array_keys( $args ), $known ) ) {
			return null;
		}

		if ( isset( $args['number'] ) && -1 !== (int) $args['number'] ) {
			return null;
		}

		/*
		 * Give_Payments_Query excludes child donations such as renewals by setting post_parent to 0, and
		 * add-ons like Recurring clear that through `give_pre_get_payments`. Give the hook the same query
		 * object it would see on the legacy path so the SQL path keeps the same rows.
		 */
		$query = new Give_Payments_Query( $args );
		$query->__set( 'post_parent', 0 );
		do_action( 'give_pre_get_payments', $query );

		$statuses = (array) ( $args['post_status'] ?? $args['status'] ?? 'publish' );
		$where    = ' AND p.post_status IN (' . implode( ',', array_map( static function ( $status ) use ( $wpdb ) {
			return $wpdb->prepare( '%s', $status );
		}, $statuses ) ) . ')';

		if ( isset( $query->args['post_parent'] ) && is_numeric( $query->args['post_parent'] ) ) {
			$where .= $wpdb->prepare( ' AND p.post_parent = %d', $query->args['post_parent'] );
		}

		if ( ! empty( $args['start_date'] ) && ! is_wp_error( $args['start_date'] ) ) {
			$where .= $wpdb->prepare( ' AND p.post_date >= %s', date( 'Y-m-d H:i:s', $args['start_date'] ) );
		}

		if ( ! empty( $args['end_date'] ) && ! is_wp_error( $args['end_date'] ) ) {
			$where .= $wpdb->prepare( ' AND p.post_date <= %s', date( 'Y-m-d H:i:s', $args['end_date'] ) );
		}

		$donation_id_col = Give()->payment_meta->get_meta_type() . '_id';
		$meta_conditions = [];

		if ( ! empty( $args['give_forms'] ) ) {
			$meta_conditions[] = [ 'key' => '_give_payment_form_id', 'value' => $args['give_forms'] ];
		}

		foreach ( (array) ( $args['meta_query'] ?? [] ) as $index => $condition ) {
			if ( 'relation' === $index ) {
				if ( 'AND' !== strtoupper( $condition ) ) {
					return null;
				}
				continue;
			}

			if ( ! is_array( $condition ) || empty( $condition['key'] ) || ! isset( $condition['value'] ) ) {
				return null;
			}

			if ( ! in_array( strtoupper( $condition['compare'] ?? '=' ), [ '=', 'IN' ], true ) ) {
				return null;
			}

			if ( array_diff( array_keys( $condition ), [ 'key', 'value', 'compare' ] ) ) {
				return null;
			}

			$meta_conditions[] = $condition;
		}

		foreach ( $meta_conditions as $condition ) {
			$values = array_map( 'strval', (array) $condition['value'] );

			if ( ! $values ) {
				return null;
			}

			$placeholders = implode( ',', array_fill( 0, count( $values ), '%s' ) );
			$where       .= $wpdb->prepare(
				" AND p.ID IN (SELECT {$donation_id_col} FROM {$wpdb->donationmeta} WHERE meta_key = %s AND meta_value IN ({$placeholders}))",
				array_merge( [ $condition['key'] ], $values )
			);
		}

		return $where;
	}

	/**
	 * Get the best selling forms
	 *
	 * @since  1.0
	 * @since 2.9.6 Added an explicit ORDER BY on which to apply a DESC order.
     * @since 2.18.0 Updated sales calculation so that the results are ordered as integers.
	 * @access public
	 * @global wpdb $wpdb
	 *
	 * @param       $number int The number of results to retrieve with the default set to 10.
	 *
	 * @return array       Best selling forms
	 */
	public function get_best_selling( $number = 10 ) {
		global $wpdb;

		$meta_table = give_v20_bc_table_details( 'form' );

		$give_forms = $wpdb->get_results(
			$wpdb->prepare(
				"SELECT {$meta_table['column']['id']} as form_id, max(ABS(meta_value)) as sales
				FROM {$meta_table['name']} WHERE meta_key='_give_form_sales' AND meta_value > 0
				GROUP BY meta_value+0
				ORDER BY sales DESC
				LIMIT %d;",
				$number
			)
		);

		return $give_forms;
	}

}