From fa5b4623129681fa29afe5220938114a26a75314 Mon Sep 17 00:00:00 2001
From: Manuel Thalmann <m@nuth.ch>
Date: Sat, 24 Aug 2024 16:15:10 +0200
Subject: [PATCH] Remove unused resources after user setup

---
 scripts/Common/Scripts/Config.ps1 |  1 +
 scripts/Windows/OS/Install.ps1    | 19 ++++++++++++++++---
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/scripts/Common/Scripts/Config.ps1 b/scripts/Common/Scripts/Config.ps1
index db3359c8..e448cea5 100644
--- a/scripts/Common/Scripts/Config.ps1
+++ b/scripts/Common/Scripts/Config.ps1
@@ -14,6 +14,7 @@ enum SetupStage {
 enum UserStage {
     Create
     Configure
+    Cleanup
     Completed
 }
 
diff --git a/scripts/Windows/OS/Install.ps1 b/scripts/Windows/OS/Install.ps1
index 34b89f61..7d638b48 100644
--- a/scripts/Windows/OS/Install.ps1
+++ b/scripts/Windows/OS/Install.ps1
@@ -37,7 +37,9 @@ $null = New-Module {
         Starts the installation loop.
     #>
     function Start-InstallationLoop {
-        $wslLocation = "$env:ProgramData\PortValhalla\Ubuntu";
+        $distribution = "PortValhalla";
+        $wslLocation = "$env:ProgramData/PortValhalla/Ubuntu";
+        $wslDisk = "$wslLocation/ext4.vhdx";
 
         while (-not (Get-IsFinished)) {
             switch (Get-Stage) {
@@ -195,8 +197,8 @@ $null = New-Module {
                 }
                 default {
                     if (-not (& { $null = wsl -l; $? })) {
-                        wsl --import-in-place "PortValhalla" "$wslLocation/ext4.vhdx";
-                        wsl --set-default "PortValhalla";
+                        wsl --import-in-place $distribution "$wslDisk";
+                        wsl --set-default $distribution;
                     }
 
                     switch ($_) {
@@ -610,7 +612,18 @@ $null = New-Module {
                                                         Add-LocalGroupMember -Member "$name" -Name "$group";
                                                     }
 
+                                                    Set-UserStage ([UserStage]::Cleanup);
+                                                }
+                                                ([UserStage]::Cleanup) {
+                                                    $user = Get-SetupUser;
+                                                    Enable-LocalUser $user;
+                                                    Set-AutologinUser $user;
+                                                    wsl --shutdown;
+                                                    $tempDisk = Rename-Item -Force $wslDisk "ext4_.vhdx" -PassThru;
+                                                    wsl --unregister $distribution;
+                                                    Move-Item $tempDisk $wslDisk;
                                                     Set-UserStage ([UserStage]::Completed);
+                                                    Restart-Intermediate;
                                                 }
                                             }
                                         }