<?php
/** @noinspection PhpMissingReturnTypeInspection */
/** @noinspection RegExpSingleCharAlternation */

namespace WPDRMS\ASP\Hooks\Filters;

use WP_Post;
use WP_Query;
use WPDRMS\ASP\Query\SearchQuery;
use WPDRMS\Utils\Archive;
use WPDRMS\ASP\Utils\Search;
use WPDRMS\Utils\WooCommerce;

if ( !defined('ABSPATH') ) {
	die('-1');
}


class SearchOverride extends AbstractFilter {

	public static $override_count = 0;

	public function handle() {}

	/**
	 * Checks and cancels the original search query made by WordPress, if required
	 *
	 * @param string   $request - The SQL query
	 * @param WP_Query $wp_query - The instance of WP_Query() for this query
	 * @return string
	 * @noinspection PhpUnused
	 */
	public function maybeCancelWPQuery( string $request, WP_Query $wp_query ) {
		// is_home check is needed for the blog/home page filtering, query reset will break it
		if ( $this->checkSearchOverride(true, $wp_query) === true && !$wp_query->is_home ) {
			$request = '';
		}
		return $request;
	}

	/**
	 * Overrides the $posts object array with the results from Ajax Search Pro
	 *
	 * @param mixed              $posts array of posts
	 * @param null|WP_Query      $wp_query The instance of WP_Query() for this query
	 * @param 'posts'|'wp_query' $return_as to return the $posts or the modified $wp_query
	 * @return ($return_as is 'posts' ? WP_Post[] : WP_Query)
	 */
	public function override( $posts, ?WP_Query $wp_query, string $return_as = 'posts' ) {
		$check_override = $this->checkSearchOverride(false, $wp_query);
		if ( $check_override === false ) {
			return $posts;
		} else {
			$_p_id  = $check_override[0];
			$s_data = $check_override[1];
		}

		// The get_query_var() is malfunctioning in some cases!!! use $_GET['paged']
		// $paged = (get_query_var('paged') != 0) ? get_query_var('paged') : 1;
		if ( isset($_GET['asp_force_reset_pagination']) ) {
			// For the correct pagination highlight
			$_GET['paged'] = 1;
			$paged         = 1;
			set_query_var('paged', 1);
			set_query_var('page', 1);
		} else {
			if ( isset($_GET['paged']) ) {
				$paged = intval($_GET['paged']);
			} elseif (
				isset($wp_query->query_vars['offset'], $wp_query->query_vars['posts_per_page']) &&
				$wp_query->query_vars['posts_per_page'] > 0 &&
				$wp_query->query_vars['offset'] > 0
			) {
				$paged = floor($wp_query->query_vars['offset'] / $wp_query->query_vars['posts_per_page']) + 1;
			} elseif ( isset($wp_query->query_vars['paged']) ) {
				$paged = $wp_query->query_vars['paged'];
			} else {
				$paged = 1;
			}
		}

		if ( !wd_asp()->instances->exists($_p_id) ) {
			return $posts;
		}

		$instance = wd_asp()->instances->get($_p_id);
		$sd       = $instance['data'];
		// First check the asp_ls, as s might be set already
		$phrase = $_GET['asp_ls'] ?? $_GET['s'] ?? $wp_query->query_vars['s'] ?? '';

		$paged = $paged <= 0 ? 1 : $paged;

		// Elementor related
		if (
			isset($wp_query->query_vars, $wp_query->query_vars['posts_per_page']) &&
			$this->queryVarsHasAspOverridePostType($wp_query)
		) {
			$posts_per_page = $wp_query->query_vars['posts_per_page'];
		} else {
			$posts_per_page = $sd['results_per_page'];
		}
		if ( $posts_per_page === 'auto' ) {
			if ( isset($wp_query->query_vars, $wp_query->query_vars['posts_per_page']) ) {
				$posts_per_page = $wp_query->query_vars['posts_per_page'];
			} else {
				$posts_per_page = get_option( 'posts_per_page' );
			}
		}
		$posts_per_page = intval($posts_per_page);
		$posts_per_page = $posts_per_page === 0 ? 1 : $posts_per_page;

		$s_data = apply_filters('asp_search_override_data', $s_data, $posts, $wp_query, $_p_id, $phrase);

		// A possible exit point for the user, if he sets the _abort argument
		if ( isset($s_data['_abort']) ) {
			return $posts;
		}

		$args = array(
			's'              => $phrase,
			'_ajax_search'   => false,
			'posts_per_page' => $posts_per_page,
			'page'           => $paged,
		);

		$args = apply_filters(
			'asp/search/override/args',
			$args,
			array(
				'posts'          => $posts,
				'wp_query'       => $wp_query,
				'search_id'      => $_p_id,
				'phrase'         => $phrase,
				'override_count' => self::$override_count,
			),
		);

		// $args = self::getAdditionalArgs($args);
		add_filter('asp_query_args', array( $this, 'getAdditionalArgs' ));
		add_filter('asp_query_args', array( new DiviFilterGrid($wp_query), 'filterGridQueryArgs' ), 99, 1);

		if ( count($s_data) === 0 ) {
			$asp_query = new SearchQuery($args, $_p_id);
		} else {

			$asp_query = new SearchQuery($args, $_p_id, $s_data);
		}

		$res = $asp_query->posts;
		if ( $sd['result_suggestions_results_page'] && count($res) === 0 ) {
			$asp_query->resultsSuggestions();
			$res = $asp_query->posts;
		}

		do_action('asp_after_search', $phrase, $res, $_p_id);

		// Elementor Posts widget no results text
		if (
			count($res) === 0 &&
			isset($wp_query->query_vars, $wp_query->query_vars['is_elementor'])
		) {
			echo Search::generateHTMLResults(array(), false, $_p_id, $phrase, 'elementor');
		}

		$wp_query->query_vars['post__in'] = $wp_query->query_vars['post__in'] ?? array();
		if ( is_array($wp_query->query_vars['post__in']) ) {
			$wp_query->query_vars['post__in'] = array_unique(
				array_merge(
					$wp_query->query_vars['post__in'],
					array_map(
						function ( $aspr ) {
							return $aspr->id;
						},
						$asp_query->all_results
					)
				)
			);
			$wp_query->query_vars['orderby']  = 'post__in';
		}

		if ( isset($wp_query->posts) ) {
			$wp_query->posts = $res;
		}
		$wp_query->post_count = count($res);

		/**
		 * Reset post type
		 * Very important. Some archive posts modules fail because asp_override is non-existent.
		 */
		if ( empty($wp_query->query_vars['post_type']) || $this->queryVarsHasAspOverridePostType( $wp_query ) ) {
			$wp_query->query_vars['post_type'] = $asp_query->args->post_type;
		}

		$wp_query->found_posts = $asp_query->found_posts;
		if ( ( $wp_query->found_posts / $posts_per_page ) > 1 ) {
			$wp_query->max_num_pages = intval(ceil($wp_query->found_posts / $posts_per_page));
		} else {
			$wp_query->max_num_pages = 1;
		}

		$res      = apply_filters('asp/search/override/results', $res, $posts, $wp_query, $asp_query);
		$wp_query = apply_filters('asp/search/override/wp_query', $wp_query, $res, $posts, $asp_query);

		++self::$override_count;
		if ( $return_as === 'posts' ) {
			return $res;
		} else {
			return $wp_query;
		}
	}

