Allow running gzip-ped script from URL

atifaziz · atifaziz · commit aea2f7b45acc · 2020-08-06T15:43:57.000+02:00
diff --git a/src/Dotnet.Script.Core/ScriptDownloader.cs b/src/Dotnet.Script.Core/ScriptDownloader.cs
@@ -1,7 +1,7 @@
 ﻿using System;
 using System.IO;
+using System.IO.Compression;
 using System.Net.Http;
-using System.Net.Mime;
 using System.Threading.Tasks;
 
 namespace Dotnet.Script.Core
@@ -10,7 +10,6 @@ public class ScriptDownloader
     {
         public async Task<string> Download(string uri)
         {
-            const string plainTextMediaType = "text/plain";
             using (HttpClient client = new HttpClient())
             {
                 using (HttpResponseMessage response = await client.GetAsync(uri))
@@ -19,14 +18,22 @@ public async Task<string> Download(string uri)
 
                     using (HttpContent content = response.Content)
                     {
-                        string mediaType = content.Headers.ContentType.MediaType;
-
-                        if (string.IsNullOrWhiteSpace(mediaType) || mediaType.Equals(plainTextMediaType, StringComparison.InvariantCultureIgnoreCase))
+                        var mediaType = content.Headers.ContentType.MediaType?.ToLowerInvariant().Trim();
+                        switch (mediaType)
                         {
-                            return await content.ReadAsStringAsync();
+                            case null:
+                            case "":
+                            case "text/plain":
+                                return await content.ReadAsStringAsync();
+                            case "application/gzip":
+                            case "application/x-gzip":
+                                using (var stream = await content.ReadAsStreamAsync())
+                                using (var gzip = new GZipStream(stream, CompressionMode.Decompress))
+                                using (var reader = new StreamReader(gzip))
+                                    return await reader.ReadToEndAsync();
+                            default:
+                                throw new NotSupportedException($"The media type '{mediaType}' is not supported when executing a script over http/https");
                         }
-
-                        throw new NotSupportedException($"The media type '{mediaType}' is not supported when executing a script over http/https");
                     }
                 }
             }

Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,7 @@`
`1`	`1`	`using System;`
`2`	`2`	`using System.IO;`
	`3`	`+using System.IO.Compression;`
`3`	`4`	`using System.Net.Http;`
`4`		`-using System.Net.Mime;`
`5`	`5`	`using System.Threading.Tasks;`
`6`	`6`
`7`	`7`	`namespace Dotnet.Script.Core`
`@@ -10,7 +10,6 @@ public class ScriptDownloader`
`10`	`10`	`{`
`11`	`11`	`public async Task<string> Download(string uri)`
`12`	`12`	`{`
`13`		`- const string plainTextMediaType = "text/plain";`
`14`	`13`	`using (HttpClient client = new HttpClient())`
`15`	`14`	`{`
`16`	`15`	`using (HttpResponseMessage response = await client.GetAsync(uri))`
`@@ -19,14 +18,22 @@ public async Task<string> Download(string uri)`
`19`	`18`
`20`	`19`	`using (HttpContent content = response.Content)`
`21`	`20`	`{`
`22`		`- string mediaType = content.Headers.ContentType.MediaType;`
`23`		`-`
`24`		`- if (string.IsNullOrWhiteSpace(mediaType) \|\| mediaType.Equals(plainTextMediaType, StringComparison.InvariantCultureIgnoreCase))`
	`21`	`+ var mediaType = content.Headers.ContentType.MediaType?.ToLowerInvariant().Trim();`
	`22`	`+ switch (mediaType)`
`25`	`23`	`{`
`26`		`- return await content.ReadAsStringAsync();`
	`24`	`+ case null:`
	`25`	`+ case "":`
	`26`	`+ case "text/plain":`
	`27`	`+ return await content.ReadAsStringAsync();`
	`28`	`+ case "application/gzip":`
	`29`	`+ case "application/x-gzip":`
	`30`	`+ using (var stream = await content.ReadAsStreamAsync())`
	`31`	`+ using (var gzip = new GZipStream(stream, CompressionMode.Decompress))`
	`32`	`+ using (var reader = new StreamReader(gzip))`
	`33`	`+ return await reader.ReadToEndAsync();`
	`34`	`+ default:`
	`35`	`+ throw new NotSupportedException($"The media type '{mediaType}' is not supported when executing a script over http/https");`
`27`	`36`	`}`
`28`		`-`
`29`		`- throw new NotSupportedException($"The media type '{mediaType}' is not supported when executing a script over http/https");`
`30`	`37`	`}`
`31`	`38`	`}`
`32`	`39`	`}`