Latte: added {linkBase}

Author: dg

src/Application/LinkGenerator.php

@@ -295,4 +295,13 @@ private function isDeprecated(?UI\Presenter $presenter, \ReflectionClass|\Reflec
 		return $presenter?->invalidLinkMode
 			&& (UI\ComponentReflection::parseAnnotation($reflection, 'deprecated') || $reflection->getAttributes(Attributes\Deprecated::class));
 	}
+
+
+	/** @internal */
+	public static function applyBase(string $link, string $base): string
+	{
+		return str_contains($link, ':') && $link[0] !== ':'
+			? ":$base:$link"
+			: $link;
+	}
 }

src/Bridges/ApplicationLatte/Nodes/LinkBaseNode.php

@@ -0,0 +1,83 @@
+<?php
+
+/**
+ * This file is part of the Nette Framework (https://nette.org)
+ * Copyright (c) 2004 David Grudl (https://davidgrudl.com)
+ */
+
+declare(strict_types=1);
+
+namespace Nette\Bridges\ApplicationLatte\Nodes;
+
+use Latte\CompileException;
+use Latte\Compiler\Node;
+use Latte\Compiler\NodeHelpers;
+use Latte\Compiler\Nodes\Php\Expression\AuxiliaryNode;
+use Latte\Compiler\Nodes\Php\ExpressionNode;
+use Latte\Compiler\Nodes\Php\Scalar\StringNode;
+use Latte\Compiler\Nodes\StatementNode;
+use Latte\Compiler\Nodes\TemplateNode;
+use Latte\Compiler\NodeTraverser;
+use Latte\Compiler\PrintContext;
+use Latte\Compiler\Tag;
+use Nette\Application\LinkGenerator;
+
+
+/**
+ * {linkBase module}
+ */
+class LinkBaseNode extends StatementNode
+{
+	public ExpressionNode $base;
+
+
+	public static function create(Tag $tag): static
+	{
+		$tag->expectArguments();
+		if (!$tag->isInHead()) {
+			throw new CompileException("{{$tag->name}} must be placed in template head.", $tag->position);
+		}
+
+		$node = new static;
+		$node->base = $tag->parser->parseUnquotedStringOrExpression();
+		return $node;
+	}
+
+
+	public function print(PrintContext $context): string
+	{
+		return '';
+	}
+
+
+	public function &getIterator(): \Generator
+	{
+		yield $this->base;
+	}
+
+
+	public static function applyLinkBasePass(TemplateNode $node): void
+	{
+		$base = NodeHelpers::findFirst($node, fn(Node $node) => $node instanceof self)?->base;
+		if ($base === null) {
+			return;
+		}
+
+		(new NodeTraverser)->traverse($node, function (Node $link) use ($base) {
+			if ($link instanceof LinkNode) {
+				if ($link->destination instanceof StringNode && $base instanceof StringNode) {
+					$link->destination->value = LinkGenerator::applyBase($link->destination->value, $base->value);
+				} else {
+					$origDestination = $link->destination;
+					$link->destination = new AuxiliaryNode(
+						fn(PrintContext $context) => $context->format(
+							LinkGenerator::class . '::applyBase(%node, %node)',
+							$origDestination,
+							$base,
+						),
+					);
+				}
+			}
+		});
+	}
+}

src/Bridges/ApplicationLatte/UIExtension.php

@@ -79,6 +79,7 @@ public function getTags(): array
 			'control' => Nodes\ControlNode::create(...),
 			'plink' => Nodes\LinkNode::create(...),
 			'link' => Nodes\LinkNode::create(...),
+			'linkBase' => Nodes\LinkBaseNode::create(...),
 			'ifCurrent' => Nodes\IfCurrentNode::create(...),
 			'templatePrint' => Nodes\TemplatePrintNode::create(...),
 			'snippet' => Nodes\SnippetNode::create(...),
@@ -93,6 +94,7 @@ public function getPasses(): array
 	{
 		return [
 			'snippetRendering' => $this->snippetRenderingPass(...),
+			'applyLinkBase' => [Nodes\LinkBaseNode::class, 'applyLinkBasePass'],
 		];
 	}
 

tests/Bridges.Latte3/{linkBase}.phpt

@@ -0,0 +1,45 @@
+<?php
+
+/**
+ * Test: {linkBase ...}
+ */
+
+declare(strict_types=1);
+
+use Tester\Assert;
+
+require __DIR__ . '/../bootstrap.php';
+
+if (version_compare(Latte\Engine::VERSION, '3', '<')) {
+	Tester\Environment::skip('Test for Latte 3');
+}
+
+
+$latte = new Latte\Engine;
+$latte->setLoader(new Latte\Loaders\StringLoader);
+$latte->addExtension(new Nette\Bridges\ApplicationLatte\UIExtension(null));
+
+
+Assert::contains(
+	'$this->global->uiControl->link(\'foo\')',
+	$latte->compile('{linkBase Base}{link foo}'),
+);
+Assert::contains(
+	'$this->global->uiControl->link(\':Base:Foo:\')',
+	$latte->compile('{linkBase Base}{link Foo:}'),
+);
+Assert::contains(
+	'$this->global->uiControl->link(\':Foo:\')',
+	$latte->compile('{linkBase Base}{link :Foo:}'),
+);
+
+
+// dynamic
+Assert::contains(
+	'$this->global->uiControl->link(Nette\Application\LinkGenerator::applyBase($link, \'Base\'))',
+	$latte->compile('{linkBase Base}{link $link}'),
+);
+Assert::contains(
+	'$this->global->uiControl->link(Nette\Application\LinkGenerator::applyBase(\'foo\', $base))',
+	$latte->compile('{linkBase $base}{link foo}'),
+);