<?php

function flatsome_live_search_script() {
	global $extensions_uri;
	$theme   = wp_get_theme( get_template() );
	$version = $theme->get( 'Version' );
	wp_enqueue_script( 'flatsome-live-search', $extensions_uri . '/flatsome-live-search/flatsome-live-search.js', false, $version, true );
}

add_action( 'wp_enqueue_scripts', 'flatsome_live_search_script' );

/**
 * Search for posts and pages.
 *
 * @param  array $args Query args.
 *
 * @return array Posts.
 */
function flatsome_ajax_search_posts( $args ) {
	$defaults = $args;

	$args['s']         = apply_filters( 'flatsome_ajax_search_query', $_REQUEST['query'] );
	$args['post_type'] = apply_filters( 'flatsome_ajax_search_post_type', array( 'post', 'page' ) );

	$search_query   = http_build_query( $args );
	$query_function = apply_filters( 'flatsome_ajax_search_function', 'get_posts', $search_query, $args, $defaults );

	return ( ( $query_function == 'get_posts' ) || ! function_exists( $query_function ) ) ? get_posts( $args ) : $query_function( $search_query, $args, $defaults );
}

/**
 * Gets products based on the search type specified.
 *
 * @param string $search_type Type of search.
 * @param array  $args        Default query args.
 *
 * @return array Products.
 */
function flatsome_ajax_search_get_products( $search_type, array $args ) {
	$order_by      = get_theme_mod( 'search_products_order_by', 'relevance' );
	$order         = get_theme_mod( 'search_products_order', 'ASC' );
	$ordering_args = WC()->query->get_catalog_ordering_args( $order_by, $order );
	$defaults      = $args;

	$args['post_type']  = 'product';
	$args['orderby']    = $ordering_args['orderby'];
	$args['order']      = $ordering_args['order'];
	$args['meta_query'] = WC()->query->get_meta_query(); // WPCS: slow query ok.
	$args['tax_query']  = array(); // WPCS: slow query ok.
	$args               = flatsome_ajax_search_catalog_visibility( $args );
	$args               = flatsome_ajax_search_query_by_category( $args );

	switch ( $search_type ) {
		case 'product':
			$args['s'] = apply_filters( 'flatsome_ajax_search_products_search_query', $_REQUEST['query'] );
			break;
		case 'tag':
			$args['s']           = '';
			$args['product_tag'] = apply_filters( 'flatsome_ajax_search_products_by_tag_search_query', $_REQUEST['query'] );
			break;
		case 'sku':
			$query                = apply_filters( 'flatsome_ajax_search_products_by_sku_search_query', $_REQUEST['query'] );
			$args['s']            = '';
			$args['post_type']    = array( 'product', 'product_variation' );
			$args['meta_query'][] = array(
				'key'   => '_sku',
				'value' => $query,
			);
			break;
	}

	$search_query   = http_build_query( $args );
	$query_function = apply_filters( 'flatsome_ajax_search_function', 'get_posts', $search_query, $args, $defaults );

	return ( ( $query_function === 'get_posts' ) || ! function_exists( $query_function ) ) ? get_posts( $args ) : $query_function( $search_query, $args, $defaults );
}

/**
 * Checks product catalog visibility with custom tax_query. (only queries the exclude-from-search term) and checks WC hide out of stock option.
 *
 * @param  array $args Query args.
 *
 * @return array Query args with addition.
 */
function flatsome_ajax_search_catalog_visibility( $args ) {
	$product_visibility_term_ids = wc_get_product_visibility_term_ids();

	// Catalog visibility.
	$args['tax_query'][] = array(
		'taxonomy' => 'product_visibility',
		'field'    => 'term_taxonomy_id',
		'terms'    => $product_visibility_term_ids['exclude-from-search'],
		'operator' => 'NOT IN',
	);

	// Hide out of stock.
	if ( 'yes' === get_option( 'woocommerce_hide_out_of_stock_items' ) ) {
		$args['tax_query'][] = array(
			'taxonomy' => 'product_visibility',
			'field'    => 'term_taxonomy_id',
			'terms'    => $product_visibility_term_ids['outofstock'],
			'operator' => 'NOT IN',
		);
	}

	return $args;
}

/**
 * Query products by category if 'product_cat' is in request.
 *
 * @param array $args Query args.
 *
 * @return array Query args with or without addition.
 */
