Handle initialization in operation code

scripts/Common/Scripts/Operations.ps1

@@ -1,12 +1,16 @@
-. "$PSScriptRoot/Config.ps1";
+using namespace System.Management.Automation.Host;
+
 . "$PSScriptRoot/../Types/OneShotTask.ps1";
-. "$PSScriptRoot/../../Windows/Scripts/PowerManagement.ps1";
-. "$PSScriptRoot/../../Windows/Scripts/Registry.ps1";
-. "$PSScriptRoot/../../Windows/Scripts/Security.ps1";
-. "$PSScriptRoot/../../Windows/Scripts/WSL.ps1";
 
 $null = New-Module {
+    . "$PSScriptRoot/Config.ps1";
+    . "$PSScriptRoot/../Scripts/SoftwareManagement.ps1";
     . "$PSScriptRoot/../Types/OneShotTask.ps1";
+    . "$PSScriptRoot/../../Windows/Scripts/Hooks.ps1";
+    . "$PSScriptRoot/../../Windows/Scripts/PowerManagement.ps1";
+    . "$PSScriptRoot/../../Windows/Scripts/Registry.ps1";
+    . "$PSScriptRoot/../../Windows/Scripts/Security.ps1";
+    . "$PSScriptRoot/../../Windows/Scripts/WSL.ps1";
     $oneShotTaskName = "PortValhalla OneShot";
     $logName = "Application";
     $oneShotTrigger = 1337;
@@ -59,6 +63,166 @@ $null = New-Module {
                     Clear-OperationResources;
                 };
             }
