<?php
/**
 * Class for saving/processing tour booking periods.
 *
 * @author    Themedelight
 * @package   Themedelight/AdventureTours
 * @version   2.4.1
 */

class AtTourBookingService extends TdComponent
{
	/**
	 * Determines what post meta key should be used for the tour booking periods storing in DB.
	 *
	 * @var string
	 */
	public $meta_key = 'tour_booking_periods';

	/**
	 * Limit that prevents too long looping during period expanding into set of days
	 * for that booking is available.
	 *
	 * @see expand_period method.
	 * @var integer
	 */
	public $expandIterationsLimit = 5000;

	/**
	 * List of order statuses booking for that is active (order statuses missed in this list will be considered as declined/inactive).
	 *
	 * @see get_booking_records
	 * @var array
	 */
	public $order_statuses_with_active_tour_booking = array(
		'wc-pending',
		'wc-processing',
		'wc-on-hold',
		'wc-completed',
		// 'wc-cancelled','wc-refunded','wc-failed',
	);

	/**
	 * If limits for same days from different period should be summarized or they should be overwritten by more recent period.
	 * 
	 * @var boolean
	 */
	public $sum_limits = false;

	/**
	 * Returns set of the saved periods related to the specefied period.
	 *
	 * @param  int $post_id
	 * @return array
	 */
	public function get_rows( $post_id ) {
		if ( $post_id > 0 && $this->meta_key ) {
			$rows = get_post_meta( $post_id, $this->meta_key, true );
		} else {
			$rows = null;
		}

		return $rows ? $rows : array();
	}

	/**
	 * Saves new period set to DB.
	 *
	 * @param int     $post_id
	 * @param assoc   $periods
	 * @param boolean $validate
	 * @return assoc  errors hppaned during validation
	 */
	public function set_rows( $post_id, $periods, $validate = true ) {
		$validation_errors = array();
		if ( $post_id < 1 || ! $this->meta_key ) {
			$validation_errors['general'] = esc_html__( 'Parametes errors. Please contact support', 'adventure-tours' );
			return $validation_errors;
		}
		if ( ! $periods ) {
			delete_post_meta( $post_id, $this->meta_key );
		} else {
			if ( $validate ) {
				foreach( $periods as $index => $period_data ) {
					$period_errors = $this->check_period_data( $period_data );
					if ( $period_errors ) {
						$validation_errors[ $index ] = $period_errors;
					}
				}
				if ( $validation_errors ) {
					return $validation_errors;
				}
			}

			update_post_meta( $post_id, $this->meta_key, $this->format_periods( $periods ) );
		}

		return $validation_errors;
	}

	protected function format_periods( $periods ) {
		if ( $periods ) {
			foreach ( $periods as $period_index => &$period ) {
				if ( ! isset( $period['mode'] ) ) {
					$period['mode'] = 'default';
				}

				if ( 'exact-dates' == $period['mode'] ) {
					// need set up from and to fields to be compatible for period intersation function
					if ( empty( $period['exact_dates'] ) ) {
						unset( $periods[ $period_index ] );
						continue;
					}

					$from = null;
					$to = null;
					foreach ( $period['exact_dates'] as $date ) {
						if ( null == $from ) {
							$from = $to = $date;
						}

						if ( $date < $from ) {
							$from = $date;
						} elseif ( $date > $to ) {
							$to = $date;
						}
					}

					$period['from'] = date('Y-m-d', strtotime( $from ) );
					$period['to'] = date('Y-m-d', strtotime( $to ) );
				} else {
					$period['from'] = date('Y-m-d', strtotime( $period['from'] ) );
					$period['to'] = date('Y-m-d', strtotime( $period['to'] ) );
				}
			}
		}

		return $periods;
	}

