Use Content for Model on Response Too

wenbinye · Jun 18, 2024 · 2993aeb · 2993aeb
1 parent 3c1d0f2
commit 2993aeb
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/utils/rector/src/OpenApiTagValueNode.php b/utils/rector/src/OpenApiTagValueNode.php
@@ -94,16 +94,17 @@ public function getValues(): array
 
                     // request body wants `content` as the ref, so handle that first
                     $isModel = 'Nelmio\\ApiDocBundle\\Annotation\\Model' === $itemClass;
-                    $isRequestBody = $attributeClass === 'OpenApi\\Attributes\\RequestBody';
-                    if ($isRequestBody && 'content' === $name && $isModel) {
+                    $needsContent = $attributeClass === 'OpenApi\\Attributes\\RequestBody'
+                        || $attributeClass === 'OpenApi\\Attributes\\Response';
+                    if ($needsContent && 'content' === $name && $isModel) {
                         $newValues[$name] = new ArrayItemNode(
                             new OpenApiTagValueNode($item->value),
                             $name,
                             $item->kindValueQuoted,
                             $item->kindKeyQuoted
                         );
                         break;
-                    } elseif (!$isRequestBody && 'ref' === $name && $isModel) {
+                    } elseif (!$needsContent && 'ref' === $name && $isModel) {
                         $newValues[$name] = new ArrayItemNode(
                             new OpenApiTagValueNode($item->value),
                             $name,