Latte: {layout auto} always call Presenter::findLayoutTemplateFile()

Author: dg

src/Bridges/ApplicationLatte/UIExtension.php

@@ -10,7 +10,10 @@
 namespace Nette\Bridges\ApplicationLatte;
 
 use Latte;
+use Latte\Compiler\Nodes\Php\Expression\AuxiliaryNode;
 use Latte\Compiler\Nodes\TemplateNode;
+use Latte\Compiler\Tag;
+use Latte\Essential\Nodes\ExtendsNode;
 use Nette;
 use Nette\Application\UI;
 
@@ -64,6 +67,8 @@ public function getTags(): array
 			'templatePrint' => [Nodes\TemplatePrintNode::class, 'create'],
 			'snippet' => [Nodes\SnippetNode::class, 'create'],
 			'snippetArea' => [Nodes\SnippetAreaNode::class, 'create'],
+			'layout' => [$this, 'createExtendsNode'],
+			'extends' => [$this, 'createExtendsNode'],
 		];
 	}
 
@@ -106,4 +111,15 @@ private function findNonce(Nette\Http\IResponse $httpResponse): ?string
 			?: $httpResponse->getHeader('Content-Security-Policy-Report-Only');
 		return preg_match('#\s\'nonce-([\w+/]+=*)\'#', (string) $header, $m) ? $m[1] : null;
 	}
+
+
+	public static function createExtendsNode(Tag $tag): ExtendsNode
+	{
+		$auto = $tag->parser->stream->is('auto');
+		$node = ExtendsNode::create($tag);
+		if ($auto) {
+			$node->extends = new AuxiliaryNode(fn() => '$this->global->uiPresenter->findLayoutTemplateFile()');
+		}
+		return $node;
+	}
 }

tests/Bridges.Latte3/Template.getParentName().phpt

@@ -73,3 +73,11 @@ Assert::exception(
 	LogicException::class, // missing template
 );
 Assert::same('layout.latte', $template->getParentName());
+
+
+$latte->setLoader(new Latte\Loaders\StringLoader([
+	'main.latte' => '{extends foo.latte}',
+	'foo.latte' => '{extends auto}',
+	'layout.latte' => 'layout',
+]));
+Assert::same('layout', $template = $latte->renderToString('main.latte'));