Added AggregateSourceStubber

src/BetterReflection.php

@@ -11,6 +11,8 @@
 use Roave\BetterReflection\Reflector\FunctionReflector;
 use Roave\BetterReflection\SourceLocator\Ast\Locator as AstLocator;
 use Roave\BetterReflection\SourceLocator\Ast\Parser\MemoizingParser;
+use Roave\BetterReflection\SourceLocator\SourceStubber\AggregateSourceStubber;
+use Roave\BetterReflection\SourceLocator\SourceStubber\PhpStormStubsSourceStubber;
 use Roave\BetterReflection\SourceLocator\SourceStubber\ReflectionSourceStubber;
 use Roave\BetterReflection\SourceLocator\SourceStubber\SourceStubber;
 use Roave\BetterReflection\SourceLocator\Type\AggregateSourceLocator;
@@ -94,6 +96,9 @@ public function findReflectionsOnLine() : FindReflectionOnLine
     public function sourceStubber() : SourceStubber
     {
         return $this->sourceStubber
-            ?? $this->sourceStubber = new ReflectionSourceStubber();
+            ?? $this->sourceStubber = new AggregateSourceStubber(
+                new PhpStormStubsSourceStubber($this->phpParser()),
+                new ReflectionSourceStubber()
+            );
     }
 }

src/SourceLocator/SourceStubber/AggregateSourceStubber.php

@@ -0,0 +1,52 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Roave\BetterReflection\SourceLocator\SourceStubber;
+
+use ReflectionClass as CoreReflectionClass;
+use ReflectionFunction as CoreReflectionFunction;
+use function array_merge;
+
+class AggregateSourceStubber implements SourceStubber
+{
+    /** @var SourceStubber[] */
+    private $sourceStubbers;
+
+    public function __construct(SourceStubber $sourceStubber, SourceStubber ...$otherSourceStubbers)
+    {
+        $this->sourceStubbers = array_merge([$sourceStubber], $otherSourceStubbers);
+    }
+
+    /**
+     * {@inheritDoc}
+     */
+    public function generateClassStub(CoreReflectionClass $classReflection) : ?string
+    {
+        foreach ($this->sourceStubbers as $sourceStubber) {
+            $stub = $sourceStubber->generateClassStub($classReflection);
+
+            if ($stub !== null) {
+                return $stub;
+            }
+        }
+
+        return null;
+    }
+
+    /**
+     * {@inheritDoc}
+     */
+    public function generateFunctionStub(CoreReflectionFunction $functionReflection) : ?string
+    {
+        foreach ($this->sourceStubbers as $sourceStubber) {
+            $stub = $sourceStubber->generateFunctionStub($functionReflection);
+
+            if ($stub !== null) {
+                return $stub;
+            }
+        }
+
+        return null;
+    }
+}