+
+            Set-Stage ((Get-Stage) ?? ([SetupStage]::Initialize));
+
+            & {
+                while ((Get-Stage) -eq ([SetupStage]::Initialize)) {
+                    if ($IsWindows) {
+                        if (-not ((Test-Command "choco") -and (Test-Command "refreshenv"))) {
+                            Invoke-Hook "Install-Chocolatey" -Fallback {
+                                # Install chocolatey
+                                New-Item -Force $PROFILE;
+                                [System.Net.ServicePointManager]::SecurityProtocol = [System.Net.ServicePointManager]::SecurityProtocol -bor 3072;
+                                Invoke-Expression ((New-Object System.Net.WebClient).DownloadString('https://community.chocolatey.org/install.ps1'));
+                                Import-Module $env:ChocolateyInstall/helpers/chocolateyProfile.psm1;
+                                refreshenv;
+                            };
+
+                            continue;
+                        }
+
+                        if (-not (Test-ChocoPackage "powershell-core")) {
+                            Invoke-Hook "Install-PowerShellCore" -Fallback {
+                                choco install -y powershell-core --install-arguments='"ADD_FILE_CONTEXT_MENU_RUNPOWERSHELL=1 ADD_EXPLORER_CONTEXT_MENU_OPENPOWERSHELL=1 REGISTER_MANIFEST=1 USER_MU=1 ENABLE_MU=1"';
+                            };
+
+                            Restart-Intermediate;
+                            return;
+                        }
+
+                        if ($env:PWSH_PATH -and (Test-Path $env:PWSH_PATH)) {
+                            attrib "-R" "$env:PWSH_PATH\*" /S /D;
+                            Remove-Item -Recurse -Force $env:PWSH_PATH;
+                            continue;
+                        }
+
+                        if ($env:DEBUG) {
+                            if (
+                                (Get-Item $env:INSTALLER_SCRIPT).IsReadOnly -and
+                                (Test-Qemu) -and
+                                ($Host.UI.PromptForChoice(
+                                    "Confirm",
+                                    "Do you wish to swap to live scripts?",
+                                    [ChoiceDescription[]]@(
+                                        [ChoiceDescription]::new("&No", "Use scripts stored in the virtual machine"),
+                                        [ChoiceDescription]::new("&Yes", "Use live scripts stored on the host")),
+                                    0) -eq 1)) {
+                                Install-ChocoPackage winfsp qemu-guest-agent;
+                                Get-Service VirtioFsSvc | Start-Service -PassThru | Set-Service -StartupType Automatic;
+
+                                while (-not (Test-Path Z:\)) {
+                                    Start-Sleep 0.1;
+                                }
+
+                                foreach ($name in @("CONFIG_MODULE", "INSTALLER_SCRIPT")) {
+                                    $variable = Get-Item "Env:\$name";
+
+                                    $path = Join-Path `
+                                        "Z:\Repositories\PortValhalla" `
+                                        ([System.IO.Path]::GetRelativePath("$PSScriptRoot/../../..", $variable.Value));
+
+                                    Set-Item "Env:\$name" $path;
+                                    Write-Host "The new value of ``$name`` is ``$path``";
+                                }
+
+                                Restart-Intermediate;
+                                exit;
+                            }
+                        }
+
+                        if (-not (Test-Command "gsudo")) {
+                            Install-ChocoPackage gsudo;
+                            refreshenv;
+                            continue;
+                        }
+
+                        if ($env:DEBUG) {
+                            & {
+                                $sys32 = "$env:WINDIR/System32";
+                                $osk = (Get-Item "$sys32/osk.exe").FullName;
+                                $cmd = (Get-Item "$sys32/cmd.exe").FullName;
+                                $tmpOsk = New-TemporaryFile;
+                                $tmpCmd = New-TemporaryFile;
+                                gsudo -d copy "$osk" "$tmpOsk";
+                                gsudo -d copy "$cmd" "$tmpCmd";
+
+                                if ((Get-FileHash $tmpOsk).Hash -ne (Get-FileHash $tmpCmd).Hash) {
+                                    Set-MpPreference -ExclusionPath $osk;
+                                    gsudo -d --ti move $osk "${osk}_";
+                                    gsudo -d -s copy $cmd $osk;
+                                    continue;
+                                }
+                            };
+                        }
+
+                        if (-not (Test-Winget)) {
+                            . "$PSScriptRoot/../../Windows/Software/winget/Manage.ps1";
+                            continue;
+                        }
+
+                        if (-not (Test-Command "git")) {
+                            Install-WingetPackage Git.Git;
+                            refreshenv;
+                            continue;
+                        }
+
+                        if (-not (Test-Wsl)) {
+                            Install-Wsl;
+                            Restart-Intermediate;
+                            return;
+                        }
+
+                        if (-not (Test-WslDistributions)) {
+                            Install-WslDistribution;
+                            continue;
+                        }
+
+                        if (-not (wsl --shell-type login type -t nix)) {
+                            wsl -- sh `<`(curl -L https://nixos.org/nix/install`) --daemon --yes;
+                            wsl --shutdown;
+                            continue;
+                        }
+
+                        Invoke-Hook "Install-PSModules" -Fallback {
+                            $modules = @(
+                                @("KnownFolders"),
+                                @("PSWindowsUpdate"),
+                                @("PSScriptAnalyzer"),
+                                @("LocalAccounts", $true),
+                                @("NuGet")
+                            );
+
+                            foreach ($module in $modules) {
+                                $parameters = @{ };
+
+                                if ($module[1]) {
+                                    $parameters = @{
+                                        allowPrerelease = $true;
+                                    };
+                                }
+
+                                if (-not (Test-PSModule $module[0])) {
+                                    Install-Module -Scope AllUsers -AcceptLicense -Force -AllowClobber $module[0] @parameters;
+                                    Import-Module $module[0];
+                                }
+                            }
+
+                            . "$PSScriptRoot/../../Windows/Software/PinnedItem/Manage.ps1";
+                        };
+
+                        if (-not (Test-PSPackage Selenium.WebDriver)) {
+                            Write-Host "Installing browser automation tools…";
+                            $null = Install-Package -Force Selenium.WebDriver -RequiredVersion 4.10.0 -SkipDependencies;
+                            continue;
+                        }
+
+                        Install-ChocoPackage selenium-gecko-driver firefox;
+                        Install-WingetPackage AutoHotkey.AutoHotkey;
+                        Set-Stage ([SetupStage]::Configure);
+                    }
+                }
+            }
         }
 
         & $Action;