From 397e38b7a6e99f689689b5bfa953b63ebf682b7b Mon Sep 17 00:00:00 2001
From: David Grudl <david@grudl.com>
Date: Tue, 27 Dec 2022 01:37:19 +0100
Subject: [PATCH] Latte: NNameNode split to two classes

---
 src/Bridges/FormsLatte/FormsExtension.php     |  4 +-
 .../{NNameNode.php => FieldNNameNode.php}     | 43 ++--------
 .../FormsLatte/Nodes/FormNNameNode.php        | 80 +++++++++++++++++++
 3 files changed, 89 insertions(+), 38 deletions(-)
 rename src/Bridges/FormsLatte/Nodes/{NNameNode.php => FieldNNameNode.php} (69%)
 create mode 100644 src/Bridges/FormsLatte/Nodes/FormNNameNode.php

diff --git a/src/Bridges/FormsLatte/FormsExtension.php b/src/Bridges/FormsLatte/FormsExtension.php
index e481544cc..3e13a0b88 100644
--- a/src/Bridges/FormsLatte/FormsExtension.php
+++ b/src/Bridges/FormsLatte/FormsExtension.php
@@ -28,7 +28,9 @@ public function getTags(): array
 			'inputError' => [Nodes\InputErrorNode::class, 'create'],
 			'formPrint' => [Nodes\FormPrintNode::class, 'create'],
 			'formClassPrint' => [Nodes\FormPrintNode::class, 'create'],
-			'n:name' => [Nodes\NNameNode::class, 'create'],
+			'n:name' => fn(Latte\Compiler\Tag $tag) => yield from strtolower($tag->htmlElement->name) === 'form'
+				? Nodes\FormNNameNode::create($tag)
+				: Nodes\FieldNNameNode::create($tag),
 		];
 	}
 
diff --git a/src/Bridges/FormsLatte/Nodes/NNameNode.php b/src/Bridges/FormsLatte/Nodes/FieldNNameNode.php
similarity index 69%
rename from src/Bridges/FormsLatte/Nodes/NNameNode.php
rename to src/Bridges/FormsLatte/Nodes/FieldNNameNode.php
index 5a6867bae..78ce293a6 100644
--- a/src/Bridges/FormsLatte/Nodes/NNameNode.php
+++ b/src/Bridges/FormsLatte/Nodes/FieldNNameNode.php
@@ -9,7 +9,6 @@
 
 namespace Nette\Bridges\FormsLatte\Nodes;
 
-use Latte;
 use Latte\Compiler\Nodes\AreaNode;
 use Latte\Compiler\Nodes\AuxiliaryNode;
 use Latte\Compiler\Nodes\Html\AttributeNode;
@@ -24,9 +23,9 @@
 
 
 /**
- * <form n:name>, <input n:name>, <select n:name>, <textarea n:name>, <label n:name> and <button n:name>
+ * <input n:name>, <select n:name>, <textarea n:name>, <label n:name> and <button n:name>
  */
-final class NNameNode extends StatementNode
+final class FieldNNameNode extends StatementNode
 {
 	public ExpressionNode $name;
 	public ?ExpressionNode $part = null;
@@ -46,11 +45,7 @@ public static function create(Tag $tag): \Generator
 
 		[$node->content] = yield;
 
-		if (strtolower($tag->htmlElement->name) === 'form') {
-			$node->initForm($tag);
-		} else {
-			$node->initElement($tag);
-		}
+		$node->init($tag);
 
 		return $node;
 	}
@@ -62,34 +57,7 @@ public function print(PrintContext $context): string
 	}
 
 
