From 98dfabc2cd86f7d86390daff46d92f59ddcf8dbc Mon Sep 17 00:00:00 2001
From: Manuel Thalmann <m@nuth.ch>
Date: Wed, 21 Aug 2024 18:20:11 +0200
Subject: [PATCH] Allow registering startup script for the default user

---
 scripts/Windows/Scripts/PowerManagement.ps1 | 26 ++++++++++++++-------
 1 file changed, 17 insertions(+), 9 deletions(-)

diff --git a/scripts/Windows/Scripts/PowerManagement.ps1 b/scripts/Windows/Scripts/PowerManagement.ps1
index f9a1d9ef..dea82d82 100644
--- a/scripts/Windows/Scripts/PowerManagement.ps1
+++ b/scripts/Windows/Scripts/PowerManagement.ps1
@@ -70,6 +70,8 @@ $null = New-Module {
         param(
             [Parameter(ParameterSetName="System")]
             [switch] $System,
+            [Parameter(ParameterSetName="DefaultUser", Mandatory)]
+            [switch] $DefaultUser,
             [Parameter(ParameterSetName="User", Mandatory)]
             [switch] $User,
             [Parameter(ParameterSetName="User")]
@@ -77,6 +79,18 @@ $null = New-Module {
             [RegistryKey] $UserKey
         )
 
+        if ($DefaultUser.IsPresent) {
+            Edit-DefaultUserKey {
+                param(
+                    [RegistryKey] $Key
+                )
+
+                Register-Setup -UserKey $Key;
+            }
+
+            return;
+        }
+
         if ($User.IsPresent -or $UserKey) {
             if (-not $UserKey) {
                 $UserKey = Get-Item "HKCU:\";
@@ -159,15 +173,9 @@ $null = New-Module {
             [switch] $CurrentUser
         )
 
-        if ($DefaultUser.ToBool()) {
-            Edit-DefaultUserKey {
-                param(
-                    [RegistryKey] $Key
-                )
-
-                Register-Setup -UserKey $Key;
-            }
-        } elseif ($CurrentUser.ToBool()) {
+        if ($DefaultUser.IsPresent) {
+            Register-Setup -DefaultUser;
+        } elseif ($CurrentUser.IsPresent) {
             Register-Setup -User;
         } else {
             Register-Setup;