From be532c56158362701a4d5704e995c743ff8b9804 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Fri, 21 Aug 2020 23:24:47 +0800
Subject: ...

---
 Timeline/Startup.cs                                |  1 +
 .../Swagger/ByteDataRequestOperationProcessor.cs   | 27 ++++++++++++++++++++++
 2 files changed, 28 insertions(+)
 create mode 100644 Timeline/Swagger/ByteDataRequestOperationProcessor.cs

diff --git a/Timeline/Startup.cs b/Timeline/Startup.cs
index 408057e5..86bdaf54 100644
--- a/Timeline/Startup.cs
+++ b/Timeline/Startup.cs
@@ -115,6 +115,7 @@ namespace Timeline
                     }));
                 document.OperationProcessors.Add(new AspNetCoreOperationSecurityScopeProcessor("JWT"));
                 document.OperationProcessors.Add(new DefaultDescriptionOperationProcessor());
+                document.OperationProcessors.Add(new ByteDataRequestOperationProcessor());
             });
 
             if (!disableFrontEnd)
diff --git a/Timeline/Swagger/ByteDataRequestOperationProcessor.cs b/Timeline/Swagger/ByteDataRequestOperationProcessor.cs
new file mode 100644
index 00000000..887831ac
--- /dev/null
+++ b/Timeline/Swagger/ByteDataRequestOperationProcessor.cs
@@ -0,0 +1,27 @@
+using NJsonSchema;
+using NSwag;
+using NSwag.Generation.Processors;
+using NSwag.Generation.Processors.Contexts;
+using System.Linq;
+using Timeline.Models;
+
+namespace Timeline.Swagger
+{
+    /// <summary>
+    /// Coerce ByteData body type into the right one.
+    /// </summary>
+    public class ByteDataRequestOperationProcessor : IOperationProcessor
+    {
+        /// <inheritdoc/>
+        public bool Process(OperationProcessorContext context)
+        {
+            var hasByteDataBody = context.MethodInfo.GetParameters().Where(p => p.ParameterType == typeof(ByteData)).Any();
+            if (hasByteDataBody)
+            {
+                var bodyParameter = context.OperationDescription.Operation.Parameters.Where(p => p.Kind == OpenApiParameterKind.Body).Single();
+                bodyParameter.Schema = JsonSchema.FromType<byte[]>();
+            }
+            return true;
+        }
+    }
+}
-- 
cgit v1.2.3