<?php

/*
 * This file is part of the Symfony package.
 *
 * (c) Fabien Potencier <fabien@symfony.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Symfony\Component\Uid\Factory;

use Symfony\Component\Uid\TimeBasedUidInterface;
use Symfony\Component\Uid\Uuid;
use Symfony\Component\Uid\UuidV3;
use Symfony\Component\Uid\UuidV4;
use Symfony\Component\Uid\UuidV5;
use Symfony\Component\Uid\UuidV7;

class MockUuidFactory extends UuidFactory
{
    private \Iterator $sequence;

    /**
     * @param iterable<string|Uuid> $uuids
     */
    public function __construct(
        iterable $uuids,
        private Uuid|string|null $timeBasedNode = null,
        private Uuid|string|null $nameBasedNamespace = null,
    ) {
        $this->sequence = match (true) {
            \is_array($uuids) => new \ArrayIterator($uuids),
            $uuids instanceof \Iterator => $uuids,
            $uuids instanceof \Traversable => new \IteratorIterator($uuids),
        };
    }

    public function create(): Uuid
    {
        if (!$this->sequence->valid()) {
            throw new \RuntimeException('No more UUIDs in sequence.');
        }
        $uuid = $this->sequence->current();
        $this->sequence->next();

        return $uuid instanceof Uuid ? $uuid : Uuid::fromString($uuid);
    }

    public function randomBased(): RandomBasedUuidFactory
    {
        if (!($uuid = $this->create()) instanceof UuidV4) {
            throw new \RuntimeException(\sprintf('Next UUID in sequence is not a UuidV4: "%s".', get_debug_type($uuid)));
        }

        return new class($uuid) extends RandomBasedUuidFactory {
            public function __construct(
                private UuidV4 $uuid,
            ) {
            }

            public function create(): UuidV4
            {
                return $this->uuid;
            }
        };
    }

    public function timeBased(Uuid|string|null $node = null): TimeBasedUuidFactory
    {
        if (!($uuid = $this->create()) instanceof TimeBasedUidInterface) {
            throw new \RuntimeException(\sprintf('Next UUID in sequence is not a TimeBasedUidInterface: "%s".', get_debug_type($uuid)));
        }

        if (\is_string($node ??= $this->timeBasedNode)) {
            $node = Uuid::fromString($node);
        }

        return new class($uuid, $node) extends TimeBasedUuidFactory {
            public function __construct(
                private TimeBasedUidInterface $uuid,
                private ?Uuid $node = null,
            ) {
                if ($uuid instanceof UuidV7) {
                    $this->node = null;
                }
            }

            public function create(?\DateTimeInterface $time = null): Uuid&TimeBasedUidInterface
            {
                if (null !== $time && $this->uuid->getDateTime() != $time) {
                    throw new \RuntimeException(\sprintf('Next UUID in sequence does not match the expected time: "%s" != "%s".', $this->uuid->getDateTime()->format('@U.uT'), $time->format('@U.uT')));
                }

                if (null !== $this->node && $this->uuid->getNode() !== substr($this->node->toRfc4122(), -12)) {
                    throw new \RuntimeException(\sprintf('Next UUID in sequence does not match the expected node: "%s" != "%s".', $this->uuid->getNode(), substr($this->node->toRfc4122(), -12)));
                }

                return $this->uuid;
            }
        };
    }

    public function nameBased(Uuid|string|null $namespace = null): NameBasedUuidFactory
    {
        if (!($uuid = $this->create()) instanceof UuidV5 && !$uuid instanceof UuidV3) {
            throw new \RuntimeException(\sprintf('Next UUID in sequence is not a UuidV5 or UuidV3: "%s".', get_debug_type($uuid)));
        }

        $factory = parent::nameBased($namespace);

        return new class ($uuid, $factory) extends NameBasedUuidFactory {
            public function __construct(
                private UuidV5|UuidV3 $uuid,
                private NameBasedUuidFactory $factory,
            ) {
            }

            public function create(string $name): UuidV5|UuidV3
            {
                if ($this->uuid->toRfc4122() !== $this->factory->create($name)->toRfc4122()) {
                    throw new \RuntimeException(\sprintf('Next UUID in sequence does not match the expected named UUID: "%s" != "%s".', $this->uuid->toRfc4122(), $this->factory->create($name)->toRfc4122()));
                }

                return $this->uuid;
            }
        };
    }
}