Merge pull request 'feature/size-check' (#17) from Schrader/SPT-AKI-Installer:feature/size-check into master

Reviewed-on: CWX/SPT-AKI-Installer#17

README.md

@@ -6,6 +6,7 @@
 - Checks if .net 4.7.2 (or higher) is installed
 - Checks if .net 6 desktop runtime is installed
 - Checks if EFT is installed,
+- Checks if there is enough space before install,
 - Checks installer is not in OG game directory,
 - Checks install folder does not have game files already in it,
 - Checks if gameversion matches aki version, if so skip patcher process,

SPTInstaller/GlobalUsings.cs

@@ -0,0 +1,3 @@
+// Global using directives
+global using System;
+global using System.IO;

SPTInstaller/Helpers/DirectorySizeHelper.cs

@@ -0,0 +1,29 @@
+using System.Linq;
+using Serilog;
+
+namespace SPTInstaller.Helpers;
+
+public static class DirectorySizeHelper
+{
+    public static bool CheckAvailableSize(string eftSourceDirPath, string installTargetDirPath)
+    {
+        try
+        {
+            var eftSourceDirectoryInfo = new DirectoryInfo(eftSourceDirPath);
+            var installTargetDirectoryInfo = new DirectoryInfo(installTargetDirPath);
+                
+            var eftSourceDirSize = GetSizeOfDirectory(eftSourceDirectoryInfo);
+            var availableSize = DriveInfo.GetDrives().FirstOrDefault(d => d.Name == installTargetDirectoryInfo.Root.Name)?.AvailableFreeSpace ?? 0;
+
+            return eftSourceDirSize < availableSize;
+        }
+        catch (Exception ex)
+        {
+            Log.Error(ex, "Error while checking available size");
+
+            return false;
+        }
+    }
+        
+    private static long GetSizeOfDirectory(DirectoryInfo sourceDir) => sourceDir.EnumerateFiles("*", SearchOption.AllDirectories).Sum(fi => fi.Length);
+}

SPTInstaller/Installer Tasks/IntializationTask.cs

@@ -1,7 +1,6 @@
 using SPTInstaller.Aki.Helper;
 using SPTInstaller.Interfaces;
 using SPTInstaller.Models;
-using System.IO;
 using System.Threading.Tasks;
 
 namespace SPTInstaller.Installer_Tasks

SPTInstaller/Installer Tasks/PreChecks/FreeSpacePreCheck.cs

@@ -0,0 +1,25 @@
+using System.Threading.Tasks;
+using SPTInstaller.Helpers;
+using SPTInstaller.Models;
+
+namespace SPTInstaller.Installer_Tasks.PreChecks;
+
+public class FreeSpacePreCheck : PreCheckBase
+{
+    private readonly InternalData _internalData;
+
+    public FreeSpacePreCheck(InternalData internalData) : base("Free Space", true)
+    {
+        _internalData = internalData;
+    }
+
+    public override async Task<bool> CheckOperation()
+    {
+        if (_internalData.OriginalGamePath is null || _internalData.TargetInstallPath is null)
+        {
+            return false;
+        }
+
+        return DirectorySizeHelper.CheckAvailableSize(_internalData.OriginalGamePath, _internalData.TargetInstallPath);
+    }
+}

SPTInstaller/Models/InternalData.cs

@@ -8,12 +8,12 @@ namespace SPTInstaller.Models
         /// <summary>
         /// The folder to install SPT into
         /// </summary>
-        public string TargetInstallPath { get; set; }
+        public string? TargetInstallPath { get; set; }
 
         /// <summary>
         /// The orginal EFT game path
         /// </summary>
-        public string OriginalGamePath { get; set; }
+        public string? OriginalGamePath { get; set; }
 
         /// <summary>
         /// The original EFT game version

SPTInstaller/Program.cs

@@ -9,8 +9,6 @@ using SPTInstaller.Installer_Tasks;
 using SPTInstaller.Installer_Tasks.PreChecks;
 using SPTInstaller.Interfaces;
 using SPTInstaller.Models;
-using System;
-using System.IO;
 using System.Linq;
 using System.Reflection;
 
@@ -35,8 +33,9 @@ namespace SPTInstaller
             ServiceHelper.Register<InternalData>();
             ServiceHelper.Register<PreCheckBase, NetFramework472PreCheck>();
             ServiceHelper.Register<PreCheckBase, NetCore6PreCheck>();
+            ServiceHelper.Register<PreCheckBase, FreeSpacePreCheck>();
 #if !TEST
-            string logPath = Path.Join(Environment.CurrentDirectory, "spt-aki-installer_.log");
+            var logPath = Path.Join(Environment.CurrentDirectory, "spt-aki-installer_.log");
 
             Log.Logger = new LoggerConfiguration()
                          .MinimumLevel.Debug()

SPTInstaller/SPTInstaller.csproj

@@ -9,8 +9,8 @@
     <PackageIcon>icon.ico</PackageIcon>
     <ApplicationIcon>Assets\icon.ico</ApplicationIcon>
     <Configurations>Debug;Release;TEST</Configurations>
-    <AssemblyVersion>2.2</AssemblyVersion>
+    <AssemblyVersion>2.3</AssemblyVersion>
-    <FileVersion>2.2</FileVersion>
+    <FileVersion>2.3</FileVersion>
   </PropertyGroup>
 
   <ItemGroup>

SPTInstaller/ViewModels/PreChecksViewModel.cs

@@ -1,59 +1,54 @@
-using ReactiveUI;
+using System.Collections.ObjectModel;
+using System.Threading.Tasks;
+using System.Windows.Input;
+using ReactiveUI;
+using SPTInstaller.Aki.Helper;
 using SPTInstaller.Controllers;
 using SPTInstaller.Helpers;
 using SPTInstaller.Models;
-using System;
-using System.Collections.ObjectModel;
-using System.Threading.Tasks;
-using System.Windows.Input;
 
-namespace SPTInstaller.ViewModels
+namespace SPTInstaller.ViewModels;
+
+public class PreChecksViewModel : ViewModelBase
 {
-    public class PreChecksViewModel : ViewModelBase
+    private string _installPath;
+    private bool _allowInstall;
+
+    public ObservableCollection<PreCheckBase> PreChecks { get; set; } = new(ServiceHelper.GetAll<PreCheckBase>());
+    public ICommand StartInstallCommand { get; set; }
+    public string InstallPath
     {
-        private string _InstallPath;
+        get => _installPath;
-        public string InstallPath
+        set => this.RaiseAndSetIfChanged(ref _installPath, value);
-        {
-            get => _InstallPath;
-            set => this.RaiseAndSetIfChanged(ref _InstallPath, value);
-        }
-
-        ObservableCollection<PreCheckBase> PreChecks { get; set; } 
-            = new ObservableCollection<PreCheckBase>(ServiceHelper.GetAll<PreCheckBase>());
-
-        ICommand StartInstallCommand { get; set; }
-
-        public PreChecksViewModel(IScreen host) : base(host)
-        {
-            var data = ServiceHelper.Get<InternalData>();
-            var installer = ServiceHelper.Get<InstallController>();
-
-            if(data == null || installer == null)
-            {
-                NavigateTo(new MessageViewModel(HostScreen, Result.FromError("Failed to get required service for prechecks")));
-                return;
-            }
-
-            data.TargetInstallPath = Environment.CurrentDirectory;
-
-            InstallPath = data.TargetInstallPath;
-
-            StartInstallCommand = ReactiveCommand.Create(() =>
-            {
-                NavigateTo(new InstallViewModel(HostScreen));
-            });
-
-
-            Task.Run(async () =>
-            {
-                var result = await installer.RunPreChecks();
-
-                if(!result.Succeeded)
-                {
-                    //if a required precheck fails, abort to message view
-                    NavigateTo(new MessageViewModel(HostScreen ,result));
-                }
-            });
-        }
     }
-}
+    
+    public bool AllowInstall 
+    {
+        get => _allowInstall;
+        set => this.RaiseAndSetIfChanged(ref _allowInstall, value);
+    }
+
+    public PreChecksViewModel(IScreen host) : base(host)
+    {
+        var data = ServiceHelper.Get<InternalData?>();
+        var installer = ServiceHelper.Get<InstallController?>();
+
+        if (data == null || installer == null)
+        {
+            NavigateTo(new MessageViewModel(HostScreen, Result.FromError("Failed to get required service for prechecks")));
+            return;
+        }
+
+        data.OriginalGamePath = PreCheckHelper.DetectOriginalGamePath();
+        data.TargetInstallPath = Environment.CurrentDirectory;
+        InstallPath = data.TargetInstallPath;
+
+        StartInstallCommand = ReactiveCommand.Create(() => NavigateTo(new InstallViewModel(HostScreen)));
+
+        Task.Run(async () =>
+        {
+            var result = await installer.RunPreChecks();
+            AllowInstall = result.Succeeded;
+        });
+    }
+}

SPTInstaller/Views/PreChecksView.axaml

@@ -23,6 +23,7 @@
 				Margin="10"
 				FontSize="15" FontWeight="SemiBold"
 				Classes="yellow"
+				IsEnabled="{Binding AllowInstall}"
 				Command="{Binding StartInstallCommand}"
 				/>