Yiisoft\Classifier\Classifier

Complexity

Total Complexity

22

Size/Duplication

Total Lines	122
Duplicated Lines	0 %

Test Coverage

Coverage

98.21%

Importance

Changes	6
Bugs	1	Features	0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	2	1
A	withInterface()	0	7	2
A	withParent()	0	5	1
A	withAttribute()	0	7	2
A	scanFiles()	0	10	2
C	find()	0	53	14

<?php

declare(strict_types=1);

namespace Yiisoft\Classifier;

use ReflectionAttribute;
use ReflectionClass;
use Symfony\Component\Finder\Finder;

final class Classifier
{
    /**
     * @var string[]
     */
    private array $interfaces = [];
    /**
     * @var string[]
     */
    private array $attributes = [];
    /**
     * @psalm-var class-string
     */
    private ?string $parent = null;

    public function __construct(private string $directory)
    {
    }

    /**
     * @param string|string[] $interfaces
     */
    public function withInterface(string|array $interfaces): self
    {
        $new = clone $this;
        foreach ((array) $interfaces as $interface) {
            $new->interfaces[] = $interface;
        }
        return $new;
    }

    /**
     * @psalm-param class-string $parent
     */
    public function withParent(string $parent): self
    {
        $new = clone $this;
        $new->parent = $parent;
        return $new;
    }

    /**
     * @param string|string[] $attributes
     */
    public function withAttribute(string|array $attributes): self
    {
        $new = clone $this;
        foreach ((array) $attributes as $attribute) {
            $new->attributes[] = $attribute;
        }
        return $new;
    }

    public function find(): iterable
    {
        $countInterfaces = count($this->interfaces);
        $countAttributes = count($this->attributes);

        if ($countInterfaces === 0 && $countAttributes === 0 && $this->parent === null) {
            return [];
        }

        $this->scanFiles();

        $classesToFind = get_declared_classes();

        $baseDirectory = $this->directory;
        if (DIRECTORY_SEPARATOR === '\\') {
            $baseDirectory = str_replace('/', '\\', $baseDirectory);
        }

        foreach ($classesToFind as $className) {
            $reflection = new ReflectionClass($className);
            $filePath = $reflection->getFileName();
            if ($filePath === false || !str_starts_with($filePath, $baseDirectory)) {
                continue;
            }

            if ($countInterfaces > 0) {
                $interfaces = $reflection->getInterfaces();
                $interfaces = array_map(fn(ReflectionClass $class) => $class->getName(), $interfaces);

                if (count(array_intersect($this->interfaces, $interfaces)) !== $countInterfaces) {
                    continue;
                }
            }

            if ($countAttributes > 0) {
                $attributes = $reflection->getAttributes();
                $attributes = array_map(
                    static fn(ReflectionAttribute $attribute) => $attribute->getName(),
                    $attributes
                );

                if (count(array_intersect($this->attributes, $attributes)) !== $countAttributes) {
                    continue;
                }
            }

            if ($this->parent !== null) {
                if (!is_subclass_of($className, $this->parent)) {
                    continue;
                }
            }

            yield $className;
        }
    }

    /**
     * @psalm-suppress UnresolvableInclude
     */
    private function scanFiles(): void
    {
        $files = (new Finder())
            ->in($this->directory)
            ->name('*.php')
            ->sortByName()
            ->files();

        foreach ($files as $file) {
            require_once $file;
        }
    }
}