	/**
	 * Returns day set for booking for that is avaliable.
	 *
	 * @param  int     $post_id
	 * @param  string  $from_date              allows limit range of days that should be involved.
	 * @param  string  $to_date                allows limit range of days that should be involved.
	 * @param  boolean $exclude_booked_tickets if already booked tickets should be taken in consideration.
	 * @return array
	 */
	public function get_expanded( $post_id, $from_date = null, $to_date = null, $exclude_booked_tickets = true ) {
		$rows = $this->get_periods( $post_id, $from_date, $to_date );
		return $this->expand_periods(
			$rows,
			$exclude_booked_tickets ? $post_id : 0,
			$from_date,
			$to_date
		);
	}

	/**
	 * Returns set of special price rules based on day basis.
	 *
	 * @param  int    $post_id
	 * @param  string $from_date allows limit range of days that should be involved.
	 * @param  string $to_date   allows limit range of days that should be involved.
	 * @return assoc
	 */
	public function get_expanded_price_rules( $post_id, $from_date = null, $to_date = null ) {
		$rows = $this->get_periods( $post_id, $from_date, $to_date );
		return $this->expand_periods(
			$rows,
			0,
			$from_date,
			$to_date,
			true
		);
	}

	/**
	 * Returns special price rule for specific date.
	 * @param  int    $post_id
	 * @param  string $date
	 * @return string|null
	 */
	public function get_date_price_rule( $post_id, $date ) {
		$rules = $this->get_expanded_price_rules( $post_id, $date, $date );
		if ( $rules ) {
			return isset( $rules[ $date ] ) ? $rules[ $date ] : null;
		}
		return null;
	}

	public function calculate_date_price( $post_id, $date, $price ) {
		$rule = $this->get_date_price_rule( $post_id, $date );
		if ( $rule ) {
			$spec_price_service = $this->get_special_price_service();
			if ( $spec_price_service ) {
				return $spec_price_service->process_rule( $rule, $price );
			}
		}
		return $price;
	}

	public function get_special_price_service() {
		return adventure_tours_di( 'tour_special_price_service' );
	}

	/**
	 * Expands passed periods into set of dates with available for booking tickets number.
	 *
	 * @param  array   $periods
	 * @param  integer $exclude_for_tour_id tour id booking for that should be deducted from expanded periods.
	 * @param  string  $from_date           allows limit range of days that should be involved.
	 * @param  string  $to_date             allows limit range of days that should be involved.
	 * @param  boolean $price_rules         if special price rules should be expanded instead of limit option value.
	 * @return array
	 */
	public function expand_periods( $periods, $exclude_for_tour_id = 0, $from_date = null, $to_date = null, $price_rules = false ) {
		$result = array();

		if ( $periods ) {
			foreach ( $periods as $period ) {
				$expandedDays = $this->expand_period( $period, $price_rules );
				if ( $expandedDays ) {
					if ( ! $price_rules && $this->sum_limits ) {
						foreach ( $expandedDays as $time => $new_limit_value ) {
							if ( isset($result[ $time ]) ) {
								$result[ $time ] = $result[ $time ] + $new_limit_value;
							} else {
								$result[ $time ] = $new_limit_value;
							}
						}
					} else {
						$result = array_merge( $result, $expandedDays );
					}
				}
			}

			if ( $result && $exclude_for_tour_id > 0 && ! $price_rules ) {
				$booked_tickets = $this->get_booking_data( $exclude_for_tour_id, $from_date, $to_date );
				if ( $booked_tickets ) {
					foreach ( $booked_tickets as $booking_date => $qnt ) {
						if ( isset( $result[$booking_date] ) ) {
							$result[$booking_date] -= $qnt;
							if ( $result[$booking_date] < 1 ) {
								unset( $result[$booking_date] );
							}
						}
					}
				}
			}

			if ( $result && ! $price_rules ) {
				foreach( $result as $booking_date => $qnt ) {
					if ( $qnt < 1 ) {
						unset( $result[ $booking_date ] );
					}
				}
			}
		}

		return $result;
	}

