getType(), getReturnType(): added option $asObject that returns Nette…

…\Utils\Type (requires nette/utils 3.2.5)
nette · Sep 23, 2021 · 9b79d3c · 9b79d3c
1 parent 3405e81
commit 9b79d3c
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 6 deletions.
diff --git a/src/PhpGenerator/Parameter.php b/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;
 	}
 
 

diff --git a/src/PhpGenerator/Property.php b/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;
 	}
 
 

diff --git a/src/PhpGenerator/Traits/FunctionLike.php b/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;
 	}