Add php-parser Node to reflection data and cleanup in the end

src/Reflection/Internal/CompleteReflection/RemoveNode.php

@@ -0,0 +1,47 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Typhoon\Reflection\Internal\CompleteReflection;
+
+use Typhoon\DeclarationId\AnonymousClassId;
+use Typhoon\DeclarationId\AnonymousFunctionId;
+use Typhoon\DeclarationId\ConstantId;
+use Typhoon\DeclarationId\NamedClassId;
+use Typhoon\DeclarationId\NamedFunctionId;
+use Typhoon\Reflection\Internal\Data;
+use Typhoon\Reflection\Internal\Hook\ClassHook;
+use Typhoon\Reflection\Internal\Hook\ConstantHook;
+use Typhoon\Reflection\Internal\Hook\FunctionHook;
+use Typhoon\Reflection\Internal\Hook\HookPriorities;
+use Typhoon\Reflection\TyphoonReflector;
+use Typhoon\TypedMap\TypedMap;
+
+/**
+ * @internal
+ * @psalm-internal Typhoon\Reflection
+ */
+enum RemoveNode implements ConstantHook, FunctionHook, ClassHook
+{
+    case Instance;
+
+    public function priority(): int
+    {
+        return HookPriorities::CLEAN_UP;
+    }
+
+    public function processConstant(ConstantId $id, TypedMap $data, TyphoonReflector $reflector): TypedMap
+    {
+        return $data->without(Data::Node);
+    }
+
+    public function processFunction(NamedFunctionId|AnonymousFunctionId $id, TypedMap $data, TyphoonReflector $reflector): TypedMap
+    {
+        return $data->without(Data::Node);
+    }
+
+    public function processClass(NamedClassId|AnonymousClassId $id, TypedMap $data, TyphoonReflector $reflector): TypedMap
+    {
+        return $data->without(Data::Node);
+    }
+}

src/Reflection/Internal/Data.php

@@ -66,4 +66,5 @@ enum Data
     public const Deprecation = Data\DeprecationKey::Key;
     public const Optional = Data\BoolKeys::Optional;
     public const Cloneable = Data\BoolKeys::Cloneable;
+    public const Node = Data\NodeKey::Key;
 }

src/Reflection/Internal/Data/NodeKey.php

@@ -0,0 +1,27 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Typhoon\Reflection\Internal\Data;
+
+use PhpParser\Node\Expr\FuncCall;
+use PhpParser\Node\Stmt\ClassLike;
+use PhpParser\Node\Stmt\Const_;
+use PhpParser\Node\Stmt\Function_;
+use Typhoon\TypedMap\OptionalKey;
+use Typhoon\TypedMap\TypedMap;
+
+/**
+ * @internal
+ * @psalm-internal Typhoon\Reflection\Internal
+ * @implements OptionalKey<null|ClassLike|Function_|FuncCall|Const_>
+ */
+enum NodeKey implements OptionalKey
+{
+    case Key;
+
+    public function default(TypedMap $map): mixed
+    {
+        return null;
+    }
+}

src/Reflection/Internal/PhpParser/NodeReflector.php

@@ -67,6 +67,7 @@ public function reflectConstant(Const_ $node, int|string $key, Context $context)
         $valueTypeReflector = new ConstantExpressionTypeReflector($context);
 
         return (new TypedMap())
+            ->with(Data::Node, $node)
             ->with(Data::Context, $context)
             ->with(Data::PhpDoc, $node->getDocComment())
             ->with(Data::Location, $this->reflectLocation($context, $node))
@@ -84,6 +85,7 @@ public function reflectDefine(FuncCall $node, Context $context): TypedMap
         $valueTypeReflector = new ConstantExpressionTypeReflector($context);
 
         return (new TypedMap())
+            ->with(Data::Node, $node)
             ->with(Data::Location, $this->reflectLocation($context, $node))
             ->with(Data::Namespace, $context->namespace())
             ->with(Data::ValueExpression, $compiler->compile($valueArg->value))
@@ -94,12 +96,14 @@ public function reflectFunction(Function_ $node, Context $context): TypedMap
     {
         return $this
             ->reflectFunctionLike($node, $context)
+            ->with(Data::Node, $node)
             ->with(Data::Namespace, $context->namespace());
     }
 
     public function reflectClassLike(ClassLike $node, Context $context): TypedMap
     {
         $data = (new TypedMap())
+            ->with(Data::Node, $node)
             ->with(Data::PhpDoc, $node->getDocComment())
             ->with(Data::Location, $this->reflectLocation($context, $node))
             ->with(Data::Context, $context)

src/Reflection/TyphoonReflector.php

@@ -30,6 +30,7 @@
 use Typhoon\Reflection\Internal\CompleteReflection\CopyPromotedParameterToProperty;
 use Typhoon\Reflection\Internal\CompleteReflection\RemoveCode;
 use Typhoon\Reflection\Internal\CompleteReflection\RemoveContext;
+use Typhoon\Reflection\Internal\CompleteReflection\RemoveNode;
 use Typhoon\Reflection\Internal\CompleteReflection\SetAttributeRepeated;
 use Typhoon\Reflection\Internal\CompleteReflection\SetClassCloneable;
 use Typhoon\Reflection\Internal\CompleteReflection\SetInterfaceMethodAbstract;
@@ -102,6 +103,7 @@ public static function build(
                 SetTemplateIndex::Instance,
                 ResolveClassInheritance::Instance,
                 RemoveContext::Instance,
+                RemoveNode::Instance,
                 RemoveCode::Instance,
                 CleanUpInternallyDefined::Instance,
             ]),