Phraseanet/lib/Alchemy/Phrasea/Command/SearchEngine/Debug/QueryParseCommand.php

<?php

/*
 * This file is part of Phraseanet
 *
 * (c) 2005-2014 Alchemy
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Alchemy\Phrasea\Command\SearchEngine\Debug;

use Alchemy\Phrasea\Command\Command;
use Alchemy\Phrasea\SearchEngine\Elastic\Indexer;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\Stopwatch\Stopwatch;

class QueryParseCommand extends Command
{
    protected function configure()
    {
        $this
            ->setName('searchengine:query:parse')
            ->setDescription('Debug a search query')
            ->addArgument(
                'query',
                InputArgument::REQUIRED,
                'The query to debug'
            )
            ->addOption(
                'compiler-dump',
                'd',
                InputOption::VALUE_NONE,
                'Output underlying compiler AST before visitor processing'
            )
            ->addOption(
                'no-compiler-postprocessing',
                'p',
                InputOption::VALUE_NONE,
                'Prevent AST optimization before visitor processing'
            )
            ->addOption(
                'raw',
                null,
                InputOption::VALUE_NONE,
                'Only output query dump'
            )
        ;
    }

    protected function doExecute(InputInterface $input, OutputInterface $output)
    {
        $string = $input->getArgument('query');
        $raw = $input->getOption('raw');

        if (!$raw) {
            $output->writeln(sprintf('Parsing search query: <comment>%s</comment>', $string));
            $output->writeln(str_repeat('-', 20));
        }

        $postprocessing = !$input->getOption('no-compiler-postprocessing');

        $compiler = $this->container['query_compiler'];

        $stopwatch = new Stopwatch();
        $stopwatch->start('parsing');

        if ($input->getOption('compiler-dump')) {
            $dump = $compiler->dump($string, $postprocessing);
        } else {
            $query = $compiler->parse($string, $postprocessing);
            $dump = $query->dump();
        }

        $event = $stopwatch->stop('parsing');

        if (!$raw) {
            $output->writeln($dump);
            $output->writeln(str_repeat('-', 20));
            $output->writeln(sprintf("Took %sms", $event->getDuration()));
        } else {
            $output->write($dump);
        }
    }
}