diff --git a/src/Application/UI/Control.php b/src/Application/UI/Control.php index d7dc2d9a1..d694665b3 100644 --- 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;