Add support for @return $this phpdoc

padawan-php · Jan 9, 2017 · 4b5594f · 4b5594f
1 parent a926bb6
commit 4b5594f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/Padawan/Domain/Project/Node/ClassData.php b/src/Padawan/Domain/Project/Node/ClassData.php
@@ -87,6 +87,11 @@ public function addInterface($interface)
     }
     public function addMethod(MethodData $method)
     {
+        if ($method->return instanceof FQCN) {
+            if ($method->return->getLast() === 'this') {
+                $method->return = $this->fqcn;
+            }
+        }
         $this->methods->add($method);
     }
     public function getMethod($methodName)