latte: yield sub-nodes

contributte · Sep 12, 2023 · c5b4c65 · c5b4c65
1 parent ef53832
commit c5b4c65
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/src/Bridge/Nette/Latte/Extension/Node/ImgNode.php b/src/Bridge/Nette/Latte/Extension/Node/ImgNode.php
@@ -2,6 +2,7 @@
 
 namespace Contributte\Imagist\Bridge\Nette\Latte\Extension\Node;
 
+use Generator;
 use Latte\Compiler\Nodes\Html\ElementNode;
 use Latte\Compiler\Nodes\Php\Expression\ArrayNode;
 use Latte\Compiler\Nodes\Php\ExpressionNode;
@@ -12,11 +13,11 @@
 final class ImgNode extends StatementNode
 {
 
-	private ?ElementNode $element;
+	public ?ElementNode $element;
 
-	private ExpressionNode $image;
+	public ExpressionNode $image;
 
-	private ArrayNode $arguments;
+	public ArrayNode $arguments;
 
 	public static function create(Tag $tag): self
 	{
@@ -51,4 +52,17 @@ public function print(PrintContext $context): string
 		return $code;
 	}
 
+	/**
+	 * @return Generator<ElementNode|ExpressionNode>
+	 */
+	public function &getIterator(): Generator
+	{
+		if ($this->element) {
+			yield $this->element;
+		}
+
+		yield $this->image;
+		yield $this->arguments;
+	}
+
 }