	/**
	 * Returns number of the open tickets for specific tour for specific date.
	 *
	 * @param  int    $tour_id
	 * @param  string $date
	 * @param  string $variation_id
	 * @return int
	 */
	public function get_open_tickets( $tour_id, $date, $variation_id = 0 ) {
		$allowed_dates = $tour_id && $date ? $this->get_expanded( $tour_id, $date, $date ) : array();

		if ( $allowed_dates && isset( $allowed_dates[ $date ] ) ) {
			return $allowed_dates[ $date ];
		} else {
			return 0;
		}
	}

	/**
	 * Returns set of periods related to the selected tour (and some specific period).
	 *
	 * @param  int    $post_id
	 * @param  string $from_date allows limit range of days that should be involved.
	 * @param  string $to_date allows limit range of days that should be involved.
	 * @return array
	 */
	public function get_periods( $post_id, $from_date = null, $to_date = null ) {
		$rows = $this->get_rows( $post_id );

		if ( $rows && ( $from_date || $to_date ) ) {
			$result = array();
			foreach ( $rows as $period ) {
				$newSet = $this->get_inersected_period( $period, array(
					'from' => $from_date,
					'to' => $to_date,
				) );

				// excluding exact dates that out of new range
				if ( $newSet && ! empty( $period['exact_dates'] ) && ! empty( $period['mode'] ) && 'exact-dates' == $period['mode'] ) {
					$new_from = $newSet['from'] > $this->to_time( $period['from'], true ) ? $newSet['from'] : null;
					$new_to = $newSet['to'] < $this->to_time( $period['to'], true ) ? $newSet['to'] : null;
					if ( $new_from || $new_to ) {
						$new_dates_set = array();
						foreach( $newSet['exact_dates'] as $_dateIndex => $_ticketDate ) {
							$_ticketTimestamp = $this->to_time( $_ticketDate, true );
							if ( ( $new_from && $new_from > $_ticketTimestamp ) || ( $new_to && $new_to < $_ticketTimestamp ) ) {
								// date is out of new range
							} else {
								$new_dates_set[] = $_ticketDate;
							}
						}

						if ( ! $new_dates_set ) {
							$newSet = null; // all dates out of new range
						} else {
							$newSet['exact_dates'] = $new_dates_set;
						}
					}
				}

				if ( $newSet ) {
					$result[] = $newSet;
				}
			}
			return $result;
		}

		return $rows;
	}

	/**
	 * Converts passed period into a set of days for that booking is available.
	 *
	 * @param  assoc  $period
	 * @return array
	 */
	public function expand_period( $period, $price_rules = false ) {
		$result = array();

		if ( !isset( $period['type'] ) || $period['type'] != '1' ) {
			return $result;
		}

		$type = isset( $period['mode'] ) ? $period['mode'] : 'default';

		$expanded_value = '1';
		if ( $price_rules ) {
			$expanded_value = isset( $period['spec_price'] ) ? $period['spec_price'] : '';
		} else if ( isset( $period['limit'] ) ) {
			$expanded_value = $period['limit'];
		}

		switch ( $type ) {
		case 'exact-dates':
			$result = $this->expand_period_exact_dates( $period, $expanded_value );
			break;

		case 'default':
		default:
			$result = $this->expand_period_default( $period, $expanded_value );
			break;
		}

		return $result;
	}

	protected function expand_period_exact_dates( $period, $expanded_value ) {
		$result = array();

		$times = !empty( $period['times'] ) ? $period['times'] : array();

		if ( ! empty( $period['exact_dates'] ) ) {
			foreach ( $period['exact_dates'] as $_date ) {
//TODO improve!
				$this->expand_times( $result, $_date, $expanded_value, $times );
			}
		}

		return $result;
	}

