ONGR\FilterManagerBundle\Filter\Widget\Search\FuzzySearch

Complexity

Total Complexity

9

Size/Duplication

Total Lines	66
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	5

Importance

Changes	1
Bugs	0	Features	0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	getParameters()	0	4	1
A	setFuzziness()	0	4	1
A	setPrefixLength()	0	4	1
A	setMaxExpansions()	0	4	1
B	modifySearch()	0	17	5

<?php

/*
 * This file is part of the ONGR package.
 *
 * (c) NFQ Technologies UAB <info@nfq.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace ONGR\FilterManagerBundle\Filter\Widget\Search;

use ONGR\ElasticsearchDSL\Query\BoolQuery;
use ONGR\ElasticsearchDSL\Query\FuzzyQuery;
use ONGR\ElasticsearchDSL\Search;
use ONGR\FilterManagerBundle\Filter\FilterState;
use ONGR\FilterManagerBundle\Search\SearchRequest;

/**
 * This class runs match search.
 */
class FuzzySearch extends AbstractSingleValue
{
    /**
     * @var array Fuzzy query parameters.
     */
    private $parameters = [];

    /**
     * {@inheritdoc}
     */
    public function modifySearch(Search $search, FilterState $state = null, SearchRequest $request = null)
    {
        if ($state && $state->isActive()) {
            if (strpos($this->getField(), ',') !== false) {
                $subQuery = new BoolQuery();
                foreach (explode(',', $this->getField()) as $field) {
                    $subQuery->add(new FuzzyQuery($field, $state->getValue(), $this->getParameters()), 'should');
                }
                $search->addQuery($subQuery, 'must');
            } else {
                $search->addQuery(
                    new FuzzyQuery($this->getField(), $state->getValue(), $this->getParameters()),
                    'must'
                );
            }
        }
    }

    /**
     * @return array
     */
    public function getParameters()
    {
        return array_filter($this->parameters);
    }

    /**
     * The maximum edit distance.
     *
     * @param string|int|float $fuzziness
     */
    public function setFuzziness($fuzziness)
    {
        $this->parameters['fuzziness'] = $fuzziness;
    }

    /**
     * The number of initial characters which will not be “fuzzified”.
     *
     * @param int $prefixLength
     */
    public function setPrefixLength($prefixLength)
    {
        $this->parameters['prefix_length'] = $prefixLength;
    }

    /**
     * The maximum number of terms that the fuzzy query will expand to.
     *
     * @param int $maxExpansions
     */
    public function setMaxExpansions($maxExpansions)
    {
        $this->parameters['max_expansions'] = $maxExpansions;
    }
}