convert Pilz.IO to C# & Add EmbeddedFilesContainer

2020-04-28 13:59:46 +02:00
parent b86e392ac2
commit 9a98cbce02
26 changed files with 1342 additions and 462 deletions
--- a/Pilz.IO/EmbeddedFilesContainer.cs
+++ b/Pilz.IO/EmbeddedFilesContainer.cs
@@ -0,0 +1,129 @@
+using Newtonsoft.Json;
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.IO.Compression;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Pilz.IO
+{
+    public class EmbeddedFilesContainer
+    {
+        [JsonProperty("CompressedFiles")]
+        private readonly Dictionary<string, byte[]> compressedFiles = new Dictionary<string, byte[]>();
+
+        [JsonIgnore]
+        public IEnumerable<string> AllFileNames
+        {
+            get => compressedFiles.Keys;
+        }
+
+        public Task<bool> AddFileAsync(string fileName, string filePath)
+        {
+            return Task.Run(() => AddFile(fileName, filePath));
+        }
+
+        public bool AddFile(string fileName, string filePath)
+        {
+            bool success;
+            FileStream fs = null;
+            MemoryStream compressed = null;
+
+#if !DEBUG
+            try
+            {
+#endif
+                fs = new FileStream(filePath, FileMode.Open, FileAccess.Read);
+                compressed = new MemoryStream();
+                using (var compressor = new DeflateStream(compressed, CompressionLevel.Optimal, true))
+                    fs.CopyTo(compressor);
+                success = true;
+#if !DEBUG
+            }
+            catch (Exception)
+            {
+                success = false;
+            }
+#endif
+
+            if (success)
+            {
+                var compressedBytes = compressed.ToArray();
+                if (compressedFiles.ContainsKey(fileName))
+                    compressedFiles[fileName] = compressedBytes;
+                else
+                    compressedFiles.Add(fileName, compressedBytes);
+            }
+
+            compressed?.Close();
+            fs?.Close();
+
+            return success;
+        }
+
+        public void RemoveFile(string fileName)
+        {
+            if (compressedFiles.ContainsKey(fileName))
+                compressedFiles.Remove(fileName);
+        }
+
+        public bool HasFile(string fileName)
+        {
+            return compressedFiles.ContainsKey(fileName);
+        }
+
+        public Task<Stream> GetStreamAsync(string fileName)
+        {
+            return Task.Run(() => GetStream(fileName));
+        }
+
+        public Stream GetStream(string fileName)
+        {
+            Stream decompressed = null;
+
+            if (compressedFiles.ContainsKey(fileName))
+            {
+                decompressed = new MemoryStream();
+                DecompressToStream(decompressed, compressedFiles[fileName]);
+            }
+
+            return decompressed;
+        }
+
+        public Task<string> GetLocalFilePathAsync(string fileName)
+        {
+            return Task.Run(() => GetLocalFilePath(fileName));
+        }
+
+        public string GetLocalFilePath(string fileName)
+        {
+            string filePath = string.Empty;
+
+            if (compressedFiles.ContainsKey(fileName))
+            {
+                filePath = Path.GetTempFileName();
+
+                if (Path.HasExtension(fileName))
+                    filePath = Path.ChangeExtension(filePath, Path.GetExtension(fileName));
+
+                var decompressed = new FileStream(filePath, FileMode.Create, FileAccess.ReadWrite);
+                DecompressToStream(decompressed, compressedFiles[fileName]);
+                decompressed.Flush();
+                decompressed.Close();
+            }
+
+            return filePath;
+        }
+
+        private void DecompressToStream(Stream decompressed, byte[] compressedData)
+        {
+            var compressed = new MemoryStream(compressedData);
+            var decompressor = new DeflateStream(compressed, CompressionMode.Decompress, true);
+            decompressor.CopyTo(decompressed);
+            decompressor.Close();
+            compressed.Close();
+        }
+    }
+}