From 20ba04392965fef934bd4d0dbb2c4399cde97a3e 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 | 24 ++++++++++++---------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/scripts/Windows/Scripts/PowerManagement.ps1 b/scripts/Windows/Scripts/PowerManagement.ps1
index b571279a..f8818d9e 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,7 +79,15 @@ $null = New-Module {
             [RegistryKey] $UserKey
         )
 
-        if ($User.IsPresent -or $UserKey) {
+        if ($DefaultUser.IsPresent) {
+            Edit-DefaultUserKey {
+                param(
+                    [RegistryKey] $Key
+                )
+
+                Register-Setup -UserKey $Key;
+            }
+        } elseif ($User.IsPresent -or $UserKey) {
             if (-not $UserKey) {
                 $UserKey = Get-Item "HKCU:\";
             }
@@ -159,15 +169,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;