extend class-string service id support

Author: Laytan Laats

src/Type/ContainerDynamicReturnTypeExtension.php

@@ -4,7 +4,10 @@
 
 use mglaman\PHPStanDrupal\Drupal\DrupalServiceDefinition;
 use mglaman\PHPStanDrupal\Drupal\ServiceMap;
+use PhpParser\Node;
+use PhpParser\Node\Expr\ClassConstFetch;
 use PhpParser\Node\Expr\MethodCall;
+use PhpParser\Node\Name\FullyQualified;
 use PhpParser\Node\Scalar\String_;
 use PhpParser\Node\VariadicPlaceholder;
 use PHPStan\Analyser\Scope;
@@ -52,13 +55,12 @@ public function getTypeFromMethodCall(
             throw new ShouldNotHappenException();
         }
         $arg1 = $arg1->value;
-        if (!$arg1 instanceof String_) {
-            // @todo determine what these types are.
+
+        $serviceId = $this->getServiceId($arg1);
+        if ($serviceId === null) {
             return $returnType;
         }
 
-        $serviceId = $arg1->value;
-
         if ($methodReflection->getName() === 'get') {
             $service = $this->serviceMap->getService($serviceId);
             if ($service instanceof DrupalServiceDefinition) {
@@ -73,4 +75,18 @@ public function getTypeFromMethodCall(
 
         throw new ShouldNotHappenException();
     }
+
+    protected function getServiceId(Node $arg1): ?string
+    {
+        if ($arg1 instanceof String_) {
+            // @todo determine what these types are.
+            return $arg1->value;
+        }
+
+        if ($arg1 instanceof ClassConstFetch && $arg1->class instanceof FullyQualified) {
+            return (string) $arg1->class;
+        }
+
+        return null;
+    }
 }

tests/src/Type/data/container.php

@@ -18,4 +18,5 @@ function test(): void {
     assertType(MyService::class, $container->get('service_map.concrete_service_with_a_parent_which_has_a_parent'));
     assertType(Override::class, $container->get('service_map.concrete_service_overriding_definition_of_its_parent'));
     assertType(Concrete::class, $container->get('service_map.concrete_overriding_its_parent_which_has_a_parent'));
+    assertType(MyService::class, $container->get(MyService::class));
 }

tests/src/Type/data/drupal-class-resolver.php

@@ -17,4 +17,9 @@ function test(): void {
     assertType(MyService::class, \Drupal::service('class_resolver')->getInstanceFromDefinition('service_map.my_service'));
     assertType(MyService::class, \Drupal::classResolver()->getInstanceFromDefinition('service_map.my_service'));
     assertType(MyService::class, \Drupal::classResolver('service_map.my_service'));
+
+    assertType(MyService::class, (new ClassResolver())->getInstanceFromDefinition(MyService::class));
+    assertType(MyService::class, \Drupal::service('class_resolver')->getInstanceFromDefinition(MyService::class));
+    assertType(MyService::class, \Drupal::classResolver()->getInstanceFromDefinition(MyService::class));
+    assertType(MyService::class, \Drupal::classResolver(MyService::class));
 }