add badge_value check to default enum case

klaviyo · Dec 18, 2024 · 06c945a · 06c945a
1 parent a5963ac
commit 06c945a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Sources/KlaviyoSwiftExtension/KlaviyoExtension.swift b/Sources/KlaviyoSwiftExtension/KlaviyoExtension.swift
@@ -46,7 +46,13 @@ public enum KlaviyoExtensionSDK {
                     }
                     bestAttemptContent.badge = (badgeValue as NSNumber)
                 }
-            default: break
+            default:
+                if let badgeValue = bestAttemptContent.userInfo["badge_value"] as? Int {
+                    if let userDefaults = UserDefaults(suiteName: Bundle.main.object(forInfoDictionaryKey: "Klaviyo_App_Group") as? String) {
+                        userDefaults.set(badgeValue, forKey: "badgeCount")
+                    }
+                    bestAttemptContent.badge = (badgeValue as NSNumber)
+                }
             }
         }