StatementOrderVisitor

ondrejmirtes · ondrejmirtes · commit 4b33bd987779 · 2020-08-18T17:03:37.000+02:00
diff --git a/conf/config.neon b/conf/config.neon
@@ -275,6 +275,7 @@ services:
 		setup:
 			- addVisitor(@PhpParser\NodeVisitor\NameResolver)
 			- addVisitor(@PhpParser\NodeVisitor\NodeConnectingVisitor)
+			- addVisitor(@PHPStan\NodeVisitor\StatementOrderVisitor)
 
 	-
 		class: PhpParser\NodeVisitor\NameResolver
@@ -471,6 +472,9 @@ services:
 		arguments:
 			fileExtensions: %fileExtensions%
 
+	-
+		class: PHPStan\NodeVisitor\StatementOrderVisitor
+
 	-
 		class: PHPStan\Parallel\ParallelAnalyser
 		arguments:
diff --git a/src/NodeVisitor/StatementOrderVisitor.php b/src/NodeVisitor/StatementOrderVisitor.php
@@ -0,0 +1,65 @@
+<?php declare(strict_types = 1);
+
+namespace PHPStan\NodeVisitor;
+
+use PhpParser\Node;
+use PhpParser\NodeVisitorAbstract;
+
+class StatementOrderVisitor extends NodeVisitorAbstract
+{
+
+	/** @var int[] */
+	private array $orderStack = [];
+
+	private int $depth = 0;
+
+	/**
+	 * @param Node[] $nodes $nodes
+	 * @return null
+	 */
+	public function beforeTraverse(array $nodes)
+	{
+		$this->orderStack = [0];
+		$this->depth = 0;
+
+		return null;
+	}
+
+	/**
+	 * @param Node $node
+	 * @return null
+	 */
+	public function enterNode(Node $node)
+	{
+		if (!$node instanceof Node\Stmt) {
+			return null;
+		}
+
+		$order = $this->orderStack[count($this->orderStack) - 1];
+		$node->setAttribute('statementOrder', $order);
+		$node->setAttribute('statementDepth', $this->depth);
+
+		$this->orderStack[count($this->orderStack) - 1] = $order + 1;
+		$this->orderStack[] = 0;
+		$this->depth++;
+
+		return null;
+	}
+
+	/**
+	 * @param Node $node
+	 * @return null
+	 */
+	public function leaveNode(Node $node)
+	{
+		if (!$node instanceof Node\Stmt) {
+			return null;
+		}
+
+		array_pop($this->orderStack);
+		$this->depth--;
+
+		return null;
+	}
+
+}