	/**
	 * Checks and gets additional arguments for the override query
	 *
	 * @param $args - query arguments for the SearchQuery()
	 */
	public static function getAdditionalArgs( $args ) {
		global $wpdb, $wp_query;

		// WooCommerce price filter
		if ( isset($_GET['min_price'], $_GET['max_price']) ) {
			$args['post_meta_filter'][] = array(
				'key'      => '_price',         // meta key
				'value'    => array( floatval($_GET['min_price']), floatval($_GET['max_price']) ),
				'operator' => 'BETWEEN',
			);
		}

		// WooCommerce or other custom Ordering
		if ( isset($_GET['orderby']) || isset($_GET['product_orderby']) ) {
			$o_by = $_GET['orderby'] ?? $_GET['product_orderby'];
			$o_by = str_replace(' ', '', ( strtolower($o_by) ));
			if ( isset($_GET['order']) || isset($_GET['product_order']) ) {
				$o_way = $_GET['order'] ?? $_GET['product_order'];
			} elseif ( $o_by == 'price' || $o_by == 'product_price' ) {
				$o_way = 'ASC';
			} elseif ( $o_by == 'alphabetical' ) {
				$o_way = 'ASC';
			} else {
				$o_way = 'DESC';
			}
			$o_way = strtoupper($o_way);
			if ( $o_way != 'DESC' && $o_way != 'ASC' ) {
				$o_way = 'DESC';
			}
			switch ( $o_by ) {
				case 'id':
				case 'post_id':
				case 'product_id':
					$args['post_primary_order'] = "id $o_way";
					break;
				case 'popularity':
				case 'post_popularity':
				case 'product_popularity':
					$args['post_primary_order']          = "customfp $o_way";
					$args['post_primary_order_metatype'] = 'numeric';
					$args['_post_primary_order_metakey'] = 'total_sales';
					break;
				case 'rating':
				case 'post_rating':
				case 'product_rating':
					// Custom query args here
					$args['cpt_query']['fields']  = "(
                            SELECT
                                IF(AVG( $wpdb->commentmeta.meta_value ) IS NULL, 0, AVG( $wpdb->commentmeta.meta_value ))
                            FROM
                                $wpdb->comments
                                LEFT JOIN $wpdb->commentmeta ON($wpdb->comments.comment_ID = $wpdb->commentmeta.comment_id)
                            WHERE
                                $wpdb->posts.ID = $wpdb->comments.comment_post_ID
                                AND ( $wpdb->commentmeta.meta_key = 'rating' OR $wpdb->commentmeta.meta_key IS null )
                        ) as average_rating, ";
					$args['cpt_query']['orderby'] = "average_rating $o_way, ";

					// Force different field order for index table
					$args['post_primary_order'] = "average_rating $o_way";
					break;
				case 'date':
				case 'post_date':
				case 'product_date':
					$args['post_primary_order'] = "post_date $o_way";
					break;
				case 'name':
				case 'post_name':
				case 'product_name':
				case 'alphabetical':
				case 'reverse_alpha':
				case 'reverse_alphabetical':
					$args['post_primary_order'] = "post_title $o_way";
					break;
				case 'price':
				case 'product_price':
				case 'price-desc':
					$args['post_primary_order']          = "customfp $o_way";
					$args['post_primary_order_metatype'] = 'numeric';
					$args['_post_primary_order_metakey'] = '_price';
					break;
				case 'relevance':
					$args['post_primary_order'] = "relevance $o_way";
					break;
			}
		}

