Stream/file returns

shinyorg · Sep 15, 2024 · 7b640ad · 7b640ad
1 parent dd6b77a
commit 7b640ad
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/Shiny.Mediator.SourceGenerators/Http/OpenApiContractGenerator.cs b/src/Shiny.Mediator.SourceGenerators/Http/OpenApiContractGenerator.cs
@@ -131,7 +131,7 @@ string GetResponseType(OpenApiOperation op)
     {
         string? responseType = null;
         if (response.TryGetValue("application/json", out var responseContent))
-            responseType = GetSchemaType(responseContent.Schema);
+            responseType = this.GetSchemaType(responseContent.Schema);
 
         return responseType;
     }
@@ -164,6 +164,9 @@ string GetResponseType(OpenApiOperation op)
                     var listType = this.GetSchemaType(schema.Items);
                     return $"global::System.Collections.Generic.List<{listType}>";
 
+                case "file":
+                    return "global::System.IO.Stream";
+
                 case "object":
                     if (schema.AdditionalProperties == null)
                     {
@@ -249,7 +252,7 @@ string GenerateComplexType(KeyValuePair<string, OpenApiSchema> schema, Action<st
             if (schema.Value.AllOf.Count == 1)
             {
                 // add inheritance
-                var baseType = GetSchemaType(schema.Value.AllOf.Single());
+                var baseType = this.GetSchemaType(schema.Value.AllOf.Single());
                 sb.AppendLine($" : {baseType}");
 
                 output($"INHERITED: {className} ({baseType})");