Snooze/dismiss support in button builders

Author: andrewleader

Microsoft.Toolkit.Uwp.Notifications/Toasts/ToastButton.cs

@@ -20,6 +20,11 @@ public sealed class ToastButton :
 
         private bool _usingCustomArguments;
 
+        private bool _usingSnoozeActivation;
+        private string _snoozeSelectionBoxId;
+
+        private bool _usingDismissActivation;
+
         /// <summary>
         /// Initializes a new instance of the <see cref="ToastButton"/> class.
         /// </summary>
@@ -213,6 +218,11 @@ private ToastButton AddArgumentHelper(string key, string value)
                 throw new InvalidOperationException("You cannot use the AddArgument methods when using protocol activation.");
             }
 
+            if (_usingDismissActivation || _usingSnoozeActivation)
+            {
+                throw new InvalidOperationException("You cannot use the AddArgument methods when using dismiss or snooze activation.");
+            }
+
             bool alreadyExists = _arguments.ContainsKey(key);
 
             _arguments[key] = value;
@@ -314,6 +324,48 @@ public ToastButton SetAfterActivationBehavior(ToastAfterActivationBehavior after
             return this;
         }
 
+        /// <summary>
+        /// Configures the button to use system snooze activation when the button is clicked, using the default system snooze time.
+        /// </summary>
+        /// <returns>The current instance of <see cref="ToastButton"/></returns>
+        public ToastButton SetSnoozeActivation()
+        {
+            return SetSnoozeActivation(null);
+        }
+
+        /// <summary>
+        /// Configures the button to use system snooze activation when the button is clicked, with a snooze time defined by the specified selection box.
+        /// </summary>
+        /// <param name="selectionBoxId">The ID of an existing <see cref="ToastSelectionBox"/> which allows the user to pick a custom snooze time. The ID's of the <see cref="ToastSelectionBoxItem"/>s inside the selection box must represent the snooze interval in minutes. For example, if the user selects an item that has an ID of "120", then the notification will be snoozed for 2 hours. When the user clicks this button, if you specified a SelectionBoxId, the system will parse the ID of the selected item and snooze by that amount of minutes.</param>
+        /// <returns>The current instance of <see cref="ToastButton"/></returns>
+        public ToastButton SetSnoozeActivation(string selectionBoxId)
+        {
+            if (_arguments.Count > 0)
+            {
+                throw new InvalidOperationException($"{nameof(SetSnoozeActivation)} cannot be used in conjunction with ${nameof(AddArgument)}.");
+            }
+
+            _usingSnoozeActivation = true;
+            _snoozeSelectionBoxId = selectionBoxId;
+
+            return this;
+        }
+
+        /// <summary>
+        /// Configures the button to use system dismiss activation when the button is clicked (the toast will simply dismiss rather than activating).
+        /// </summary>
+        /// <returns>The current instance of <see cref="ToastButton"/></returns>
+        public ToastButton SetDismissActivation()
+        {
+            if (_arguments.Count > 0)
+            {
+                throw new InvalidOperationException($"{nameof(SetDismissActivation)} cannot be used in conjunction with ${nameof(AddArgument)}.");
+            }
+
+            _usingDismissActivation = true;
+            return this;
+        }
+
         /// <summary>
         /// Sets an identifier used in telemetry to identify your category of action. This should be something like "Delete", "Reply", or "Archive". In the upcoming toast telemetry dashboard in Dev Center, you will be able to view how frequently your actions are being clicked.
         /// </summary>
@@ -349,7 +401,7 @@ public ToastButton SetTextBoxId(string textBoxId)
 
         internal bool CanAddArguments()
         {
-            return ActivationType != ToastActivationType.Protocol && !_usingCustomArguments;
+            return ActivationType != ToastActivationType.Protocol && !_usingCustomArguments && !_usingDismissActivation && !_usingSnoozeActivation;
         }
 
         internal bool ContainsArgument(string key)
@@ -362,13 +414,32 @@ internal Element_ToastAction ConvertToElement()
             var el = new Element_ToastAction()
             {
                 Content = Content,
-                Arguments = Arguments,
-                ActivationType = Element_Toast.ConvertActivationType(ActivationType),
                 ImageUri = ImageUri,
                 InputId = TextBoxId,
                 HintActionId = HintActionId
             };
 
+            if (_usingSnoozeActivation)
+            {
+                el.ActivationType = Element_ToastActivationType.System;
+                el.Arguments = "snooze";
+
+                if (_snoozeSelectionBoxId != null)
+                {
+                    el.InputId = _snoozeSelectionBoxId;
+                }
+            }
+            else if (_usingDismissActivation)
+            {
+                el.ActivationType = Element_ToastActivationType.System;
+                el.Arguments = "dismiss";
+            }
+            else
+            {
+                el.ActivationType = Element_Toast.ConvertActivationType(ActivationType);
+                el.Arguments = Arguments;
+            }
+
             ActivationOptions?.PopulateElement(el);
 
             return el;

