From 92dbeab27e73d60ec16a98ab7c889d877c73e9ce Mon Sep 17 00:00:00 2001
From: David Grudl <david@grudl.com>
Date: Mon, 9 Jan 2023 14:07:48 +0100
Subject: [PATCH] LatteExtension: added option 'variables'

---
 src/Bridges/ApplicationDI/LatteExtension.php  |  7 +-
 .../ApplicationLatte/DefaultTemplate.php      |  1 +
 .../ApplicationLatte/TemplateFactory.php      |  2 +
 .../Bridges.DI/LatteExtension.variables.phpt  | 74 +++++++++++++++++++
 4 files changed, 82 insertions(+), 2 deletions(-)
 create mode 100644 tests/Bridges.DI/LatteExtension.variables.phpt

diff --git a/src/Bridges/ApplicationDI/LatteExtension.php b/src/Bridges/ApplicationDI/LatteExtension.php
index f26c5d393..82d0c08dd 100644
--- a/src/Bridges/ApplicationDI/LatteExtension.php
+++ b/src/Bridges/ApplicationDI/LatteExtension.php
@@ -38,6 +38,7 @@ public function getConfigSchema(): Nette\Schema\Schema
 			'strictTypes' => Expect::bool(false),
 			'strictParsing' => Expect::bool(false),
 			'phpLinter' => Expect::string(),
+			'variables' => Expect::array([]),
 		]);
 	}
 
@@ -75,8 +76,10 @@ public function loadConfiguration(): void
 		}
 
 		$builder->addDefinition($this->prefix('templateFactory'))
-			->setFactory(ApplicationLatte\TemplateFactory::class)
-			->setArguments(['templateClass' => $config->templateClass]);
+			->setFactory(ApplicationLatte\TemplateFactory::class, [
+				'templateClass' => $config->templateClass,
+				'configVars' => $config->variables,
+			]);
 
 		if ($this->name === 'latte') {
 			$builder->addAlias('nette.latteFactory', $this->prefix('latteFactory'));
diff --git a/src/Bridges/ApplicationLatte/DefaultTemplate.php b/src/Bridges/ApplicationLatte/DefaultTemplate.php
index 29d761663..85eae9c6d 100644
--- a/src/Bridges/ApplicationLatte/DefaultTemplate.php
+++ b/src/Bridges/ApplicationLatte/DefaultTemplate.php
@@ -29,6 +29,7 @@ final class DefaultTemplate extends Template
 
 	/** @var \stdClass[] */
 	public array $flashes = [];
+	public \stdClass $config;
 
 
 	/**
diff --git a/src/Bridges/ApplicationLatte/TemplateFactory.php b/src/Bridges/ApplicationLatte/TemplateFactory.php
index 1f2cbb764..5dc9cf50f 100644
--- a/src/Bridges/ApplicationLatte/TemplateFactory.php
+++ b/src/Bridges/ApplicationLatte/TemplateFactory.php
@@ -28,6 +28,7 @@ public function __construct(
 		private readonly ?Nette\Http\IRequest $httpRequest = null,
 		private readonly ?Nette\Security\User $user = null,
 		$templateClass = null,
+		private array $configVars = [],
 	) {
 		if ($templateClass && (!class_exists($templateClass) || !is_a($templateClass, Template::class, true))) {
 			throw new Nette\InvalidArgumentException("Class $templateClass does not implement " . Template::class . ' or it does not exist.');
@@ -63,6 +64,7 @@ public function createTemplate(?UI\Control $control = null, ?string $class = nul
 			'flashes' => $flashes,
 			'control' => $control,
 			'presenter' => $presenter,
+			'config' => $control instanceof UI\Presenter && $this->configVars ? (object) $this->configVars : null,
 		];
 
 		foreach ($params as $key => $value) {
diff --git a/tests/Bridges.DI/LatteExtension.variables.phpt b/tests/Bridges.DI/LatteExtension.variables.phpt
new file mode 100644
index 000000000..c7051e5d8
--- /dev/null
+++ b/tests/Bridges.DI/LatteExtension.variables.phpt
@@ -0,0 +1,74 @@
+<?php
+
+/**
+ * Test: LatteExtension v3
+ */
+
+declare(strict_types=1);
+
+use Nette\DI;
+use Tester\Assert;
+
+require __DIR__ . '/../bootstrap.php';
+
+Tester\Environment::bypassFinals();
+
+
+test('no variables', function () {
+	$loader = new DI\Config\Loader;
+	$config = $loader->load(Tester\FileMock::create('
+	latte:
+		variables:
+	', 'neon'));
+
+	$compiler = new DI\Compiler;
+	$compiler->addExtension('latte', new Nette\Bridges\ApplicationDI\LatteExtension('', false));
+	$code = $compiler
+		->addConfig($config)
+		->setClassName('Container1')
+		->compile();
+	eval($code);
+
+	$container = new Container1;
+	$latteFactory = $container->getService('latte.templateFactory');
+	$presenter = Mockery::mock(Nette\Application\UI\Presenter::class);
+	$presenter->shouldReceive('getHttpResponse')->andReturn(Mockery::mock(Nette\Http\IResponse::class)->shouldIgnoreMissing());
+	$presenter->shouldIgnoreMissing();
+
+	$template = $latteFactory->createTemplate($presenter);
+	Assert::notContains('config', $template->getParameters());
+});
+
+
+test('presenter presence', function () {
+	$loader = new DI\Config\Loader;
+	$config = $loader->load(Tester\FileMock::create('
+	latte:
+		variables:
+			foo: bar
+	', 'neon'));
+
+	$compiler = new DI\Compiler;
+	$compiler->addExtension('latte', new Nette\Bridges\ApplicationDI\LatteExtension('', false));
+	$code = $compiler
+		->addConfig($config)
+		->setClassName('Container2')
+		->compile();
+	eval($code);
+
+	$container = new Container2;
+	$latteFactory = $container->getService('latte.templateFactory');
+	$template = $latteFactory->createTemplate();
+	Assert::notContains('config', $template->getParameters());
+
+
+	$presenter = Mockery::mock(Nette\Application\UI\Presenter::class);
+	$presenter->shouldReceive('getHttpResponse')->andReturn(Mockery::mock(Nette\Http\IResponse::class)->shouldIgnoreMissing());
+	$presenter->shouldIgnoreMissing();
+
+	$template = $latteFactory->createTemplate($presenter);
+	Assert::equal(
+		(object) ['foo' => 'bar'],
+		$template->getParameters()['config'],
+	);
+});