<?php
/**
 * SPDX-FileCopyrightText: 2023 Nextcloud GmbH and Nextcloud contributors
 * SPDX-License-Identifier: AGPL-3.0-or-later
 */
namespace Test\Files\Search\QueryOptimizer;

use OC\Files\Search\QueryOptimizer\FlattenSingleArgumentBinaryOperation;
use OC\Files\Search\QueryOptimizer\MergeDistributiveOperations;
use OC\Files\Search\SearchBinaryOperator;
use OC\Files\Search\SearchComparison;
use OCP\Files\Search\ISearchBinaryOperator;
use OCP\Files\Search\ISearchComparison;
use Test\TestCase;

class MergeDistributiveOperationsTest extends TestCase {
	private $optimizer;
	private $simplifier;

	protected function setUp(): void {
		parent::setUp();

		$this->optimizer = new MergeDistributiveOperations();
		$this->simplifier = new FlattenSingleArgumentBinaryOperation();
	}

	public function testBasicOrOfAnds() {
		$operator = new SearchBinaryOperator(
			ISearchBinaryOperator::OPERATOR_OR,
			[
				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 1),
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "foo"),
				]),
				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 1),
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "bar"),
				]),
				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 1),
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "asd"),
				])
			]
		);
		$this->assertEquals('((storage eq 1 and path eq "foo") or (storage eq 1 and path eq "bar") or (storage eq 1 and path eq "asd"))', $operator->__toString());

		$this->optimizer->processOperator($operator);
		$this->simplifier->processOperator($operator);

		$this->assertEquals('(storage eq 1 and (path eq "foo" or path eq "bar" or path eq "asd"))', $operator->__toString());
	}

	public function testDontTouchIfNotSame() {
		$operator = new SearchBinaryOperator(
			ISearchBinaryOperator::OPERATOR_OR,
			[
				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 1),
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "foo"),
				]),
				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 2),
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "bar"),
				]),
				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 3),
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "asd"),
				])
			]
		);
		$this->assertEquals('((storage eq 1 and path eq "foo") or (storage eq 2 and path eq "bar") or (storage eq 3 and path eq "asd"))', $operator->__toString());

		$this->optimizer->processOperator($operator);
		$this->simplifier->processOperator($operator);

		$this->assertEquals('((storage eq 1 and path eq "foo") or (storage eq 2 and path eq "bar") or (storage eq 3 and path eq "asd"))', $operator->__toString());
	}

	public function testMergePartial() {
		$operator = new SearchBinaryOperator(
			ISearchBinaryOperator::OPERATOR_OR,
			[
				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 1),
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "foo"),
				]),
				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 1),
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "bar"),
				]),
				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 2),
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "asd"),
				])
			]
		);
		$this->assertEquals('((storage eq 1 and path eq "foo") or (storage eq 1 and path eq "bar") or (storage eq 2 and path eq "asd"))', $operator->__toString());

		$this->optimizer->processOperator($operator);
		$this->simplifier->processOperator($operator);

		$this->assertEquals('((storage eq 1 and (path eq "foo" or path eq "bar")) or (storage eq 2 and path eq "asd"))', $operator->__toString());
	}

	public function testOptimizeInside() {
		$operator = new SearchBinaryOperator(
			ISearchBinaryOperator::OPERATOR_AND,
			[
				new SearchBinaryOperator(
					ISearchBinaryOperator::OPERATOR_OR,
					[
						new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
							new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 1),
							new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "foo"),
						]),
						new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
							new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 1),
							new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "bar"),
						]),
						new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
							new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 1),
							new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "asd"),
						])
					]
				),
				new SearchComparison(ISearchComparison::COMPARE_EQUAL, "mimetype", "text")
			]
		);
		$this->assertEquals('(((storage eq 1 and path eq "foo") or (storage eq 1 and path eq "bar") or (storage eq 1 and path eq "asd")) and mimetype eq "text")', $operator->__toString());

		$this->optimizer->processOperator($operator);
		$this->simplifier->processOperator($operator);

		$this->assertEquals('((storage eq 1 and (path eq "foo" or path eq "bar" or path eq "asd")) and mimetype eq "text")', $operator->__toString());
	}

	public function testMoveInnerOperations() {
		$operator = new SearchBinaryOperator(
			ISearchBinaryOperator::OPERATOR_OR,
			[
				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 1),
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "foo"),
				]),
				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 1),
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "bar"),
				]),
				new SearchBinaryOperator(ISearchBinaryOperator::OPERATOR_AND, [
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "storage", 1),
					new SearchComparison(ISearchComparison::COMPARE_EQUAL, "path", "asd"),
					new SearchComparison(ISearchComparison::COMPARE_GREATER_THAN, "size", "100"),
				])
			]
		);
		$this->assertEquals('((storage eq 1 and path eq "foo") or (storage eq 1 and path eq "bar") or (storage eq 1 and path eq "asd" and size gt "100"))', $operator->__toString());

		$this->optimizer->processOperator($operator);
		$this->simplifier->processOperator($operator);

		$this->assertEquals('(storage eq 1 and (path eq "foo" or path eq "bar" or (path eq "asd" and size gt "100")))', $operator->__toString());
	}
}