allow {exitIf} in {define} (#334)

nette · Aug 7, 2023 · be11600 · be11600
1 parent b9c9433
commit be11600
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 1 deletion.
diff --git a/src/Latte/Essential/Nodes/JumpNode.php b/src/Latte/Essential/Nodes/JumpNode.php
@@ -35,7 +35,7 @@ public static function create(Tag $tag): static
 		$allowed = match ($tag->name) {
 			'breakIf', 'continueIf' => ['for', 'foreach', 'while'],
 			'skipIf' => ['foreach'],
-			'exitIf' => ['block', null],
+			'exitIf' => ['block', 'define', null],
 		};
 		for (
 			$parent = $tag->parent;

diff --git a/tests/tags/exitIf.phpt b/tests/tags/exitIf.phpt
@@ -80,3 +80,32 @@ Assert::match(
 		XX,
 	$latte->compile($template),
 );
+
+
+$template = <<<'XX'
+	{define foo}
+		a
+		{exitIf true}
+		b
+		{exitIf false}
+		c
+	{/define}
+	XX;
+
+Assert::match(
+	<<<'XX'
+		%A%
+			{
+				echo '	a
+		';
+				if (true) /* line 3 */ return;
+				echo '	b
+		';
+				if (false) /* line 5 */ return;
+				echo '	c
+		';
+			}
+		%A%
+		XX,
+	$latte->compile($template),
+);