Before download check available diskspace

Author: sweiland-openrails

Source/Menu/DownloadContentForm.cs

@@ -90,6 +90,25 @@ private void DownloadContentButton_Click(object sender, EventArgs e)
 
             // various checks for the directory where the route is installed
 
+            DriveInfo dInfo = new DriveInfo(installPathRoute);
+
+            long size = Routes[RouteName].InstallSize + Routes[RouteName].DownloadSize;
+
+            if (size > (dInfo.AvailableFreeSpace * 1.1))
+            {
+                message = Catalog.GetStringFmt("Not enough diskspace on drive {0} ({1}), available {2} kB, needed {3} kB, still continue?", 
+                    dInfo.Name, 
+                    dInfo.VolumeLabel,
+                    (dInfo.AvailableFreeSpace / 1024).ToString("N0"), 
+                    (size / 1024).ToString("N0"));
+
+                if (MessageBox.Show(message, Catalog.GetString("Attention"), MessageBoxButtons.OKCancel, MessageBoxIcon.Warning) != DialogResult.OK)
+                {
+                    // cancelled
+                    return;
+                }
+            }
+
             message = Catalog.GetStringFmt("Route to be installed in \"{0}\", are you sure?", installPathRoute);
 
             if (MessageBox.Show(message, Catalog.GetString("Attention"), MessageBoxButtons.OKCancel, MessageBoxIcon.Warning) != DialogResult.OK)

Source/ORTS.Settings/RouteSettings.cs

@@ -33,10 +33,15 @@ public class Route
             public string DateInstalled { get; set; }
             public string Url { get; set; }
 
-            public Route(string dateInstalled, string url)
+            public long DownloadSize;
+            public long InstallSize;
+
+            public Route(string dateInstalled, string url, long downloadSize, long installSize)
             { 
                 DateInstalled = dateInstalled;
                 Url = url;
+                DownloadSize = downloadSize;
+                InstallSize = installSize;
             }
         }
 
@@ -109,12 +114,14 @@ public void LoadContentAndInstalled()
                     {
                         string routeName = result["name"].ToString();
                         string url = result["url"].ToString();
+                        long downloadSize = convertResultToLong(result, "downloadSize");
+                        long installSize = convertResultToLong(result, "installSize");
 
                         if (url.EndsWith(".git") || url.EndsWith(".zip"))
                         {
                             if (!Routes.ContainsKey(routeName))
                             {
-                                Routes.Add(routeName, new RouteSettings.Route("", url));
+                                Routes.Add(routeName, new RouteSettings.Route("", url, downloadSize, installSize));
                             }
                         }
                     }
@@ -130,6 +137,18 @@ public void LoadContentAndInstalled()
             return;
         }
 
+        long convertResultToLong(JToken result, string fieldName)
+        {
+            if (result[fieldName] != null)
+            {
+                return (long)Convert.ToDouble(result[fieldName].ToString());
+            }
+            else
+            {
+                return 0;
+            }
+        }
+
         private void directoryDelete(string directoryName)
         {
             if (Directory.Exists(directoryName))