Merge pull request swagger-api#1217 from swagger-api/issue-1174

identify file type in api response schemas.
mwiehl · Nov 25, 2023 · a0a349a · a0a349a
2 parents 1f0254f + 01bc481
commit a0a349a
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/src/main/java/io/swagger/codegen/v3/generators/DefaultCodegenConfig.java b/src/main/java/io/swagger/codegen/v3/generators/DefaultCodegenConfig.java
@@ -83,6 +83,7 @@
 import java.nio.charset.StandardCharsets;
 import java.util.ArrayList;
 import java.util.Arrays;
+import java.util.Collection;
 import java.util.Collections;
 import java.util.Comparator;
 import java.util.HashMap;
@@ -2338,6 +2339,9 @@ public CodegenResponse fromResponse(String responseCode, ApiResponse response) {
                 } else {
                     codegenResponse.baseType = codegenProperty.baseType;
                 }
+                if (isFileTypeSchema(responseSchema)) {
+                    codegenResponse.getVendorExtensions().put(CodegenConstants.IS_FILE_EXT_NAME, Boolean.TRUE);
+                }
             }
             codegenResponse.dataType = codegenProperty.datatype;
 
@@ -4403,6 +4407,21 @@ protected void setParameterNullable(CodegenParameter parameter, CodegenProperty
         parameter.nullable = property.nullable;
     }
 
+    protected boolean isFileTypeSchema(Schema schema) {
+        final Schema fileTypeSchema;
+        if (StringUtils.isNotBlank(schema.get$ref())) {
+            fileTypeSchema = OpenAPIUtil.getSchemaFromRefSchema(schema, openAPI);
+        } else {
+            fileTypeSchema = schema;
+        }
+        if (fileTypeSchema.getProperties() != null) {
+            final Collection<Schema> propertySchemas = fileTypeSchema.getProperties().values();
+            return propertySchemas.stream().anyMatch(propertySchema -> "string".equalsIgnoreCase(propertySchema.getType())
+                && "binary".equalsIgnoreCase(propertySchema.getFormat()));
+        }
+        return false;
+    }
+
     @Override
     public boolean needsUnflattenedSpec() {
         return false;