From 2993aebdc5ac58ae98dc50bc93265841118bf2a3 Mon Sep 17 00:00:00 2001 From: Christopher Davis Date: Tue, 18 Jun 2024 07:59:23 -0500 Subject: [PATCH] Use Content for `Model` on Response Too --- utils/rector/src/OpenApiTagValueNode.php | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/utils/rector/src/OpenApiTagValueNode.php b/utils/rector/src/OpenApiTagValueNode.php index d08af7d..be65919 100644 --- a/utils/rector/src/OpenApiTagValueNode.php +++ b/utils/rector/src/OpenApiTagValueNode.php @@ -94,8 +94,9 @@ 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, @@ -103,7 +104,7 @@ public function getValues(): array $item->kindKeyQuoted ); break; - } elseif (!$isRequestBody && 'ref' === $name && $isModel) { + } elseif (!$needsContent && 'ref' === $name && $isModel) { $newValues[$name] = new ArrayItemNode( new OpenApiTagValueNode($item->value), $name,