From 048ea9da73bd2a82f04e06f40157069f71e839db Mon Sep 17 00:00:00 2001
From: Vadim Kravchuk <vadimkravchuk1994@gmail.com>
Date: Thu, 3 Aug 2023 23:55:18 +0300
Subject: [PATCH] Support Multiple Enums class-names in Schema::enum property
 (#1460)

---
 src/Processors/ExpandEnums.php                  | 14 +++++++++++++-
 .../Fixtures/PHP/Enums/TypeEnumStringBacked.php | 17 +++++++++++++++++
 tests/Fixtures/PHP/ReferencesEnum.php           | 17 +++++++++++++++++
 tests/Processors/ExpandEnumsTest.php            | 11 +++++++++++
 4 files changed, 58 insertions(+), 1 deletion(-)
 create mode 100644 tests/Fixtures/PHP/Enums/TypeEnumStringBacked.php

diff --git a/src/Processors/ExpandEnums.php b/src/Processors/ExpandEnums.php
index 6865fb85..4a7cd30d 100644
--- a/src/Processors/ExpandEnums.php
+++ b/src/Processors/ExpandEnums.php
@@ -79,7 +79,19 @@ protected function expandSchemaEnum(Analysis $analysis): void
             } else {
                 // might be an array of \UnitEnum::class, string, int, etc...
                 assert(is_array($schema->enum));
-                $cases = $schema->enum;
+
+                $cases = [];
+
+                // transform each Enum cases into UnitEnum
+                foreach ($schema->enum as $enum) {
+                    if (is_string($enum) && enum_exists($enum)) {
+                        foreach ($enum::cases() as $case) {
+                            $cases[] = $case;
+                        }
+                    } else {
+                        $cases[] = $enum;
+                    }
+                }
             }
 
             $enums = [];
diff --git a/tests/Fixtures/PHP/Enums/TypeEnumStringBacked.php b/tests/Fixtures/PHP/Enums/TypeEnumStringBacked.php
new file mode 100644
index 00000000..586bd2f1
--- /dev/null
+++ b/tests/Fixtures/PHP/Enums/TypeEnumStringBacked.php
@@ -0,0 +1,17 @@
+<?php declare(strict_types=1);
+
+/**
+ * @license Apache 2.0
+ */
+
+namespace OpenApi\Tests\Fixtures\PHP\Enums;
+
+use OpenApi\Attributes as OAT;
+
+#[OAT\Schema(type: 'string')]
+enum TypeEnumStringBacked: string
+{
+    case FEATURE = 'feature';
+    case BUG = 'bug';
+    case IMPROVEMENT = 'improvement';
+}
diff --git a/tests/Fixtures/PHP/ReferencesEnum.php b/tests/Fixtures/PHP/ReferencesEnum.php
index dc810233..2e8cf933 100644
--- a/tests/Fixtures/PHP/ReferencesEnum.php
+++ b/tests/Fixtures/PHP/ReferencesEnum.php
@@ -12,6 +12,7 @@
 use OpenApi\Tests\Fixtures\PHP\Enums\StatusEnumBacked;
 use OpenApi\Tests\Fixtures\PHP\Enums\StatusEnumIntegerBacked;
 use OpenApi\Tests\Fixtures\PHP\Enums\StatusEnumStringBacked;
+use OpenApi\Tests\Fixtures\PHP\Enums\TypeEnumStringBacked;
 
 #[OAT\Schema()]
 class ReferencesEnum
@@ -51,6 +52,22 @@ class ReferencesEnum
     #[OAT\Property(title: 'statusEnumStringBackedMixed', description: 'Status enum string backed mixed', type: 'string', enum: [StatusEnumStringBacked::DRAFT, StatusEnumStringBacked::ARCHIVED, 'other'], nullable: true)]
     public ?string $statusEnumStringBackedMixed;
 
+    #[OAT\Property(
+        title: 'multipleEnumsClassNames',
+        description: 'MultipleEnumsClassNames array',
+        type: 'string',
+        enum: [StatusEnumStringBacked::class, TypeEnumStringBacked::class],
+    )]
+    public string $multipleEnumsClassNames;
+
+    #[OAT\Property(
+        title: 'mixedMultipleEnumsClassNames',
+        description: 'MultipleEnumsClassNames array',
+        type: 'string',
+        enum: [StatusEnumStringBacked::class, TypeEnumStringBacked::class, 'OTHER'],
+    )]
+    public string $mixedMultipleEnumsClassNames;
+
     /** @var list<string> StatusEnumStringBacked array */
     #[OAT\Property(
         title: 'statusEnums',
diff --git a/tests/Processors/ExpandEnumsTest.php b/tests/Processors/ExpandEnumsTest.php
index c917e64d..1af5af41 100644
--- a/tests/Processors/ExpandEnumsTest.php
+++ b/tests/Processors/ExpandEnumsTest.php
@@ -14,6 +14,7 @@
 use OpenApi\Tests\Fixtures\PHP\Enums\StatusEnumBacked;
 use OpenApi\Tests\Fixtures\PHP\Enums\StatusEnumIntegerBacked;
 use OpenApi\Tests\Fixtures\PHP\Enums\StatusEnumStringBacked;
+use OpenApi\Tests\Fixtures\PHP\Enums\TypeEnumStringBacked;
 use OpenApi\Tests\OpenApiTestCase;
 
 class ExpandEnumsTest extends OpenApiTestCase
@@ -136,6 +137,16 @@ public function expandEnumClassStringFixtures(): iterable
                 'itemsStatusEnumStringBackedMixed',
                 $mapValues([StatusEnumStringBacked::DRAFT, StatusEnumStringBacked::ARCHIVED, 'other']),
             ],
+            'multipleEnumsClassNames' => [
+                ['PHP/ReferencesEnum.php'],
+                'multipleEnumsClassNames',
+                $mapValues(array_merge(StatusEnumStringBacked::cases(), TypeEnumStringBacked::cases())),
+            ],
+            'mixedMultipleEnumsClassNames' => [
+                ['PHP/ReferencesEnum.php'],
+                'mixedMultipleEnumsClassNames',
+                $mapValues(array_merge(StatusEnumStringBacked::cases(), TypeEnumStringBacked::cases(), ['OTHER'])),
+            ],
         ];
     }