Some more moved to traits.

2025-09-05 12:12:18 +00:00 · 2020-03-21 09:01:14 +01:00
parent d16389748f
commit 5701597e20
3 changed files with 211 additions and 147 deletions
--- a/app/Helpers/Collector/Extensions/TimeCollection.php
+++ b/app/Helpers/Collector/Extensions/TimeCollection.php
@@ -0,0 +1,120 @@
+<?php
+declare(strict_types=1);
+/**
+ * TimeCollection.php
+ * Copyright (c) 2020 james@firefly-iii.org
+ *
+ * This file is part of Firefly III (https://github.com/firefly-iii).
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+namespace FireflyIII\Helpers\Collector\Extensions;
+
+use Carbon\Carbon;
+use FireflyIII\Helpers\Collector\GroupCollectorInterface;
+
+/**
+ * Trait TimeCollection
+ */
+trait TimeCollection
+{
+
+    /**
+     * Collect transactions after a specific date.
+     *
+     * @param Carbon $date
+     *
+     * @return GroupCollectorInterface
+     */
+    public function setAfter(Carbon $date): GroupCollectorInterface
+    {
+        $afterStr = $date->format('Y-m-d 00:00:00');
+        $this->query->where('transaction_journals.date', '>=', $afterStr);
+
+        return $this;
+    }
+
+    /**
+     * Collect transactions before a specific date.
+     *
+     * @param Carbon $date
+     *
+     * @return GroupCollectorInterface
+     */
+    public function setBefore(Carbon $date): GroupCollectorInterface
+    {
+        $beforeStr = $date->format('Y-m-d 00:00:00');
+        $this->query->where('transaction_journals.date', '<=', $beforeStr);
+
+        return $this;
+    }
+
+    /**
+     * Collect transactions created on a specific date.
+     *
+     * @param Carbon $date
+     *
+     * @return GroupCollectorInterface
+     */
+    public function setCreatedAt(Carbon $date): GroupCollectorInterface
+    {
+        $after  = $date->format('Y-m-d 00:00:00');
+        $before = $date->format('Y-m-d 23:59:59');
+        $this->query->where('transaction_journals.created_at', '>=', $after);
+        $this->query->where('transaction_journals.created_at', '<=', $before);
+
+        return $this;
+    }
+
+    /**
+     * Set the start and end time of the results to return.
+     *
+     * @param Carbon $start
+     * @param Carbon $end
+     *
+     * @return GroupCollectorInterface
+     */
+    public function setRange(Carbon $start, Carbon $end): GroupCollectorInterface
+    {
+        if ($end < $start) {
+            [$start, $end] = [$end, $start];
+        }
+        $startStr = $start->format('Y-m-d H:i:s');
+        $endStr   = $end->format('Y-m-d H:i:s');
+
+        $this->query->where('transaction_journals.date', '>=', $startStr);
+        $this->query->where('transaction_journals.date', '<=', $endStr);
+
+        return $this;
+    }
+
+
+    /**
+     * Collect transactions updated on a specific date.
+     *
+     * @param Carbon $date
+     *
+     * @return GroupCollectorInterface
+     */
+    public function setUpdatedAt(Carbon $date): GroupCollectorInterface
+    {
+        $after  = $date->format('Y-m-d 00:00:00');
+        $before = $date->format('Y-m-d 23:59:59');
+        $this->query->where('transaction_journals.updated_at', '>=', $after);
+        $this->query->where('transaction_journals.updated_at', '<=', $before);
+
+        return $this;
+    }
+}