From 6dd92e540dfc4064e54aa061c42ee95fd5867889 Mon Sep 17 00:00:00 2001
From: "waffle.lord" <waffle.lord@hotmail.com>
Date: Tue, 22 Aug 2023 10:21:52 -0400
Subject: [PATCH] add problem folder warning

---
 SPTInstaller/App.axaml                        |  2 ++
 .../Dialogs/ConfirmationDialog.axaml          | 28 +++++++++++++++++++
 .../Dialogs/ConfirmationDialog.axaml.cs       | 22 +++++++++++++++
 SPTInstaller/Helpers/FileHelper.cs            |  9 ++++++
 SPTInstaller/Program.cs                       |  3 +-
 SPTInstaller/SPTInstaller.csproj              |  1 +
 SPTInstaller/ViewModels/PreChecksViewModel.cs | 17 +++++++++--
 SPTInstaller/Views/MainWindow.axaml           |  7 +++--
 8 files changed, 84 insertions(+), 5 deletions(-)
 create mode 100644 SPTInstaller/CustomControls/Dialogs/ConfirmationDialog.axaml
 create mode 100644 SPTInstaller/CustomControls/Dialogs/ConfirmationDialog.axaml.cs

diff --git a/SPTInstaller/App.axaml b/SPTInstaller/App.axaml
index 94f1bf0..5a734ba 100644
--- a/SPTInstaller/App.axaml
+++ b/SPTInstaller/App.axaml
@@ -2,6 +2,7 @@
              xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
              xmlns:local="using:SPTInstaller"
              x:Class="SPTInstaller.App"
+             xmlns:dialogHostAvalonia="clr-namespace:DialogHostAvalonia;assembly=DialogHost.Avalonia"
              RequestedThemeVariant="Light">
     <Application.DataTemplates>
         <local:ViewLocator/>
@@ -9,6 +10,7 @@
 
     <Application.Styles>
         <FluentTheme />
+        <dialogHostAvalonia:DialogHostStyles />
     </Application.Styles>
 
 	<Application.Resources>
diff --git a/SPTInstaller/CustomControls/Dialogs/ConfirmationDialog.axaml b/SPTInstaller/CustomControls/Dialogs/ConfirmationDialog.axaml
new file mode 100644
index 0000000..572110f
--- /dev/null
+++ b/SPTInstaller/CustomControls/Dialogs/ConfirmationDialog.axaml
@@ -0,0 +1,28 @@
+<UserControl xmlns="https://github.com/avaloniaui"
+             xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+             xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
+             xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
+             xmlns:dialogHost="clr-namespace:DialogHostAvalonia;assembly=DialogHost.Avalonia"
+             mc:Ignorable="d" d:DesignWidth="400" d:DesignHeight="300"
+             x:Class="SPTInstaller.CustomControls.Dialogs.ConfirmationDialog"
+             MinWidth="300" MinHeight="100" 
+             MaxWidth="600" MaxHeight="300">
+    <Grid RowDefinitions="10,AUTO,*,AUTO,10" ColumnDefinitions="10,*,AUTO,10,AUTO,10"
+          Background="{StaticResource AKI_Background_Light}">
+        <TextBlock Grid.Row="1" Grid.Column="1" Grid.ColumnSpan="4" 
+                   Text="{Binding Message, RelativeSource={RelativeSource AncestorType=UserControl}}"
+                   TextWrapping="Wrap"
+                   />
+        <Button Content="No" Grid.Row="3" Grid.Column="2"
+                Width="50" VerticalContentAlignment="Center" HorizontalContentAlignment="Center"
+                Command="{Binding RelativeSource={RelativeSource FindAncestor, AncestorType=dialogHost:DialogHost}, Path=CloseDialogCommand}"
+                CommandParameter="False"
+                Classes="yellow"
+                />
+        <Button Content="Yes" Grid.Row="3" Grid.Column="4"
+                Width="50" VerticalContentAlignment="Center" HorizontalContentAlignment="Center"
+                Command="{Binding RelativeSource={RelativeSource FindAncestor, AncestorType=dialogHost:DialogHost}, Path=CloseDialogCommand}"
+                CommandParameter="True"
+                />
+    </Grid>
+</UserControl>
diff --git a/SPTInstaller/CustomControls/Dialogs/ConfirmationDialog.axaml.cs b/SPTInstaller/CustomControls/Dialogs/ConfirmationDialog.axaml.cs
new file mode 100644
index 0000000..5367762
--- /dev/null
+++ b/SPTInstaller/CustomControls/Dialogs/ConfirmationDialog.axaml.cs
@@ -0,0 +1,22 @@
+using Avalonia;
+using Avalonia.Controls;
+
+namespace SPTInstaller.CustomControls.Dialogs;
+public partial class ConfirmationDialog : UserControl
+{
+    public ConfirmationDialog(string message)
+    {
+        InitializeComponent();
+
+        Message = message;
+    }
+
+    public string Message
+    {
+        get => GetValue(MessageProperty);
+        set => SetValue(MessageProperty, value);
+    }
+
+    public static readonly StyledProperty<string> MessageProperty =
+        AvaloniaProperty.Register<ConfirmationDialog, string>(nameof(Message));
+}
diff --git a/SPTInstaller/Helpers/FileHelper.cs b/SPTInstaller/Helpers/FileHelper.cs
index e240fee..27319b7 100644
--- a/SPTInstaller/Helpers/FileHelper.cs
+++ b/SPTInstaller/Helpers/FileHelper.cs
@@ -111,4 +111,13 @@ public static class FileHelper
         }
     }
 
