TemplateFactory: added onCreate event [closes #195] (#196)

Author: Milan Felix Šulc

src/Bridges/ApplicationLatte/TemplateFactory.php

@@ -21,6 +21,9 @@ class TemplateFactory implements UI\ITemplateFactory
 {
 	use Nette\SmartObject;
 
+	/** @var callable[]  function (Template $template); Occurs when a new template is created */
+	public $onCreate;
+
 	/** @var ILatteFactory */
 	private $latteFactory;
 
@@ -126,6 +129,8 @@ public function createTemplate(UI\Control $control = null): UI\ITemplate
 			$template->flashes = (array) $presenter->getFlashSession()->$id;
 		}
 
+		$this->onCreate($template);
+
 		return $template;
 	}
 }

tests/Bridges.Latte/TemplateFactory.onCreate.phpt

@@ -0,0 +1,33 @@
+<?php
+
+/**
+ * Test: TemplateFactory in Bridge properly handles TemplateFactory::onCreate
+ */
+
+declare(strict_types=1);
+
+use Nette\Bridges\ApplicationLatte\ILatteFactory;
+use Nette\Bridges\ApplicationLatte\Template;
+use Nette\Bridges\ApplicationLatte\TemplateFactory;
+use Nette\Http;
+use Tester\Assert;
+
+
+require __DIR__ . '/../bootstrap.php';
+
+
+test(function () {
+	$engine = new Latte\Engine;
+	$latteFactory = Mockery::mock(ILatteFactory::class);
+	$latteFactory->shouldReceive('create')->andReturn($engine);
+	$factory = new TemplateFactory($latteFactory, new Http\Request(new Http\UrlScript('http://nette.org')));
+	$factory->onCreate[] = $callback = function (Template $template) {
+		$template->add('foo', 'bar');
+	};
+
+	$template = $factory->createTemplate();
+
+	Assert::type('array', $factory->onCreate);
+	Assert::same($callback, $factory->onCreate[0]); // our callback
+	Assert::equal('bar', $template->foo);
+});