Clean-up Choreo if previously installed

wpilibsuite · Dec 17, 2024 · 252b4f9 · 252b4f9
1 parent a62c1f1
commit 252b4f9
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/WPILibInstaller-Avalonia/ViewModels/InstallPageViewModel.cs b/WPILibInstaller-Avalonia/ViewModels/InstallPageViewModel.cs
@@ -164,6 +164,10 @@ private async Task InstallEverything(CancellationToken token)
                     ProgressTotal = 88;
                     TextTotal = "Creating Shortcuts";
                     await RunShortcutCreator(token);
+                    if (token.IsCancellationRequested) break;
+                    ProgressTotal = 99;
+                    TextTotal = "Cleaning Up";
+                    await RunChoreoCleanup();
                 } while (false);
             }
             catch (OperationCanceledException)
@@ -984,5 +988,26 @@ await Task.Run(() =>
                 }, token);
             }
         }
+
+        private async Task RunChoreoCleanup()
+        {
+            if (Directory.Exists(Path.Combine(configurationProvider.InstallDirectory, "choreo"))) {
+                Text = "Clean up Choreo Directory";
+                Progress = 25;
+                Directory.Delete(Path.Combine(configurationProvider.InstallDirectory, "choreo"), true);
+
+                Text = "Clean up Choreo Scripts";
+                Progress = 50;
+                File.Delete(Path.Combine(configurationProvider.InstallDirectory, "tools", "choreo.vbs"));
+                File.Delete(Path.Combine(configurationProvider.InstallDirectory, "tools", "choreo.sh"));
+
+                Text = "Clean up Choreo Vendordeps";
+                Progress = 75;
+                File.Delete(Path.Combine(configurationProvider.InstallDirectory, "vendordeps", "ChoreoLib2025Beta.json"));
+            }
+
+            await Task.Yield();
+        }
+
     }
 }