feat: add support for DATE_ADD(), DATE_SUBTRACT() and DATE_BIN()

src/MartinGeorgiev/Doctrine/ORM/Query/AST/Functions/DateAdd.php

@@ -0,0 +1,45 @@
+<?php
+
+declare(strict_types=1);
+
+namespace MartinGeorgiev\Doctrine\ORM\Query\AST\Functions;
+
+use Doctrine\ORM\Query\AST\Node;
+use MartinGeorgiev\Doctrine\ORM\Query\AST\Functions\Exception\InvalidArgumentForVariadicFunctionException;
+use MartinGeorgiev\Doctrine\ORM\Query\AST\Functions\Traits\TimezoneValidationTrait;
+
+/**
+ * Implementation of PostgreSQL DATE_ADD().
+ *
+ * Adds an interval to a timestamp with time zone, computing times of day and daylight-savings
+ * adjustments according to the time zone.
+ *
+ * @see https://www.postgresql.org/docs/16/functions-datetime.html
+ * @since 3.1
+ *
+ * @author Martin Georgiev <martin.georgiev@gmail.com>
+ *
+ * @example Using it in DQL: "SELECT DATE_ADD(e.timestampWithTz, '1 day', 'Europe/Sofia') FROM Entity e"
+ */
+class DateAdd extends BaseVariadicFunction
+{
+    use TimezoneValidationTrait;
+
+    protected function customizeFunction(): void
+    {
+        $this->setFunctionPrototype('date_add(%s)');
+    }
+
+    protected function validateArguments(Node ...$arguments): void
+    {
+        $argumentCount = \count($arguments);
+        if ($argumentCount < 2 || $argumentCount > 3) {
+            throw InvalidArgumentForVariadicFunctionException::between('date_add', 2, 3);
+        }
+
+        // Validate that the third parameter is a valid timezone if provided
+        if ($argumentCount === 3) {
+            $this->validateTimezone($arguments[2], 'DATE_ADD');
+        }
+    }
+}

src/MartinGeorgiev/Doctrine/ORM/Query/AST/Functions/DateBin.php

@@ -0,0 +1,28 @@
+<?php
+
+declare(strict_types=1);
+
+namespace MartinGeorgiev\Doctrine\ORM\Query\AST\Functions;
+
+/**
+ * Implementation of PostgreSQL DATE_BIN().
+ *
+ * Bins input into specified interval aligned with specified origin.
+ *
+ * @see https://www.postgresql.org/docs/14/functions-datetime.html
+ * @since 3.1
+ *
+ * @author Martin Georgiev <martin.georgiev@gmail.com>
+ *
+ * @example Using it in DQL: "SELECT DATE_BIN('15 minutes', e.createdAt, '2001-02-16 20:05:00') FROM Entity e"
+ */
+class DateBin extends BaseFunction
+{
+    protected function customizeFunction(): void
+    {
+        $this->setFunctionPrototype('date_bin(%s, %s, %s)');
+        $this->addNodeMapping('StringPrimary');
+        $this->addNodeMapping('StringPrimary');
+        $this->addNodeMapping('StringPrimary');
+    }
+}

src/MartinGeorgiev/Doctrine/ORM/Query/AST/Functions/DateSubtract.php

@@ -0,0 +1,45 @@
+<?php
+
+declare(strict_types=1);
+
+namespace MartinGeorgiev\Doctrine\ORM\Query\AST\Functions;
+
+use Doctrine\ORM\Query\AST\Node;
+use MartinGeorgiev\Doctrine\ORM\Query\AST\Functions\Exception\InvalidArgumentForVariadicFunctionException;
+use MartinGeorgiev\Doctrine\ORM\Query\AST\Functions\Traits\TimezoneValidationTrait;
+
+/**
+ * Implementation of PostgreSQL DATE_SUBTRACT().
+ *
+ * Subtracts an interval from a timestamp with time zone, computing times of day and daylight-savings
+ * adjustments according to the time zone.
+ *
+ * @see https://www.postgresql.org/docs/16/functions-datetime.html
+ * @since 3.1
+ *
+ * @author Martin Georgiev <martin.georgiev@gmail.com>
+ *
+ * @example Using it in DQL: "SELECT DATE_SUBTRACT(e.timestampWithTz, '1 day', 'Europe/Sofia') FROM Entity e"
+ */
+class DateSubtract extends BaseVariadicFunction
+{
+    use TimezoneValidationTrait;
+
+    protected function customizeFunction(): void
+    {
+        $this->setFunctionPrototype('date_subtract(%s)');
+    }
+
+    protected function validateArguments(Node ...$arguments): void
+    {
+        $argumentCount = \count($arguments);
+        if ($argumentCount < 2 || $argumentCount > 3) {
+            throw InvalidArgumentForVariadicFunctionException::between('date_subtract', 2, 3);
+        }
+
+        // Validate that the third parameter is a valid timezone if provided
+        if ($argumentCount === 3) {
+            $this->validateTimezone($arguments[2], 'DATE_SUBTRACT');
+        }
+    }
+}

