<?php

declare(strict_types=1);

/*
 * This file is part of Contao.
 *
 * (c) Leo Feyer
 *
 * @license LGPL-3.0-or-later
 */

namespace Contao\CoreBundle\Security\Voter\DataContainer;

use Contao\CoreBundle\Security\ContaoCorePermissions;
use Contao\CoreBundle\Security\DataContainer\CreateAction;
use Contao\CoreBundle\Security\DataContainer\DeleteAction;
use Contao\CoreBundle\Security\DataContainer\ReadAction;
use Contao\CoreBundle\Security\DataContainer\UpdateAction;
use Doctrine\DBAL\Connection;
use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
use Symfony\Component\Security\Core\Authorization\AccessDecisionManagerInterface;
use Symfony\Contracts\Service\ResetInterface;

/**
 * @internal
 */
class PageTypeAccessVoter extends AbstractDataContainerVoter implements ResetInterface
{
    use TypeAccessTrait;

    private const FIRST_LEVEL_TYPES = ['error_401', 'error_403', 'error_404', 'error_503'];

    private array|null $rootIds = null;

    private array|null $rootTypes = null;

    public function __construct(
        private readonly AccessDecisionManagerInterface $accessDecisionManager,
        private readonly Connection $connection,
    ) {
    }

    public function reset(): void
    {
        $this->rootIds = null;
        $this->rootTypes = null;
    }

    protected function getTable(): string
    {
        return 'tl_page';
    }

    protected function hasAccess(TokenInterface $token, CreateAction|DeleteAction|ReadAction|UpdateAction $action): bool
    {
        return $this->validateAccessToPageType($token, $action)
            && $this->validateFirstLevelType($action)
            && $this->validateRootType($action);
    }

    private function validateAccessToPageType(TokenInterface $token, CreateAction|DeleteAction|ReadAction|UpdateAction $action): bool
    {
        if ($action instanceof ReadAction) {
            return true;
        }

        return $this->hasAccessToType($token, ContaoCorePermissions::USER_CAN_ACCESS_PAGE_TYPE, $action);
    }

    private function validateFirstLevelType(CreateAction|DeleteAction|ReadAction|UpdateAction $action): bool
    {
        if ($action instanceof ReadAction || $action instanceof DeleteAction) {
            return true;
        }

        // Allow if current page type is not an error page, or the new page type is not
        // set to an error page type.
        if (
            (!$action instanceof UpdateAction || !\in_array($action->getCurrent()['type'], self::FIRST_LEVEL_TYPES, true))
            && (!isset($action->getNew()['type']) || !\in_array($action->getNew()['type'], self::FIRST_LEVEL_TYPES, true))
        ) {
            return true;
        }

        // Cannot create a page without a parent ID
        if ($action instanceof CreateAction && null === $action->getNewPid()) {
            return false;
        }

        $type = $action->getNew()['type'] ?? ($action instanceof UpdateAction ? $action->getCurrent()['type'] : null);
        $currentPid = $action instanceof UpdateAction ? (int) $action->getCurrentPid() : null;
        $pid = (int) ($action->getNewPid() ?? $currentPid);

        if (
            (null !== $action->getNewPid() || null !== ($action->getNew()['sorting'] ?? null))
            && (!$action instanceof UpdateAction || \in_array($type, self::FIRST_LEVEL_TYPES, true))
            && ($pid !== $currentPid)
            && (!$this->isRootPage($pid) || $this->hasPageTypeInRoot($type, $pid))
        ) {
            return false;
        }

        return !($action instanceof UpdateAction
            && isset($action->getNew()['type'])
            && \in_array($action->getNew()['type'], self::FIRST_LEVEL_TYPES, true)
            && (!$this->isRootPage((int) $action->getCurrentPid()) || $this->hasPageTypeInRoot($type, (int) $action->getCurrentPid())));
    }

    private function validateRootType(CreateAction|DeleteAction|ReadAction|UpdateAction $action): bool
    {
        if ($action instanceof ReadAction || $action instanceof DeleteAction) {
            return true;
        }

        if (($action instanceof CreateAction || null === $action->getNewPid()) && !isset($action->getNew()['type'])) {
            return true;
        }

        $type = $action->getNew()['type'] ?? ($action instanceof UpdateAction ? ($action->getCurrent()['type'] ?? null) : null);
        $pid = (int) ($action->getNewPid() ?? ($action instanceof UpdateAction ? $action->getCurrentPid() : -1));

        return ('root' !== $type || 0 === $pid) && (0 !== $pid || 'root' === $type);
    }

    private function isRootPage(int $pageId): bool
    {
        if (null === $this->rootIds) {
            $this->rootIds = $this->connection->fetchFirstColumn("SELECT id FROM tl_page WHERE type = 'root'");
            $this->rootIds = array_map(\intval(...), $this->rootIds);
        }

        return \in_array($pageId, $this->rootIds, true);
    }

    private function hasPageTypeInRoot(string $type, int $rootId): bool
    {
        if (null === $this->rootTypes) {
            $this->rootTypes = [];
            $records = $this->connection->fetchAllNumeric("SELECT p.pid, p.type FROM tl_page p JOIN tl_page r ON p.pid = r.id WHERE r.type = 'root' GROUP BY p.pid, p.type");

            foreach ($records as $row) {
                $this->rootTypes[$row[0]][$row[1]] = true;
            }
        }

        return $this->rootTypes[$rootId][$type] ?? false;
    }
}
