AttributeSubscriberMetadataFactory.php

<?php

declare(strict_types=1);

namespace Patchlevel\EventSourcing\Metadata\Subscriber;

use Patchlevel\EventSourcing\Attribute\Cleanup;
use Patchlevel\EventSourcing\Attribute\OnFailed;
use Patchlevel\EventSourcing\Attribute\RetryStrategy;
use Patchlevel\EventSourcing\Attribute\Setup;
use Patchlevel\EventSourcing\Attribute\Subscribe;
use Patchlevel\EventSourcing\Attribute\Subscriber;
use Patchlevel\EventSourcing\Attribute\Teardown;
use ReflectionAttribute;
use ReflectionClass;
use ReflectionMethod;
use Symfony\Component\TypeInfo\TypeResolver\TypeResolver;

use function array_key_exists;
use function count;

final class AttributeSubscriberMetadataFactory implements SubscriberMetadataFactory
{
    private readonly TypeResolver $typeResolver;

    public function __construct()
    {
        $this->typeResolver = TypeResolver::create();
    }

    /** @var array<class-string, SubscriberMetadata> */
    private array $subscriberMetadata = [];

    /** @param class-string $subscriber */
    public function metadata(string $subscriber): SubscriberMetadata
    {
        if (array_key_exists($subscriber, $this->subscriberMetadata)) {
            return $this->subscriberMetadata[$subscriber];
        }

        $reflector = new ReflectionClass($subscriber);

        $attributes = $reflector->getAttributes(Subscriber::class, ReflectionAttribute::IS_INSTANCEOF);

        if ($attributes === []) {
            throw new ClassIsNotASubscriber($subscriber);
        }

        $subscriberInfo = $attributes[0]->newInstance();

        $methods = $reflector->getMethods();

        $subscribeMethods = [];
        $setupMethod = null;
        $teardownMethod = null;
        $cleanupMethod = null;
        $failedMethod = null;

        foreach ($methods as $method) {
            $attributes = $method->getAttributes(Subscribe::class);

            foreach ($attributes as $attribute) {
                $instance = $attribute->newInstance();
                $eventClass = $instance->eventClass;

                if (array_key_exists($eventClass, $subscribeMethods)) {
                    throw DuplicateSubscribeMethod::duplicateEvent(
                        $subscriber,
                        $eventClass,
                        $subscribeMethods[$eventClass]->name,
                        $method->getName(),
                    );
                }

                $subscribeMethods[$eventClass] = $this->subscribeMethod($method);
            }

            if ($method->getAttributes(OnFailed::class)) {
                if ($failedMethod !== null) {
                    throw new DuplicateFailedMethod(
                        $subscriber,
                        $failedMethod,
                        $method->getName(),
                    );
                }

                $failedMethod = $method->getName();
            }

            if ($method->getAttributes(Setup::class)) {
                if ($setupMethod !== null) {
                    throw new DuplicateSetupMethod(
                        $subscriber,
                        $setupMethod,
                        $method->getName(),
                    );
                }

                $setupMethod = $method->getName();
            }

            if ($method->getAttributes(Cleanup::class)) {
                if ($cleanupMethod !== null) {
                    throw new DuplicateCleanupMethod(
                        $subscriber,
                        $cleanupMethod,
                        $method->getName(),
                    );
                }

                if ($teardownMethod !== null) {
                    throw new MixedTeardownAndCleanupMethods(
                        $subscriber,
                        $teardownMethod,
                        $method->getName(),
                    );
                }

                $cleanupMethod = $method->getName();
            }

            if (!$method->getAttributes(Teardown::class)) {
                continue;
            }

            if ($teardownMethod !== null) {
                throw new DuplicateTeardownMethod(
                    $subscriber,
                    $teardownMethod,
                    $method->getName(),
                );
            }

            if ($cleanupMethod !== null) {
                throw new MixedTeardownAndCleanupMethods(
                    $subscriber,
                    $method->getName(),
                    $cleanupMethod,
                );
            }

            $teardownMethod = $method->getName();
        }

        if (array_key_exists(Subscribe::ALL, $subscribeMethods) && count($subscribeMethods) > 1) {
            throw DuplicateSubscribeMethod::mixedWithAll($subscriber);
        }

        $metadata = new SubscriberMetadata(
            $subscriberInfo->id,
            $subscriberInfo->group,
            $subscriberInfo->runMode,
            $subscribeMethods,
            $setupMethod,
            $teardownMethod,
            $failedMethod,
            $this->retryStrategy($reflector),
            $cleanupMethod,
        );

        $this->subscriberMetadata[$subscriber] = $metadata;

        return $metadata;
    }

    private function subscribeMethod(ReflectionMethod $method): SubscribeMethodMetadata
    {
        $arguments = [];

        foreach ($method->getParameters() as $parameter) {
            $type = $parameter->getType();

            if ($type === null) {
                throw ArgumentTypeNotSupported::missingType(
                    $method->getDeclaringClass()->getName(),
                    $method->getName(),
                    $parameter->getName(),
                );
            }

            $arguments[] = new ArgumentMetadata(
                $parameter->getName(),
                $this->typeResolver->resolve($type),
            );
        }

        return new SubscribeMethodMetadata(
            $method->getName(),
            $arguments,
        );
    }

    private function retryStrategy(ReflectionClass $reflector): string|null
    {
        $attributes = $reflector->getAttributes(RetryStrategy::class);

        if ($attributes === []) {
            return null;
        }

        $instance = $attributes[0]->newInstance();

        return $instance->name;
    }
}