From 6c2f5457436abed1e0b4b231d5f50eb3984734c1 Mon Sep 17 00:00:00 2001
From: "waffle.lord" <waffle.lord@hotmail.com>
Date: Fri, 20 Oct 2023 22:26:13 -0400
Subject: [PATCH] add a try-catch to dir size calculation, other stuff

---
 .idea/.idea.SPTInstaller/.idea/avalonia.xml   |  1 +
 SPTInstaller/Helpers/DirectorySizeHelper.cs   | 37 +++++++++++++++++--
 SPTInstaller/Helpers/DownloadCacheHelper.cs   |  5 +++
 .../PreChecks/FreeSpacePreCheck.cs            |  6 +++
 SPTInstaller/SPTInstaller.csproj              |  4 +-
 5 files changed, 47 insertions(+), 6 deletions(-)

diff --git a/.idea/.idea.SPTInstaller/.idea/avalonia.xml b/.idea/.idea.SPTInstaller/.idea/avalonia.xml
index 400e2f0..613e7e9 100644
--- a/.idea/.idea.SPTInstaller/.idea/avalonia.xml
+++ b/.idea/.idea.SPTInstaller/.idea/avalonia.xml
@@ -3,6 +3,7 @@
   <component name="AvaloniaProject">
     <option name="projectPerEditor">
       <map>
+        <entry key="SPTInstaller/App.axaml" value="SPTInstaller/SPTInstaller.csproj" />
         <entry key="SPTInstaller/CustomControls/DetailedPreCheckItem.axaml" value="SPTInstaller/SPTInstaller.csproj" />
         <entry key="SPTInstaller/CustomControls/PreCheckItem.axaml" value="SPTInstaller/SPTInstaller.csproj" />
         <entry key="SPTInstaller/CustomControls/SPTInstallButton.axaml" value="SPTInstaller/SPTInstaller.csproj" />
diff --git a/SPTInstaller/Helpers/DirectorySizeHelper.cs b/SPTInstaller/Helpers/DirectorySizeHelper.cs
index d87b6c4..52acdbd 100644
--- a/SPTInstaller/Helpers/DirectorySizeHelper.cs
+++ b/SPTInstaller/Helpers/DirectorySizeHelper.cs
@@ -1,4 +1,5 @@
 using System.Linq;
+using Serilog;
 
 namespace SPTInstaller.Helpers;
 
@@ -8,11 +9,23 @@ public static class DirectorySizeHelper
     // https://stackoverflow.com/a/14488941
     static readonly string[] SizeSuffixes =
                    { "bytes", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB" };
+
     public static string SizeSuffix(Int64 value, int decimalPlaces = 1)
     {
-        if (decimalPlaces < 0) { throw new ArgumentOutOfRangeException("decimalPlaces"); }
-        if (value < 0) { return "-" + SizeSuffix(-value, decimalPlaces); }
-        if (value == 0) { return string.Format("{0:n" + decimalPlaces + "} bytes", 0); }
+        if (decimalPlaces < 0)
+        {
+            throw new ArgumentOutOfRangeException("decimalPlaces");
+        }
+
+        if (value < 0)
+        {
+            return "-" + SizeSuffix(-value, decimalPlaces);
+        }
+
+        if (value == 0)
+        {
+            return string.Format("{0:n" + decimalPlaces + "} bytes", 0);
+        }
 
         // mag is 0 for bytes, 1 for KB, 2, for MB, etc.
         int mag = (int)Math.Log(value, 1024);
@@ -34,5 +47,21 @@ public static class DirectorySizeHelper
             SizeSuffixes[mag]);
     }
 
-    public static long GetSizeOfDirectory(DirectoryInfo sourceDir) => sourceDir.EnumerateFiles("*", SearchOption.AllDirectories).Sum(fi => fi.Length);
+    /// <summary>
+    /// Gets the size of a directory in bytes
+    /// </summary>
+    /// <param name="sourceDir">The directory to get the size of</param>
+    /// <returns>the size of the <paramref name="sourceDir"/> in bytes or -1 if an error occurred</returns>
+    public static long GetSizeOfDirectory(DirectoryInfo sourceDir)
+    {
+        try
+        {
+            return sourceDir.EnumerateFiles("*", SearchOption.AllDirectories).Sum(fi => fi.Length);
+        }
+        catch (Exception ex)
+        {
+            Log.Error(ex, "Something went wrong calculating dir size");
+            return -1;
+        }
+    }
 }
\ No newline at end of file
diff --git a/SPTInstaller/Helpers/DownloadCacheHelper.cs b/SPTInstaller/Helpers/DownloadCacheHelper.cs
index 819d719..29e343e 100644
--- a/SPTInstaller/Helpers/DownloadCacheHelper.cs
+++ b/SPTInstaller/Helpers/DownloadCacheHelper.cs
@@ -20,6 +20,11 @@ public static class DownloadCacheHelper
 
         var cacheSize = DirectorySizeHelper.GetSizeOfDirectory(cacheDir);
 
+        if (cacheSize == -1)
+        {
+            return "An error occurred while getting the cache size :(";
+        }
+
         if (cacheSize == 0)
             return "Empty";
 
diff --git a/SPTInstaller/Installer Tasks/PreChecks/FreeSpacePreCheck.cs b/SPTInstaller/Installer Tasks/PreChecks/FreeSpacePreCheck.cs
index db3b2c5..5047208 100644
--- a/SPTInstaller/Installer Tasks/PreChecks/FreeSpacePreCheck.cs	
+++ b/SPTInstaller/Installer Tasks/PreChecks/FreeSpacePreCheck.cs	
@@ -28,6 +28,12 @@ public class FreeSpacePreCheck : PreCheckBase
             var installTargetDirectoryInfo = new DirectoryInfo(_internalData.TargetInstallPath);
 
             var eftSourceDirSize = DirectorySizeHelper.GetSizeOfDirectory(eftSourceDirectoryInfo);
+
+            if (eftSourceDirSize == -1)
+            {
+                return PreCheckResult.FromError("An error occurred while getting the EFT source directory size");
+            }
+            
             var availableSize = DriveInfo.GetDrives().FirstOrDefault(d => d.Name.ToLower() == installTargetDirectoryInfo.Root.Name.ToLower())?.AvailableFreeSpace ?? 0;
 
             var availableSpaceMessage = $"Available Space: {DirectorySizeHelper.SizeSuffix(availableSize, 2)}";
diff --git a/SPTInstaller/SPTInstaller.csproj b/SPTInstaller/SPTInstaller.csproj
index 0fe4d30..cfc7ed3 100644
--- a/SPTInstaller/SPTInstaller.csproj
+++ b/SPTInstaller/SPTInstaller.csproj
@@ -9,8 +9,8 @@
     <PackageIcon>icon.ico</PackageIcon>
     <ApplicationIcon>Assets\icon.ico</ApplicationIcon>
     <Configurations>Debug;Release;TEST</Configurations>
-    <AssemblyVersion>2.17</AssemblyVersion>
-    <FileVersion>2.17</FileVersion>
+    <AssemblyVersion>2.18</AssemblyVersion>
+    <FileVersion>2.18</FileVersion>
     <Company>SPT-AKI</Company>
   </PropertyGroup>