Yiisoft\Classifier\PhpParserClassifier

Complexity

Total Complexity

4

Size/Duplication

Total Lines	36
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	13	1
A	getAvailableClasses()	0	12	3

<?php

declare(strict_types=1);

namespace Yiisoft\Classifier;

use PhpParser\NodeTraverser;
use PhpParser\NodeVisitor\NameResolver;
use PhpParser\Parser;
use PhpParser\ParserFactory;

final class PhpParserClassifier extends AbstractClassifier
{
    private Parser $parser;
    private ClassifierVisitor $visitor;
    private NodeTraverser $nodeTraverser;

    public function __construct(string $directory, string ...$directories)
    {
        parent::__construct($directory, ...$directories);
        $this->parser = (new ParserFactory())->create(ParserFactory::PREFER_PHP7);

        $filter = fn(string $className): bool => /** @psalm-var class-string $className */ $this->skipClass($className);
        $visitor = new ClassifierVisitor($filter);
        $traverser = new NodeTraverser();
        $traverser->addVisitor(new NameResolver());
        $traverser->addVisitor($visitor);

        $this->visitor = $visitor;
        $this->nodeTraverser = $traverser;
    }

    /**
     * @return iterable<class-string>
     */
    protected function getAvailableClasses(): iterable
    {
        $files = $this->getFiles();

        foreach ($files as $file) {
            $nodes = $this->parser->parse($file->getContents());
            if ($nodes !== null) {
                $this->nodeTraverser->traverse($nodes);
            }
        }

        yield from new \ArrayIterator($this->visitor->getClassNames());
    }
}