Skip to content
Snippets Groups Projects
HasChildFilter.php 2.03 KiB
Newer Older
juliensantos87's avatar
juliensantos87 committed
<?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\ElasticsearchBundle\DSL\Filter;

use ONGR\ElasticsearchBundle\DSL\BuilderInterface;
use ONGR\ElasticsearchBundle\DSL\ParametersTrait;

/**
 * Elasticsearch has_child filter.
 */
class HasChildFilter implements BuilderInterface
{
    use ParametersTrait;

    const INNER_QUERY = 'query';
    const INNER_FILTER = 'filter';

juliensantos87's avatar
juliensantos87 committed
    /**
     * @var string
     */
    private $type;

    /**
Mantas Jonušas's avatar
Mantas Jonušas committed
     * @var BuilderInterface
juliensantos87's avatar
juliensantos87 committed
     */
juliensantos87's avatar
juliensantos87 committed
    private $filter;
    /**
     * @var BuilderInterface
     */
    private $query;

juliensantos87's avatar
juliensantos87 committed
    /**
     * @param string           $type
     * @param BuilderInterface $block
juliensantos87's avatar
juliensantos87 committed
     * @param array            $parameters
     * @param string           $inner
     *
     * @throws \InvalidArgumentException
juliensantos87's avatar
juliensantos87 committed
     */
    public function __construct($type, BuilderInterface $block, array $parameters = [], $inner = self::INNER_FILTER)
juliensantos87's avatar
juliensantos87 committed
    {
        $this->type = $type;

        switch ($inner) {
            case 'filter':
                $this->filter = $block;
                break;
            case 'query':
                $this->query = $block;
                break;
            default:
                throw new \InvalidArgumentException('Not supported argument type');
        }

juliensantos87's avatar
juliensantos87 committed
        $this->setParameters($parameters);
    }

    /**
     * {@inheritdoc}
     */
    public function getType()
    {
        return 'has_child';
    }

    /**
     * {@inheritdoc}
     */
    public function toArray()
    {
        $query = [ 'type' => $this->type ];

        $queries = ['filter', 'query'];

        foreach ($queries as $type) {
            if ($this->{$type}) {
                $query[$type] = [
                    $this->{$type}->getType() => $this->{$type}->toArray(),
                ];
            }
        }
juliensantos87's avatar
juliensantos87 committed

        $output = $this->processArray($query);

        return $output;
    }
}