Update task time tracking based on subtask time tracking

1 files changed, 39 insertions, 0 deletions

diff --git a/app/Model/Subtask.php b/app/Model/Subtask.php
index 9ecd2c6a..1d5ed566 100644
--- a/app/Model/Subtask.php
+++ b/app/Model/Subtask.php
@@ -176,6 +176,9 @@ class Subtask extends Base
         $subtask_id = $this->persist(self::TABLE, $values);
 
         if ($subtask_id) {
+
+            $this->updateTaskTimeTracking($values['task_id']);
+
             $this->container['dispatcher']->dispatch(
                 self::EVENT_CREATE,
                 new SubtaskEvent(array('id' => $subtask_id) + $values)
@@ -198,6 +201,11 @@ class Subtask extends Base
         $result = $this->db->table(self::TABLE)->eq('id', $values['id'])->save($values);
 
         if ($result) {
+
+            if (isset($values['task_id'])) {
+                $this->updateTaskTimeTracking($values['task_id']);
+            }
+
             $this->container['dispatcher']->dispatch(
                 self::EVENT_UPDATE,
                 new SubtaskEvent($values)
@@ -260,6 +268,37 @@ class Subtask extends Base
     }
 
     /**
+     * Update task time tracking based on subtasks time tracking
+     *
+     * @access public
+     * @param  integer   $task_id    Task id
+     * @return bool
+     */
+    public function updateTaskTimeTracking($task_id)
+    {
+        $result = $this->db
+                       ->table(self::TABLE)
+                       ->eq('task_id', $task_id)
+                       ->columns(
+                           'SUM(time_spent) AS total_spent',
+                           'SUM(time_estimated) AS total_estimated'
+                       )
+                       ->findOne();
+
+        if (empty($result['total_spent']) && empty($result['total_estimated'])) {
+            return true;
+        }
+
+        return $this->db
+                    ->table(Task::TABLE)
+                    ->eq('id', $task_id)
+                    ->update(array(
+                        'time_spent' => $result['total_spent'],
+                        'time_estimated' => $result['total_estimated'],
+                    ));
+    }
+
+    /**
      * Remove
      *
      * @access public