1 files changed, 56 insertions, 0 deletions
diff --git a/app/Formatter/TaskListSubtaskAssigneeFormatter.php b/app/Formatter/TaskListSubtaskAssigneeFormatter.php
new file mode 100644
index 00000000..73391766
--- /dev/null
+++ b/app/Formatter/TaskListSubtaskAssigneeFormatter.php
@@ -0,0 +1,56 @@
+<?php
+
+namespace Kanboard\Formatter;
+
+/**
+ * Class TaskListSubtaskAssigneeFormatter
+ *
+ * @package Kanboard\Formatter
+ * @author  Frederic Guillot
+ */
+class TaskListSubtaskAssigneeFormatter extends TaskListFormatter
+{
+    protected $userId = 0;
+    protected $withoutEmptyTasks = false;
+
+    /**
+     * Set assignee
+     *
+     * @param  integer $userId
+     * @return $this
+     */
+    public function withUserId($userId)
+    {
+        $this->userId = $userId;
+        return $this;
+    }
+
+    public function withoutEmptyTasks()
+    {
+        $this->withoutEmptyTasks = true;
+        return $this;
+    }
+
+    /**
+     * Apply formatter
+     *
+     * @access public
+     * @return array
+     */
+    public function format()
+    {
+        $tasks = parent::format();
+        $taskIds = array_column($tasks, 'id');
+        $subtasks = $this->subtaskModel->getAllByTaskIdsAndAssignee($taskIds, $this->userId);
+        $subtasks = array_column_index($subtasks, 'task_id');
+        array_merge_relation($tasks, $subtasks, 'subtasks', 'id');
+
+        if ($this->withoutEmptyTasks) {
+            $tasks = array_filter($tasks, function (array $task) {
+                return count($task['subtasks']) > 0;
+            });
+        }
+
+        return $tasks;
+    }
+}