add getFeatureConfig implementation

Author: cmonfortep

subscriptions/subscriptions-impl/src/main/java/com/duckduckgo/subscriptions/impl/messaging/SubscriptionsJSHelper.kt

@@ -20,6 +20,7 @@ import com.duckduckgo.di.scopes.AppScope
 import com.duckduckgo.js.messaging.api.JsCallbackData
 import com.duckduckgo.subscriptions.api.SubscriptionsJSHelper
 import com.duckduckgo.subscriptions.impl.AccessTokenResult
+import com.duckduckgo.subscriptions.impl.PrivacyProFeature
 import com.duckduckgo.subscriptions.impl.SubscriptionsManager
 import com.squareup.anvil.annotations.ContributesBinding
 import javax.inject.Inject
@@ -29,6 +30,7 @@ import org.json.JSONObject
 @ContributesBinding(AppScope::class)
 class RealSubscriptionsJSHelper @Inject constructor(
     private val subscriptionsManager: SubscriptionsManager,
+    private val privacyProFeature: PrivacyProFeature,
 ) : SubscriptionsJSHelper {
 
     override suspend fun processJsCallbackMessage(
@@ -57,6 +59,10 @@ class RealSubscriptionsJSHelper @Inject constructor(
             getAuthAccessTokenData(featureName, method, it)
         }
 
+        METHOD_GET_FEATURE_CONFIG -> id?.let {
+            getFeatureConfigData(featureName, method, it)
+        }
+
         else -> null
     }
 
@@ -88,6 +94,14 @@ class RealSubscriptionsJSHelper @Inject constructor(
         return JsCallbackData(jsonPayload, featureName, method, id)
     }
 
+    private suspend fun getFeatureConfigData(featureName: String, method: String, id: String): JsCallbackData {
+        val jsonPayload = JSONObject().apply {
+            put(USE_PAID_DUCK_AI, privacyProFeature.duckAiPlus().isEnabled())
+        }
+
+        return JsCallbackData(jsonPayload, featureName, method, id)
+    }
+
     companion object {
         private const val METHOD_HANDSHAKE = "handshake"
         private const val METHOD_SUBSCRIPTION_DETAILS = "subscriptionDetails"
@@ -106,5 +120,6 @@ class RealSubscriptionsJSHelper @Inject constructor(
         private const val PAYMENT_PLATFORM = "paymentPlatform"
         private const val STATUS = "status"
         private const val ACCESS_TOKEN = "accessToken"
+        private const val USE_PAID_DUCK_AI = "usePaidDuckAi"
     }
 }

subscriptions/subscriptions-impl/src/test/java/com/duckduckgo/subscriptions/impl/messaging/RealSubscriptionsJSHelperTest.kt

@@ -11,11 +11,14 @@ import com.duckduckgo.subscriptions.impl.SubscriptionsConstants.MONTHLY
 import com.duckduckgo.subscriptions.impl.SubscriptionsConstants.YEARLY
 import com.duckduckgo.subscriptions.impl.SubscriptionsManager
 import com.duckduckgo.subscriptions.impl.repository.Subscription
+import com.duckduckgo.subscriptions.impl.PrivacyProFeature
+import com.duckduckgo.feature.toggles.api.FakeFeatureToggleFactory
 import kotlinx.coroutines.test.runTest
 import org.json.JSONArray
 import org.json.JSONObject
 import org.junit.Assert.assertEquals
 import org.junit.Assert.assertNull
+import org.junit.Before
 import org.junit.Rule
 import org.junit.Test
 import org.junit.runner.RunWith
@@ -30,11 +33,17 @@ class RealSubscriptionsJSHelperTest {
     var coroutineRule = CoroutineTestRule()
 
     private val mockSubscriptionsManager: SubscriptionsManager = mock()
+    private val privacyProFeature = FakeFeatureToggleFactory.create(PrivacyProFeature::class.java)
 
-    private val testee = RealSubscriptionsJSHelper(mockSubscriptionsManager)
+    private val testee = RealSubscriptionsJSHelper(mockSubscriptionsManager, privacyProFeature)
 
     private val featureName = "subscriptions"
 
+    @Before
+    fun setUp() {
+        // Set up any necessary initializations or mocks
+    }
+
     @Test
     fun whenMethodIsUnknownThenReturnNull() = runTest {
         val method = "unknownMethod"
@@ -211,4 +220,48 @@ class RealSubscriptionsJSHelperTest {
         assertEquals(expected.method, result?.method)
         assertEquals(expected.params.toString(), result?.params.toString())
     }
+
+    @Test
+    fun whenGetFeatureConfigRequestThenReturnJsCallbackDataWithUsePaidDuckAiFlag() = runTest {
+        val method = "getFeatureConfig"
+        val id = "123"
+        val usePaidDuckAi = true
+
+        privacyProFeature.duckAiPlus().setRawStoredState(com.duckduckgo.feature.toggles.api.Toggle.State(usePaidDuckAi))
+
+        val result = testee.processJsCallbackMessage(featureName, method, id, null)
+
+        val jsonPayload = JSONObject().apply {
+            put("usePaidDuckAi", usePaidDuckAi)
+        }
+
+        val expected = JsCallbackData(jsonPayload, featureName, method, id)
+
+        assertEquals(expected.id, result?.id)
+        assertEquals(expected.featureName, result?.featureName)
+        assertEquals(expected.method, result?.method)
+        assertEquals(expected.params.toString(), result?.params.toString())
+    }
+
+    @Test
+    fun whenGetFeatureConfigRequestWithDisabledFlagThenReturnJsCallbackDataWithUsePaidDuckAiFalse() = runTest {
+        val method = "getFeatureConfig"
+        val id = "123"
+        val usePaidDuckAi = false
+
+        privacyProFeature.duckAiPlus().setRawStoredState(com.duckduckgo.feature.toggles.api.Toggle.State(usePaidDuckAi))
+
+        val result = testee.processJsCallbackMessage(featureName, method, id, null)
+
+        val jsonPayload = JSONObject().apply {
+            put("usePaidDuckAi", usePaidDuckAi)
+        }
+
+        val expected = JsCallbackData(jsonPayload, featureName, method, id)
+
+        assertEquals(expected.id, result?.id)
+        assertEquals(expected.featureName, result?.featureName)
+        assertEquals(expected.method, result?.method)
+        assertEquals(expected.params.toString(), result?.params.toString())
+    }
 }