src/MartinGeorgiev/Doctrine/ORM/Query/AST/Functions/Exception/InvalidTimezoneException.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace MartinGeorgiev\Doctrine\ORM\Query\AST\Functions\Exception;
+
+use Doctrine\DBAL\Types\ConversionException;
+
+/**
+ * @since 3.1
+ *
+ * @author Martin Georgiev <martin.georgiev@gmail.com>
+ */
+class InvalidTimezoneException extends ConversionException
+{
+    public static function forNonLiteralNode(string $nodeClass, string $functionName): self
+    {
+        return new self(\sprintf(
+            'The timezone parameter for %s must be a string literal, got %s',
+            $functionName,
+            $nodeClass
+        ));
+    }
+
+    public static function forInvalidTimezone(string $timezone, string $functionName): self
+    {
+        return new self(\sprintf(
+            'Invalid timezone "%s" provided for %s. Must be a valid PHP timezone identifier.',
+            $timezone,
+            $functionName
+        ));
+    }
+}

src/MartinGeorgiev/Doctrine/ORM/Query/AST/Functions/Traits/TimezoneValidationTrait.php

@@ -0,0 +1,48 @@
+<?php
+
+declare(strict_types=1);
+
+namespace MartinGeorgiev\Doctrine\ORM\Query\AST\Functions\Traits;
+
+use Doctrine\ORM\Query\AST\Literal;
+use Doctrine\ORM\Query\AST\Node;
+use MartinGeorgiev\Doctrine\ORM\Query\AST\Functions\Exception\InvalidTimezoneException;
+
+/**
+ * Provides timezone validation functionality for functions that use valid PHP timezones.
+ *
+ * @since 3.1
+ *
+ * @author Martin Georgiev <martin.georgiev@gmail.com>
+ */
+trait TimezoneValidationTrait
+{
+    /**
+     * Validates that the given node represents a valid PHP timezone.
+     *
+     * @throws InvalidTimezoneException If the timezone is invalid
+     */
+    protected function validateTimezone(Node $node, string $functionName): void
+    {
+        if (!$node instanceof Literal || !\is_string($node->value)) {
+            throw InvalidTimezoneException::forNonLiteralNode($node::class, $functionName);
+        }
+
+        $timezone = \trim((string) $node->value, "'\"");
+
+        if (!$this->isValidTimezone($timezone)) {
+            throw InvalidTimezoneException::forInvalidTimezone($timezone, $functionName);
+        }
+    }
+
+    private function isValidTimezone(string $timezone): bool
+    {
+        try {
+            new \DateTimeZone($timezone);
+
+            return true;
+        } catch (\Exception) {
+            return false;
+        }
+    }
+}

tests/MartinGeorgiev/Doctrine/ORM/Query/AST/Functions/DateAddTest.php

@@ -0,0 +1,46 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Tests\MartinGeorgiev\Doctrine\ORM\Query\AST\Functions;
+
+use Fixtures\MartinGeorgiev\Doctrine\Entity\ContainsDates;
+use MartinGeorgiev\Doctrine\ORM\Query\AST\Functions\DateAdd;
+use MartinGeorgiev\Doctrine\ORM\Query\AST\Functions\Exception\InvalidTimezoneException;
+
+class DateAddTest extends TestCase
+{
+    protected function getStringFunctions(): array
+    {
+        return [
+            'DATE_ADD' => DateAdd::class,
+        ];
+    }
+
+    protected function getExpectedSqlStatements(): array
+    {
+        return [
+            'adds 1 day with timezone' => "SELECT date_add(c0_.datetimetz1, '1 day', 'Europe/Sofia') AS sclr_0 FROM ContainsDates c0_",
+            'adds 2 hours with timezone' => "SELECT date_add(c0_.datetimetz1, '2 hours', 'UTC') AS sclr_0 FROM ContainsDates c0_",
+            'adds 3 days without timezone' => "SELECT date_add(c0_.datetimetz1, '3 days') AS sclr_0 FROM ContainsDates c0_",
+        ];
+    }
+
+    protected function getDqlStatements(): array
+    {
+        return [
+            'adds 1 day with timezone' => \sprintf("SELECT DATE_ADD(e.datetimetz1, '1 day', 'Europe/Sofia') FROM %s e", ContainsDates::class),
+            'adds 2 hours with timezone' => \sprintf("SELECT DATE_ADD(e.datetimetz1, '2 hours', 'UTC') FROM %s e", ContainsDates::class),
+            'adds 3 days without timezone' => \sprintf("SELECT DATE_ADD(e.datetimetz1, '3 days') FROM %s e", ContainsDates::class),
+        ];
+    }
+
+    public function test_invalid_timezone_throws_exception(): void
+    {
+        $this->expectException(InvalidTimezoneException::class);
+        $this->expectExceptionMessage('Invalid timezone "Invalid/Timezone" provided for DATE_ADD');
+
+        $dql = \sprintf("SELECT DATE_ADD(e.datetimetz1, '1 day', 'Invalid/Timezone') FROM %s e", ContainsDates::class);
+        $this->buildEntityManager()->createQuery($dql)->getSQL();
+    }
+}