	protected function expand_period_default( $period, $expanded_value, $step = '+1 day' ) {
		$result = array();

		$iterationsLimit = $this->expandIterationsLimit > 1 ? $this->expandIterationsLimit : 5000;

		$curTime = $this->to_time( $period['from'] );
		$endTime = $this->to_time( $period['to'] );

		$times = !empty( $period['times'] ) ? $period['times'] : array();

		if ( $curTime && $endTime && $curTime <= $endTime ) {
			$allowedDays = ! empty( $period['days'] ) ? $period['days'] : array();
			while ( $curTime <= $endTime && $iterationsLimit ) {
				$nDay = date( 'D', $curTime );
				if ( in_array( $nDay, $allowedDays ) ) {
					$this->expand_times( $result, date( 'Y-m-d', $curTime ), $expanded_value, $times );
				}
				$curTime = strtotime( $step, $curTime );
				$iterationsLimit--;
			}
		}

		return $result;
	}

	protected function expand_times( array &$expand_to, $date, $expanded_value, $times ) {
		if ( $times ) {
			foreach ( $times as $time ) {
				$expand_to[ $date . ' ' . $time ] = $expanded_value;
			}
		} else {
			$expand_to[ $date ] = $expanded_value;
		}
	}

	/**
	 * Makes intersected period based on dates from period and restriction period.
	 *
	 * @param  assoc $p1
	 * @param  assoc $p2
	 * @return assoc
	 */
	public function get_inersected_period($p1, $p2) {
		$s1 = $this->to_time( $p1['from'], true );
		$e1 = $this->to_time( $p1['to'], true );

		$s2 = $this->to_time( $p2['from'], true );
		$e2 = $this->to_time( $p2['to'], true );

		if ( $s1 && $e1 && $s2 && $e2 ) {
			if ( $s2 <= $e1 && $e2 >= $s1 ) {
				return array_merge( $p1, array(
					'from' => max( $s1, $s2 ),
					'to' => min( $e1, $e2 ),
				) );
			}
		}
		return null;
	}

	/**
	 * Converts sting date presentation into timestamp.
	 *
	 * @param  string|int $date_string date string or timestamp
	 * @param  boolean    $trim_time   set to true if time defenition should be ignored
	 * @return int
	 */
	public function to_time( $date_string, $trim_time = false ) {
		if ( ! $date_string ) {
			return null;
		}
		if ( is_int( $date_string ) ) {
			return $date_string;
		}
		if ( $trim_time ) {
			$date_string = preg_replace('/ \d{2}:\d{2}$/', '', $date_string );
		}

		return strtotime( $date_string );
	}

	/**
	 * Returns data related the tour booking.
	 *
	 * @param  int    $tour_id
	 * @param  string $from_date optional, allow restict search timeframe.
	 * @param  string $to_date   optional, allow restict search timeframe.
	 * @return assoc
	 */
	public function get_booking_data( $tour_id, $from_date = null, $to_date = null ) {

		$tour_aliases = array();
		if ( $tour_id && adventure_tours_check( 'is_wpml_in_use' ) ) {
			// retriving all translation post ids
			// as different product_id may be saved in order_item if default language has been changed
			global $sitepress;
			$translations = $sitepress->get_element_translations( $sitepress->get_element_trid( $tour_id ) );
			if ( $translations && count( $translations ) > 1 ) {
				foreach ($translations as $lang_code => $translation ) {
					$tour_aliases[] = $translation->element_id;
				}
			}
		}

		$records_set = $this->get_booking_records( $tour_aliases ? $tour_aliases : $tour_id, $from_date, $to_date );

		$result = array();
		if ( $records_set ) {
			foreach ( $records_set as $order_report ) {
				$cur_tour_id = $order_report['tour_id'];
				// to "compound" all translation sales in a single set checking if tour belongs to our translations
				$save_as = $tour_aliases && in_array( $cur_tour_id, $tour_aliases ) ? $tour_id : $cur_tour_id;
				$date = $order_report['booking_date'];
				$qty = $order_report['qty'];
				if ( ! isset( $result[ $save_as ] ) ) {
					$result[ $save_as ] = array();
				}
				if ( ! empty( $result[ $save_as ][$date] ) ) {
					$result[ $save_as ][ $date ] += $qty;
				} else {
					$result[ $save_as ][ $date ] = $qty;
				}
			}
		}

		if ( $tour_id ) {
			return isset( $result[$tour_id] ) ? $result[$tour_id] : array();
		}

		return $result;
	}

