Add output file type, collect all files

2016-05-18 13:05:24 -04:00
parent 4343f08733
commit 77ae03534e
19 changed files with 288 additions and 48 deletions
@@ -2,7 +2,6 @@ using System;
 using System.Collections.Generic;
 using System.ComponentModel.DataAnnotations;
 using System.ComponentModel.DataAnnotations.Schema;
-using System.Data;
 using System.IO;
 using System.IO.Compression;
 using System.Linq;
@@ -55,12 +54,12 @@ namespace LeafWeb.Core.Entities
 		/// <summary>
 		/// Contains all output files in a zip
 		/// </summary>
-		public byte[] GetOutputFileZip()
+		public byte[] GetOutputFileZip(LeafOutputFileType? fileType)
 		{
 			using (var compressedFileStream = new MemoryStream())
 			using (var archive = new ZipArchive(compressedFileStream, ZipArchiveMode.Create, true))
 			{
-				foreach (var outputFile in OutputFiles)
+				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.Contents))