Ensure the script path is detected properly

profiles/DerGeret/config.nix

@@ -32,6 +32,7 @@
       hardware = {
         components = [
           "ROG Zenith Extreme Alpha"
+          "Predator Z301C"
         ];
 
         eyeX = true;

scripts/Common/Scripts/Config.ps1

@@ -7,6 +7,7 @@ enum SetupStage {
     Configure
     Install
     CreateUser
+    ConfigureUser
 }
 
 $null = New-Module {
@@ -149,9 +150,21 @@ $null = New-Module {
         Gets the names of the users to create.
     #>
     function Get-Users {
+        [OutputType([string[]])]
+        param()
         Get-Attributes "valhalla.windows.users";
     }
 
+    <#
+        .SYNOPSIS
+        Gets the name of the setup user.
+    #>
+    function Get-SetupUser {
+        [OutputType([string])]
+        param()
+        Get-Config "valhalla.windows.setupUser";
+    }
+
     <#
         .SYNOPSIS
         Gets the value of an option related to the setup.

scripts/Common/Scripts/Operations.ps1

@@ -10,5 +10,6 @@ function Start-Operation {
         $env:CONFIG_MODULE = Resolve-Path $env:CONFIG_MODULE;
     }
 
+    New-Alias "sudo" gsudo;
     & $Action;
 }

scripts/Common/Scripts/Software.ps1

@@ -72,7 +72,8 @@ $null = New-Module {
                 winget install `
                     --accept-source-agreements --accept-package-agreements `
                     --source winget `
-                    --exact --id $name $ArgumentList;
+                    $ArgumentList `
+                    --exact --id $name ;
             } else {
                 Write-Host "Package ``$name`` is already installed"
             }

scripts/Windows/OS/Install.ps1

@@ -1,6 +1,9 @@
 #!/bin/pwsh
+using namespace System.Security.Principal;
+
 . "$PSScriptRoot/../../../scripts/Windows/Scripts/Prerequisites.ps1";
 . "$PSScriptRoot/../../Common/Scripts/Context.ps1";
+. "$PSScriptRoot/../Scripts/Security.ps1";
 . "$PSScriptRoot/../Software/Firefox/Install.ps1";
 . "$PSScriptRoot/Manage.ps1";
 . "$PSScriptRoot/User/Install.ps1";
@@ -8,6 +11,7 @@
 $null = New-Module {
     . "$PSScriptRoot/../Scripts/Hooks.ps1";
     . "$PSScriptRoot/../Scripts/PowerManagement.ps1";
+    . "$PSScriptRoot/../Scripts/Registry.ps1";
     . "$PSScriptRoot/../Scripts/Update.ps1";
     . "$PSScriptRoot/../Scripts/Users.ps1";
     . "$PSScriptRoot/../../Common/Scripts/Config.ps1";
@@ -32,9 +36,12 @@ $null = New-Module {
     #>
     function Start-InstallationLoop {
         while (-not (Get-IsFinished)) {
-            if ($null -eq (Get-Stage)) {
+            switch (Get-Stage) {
+                ($null) {
                     Set-Stage ([SetupStage]::Initialize);
-            } elseif ((Get-Stage) -eq ([SetupStage]::Initialize)) {
+                    break;
+                }
+                ([SetupStage]::Initialize) {
                     if (-not ((Test-Command "choco") -and (Test-Command "refreshenv"))) {
                         Invoke-Hook "Install-Chocolatey" -Fallback {
                             # Install chocolatey
@@ -47,6 +54,12 @@ $null = New-Module {
                         continue;
                     }
 
+                    if (-not (Test-Command "gsudo")) {
+                        Install-ChocoPackage gsudo;
+                        refreshenv;
+                        continue;
+                    }
+
                     if (-not (Test-Command "git")) {
                         Install-ChocoPackage git;
                         refreshenv;
@@ -62,7 +75,7 @@ $null = New-Module {
                         return;
                     }
     
-                if (Test-Path $env:PWSH_PATH) {
+                    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;
@@ -107,7 +120,9 @@ $null = New-Module {
                     };
 
                     Set-Stage ([SetupStage]::Configure);
-            } else {
+                    break;
+                }
+                default {
                     $null = Import-Module PSWindowsUpdate;
         
                     Invoke-Hook "Invoke-WindowsUpdate" -Fallback {
@@ -135,10 +150,10 @@ $null = New-Module {
                         $arguments = [hashtable]@{ };
 
                         if ($Action) {
-                        $install = $true;
+                            $install = ($Action -eq ([InstallerAction]::Install));
                             $null = $arguments.Add("action", $Action);
                         } else {
-                        $install = ($Action -eq ([InstallerAction]::Install));
+                            $install = $true;
                         }
 
                         # Drivers
@@ -173,7 +188,7 @@ $null = New-Module {
                                     Remove-DesktopIcon "*Geforce*";
                                 }
 
-                            if (Get-Config "valhalla.hardware.corsairLighting") {
+                                if (Get-Config "valhalla.hardware.corsairDevice") {
                                     Install-ChocoPackage icue;
                                 }
 
@@ -387,7 +402,7 @@ $null = New-Module {
                                 . "$softwarePath/ManiaPlanet/Manage.ps1" @arguments;
                                 . "$softwarePath/osu!/Manage.ps1" @arguments;
                                 . "$softwarePath/osu!lazer/Manage.ps1" @arguments;
-                            . "$softwarePath/RetorArch/Manage.ps1" @arguments;
+                                . "$softwarePath/RetroArch/Manage.ps1" @arguments;
                                 . "$softwarePath/reWASD/Manage.ps1" @arguments;
                             }
                         };
@@ -419,12 +434,56 @@ $null = New-Module {
                         }
                         ([SetupStage]::CreateUser) {
                             Start-ValhallaUserSetup;
+                            Set-Stage ([SetupStage]::ConfigureUser);
+                        }
+                        ([SetupStage]::ConfigureUser) {
+                            $userOption = "CurrentUser";
+
+                            function Get-CurrentUser {
+                                (Get-SetupOption $userOption) ?? 0;
+                            }
+
+                            function Set-CurrentUser {
+                                param([int] $Value)
+                                Set-SetupOption $userOption $Value;
+                            }
+
+                            [string[]] $users = Get-Users;
+                            $currentUser = Get-CurrentUser;
+                            Disable-BootMessage;
+
+                            if ($currentUser -lt $users.Count) {
+                                $user = Get-LocalUser $users[$currentUser];
+
+                                Add-LocalGroupMember `
+                                    -SID ([SecurityIdentifier]::new([WellKnownSidType]::BuiltinAdministratorsSid, $null))`
+                                    $user `
+                                    -ErrorAction SilentlyContinue;
+
+                                if ($env:UserName -ne "$user") {
+                                    Disable-LocalUser $env:UserName;
+                                    Enable-LocalUser $user;
+
+                                    if (Get-UserConfig -UserName "$user" -Name "microsoftAccount") {
+                                        Enable-UAC;
+                                        Disable-Autologin;
+                                        Set-BootMessage -Caption "Login" -Message "Please login using your account.";
+                                    } else {
+                                        Set-AutologinUser "$user";
+                                    }
+
+                                    Restart-Intermediate -DefaultUser;
+                                    return;
+                                }
+                            }
+
                             Set-IsFinished $true;
                         }
                     }
                 }
             }
         }
+    }
     
     function Invoke-WindowsInstallation([Context] $context) {
         $Global:InformationPreference = "Continue";

scripts/Windows/Scripts/AppAssociations.ps1

@@ -61,7 +61,6 @@ $null = New-Module {
         [System.Xml.XmlNode] $association = $null;
         $document = Get-DefaultAppAssociations;
         $candidates = $document.SelectNodes((& $getSelector $Identifier));
-        Write-Host "Number of potential associations: $($candidates.Count)"
 
         if ($candidates.Count -eq 1) {
             $association = $candidates[0];

scripts/Windows/Scripts/PowerManagement.ps1

@@ -1,17 +1,41 @@
 using namespace Microsoft.Win32;
 
 $null = New-Module {
+    . "$PSScriptRoot/../Scripts/Registry.ps1";
+    . "$PSScriptRoot/../../Common/Scripts/Config.ps1";
     . "$PSScriptRoot/../../Common/Scripts/Scripting.ps1";
     [RegistryKey] $key = $null;
-    $runOncePath = "HKLM:\Software\Microsoft\Windows\CurrentVersion\RunOnce";
+    $runOncePath = "Software\Microsoft\Windows\CurrentVersion\RunOnce";
+    $systemRunOncePath = "HKLM:\$runOncePath";
+    $logonPath = "HKLM:\SOFTWARE\Microsoft\Windows NT\CurrentVersion\Winlogon"
     $runOnceName = "PortValhalla";
+    $autologinOption = "AutoAdminLogon";
+    $domainOption = "DefaultDomainName";
+    $userOption = "DefaultUserName";
+    $passwordOption = "DefaultPassword";
 
     <#
         .SYNOPSIS
         Gets the reghistry key containing the `RunOnce` commands.
     #>
     function Get-RunOnceKey {
-        Get-Item $runOncePath;
+        param(
+            [RegistryKey] $UserKey
+        )
+
+        [string] $path;
+
+        if ($UserKey) {
+            $path = "$($UserKey.PSPath)\$runOncePath";
+        } else {
+            $path = $systemRunOncePath;
+        }
+
+        if (-not (Test-Path $path)) {
+            New-Item $path;
+        } else {
+            Get-Item $path;
+        }
     }
 
     <#
@@ -23,15 +47,34 @@ $null = New-Module {
     #>
     function Register-Setup {
         param(
+            [Parameter(ParameterSetName="System")]
+            [switch] $System,
+            [Parameter(ParameterSetName="User",Mandatory)]
+            [switch] $User,
+            [Parameter(Mandatory)]
             [RegistryKey] $UserKey
         )
 
+        if ($User.IsPresent) {
+            if (-not $UserKey) {
+                $UserKey = Get-Item "HKCU:\";
+            }
+
             $key = Get-RunOnceKey $UserKey;
+        } else {
+            $key = Get-RunOnceKey;
+        }
 
         Set-ItemProperty -Path $key.PSPath -Name $runOnceName -Type "ExpandString" -Value (
             "pwsh -Command " +
-            "`$env:PWSH_PATH = $(ConvertTo-Injection $env:PWSH_PATH);" +
+            (& {
-            "`$env:INSTALLER_SCRIPT = $(ConvertTo-Injection $env:INSTALLER_SCRIPT);" +
+                if ($env:PWSH_PATH) {
+                    "`$env:PWSH_PATH = $(ConvertTo-Injection $env:PWSH_PATH);"
+                } else {
+                    ""
+                }
+            }) +
+            "`$env:INSTALLER_SCRIPT = $(ConvertTo-Injection (Resolve-Path $env:INSTALLER_SCRIPT));" +
             "`$env:CONFIG_MODULE = $(ConvertTo-Injection $env:CONFIG_MODULE);" +
             "& `$env:INSTALLER_SCRIPT;"
         );
@@ -39,12 +82,79 @@ $null = New-Module {
         $key.Handle.Close();
     }
 
+    <#
+        .SYNOPSIS
+        Sets the user to login automatically on boot.
+
+        .PARAMETER Name
+        The name of the user to login automatically.
+    #>
+    function Set-AutologinUser {
+        param(
+            [string] $Name
+        )
+
+        Set-ItemProperty $autologinOption -Name $autologinOption "1";
+
+        if (-not $Name) {
+            $Name = Get-SetupUser;
+        }
+
+        $options = @{
+            $domainOption = "";
+            $userOption = $Name;
+            $passwordOption = "";
+        };
+
+        foreach ($key in $options.Keys) {
+            Set-ItemProperty -Name $key
+            $displayName = Get-UserConfig -UserName $name "displayName";
+
+            $userArguments = @{
+                name = $name;
+            };
+
+            if ($displayName) {
+                $userArguments.fullName = $displayName;
+            } -Value $options[$key];
+        }
+    }
+
+    <#
+        .SYNOPSIS
+        Disables the automatic login.
+    #>
+    function Disable-Autologin {
+        Set-ItemProperty $logonPath -Name $autologinOption "0";
+
+        foreach ($key in @($domainOption, $userOption, $passwordOption)) {
+            Remove-ItemProperty $logonPath -Name $key -ErrorAction SilentlyContinue;
+        }
+    }
+
     <#
         .SYNOPSIS
         Reboots the machine intermediately and restarts the setup after the next login.
     #>
     function Restart-Intermediate {
-        Register-Setup;
+        param(
-        Restart-Computer;
+            [switch] $DefaultUser
+        )
+
+        $register = { param($Key) Register-Setup -UserKey $Key; };
+
+        if ($DefaultUser) {
+            Edit-DefaultUserKey {
+                param(
+                    [RegistryKey] $Key
+                )
+
+                & $register -Key $Key;
+            }
+        }
+
+        & $register;
+        Read-Host "Press enter to reboot";
+        Restart-Computer -Force;
     }
 }

scripts/Windows/Scripts/Registry.ps1

@@ -16,4 +16,41 @@ $null = New-Module {
         [System.GC]::Collect();
         & reg unload $regRootPath;
     }
+
+    <#
+        .SYNOPSIS
+        Sets a message to show on the login screen.
+
+        .PARAMETER Caption
+        The title of the message.
+
+        .PARAMETER Message
+        The text of the message.
+    #>
+    function Set-BootMessage {
+        param(
+            [string] $Caption,
+            [string] $Message
+        )
+
+        $options = @{
+            legalnoticecaption = $Caption;
+            legalnoticetext = $Message;
+        };
+
+        foreach ($key in $options.Keys) {
+            Set-ItemProperty "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Policies\System" `
+                -Name $key `
+                -Type "String" `
+                -Value ($options[$key]);
+        }
+    }
+
+    <#
+        .SYNOPSIS
+        Disables the boot message.
+    #>
+    function Disable-BootMessage {
+        Set-BootMessage;
+    }
 }

scripts/Windows/Scripts/Security.ps1

@@ -0,0 +1,36 @@
+$null = New-Module {
+    $uacOption = "EnableLUA";
+    $systemPolicyPath = "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Policies\System";
+
+    $uacSetter = {
+        param(
+            [bool] $Value
+        )
+
+        Set-ItemProperty -Path $systemPolicyPath -Name $uacOption -Value ([int]$Value);
+    }
+
+    <#
+        .SYNOPSIS
+        Determines whether UAC is enabled.
+    #>
+    function Get-UACStatus {
+        [bool](Get-ItemProperty -Path $systemPolicyPath -Name $uacOption);
+    }
+
+    <#
+        .SYNOPSIS
+        Enables UAC.
+    #>
+    function Enable-UAC {
+        & $uacSetter $true;
+    }
+
+    <#
+        .SYNOPSIS
+        Disables UAC.
+    #>
+    function Disable-UAC {
+        & $uacSetter $false;
+    }
+};

scripts/Windows/Scripts/Users.ps1

@@ -57,7 +57,7 @@ $null = New-Module {
                             "So sorry…") -join "`n");
 
                     Write-Host "Create a user for ``$Name`` manually (because Windows is too stupid)…";
-                    Read-Host "Hit enter once you're done";
+                    $null = Read-Host "Hit enter once you're done";
 
                     $newUsers = @(Get-LocalUser | Where-Object { -not ($currentUsers -contains $_.Name) });
 
@@ -96,7 +96,8 @@ $null = New-Module {
                                 return $newUsers[$result - 1];
                             }
                         }
-
+                    } else {
+                        Write-Host "";
                         Write-Host "Unable to determine the new user";
                         Write-Host "Retrying…";
                     }
@@ -111,10 +112,6 @@ $null = New-Module {
             Set-SetupOption $userOption $i;
             $name = $users[$i];
             Write-Host "Creating personal user ``$name``…";
-
-            if (Get-UserConfig -UserName $name "microsoftAccount") {
-                Add-MicrosoftAccount $name;
-            } else {
             $displayName = Get-UserConfig -UserName $name "displayName";
 
             $userArguments = @{
@@ -125,8 +122,13 @@ $null = New-Module {
                 $userArguments.fullName = $displayName;
             }
 
-                New-LocalUser -NoPassword @userArguments;
+            if (Get-UserConfig -UserName $name "microsoftAccount") {
+                Add-MicrosoftAccount $name;
+            } else {
+                New-LocalUser -Disabled -NoPassword @userArguments;
             }
+
+            Set-LocalUser @userArguments;
         }
     }
 };

scripts/Windows/Software/ManiaPlanet/Manage.ps1

@@ -4,10 +4,12 @@ param(
 )
 
 . "$PSScriptRoot/../../../Common/Scripts/Software.ps1";
+. "$PSScriptRoot/../../../Common/Scripts/System.ps1";
 
 Start-SoftwareInstaller @PSBoundParameters `
     -Installer {
         Install-WingetPackage Nadeo.ManiaPlanet;
+        Remove-DesktopIcon "ManiaPlanet*";
     };
 
 # ToDo: Add restoration

scripts/Windows/Software/PuTTY/Manage.ps1

@@ -5,8 +5,8 @@ param(
     [hashtable] $Arguments
 )
 
-. "$PSScriptRoot/../../Scripts/Software.ps1";
+. "$PSScriptRoot/../../Scripts/AppAssociations.ps1";
-. "$PSScriptRoot/../../../Common/Scripts/AppAssociations.ps1";
+. "$PSScriptRoot/../../../Common/Scripts/Software.ps1";
 
 Start-SoftwareInstaller @PSBoundParameters `
     -Installer {

scripts/Windows/Software/VisualStudio/Manage.ps1

@@ -21,9 +21,10 @@ $null = New-Module {
                 $packageName = $version[0];
                 Write-Host "Installing ``$packageName``…";
                 Install-ChocoPackage $packageName;
+            }
+
             Remove-DesktopIcon "CocosCreator*";
             Remove-DesktopIcon "Unity Hub*";
-            }
         };
     
     # ToDo: Add restoration