enable multiple types for enum

Author: arnedesmedt

src/Type/EnumType.php

@@ -20,13 +20,22 @@ class EnumType implements AnnotatedType
     use NullableType,
         HasAnnotations;
 
+    /**
+     * @var string[]
+     */
+    private const VALID_TYPES = [
+        JsonSchema::TYPE_STRING,
+        JsonSchema::TYPE_INT,
+        JsonSchema::TYPE_FLOAT,
+    ];
+
     /**
      * @var string|array
      */
     private $type = JsonSchema::TYPE_STRING;
 
     /**
-     * @var string[]
+     * @var string[]|int[]|float[]
      */
     private $entries;
 
@@ -54,4 +63,71 @@ public function asNullable(): Type
 
         return $cp;
     }
+
+    public function setType(string $type) : Type
+    {
+        $cp = clone $this;
+
+        $nullable = is_array($cp->type) && in_array(JsonSchema::TYPE_NULL, $cp->type);
+
+        $cp->type = $nullable ? [$type, JsonSchema::TYPE_NULL] : $type;
+        $cp->entries = $this->castEntries($type);
+
+        return $cp;
+    }
+
+    private function castEntries(string $type)
+    {
+        $castCallback = $this->castCallback($type);
+
+        return array_map($castCallback, $this->entries);
+    }
+
+    private function castCallback(string $type)
+    {
+        $this->typeIsValid($type);
+
+        switch ($type) {
+            case JsonSchema::TYPE_INT:
+                return static function ($entry) {
+                    if ($entry === null) {
+                        return $entry;
+                    }
+
+                    return intval($entry);
+                };
+            case JsonSchema::TYPE_STRING:
+                return static function ($entry) {
+                    if ($entry === null) {
+                        return $entry;
+                    }
+
+                    return (string) $entry;
+                };
+            case JsonSchema::TYPE_FLOAT:
+                return static function ($entry) {
+                    if ($entry === null) {
+                        return $entry;
+                    }
+
+                    return floatval($entry);
+                };
+        }
+    }
+
+    private function typeIsValid(string $type): bool
+    {
+        if (!in_array($type, self::VALID_TYPES)) {
+            throw new \InvalidArgumentException(
+                sprintf(
+                    'Invalid schema type %s',
+                    $type
+                )
+            );
+        }
+
+        return true;
+    }
+
+
 }

tests/Type/EnumTypeTest.php

@@ -11,6 +11,7 @@
 
 namespace EventEngineTest\JsonSchema\Type;
 
+use EventEngine\JsonSchema\JsonSchema;
 use EventEngine\JsonSchema\Type\EnumType;
 use EventEngineTest\JsonSchema\BasicTestCase;
 
@@ -47,4 +48,40 @@ public function it_creates_nullable_enum_type()
             $enumType->toArray()
         );
     }
+
+
+    /**
+     * @test
+     */
+    public function it_creates_adjustable_enum_type()
+    {
+        $enumType = (new EnumType('0', '1'))->setType(JsonSchema::TYPE_INT);
+
+        $this->assertEquals(
+            [
+                'type' => 'integer',
+                'enum' => [0, 1],
+            ],
+            $enumType->toArray()
+        );
+    }
+
+
+    /**
+     * @test
+     */
+    public function it_creates_adjustable_nullable_enum_type()
+    {
+        $enumType = (new EnumType('0', '1'))
+            ->asNullable()
+            ->setType(JsonSchema::TYPE_INT);
+
+        $this->assertEquals(
+            [
+                'type' => ['integer', 'null'],
+                'enum' => [0, 1, null],
+            ],
+            $enumType->toArray()
+        );
+    }
 }