Entity: FindEntity trait

src/LeanMapperWorkflow/Repository.php

@@ -14,12 +14,13 @@ class Repository extends AbstractRepository
 
 	/**
 	 * @param  int|Restriction $id
+	 * @param  callback|null   $notFoundCallaback
 	 *
 	 * @return Entity
 	 *
 	 * @throws InvalidArgumentException
 	 */
-	public function find($id)
+	public function find($id, $notFoundCallback = NULL)
 	{
 		if (is_numeric($id)) {
 			$restriction = $this->createRestriction();
@@ -35,7 +36,11 @@ class Repository extends AbstractRepository
 
 		$row = $statement->fetch();
 		if ($row === FALSE) {
-			throw new \InvalidArgumentException('Not found: ' . var_export($id, TRUE));
+			if (is_callable($notFoundCallback)) {
+				\Nette\Utils\Callback::invokeArgs($notFoundCallback, [$id]);
+			} else {
+				throw new InvalidArgumentException('Not found: ' . var_export($id, TRUE));
+			}
 		}
 
 		return $this->createEntity($row);

src/LeanMapperWorkflow/Traits/FindEntity.php

+<?php
+
+namespace LeanMapperWorkflow\Traits;
+
+use LeanMapperWorkflow\Repository;
+
+
+trait FindEntity
+{
+
+	/**
+	 * @param  Repository $repository
+	 * @param  mixed      $id
+	 * @param  bool       $showMessage
+	 * @param  string     $redirectLink
+	 *
+	 * @throws \InvalidArgumentException
+	 */
+	protected final function findEntity(Repository $repository, $id, $showMessage = FALSE, $redirectLink = NULL)
+	{
+		if ($this instanceof \Nette\Application\IPresenter && ($showMessage !== FALSE || $redirectLink !== NULL)) {
+			$callback = function() use ($showMessage, $redirectLink) {
+				if ($showMessage) {
+					$this->flashMessage('@todo', 'error');
+				}
+
+				if ($link !== NULL) {
+					$this->redirect($link);
+				}
+			};
+		} else {
+			$callback = NULL;
+		}
+
+		return $repository->find($id, $callback);
+	}
+
+}
\ No newline at end of file