	/**
	 * Retrives set of records about the booking events from DB.
	 *
	 * @param  int    $tour_id   optional, allow restrict loaded set with 1 specific tour.
	 * @param  string $from_date optional, allow restict search timeframe.
	 * @param  string $to_date   optional, allow restict search timeframe.
	 * @return array
	 */
	public function get_booking_records( $tour_id = null, $from_date = null, $to_date = null ) {
		//improve filtering for booking records loading
		global $wpdb;
		$dates_condition = '';
		if ( $from_date || $to_date ) {
			if ( $from_date && $to_date ) {
				if ( $from_date == $to_date ) {
					$dates_condition = " AND im.meta_value = '{$from_date}'";
				} else {
					// make convertation to date?
					//$dates_condition = 'im.date BETWEEN "{$from_date}" AND "{$to_date}"';
				}
			} else if ( $from_date ) {
				// make convertation to date?
			} else if ( $to_date ) {
				// make convertation to date?
			}
		}

		$tour_condition = '';
		if ( $tour_id ) {
			if ( is_array( $tour_id ) ) {
				$tour_condition = ' AND pidmeta.meta_value IN (' .
						join( ',', array_map( 'intval', $tour_id ) ) .
					')';
			} else {
				$tour_condition = ' AND pidmeta.meta_value = ' . intval( $tour_id );
			}
		}

		$tour_date_meta_key = 'tour_date';
		$product_id_meta_key = '_product_id';
		$qty_meta_key = '_qty';

		$status_condition = $this->order_statuses_with_active_tour_booking ? " AND o.post_status IN ('" . join( "','", $this->order_statuses_with_active_tour_booking ) . "')" : '';

		// $wpdb->prepare( 
		$query = "SELECT i.order_id, i.order_item_id, pidmeta.meta_value as tour_id, qntmeta.meta_value as qty, im.meta_value as booking_date, o.post_status as order_status 
			FROM `{$wpdb->prefix}woocommerce_order_itemmeta` im
			RIGHT JOIN `{$wpdb->prefix}woocommerce_order_items` i on im.order_item_id = i.order_item_id
			RIGHT JOIN `{$wpdb->prefix}posts` o on i.order_id = o.ID
			RIGHT JOIN `{$wpdb->prefix}woocommerce_order_itemmeta` pidmeta on pidmeta.order_item_id = im.order_item_id AND pidmeta.meta_key = '{$product_id_meta_key}'
			RIGHT JOIN `{$wpdb->prefix}woocommerce_order_itemmeta` qntmeta on qntmeta.order_item_id = im.order_item_id AND qntmeta.meta_key = '{$qty_meta_key}'
			WHERE im.meta_key = '{$tour_date_meta_key}'"
				. $tour_condition
				. $status_condition
				. $dates_condition;

		return $wpdb->get_results( $query, ARRAY_A );
	}

