fix union type returns

scrumble-nl · Mar 5, 2024 · 8947c6c · 8947c6c
1 parent 12d6bb8
commit 8947c6c
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/Support/Generators/AttributePropertyGenerator.php b/src/Support/Generators/AttributePropertyGenerator.php
@@ -6,6 +6,7 @@
 
 use ReflectionClass;
 use ReflectionMethod;
+use ReflectionUnionType;
 use Illuminate\Database\Eloquent\Model;
 use Scrumble\TypeGenerator\Interfaces\IPropertyGenerator;
 
@@ -90,14 +91,13 @@ private function getPropertyType(ReflectionMethod $method): string
     {
         if (null !== ($returnType = $method->getReturnType())) {
             if($returnType instanceof ReflectionUnionType) {
-                $returnTypes = collect($returnType->getTypes())
-                ->map(function($returnType) {
-                    // @phpstan-ignore-next-line
-                    return $this->formatPhpReturnType($returnType->getName()) ;
-                })
-                ->join(' | ') . ($returnType->allowsNull() ? ' | null' : '');
+                return collect($returnType->getTypes())
+                    ->map(function($returnType) {
+                        return $this->formatPhpReturnType($returnType->getName()) ;
+                    })
+                    ->join(' | ') . ($returnType->allowsNull() ? ' | null' : '');
             }
-            
+
             // @phpstan-ignore-next-line
             return $this->formatPhpReturnType($returnType->getName()) . ($returnType->allowsNull() ? ' | null' : '');
         }