Update zip archive functionality

add download buttons for multiple input/output

Core/Entities/LeafInput.cs

@@ -5,6 +5,7 @@ using System.ComponentModel.DataAnnotations.Schema;
 using System.IO;
 using System.IO.Compression;
 using System.Linq;
+using LeafWeb.Core.Utility;
 
 namespace LeafWeb.Core.Entities
 {
@@ -143,41 +144,87 @@ namespace LeafWeb.Core.Entities
 		/// </summary>
 		public byte[] GetOutputFileZip(LeafOutputFileType? fileType)
         {
-			using (var compressedFileStream = new MemoryStream())
-			{
-				using (var archive = new ZipArchive(compressedFileStream, ZipArchiveMode.Create, true))
-				{
-					foreach (var outputFile in OutputFiles.Where(f => !fileType.HasValue || f.FileType == fileType.Value))
-					{
-						var entry = archive.CreateEntry(outputFile.Filename);
-						using (var originalFileStream = new MemoryStream(outputFile.FileContents.Contents))
-						using (var entryStream = entry.Open())
-							originalFileStream.CopyTo(entryStream);
-					}
-				}
-				return compressedFileStream.ToArray();
-			}
+            return NewMemoryZipArchive((archive, stream) => 
+                CompressOutputFiles(
+                    OutputFiles.Where(f => !fileType.HasValue || f.FileType == fileType.Value),
+                    archive));
 		}
 
         /// <summary>
         /// Contains all input files in a zip
         /// </summary>
-		public byte[] GetInputFileZip()
+        public byte[] GetInputFileZip() =>
+            NewMemoryZipArchive((archive, stream) => CompressInputFiles(InputFiles, archive));
+
+        /// <summary>
+        /// Contains all input files in a zip, ordered into directories
+        /// </summary>
+        public static byte[] GetInputFilesZip(IEnumerable<LeafInput> leafInputs) =>
+            NewMemoryZipArchive((archive, stream) =>
+            {
+                var inputs = leafInputs.ToList();
+
+                foreach (var leafInput in inputs)
+                {
+                    var directory = $"{leafInput.Added:yyyy-dd-MM--HH-mm}_{leafInput.Identifier.FilterValidFilename()}";
+                    CompressInputFiles(leafInput.InputFiles, archive, directory);
+                }
+            });
+
+        /// <summary>
+        /// Contains all input files in a zip, ordered into directories
+        /// </summary>
+        public static byte[] GetOutputFilesZip_ToUser(IEnumerable<LeafInput> leafInputs) =>
+            NewMemoryZipArchive((archive, stream) =>
+            {
+                var inputs = leafInputs.ToList();
+
+                foreach (var leafInput in inputs)
+                {
+                    if (stream.Length > 2560000L)
+                        continue;
+                    var directory = $"{leafInput.Added:yyyy-dd-MM--HH-mm}_{leafInput.Identifier.FilterValidFilename()}";
+                    CompressOutputFiles(leafInput.OutputFiles.Where(f => f.FileType == LeafOutputFileType.ToUser), archive, directory);
+                }
+            });
+
+        private static byte[] NewMemoryZipArchive(Action<ZipArchive, MemoryStream> addFiles)
         {
             using (var compressedFileStream = new MemoryStream())
-			{
             using (var archive = new ZipArchive(compressedFileStream, ZipArchiveMode.Create, true))
             {
-					foreach (var inputFile in InputFiles)
+                addFiles(archive, compressedFileStream);
+                return compressedFileStream.ToArray();
+            }
+        }
+
+        private static long CompressInputFiles(IEnumerable<LeafInputFile> inputFiles, ZipArchive archive, string prefix = "")
         {
-						var entry = archive.CreateEntry(inputFile.Filename);
+            var contentsLength = 0L;
+            foreach (var inputFile in inputFiles)
+            {
+                contentsLength += inputFile.Contents.LongLength;
+                var entry = archive.CreateEntry(Path.Combine(prefix, inputFile.Filename));
                 using (var originalFileStream = new MemoryStream(inputFile.Contents))
                 using (var entryStream = entry.Open())
                     originalFileStream.CopyTo(entryStream);
             }
+
+            return contentsLength;
         }
-				return compressedFileStream.ToArray();
+
+        private static long CompressOutputFiles(IEnumerable<LeafOutputFile> outputFiles, ZipArchive archive, string prefix = "")
+        {
+            var contentsLength = 0L;
+            foreach (var outputFile in outputFiles)
+            {
+                contentsLength += outputFile.FileContents.Contents.LongLength;
+                var entry = archive.CreateEntry(Path.Combine(prefix, outputFile.Filename));
+                using (var originalFileStream = new MemoryStream(outputFile.FileContents.Contents))
+                using (var entryStream = entry.Open())
+                    originalFileStream.CopyTo(entryStream);
             }
+            return contentsLength;
         }
 
         public int GetOutputFileSizeSum()

WebCms/Controllers/QueueController.cs

@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 using System.Linq;
 using System.Web;
 using System.Web.Caching;
@@ -126,6 +127,16 @@ namespace LeafWeb.WebCms.Controllers
 			return GetOutputZip(id, LeafOutputFileType.CleanedInput);
 		}
 
+        public ActionResult DownloadResultsInputZip(LeafDataQuery q)
+        {
+            return GetResults(q, LeafInput.GetInputFilesZip, $"LeafWeb_{DateTime.Now:yyyy-dd-MM--HH-mm-ss}_Input.zip");
+        }
+
+        public ActionResult DownloadResultsOutputZip(LeafDataQuery q)
+        {
+            return GetResults(q, LeafInput.GetOutputFilesZip_ToUser, $"LeafWeb_{DateTime.Now:yyyy-dd-MM--HH-mm-ss}_Output.zip");
+        }
+
         private ActionResult GetOutputZip(int id, LeafOutputFileType type)
 	    {
 			var leafInput = DataService.GetLeafInput(id);
@@ -154,6 +165,23 @@ namespace LeafWeb.WebCms.Controllers
 			return new FileContentResult(zip, "application/zip") { FileDownloadName = filename };
 		}
 
+        private ActionResult GetResults(LeafDataQuery query, Func<IEnumerable<LeafInput>, byte[]> getZip, string filename)
+	    {
+            var resultItems =
+                DataService.GetLeafInputsOrdered();
+
+            resultItems = QueryFilter.Search(resultItems, query);
+
+            if (resultItems == null)
+                return View("DownloadNotFound");
+
+            var zip = getZip(resultItems);
+
+			//var filename = $"LeafWeb_{DateTime.Now:yyyy-dd-MM--HH-mm-ss}_Input.zip";
+
+			return new FileContentResult(zip, "application/zip") { FileDownloadName = filename };
+		}
+
         [ActionLog]
 		public ActionResult Delete(int id)
 		{

WebCms/Utility/NameValueCollectionUtil.cs

@@ -3,6 +3,7 @@ using System.Collections.Generic;
 using System.Collections.Specialized;
 using System.Linq;
 using System.Web;
+using System.Web.Routing;
 
 namespace LeafWeb.WebCms.Utility
 {
@@ -27,5 +28,13 @@ namespace LeafWeb.WebCms.Utility
 
             return nvs;
         }
+
+        public static RouteValueDictionary ToRouteValueDictionary(this NameValueCollection nvc)
+        {
+            var routeValues = new RouteValueDictionary();
+            foreach (var key in nvc.AllKeys)
+                routeValues.Add(key, nvc[key]);
+            return routeValues;
+        }
     }
 }

