Spiral\Reactor\Traits\ClassAware

Complexity

Total Complexity

3

Size/Duplication

Total Lines	26
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	getClass()	0	7	1
A	addClass()	0	3	1
A	getClasses()	0	10	1

<?php

declare(strict_types=1);

namespace Spiral\Reactor\Traits;

use Nette\PhpGenerator\ClassLike;
use Nette\PhpGenerator\ClassType;
use Nette\PhpGenerator\Helpers;
use Spiral\Reactor\Aggregator\Classes;
use Spiral\Reactor\ClassDeclaration;

/**
 * @internal
 */
trait ClassAware
{
    public function addClass(string $name): ClassDeclaration
    {
        return ClassDeclaration::fromElement($this->element->addClass($name));
    }

    public function getClass(string $name): ClassDeclaration
    {
        /**
         * @psalm-suppress InternalClass
         * @psalm-suppress InternalMethod
         */
        return $this->getClasses()->get(Helpers::extractShortName($name));
    }

    public function getClasses(): Classes
    {
        $classes = \array_filter(
            $this->element->getClasses(),
            static fn (ClassLike $element): bool => $element instanceof ClassType
        );

        return new Classes(\array_map(
            static fn (ClassType $class): ClassDeclaration => ClassDeclaration::fromElement($class),
            $classes
        ));
    }
}