Advertisement
Guest User

Untitled

a guest
Mar 25th, 2019
131
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C# 2.20 KB | None | 0 0
  1. /// <summary>
  2. /// Filter to enable handling file upload in swagger
  3. /// </summary>
  4. public class FormFileSwaggerFilter : IOperationFilter
  5. {
  6.     private const string formDataMimeType = "multipart/form-data";
  7.     private static readonly string[] formFilePropertyNames =
  8.         typeof(IFormFile).GetTypeInfo().DeclaredProperties.Select(p => p.Name).ToArray();
  9.  
  10.     public void Apply(Operation operation, OperationFilterContext context)
  11.     {
  12.         var parameters = operation.Parameters;
  13.         if (parameters == null || parameters.Count == 0) return;
  14.  
  15.         var formFileParameterNames = new List<string>();
  16.         var formFileSubParameterNames = new List<string>();
  17.  
  18.         foreach (var actionParameter in context.ApiDescription.ActionDescriptor.Parameters)
  19.         {
  20.             var properties =
  21.                 actionParameter.ParameterType.GetProperties()
  22.                     .Where(p => p.PropertyType == typeof(IFormFile))
  23.                     .Select(p => p.Name)
  24.                     .ToArray();
  25.  
  26.             if (properties.Length != 0)
  27.             {
  28.                 formFileParameterNames.AddRange(properties);
  29.                 formFileSubParameterNames.AddRange(properties);
  30.                 continue;
  31.             }
  32.  
  33.             if (actionParameter.ParameterType != typeof(IFormFile)) continue;
  34.             formFileParameterNames.Add(actionParameter.Name);
  35.         }
  36.  
  37.         if (!formFileParameterNames.Any()) return;
  38.  
  39.         var consumes = operation.Consumes;
  40.         consumes.Clear();
  41.         consumes.Add(formDataMimeType);
  42.  
  43.         foreach (var parameter in parameters.ToArray())
  44.         {
  45.             if (!(parameter is NonBodyParameter) || parameter.In != "formData") continue;
  46.  
  47.             if (formFileSubParameterNames.Any(p => parameter.Name.StartsWith(p + "."))
  48.                 || formFilePropertyNames.Contains(parameter.Name))
  49.                 parameters.Remove(parameter);
  50.         }
  51.  
  52.         foreach (var formFileParameter in formFileParameterNames)
  53.         {
  54.             parameters.Add(new NonBodyParameter()
  55.             {
  56.                 Name = formFileParameter,
  57.                 Type = "file",
  58.                 In = "formData"
  59.             });
  60.         }
  61.     }
  62. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement