Phraseanet/lib/Alchemy/Phrasea/SearchEngine/Elastic/AST/AbstractTermNode.php

<?php

namespace Alchemy\Phrasea\SearchEngine\Elastic\AST;

use Alchemy\Phrasea\SearchEngine\Elastic\Search\QueryContext;
use Alchemy\Phrasea\SearchEngine\Elastic\Thesaurus\Concept;
use Alchemy\Phrasea\SearchEngine\Elastic\Thesaurus\TermInterface;

abstract class AbstractTermNode extends Node implements TermInterface
{
    protected $text;
    protected $context;
    private $concepts = [];

    public function __construct($text, Context $context = null)
    {
        $this->text = $text;
        $this->context = $context;
    }

    public function setConcepts(array $concepts)
    {
        $this->concepts = $concepts;
    }

    protected function buildConceptQueries(QueryContext $context)
    {
        $queries = [];
        $concepts = Concept::pruneNarrowConcepts($this->concepts);
        if (!$concepts) {
            return [];
        }
        $fields = $context->getFields();
        if (empty($fields)) {
            $fields = ['*'];
        }
        $prefixedFields = array();
        foreach ($fields as $field) {
            $prefixedFields[] = 'concept_path.' . $field;
        }
        foreach ($concepts as $concept) {
            $queries[]['multi_match'] = [
                'fields' => $prefixedFields,
                'query' => $concept->getPath()
            ];
        }

        return $queries;
    }

    public function getValue()
    {
        return $this->text;
    }

    public function hasContext()
    {
        return $this->context !== null;
    }

    public function getContext()
    {
        return $this->context->getValue();
    }

    public function getTermNodes()
    {
        return [$this];
    }
}