1 files changed, 105 insertions, 0 deletions
diff --git a/app/Formatter/BoardSwimlaneFormatter.php b/app/Formatter/BoardSwimlaneFormatter.php
new file mode 100644
index 00000000..91b4bfd7
--- /dev/null
+++ b/app/Formatter/BoardSwimlaneFormatter.php
@@ -0,0 +1,105 @@
+<?php
+
+namespace Kanboard\Formatter;
+
+use Kanboard\Core\Filter\FormatterInterface;
+
+/**
+ * Board Swimlane Formatter
+ *
+ * @package formatter
+ * @author  Frederic Guillot
+ */
+class BoardSwimlaneFormatter extends BaseFormatter implements FormatterInterface
+{
+    protected $swimlanes = array();
+    protected $columns = array();
+    protected $tasks = array();
+
+    /**
+     * Set swimlanes
+     *
+     * @access public
+     * @param  array $swimlanes
+     * @return $this
+     */
+    public function withSwimlanes($swimlanes)
+    {
+        $this->swimlanes = $swimlanes;
+        return $this;
+    }
+
+    /**
+     * Set columns
+     *
+     * @access public
+     * @param  array $columns
+     * @return $this
+     */
+    public function withColumns($columns)
+    {
+        $this->columns = $columns;
+        return $this;
+    }
+
+    /**
+     * Set tasks
+     *
+     * @access public
+     * @param  array $tasks
+     * @return $this
+     */
+    public function withTasks(array $tasks)
+    {
+        $this->tasks = $tasks;
+        return $this;
+    }
+
+    /**
+     * Apply formatter
+     *
+     * @access public
+     * @return array
+     */
+    public function format()
+    {
+        $nb_swimlanes = count($this->swimlanes);
+        $nb_columns = count($this->columns);
+
+        foreach ($this->swimlanes as &$swimlane) {
+            $swimlane['columns'] = BoardColumnFormatter::getInstance($this->container)
+                ->withSwimlaneId($swimlane['id'])
+                ->withColumns($this->columns)
+                ->withTasks($this->tasks)
+                ->format();
+
+            $swimlane['nb_swimlanes'] = $nb_swimlanes;
+            $swimlane['nb_columns'] = $nb_columns;
+            $swimlane['nb_tasks'] = array_column_sum($swimlane['columns'], 'nb_tasks');
+            $swimlane['score'] = array_column_sum($swimlane['columns'], 'score');
+
+            $this->calculateStatsByColumnAcrossSwimlanes($swimlane['columns']);
+        }
+
+        return $this->swimlanes;
+    }
+
+    /**
+     * Calculate stats for each column acrosss all swimlanes
+     *
+     * @access protected
+     * @param  array $columns
+     */
+    protected function calculateStatsByColumnAcrossSwimlanes(array $columns)
+    {
+        foreach ($columns as $columnIndex => $column) {
+            if (! isset($this->swimlanes[0]['columns'][$columnIndex]['column_nb_tasks'])) {
+                $this->swimlanes[0]['columns'][$columnIndex]['column_nb_tasks'] = 0;
+                $this->swimlanes[0]['columns'][$columnIndex]['column_score'] = 0;
+            }
+
+            $this->swimlanes[0]['columns'][$columnIndex]['column_nb_tasks'] += $column['nb_tasks'];
+            $this->swimlanes[0]['columns'][$columnIndex]['column_score'] += $column['score'];
+        }
+    }
+}