jaroslavtyc/drd-plus-codes

<?php declare(strict_types=1);

namespace DrdPlus\Tests\Codes;

use DrdPlus\Codes\Code;
use DrdPlus\Codes\Partials\AbstractCode;

/**
 * @method static assertMatchesRegularExpression($regexp, $value, $message = '')
 */
trait GetCodeClassesTrait
{

    private static $codeClasses;

    /**
     * @param string $rootClass
     * @return array|string[]|AbstractCode[]
     * @throws \ReflectionException
     */
    protected function getCodeClasses(string $rootClass = Code::class): array
    {
        if (self::$codeClasses === null) {
            $codeReflection = new \ReflectionClass($rootClass);
            $rootDir = \dirname($codeReflection->getFileName());
            $rootNamespace = $codeReflection->getNamespaceName();

            self::$codeClasses = $this->scanForCodeClasses($rootDir, $rootNamespace);
        }

        return self::$codeClasses;
    }

    /**
     * @param string $rootDir
     * @param string $rootNamespace
     * @return array|string[]
     * @throws \ReflectionException
     */
    protected function scanForCodeClasses(string $rootDir, string $rootNamespace): array
    {
        $codeClasses = [];
        foreach (\scandir($rootDir, SCANDIR_SORT_NONE) as $folder) {
            $folderFullPath = $rootDir . DIRECTORY_SEPARATOR . $folder;
            if ($folder !== '.' && $folder !== '..') {
                if (\is_dir($folderFullPath)) {
                    foreach ($this->scanForCodeClasses($folderFullPath, $rootNamespace . '\\' . $folder) as $foundCode) {
                        $codeClasses[] = $foundCode;
                    }
                } elseif (\is_file($folderFullPath) && \preg_match('~(?<classBasename>\w+(?:Code)?)\.php$~', $folder, $matches)) {
                    $reflectionClass = new \ReflectionClass($rootNamespace . '\\' . $matches['classBasename']);
                    if (!$reflectionClass->isAbstract() && $reflectionClass->implementsInterface(Code::class)) {
                        self::assertMatchesRegularExpression(
                            '~Code$~',
                            $reflectionClass->getName(),
                            'Every single code should ends by "Code"'
                        );
                        $codeClasses[] = $reflectionClass->getName();
                    }
                }
            }
        }

        return $codeClasses;
    }
}