|  1:  | <?php | 
|  2:  |  | 
|  3:  | declare(strict_types=1); | 
|  4:  |  | 
|  5:  | namespace PHPStan\BetterReflection\SourceLocator\Type; | 
|  6:  |  | 
|  7:  | use Iterator; | 
|  8:  | use PHPStan\BetterReflection\Identifier\Identifier; | 
|  9:  | use PHPStan\BetterReflection\Identifier\IdentifierType; | 
| 10:  | use PHPStan\BetterReflection\Reflection\Reflection; | 
| 11:  | use PHPStan\BetterReflection\Reflector\Reflector; | 
| 12:  | use PHPStan\BetterReflection\SourceLocator\Ast\Locator; | 
| 13:  | use PHPStan\BetterReflection\SourceLocator\Exception\InvalidFileInfo; | 
| 14:  | use PHPStan\BetterReflection\SourceLocator\Exception\InvalidFileLocation; | 
| 15:  | use SplFileInfo; | 
| 16:  |  | 
| 17:  | use function array_filter; | 
| 18:  | use function array_map; | 
| 19:  | use function array_values; | 
| 20:  | use function iterator_to_array; | 
| 21:  | use function pathinfo; | 
| 22:  |  | 
| 23:  | use const PATHINFO_EXTENSION; | 
| 24:  |  | 
| 25:  |  | 
| 26:  |  | 
| 27:  |  | 
| 28:  | class FileIteratorSourceLocator implements SourceLocator | 
| 29:  | { | 
| 30:  |      | 
| 31:  |  | 
| 32:  |  | 
| 33:  |     private $aggregateSourceLocator = null; | 
| 34:  |  | 
| 35:  |      | 
| 36:  |     private $fileSystemIterator; | 
| 37:  |      | 
| 38:  |  | 
| 39:  |  | 
| 40:  |     private $astLocator; | 
| 41:  |  | 
| 42:  |      | 
| 43:  |  | 
| 44:  |  | 
| 45:  |  | 
| 46:  |  | 
| 47:  |     public function __construct(Iterator $fileInfoIterator, Locator $astLocator) | 
| 48:  |     { | 
| 49:  |         $this->astLocator = $astLocator; | 
| 50:  |         foreach ($fileInfoIterator as $fileInfo) { | 
| 51:  |             if (! $fileInfo instanceof SplFileInfo) { | 
| 52:  |                 throw InvalidFileInfo::fromNonSplFileInfo($fileInfo); | 
| 53:  |             } | 
| 54:  |         } | 
| 55:  |  | 
| 56:  |         $this->fileSystemIterator = $fileInfoIterator; | 
| 57:  |     } | 
| 58:  |  | 
| 59:  |      | 
| 60:  |     private function getAggregatedSourceLocator(): AggregateSourceLocator | 
| 61:  |     { | 
| 62:  |          | 
| 63:  |         return $this->aggregateSourceLocator | 
| 64:  |             ?? $this->aggregateSourceLocator = new AggregateSourceLocator(array_values(array_filter(array_map(function (SplFileInfo $item): ?\PHPStan\BetterReflection\SourceLocator\Type\SingleFileSourceLocator { | 
| 65:  |                 $realPath = $item->getRealPath(); | 
| 66:  |  | 
| 67:  |                 if (! ($item->isFile() && pathinfo($realPath, PATHINFO_EXTENSION) === 'php')) { | 
| 68:  |                     return null; | 
| 69:  |                 } | 
| 70:  |  | 
| 71:  |                 return new SingleFileSourceLocator($realPath, $this->astLocator); | 
| 72:  |             }, iterator_to_array($this->fileSystemIterator))))); | 
| 73:  |     } | 
| 74:  |  | 
| 75:  |      | 
| 76:  |  | 
| 77:  |  | 
| 78:  |  | 
| 79:  |  | 
| 80:  |     public function locateIdentifier(Reflector $reflector, Identifier $identifier): ?\PHPStan\BetterReflection\Reflection\Reflection | 
| 81:  |     { | 
| 82:  |         return $this->getAggregatedSourceLocator()->locateIdentifier($reflector, $identifier); | 
| 83:  |     } | 
| 84:  |  | 
| 85:  |      | 
| 86:  |  | 
| 87:  |  | 
| 88:  |  | 
| 89:  |  | 
| 90:  |     public function locateIdentifiersByType(Reflector $reflector, IdentifierType $identifierType): array | 
| 91:  |     { | 
| 92:  |         return $this->getAggregatedSourceLocator()->locateIdentifiersByType($reflector, $identifierType); | 
| 93:  |     } | 
| 94:  | } | 
| 95:  |  |