Change folder name

RageCoop.Client/Networking/DownloadManager.cs

@@ -0,0 +1,185 @@
+using System.IO;
+using System.Linq;
+using System.Collections.Generic;
+
+namespace RageCoop.Client
+{
+    internal static class DownloadManager
+    {
+        private static readonly List<DownloadFile> _downloadFiles = new List<DownloadFile>();
+        private static readonly Dictionary<int, FileStream> _streams = new Dictionary<int, FileStream>();
+        private static readonly List<int> _filesFinished = new List<int>();
+        public static bool DownloadComplete = false;
+
+        public static void AddFile(int id, string name, long length)
+        {
+            string downloadFolder = $"Scripts\\RageCoop\\Resources\\{Main.Settings.LastServerAddress.Replace(":", ".")}";
+
+            if (!Directory.Exists(downloadFolder))
+            {
+                Directory.CreateDirectory(downloadFolder);
+            }
+
+            if (FileAlreadyExists(downloadFolder, name, length))
+            {
+                // Send the server we are already done
+                Networking.SendDownloadFinish(id);
+
+                Cancel(id);
+                return;
+            }
+            
+            if (!Path.GetExtension(name).EndsWith(".zip"))
+            {
+                Cancel(id);
+
+                GTA.UI.Notification.Show($"The download of a file from the server was blocked! [{name}]", true);
+                Main.Logger.Error($"The download of a file from the server was blocked! [{name}]");
+                return;
+            }
+
+            lock (_downloadFiles)
+            {
+                _downloadFiles.Add(new DownloadFile()
+                {
+                    FileID = id,
+                    FileName = name,
+                    FileLength = length
+                });
+            }
+
+            lock (_streams)
+            {
+                _streams.Add(id, new FileStream($"{downloadFolder}\\{name}", FileMode.CreateNew, FileAccess.Write, FileShare.ReadWrite));
+            }
+        }
+
+        /// <summary>
+        /// Check if the file already exists and if the size correct otherwise delete this file
+        /// </summary>
+        /// <param name="folder"></param>
+        /// <param name="name"></param>
+        /// <param name="length"></param>
+        /// <returns></returns>
+        private static bool FileAlreadyExists(string folder, string name, long length)
+        {
+            string filePath = $"{folder}\\{name}";
+
+            if (File.Exists(filePath))
+            {
+                if (new FileInfo(filePath).Length == length)
+                {
+                    return true;
+                }
+
+                // Delete the file because the length is wrong (maybe the file was updated)
+                File.Delete(filePath);
+            }
+
+            return false;
+        }
+
+        public static void RenderProgress()
+        {
+            if (_downloadFiles.Count == 0)
+            {
+                return;
+            }
+            
+            lock (_downloadFiles) lock (_filesFinished)
+            {
+                new LemonUI.Elements.ScaledText(new System.Drawing.PointF(System.Windows.Forms.Screen.PrimaryScreen.Bounds.Width / 2, System.Windows.Forms.Screen.PrimaryScreen.Bounds.Height - 60), $"Downloading files {_filesFinished.Count()} / {_downloadFiles.Count() + _filesFinished.Count()}", 0.5f) { Alignment = GTA.UI.Alignment.Center }.Draw();
+            }
+        }
+
+        public static void Write(int id, byte[] chunk)
+        {
+            lock (_filesFinished)
+            {
+                if (_filesFinished.Contains(id))
+                {
+                    return;
+                }
+            }
+
+            lock (_streams)
+            {
+                FileStream fs = _streams.FirstOrDefault(x => x.Key == id).Value;
+                if (fs == null)
+                {
+                    Main.Logger.Error($"Stream for file {id} not found!");
+                    return;
+                }
+
+                fs.Write(chunk, 0, chunk.Length);
+
+                lock (_downloadFiles)
+                {
+                    DownloadFile file = _downloadFiles.FirstOrDefault(x => x.FileID == id);
+                    if (file == null)
+                    {
+                        Main.Logger.Error($"File {id} couldn't be found in the list!");
+                        return;
+                    }
+
+                    file.FileWritten += chunk.Length;
+
+                    if (file.FileWritten >= file.FileLength)
+                    {
+                        Cancel(id);
+                    }
+                }
+            }
+        }
+
+        public static void Cancel(int id)
+        {
+            lock (_streams) lock (_downloadFiles) lock (_filesFinished)
+            {
+                FileStream fs = _streams.ContainsKey(id) ? _streams[id] : null;
+                if (fs != null)
+                {
+                    fs.Close();
+                    fs.Dispose();
+                
+                    _streams.Remove(id);
+                }
+
+                if (_downloadFiles.Any(x => x.FileID == id))
+                {
+                    _downloadFiles.Remove(_downloadFiles.First(x => x.FileID == id));
+                }
+
+                _filesFinished.Add(id);
+            }
+        }
+
+        public static void Cleanup(bool everything)
+        {
+            lock (_streams) lock (_downloadFiles) lock (_filesFinished)
+            {
+                foreach (var stream in _streams)
+                {
+                    stream.Value.Close();
+                    stream.Value.Dispose();
+                }
+                _streams.Clear();
+                _downloadFiles.Clear();
+                _filesFinished.Clear();
+            }
+
+            if (everything)
+            {
+                DownloadComplete = false;
+            }
+        }
+    }
+
+    public class DownloadFile
+    {
+        public int FileID { get; set; } = 0;
+        public string FileName { get; set; } = string.Empty;
+        public long FileLength { get; set; } = 0;
+        public long FileWritten { get; set; } = 0;
+    }
+}