rector/rules/Restoration/ClassMap/ExistingClassesProvider.php

<?php

declare(strict_types=1);

namespace Rector\Restoration\ClassMap;

use Nette\Loaders\RobotLoader;
use Nette\Utils\Arrays;
use Symplify\ComposerJsonManipulator\ComposerJsonFactory;

final class ExistingClassesProvider
{
    /**
     * @var string[]
     */
    private $existingClasses = [];

    /**
     * @var ComposerJsonFactory
     */
    private $composerJsonFactory;

    public function __construct(ComposerJsonFactory $composerJsonFactory)
    {
        $this->composerJsonFactory = $composerJsonFactory;
    }

    /**
     * @return string[]
     */
    public function provide(): array
    {
        if ($this->existingClasses === []) {
            $psr4Paths = $this->getPsr4PathFromComposerJson();

            /** @var string[] $existingClasses */
            $existingClasses = $this->findClassesInDirectories($psr4Paths);

            $declaredClasses = get_declared_classes();
            $existingClasses = array_merge($existingClasses, $declaredClasses);

            $this->existingClasses = $existingClasses;
        }

        return $this->existingClasses;
    }

    /**
     * @return string[]
     */
    private function getPsr4PathFromComposerJson(): array
    {
        $composerJsonFilePath = getcwd() . '/composer.json';
        $composerJson = $this->composerJsonFactory->createFromFilePath($composerJsonFilePath);

        $psr4AndClassmapDirectories = $composerJson->getPsr4AndClassmapDirectories();
        return Arrays::flatten($psr4AndClassmapDirectories);
    }

    /**
     * @param string[] $directories
     * @return string[]
     */
    private function findClassesInDirectories(array $directories): array
    {
        $robotLoader = new RobotLoader();
        $robotLoader->setTempDirectory(sys_get_temp_dir() . '/rector');

        foreach ($directories as $directory) {
            $robotLoader->addDirectory(getcwd() . '/' . $directory);
        }

        $classNames = [];
        foreach (array_keys($robotLoader->getIndexedClasses()) as $className) {
            if (! is_string($className)) {
                continue;
            }

            $classNames[] = $className;
        }

        return $classNames;
    }
}