diff --git a/src/Action/DeleteAction.php b/src/Action/DeleteAction.php index f9ebac3..b8a7076 100644 --- a/src/Action/DeleteAction.php +++ b/src/Action/DeleteAction.php @@ -19,6 +19,9 @@ class DeleteAction extends BaseAction public $deleteStatus = 0; public $errorMessage = 'Record not found'; + public $extraFields = null; + public $formName = ''; + /** * @author Adegoke Obasa * @author Akinwunmi Taiwo @@ -33,16 +36,18 @@ public function run() $controller->isPostCheck($referrerUrl); if (!$this->model) { - $controller->flashError($this->errorMessage); - } else { - $this->model->{$this->deleteAttribute} = $this->deleteStatus; - if (!$this->model->save()) { - $controller->flashError($this->model->getErrors()); - } else { - $controller->flashSuccess($this->successMessage); - } + return $controller->returnError($this->errorMessage, $this->returnUrl); + } + + $this->model->{$this->deleteAttribute} = $this->deleteStatus; + if ($this->extraFields) { + $this->model->load($this->extraFields,$this->formName); + } + + if (!$this->model->save()) { + return $controller->returnError($this->model->getErrors(), $this->returnUrl); } - return $controller->redirect($this->returnUrl); + return $controller->returnSuccess($this->successMessage, $this->returnUrl); } }