		if ( isset($_GET['post_type']) && $_GET['post_type'] == 'product' ) {
			$args['search_type'] = array( 'cpt' );
			$old_ptype           = $args['post_type'];
			$args['post_type']   = array();
			if ( in_array('product', $old_ptype) ) {
				$args['post_type'][] = 'product';
			}
			if ( in_array('product_variation', $old_ptype) ) {
				$args['post_type'][] = 'product_variation';
			}

			// Exclude from search products
			$visibility_term = get_term_by('slug', 'exclude-from-search', 'product_visibility');

			if ( $visibility_term !== false ) {
				$found = false;
				if ( isset($args['post_tax_filter']) ) {
					foreach ( $args['post_tax_filter'] as &$filter ) {
						if ( $filter['taxonomy'] == 'product_visibility' ) {
							$filter['exclude'] = $filter['exclude'] ?? array();
							$filter['exclude'] = array_merge($filter['exclude'], array( $visibility_term->term_id ));
							$found             = true;
							break;
						}
					}
				}
				if ( !$found ) {
					$args['post_tax_filter'][] = array(
						'taxonomy'    => 'product_visibility',
						'include'     => array(),
						'exclude'     => array( $visibility_term->term_id ),
						'allow_empty' => true,
					);
				}
			}
		}

		// Archive pages
		if ( isset($_GET['asp_ls']) && !isset($wp_query->query_vars['asp_not_archive']) ) {
			if ( WooCommerce::isShop() ) {
				$args['search_type'] = array( 'cpt' );
				// Only allow product or product variations if selected
				$args['post_type'] = in_array('product_variation', $args['post_type']) ? array( 'product', 'product_variation' ) : array( 'product' );
				// Exclude from catalog products
				$visibility_term = get_term_by('slug', 'exclude-from-search', 'product_visibility');
				$catalog_term    = get_term_by('slug', 'exclude-from-catalog', 'product_visibility');
				if ( $visibility_term !== false && $catalog_term !== false ) {
					$found = false;
					if ( isset($args['post_tax_filter']) ) {
						foreach ( $args['post_tax_filter'] as &$filter ) {
							if ( $filter['taxonomy'] == 'product_visibility' ) {
								$filter['exclude'] = $filter['exclude'] ?? array();
								$filter['exclude'] = array_merge(
									$filter['exclude'],
									array(
										$visibility_term->term_id,
										$catalog_term->term_id,
									)
								);
								$found             = true;
								break;
							}
						}
					}
					if ( !$found ) {
						$args['post_tax_filter'][] = array(
							'taxonomy'    => 'product_visibility',
							'include'     => array(),
							'exclude'     => array(
								$visibility_term->term_id,
								$catalog_term->term_id,
							),
							'allow_empty' => true,
						);
					}
				}
			} elseif ( Archive::isTaxonomyArchive() ) {
				$args['search_type'] = array( 'cpt' );
				$found               = false;
				$children            = get_term_children(get_queried_object()->term_id, get_queried_object()->taxonomy);
				$include             = !is_wp_error($children) ? array_merge(array( get_queried_object()->term_id ), $children) : array( get_queried_object()->term_id );
				if ( isset($args['post_tax_filter']) ) {
					foreach ( $args['post_tax_filter'] as &$filter ) {
						if ( $filter['taxonomy'] == get_queried_object()->taxonomy ) {
							$filter['include'] = array_diff($include, $filter['exclude']);
							$found             = true;
							break;
						}
					}
				}
				if ( !$found ) {
					$args['post_tax_filter'][] = array(
						'taxonomy'    => get_queried_object()->taxonomy,
						'include'     => $include,
						'exclude'     => array(),
						'allow_empty' => true,
					);
				}
			} elseif ( Archive::isPostTypeArchive() ) {
				$args['search_type'] = array( 'cpt' );
				$post_type           = $wp_query->get( 'post_type' );
				$args['post_type']   = $post_type == '' ? array( 'post' ) : array( $post_type );
			}
		}
		return $args;
	}

	/**
	 * Checks if the default WordPress search query is executed right now, and if it needs an override.
	 * Also sets some cookie and request variables, if needed.
	 *
	 * @param bool          $check_only - when true, only checks if the override should be initiated, no variable changes
	 * @param null|WP_Query $wp_query - The instance of WP_Query() for this query
	 * @return array|bool
	 */
	public function checkSearchOverride( bool $check_only, ?WP_Query $wp_query ) {
		// Check the search query
		if ( !$this->isSearch($wp_query) ) {
			return false;
		}
		// If you get method is used, then the cookies are not present or not used
		if ( isset($_GET['p_asp_data']) ) {
			if ( $check_only ) {
				return true;
			}
			$_p_id = $_GET['p_asid'] ?? $_GET['np_asid'];
			if ( $_GET['p_asp_data'] == 1 ) {
				$s_data = $_GET;
			} else {
				// Legacy support
				parse_str(base64_decode($_GET['p_asp_data']), $s_data);
			}
		} elseif (
			isset($_GET['s'], $_COOKIE['asp_data']) && (
				( $_GET['s'] != '' && isset($_COOKIE['asp_phrase']) && $_COOKIE['asp_phrase'] == $_GET['s'] ) ||
				( $_GET['s'] == '' )
			)
		) {
			if ( $check_only ) {
				return true;
			}
			parse_str($_COOKIE['asp_data'], $s_data);
			$_POST['np_asp_data'] = $_COOKIE['asp_data'];
			$_POST['np_asid']     = $_COOKIE['asp_id'];
			$_p_id                = $_COOKIE['asp_id'];
		} elseif (
			isset($wp_query, $wp_query->query_vars['asp_id'])
		) {
			if ( $check_only ) {
				return true;
			}
			$_p_id  = $wp_query->query_vars['asp_id'];
			$s_data = array();
		} else {
			// Probably the search results page visited via URL, not triggered via search bar
			if ( isset($_GET['post_type']) && $_GET['post_type'] == 'product' ) {
				$override_id = get_option('asp_woo_override', -1);
			} else {
				$override_id = get_option('asp_st_override', -1);
			}
			if ( $override_id > -1 && wd_asp()->instances->exists( $override_id ) ) {
				$inst = wd_asp()->instances->get( $override_id );
				if ( $inst['data']['override_default_results'] == 1 ) {
					return array( $override_id, array() );
				}
			}

			// Something is not right
			return false;
		}

		return array( $_p_id, $s_data );
	}

	public function isSearch( ?WP_Query $wp_query ): bool {
		$is_search  = true;
		$soft_check =
			defined('ELEMENTOR_VERSION') || // Elementor
			defined( 'ET_CORE' ) || // Divi
			wd_asp()->o['asp_compatibility']['query_soft_check'];

		// This can't be a search query if none of this is set
		if ( !isset($wp_query, $wp_query->query_vars, $_GET['s']) ) {
			$is_search = false;
		} else {
			// Possible candidates for search below
			if ( $soft_check ) {
				// In soft check mode, it does not have to be the main query
				if ( !$wp_query->is_search() ) {
					$is_search = false;
				}
			} elseif ( !$wp_query->is_search() || !$wp_query->is_main_query() ) {
				$is_search = false;
			}
			if ( !$is_search && isset($wp_query->query_vars['aps_title']) ) {
				$is_search = true;
			}
		}

		// GEO directory search, do not override
		if ( $is_search && isset($_GET['geodir_search']) ) {
			$is_search = false;
		}

		// Elementor or other forced override
		// Only allow override once in this case, otherwise duplicates will appear
		// @ticket: https://wp-dreams.com/forums/topic/filtered-results-show-duplicate-posts/
		if ( $this->queryVarsHasAspOverridePostType($wp_query) && self::$override_count === 0 ) {
			$is_search = true;
		}

		// Archive pages
		if ( isset($_GET['asp_ls'], $_GET['p_asid']) && $wp_query->is_main_query() ) {
			if (
				( wd_asp()->instances->getOption($_GET['p_asid'], 'woo_shop_live_search') && WooCommerce::isShop() ) ||
				( wd_asp()->instances->getOption($_GET['p_asid'], 'cpt_archive_live_search') && Archive::isPostTypeArchive() ) ||
				( wd_asp()->instances->getOption($_GET['p_asid'], 'taxonomy_archive_live_search') && Archive::isTaxonomyArchive() ) ) {
				$is_search = true;
			}
		}

		// Is this the admin area?
		if ( $is_search && is_admin() ) {
			$is_search = false;
		}

		// Forced non-override
		if ( isset($wp_query->query_vars['asp_override']) ) {
			$is_search = (bool) $wp_query->query_vars['asp_override'];
		}

		// Possibility to add exceptions
		return (bool) apply_filters('asp_query_is_search', $is_search, $wp_query);
	}

	/**
	 * Fixes the non-live result URLs for generic themes
	 *
	 * @param $url
	 * @param $post
	 * @return mixed
	 * @noinspection PhpUnused
	 */
	public function fixUrls( $url, $post ) {
		if ( isset($post->asp_data, $post->asp_data->link) ) {
			return $post->asp_data->link;
		} elseif ( isset($post->asp_guid) ) {
			return $post->asp_guid;
		}
		return $url;
	}

	/**
	 * Fixes the URLs of the non-live search results, when using the Genesis Framework
	 *
	 * @param $output
	 * @param $wrap
	 * @param $title
	 * @return mixed
	 * @noinspection PhpUnused
	 */
	public function fixUrlsGenesis( $output, $wrap, $title ) {
		global $post;

		if ( isset($post, $post->asp_guid) && is_object($post) && function_exists('genesis_markup') ) {
			$pattern = "/(?<=href=(\"|'))[^\"']+(?=(\"|'))/";
			$title   = preg_replace($pattern, $post->asp_guid, $title);

			$output = genesis_markup(
				array(
					'open'    => "<{$wrap} %s>",
					'close'   => "</{$wrap}>",
					'content' => $title,
					'context' => 'entry-title',
					'params'  => array(
						'wrap' => $wrap,
					),
					'echo'    => false,
				)
			);
		}

		return $output;
	}

	private function queryVarsHasAspOverridePostType( ?WP_Query $wp_query ): bool {
		if ( $wp_query === null || !isset($wp_query->query_vars['post_type']) ) {
			return false;
		}

		if ( is_array($wp_query->query_vars['post_type']) ) {
			return in_array('asp_override', $wp_query->query_vars['post_type']);
		} else {
			return $wp_query->query_vars['post_type'] === 'asp_override';
		}
	}
}