+    public static bool CheckPathForProblemLocations(string path)
+    {
+        if (path.EndsWith("Desktop")) return false;
+
+        var problemNames = new string[] {"OneDrive", "NextCloud", "DropBox", "Google" };
+
+        return problemNames.Where(x => path.Contains(x)).Count() > 0;
+    }
+
 }
\ No newline at end of file
diff --git a/SPTInstaller/Program.cs b/SPTInstaller/Program.cs
index ccc1b17..d2635e7 100644
--- a/SPTInstaller/Program.cs
+++ b/SPTInstaller/Program.cs
@@ -33,8 +33,9 @@ internal class Program
         ServiceHelper.Register<InternalData>();
         ServiceHelper.Register<PreCheckBase, NetFramework472PreCheck>();
         ServiceHelper.Register<PreCheckBase, NetCore6PreCheck>();
-        ServiceHelper.Register<PreCheckBase, FreeSpacePreCheck>();
+
 #if !TEST
+        ServiceHelper.Register<PreCheckBase, FreeSpacePreCheck>();
         var logPath = Path.Join(Environment.CurrentDirectory, "spt-aki-installer_.log");
 
         Log.Logger = new LoggerConfiguration()
diff --git a/SPTInstaller/SPTInstaller.csproj b/SPTInstaller/SPTInstaller.csproj
index b425c08..a2fb9e9 100644
--- a/SPTInstaller/SPTInstaller.csproj
+++ b/SPTInstaller/SPTInstaller.csproj
@@ -35,6 +35,7 @@
     <PackageReference Include="Avalonia.Diagnostics" Version="11.0.4" />
     <PackageReference Include="Avalonia.ReactiveUI" Version="11.0.4" />
     <PackageReference Include="Avalonia.Themes.Fluent" Version="11.0.4" />
+    <PackageReference Include="DialogHost.Avalonia" Version="0.7.6" />
     <PackageReference Include="FubarCoder.RestSharp.Portable.HttpClient" Version="4.0.8" />
     <PackageReference Include="MegaApiClient" Version="1.10.3" />
     <PackageReference Include="Newtonsoft.Json" Version="13.0.3" />
diff --git a/SPTInstaller/ViewModels/PreChecksViewModel.cs b/SPTInstaller/ViewModels/PreChecksViewModel.cs
index a4a25b9..93eea80 100644
--- a/SPTInstaller/ViewModels/PreChecksViewModel.cs
+++ b/SPTInstaller/ViewModels/PreChecksViewModel.cs
@@ -1,11 +1,12 @@
 using System.Collections.ObjectModel;
-using System.Diagnostics.Metrics;
 using System.Reflection;
 using System.Threading.Tasks;
 using System.Windows.Input;
+using DialogHostAvalonia;
 using ReactiveUI;
 using Serilog;
 using SPTInstaller.Controllers;
+using SPTInstaller.CustomControls.Dialogs;
 using SPTInstaller.Helpers;
 using SPTInstaller.Models;
 
@@ -58,18 +59,30 @@ public class PreChecksViewModel : ViewModelBase
 
         data.OriginalGamePath = PreCheckHelper.DetectOriginalGamePath();
 
+#if !TEST
         if (data.OriginalGamePath == null)
         {
             NavigateTo(new MessageViewModel(HostScreen, Result.FromError("Could not find EFT install.\n\nDo you own and have the game installed?")));
         }
+#endif
 
         data.TargetInstallPath = Environment.CurrentDirectory;
         InstallPath = data.TargetInstallPath;
 
         Log.Information($"Install Path: {FileHelper.GetRedactedPath(InstallPath)}");
 
-        StartInstallCommand = ReactiveCommand.Create(() =>
+        StartInstallCommand = ReactiveCommand.Create(async () =>
         {
+            if(FileHelper.CheckPathForProblemLocations(InstallPath))
+            {
+                var confirmation = await DialogHost.Show(new ConfirmationDialog($"We suspect you may be installing to your desktop or a cloud synced folder.\nYou might want to considering installing somewhere else to avoid issues.\n\nAre you sure you want to install to this path?\n{InstallPath}"));
+
+                if (confirmation == null || !bool.TryParse(confirmation.ToString(), out var confirm) || !confirm)
+                {
+                    return;
+                }
+            }
+
             UpdateInfo.ShowCard = false;
             NavigateTo(new InstallViewModel(HostScreen));
         });
diff --git a/SPTInstaller/Views/MainWindow.axaml b/SPTInstaller/Views/MainWindow.axaml
index 963c638..9a82919 100644
--- a/SPTInstaller/Views/MainWindow.axaml
+++ b/SPTInstaller/Views/MainWindow.axaml
@@ -5,6 +5,7 @@
         xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
         xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
 		xmlns:cc="using:SPTInstaller.CustomControls"
+        xmlns:dialogHost="clr-namespace:DialogHostAvalonia;assembly=DialogHost.Avalonia"
         mc:Ignorable="d" d:DesignWidth="800" d:DesignHeight="450"
         x:Class="SPTInstaller.Views.MainWindow"
 		Icon="/Assets/icon.ico"
@@ -32,7 +33,9 @@
 			         XButtonCommand="{Binding CloseCommand}"
 			         MinButtonCommand="{Binding MinimizeCommand}"
 					 />
-        
-		<rxui:RoutedViewHost Router="{Binding Router}" Grid.Row="1"/>
+
+        <dialogHost:DialogHost  Grid.Row="1" Background="Transparent">
+            <rxui:RoutedViewHost Router="{Binding Router}"/>
+        </dialogHost:DialogHost>
 	</Grid>
 </Window>