improved Latte 3 nodes

contributte · Dec 3, 2022 · 80e15b6 · 80e15b6
1 parent 5bafb56
commit 80e15b6
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 6 deletions.
diff --git a/src/Latte/Extension/Node/MultiplierAddNode.php b/src/Latte/Extension/Node/MultiplierAddNode.php
@@ -19,13 +19,13 @@ final class MultiplierAddNode extends StatementNode
 {
 
 	/** @var ExpressionNode */
-	private $name;
+	public $name;
 
 	/** @var ArrayNode */
-	private $attributes;
+	public $attributes;
 
 	/** @var ExpressionNode */
-	private $part;
+	public $part;
 
 	public static function create(Tag $tag): self
 	{
@@ -74,4 +74,10 @@ public static function getCreateButton(Multiplier $multiplier, $buttonId): ?Subm
 		return $multiplier->getCreateButtons()[$buttonId] ?? null;
 	}
 
+	public function &getIterator(): \Generator
+	{
+		yield $this->name;
+		yield $this->attributes;
+		yield $this->part;
+	}
 }
diff --git a/src/Latte/Extension/Node/MultiplierNode.php b/src/Latte/Extension/Node/MultiplierNode.php
@@ -15,10 +15,10 @@ final class MultiplierNode extends StatementNode
 {
 
 	/** @var ExpressionNode */
-	private $name;
+	public $name;
 
 	/** @var AreaNode */
-	private $content;
+	public $content;
 
 	/**
 	 * @return Generator<int, ?mixed[], array{AreaNode, ?Tag}, self>

diff --git a/src/Latte/Extension/Node/MultiplierRemoveNode.php b/src/Latte/Extension/Node/MultiplierRemoveNode.php
@@ -16,7 +16,7 @@ final class MultiplierRemoveNode extends StatementNode
 {
 
 	/** @var ArrayNode */
-	private $attributes;
+	public $attributes;
 
 	public static function create(Tag $tag): self
 	{
@@ -54,4 +54,9 @@ public static function getRemoveButton(array $formsStack): ?IComponent
 		return $container->getComponent(Multiplier::SUBMIT_REMOVE_NAME, false);
 	}
 
+	public function &getIterator(): \Generator
+	{
+		yield $this->attributes;
+	}
+
 }