Introduce InTraitNode

Author: ondrejmirtes

src/Analyser/NodeScopeResolver.php

@@ -89,6 +89,7 @@
 use PHPStan\Node\InForeachNode;
 use PHPStan\Node\InFunctionNode;
 use PHPStan\Node\InstantiationCallableNode;
+use PHPStan\Node\InTraitNode;
 use PHPStan\Node\LiteralArrayItem;
 use PHPStan\Node\LiteralArrayNode;
 use PHPStan\Node\MatchExpressionArm;
@@ -4362,7 +4363,10 @@ private function processNodesForTraitUse($node, ClassReflection $traitReflection
 					$methodAst->setAttribute('originalTraitMethodName', $methodAst->name->toLowerString());
 					$methodAst->name = $methodNames[$methodName];
 				}
-				$this->processStmtNodes($node, $stmts, $scope->enterTrait($traitReflection), $nodeCallback, StatementContext::createTopLevel());
+
+				$traitScope = $scope->enterTrait($traitReflection);
+				$nodeCallback(new InTraitNode($node, $traitReflection), $traitScope);
+				$this->processStmtNodes($node, $stmts, $traitScope, $nodeCallback, StatementContext::createTopLevel());
 				return;
 			}
 			if ($node instanceof Node\Stmt\ClassLike) {

src/Node/InTraitNode.php

@@ -0,0 +1,40 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\Node;
+
+use PhpParser\Node;
+use PHPStan\Reflection\ClassReflection;
+
+/** @api */
+class InTraitNode extends Node\Stmt implements VirtualNode
+{
+
+	public function __construct(private Node\Stmt\Trait_ $originalNode, private ClassReflection $traitReflection)
+	{
+		parent::__construct($originalNode->getAttributes());
+	}
+
+	public function getOriginalNode(): Node\Stmt\Trait_
+	{
+		return $this->originalNode;
+	}
+
+	public function getTraitReflection(): ClassReflection
+	{
+		return $this->traitReflection;
+	}
+
+	public function getType(): string
+	{
+		return 'PHPStan_Stmt_InTraitNode';
+	}
+
+	/**
+	 * @return string[]
+	 */
+	public function getSubNodeNames(): array
+	{
+		return [];
+	}
+
+}