{exitIf} must not be in prepare()

nette · Nov 11, 2023 · 48fd9c1 · 48fd9c1
1 parent af8c0aa
commit 48fd9c1
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 1 deletion.
diff --git a/src/Latte/Essential/Nodes/JumpNode.php b/src/Latte/Essential/Nodes/JumpNode.php
@@ -31,6 +31,10 @@ class JumpNode extends StatementNode
 	public static function create(Tag $tag): static
 	{
 		$tag->expectArguments();
+		$tag->outputMode = $tag->name === 'exitIf' // to not be in prepare()
+			? $tag::OutputRemoveIndentation
+			: $tag::OutputNone;
+
 		for (
 			$parent = $tag->parent;
 			$parent?->node instanceof IfNode || $parent?->node instanceof IfContentNode;

diff --git a/tests/tags/exitIf.nodes.phpt b/tests/tags/exitIf.nodes.phpt
@@ -13,11 +13,11 @@ require __DIR__ . '/../bootstrap.php';
 
 Assert::match(<<<'XX'
 	Template:
+		Fragment:
 		Fragment:
 			Jump:
 				Variable:
 					name: a
-		Fragment:
 			Text:
 				content: ' '
 			Block:

diff --git a/tests/tags/exitIf.phpt b/tests/tags/exitIf.phpt
@@ -27,6 +27,24 @@ Assert::exception(
 );
 
 
+$template = <<<'XX'
+	{exitIf true}
+	c
+	XX;
+
+Assert::match(
+	<<<'XX'
+		%A%
+			{
+				if (true) /* line 1 */ return;
+				echo 'c';
+			}
+		%A%
+		XX,
+	$latte->compile($template),
+);
+
+
 $template = <<<'XX'
 	a
 	{exitIf true}