WebCms/Views/Queue/Index.cshtml

@@ -94,6 +94,8 @@
 
 @if (Model.Items.Any())
 {
+    @DownloadResultsInput()
+    @DownloadResultsOutput_ToUser()
     <div id="queue" class="table-responsive mt-3">
         @grid.Table(columns:
             grid.Columns(
@@ -155,14 +157,14 @@ else
 
 @helper DetailsLink(dynamic item)
 {
-    var cssClass= CssClassUtil.CreateCssClassDataDictionary("dropdown-item");
+    var cssClass = CssClassUtil.CreateCssClassDataDictionary("dropdown-item");
 
     @Html.Partial("DisplayTemplates/_DetailsLink", (int)item.Id, cssClass)
 }
 
 @helper ChartLink(dynamic item)
 {
-    var cssClass= CssClassUtil.CreateCssClassDataDictionary("dropdown-item");
+    var cssClass = CssClassUtil.CreateCssClassDataDictionary("dropdown-item");
 
     if (!item.HasLeafChart)
     {
@@ -257,3 +259,18 @@ else
     if (disabled)
     {<text>disabled</text>}
 }
+
+
+@helper DownloadResultsInput()
+{
+    <a href="@Url.Action("DownloadResultsInputZip", "Queue", Model.Q.GetNameValueCollection().ToRouteValueDictionary())">
+        <span class="fa fa-download"></span> Input
+    </a>
+}
+
+@helper DownloadResultsOutput_ToUser()
+{
+    <a href="@Url.Action("DownloadResultsOutputZip", "Queue", Model.Q.GetNameValueCollection().ToRouteValueDictionary())">
+        <span class="fa fa-download"></span> Output
+    </a>
+}