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('getExtends', '
+		return !$this->getParent() && $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('getExtends', 'return $this->params["_presenter"]->findLayoutTemplateFile();');
+	}
+
+
 	/** @deprecated */
 	public static function renderSnippets(Nette\Application\UI\Control $control, \stdClass $local, array $params)
 	{

tests/Bridges.Latte/Template.getExtends.phpt

@@ -0,0 +1,62 @@
+<?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('');
+$template->params['_control'] = new MockPresenter;
+$template->initialize(NULL);
+Assert::same('layout.latte', $template->getExtends());
+
+$template = $latte->createTemplate('{block}...{/block}');
+$template->params['_control'] = new MockPresenter;
+$template->initialize(NULL);
+Assert::same('layout.latte', $template->getExtends());
+
+$template = $latte->createTemplate('{block name}...{/block}');
+$template->params['_control'] = new MockPresenter;
+$template->initialize(NULL);
+Assert::same('layout.latte', $template->getExtends());
+
+$template = $latte->createTemplate('{extends "file.latte"} {block name}...{/block}');
+$template->params['_control'] = new MockPresenter;
+$template->initialize(NULL);
+Assert::same('file.latte', $template->getExtends());
+
+$template = $latte->createTemplate('{extends "file.latte"}');
+$template->params['_control'] = new MockPresenter;
+$template->initialize(NULL);
+Assert::same('file.latte', $template->getExtends());
+
+$template = $latte->createTemplate('{extends $file} {block name}...{/block}');
+$template->params['_control'] = new MockPresenter;
+$template->params['file'] = 'file.latte';
+$template->initialize(NULL);
+Assert::same('file.latte', $template->getExtends());
+
+$template = $latte->createTemplate('{extends none}');
+$template->params['_control'] = new MockPresenter;
+$template->initialize(NULL);
+Assert::null($template->getExtends());
+
+$template = $latte->createTemplate('{extends auto}');
+$template->params['_presenter'] = new MockPresenter;
+$template->initialize(NULL);
+Assert::same('layout.latte', $template->getExtends());

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

@@ -25,6 +25,12 @@ if (Latte\Macros\BlockMacrosRuntime::progress($this, get_defined_vars())) return
 	}
 
 
+	function getExtends()
+	{
+		return %A%
+	}
+
+
 	function blockOuter1($_b, $_args)
 	{
 		unset($_args["this"]); foreach ($_args as $__k => $__v) $$__k = $__v;
@@ -54,9 +60,4 @@ $_control->redrawControl('outer2', FALSE);
 	<?php if (isset($_l->dynSnippets)) return $_l->dynSnippets;
 	}
 
-
-	function getExtends()
-	{
-	}
-
 }

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

@@ -21,6 +21,12 @@ if (Latte\Macros\BlockMacrosRuntime::progress($this, get_defined_vars())) return
 	}
 
 
+	function getExtends()
+	{
+		return %A%
+	}
+
+
 	function blockOuter($_b, $_args)
 	{
 		unset($_args["this"]); foreach ($_args as $__k => $__v) $$__k = $__v;
@@ -38,9 +44,4 @@ $_control->redrawControl('outer', FALSE);
 	<?php if (isset($_l->dynSnippets)) return $_l->dynSnippets;
 	}
 
-
-	function getExtends()
-	{
-	}
-
 }

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

@@ -31,6 +31,12 @@ if (Nette\Bridges\ApplicationLatte\UIRuntime::initialize($this, $_b)) return;
 	}
 
 
+	function getExtends()
+	{
+		return %A%
+	}
+
+
 	function blockOuter($_b, $_args)
 	{
 		unset($_args["this"]); foreach ($_args as $__k => $__v) $$__k = $__v;
@@ -57,9 +63,4 @@ $_control->redrawControl('gallery', FALSE);
 
 	}
 
-
-	function getExtends()
-	{
-	}
-
 }

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

@@ -40,7 +40,13 @@ if (Nette\Bridges\ApplicationLatte\UIRuntime::initialize($this, $_b)) return;
 	}
 
 
-	function block_%h%($_b, $_args)
+	function getExtends()
+	{
+		return %A%
+	}
+
+
+	function block_b14a7($_b, $_args)
 	{
 		unset($_args["this"]); foreach ($_args as $__k => $__v) $$__k = $__v;
 $_control->redrawControl('', FALSE);
@@ -83,9 +89,4 @@ $_control->redrawControl('title2', FALSE)
 ?>Title 2<?php
 	}
 
-
-	function getExtends()
-	{
-	}
-
 }