Extract Universe Flux functions into a trait

Signed-off-by: David Arendsen <darendsen@gamepoint.com>

src/Builder/Basics.php

@@ -0,0 +1,104 @@
+<?php
+
+namespace Arendsen\FluxQueryBuilder\Builder;
+
+use DateTime;
+use Arendsen\FluxQueryBuilder\QueryBuilder;
+use Arendsen\FluxQueryBuilder\Builder\QueryBuilderInterface;
+use Arendsen\FluxQueryBuilder\Builder\FluxPart;
+use Arendsen\FluxQueryBuilder\Expression\KeyValue;
+use Arendsen\FluxQueryBuilder\Expression\KeyFilter;
+use Arendsen\FluxQueryBuilder\Functions\Filter;
+use Arendsen\FluxQueryBuilder\Functions\From;
+use Arendsen\FluxQueryBuilder\Functions\Range;
+use Arendsen\FluxQueryBuilder\Functions\RawFunction;
+
+trait Basics
+{
+    public function from(array $from): QueryBuilderInterface
+    {
+        $this->addRequiredData(QueryBuilder::REQUIRED_INPUT_FROM, $from);
+        $this->addToQuery(
+            FluxPart::FROM,
+            new From($from)
+        );
+        return $this;
+    }
+
+    public function fromBucket(string $bucket): QueryBuilderInterface
+    {
+        $this->from(['bucket' => $bucket]);
+        return $this;
+    }
+
+    public function fromMeasurement(string $measurement): QueryBuilderInterface
+    {
+        $this->addRequiredData(QueryBuilder::REQUIRED_INPUT_MEASUREMENT, $measurement);
+        $this->addKeyFilter(KeyFilter::setEqualTo('_measurement', $measurement));
+        return $this;
+    }
+
+    /**
+     * @deprecated
+     *
+     * @param KeyValue $keyValue
+     * @return QueryBuilder
+     */
+    public function addFilter(KeyValue $keyValue): QueryBuilderInterface
+    {
+        $this->addToQuery(
+            FluxPart::FILTERS,
+            new Filter($keyValue)
+        );
+        return $this;
+    }
+
+    public function addKeyFilter(KeyFilter $keyFilter): QueryBuilderInterface
+    {
+        $this->addToQuery(
+            FluxPart::FILTERS,
+            new Filter($keyFilter)
+        );
+        return $this;
+    }
+
+    public function addFieldFilter(array $fields): QueryBuilderInterface
+    {
+        $this->addToQuery(
+            FluxPart::FILTERS,
+            new Filter($fields)
+        );
+        return $this;
+    }
+
+    public function addRange(DateTime $start, ?DateTime $stop = null): QueryBuilderInterface
+    {
+        $this->addRequiredData(QueryBuilder::REQUIRED_INPUT_RANGE, [$start, $stop]);
+        $this->addToQuery(
+            FluxPart::RANGE,
+            new Range($start, $stop)
+        );
+        return $this;
+    }
+
+    public function addRangeStart(DateTime $start): QueryBuilderInterface
+    {
+        $this->addRange($start);
+        return $this;
+    }
+
+    public function addRangeInBetween(DateTime $start, DateTime $stop): QueryBuilderInterface
+    {
+        $this->addRange($start, $stop);
+        return $this;
+    }
+
+    public function addRawFunction(string $input): QueryBuilderInterface
+    {
+        $this->addToQuery(
+            FluxPart::RAWFUNCTION,
+            new RawFunction($input)
+        );
+        return $this;
+    }
+}