|  | 
|  | 1 | +<?php declare(strict_types = 1); | 
|  | 2 | + | 
|  | 3 | +namespace PHPStan\Symfony; | 
|  | 4 | + | 
|  | 5 | +use PHPStan\Reflection\ClassReflection; | 
|  | 6 | +use PHPStan\Reflection\ReflectionProvider; | 
|  | 7 | +use Symfony\Component\Messenger\Handler\MessageSubscriberInterface; | 
|  | 8 | +use function class_exists; | 
|  | 9 | +use function count; | 
|  | 10 | +use function is_array; | 
|  | 11 | +use function is_int; | 
|  | 12 | +use function is_string; | 
|  | 13 | + | 
|  | 14 | +final class MessageMapFactory | 
|  | 15 | +{ | 
|  | 16 | + | 
|  | 17 | +	private const MESSENGER_HANDLER_TAG = 'messenger.message_handler'; | 
|  | 18 | +	private const DEFAULT_HANDLER_METHOD = '__invoke'; | 
|  | 19 | + | 
|  | 20 | +	/** @var ReflectionProvider */ | 
|  | 21 | +	private $reflectionProvider; | 
|  | 22 | + | 
|  | 23 | +	/** @var ServiceMap */ | 
|  | 24 | +	private $serviceMap; | 
|  | 25 | + | 
|  | 26 | +	public function __construct(ServiceMap $symfonyServiceMap, ReflectionProvider $reflectionProvider) | 
|  | 27 | +	{ | 
|  | 28 | +		$this->serviceMap = $symfonyServiceMap; | 
|  | 29 | +		$this->reflectionProvider = $reflectionProvider; | 
|  | 30 | +	} | 
|  | 31 | + | 
|  | 32 | +	public function create(): MessageMap | 
|  | 33 | +	{ | 
|  | 34 | +		$returnTypesMap = []; | 
|  | 35 | + | 
|  | 36 | +		foreach ($this->serviceMap->getServices() as $service) { | 
|  | 37 | +			$serviceClass = $service->getClass(); | 
|  | 38 | + | 
|  | 39 | +			if ($serviceClass === null) { | 
|  | 40 | +				continue; | 
|  | 41 | +			} | 
|  | 42 | + | 
|  | 43 | +			foreach ($service->getTags() as $tag) { | 
|  | 44 | +				if ($tag->getName() !== self::MESSENGER_HANDLER_TAG) { | 
|  | 45 | +					continue; | 
|  | 46 | +				} | 
|  | 47 | + | 
|  | 48 | +				if (!$this->reflectionProvider->hasClass($serviceClass)) { | 
|  | 49 | +					continue; | 
|  | 50 | +				} | 
|  | 51 | + | 
|  | 52 | +				$reflectionClass = $this->reflectionProvider->getClass($serviceClass); | 
|  | 53 | + | 
|  | 54 | +				/** @var array{handles?: class-string, method?: string} $tagAttributes */ | 
|  | 55 | +				$tagAttributes = $tag->getAttributes(); | 
|  | 56 | + | 
|  | 57 | +				if (isset($tagAttributes['handles'])) { | 
|  | 58 | +					$handles = [$tagAttributes['handles'] => ['method' => $tagAttributes['method'] ?? self::DEFAULT_HANDLER_METHOD]]; | 
|  | 59 | +				} else { | 
|  | 60 | +					$handles = $this->guessHandledMessages($reflectionClass); | 
|  | 61 | +				} | 
|  | 62 | + | 
|  | 63 | +				foreach ($handles as $messageClassName => $options) { | 
|  | 64 | +					$methodName = $options['method'] ?? self::DEFAULT_HANDLER_METHOD; | 
|  | 65 | + | 
|  | 66 | +					if (!$reflectionClass->hasNativeMethod($methodName)) { | 
|  | 67 | +						continue; | 
|  | 68 | +					} | 
|  | 69 | + | 
|  | 70 | +					$methodReflection = $reflectionClass->getNativeMethod($methodName); | 
|  | 71 | + | 
|  | 72 | +					foreach ($methodReflection->getVariants() as $variant) { | 
|  | 73 | +						$returnTypesMap[$messageClassName][] = $variant->getReturnType(); | 
|  | 74 | +					} | 
|  | 75 | +				} | 
|  | 76 | +			} | 
|  | 77 | +		} | 
|  | 78 | + | 
|  | 79 | +		$messageMap = []; | 
|  | 80 | +		foreach ($returnTypesMap as $messageClassName => $returnTypes) { | 
|  | 81 | +			if (count($returnTypes) !== 1) { | 
|  | 82 | +				continue; | 
|  | 83 | +			} | 
|  | 84 | + | 
|  | 85 | +			$messageMap[$messageClassName] = $returnTypes[0]; | 
|  | 86 | +		} | 
|  | 87 | + | 
|  | 88 | +		return new MessageMap($messageMap); | 
|  | 89 | +	} | 
|  | 90 | + | 
|  | 91 | +	/** @return iterable<string, array<string, string>> */ | 
|  | 92 | +	private function guessHandledMessages(ClassReflection $reflectionClass): iterable | 
|  | 93 | +	{ | 
|  | 94 | +		if ($reflectionClass->implementsInterface(MessageSubscriberInterface::class)) { | 
|  | 95 | +			$className = $reflectionClass->getName(); | 
|  | 96 | + | 
|  | 97 | +			foreach ($className::getHandledMessages() as $index => $value) { | 
|  | 98 | +				$containOptions = self::containOptions($index, $value); | 
|  | 99 | +				if ($containOptions === true) { | 
|  | 100 | +					yield $index => $value; | 
|  | 101 | +				} elseif ($containOptions === false) { | 
|  | 102 | +					yield $value => ['method' => self::DEFAULT_HANDLER_METHOD]; | 
|  | 103 | +				} | 
|  | 104 | +			} | 
|  | 105 | + | 
|  | 106 | +			return; | 
|  | 107 | +		} | 
|  | 108 | + | 
|  | 109 | +		if (!$reflectionClass->hasNativeMethod(self::DEFAULT_HANDLER_METHOD)) { | 
|  | 110 | +			return; | 
|  | 111 | +		} | 
|  | 112 | + | 
|  | 113 | +		$methodReflection = $reflectionClass->getNativeMethod(self::DEFAULT_HANDLER_METHOD); | 
|  | 114 | + | 
|  | 115 | +		$variants = $methodReflection->getVariants(); | 
|  | 116 | +		if (count($variants) !== 1) { | 
|  | 117 | +			return; | 
|  | 118 | +		} | 
|  | 119 | + | 
|  | 120 | +		$parameters = $variants[0]->getParameters(); | 
|  | 121 | + | 
|  | 122 | +		if (count($parameters) !== 1) { | 
|  | 123 | +			return; | 
|  | 124 | +		} | 
|  | 125 | + | 
|  | 126 | +		$classNames = $parameters[0]->getType()->getObjectClassNames(); | 
|  | 127 | + | 
|  | 128 | +		if (count($classNames) !== 1) { | 
|  | 129 | +			return; | 
|  | 130 | +		} | 
|  | 131 | + | 
|  | 132 | +		yield $classNames[0] => ['method' => self::DEFAULT_HANDLER_METHOD]; | 
|  | 133 | +	} | 
|  | 134 | + | 
|  | 135 | +	/** | 
|  | 136 | +	 * @param mixed $index | 
|  | 137 | +	 * @param mixed $value | 
|  | 138 | +	 * @phpstan-assert-if-true =class-string $index | 
|  | 139 | +	 * @phpstan-assert-if-true =array<string, mixed> $value | 
|  | 140 | +	 * @phpstan-assert-if-false =int $index | 
|  | 141 | +	 * @phpstan-assert-if-false =class-string $value | 
|  | 142 | +	 */ | 
|  | 143 | +	private static function containOptions($index, $value): ?bool | 
|  | 144 | +	{ | 
|  | 145 | +		if (is_string($index) && class_exists($index) && is_array($value)) { | 
|  | 146 | +			return true; | 
|  | 147 | +		} elseif (is_int($index) && is_string($value) && class_exists($value)) { | 
|  | 148 | +			return false; | 
|  | 149 | +		} | 
|  | 150 | + | 
|  | 151 | +		return null; | 
|  | 152 | +	} | 
|  | 153 | + | 
|  | 154 | +} | 
0 commit comments