0byt3m1n1-V2

Path: /home/dev2.destoffenstraat.com/app/code/Mirasvit/Search/Model/Index/ [ Home ]
File: Searcher.php
<?php
/**
 * Mirasvit
 *
 * This source file is subject to the Mirasvit Software License, which is available at https://mirasvit.com/license/.
 * Do not edit or add to this file if you wish to upgrade the to newer versions in the future.
 * If you wish to customize this module for your needs.
 * Please refer to http://www.magentocommerce.com for more information.
 *
 * @category  Mirasvit
 * @package   mirasvit/module-search-ultimate
 * @version   2.3.2
 * @copyright Copyright (C) 2024 Mirasvit (https://mirasvit.com/)
 */



namespace Mirasvit\Search\Model\Index;

use Magento\CatalogSearch\Model\Advanced\Request\BuilderFactory as RequestBuilderFactory;
use Magento\Framework\App\RequestInterface;
use Magento\Framework\App\ResponseFactory;
use Magento\Framework\App\ScopeResolverInterface;
use Magento\Framework\App\State as AppState;
use Magento\Framework\Data\Collection\AbstractDb;
use Magento\Framework\Search\Response\Aggregation;
use Magento\Search\Model\QueryFactory;
use Magento\Search\Model\SearchEngine;
use Mirasvit\Misspell\Model\GraphQL\Suggester;
use Mirasvit\Search\Api\Data\Index\InstanceInterface;

class Searcher
{
    private $instance;

    private $queryFactory;

    private $requestBuilderFactory;

    private $searchEngine;

    private $scopeResolver;

    private $request;

    private $responseFactory;

    private $appState;

    private $suggester;

    private $total        = 0;

    private $aggregations = null;

    public function __construct(
        QueryFactory           $queryFactory,
        RequestBuilderFactory  $requestBuilderFactory,
        SearchEngine           $searchEngine,
        ScopeResolverInterface $scopeResolver,
        RequestInterface       $request,
        ResponseFactory        $responseFactory,
        AppState               $appState,
        Suggester              $suggester
    ) {
        $this->queryFactory          = $queryFactory;
        $this->requestBuilderFactory = $requestBuilderFactory;
        $this->searchEngine          = $searchEngine;
        $this->scopeResolver         = $scopeResolver;
        $this->request               = $request;
        $this->responseFactory       = $responseFactory;
        $this->appState              = $appState;
        $this->suggester             = $suggester;
    }

    public function getInstance(): InstanceInterface
    {
        return $this->instance;
    }

    public function setInstance(InstanceInterface $instance)
    {
        $this->instance = $instance;

        return $this;
    }

    /**
     * Join matches to collection
     *
     * @param AbstractDb $collection
     * @param string     $field
     * @param array      $args
     *
     * @return $this
     */
    public function joinMatches($collection, string $field = 'e.entity_id', array $args = [], \Zend_Db_Expr $orderExpr = null)
    {
        $queryResponse = $this->getQueryResponse($args);

        if ($this->appState->getAreaCode() === 'graphql' && $queryResponse) {
            if ($queryResponse->getTotal() == 0) {
                $term = $this->suggester->suggest();
                if ($term) {
                    $this->queryFactory->get()->setQueryText($term);
                    $queryResponse = $this->getQueryResponse($args);
                }
            }

            $this->total        = $queryResponse->getTotal();
            $this->aggregations = $queryResponse->getAggregations();
        }

        $ids = [0];

        if ($queryResponse) {
            foreach ($queryResponse->getIterator() as $item) {
                $ids[] = $item->getId();
            }
        }

        $idList    = join(',', $ids);
        $whereExpr = new \Zend_Db_Expr("$field IN ($idList)");
        if ($orderExpr == null) {
            $orderExpr = new \Zend_Db_Expr("FIELD($field, $idList)");
        }

        $collection->getSelect()
            ->reset(\Magento\Framework\DB\Select::ORDER)
            ->where($whereExpr->__toString())
            ->order($orderExpr);

        return $this;
    }

    /**
     * @return \Magento\Framework\Search\Response\QueryResponse|\Magento\Framework\Search\ResponseInterface
     */
    public function getQueryResponse(array $args = [])
    {
        /** @var \Magento\Search\Model\Query $query */
        $query = $this->queryFactory->get();

        if ($query->isQueryTextShort()) {
            return false;
        }

        $queryText = $this->queryFactory->get()->getQueryText();
        if (empty($queryText)) {
            return false;
        }

        $requestBuilder = $this->requestBuilderFactory->create();
        $requestBuilder->bind('search_term', $queryText);

        $requestBuilder->bindDimension('scope', $this->scopeResolver->getScope());

        if (!empty($args) && $this->appState->getAreaCode() === 'graphql') {
            $requestBuilder->setRequestName('quick_search_container');
            $requestBuilder->setFrom(($args['currentPage'] == 1) ? 0 : ($args['currentPage'] - 1) * $args['pageSize']);
            $requestBuilder->setSize($args['pageSize']);
            $requestBuilder->setSort($args['sort']);
            foreach ($args['filter'] as $attribute => $value) {
                if ($attribute == 'category_id') {
                    $requestBuilder->bind('category_ids', array_values($value)[0]);
                } elseif (array_key_exists('from', $value)) {
                    $requestBuilder->bind($attribute . '.from', $value['from']);
                } elseif (array_key_exists('to', $value)) {
                    $requestBuilder->bind($attribute . '.to', $value['to']);
                } else {
                    $requestBuilder->bind($attribute, $value);
                }
            }
        } else {
            $requestBuilder->setRequestName($this->getInstance()->getIdentifier());
        }

        if ($this->getInstance()->getIdentifier() === 'catalogsearch_fulltext') {
            $requestBuilder->setRequestName('quick_search_container');
            $requestBuilder->bind('visibility', [3, 4]);

            $cat = $this->request->getParam('cat');

            if ($cat && (int)$cat > 0) {
                $requestBuilder->bind('category_ids', (int)$cat);
            }
        }

        $queryRequest = $requestBuilder->create();

        return $this->searchEngine->search($queryRequest);
    }

    /**
     * @return array
     */
    public function getMatchedIds()
    {
        /** @var \Magento\Search\Model\Query $query */
        $query = $this->queryFactory->get();
        if ($query->isQueryTextShort()) {
            return [];
        }

        $queryText = $query->getQueryText();

        $requestBuilder = $this->requestBuilderFactory->create();

        $requestBuilder->bind('search_term', $queryText);

        $requestBuilder->bindDimension('scope', $this->scopeResolver->getScope());

        $requestBuilder->setRequestName($this->getInstance()->getIdentifier());

        /** @var \Magento\Framework\Search\Request $queryRequest */
        $queryRequest = $requestBuilder->create();

        $queryResponse = $this->searchEngine->search($queryRequest);
        $ids           = [];
        foreach ($queryResponse->getIterator() as $item) {
            $ids[] = $item->getId();
        }

        return $ids;
    }

    public function getTotal(): int
    {
        return $this->total;
    }

    public function getAggregations(): ?Aggregation
    {
        return $this->aggregations;
    }
}