tests/MartinGeorgiev/Doctrine/ORM/Query/AST/Functions/DateBinTest.php

@@ -0,0 +1,34 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Tests\MartinGeorgiev\Doctrine\ORM\Query\AST\Functions;
+
+use Fixtures\MartinGeorgiev\Doctrine\Entity\ContainsDates;
+use MartinGeorgiev\Doctrine\ORM\Query\AST\Functions\DateBin;
+
+class DateBinTest extends TestCase
+{
+    protected function getStringFunctions(): array
+    {
+        return [
+            'DATE_BIN' => DateBin::class,
+        ];
+    }
+
+    protected function getExpectedSqlStatements(): array
+    {
+        return [
+            'bins by 15 minutes' => "SELECT date_bin('15 minutes', c0_.datetime1, '2001-02-16 20:05:00') AS sclr_0 FROM ContainsDates c0_",
+            'bins by 1 hour' => "SELECT date_bin('1 hour', c0_.datetime1, '2001-02-16 00:00:00') AS sclr_0 FROM ContainsDates c0_",
+        ];
+    }
+
+    protected function getDqlStatements(): array
+    {
+        return [
+            'bins by 15 minutes' => \sprintf("SELECT DATE_BIN('15 minutes', e.datetime1, '2001-02-16 20:05:00') FROM %s e", ContainsDates::class),
+            'bins by 1 hour' => \sprintf("SELECT DATE_BIN('1 hour', e.datetime1, '2001-02-16 00:00:00') FROM %s e", ContainsDates::class),
+        ];
+    }
+}

tests/MartinGeorgiev/Doctrine/ORM/Query/AST/Functions/DateSubtractTest.php

@@ -0,0 +1,46 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Tests\MartinGeorgiev\Doctrine\ORM\Query\AST\Functions;
+
+use Fixtures\MartinGeorgiev\Doctrine\Entity\ContainsDates;
+use MartinGeorgiev\Doctrine\ORM\Query\AST\Functions\DateSubtract;
+use MartinGeorgiev\Doctrine\ORM\Query\AST\Functions\Exception\InvalidTimezoneException;
+
+class DateSubtractTest extends TestCase
+{
+    protected function getStringFunctions(): array
+    {
+        return [
+            'DATE_SUBTRACT' => DateSubtract::class,
+        ];
+    }
+
+    protected function getExpectedSqlStatements(): array
+    {
+        return [
+            'subtracts 1 day with timezone' => "SELECT date_subtract(c0_.datetimetz1, '1 day', 'Europe/Sofia') AS sclr_0 FROM ContainsDates c0_",
+            'subtracts 2 hours with timezone' => "SELECT date_subtract(c0_.datetimetz1, '2 hours', 'UTC') AS sclr_0 FROM ContainsDates c0_",
+            'subtracts 3 days without timezone' => "SELECT date_subtract(c0_.datetimetz1, '3 days') AS sclr_0 FROM ContainsDates c0_",
+        ];
+    }
+
+    protected function getDqlStatements(): array
+    {
+        return [
+            'subtracts 1 day with timezone' => \sprintf("SELECT DATE_SUBTRACT(e.datetimetz1, '1 day', 'Europe/Sofia') FROM %s e", ContainsDates::class),
+            'subtracts 2 hours with timezone' => \sprintf("SELECT DATE_SUBTRACT(e.datetimetz1, '2 hours', 'UTC') FROM %s e", ContainsDates::class),
+            'subtracts 3 days without timezone' => \sprintf("SELECT DATE_SUBTRACT(e.datetimetz1, '3 days') FROM %s e", ContainsDates::class),
+        ];
+    }
+
+    public function test_invalid_timezone_throws_exception(): void
+    {
+        $this->expectException(InvalidTimezoneException::class);
+        $this->expectExceptionMessage('Invalid timezone "Invalid/Timezone" provided for DATE_SUBTRACT');
+
+        $dql = \sprintf("SELECT DATE_SUBTRACT(e.datetimetz1, '1 day', 'Invalid/Timezone') FROM %s e", ContainsDates::class);
+        $this->buildEntityManager()->createQuery($dql)->getSQL();
+    }
+}

tests/MartinGeorgiev/Doctrine/ORM/Query/AST/Functions/TestCase.php

@@ -113,7 +113,7 @@ protected function assertSqlFromDql(string $expectedSql, string $dql, string $me
         self::assertEquals($expectedSql, $query->getSQL(), $message);
     }
 
-    private function buildEntityManager(): EntityManager
+    protected function buildEntityManager(): EntityManager
     {
         return new EntityManager(DriverManager::getConnection(['driver' => 'pdo_sqlite', 'memory' => true], $this->configuration), $this->configuration);
     }