Merge pull request #94 from dragonfruitnetwork/request-stream-types

Request stream types

DragonFruit.Common.Data.Tests/Requests/StreamTests.cs

@@ -0,0 +1,22 @@
+// DragonFruit.Common Copyright 2021 DragonFruit Network
+// Licensed under the MIT License. Please refer to the LICENSE file at the root of this project for details
+
+using System.IO;
+using System.Threading.Tasks;
+using NUnit.Framework;
+
+namespace DragonFruit.Common.Data.Tests.Requests
+{
+    [TestFixture]
+    public class StreamTests : ApiTest
+    {
+        [Test]
+        public async Task TestStreamRequests()
+        {
+            var networkStream = await Client.PerformAsync<MemoryStream>("https://google.com");
+            var fileStream = await Client.PerformAsync<FileStream>("https://google.com");
+
+            Assert.AreEqual(networkStream.Length, fileStream.Length);
+        }
+    }
+}

DragonFruit.Common.Data/ApiClient.cs

@@ -3,6 +3,7 @@
 
 using System;
 using System.Globalization;
+using System.IO;
 using System.Linq;
 using System.Net.Http;
 using System.Threading;
@@ -265,6 +266,25 @@ protected virtual async Task<T> ValidateAndProcess<T>(HttpResponseMessage respon
             response.EnsureSuccessStatusCode();
 
             using var stream = await response.Content.ReadAsStreamAsync();
+
+            if (typeof(Stream).IsAssignableFrom(typeof(T)))
+            {
+                Stream result;
+
+                if (typeof(T) == typeof(Stream) && response.Content.Headers.ContentLength < 80000 || typeof(T) == typeof(MemoryStream))
+                {
+                    result = new MemoryStream();
+                }
+                else
+                {
+                    result = File.Create(Path.GetTempFileName(), 4096, FileOptions.Asynchronous | FileOptions.SequentialScan | FileOptions.DeleteOnClose);
+                }
+
+                await stream.CopyToAsync(result).ConfigureAwait(false);
+                result.Seek(0, SeekOrigin.Begin);
+                return result as T;
+            }
+
             return Serializer.Resolve<T>(DataDirection.In).Deserialize<T>(stream);
         }