Control, Presenter::createTemplate() tries to create custom Template according to naming convention

Author: dg

src/Application/UI/Control.php

@@ -54,7 +54,16 @@ final public function getTemplate(): ITemplate
 	protected function createTemplate(): ITemplate
 	{
 		$templateFactory = $this->templateFactory ?: $this->getPresenter()->getTemplateFactory();
-		return $templateFactory->createTemplate($this);
+		return $templateFactory->createTemplate($this, self::formatTemplateClass());
+	}
+
+
+	public static function formatTemplateClass(): ?string
+	{
+		$class = preg_replace('#Presenter$|Control$#', 'Template', static::class);
+		return is_a($class, Nette\Bridges\ApplicationLatte\LatteTemplate::class, true)
+			? $class
+			: null;
 	}
 
 

src/Application/UI/Presenter.php

@@ -561,7 +561,7 @@ public static function formatRenderMethod(string $view): string
 
 	protected function createTemplate(): ITemplate
 	{
-		return $this->getTemplateFactory()->createTemplate($this);
+		return $this->getTemplateFactory()->createTemplate($this, self::formatTemplateClass());
 	}
 
 

src/Bridges/ApplicationLatte/TemplateFactory.php

@@ -54,10 +54,15 @@ public function __construct(ILatteFactory $latteFactory, Nette\Http\IRequest $ht
 	}
 
 
-	public function createTemplate(UI\Control $control = null): UI\ITemplate
+	public function createTemplate(UI\Control $control = null, string $class = null): UI\ITemplate
 	{
 		$latte = $this->latteFactory->create();
-		$template = new $this->templateClass($latte);
+		$class = $class ?: $this->templateClass;
+		$template = new $class($latte);
+		if (!$template instanceof LatteTemplate) {
+			throw new Nette\InvalidArgumentException("Class $class does not extend " . LatteTemplate::class . '.');
+		}
+
 		$presenter = $control ? $control->getPresenterIfExists() : null;
 
 		if ($latte->onCompile instanceof \Traversable) {