addRemoveButton accepts Nette\Utils\Html (#87)

contributte · Nov 21, 2023 · 387a2cd · 387a2cd
1 parent b89d552
commit 387a2cd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/Buttons/RemoveButton.php b/src/Buttons/RemoveButton.php
@@ -5,6 +5,7 @@
 use Contributte\FormMultiplier\Multiplier;
 use Nette\Forms\Controls\SubmitButton;
 use Nette\SmartObject;
+use Nette\Utils\Html;
 
 final class RemoveButton
 {
@@ -14,12 +15,12 @@ final class RemoveButton
 	/** @var callable[] */
 	public array $onCreate = [];
 
-	private ?string $caption = null;
+	private Html|string|null $caption = null;
 
 	/** @var string[] */
 	private array $classes = [];
 
-	public function __construct(?string $caption)
+	public function __construct(Html|string|null $caption)
 	{
 		$this->caption = $caption;
 	}

diff --git a/src/Multiplier.php b/src/Multiplier.php
@@ -13,6 +13,7 @@
 use Nette\Forms\Form;
 use Nette\Utils\ArrayHash;
 use Nette\Utils\Arrays;
+use Nette\Utils\Html;
 use Traversable;
 
 class Multiplier extends Container
@@ -153,7 +154,7 @@ public function getCopyNumber(): int
 		return $this->copyNumber;
 	}
 
-	public function addRemoveButton(?string $caption = null): RemoveButton
+	public function addRemoveButton(Html|string|null $caption = null): RemoveButton
 	{
 		return $this->removeButton = new RemoveButton($caption);
 	}