1 files changed, 9 insertions, 28 deletions

diff --git a/app/Controller/SubtaskStatusController.php b/app/Controller/SubtaskStatusController.php
index d4d356c3..72feb685 100644
--- a/app/Controller/SubtaskStatusController.php
+++ b/app/Controller/SubtaskStatusController.php
@@ -21,15 +21,9 @@ class SubtaskStatusController extends BaseController
         $subtask = $this->getSubtask();
 
         $status = $this->subtaskStatusModel->toggleStatus($subtask['id']);
+        $subtask['status'] = $status;
 
-        if ($this->request->getIntegerParam('refresh-table') === 0) {
-            $subtask['status'] = $status;
-            $html = $this->helper->subtask->toggleStatus($subtask, $task['project_id']);
-        } else {
-            $html = $this->renderTable($task);
-        }
-
-        $this->response->html($html);
+        $this->response->html($this->helper->subtask->renderToggleStatus($task, $subtask));
     }
 
     /**
@@ -40,32 +34,19 @@ class SubtaskStatusController extends BaseController
     public function timer()
     {
         $task = $this->getTask();
-        $subtask_id = $this->request->getIntegerParam('subtask_id');
+        $subtaskId = $this->request->getIntegerParam('subtask_id');
         $timer = $this->request->getStringParam('timer');
 
         if ($timer === 'start') {
-            $this->subtaskTimeTrackingModel->logStartTime($subtask_id, $this->userSession->getId());
+            $this->subtaskTimeTrackingModel->logStartTime($subtaskId, $this->userSession->getId());
         } elseif ($timer === 'stop') {
-            $this->subtaskTimeTrackingModel->logEndTime($subtask_id, $this->userSession->getId());
+            $this->subtaskTimeTrackingModel->logEndTime($subtaskId, $this->userSession->getId());
             $this->subtaskTimeTrackingModel->updateTaskTimeTracking($task['id']);
         }
 
-        $this->response->html($this->renderTable($task));
-    }
-
-    /**
-     * Render table
-     *
-     * @access private
-     * @param  array  $task
-     * @return string
-     */
-    private function renderTable(array $task)
-    {
-        return $this->template->render('subtask/table', array(
-            'task' => $task,
-            'subtasks' => $this->subtaskModel->getAll($task['id']),
-            'editable' => true,
-        ));
+        $this->response->html($this->template->render('subtask/timer', array(
+            'task'    => $task,
+            'subtask' => $this->subtaskModel->getByIdWithDetails($subtaskId),
+        )));
     }
 }