added option to provide custom template

Author: peldax

src/MultistepForm/MultistepForm.php

@@ -12,6 +12,7 @@ final class MultistepForm extends \Nette\Application\UI\Control
     private string $subsessionId;
     private int $currentStep;
     private array $factories = [];
+    private array $templates = [];
     private array $defaults = [];
     private bool $editing = false;
     private bool $stepSave = false;
@@ -28,6 +29,13 @@ public function __construct(
 
     public function render() : void
     {
+        if (\count($this->factories) < $this->currentStep ||
+            \count($this->templates) < $this->currentStep
+        ) {
+            throw new \Nette\InvalidStateException('No factory given');
+        }
+
+        $this->template->customTemplate = $this->templates[$this->currentStep - 1];
         $this->template->setFile(__DIR__ . '/MultistepForm.latte');
         $this->template->render();
     }
@@ -39,9 +47,10 @@ public function setFactories(array $factories) : self
         return $this;
     }
 
-    public function addFactory($factory) : self
+    public function addFactory($factory, string $templatePath = null) : self
     {
         $this->factories[] = $factory;
+        $this->templates[] = $templatePath;
 
         return $this;
     }
@@ -116,7 +125,7 @@ public function getMaxStep() : int
 
     /**
      * Returns all filled values.
-     * 
+     *
      * @return array
      */
     public function getValues() : array
@@ -148,10 +157,6 @@ public function loadState(array $params): void
 
     protected function createComponentForm() : \Nette\Forms\Form
     {
-        if (\count($this->factories) < $this->currentStep) {
-            throw new \Nette\InvalidStateException('No factory given');
-        }
-
         $factory = $this->factories[$this->currentStep - 1];
 
         if (\is_object($factory) && \method_exists($factory, 'create')) {
@@ -187,9 +192,9 @@ private function appendButtons(\Nette\Forms\Form $form) : \Nette\Forms\Form
             $previous->setDisabled();
         }
 
-        /*if ($this->isEdit()) {
+        if ($this->stepSave && $this->isEdit()) {
             $form->addSubmit('quicksave', 'Uložit');
-        }*/
+        }
 
         $next->onClick[] = function (\Nette\Forms\Controls\SubmitButton $button) : void {
             $form = $button->getForm();