Basic support for simple Symfony #[AutowireLocator] attribute

https://symfony.com/blog/new-in-symfony-6-4-autowirelocator-and-autowireiterator-attributes

Author: RafaelKr

src/Rules/Symfony/ContainerInterfacePrivateServiceRule.php

@@ -5,8 +5,12 @@
 use PhpParser\Node;
 use PhpParser\Node\Expr\MethodCall;
 use PHPStan\Analyser\Scope;
+use PHPStan\BetterReflection\Reflection\Adapter\FakeReflectionAttribute;
+use PHPStan\BetterReflection\Reflection\Adapter\ReflectionProperty;
+use PHPStan\BetterReflection\Reflection\ReflectionAttribute;
 use PHPStan\Rules\Rule;
 use PHPStan\Rules\RuleErrorBuilder;
+use PHPStan\Symfony\Service;
 use PHPStan\Symfony\ServiceMap;
 use PHPStan\TrinaryLogic;
 use PHPStan\Type\ObjectType;
@@ -66,15 +70,29 @@ public function processNode(Node $node, Scope $scope): array
 		}
 
 		$serviceId = $this->serviceMap::getServiceIdFromNode($node->getArgs()[0]->value, $scope);
-		if ($serviceId !== null) {
-			$service = $this->serviceMap->getService($serviceId);
-			if ($service !== null && !$service->isPublic()) {
-				return [
-					RuleErrorBuilder::message(sprintf('Service "%s" is private.', $serviceId))
-						->identifier('symfonyContainer.privateService')
-						->build(),
-				];
-			}
+		if ($serviceId === null) {
+			return [];
+		}
+
+		$service = $this->serviceMap->getService($serviceId);
+		if ($service === null) {
+			return [];
+		}
+
+		$isContainerInterfaceType = $isContainerType->yes() || $isPsrContainerType->yes();
+		if (
+			$isContainerInterfaceType &&
+			$this->isAutowireLocator($node, $scope, $service)
+		) {
+			return [];
+		}
+
+		if (!$service->isPublic()) {
+			return [
+				RuleErrorBuilder::message(sprintf('Service "%s" is private.', $serviceId))
+					->identifier('symfonyContainer.privateService')
+					->build(),
+			];
 		}
 
 		return [];
@@ -92,4 +110,61 @@ private function isServiceSubscriber(Type $containerType, Scope $scope): Trinary
 		return $isContainerServiceSubscriber->or($serviceSubscriberInterfaceType->isSuperTypeOf($containedClassType));
 	}
 
+	private function isAutowireLocator(Node $node, Scope $scope, Service $service): bool
+	{
+		if (!$node->var instanceof Node\Expr\PropertyFetch) {
+			return false;
+		}
+
+		$containerInterfacePropertyName = $node->var->name->name;
+		$classProperty = $scope
+			->getClassReflection()
+			->getProperty($containerInterfacePropertyName, $scope);
+
+		if (!$classProperty) {
+			return false;
+		}
+
+		/* @var ReflectionProperty $classPropertyReflection */
+		$classPropertyReflection = $classProperty->getNativeReflection();
+		$autowireLocatorAttributes = $classPropertyReflection
+			->getAttributes('Symfony\Component\DependencyInjection\Attribute\AutowireLocator');
+
+		return $this->isAutowireLocatorService($autowireLocatorAttributes, $service);
+	}
+
+	/**
+	 * @param  array<FakeReflectionAttribute|ReflectionAttribute>  $autowireLocatorAttributes
+	 * @param  Service  $service
+	 * @return bool
+	 */
+	private function isAutowireLocatorService(array $autowireLocatorAttributes, Service $service): bool
+	{
+		foreach ($autowireLocatorAttributes as $autowireLocatorAttribute) {
+			foreach ($autowireLocatorAttribute->getArgumentsExpressions() as $autowireLocatorServices) {
+				foreach ($autowireLocatorServices->items as $autowireLocatorServiceNode) {
+					/** @var Node\Expr $autowireLocatorService */
+					$autowireLocatorServiceExpr = $autowireLocatorServiceNode->value;
+
+					switch (get_class($autowireLocatorServiceExpr)) {
+						case Node\Scalar\String_::class:
+							$autowireLocatorServiceClass = $autowireLocatorServiceExpr->value;
+							break;
+						case Node\Expr\ClassConstFetch::class:
+							$autowireLocatorServiceClass = $autowireLocatorServiceExpr->class->toString();
+							break;
+						default:
+							$autowireLocatorServiceClass = null;
+					}
+
+					if ($service->getId() === $autowireLocatorServiceClass) {
+						return true;
+					}
+				}
+			}
+		}
+
+		return false;
+	}
+
 }