nguyen dung 39b8cb3612 init
2022-02-18 16:43:41 +07:00

234 lines
7.7 KiB
PHP

<?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' );
$ordering_args = WC()->query->get_catalog_ordering_args( $order_by, 'ASC' );
$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 ) {
$suggestions = array_map( 'unserialize', array_unique( array_map( 'serialize', $suggestions ) ) );
}
return $suggestions;
}