Add Group, Limit, Map, Sort functions

Signed-off-by: davidarendsen <davidarendsen@hey.com>

src/Function/Group.php

@@ -0,0 +1,32 @@
+<?php
+
+namespace Arendsen\FluxQueryBuilder\Function;
+
+class Group extends Base {
+
+    /**
+     * @var array $columns
+     */
+    private $columns;
+
+    /**
+     * @var string $mode
+     */
+    private $mode;
+
+    public function __construct(array $columns, string $mode = 'by')
+    {
+        $this->columns = $columns;
+        $this->mode = $mode;
+    }
+
+    public function __toString()
+    {
+        $columns = array_map(function($column) {
+			return '"' . $column . '"';
+		}, $this->columns);
+
+        return '|> group(columns: [' . implode(', ', $columns) . '], mode: "' . $this->mode . '") ';
+    }
+
+}

src/Function/Limit.php

@@ -0,0 +1,22 @@
+<?php
+
+namespace Arendsen\FluxQueryBuilder\Function;
+
+class Limit extends Base {
+
+    /**
+     * @var int $limit
+     */
+    private $limit;
+
+    public function __construct(int $limit)
+    {
+        $this->limit = $limit;
+    }
+
+    public function __toString()
+    {
+        return '|> limit(n:' . (string)$this->limit . ') ';
+    }
+
+}

src/Function/Map.php

@@ -0,0 +1,22 @@
+<?php
+
+namespace Arendsen\FluxQueryBuilder\Function;
+
+class Map extends Base {
+
+    /**
+     * @var array $query
+     */
+    private $query;
+
+    public function __construct(string $query)
+    {
+        $this->query = $query;
+    }
+
+    public function __toString()
+    {
+        return '|> map(fn: (r) => ({ ' . $this->query . ' })) ';
+    }
+
+}

src/Function/Sort.php

@@ -0,0 +1,33 @@
+<?php
+
+namespace Arendsen\FluxQueryBuilder\Function;
+
+class Sort extends Base {
+
+    /**
+     * @var array $columns
+     */
+    private $columns;
+
+    /**
+     * @var bool $desc
+     */
+    private $desc;
+
+    public function __construct(array $columns, bool $desc = false)
+    {
+        $this->columns = $columns;
+        $this->desc = $desc;
+    }
+
+    public function __toString()
+    {
+        $columns = array_map(function($column) {
+			return '"' . $column . '"';
+		}, $this->columns);
+        $desc = $this->desc ? 'true' : 'false';
+
+        return '|> sort(columns: [' . implode(', ', $columns) . '], desc: ' . $desc . ') ';
+    }
+
+}

src/QueryBuilder.php

@@ -7,6 +7,10 @@ use Arendsen\FluxQueryBuilder\Function\Filter;
 use Arendsen\FluxQueryBuilder\Function\From;
 use Arendsen\FluxQueryBuilder\Function\Range;
 use Arendsen\FluxQueryBuilder\Function\Reduce;
+use Arendsen\FluxQueryBuilder\Function\Sort;
+use Arendsen\FluxQueryBuilder\Function\Map;
+use Arendsen\FluxQueryBuilder\Function\Group;
+use Arendsen\FluxQueryBuilder\Function\Limit;
 use Exception;
 
 class QueryBuilder {
@@ -15,12 +19,20 @@ class QueryBuilder {
     const FLUX_PART_RANGE = 'range';
     const FLUX_PART_FILTERS = 'filters';
     const FLUX_PART_REDUCE = 'reduce';
+    const FLUX_PART_MAP = 'map';
+    const FLUX_PART_SORT = 'sort';
+    const FLUX_PART_GROUP = 'group';
+    const FLUX_PART_LIMIT = 'limit';
 
     const PARTS = [
         self::FLUX_PART_FROM,
         self::FLUX_PART_RANGE,
         self::FLUX_PART_REDUCE,
-        self::FLUX_PART_FILTERS
+        self::FLUX_PART_FILTERS,
+        self::FLUX_PART_MAP,
+        self::FLUX_PART_SORT,
+        self::FLUX_PART_GROUP,
+        self::FLUX_PART_LIMIT,
     ];
 
     const REQUIRED_INPUT_FROM = 'from';
@@ -103,6 +115,42 @@ class QueryBuilder {
         return $this;
     }
 
+    public function addSort(array $columns, $desc): QueryBuilder
+    {
+        $this->addToQueryArray(
+            self::FLUX_PART_SORT,
+            new Sort($columns, $desc)
+        );
+        return $this;
+    }
+
+    public function addMap(string $query): QueryBuilder
+    {
+        $this->addToQueryArray(
+            self::FLUX_PART_MAP,
+            new Map($query)
+        );
+        return $this;
+    }
+
+    public function addGroup(array $columns, $mode = 'by'): QueryBuilder
+    {
+        $this->addToQueryArray(
+            self::FLUX_PART_GROUP,
+            new Group($columns, $mode)
+        );
+        return $this;
+    }
+
+    public function addLimit(int $limit): QueryBuilder
+    {
+        $this->addToQueryArray(
+            self::FLUX_PART_LIMIT,
+            new Limit($limit)
+        );
+        return $this;
+    }
+
     protected function addToQuery($key, $query)
     {
         $this->fluxQueryParts[$key] = $query;