UnitTests/UnitTests.Notifications.Shared/TestToastContentBuilder.cs

@@ -405,6 +405,36 @@ public void ToastButtonBuilders_InvalidProtocolAfterArguments_ReturnSelf()
                 .SetProtocolActivation(new Uri("https://msn.com"));
         }
 
+        [TestMethod]
+        [ExpectedException(typeof(InvalidOperationException))]
+        public void ToastButtonBuilders_InvalidDismissAfterArguments_ReturnSelf()
+        {
+            new ToastButton()
+                .SetContent("View")
+                .AddArgument("action", "view")
+                .SetDismissActivation();
+        }
+
+        [TestMethod]
+        [ExpectedException(typeof(InvalidOperationException))]
+        public void ToastButtonBuilders_InvalidSnoozeAfterArguments_ReturnSelf()
+        {
+            new ToastButton()
+                .SetContent("View")
+                .AddArgument("action", "view")
+                .SetSnoozeActivation();
+        }
+
+        [TestMethod]
+        [ExpectedException(typeof(InvalidOperationException))]
+        public void ToastButtonBuilders_InvalidSnoozeWithIdAfterArguments_ReturnSelf()
+        {
+            new ToastButton()
+                .SetContent("View")
+                .AddArgument("action", "view")
+                .SetSnoozeActivation("snoozeId");
+        }
+
         [TestMethod]
         [ExpectedException(typeof(InvalidOperationException))]
         public void ToastButtonBuilders_InvalidArgumentsAfterProtocol_ReturnSelf()
@@ -424,6 +454,36 @@ public void ToastButtonBuilders_InvalidArgumentsAfterCustomArguments_ReturnSelf(
             button.AddArgument("action", "view");
         }
 
+        [TestMethod]
+        [ExpectedException(typeof(InvalidOperationException))]
+        public void ToastButtonBuilders_InvalidArgumentsAfterSnooze_ReturnSelf()
+        {
+            new ToastButton()
+                .SetContent("Later")
+                .SetSnoozeActivation()
+                .AddArgument("action", "later");
+        }
+
+        [TestMethod]
+        [ExpectedException(typeof(InvalidOperationException))]
+        public void ToastButtonBuilders_InvalidArgumentsAfterSnoozeWithId_ReturnSelf()
+        {
+            new ToastButton()
+                .SetContent("Later")
+                .SetSnoozeActivation("myId")
+                .AddArgument("action", "later");
+        }
+
+        [TestMethod]
+        [ExpectedException(typeof(InvalidOperationException))]
+        public void ToastButtonBuilders_InvalidArgumentsAfterDismissActivation_ReturnSelf()
+        {
+            new ToastButton()
+                .SetContent("Later")
+                .SetDismissActivation()
+                .AddArgument("action", "later");
+        }
+
         [TestMethod]
         public void SetToastDurationTest_WithCustomToastDuration_ReturnSelfWithCustomToastDurationSet()
         {

UnitTests/UnitTests.Notifications.Shared/Test_Toast_Xml.cs

@@ -977,6 +977,37 @@ public void Test_Toast_Xml_Actions_SixTotal()
             Assert.Fail("Exception should have been thrown, only 5 actions are allowed.");
         }
 
+        [TestMethod]
+        public void Test_Toast_Xml_Actions_SnoozeAndDismissUsingBuilders()
+        {
+            AssertActionsPayload("<actions><action activationType='system' arguments='snooze'/><action activationType='system' arguments='dismiss'/><action content='Hide' activationType='system' arguments='dismiss'/><action content='Later' activationType='system' arguments='snooze'/><action content='Remind me' activationType='system' arguments='snooze' hint-inputId='snoozePicker'/></actions>", new ToastActionsCustom()
+            {
+                Buttons =
+                {
+                    // Allowing system to auto-generate text content
+                    new ToastButton()
+                        .SetSnoozeActivation(),
+
+                    // Allowing system to auto-generate text content
+                    new ToastButton()
+                        .SetDismissActivation(),
+
+                    // Specifying specific content
+                    new ToastButton()
+                        .SetContent("Hide")
+                        .SetDismissActivation(),
+
+                    new ToastButton()
+                        .SetContent("Later")
+                        .SetSnoozeActivation(),
+
+                    new ToastButton()
+                        .SetContent("Remind me")
+                        .SetSnoozeActivation("snoozePicker")
+                }
+            });
+        }
+
         [TestMethod]
         public void Test_Toast_Xml_Actions_TwoContextMenuItems()
         {