From 13497c7a4c18f0df8ffa95be27aacc6a19ec2cba Mon Sep 17 00:00:00 2001
From: mrceperka <mrceperka@gmail.com>
Date: Wed, 19 Oct 2022 11:23:45 +0200
Subject: [PATCH] backed enum support for i, i[], s, s[]

---
 src/SqlProcessor.php                          |  13 ++
 .../unit/SqlProcessorTest.backed_enum.phpt    | 121 ++++++++++++++++++
 2 files changed, 134 insertions(+)
 create mode 100644 tests/cases/unit/SqlProcessorTest.backed_enum.phpt

diff --git a/src/SqlProcessor.php b/src/SqlProcessor.php
index 925cd93..e9dc34b 100644
--- a/src/SqlProcessor.php
+++ b/src/SqlProcessor.php
@@ -316,6 +316,19 @@ public function processModifier(string $type, $value): string
 						case '_like_':
 							return $this->platform->formatStringLike((string) $value, 0);
 					}
+				} elseif ($value instanceof \BackedEnum) {
+					$valueType = gettype($value->value);
+					if ($type === 's') {
+						if ($valueType === 'string') {
+							return $this->platform->formatString($value->value);
+						} elseif ($valueType === 'integer') {
+							return (string) $value->value;
+						}
+					} elseif ($type === 'i') {
+						if ($valueType === 'integer') {
+							return (string) $value->value;
+						}
+					}
 				}
 
 				break;
diff --git a/tests/cases/unit/SqlProcessorTest.backed_enum.phpt b/tests/cases/unit/SqlProcessorTest.backed_enum.phpt
new file mode 100644
index 0000000..a8d6f1d
--- /dev/null
+++ b/tests/cases/unit/SqlProcessorTest.backed_enum.phpt
@@ -0,0 +1,121 @@
+<?php declare(strict_types = 1);
+
+/**
+ * @testCase
+ * @phpVersion >= 8.1
+ */
+
+namespace NextrasTests\Dbal;
+
+
+use Mockery;
+use Nextras\Dbal\Exception\InvalidArgumentException;
+use Nextras\Dbal\Platforms\IPlatform;
+use Nextras\Dbal\SqlProcessor;
+use Tester\Assert;
+
+
+require_once __DIR__ . '/../../bootstrap.php';
+
+
+enum DirectionEnum: string
+{
+	case LEFT = 'left';
+	case RIGHT = 'right';
+}
+
+
+enum BinaryEnum: int
+{
+	case ZERO = 0;
+	case ONE = 1;
+}
+
+
+class SqlProcessorBackedEnumTest extends TestCase
+{
+	/** @var IPlatform|Mockery\MockInterface */
+	private $platform;
+
+	/** @var SqlProcessor */
+	private $parser;
+
+
+	protected function setUp()
+	{
+		parent::setUp();
+		$this->platform = \Mockery::mock(IPlatform::class);
+		$this->parser = new SqlProcessor($this->platform);
+	}
+
+
+	public function testString()
+	{
+		$cases = DirectionEnum::cases();
+		foreach ($cases as $case) {
+			$this->platform->shouldReceive('formatString')->once()->with($case->value)->andReturn('hit');
+			Assert::same('hit', $this->parser->processModifier('s', $case));
+		}
+
+		$cases = BinaryEnum::cases();
+		foreach ($cases as $case) {
+			Assert::same((string) $case->value, $this->parser->processModifier('s', $case));
+		}
+
+	}
+
+
+	public function testStringArray()
+	{
+
+		$cases = DirectionEnum::cases();
+		$this->platform->shouldReceive('formatString')->times(count($cases))
+			->andReturnArg(0);
+		Assert::same('(left, right)', $this->parser->processModifier('s[]', $cases));
+
+		$cases = BinaryEnum::cases();
+		Assert::same('(0, 1)', $this->parser->processModifier('s[]', $cases));
+
+	}
+
+
+	public function testInt()
+	{
+		$cases = DirectionEnum::cases();
+		foreach ($cases as $case) {
+			Assert::exception(
+				function () use ($case) {
+					$this->parser->processModifier('i', $case);
+				},
+				InvalidArgumentException::class,
+				'Modifier %i expects value to be int, NextrasTests\Dbal\DirectionEnum given.');
+		}
+
+		$cases = BinaryEnum::cases();
+		foreach ($cases as $case) {
+			Assert::same((string) $case->value, $this->parser->processModifier('i', $case));
+		}
+
+	}
+
+
+	public function testIntArray()
+	{
+		$cases = DirectionEnum::cases();
+		Assert::exception(
+			function () use ($cases) {
+				$this->parser->processModifier('i[]', $cases);
+			},
+			InvalidArgumentException::class,
+			'Modifier %i expects value to be int, NextrasTests\Dbal\DirectionEnum given.'
+		);
+
+		$cases = BinaryEnum::cases();
+		Assert::same('(0, 1)', $this->parser->processModifier('i[]', $cases));
+
+	}
+}
+
+
+$test = new SqlProcessorBackedEnumTest();
+$test->run();