Updated with download factory

Author: FlorianRappl

CHANGELOG.md

@@ -5,6 +5,7 @@ Released on Thursday, May 2 2019.
 - Reference latest AngleSharp
 - Added `InputFile` class as a standard `IFile` implementation
 - Added `AppendFile` extensions for `IHtmlInputElement`
+- Added ability for binary data restore on assets (#19)
 
 # 0.10.1
 

src/AngleSharp.Io.Tests/Network/CookieHandlingTests.cs

@@ -95,7 +95,7 @@ public async Task SettingCookieIsPreservedViaRedirect()
         }
 
         [Test]
-        public async Task SettingCookieIsNotPreservedViaRedirectToDifferentProtocol()
+        public async Task SettingCookieIsPreservedViaRedirectToDifferentProtocol()
         {
             if (Helper.IsNetworkAvailable())
             {
@@ -107,7 +107,9 @@ public async Task SettingCookieIsNotPreservedViaRedirectToDifferentProtocol()
                 var document = await context.OpenAsync(redirectUrl);
 
                 Assert.AreEqual(@"{
-  ""cookies"": {}
+  ""cookies"": {
+    ""test"": ""baz""
+  }
 }
 ".Replace(Environment.NewLine, "\n"), document.Body.TextContent);
             }

src/AngleSharp.Io.nuspec

@@ -14,7 +14,7 @@
     <copyright>Copyright 2016-2019, AngleSharp</copyright>
     <tags>html html5 css css3 dom requester http https io filesystem storage httpclient cache</tags>
     <dependencies>
-      <dependency id="AngleSharp" version="0.10.1" />
+      <dependency id="AngleSharp" version="0.12.0" />
     </dependencies>
   </metadata>
 </package>

src/AngleSharp.Io/AngleSharp.Io.csproj

@@ -9,7 +9,7 @@
   </PropertyGroup>
 
   <ItemGroup>
-    <PackageReference Include="AngleSharp" Version="0.10.1" />
+    <PackageReference Include="AngleSharp" Version="0.12.0" />
   </ItemGroup>
 
   <PropertyGroup Condition=" '$(OS)' == 'Windows_NT' ">

src/AngleSharp.Io/GeneralExtensions.cs

@@ -1,4 +1,4 @@
-namespace AngleSharp.Io
+namespace AngleSharp.Io
 {
     using System;
     using System.Threading.Tasks;

src/AngleSharp.Io/IoConfigurationExtensions.cs

@@ -1,23 +1,41 @@
-namespace AngleSharp
-{
+namespace AngleSharp
+{
+    using AngleSharp.Dom;
+    using AngleSharp.Io;
     using AngleSharp.Io.Network;
-    using System;
-    using System.Net.Http;
-
+    using System;
+    using System.Linq;
+    using System.Net.Http;
+
     /// <summary>
     /// Additional extensions for improved requesters.
     /// </summary>
     public static class IoConfigurationExtensions
     {
+        /// <summary>
+        /// Adds capability to start a download when following some link to the
+        /// configuration.
+        /// </summary>
+        /// <param name="configuration">The configuration to extend.</param>
+        /// <param name="download">
+        /// The callback to invoke when a download should be started. Returns true
+        /// to signal an interest in downloading the response, otherwise false.
+        /// </param>
+        /// <returns>The configuration.</returns>
+        public static IConfiguration WithDownload(this IConfiguration configuration, Func<MimeType, IResponse, Boolean> download)
+        {
+            var oldFactory = configuration.Services.OfType<IDocumentFactory>().FirstOrDefault();
+            var newFactory = new DownloadFactory(oldFactory, download);
+            return configuration.WithOnly<IDocumentFactory>(newFactory);
+        }
+
         /// <summary>
         /// Adds the requesters from the AngleSharp.Io package.
         /// </summary>
         /// <param name="configuration">The configuration to use.</param>
         /// <returns>The new configuration.</returns>
-        public static IConfiguration WithRequesters(this IConfiguration configuration)
-        {
-            return configuration.WithRequesters(new HttpClientHandler { UseCookies = false, AllowAutoRedirect = false });
-        }
+        public static IConfiguration WithRequesters(this IConfiguration configuration) =>
+            configuration.WithRequesters(new HttpClientHandler { UseCookies = false, AllowAutoRedirect = false });
 
         /// <summary>
         /// Adds the requesters from the AngleSharp.Io package.
@@ -36,7 +54,7 @@ public static IConfiguration WithRequesters(this IConfiguration configuration, H
                 new DataRequester(),
                 new FtpRequester(),
                 new FileRequester(),
-                new AboutRequester()
+                new AboutRequester(),
             });
         }
     }

src/AngleSharp.Io/Network/DownloadFactory.cs

@@ -0,0 +1,29 @@
+namespace AngleSharp.Io.Network
+{
+    using AngleSharp.Dom;
+    using System;
+    using System.Threading;
+    using System.Threading.Tasks;
+
+    sealed class DownloadFactory : DefaultDocumentFactory
+    {
+        private readonly IDocumentFactory _documentFactory;
+        private readonly Func<MimeType, IResponse, Boolean> _download;
+
+        public DownloadFactory(IDocumentFactory documentFactory, Func<MimeType, IResponse, Boolean> download)
+        {
+            _documentFactory = documentFactory;
+            _download = download;
+        }
+
+        protected override Task<IDocument> CreateDefaultAsync(IBrowsingContext context, CreateDocumentOptions options, CancellationToken cancellationToken)
+        {
+            if (_download(options.ContentType, options.Response))
+            {
+                return context.OpenNewAsync(options.Response.Address.Href, cancellationToken);
+            }
+
+            return _documentFactory.CreateAsync(context, options, cancellationToken);
+        }
+    }
+}