Smile-SA/elasticsuite

<?php
/**
 * DISCLAIMER
 *
 * Do not edit or add to this file if you wish to upgrade Smile ElasticSuite to newer
 * versions in the future.
 *
 * @category  Smile
 * @package   Smile\ElasticsuiteCatalogOptimizer
 * @author    Fanny DECLERCK <fadec@smile.fr>
 * @copyright 2020 Smile
 * @license   Open Software License ("OSL") v. 3.0
 */
namespace Smile\ElasticsuiteCatalogOptimizer\Plugin;

use Smile\ElasticsuiteCore\Search\Request\Query\Builder as QueryBuilder;
use Smile\ElasticsuiteCore\Api\Search\Request\ContainerConfigurationInterface;
use Smile\ElasticsuiteCore\Search\Request\QueryInterface;
use Smile\ElasticsuiteCatalogOptimizer\Model\Optimizer\ApplierList;

/**
 * Plugin that handle query rewriting (synonym substitution) during fulltext query building phase.
 *
 * @category Smile
 * @package  Smile\ElasticsuiteCatalogOptimizer
 * @author   Fanny DECLERCK <fadec@smile.fr>
 */
class QueryRewrite
{
    /**
     * @var ApplierList
     */
    private $applierList;

    /**
     * Constructor.
     *
     * @param ApplierList $applierList Applier list.
     */
    public function __construct(ApplierList $applierList)
    {
        $this->applierList = $applierList;
    }

    /**
     * Rewrite the query.
     *
     * @SuppressWarnings(PHPMD.UnusedFormalParameter)
     *
     * @param QueryBuilder                    $subject                Original query builder.
     * @param \Closure                        $proceed                Original create func.
     * @param ContainerConfigurationInterface $containerConfiguration Search request container config.
     * @param string                          $queryText              Current query text.
     * @param array                           $filters                Filters.
     * @param string                          $spellingType           Spelling type of the query.
     *
     * @return QueryInterface
     */
    public function aroundCreateQuery(
        QueryBuilder $subject,
        \Closure $proceed,
        ContainerConfigurationInterface $containerConfiguration,
        $queryText,
        array $filters,
        $spellingType
    ) {
        $query = $proceed($containerConfiguration, $queryText, $filters, $spellingType);

        return $this->applierList->applyOptimizers($containerConfiguration, $query);
    }
}