<?php
/**
 * For the full copyright and license information, please view the
 * docs/licenses/LICENSE.txt file that was distributed with this source code.
 */

declare(strict_types=1);

namespace PrestaShop\PrestaShop\Adapter\Search\CommandHandler;

use Exception;
use PrestaShop\PrestaShop\Adapter\Product\Repository\ProductRepository;
use PrestaShop\PrestaShop\Adapter\Shop\Repository\ShopGroupRepository;
use PrestaShop\PrestaShop\Adapter\Shop\Repository\ShopRepository;
use PrestaShop\PrestaShop\Core\CommandBus\Attributes\AsCommandHandler;
use PrestaShop\PrestaShop\Core\Domain\Product\Exception\ProductNotFoundException;
use PrestaShop\PrestaShop\Core\Domain\Product\ValueObject\ProductId;
use PrestaShop\PrestaShop\Core\Domain\Search\Command\SearchIndexationCommand;
use PrestaShop\PrestaShop\Core\Domain\Search\CommandHandler\SearchIndexationHandlerInterface;
use PrestaShop\PrestaShop\Core\Domain\Search\Exception\SearchIndexationException;
use PrestaShop\PrestaShop\Core\Domain\Search\Exception\SearchIndexationProductNotFoundException;
use PrestaShop\PrestaShop\Core\Domain\Search\Exception\SearchIndexationShopGroupNotFoundException;
use PrestaShop\PrestaShop\Core\Domain\Search\Exception\SearchIndexationShopNotFoundException;
use PrestaShop\PrestaShop\Core\Domain\Shop\Exception\ShopGroupNotFoundException;
use PrestaShop\PrestaShop\Core\Domain\Shop\Exception\ShopNotFoundException;
use PrestaShop\PrestaShop\Core\Domain\Shop\ValueObject\ShopGroupId;
use PrestaShop\PrestaShop\Core\Domain\Shop\ValueObject\ShopId;
use Search;
use Shop;

/**
 * Handles search indexation command.
 */
#[AsCommandHandler]
final class SearchIndexationHandler implements SearchIndexationHandlerInterface
{
    public function __construct(
        protected readonly ProductRepository $productRepository,
        protected readonly ShopRepository $shopRepository,
        protected readonly ShopGroupRepository $shopGroupRepository,
    ) {
    }

    /**
     * {@inheritdoc}
     */
    public function handle(SearchIndexationCommand $command): void
    {
        $shopConstraint = $command->getShopConstraint();

        $shopId = $shopConstraint->getShopId();
        if ($shopId instanceof ShopId) {
            try {
                $this->shopRepository->assertShopExists($shopId);
            } catch (ShopNotFoundException) {
                throw new SearchIndexationShopNotFoundException($shopId);
            }
        }
        $shopGroupId = $shopConstraint->getShopGroupId();
        if ($shopGroupId instanceof ShopGroupId) {
            try {
                $this->shopGroupRepository->assertShopGroupExists($shopGroupId);
            } catch (ShopGroupNotFoundException) {
                throw new SearchIndexationShopGroupNotFoundException($shopGroupId);
            }
        }

        $productId = $command->getProductId();
        if ($productId instanceof ProductId) {
            try {
                $this->productRepository->assertProductExists($productId);
            } catch (ProductNotFoundException) {
                throw new SearchIndexationProductNotFoundException($productId);
            }
        }

        $isFull = $command->isFull();
        try {
            if ($shopId instanceof ShopId) {
                Shop::setContext(Shop::CONTEXT_SHOP, $shopId->getValue());
            } elseif ($shopGroupId instanceof ShopGroupId) {
                Shop::setContext(Shop::CONTEXT_GROUP, $shopGroupId->getValue());
            } else {
                Shop::setContext(Shop::CONTEXT_ALL);
            }

            if (!Search::indexation($isFull, $productId?->getValue())) {
                throw new SearchIndexationException('Search indexation failed');
            }
        } catch (Exception $e) {
            throw new SearchIndexationException('Search indexation failed', 0, $e);
        }
    }
}