Fix onAcceptAll, onDenyAll and load preferences to let 'N/A' value

segmentio · Jul 19, 2023 · 46a45a4 · 46a45a4
1 parent 0670b5f
commit 46a45a4
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 9 deletions.
diff --git a/src/consent-manager/container.tsx b/src/consent-manager/container.tsx
@@ -92,19 +92,30 @@ const Container: React.FC<ContainerProps> = props => {
   } = normalizeDestinations(props.destinations)
 
   const onAcceptAll = () => {
-    const truePreferences = Object.keys(props.preferences).reduce((acc, category) => {
-      acc[category] = true
-      return acc
-    }, {})
+    const truePreferences: CategoryPreferences = props.preferences
+    for (const preferenceName of Object.keys(props.preferences)) {
+      const value = props.preferences[preferenceName]
+      if (typeof value === 'string') {
+        truePreferences[preferenceName] = value
+      } else {
+        truePreferences[preferenceName] = true
+      }
+    }
+
     props.setPreferences(truePreferences)
     return props.saveConsent()
   }
 
   const onDenyAll = () => {
-    const falsePreferences = Object.keys(props.preferences).reduce((acc, category) => {
-      acc[category] = false
-      return acc
-    }, {})
+    const falsePreferences: CategoryPreferences = props.preferences
+    for (const preferenceName of Object.keys(props.preferences)) {
+      const value = props.preferences[preferenceName]
+      if (typeof value === 'string') {
+        falsePreferences[preferenceName] = value
+      } else {
+        falsePreferences[preferenceName] = false
+      }
+    }
 
     props.setPreferences(falsePreferences)
     return props.saveConsent()

diff --git a/src/consent-manager/index.tsx b/src/consent-manager/index.tsx
@@ -248,7 +248,7 @@ export default class ConsentManager extends PureComponent<ConsentManagerProps, {
     if (customCategories) {
       for (const preferenceName of Object.keys(customCategories)) {
         const value = preferences[preferenceName]
-        if (typeof value === 'boolean') {
+        if (typeof value === 'boolean' || typeof value === 'string') {
           customPreferences[preferenceName] = value
         } else {
           customPreferences[preferenceName] = true