ControllerAnalyzer.php

<?php

declare(strict_types=1);

namespace Rector\Symfony\TypeAnalyzer;

use PhpParser\Node;
use PhpParser\Node\Stmt\Class_;
use PHPStan\Analyser\Scope;
use PHPStan\Reflection\ClassReflection;
use PHPStan\Type\ObjectType;
use PHPStan\Type\ThisType;
use PHPStan\Type\TypeWithClassName;
use Rector\NodeTypeResolver\Node\AttributeKey;
use Rector\Php80\NodeAnalyzer\PhpAttributeAnalyzer;
use Rector\Reflection\ReflectionResolver;
use Rector\Symfony\Enum\SymfonyAttribute;
use Rector\Symfony\Enum\SymfonyClass;

final class ControllerAnalyzer
{
    public function __construct(
        private ReflectionResolver $reflectionResolver,
    ) {
    }

    public function isController(Expr|Class_ $node): bool
    {
        if ($node instanceof Class_) {
            return $this->isControllerClass($node);
        }

        $scope = $node->getAttribute(AttributeKey::SCOPE);

        // might be missing in a trait
        if (! $scope instanceof Scope) {
            return false;
        }

        $nodeType = $scope->getType($node);
        if (! $nodeType instanceof TypeWithClassName) {
            return false;
        }

        if ($nodeType instanceof ThisType) {
            $nodeType = $nodeType->getStaticObjectType();
        }

        if (! $nodeType instanceof ObjectType) {
            return false;
        }

        $classReflection = $nodeType->getClassReflection();
        if (! $classReflection instanceof ClassReflection) {
            return false;
        }

        return $this->isControllerClassReflection($classReflection);
    }

    public function isInsideController(Node $node): bool
    {
        $classReflection = $this->reflectionResolver->resolveClassReflection($node);
        if (! $classReflection instanceof ClassReflection) {
            return false;
        }

        return $this->isControllerClassReflection($classReflection);
    }

    private function isControllerClassReflection(ClassReflection $classReflection): bool
    {
        if ($classReflection->is(SymfonyClass::CONTROLLER)) {
            return true;
        }

        foreach ($classReflection->getAttributes() as $attribute) {
            if ($attribute->getName() === SymfonyAttribute::AS_CONTROLLER) {
                return true;
            }
        }

        return $classReflection->is(SymfonyClass::ABSTRACT_CONTROLLER);
    }

    private function isControllerClass(Class_ $class): bool
    {
        $classReflection = $this->reflectionResolver->resolveClassReflection($class);
        if (! $classReflection instanceof ClassReflection) {
            return false;
        }
        return $this->isControllerClassReflection($classReflection);
    }
}