add action icons to notification actions

OneSignal · Sep 3, 2021 · 506f2f9 · 506f2f9
1 parent d72573b
commit 506f2f9
Showing 1 changed file with 20 additions and 3 deletions.
diff --git a/iOS_SDK/OneSignalSDK/Source/OneSignalHelper.m b/iOS_SDK/OneSignalSDK/Source/OneSignalHelper.m
@@ -566,16 +566,33 @@ + (UNNotificationRequest*)prepareUNNotificationRequest:(OSNotification*)notifica
     return [UNNotificationRequest requestWithIdentifier:identifier content:content trigger:trigger];
 }
 
++ (UNNotificationAction *)createActionForButton:(NSDictionary *)button {
+    if (@available(iOS 15.0, *)) {
+        UNNotificationActionIcon *icon;
+        if (button[@"systemIcon"]) {
+            icon = [UNNotificationActionIcon iconWithSystemImageName:button[@"systemIcon"]];
+        } else if (button[@"templateIcon"]) {
+            icon = [UNNotificationActionIcon iconWithTemplateImageName:button[@"templateIcon"]];
+        }
+        return [UNNotificationAction actionWithIdentifier:button[@"id"]
+                                                    title:button[@"text"]
+                                                  options:UNNotificationActionOptionForeground
+                                                     icon:icon];
+    } else {
+        return [UNNotificationAction actionWithIdentifier:button[@"id"]
+                                                    title:button[@"text"]
+                                                  options:UNNotificationActionOptionForeground];
+    }
+}
+
 + (void)addActionButtons:(OSNotification*)notification
    toNotificationContent:(UNMutableNotificationContent*)content {
     if (!notification.actionButtons || notification.actionButtons.count == 0)
         return;
 
     let actionArray = [NSMutableArray new];
     for(NSDictionary* button in notification.actionButtons) {
-        let action = [UNNotificationAction actionWithIdentifier:button[@"id"]
-                                                          title:button[@"text"]
-                                                        options:UNNotificationActionOptionForeground];
+        let action = [self createActionForButton:button];
         [actionArray addObject:action];
     }