Add Task_scripts/Win - Toast notification for reboot.ps1

Task_scripts/Win - Toast notification for reboot.ps1

@@ -0,0 +1,80 @@
+<#
+    .SYNOPSIS
+        If the agent has a pending reboot display a notification on the screen.
+
+    .DESCRIPTION 
+        This script will use the Toast function in Windows 11 in order to display a notification to users to let them know they need to reboot.
+        This script can be used in order to let users reboot for Windows Updates for example. 
+
+    .OUTPUTS
+        n/a
+        
+        Errorcodes: 
+        n/a
+
+    .EXAMPLE
+        Win - Toast notification for reboot.ps1 {{agent.needs_reboot}}
+
+    .NOTES
+        Author: D.de Kooker <info@dcomputers.nl>
+        Source: n/a
+
+    .CHANGELOG
+        24-06-2022 - Initial script. 
+#>
+
+param ([Parameter(Mandatory)]$Needs_reboot)
+
+if ($Needs_reboot -ne "False") {
+    Write-Host "Reboot is needed, preparing popup."
+    #Checking if ToastReboot:// protocol handler is present
+    New-PSDrive -Name HKCR -PSProvider Registry -Root HKEY_CLASSES_ROOT -erroraction silentlycontinue | out-null
+    $ProtocolHandler = get-item 'HKCR:\ToastReboot' -erroraction 'silentlycontinue'
+    if (!$ProtocolHandler) {
+        #create handler for reboot
+        New-item 'HKCR:\ToastReboot' -force
+        set-itemproperty 'HKCR:\ToastReboot' -name '(DEFAULT)' -value 'url:ToastReboot' -force
+        set-itemproperty 'HKCR:\ToastReboot' -name 'URL Protocol' -value '' -force
+        new-itemproperty -path 'HKCR:\ToastReboot' -propertytype dword -name 'EditFlags' -value 2162688
+        New-item 'HKCR:\ToastReboot\Shell\Open\command' -force
+        set-itemproperty 'HKCR:\ToastReboot\Shell\Open\command' -name '(DEFAULT)' -value 'C:\Windows\System32\shutdown.exe -r -t 00' -force
+    }
+ 
+    if (!(Get-Module -ListAvailable -Name BurntToast)) {
+        Write-Host "Cannot find BurntToast, installing module" 
+        Install-Module -Name BurntToast -Force
+        }
+
+    if (!(Get-Module -ListAvailable -Name RunAsUser)) {
+        Write-Host "Cannot find RunAsUser, installing module" 
+        Install-Module -Name RunAsUser -Force
+        }
+
+    invoke-ascurrentuser -scriptblock {
+ 
+        $heroimage = New-BTImage -Source 'https://dcomputers.nl/wp-content/uploads/2023/08/trmm-windowsupdates.png' -HeroImage
+        $Text1 = New-BTText -Content  "Dcomputers"
+        $Text2 = New-BTText -Content "Er zijn op $(get-date) updates geinstalleerd en er is een herstart nodig. Herstart de computer nu of stel deze tot maximaal een dag uit."
+        $Button = New-BTButton -Content "Herstart NU" -Arguments "ToastReboot:" -ActivationType Protocol
+        $Button2 = New-BTButton -Content "Snooze" -snooze -id 'SnoozeTime'
+        $5Min = New-BTSelectionBoxItem -Id 5 -Content '5 minuten'
+        $10Min = New-BTSelectionBoxItem -Id 10 -Content '10 minuten'
+        $1Hour = New-BTSelectionBoxItem -Id 60 -Content '1 uur'
+        $4Hour = New-BTSelectionBoxItem -Id 240 -Content '4 uur'
+        $8Hour = New-BTSelectionBoxItem -Id 480 -Content '8 uur'
+        $1Day = New-BTSelectionBoxItem -Id 1440 -Content '1 dag'
+        $Items = $5Min, $10Min, $1Hour, $4Hour, $8Hour, $1Day
+        $SelectionBox = New-BTInput -Id 'SnoozeTime' -DefaultSelectionBoxItemId 10 -Items $Items
+        $action = New-BTAction -Buttons $Button, $Button2 -inputs $SelectionBox
+        $Binding = New-BTBinding -Children $text1, $text2 -HeroImage $heroimage
+        $Visual = New-BTVisual -BindingGeneric $Binding
+        $Content = New-BTContent -Visual $Visual -Actions $action
+        Submit-BTNotification -Content $Content
+    }
+    exit 0
+}
+
+else {
+    Write-Host "There is no reboot needed"
+    exit 0
+}