#!/bin/pwsh
. "$PSScriptRoot/../../../scripts/Windows/Scripts/Prerequisites.ps1";
. "$PSScriptRoot/../Scripts/Choco.ps1";
. "$PSScriptRoot/../Scripts/Config.ps1";
. "$PSScriptRoot/../Scripts/Hooks.ps1";
. "$PSScriptRoot/../Scripts/PowerManagement.ps1";
. "$PSScriptRoot/../Scripts/Update.ps1";
. "$PSScriptRoot/../Scripts/Context.ps1";
. "$PSScriptRoot/../Software/Firefox/Install.ps1";
. "$PSScriptRoot/Manage.ps1";
. "$PSScriptRoot/User/Install.ps1";

<#
    .SYNOPSIS
    Finishes the installation of a running Windows machine.
#>
function Start-WindowsInstallation {
    Start-InstallationLoop;
}

<#
    .SYNOPSIS
    Starts the installation loop.
#>
function Start-InstallationLoop {
    while (-not (Get-IsFinished)) {
        if ($null -eq (Get-Stage)) {
            Set-Stage ([SetupStage]::Initialize);
        } elseif ((Get-Stage) -eq ([SetupStage]::Initialize)) {
            if (-not ((Get-Command "choco") -and (Get-Command "refreshenv"))) {
                Invoke-Hook "Install-Chocolatey" -Fallback {
                    # Install chocolatey
                    [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-ChocoSoftware "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;
            }

            Invoke-Hook "Install-PSModules" -Fallback {
                Install-Module -AcceptLicense -Force PSWindowsUpdate;
            };

            if (Test-Path $env:PWSH_PATH) {
                attrib "-R" "$env:PWSH_PATH\*" /S /D;
                Remove-Item -Recurse -Force $env:PWSH_PATH;
                continue;
            }

            if (-not (& { wsl --status; $?; })) {
                wsl --install --no-launch;
                Restart-Intermediate;
                return;
            } else {
                ubuntu install --root;
                wsl -- sh `<`(curl -L https://nixos.org/nix/install`) --daemon --yes;
            }
        } else {
            $null = Import-Module PSWindowsUpdate;

            Invoke-Hook "Invoke-WindowsUpdate" -Fallback {
                Update-WindowsInstallation;
            };

            if ((Get-WURebootStatus -Silent)) {
                Restart-Intermediate;
                return;
            }

            switch (Get-Stage) {
                ([SetupStage]::Install) {
                }
                Default {}
            }
        }
    }
}

function Invoke-WindowsInstallation([Context] $context) {
    $Global:InformationPreference = "Continue";
    $Global:ErrorActionPreference = "Inquire";
    $context.UserNames ??= @("Manuel");
    Start-OldWindowsInstallationScript $context;
}

function Start-OldWindowsInstallationScript([Context] $context) {
    . "$PSScriptRoot/Upgrade.ps1";
    $finished = $false;
    Remove-Item Env:\POSH_THEME -ErrorAction SilentlyContinue;
    $configPath = "$PSScriptRoot/../Config";
    $softwarePath = "$PSScriptRoot/../Software";
    $initialConfigStage = "InitialConfiguration";
    $prerequisitesStage = "InstallationPrerequisites";
    $driverStage = "DriverInstallation";
    $softwareStage = "MachineWideSoftwareInstallation";
    $userStage = "CreatingUser";
    $restorationStage = "Restoring";


    while (-not $finished) {
        switch ($context.GetStage()) {
            { (-not $_) -or ($_ -eq $initialConfigStage) } {
                $context.SetStage($initialConfigStage);
        
                if ((Get-Command Initialize-Configuration -ErrorAction SilentlyContinue)) {
                    Write-Information "Configuration initialization function was found. Running...";
                    Initialize-Configuration $context;
                }
        
                $null = Enable-WindowsOptionalFeature -Online -All -FeatureName "NetFx3";
        
                . "$configPath/Windows/Install.ps1" $context;
                . "$configPath/Explorer/Install.ps1" $context;
                . "$configPath/OpenSSH/Install.ps1" $context;
                . "$configPath/chocolatey/Install.ps1";
        
                $context.RemoveDesktopIcon("*Microsoft Edge*");
                $context.SetStage($prerequisitesStage);
                break;
            }
            # Always install updates
            default {
                Write-Host "Starting Installation and Restoration of Windows";
                Update-WindowsInstallation $context;
            }
            $prerequisitesStage {
                Write-Host "Installing prerequisites for installing software";
        
                if (-not $(Get-Command winget)) {
                    choco install -y winget;
                }
        
                Install-Module -AcceptLicense -Force "NuGet";
                Import-Module NuGet;
        
                Install-Firefox $context;
                choco install -y selenium-gecko-driver;
                $null = Install-Package -Force Selenium.WebDriver -RequiredVersion 4.10.0 -SkipDependencies;
        
                winget install --accept-source-agreements --accept-package-agreements -e --id AutoHotkey.AutoHotkey;
        
                $context.SetStage($driverStage);
                break;
            }
            $driverStage {
                Write-Host "Installing drivers";
        
                if ((Get-Command Install-PortValhallaDrivers -ErrorAction SilentlyContinue)) {
                    Write-Information "Driver installation function was found. Starting installation";
                    Install-PortValhallaDrivers $context;
                }
        
                Write-Information "Finished installing drivers";
                $context.SetStage($softwareStage);
                $context.Reboot();
                exit;
            }
            $softwareStage {
                Write-Host "Setting up software with default app associations";
                . "$softwarePath/WinSCP/Install.ps1" $context;
                . "$softwarePath/Thunderbird/Install.ps1" $context;
    
                Write-Host "Installing default settings for new users";
                . "$softwarePath/aliae/Install.ps1" $context;
                . "$softwarePath/posh-git/Install.ps1";
                . "$softwarePath/Terminal-Icons/Install.ps1";
                . "$softwarePath/Oh My Posh/Install.ps1" $context;
                $context.SetStage($userStage);
                break;
            }
            $userStage {
                Install-PersonalUsers $context;
                $context.SetStage($restorationStage);
                break;
            }
            $restorationStage {
                Restore-WindowsInstallation $context;
                $finished = $true;
                break;
            }
        }
    }
}