UIMacros: own handler for {extends}

Author: dg

src/Bridges/ApplicationLatte/UIMacros.php

@@ -36,6 +36,21 @@ public static function install(Latte\Compiler $compiler)
 		$me->addMacro('plink', [$me, 'macroLink']);
 		$me->addMacro('link', [$me, 'macroLink']);
 		$me->addMacro('ifCurrent', [$me, 'macroIfCurrent'], '}'); // deprecated; use n:class="$presenter->linkCurrent ? ..."
+		$me->addMacro('extends', [$me, 'macroExtends']);
+		$me->addMacro('layout', [$me, 'macroExtends']);
+	}
+
+
+	/**
+	 * Initializes before template parsing.
+	 * @return void
+	 */
+	public function initialize()
+	{
+		$this->getCompiler()->addMethod('getParentName', '
+		return !$this->getReferrerTemplate() && $this->params["_control"] instanceof Nette\Application\UI\Presenter
+			? $this->params["_control"]->findLayoutTemplateFile() : NULL;
+		');
 	}
 
 
@@ -103,6 +118,18 @@ public function macroIfCurrent(MacroNode $node, PhpWriter $writer)
 	}
 
 
+	/**
+	 * {extends auto}
+	 */
+	public function macroExtends(MacroNode $node, PhpWriter $writer)
+	{
+		if ($node->modifiers || $node->parentNode || $node->args !== 'auto') {
+			return FALSE;
+		}
+		$this->getCompiler()->addMethod('getParentName', 'return $this->params["_presenter"]->findLayoutTemplateFile();');
+	}
+
+
 	/** @deprecated */
 	public static function renderSnippets(Nette\Application\UI\Control $control, \stdClass $local, array $params)
 	{

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

@@ -0,0 +1,69 @@
+<?php
+
+use Nette\Bridges\ApplicationLatte\UIMacros;
+use Tester\Assert;
+
+
+require __DIR__ . '/../bootstrap.php';
+
+
+class MockPresenter extends Nette\Application\UI\Presenter
+{
+	function findLayoutTemplateFile()
+	{
+		return 'layout.latte';
+	}
+}
+
+
+$latte = new Latte\Engine;
+$latte->setLoader(new Latte\Loaders\StringLoader);
+UIMacros::install($latte->getCompiler());
+
+$template = $latte->createTemplate(
+	'',
+	['_control' => new MockPresenter]
+);
+Assert::same('layout.latte', $template->getParentName());
+
+$template = $latte->createTemplate(
+	'{block}...{/block}',
+	['_control' => new MockPresenter]
+);
+Assert::same('layout.latte', $template->getParentName());
+
+$template = $latte->createTemplate(
+	'{block name}...{/block}',
+	['_control' => new MockPresenter]
+);
+Assert::same('layout.latte', $template->getParentName());
+
+$template = $latte->createTemplate(
+	'{extends "file.latte"} {block name}...{/block}',
+	['_control' => new MockPresenter]
+);
+Assert::same('file.latte', $template->getParentName());
+
+$template = $latte->createTemplate(
+	'{extends "file.latte"}',
+	['_control' => new MockPresenter]
+);
+Assert::same('file.latte', $template->getParentName());
+
+$template = $latte->createTemplate(
+	'{extends $file} {block name}...{/block}',
+	['_control' => new MockPresenter, 'file' => 'file.latte']
+);
+Assert::same('file.latte', $template->getParentName());
+
+$template = $latte->createTemplate(
+	'{extends none}',
+	['_control' => new MockPresenter]
+);
+Assert::null($template->getParentName());
+
+$template = $latte->createTemplate(
+	'{extends auto}',
+	['_presenter' => new MockPresenter]
+);
+Assert::same('layout.latte', $template->getParentName());

tests/Bridges.Latte/expected/UIMacros.dynamicsnippets.alt.phtml

@@ -25,6 +25,12 @@ if ($this->tryRenderParent(get_defined_vars())) return ?>
 	}
 
 
+	function getParentName()
+	{
+		%A%
+	}
+
+
 	function blockOuter1($_args)
 	{
 		extract($_args);

tests/Bridges.Latte/expected/UIMacros.dynamicsnippets.phtml

@@ -21,6 +21,12 @@ if ($this->tryRenderParent(get_defined_vars())) return ?>
 	}
 
 
+	function getParentName()
+	{
+		%A%
+	}
+
+
 	function blockOuter($_args)
 	{
 		extract($_args);

tests/Bridges.Latte/expected/UIMacros.snippet.alt.phtml

@@ -31,6 +31,12 @@ if (Nette\Bridges\ApplicationLatte\UIRuntime::initialize($this, $this->blockQueu
 	}
 
 
+	function getParentName()
+	{
+		%A%
+	}
+
+
 	function blockOuter($_args)
 	{
 		extract($_args);

tests/Bridges.Latte/expected/UIMacros.snippet.phtml

@@ -40,6 +40,12 @@ if (Nette\Bridges\ApplicationLatte\UIRuntime::initialize($this, $this->blockQueu
 	}
 
 
+	function getParentName()
+	{
+		%A%
+	}
+
+
 	function block_%h%($_args)
 	{
 		extract($_args);