TSL: Assign stack for nodes after remove stack (#32137)

* assign stack for nodes after remove stack

* remove define node stack from setup

Author: sunag

src/nodes/core/NodeBuilder.js

@@ -1683,6 +1683,14 @@ class NodeBuilder {
 	removeStack() {
 
 		const lastStack = this.stack;
+
+		for ( const node of lastStack.nodes ) {
+
+			const nodeData = this.getDataFromNode( node );
+			nodeData.stack = lastStack;
+
+		}
+
 		this.stack = lastStack.parent;
 
 		setCurrentStack( this.stacks.pop() );

src/nodes/core/StackNode.js

@@ -311,9 +311,6 @@ class StackNode extends Node {
 
 			if ( buildStage === 'setup' ) {
 
-				const nodeData = builder.getDataFromNode( node );
-				nodeData.stack = this;
-
 				node.build( builder );
 
 			} else if ( buildStage === 'analyze' ) {