-	private function initForm(Tag $tag)
-	{
-		$el = $tag->htmlElement;
-
-		$tag->replaceNAttribute(new AuxiliaryNode(fn(PrintContext $context) => $context->format(
-			'$form = $this->global->formsStack[] = '
-			. ($this->name instanceof StringNode
-				? '$this->global->uiControl[%node]'
-				: 'is_object($ʟ_tmp = %node) ? $ʟ_tmp : $this->global->uiControl[$ʟ_tmp]')
-			. ' %line;'
-			. 'echo Nette\Bridges\FormsLatte\Runtime::renderFormBegin(end($this->global->formsStack), %dump, false) %line;',
-			$this->name,
-			$this->position,
-			array_fill_keys(self::findUsedAttributes($el), null),
-			$this->position,
-		)));
-
-		$el->content = new Latte\Compiler\Nodes\FragmentNode([
-			$el->content,
-			new AuxiliaryNode(fn(PrintContext $context) => $context->format(
-				'echo Nette\Bridges\FormsLatte\Runtime::renderFormEnd(array_pop($this->global->formsStack), false) %line;',
-				$this->position,
-			)),
-		]);
-	}
-
-
-	private function initElement(Tag $tag)
+	private function init(Tag $tag)
 	{
 		$el = $tag->htmlElement;
 		$usedAttributes = self::findUsedAttributes($el);
@@ -129,7 +97,8 @@ private function initElement(Tag $tag)
 	}
 
 
-	private static function findUsedAttributes(ElementNode $el): array
+	/** @internal */
+	public static function findUsedAttributes(ElementNode $el): array
 	{
 		$res = [];
 		foreach ($el->attributes?->children as $child) {
diff --git a/src/Bridges/FormsLatte/Nodes/FormNNameNode.php b/src/Bridges/FormsLatte/Nodes/FormNNameNode.php
new file mode 100644
index 000000000..e6869467a
--- /dev/null
+++ b/src/Bridges/FormsLatte/Nodes/FormNNameNode.php
@@ -0,0 +1,80 @@
+<?php
+
+/**
+ * This file is part of the Latte (https://latte.nette.org)
+ * Copyright (c) 2008 David Grudl (https://davidgrudl.com)
+ */
+
+declare(strict_types=1);
+
+namespace Nette\Bridges\FormsLatte\Nodes;
+
+use Latte;
+use Latte\Compiler\Nodes\AreaNode;
+use Latte\Compiler\Nodes\AuxiliaryNode;
+use Latte\Compiler\Nodes\Php\ExpressionNode;
+use Latte\Compiler\Nodes\Php\Scalar\StringNode;
+use Latte\Compiler\Nodes\StatementNode;
+use Latte\Compiler\PrintContext;
+use Latte\Compiler\Tag;
+
+
+/**
+ * <form n:name>
+ */
+final class FormNNameNode extends StatementNode
+{
+	public ExpressionNode $name;
+	public AreaNode $content;
+
+
+	public static function create(Tag $tag): \Generator
+	{
+		$tag->expectArguments();
+		$node = new static;
+		$node->name = $tag->parser->parseUnquotedStringOrExpression(colon: false);
+		[$node->content] = yield;
+		$node->init($tag);
+		return $node;
+	}
+
+
+	public function print(PrintContext $context): string
+	{
+		return $this->content->print($context);
+	}
+
+
+	private function init(Tag $tag)
+	{
+		$el = $tag->htmlElement;
+
+		$tag->replaceNAttribute(new AuxiliaryNode(fn(PrintContext $context) => $context->format(
+			'$form = $this->global->formsStack[] = '
+			. ($this->name instanceof StringNode
+				? '$this->global->uiControl[%node]'
+				: 'is_object($ʟ_tmp = %node) ? $ʟ_tmp : $this->global->uiControl[$ʟ_tmp]')
+			. ' %line;'
+			. 'echo Nette\Bridges\FormsLatte\Runtime::renderFormBegin(end($this->global->formsStack), %dump, false) %line;',
+			$this->name,
+			$this->position,
+			array_fill_keys(FieldNNameNode::findUsedAttributes($el), null),
+			$this->position,
+		)));
+
+		$el->content = new Latte\Compiler\Nodes\FragmentNode([
+			$el->content,
+			new AuxiliaryNode(fn(PrintContext $context) => $context->format(
+				'echo Nette\Bridges\FormsLatte\Runtime::renderFormEnd(array_pop($this->global->formsStack), false) %line;',
+				$this->position,
+			)),
+		]);
+	}
+
+
+	public function &getIterator(): \Generator
+	{
+		yield $this->name;
+		yield $this->content;
+	}
+}