From a55d2aa6247519984c4627ce6396f45ab4b138ae Mon Sep 17 00:00:00 2001
From: Ben Whitmore <byteben@users.noreply.github.com>
Date: Wed, 30 Dec 2020 14:22:24 +0000
Subject: [PATCH] Added a Snooze switch

Added a SNooze switch to allow users to Snooze the toast. Default snooze timer is set to 1min but can be changed in the XML Section of Toast_Notify.ps1 line 190

Also fixed an error in resolving the logged on user
---
 CustomMessage.xml |  1 +
 Toast_Notify.ps1  | 50 ++++++++++++++++++++++++++++++++++++++++++++---
 2 files changed, 48 insertions(+), 3 deletions(-)

diff --git a/CustomMessage.xml b/CustomMessage.xml
index 1b39fe2..809f1be 100644
--- a/CustomMessage.xml
+++ b/CustomMessage.xml
@@ -6,4 +6,5 @@
     <EventText>We are currently experiencing problems with all our systems. We are drinking coffee will provide an update shortly. Thank you for your patience</EventText>
     <ButtonTitle>Details</ButtonTitle>
     <ButtonAction>https://byteben.com</ButtonAction>
+	<SnoozeTitle>Snooze</SnoozeTitle>
 </ToastContent>
\ No newline at end of file
diff --git a/Toast_Notify.ps1 b/Toast_Notify.ps1
index eb1c2f6..b3016c0 100644
--- a/Toast_Notify.ps1
+++ b/Toast_Notify.ps1
@@ -36,6 +36,7 @@ Toast_Notify.ps1
 Param
 (
     [Parameter(Mandatory = $False)]
+    [Switch]$Snooze,
     [String]$XMLScriptDirSource = "CustomMessage.xml",
     [String]$XMLOtherSource
 
@@ -82,6 +83,7 @@ If ($XMLValid -eq $True) {
     $EventText = $XMLToast.ToastContent.EventText
     $ButtonTitle = $XMLToast.ToastContent.ButtonTitle
     $ButtonAction = $XMLToast.ToastContent.ButtonAction
+    $SnoozeTitle = $XMLToast.ToastContent.SnoozeTitle
 
     #ToastDuration: Short = 7s, Long = 25s
     $ToastDuration = "long"
@@ -99,7 +101,7 @@ If ($XMLValid -eq $True) {
     $Launcherid = "MSEdge"
 
     #Get last(current) logged on user
-    $LoggedOnUserPath = "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Authentication\LogonUI"
+    $LoggedOnUserPath = "HKLM:\SOFTWARE\Microsoft\Windows\CurrentVersion\Authentication\LogonUI\LastLoggedOnDisplayName"
     If (!(Test-Path $LoggedOnUserPath)) {
         Try {
             $Firstname = $Null
@@ -127,8 +129,10 @@ If ($XMLValid -eq $True) {
     [Windows.UI.Notifications.ToastNotificationManager, Windows.UI.Notifications, ContentType = WindowsRuntime] | Out-Null
     [Windows.Data.Xml.Dom.XmlDocument, Windows.Data.Xml.Dom.XmlDocument, ContentType = WindowsRuntime] | Out-Null
 
-    #Build XML Template
-    [xml]$ToastTemplate = @"
+    If (!$Snooze) {
+
+        #Build XML Template when Snooze Timer is not specified
+        [xml]$ToastTemplate = @"
 <toast duration="$ToastDuration" scenario="reminder">
     <visual>
         <binding template="ToastGeneric">
@@ -156,7 +160,47 @@ If ($XMLValid -eq $True) {
     </actions>
 </toast>
 "@
+    }
+    else {
 
+        #Build XML Template when Snooze Timer is specified
+        [xml]$ToastTemplate = @"
+<toast duration="$ToastDuration" scenario="reminder">
+    <visual>
+        <binding template="ToastGeneric">
+            <text>$CustomHello</text>
+            <text>$ToastTitle</text>
+            <text placement="attribution">$Signature</text>
+            <image placement="hero" src="$HeroImage"/>
+            <image placement="appLogoOverride" hint-crop="circle" src="$BadgeImage"/>
+            <group>
+                <subgroup>
+                    <text hint-style="title" hint-wrap="true" >$EventTitle</text>
+                </subgroup>
+            </group>
+            <group>
+                <subgroup>
+                    <text hint-style="body" hint-wrap="true" >$EventText</text>
+                </subgroup>
+            </group>
+        </binding>
+    </visual>
+    <audio src="ms-winsoundevent:notification.default"/>
+    <actions>
+        <input id="SnoozeTimer" type="selection" title="Select a Snooze Interval" defaultInput="1">
+            <selection id="1" content="1 Minute"/>
+            <selection id="30" content="30 Minutes"/>
+            <selection id="60" content="1 Hour"/>
+            <selection id="120" content="2 Hours"/>
+            <selection id="240" content="4 Hours"/>
+        </input> 
+        <action arguments="$ButtonAction" content="$ButtonTitle" activationType="protocol" />
+        <action activationType="system" arguments="snooze" hint-inputId="SnoozeTimer" content="$SnoozeTitle" id="test-snooze"/>
+        <action arguments="dismiss" content="Dismiss" activationType="system"/>
+    </actions>
+</toast>
+"@
+    }
     #Prepare XML
     $ToastXml = [Windows.Data.Xml.Dom.XmlDocument]::New()
     $ToastXml.LoadXml($ToastTemplate.OuterXml)