getType(), getReturnType(): added option $asObject that returns Nette\Utils\Type

(requires nette/utils 3.2.5)

Author: dg

src/PhpGenerator/Parameter.php

@@ -10,6 +10,7 @@
 namespace Nette\PhpGenerator;
 
 use Nette;
+use Nette\Utils\Type;
 
 
 /**
@@ -61,9 +62,14 @@ public function setType(?string $type): self
 	}
 
 
-	public function getType(): ?string
+	/**
+	 * @return Type|string|null
+	 */
+	public function getType(bool $asObject = false)
 	{
-		return $this->type;
+		return $asObject && $this->type
+			? Type::fromString($this->type)
+			: $this->type;
 	}
 
 

src/PhpGenerator/Property.php

@@ -10,6 +10,7 @@
 namespace Nette\PhpGenerator;
 
 use Nette;
+use Nette\Utils\Type;
 
 
 /**
@@ -81,9 +82,14 @@ public function setType(?string $type): self
 	}
 
 
-	public function getType(): ?string
+	/**
+	 * @return Type|string|null
+	 */
+	public function getType(bool $asObject = false)
 	{
-		return $this->type;
+		return $asObject && $this->type
+			? Type::fromString($this->type)
+			: $this->type;
 	}
 
 

src/PhpGenerator/Traits/FunctionLike.php

@@ -12,6 +12,7 @@
 use Nette;
 use Nette\PhpGenerator\Dumper;
 use Nette\PhpGenerator\Parameter;
+use Nette\Utils\Type;
 
 
 /**
@@ -130,9 +131,14 @@ public function setReturnType(?string $type): self
 	}
 
 
-	public function getReturnType(): ?string
+	/**
+	 * @return Type|string|null
+	 */
+	public function getReturnType(bool $asObject = false)
 	{
-		return $this->returnType;
+		return $asObject && $this->returnType
+			? Type::fromString($this->returnType)
+			: $this->returnType;
 	}