Control::flashMessage accepts custom flash object

nette · Nov 16, 2020 · ea5ac15 · ea5ac15
1 parent 67edb5b
commit ea5ac15
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/Application/UI/Control.php b/src/Application/UI/Control.php
@@ -93,15 +93,17 @@ public function templatePrepareFilters(ITemplate $template): void
 
 	/**
 	 * Saves the message to template, that can be displayed after redirect.
+	 * @param  string|\stdClass  $message
 	 */
 	public function flashMessage($message, string $type = 'info'): \stdClass
 	{
 		$id = $this->getParameterId('flash');
-		$messages = $this->getPresenter()->getFlashSession()->$id;
-		$messages[] = $flash = (object) [
+		$flash = $message instanceof \stdClass ? $message : (object) [
 			'message' => $message,
 			'type' => $type,
 		];
+		$messages = $this->getPresenter()->getFlashSession()->$id;
+		$messages[] = $flash;
 		$this->getTemplate()->flashes = $messages;
 		$this->getPresenter()->getFlashSession()->$id = $messages;
 		return $flash;