Repair PhpParser\Node\Stmt\Class_::isAnonymous()

Author: tscni

src/Analyser/NodeScopeResolver.php

@@ -117,7 +117,6 @@
 use PHPStan\Node\UnreachableStatementNode;
 use PHPStan\Node\VariableAssignNode;
 use PHPStan\Node\VarTagChangedExpressionTypeNode;
-use PHPStan\Parser\AnonymousClassVisitor;
 use PHPStan\Parser\ArrowFunctionArgVisitor;
 use PHPStan\Parser\ClosureArgVisitor;
 use PHPStan\Parser\Parser;
@@ -856,7 +855,7 @@ private function processStmtNode(
 				if ($stmt->name === null) {
 					throw new ShouldNotHappenException();
 				}
-				if ($stmt->getAttribute(AnonymousClassVisitor::ATTRIBUTE_ANONYMOUS_CLASS, false) === false) {
+				if (!$stmt->isAnonymous()) {
 					$classReflection = $this->reflectionProvider->getClass($stmt->name->toString());
 				} else {
 					$classReflection = $this->reflectionProvider->getAnonymousClassReflection($stmt, $scope);

src/Node/AnonymousClassNode.php

@@ -0,0 +1,18 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Node;
+
+use PhpParser\Node\Stmt\Class_;
+
+/**
+ * The only purpose of this is to fix {@see Class_::isAnonymous()}, broken by us giving anonymous classes a name.
+ */
+final class AnonymousClassNode extends Class_
+{
+
+	public function isAnonymous(): bool
+	{
+		return true;
+	}
+
+}

src/Parser/AnonymousClassVisitor.php

@@ -4,15 +4,15 @@
 
 use PhpParser\Node;
 use PhpParser\NodeVisitorAbstract;
+use PHPStan\Node\AnonymousClassNode;
 use function count;
 
 class AnonymousClassVisitor extends NodeVisitorAbstract
 {
 
-	public const ATTRIBUTE_ANONYMOUS_CLASS = 'anonymousClass';
 	public const ATTRIBUTE_LINE_INDEX = 'anonymousClassLineIndex';
 
-	/** @var array<int, non-empty-list<Node\Stmt\Class_>> */
+	/** @var array<int, non-empty-list<AnonymousClassNode>> */
 	private array $nodesPerLine = [];
 
 	public function beforeTraverse(array $nodes): ?array
@@ -27,10 +27,20 @@ public function enterNode(Node $node): ?Node
 			return null;
 		}
 
-		$node->setAttribute(self::ATTRIBUTE_ANONYMOUS_CLASS, true);
+		$node = new AnonymousClassNode(
+			$node->name,
+			[
+				'flags' => $node->flags,
+				'extends' => $node->extends,
+				'implements' => $node->implements,
+				'stmts' => $node->stmts,
+				'attrGroups' => $node->attrGroups,
+			],
+			$node->getAttributes(),
+		);
 		$this->nodesPerLine[$node->getStartLine()][] = $node;
 
-		return null;
+		return $node;
 	}
 
 	public function afterTraverse(array $nodes): ?array

src/Type/FileTypeMapper.php

@@ -8,7 +8,6 @@
 use PHPStan\BetterReflection\Util\GetLastDocComment;
 use PHPStan\Broker\AnonymousClassNameHelper;
 use PHPStan\File\FileHelper;
-use PHPStan\Parser\AnonymousClassVisitor;
 use PHPStan\Parser\Parser;
 use PHPStan\PhpDoc\PhpDocNodeResolver;
 use PHPStan\PhpDoc\PhpDocStringResolver;
@@ -261,7 +260,7 @@ function (Node $node) use ($fileName, $lookForTrait, &$traitFound, $traitMethodA
 							}
 
 							$className = $this->anonymousClassNameHelper->getAnonymousClassName($node, $fileName);
-						} elseif ((bool) $node->getAttribute(AnonymousClassVisitor::ATTRIBUTE_ANONYMOUS_CLASS, false)) {
+						} elseif ($node instanceof Node\Stmt\Class_ && $node->isAnonymous()) {
 							$className = $node->name->name;
 						} else {
 							if ($traitFound) {
@@ -452,7 +451,7 @@ function (Node $node) use ($fileName, $lookForTrait, $phpDocNodeMap, &$traitFoun
 							}
 
 							$className = $this->anonymousClassNameHelper->getAnonymousClassName($node, $fileName);
-						} elseif ((bool) $node->getAttribute(AnonymousClassVisitor::ATTRIBUTE_ANONYMOUS_CLASS, false)) {
+						} elseif ($node instanceof Node\Stmt\Class_ && $node->isAnonymous()) {
 							$className = $node->name->name;
 						} else {
 							if ($traitFound) {

tests/PHPStan/Reflection/AnonymousClassReflectionTest.php

@@ -5,7 +5,6 @@
 use PhpParser\Node;
 use PhpParser\Node\Stmt\Class_;
 use PHPStan\Analyser\Scope;
-use PHPStan\Parser\AnonymousClassVisitor;
 use PHPStan\Rules\Rule;
 use PHPStan\Rules\RuleErrorBuilder;
 use PHPStan\Testing\RuleTestCase;
@@ -36,7 +35,7 @@ public function getNodeType(): string
 
 			public function processNode(Node $node, Scope $scope): array
 			{
-				if (!(bool) $node->getAttribute(AnonymousClassVisitor::ATTRIBUTE_ANONYMOUS_CLASS)) {
+				if (!$node->isAnonymous()) {
 					return [];
 				}