function flatsome_ajax_search_query_by_category( $args ) {
	if ( isset( $_REQUEST['product_cat'] ) ) {
		$args['tax_query'][] = array(
			'taxonomy' => 'product_cat',
			'field'    => 'slug',
			'terms'    => esc_attr( $_REQUEST['product_cat'] ),
		);
	}

	return $args;
}

/**
 * Search AJAX handler.
 */
function flatsome_ajax_search() {
	// The string from search text field.
	$query        = apply_filters( 'flatsome_ajax_search_query', $_REQUEST['query'] );
	$wc_activated = is_woocommerce_activated();
	$products     = array();
	$posts        = array();
	$sku_products = array();
	$tag_products = array();
	$suggestions  = array();

	$args = array(
		's'                   => $query,
		'orderby'             => '',
		'post_type'           => array(),
		'post_status'         => 'publish',
		'posts_per_page'      => 100,
		'ignore_sticky_posts' => 1,
		'post_password'       => '',
		'suppress_filters'    => false,
	);

	if ( $wc_activated ) {
		$products     = flatsome_ajax_search_get_products( 'product', $args );
		$sku_products = get_theme_mod( 'search_by_sku', 0 ) ? flatsome_ajax_search_get_products( 'sku', $args ) : array();
		$tag_products = get_theme_mod( 'search_by_product_tag', 0 ) ? flatsome_ajax_search_get_products( 'tag', $args ) : array();
	}

	if ( ( ! $wc_activated || get_theme_mod( 'search_result', 1 ) ) && ! isset( $_REQUEST['product_cat'] ) ) {
		$posts = flatsome_ajax_search_posts( $args );
	}

	$results = array_merge( $products, $sku_products, $tag_products, $posts );

	foreach ( $results as $key => $post ) {
		if ( $wc_activated && ( $post->post_type === 'product' || $post->post_type === 'product_variation' ) ) {
			$product = wc_get_product( $post );

			if ( $product->get_parent_id() ) {
				$parent_product = wc_get_product( $product->get_parent_id() );
				$visible        = $parent_product->get_catalog_visibility() === 'visible' || $parent_product->get_catalog_visibility() === 'search';
				if ( $parent_product->get_status() !== 'publish' || ! $visible ) {
					unset( $results[ $key ] );
					continue;
				}
			}

			$product_image = wp_get_attachment_image_src( get_post_thumbnail_id( $product->get_id() ) );

			$suggestions[] = array(
				'type'  => 'Product',
				'id'    => $product->get_id(),
				'value' => $product->get_title(),
				'url'   => $product->get_permalink(),
				'img'   => $product_image ? $product_image[0] : '',
				'price' => $product->get_price_html(),
			);
		} else {
			$suggestions[] = array(
				'type'  => 'Page',
				'id'    => $post->ID,
				'value' => get_the_title( $post->ID ),
				'url'   => get_the_permalink( $post->ID ),
				'img'   => get_the_post_thumbnail_url( $post->ID, 'thumbnail' ),
				'price' => '',
			);
		}
	}

	if ( empty( $results ) ) {
		$no_results = $wc_activated ? __( 'No products found.', 'woocommerce' ) : __( 'No matches found', 'flatsome' );

		$suggestions[] = array(
			'id'    => -1,
			'value' => $no_results,
			'url'   => '',
		);
	}

	$suggestions = flatsome_unique_suggestions( array( $products, $sku_products, $tag_products ), $suggestions );

	wp_send_json( array( 'suggestions' => $suggestions ) );
}

add_action( 'wp_ajax_flatsome_ajax_search_products', 'flatsome_ajax_search' );
add_action( 'wp_ajax_nopriv_flatsome_ajax_search_products', 'flatsome_ajax_search' );

/**
 * Makes search suggestions unique if multiple raw_results have values.
 *
 * @param array $raw_results Raw search results supplied unmerged.
 * @param array $suggestions Suggestions that may or may not be unique.
 *
 * @return array Unique suggestions.
 */
function flatsome_unique_suggestions( array $raw_results, array $suggestions ) {
	$results         = array_map( function ( $n ) { return $n ? true : false; }, $raw_results );
	$needs_filtering = count( array_filter( $results ) ) > 1;

	if ( $needs_filtering ) {
		$unique_suggestions = array();
		foreach ( $suggestions as $suggestion ) {
			$id = $suggestion['id'];
			if ( ! isset( $unique_suggestions[ $id ] ) ) {
				$unique_suggestions[ $id ] = $suggestion;
			}
		}
		$suggestions = array_values( $unique_suggestions );
	}

	return $suggestions;
}

