Adds return type to function

Author: jaapio

src/phpDocumentor/Reflection/Php/Factory/Function_.php

@@ -19,6 +19,7 @@
 use phpDocumentor\Reflection\Php\Function_ as FunctionDescriptor;
 use phpDocumentor\Reflection\Php\ProjectFactoryStrategy;
 use phpDocumentor\Reflection\Php\StrategyContainer;
+use phpDocumentor\Reflection\TypeResolver;
 use phpDocumentor\Reflection\Types\Context;
 use PhpParser\Comment\Doc;
 use PhpParser\Node\Stmt\Function_ as FunctionNode;
@@ -57,7 +58,13 @@ protected function doCreate($object, StrategyContainer $strategies, Context $con
     {
         $docBlock = $this->createDocBlock($strategies, $object->getDocComment(), $context);
 
-        $function = new FunctionDescriptor($object->fqsen, $docBlock, new Location($object->getLine()));
+        $returnType = null;
+        if ($object->returnType !== null) {
+            $typeResolver = new TypeResolver();
+            $returnType = $typeResolver->resolve($object->returnType, $context);
+        }
+
+        $function = new FunctionDescriptor($object->fqsen, $docBlock, new Location($object->getLine()), $returnType);
 
         foreach ($object->params as $param) {
             $strategy = $strategies->findMatching($param);

src/phpDocumentor/Reflection/Php/Function_.php

@@ -16,6 +16,8 @@
 use phpDocumentor\Reflection\Element;
 use phpDocumentor\Reflection\Fqsen;
 use phpDocumentor\Reflection\Location;
+use phpDocumentor\Reflection\Type;
+use phpDocumentor\Reflection\Types\Mixed_;
 
 /**
  * Descriptor representing a function
@@ -42,21 +44,37 @@ final class Function_ implements Element
      */
     private $location;
 
+    /**
+     * @var Type
+     */
+    private $returnType;
+
     /**
      * Initializes the object.
      *
      * @param Fqsen $fqsen
      * @param DocBlock|null $docBlock
+     * @param Location|null $location
+     * @param Type|null $returnType
      */
-    public function __construct(Fqsen $fqsen, DocBlock $docBlock = null, Location $location = null)
-    {
+    public function __construct(
+        Fqsen $fqsen,
+        DocBlock $docBlock = null,
+        Location $location = null,
+        Type $returnType = null
+    ) {
         if ($location === null) {
             $location = new Location(-1);
         }
 
+        if ($returnType ===  null) {
+            $returnType = new Mixed_();
+        }
+
         $this->fqsen = $fqsen;
         $this->docBlock = $docBlock;
         $this->location = $location;
+        $this->returnType = $returnType;
     }
 
     /**
@@ -116,4 +134,12 @@ public function getLocation()
     {
         return $this->location;
     }
+
+    /**
+     * @return Type
+     */
+    public function getReturnType() : Type
+    {
+        return $this->returnType;
+    }
 }

tests/unit/phpDocumentor/Reflection/Php/Function_Test.php

@@ -15,6 +15,8 @@
 use \Mockery as m;
 use phpDocumentor\Reflection\DocBlock;
 use phpDocumentor\Reflection\Fqsen;
+use phpDocumentor\Reflection\Types\Mixed_;
+use phpDocumentor\Reflection\Types\String_;
 
 /**
  * Tests the functionality for the Function_ class.
@@ -81,4 +83,31 @@ public function testGetDocblock()
     {
         $this->assertSame($this->docBlock, $this->fixture->getDocBlock());
     }
+
+    /**
+     * @covers ::getReturnType
+     * @covers ::__construct
+     */
+    public function testGetDefaultReturnType()
+    {
+        $method = new Function_($this->fqsen);
+        $this->assertEquals(new Mixed_(), $method->getReturnType());
+    }
+
+    /**
+     * @covers ::getReturnType
+     * @covers ::__construct
+     */
+    public function testGetReturnTypeFromConstructor()
+    {
+        $returnType = new String_();
+        $method = new Function_(
+            $this->fqsen,
+            null,
+            null,
+            $returnType
+        );
+
+        $this->assertSame($returnType, $method->getReturnType());
+    }
 }