Namespace prefixes fix.

igniphp · May 14, 2018 · 051b88c · 051b88c
1 parent e8f1ebc
commit 051b88c
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/Exception/ReflectionApiException.php b/src/Exception/ReflectionApiException.php
@@ -40,4 +40,9 @@ public static function forFinalAbstract(): self
     {
         return new self("Expression cannot be final and abstract at the same time.");
     }
+
+    public static function forUnknownType(string $type): self
+    {
+        return new self("Passed type {$type} is not recognized. Did you forgot to include it in the autoload?");
+    }
 }
diff --git a/src/ReflectionApi/RuntimeMethod.php b/src/ReflectionApi/RuntimeMethod.php
@@ -3,6 +3,7 @@
 namespace Igni\Utils\ReflectionApi;
 
 use Igni\Utils\Exception\ReflectionApiException;
+use Igni\Utils\ReflectionApi;
 
 final class RuntimeMethod implements CodeGenerator
 {
@@ -16,6 +17,8 @@ final class RuntimeMethod implements CodeGenerator
     private $body = [];
     private $name;
 
+    private const DEFAULT_TYPES = ['string', 'int', 'object', 'bool', 'float', 'array', 'callable', 'void'];
+
     public function __construct(string $name)
     {
         $this->name = $name;
@@ -28,6 +31,10 @@ public function getName(): string
 
     public function setReturnType(string $type): self
     {
+        if (!in_array($type, self::DEFAULT_TYPES) && !class_exists($type) && !interface_exists($type)) {
+            throw ReflectionApiException::forUnknownType($type);
+        }
+
         $this->returnType = $type;
 
         return $this;
@@ -97,7 +104,11 @@ public function generateCode(): string
 
         $code .= ')';
         if ($this->returnType) {
-            $code .= ": {$this->returnType}";
+            if (class_exists($this->returnType) || interface_exists($this->returnType)) {
+                $code .= ": \\{$this->returnType}";
+            } else {
+                $code .= ": {$this->returnType}";
+            }
         }
 
         if ($this->isAbstract()) {