Handle new attributes.

duckduckgo · Jun 1, 2024 · f0c629e · f0c629e
1 parent e6f3e36
commit f0c629e
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/Sources/RemoteMessaging/Matchers/UserAttributeMatcher.swift b/Sources/RemoteMessaging/Matchers/UserAttributeMatcher.swift
@@ -124,6 +124,18 @@ public struct UserAttributeMatcher: AttributeMatcher {
             }
 
             return BooleanMatchingAttribute(value).matches(value: isPrivacyProSubscriber)
+        case let matchingAttribute as PrivacyProDaysSinceSubscribedMatchingAttribute:
+            if matchingAttribute.value != MatchingAttributeDefaults.intDefaultValue {
+                return IntMatchingAttribute(matchingAttribute.value).matches(value: privacyProDaysSinceSubscribed)
+            } else {
+                return RangeIntMatchingAttribute(min: matchingAttribute.min, max: matchingAttribute.max).matches(value: privacyProDaysSinceSubscribed)
+            }
+        case let matchingAttribute as PrivacyProDaysUntilExpiryMatchingAttribute:
+            if matchingAttribute.value != MatchingAttributeDefaults.intDefaultValue {
+                return IntMatchingAttribute(matchingAttribute.value).matches(value: privacyProDaysUntilExpiry)
+            } else {
+                return RangeIntMatchingAttribute(min: matchingAttribute.min, max: matchingAttribute.max).matches(value: privacyProDaysUntilExpiry)
+            }
         default:
             assertionFailure("Could not find matching attribute")
             return nil