	/**
	 * Validation function. Checks single period data for errors and returns assoc with field errors.
	 *
	 * @param  assoc $data period fileds.
	 * @return assoc
	 */
	public function check_period_data( $data ) {
		$mode = !empty( $data['mode'] ) ? $data['mode'] : 'default';

		$errors = array();
		$from = null;
		$to = null;

		$keys = array(
			'mode',
			'limit',
			'spec_price',
		);

		if ( $mode == 'exact-dates' ) {
			$keys[] = 'exact_dates';
		} else {
			$keys = array_merge( $keys, array( 'from', 'to', 'days' ) );

			$from = !empty( $data['from'] ) ? $this->to_time( $data['from'] ) : null;
			$to = !empty( $data['to'] ) ? $this->to_time( $data['to'] ) : null;
		}

		if ( !empty( $data['times'] ) ) {
			$keys[] = 'times';
		}

		foreach ( $keys as $field_key ) {
			$field_errors = array();
			$value = isset( $data[ $field_key ] ) ? $data[ $field_key ] : null;

			if ( empty( $value ) ) {
				if ( 'days' == $field_key ) {
					$field_errors[] = esc_html__( 'Please select at least one day.', 'adventure-tours' );
				} else {
					$field_errors[] = esc_html__( 'The field is required.', 'adventure-tours' );
				}
				if ( 'limit' == $field_key && '0' == $value ) {
					continue;
				}
				if ( 'spec_price' == $field_key ) {
					continue;
				}
			} else {
				switch( $field_key ) {
				case 'limit':
					if ( $value < 1 ) {
						$field_errors[] = esc_html__( 'Minimum allowed value is 1.', 'adventure-tours' );
					}
					break;

				case 'from':
					if ( ! $from ) {
						$field_errors[] = esc_html__( 'Please check the date format.', 'adventure-tours' );
					}
					break;

				case 'to':
					if ( ! $to ) {
						$field_errors[] = esc_html__( 'Please check the date format.', 'adventure-tours' );
					} elseif ( $from && $to < $from ) {
						$field_errors[] = sprintf( esc_html__( 'The date should be grater than %s.', 'adventure-tours' ), $data['from'] );
					}
					break;

				case 'exact_dates':
					if ( ! is_array( $value ) ) {
						$field_errors[] = esc_html__( 'Parameters error. Please contact support.', 'adventure-tours' );
					} else {
						foreach ( $value as $_index => $_date ) {
							if ( ! $_date ) {
								$field_errors[] = sprintf(
									'#%s: ' . esc_html__( 'The field is required.', 'adventure-tours' ),
									$_index + 1
								);
							} else {
								$_time = $this->to_time( $_date );
								if ( ! $_time ) {
									$field_errors[] = sprintf(
										'#%s: ' . esc_html__( 'Please check the date format.', 'adventure-tours' ),
										$_index + 1
									);
								}
							}
						}
					}
					break;

				case 'times':
					if ( ! is_array( $value ) ) {
						$field_errors[] = esc_html__( 'Parameters error. Please contact support.', 'adventure-tours' );
					} else {
						foreach ( $value as $_index => $_time ) {
							if ( ! $_time ) {
								$field_errors[] = sprintf(
									'#%s: ' . esc_html__( 'The field is required.', 'adventure-tours' ),
									$_index + 1
								);
							} else {
								$is_valid_time = false;
								$parts = preg_match( '/^\d{2}:\d{2}$/', $_time ) ? explode( ':', $_time ) : array();
								if ( $parts ) {
									if ( count( $parts ) == 2 ) {
										if ( $parts[0] >= 0 && $parts[1] >= 0 && $parts[0] < 24 && $parts[1] < 60 ) {
											$is_valid_time = true;
										}
									}
								}
								if ( ! $is_valid_time ) {
									$field_errors[] = sprintf(
										'#%s: ' . esc_html__( 'Please check the time format.', 'adventure-tours' ),
										$_index + 1
									);
								}
							}
						}
					}
					break;

				case 'spec_price':
					$spec_price_service = $this->get_special_price_service();
					if ( $spec_price_service && ! $spec_price_service->is_valid_rule( $value ) ) {
						$field_errors[] = esc_html__( 'Please check the format.', 'adventure-tours' );
					}
					break;
				}
			}
			if ( $field_errors ) {
				$errors[ $field_key ] = $field_errors;
			}
		}

		return $errors;
	}
}