ArgumentsProcessor - allow only mapped classes

phpstan · Jan 15, 2024 · 0edf5b0 · 0edf5b0
1 parent 08b193c
commit 0edf5b0
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/src/Type/Doctrine/ArgumentsProcessor.php b/src/Type/Doctrine/ArgumentsProcessor.php
@@ -13,6 +13,14 @@
 class ArgumentsProcessor
 {
 
+	/** @var ObjectMetadataResolver */
+	private $objectMetadataResolver;
+
+	public function __construct(ObjectMetadataResolver $objectMetadataResolver)
+	{
+		$this->objectMetadataResolver = $objectMetadataResolver;
+	}
+
 	/**
 	 * @param Arg[] $methodCallArgs
 	 * @return list<mixed>
@@ -49,7 +57,13 @@ public function processArgs(
 			}
 
 			if ($value->isClassStringType()->yes() && count($value->getClassStringObjectType()->getObjectClassNames()) === 1) {
-				$args[] = $value->getClassStringObjectType()->getObjectClassNames()[0];
+				/** @var class-string $className */
+				$className = $value->getClassStringObjectType()->getObjectClassNames()[0];
+				if ($this->objectMetadataResolver->isTransient($className)) {
+					throw new DynamicQueryBuilderArgumentException();
+				}
+
+				$args[] = $className;
 				continue;
 			}