Send termsAccepted on native Duck.ai prompts

duckchat/duckchat-impl/src/main/java/com/duckduckgo/duckchat/impl/helper/DuckChatJSHelper.kt

@@ -47,6 +47,7 @@ import com.duckduckgo.duckchat.impl.models.AIChatAttachmentUsage
 import com.duckduckgo.duckchat.impl.pixel.DuckChatPixelSurface
 import com.duckduckgo.duckchat.impl.pixel.DuckChatPixels
 import com.duckduckgo.duckchat.impl.store.DuckChatDataStore
+import com.duckduckgo.duckchat.impl.terms.DuckAiTermsRepository
 import com.duckduckgo.duckchat.impl.ui.nativeinput.attachment.LimitsHandler
 import com.duckduckgo.duckchat.impl.ui.nativeinput.edit.SubmittedFile
 import com.duckduckgo.duckchat.impl.ui.nativeinput.edit.SubmittedImage
@@ -131,6 +132,7 @@ class RealDuckChatJSHelper @Inject constructor(
     private val editPromptSessionStore: EditPromptSessionStore,
     private val browserInteractionsPlugins: PluginPoint<BrowserInteractionsPlugin>,
     private val duckAiSessionCallback: DuckAiSessionCallback,
+    private val termsRepository: DuckAiTermsRepository,
 ) : DuckChatJSHelper {
 
     private val registerOpenedJob = ConflatedJob()
@@ -174,7 +176,7 @@ class RealDuckChatJSHelper @Inject constructor(
 
             METHOD_GET_AI_CHAT_NATIVE_PROMPT ->
                 id?.let {
-                    getAIChatNativePrompt(featureName, method, it)
+                    getAIChatNativePrompt(featureName, method, it, browserMode)
                 }
 
             METHOD_OPEN_AI_CHAT -> {
@@ -399,6 +401,7 @@ class RealDuckChatJSHelper @Inject constructor(
 
         val params = JSONObject().apply {
             put(PLATFORM, ANDROID)
+            if (duckChatFeature.nativeToSConsent().isEnabled()) put(TERMS_ACCEPTED, true)
             put("tool", "query")
             put(
                 "query",
@@ -509,14 +512,18 @@ class RealDuckChatJSHelper @Inject constructor(
             else -> 5
         }
 
-    private fun getAIChatNativePrompt(
+    private suspend fun getAIChatNativePrompt(
         featureName: String,
         method: String,
         id: String,
+        browserMode: BrowserMode,
     ): JsCallbackData {
         val pending = pendingNativePromptStore.consume()
+        val termsAccepted = pending != null && duckChatFeature.nativeToSConsent().isEnabled()
+        if (termsAccepted) termsRepository.markTermsAccepted(browserMode)
         val jsonPayload = JSONObject().apply {
             put(PLATFORM, ANDROID)
+            if (termsAccepted) put(TERMS_ACCEPTED, true)
             if (pending != null) {
                 put("tool", "query")
                 put(
@@ -775,6 +782,7 @@ class RealDuckChatJSHelper @Inject constructor(
         private const val SUPPORTS_PAGE_CONTEXT = "supportsPageContext"
         private const val SUPPORTS_SUGGESTIONS = "supportsSuggestions"
         private const val SUPPORTS_MULTIPLE_PAGE_CONTEXT = "supportsMultipleContexts"
+        private const val TERMS_ACCEPTED = "termsAccepted"
         private const val SUPPORTS_NATIVE_STORAGE = "supportsNativeStorage"
         private const val SUPPORTS_NATIVE_USAGE_WARNINGS = "supportsNativeUsageWarnings"
         private const val SUPPORTS_SUBSCRIPTION = "supportsSubscription"

duckchat/duckchat-impl/src/main/java/com/duckduckgo/duckchat/impl/terms/DuckAiTermsRepository.kt

@@ -0,0 +1,58 @@
+/*
+ * Copyright (c) 2026 DuckDuckGo
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.duckduckgo.duckchat.impl.terms
+
+import com.duckduckgo.browsermode.api.BrowserMode
+import com.duckduckgo.browsermode.api.BrowserModeDataProvider
+import com.duckduckgo.common.utils.DispatcherProvider
+import com.duckduckgo.di.scopes.AppScope
+import com.duckduckgo.duckchat.store.impl.DuckAiBridgeStorage
+import com.duckduckgo.duckchat.store.impl.store.DuckAiBridgeSettingEntity
+import com.squareup.anvil.annotations.ContributesBinding
+import kotlinx.coroutines.withContext
+import javax.inject.Inject
+
+/**
+ * Duck.ai ToS acceptance, shared with the frontend through the native storage setting it already reads.
+ */
+interface DuckAiTermsRepository {
+    suspend fun hasAcceptedTerms(browserMode: BrowserMode): Boolean
+
+    suspend fun markTermsAccepted(browserMode: BrowserMode)
+}
+
+@ContributesBinding(AppScope::class)
+class RealDuckAiTermsRepository @Inject constructor(
+    private val storageProvider: BrowserModeDataProvider<DuckAiBridgeStorage>,
+    private val dispatchers: DispatcherProvider,
+) : DuckAiTermsRepository {
+
+    override suspend fun hasAcceptedTerms(browserMode: BrowserMode): Boolean = withContext(dispatchers.io()) {
+        storageProvider.forMode(browserMode).settings.get(TERMS_KEY)?.value == ACCEPTED
+    }
+
+    override suspend fun markTermsAccepted(browserMode: BrowserMode) {
+        withContext(dispatchers.io()) {
+            storageProvider.forMode(browserMode).settings.upsert(DuckAiBridgeSettingEntity(key = TERMS_KEY, value = ACCEPTED))
+        }
+    }
+
+    private companion object {
+        const val TERMS_KEY = "duckaiHasAgreedToTerms"
+        const val ACCEPTED = "true"
+    }
+}

duckchat/duckchat-impl/src/test/kotlin/com/duckduckgo/duckchat/impl/helper/RealDuckChatJSHelperTest.kt

@@ -45,6 +45,7 @@ import com.duckduckgo.duckchat.impl.helper.RealDuckChatJSHelper.Companion.DUCK_C
 import com.duckduckgo.duckchat.impl.helper.RealDuckChatJSHelper.Companion.METHOD_GET_PAGE_CONTEXT
 import com.duckduckgo.duckchat.impl.pixel.DuckChatPixels
 import com.duckduckgo.duckchat.impl.store.DuckChatDataStore
+import com.duckduckgo.duckchat.impl.terms.DuckAiTermsRepository
 import com.duckduckgo.duckchat.impl.ui.nativeinput.attachment.LimitsHandler
 import com.duckduckgo.duckchat.impl.ui.nativeinput.edit.SubmittedFile
 import com.duckduckgo.duckchat.impl.ui.nativeinput.edit.SubmittedImage
@@ -121,6 +122,7 @@ class RealDuckChatJSHelperTest {
     private val mockBrowserInteractionsPlugin: BrowserInteractionsPlugin = mock()
     private val mockBrowserInteractionsPlugins: PluginPoint<BrowserInteractionsPlugin> = mock()
     private val mockDuckAiSessionCallback: DuckAiSessionCallback = mock()
+    private val mockTermsRepository: DuckAiTermsRepository = mock()
     private val testee = RealDuckChatJSHelper(
         duckChat = mockDuckChat,
         duckChatPixels = mockDuckChatPixels,
@@ -141,6 +143,7 @@ class RealDuckChatJSHelperTest {
         editPromptSessionStore = mockEditPromptSessionStore,
         browserInteractionsPlugins = mockBrowserInteractionsPlugins,
         duckAiSessionCallback = mockDuckAiSessionCallback,
+        termsRepository = mockTermsRepository,
     )
 
     init {
@@ -2490,6 +2493,39 @@ class RealDuckChatJSHelperTest {
         assertFalse(query.has("reasoningEffort"))
     }
 
+    @Test
+    fun whenGetAIChatNativePromptWithPendingPromptAndToSFlagOnThenTermsAcceptedAndStored() = runTest {
+        mockDuckChatFeature.nativeToSConsent().setRawStoredState(Toggle.State(enable = true))
+        whenever(mockPendingNativePromptStore.consume()).thenReturn(PendingNativePrompt("test prompt", null, null))
+
+        val result = testee.processJsCallbackMessage("aiChat", "getAIChatNativePrompt", "123", null, pageContext = viewModel.updatedPageContext)
+
+        assertTrue(result!!.params.getBoolean("termsAccepted"))
+        verify(mockTermsRepository).markTermsAccepted(any())
+    }
+
+    @Test
+    fun whenGetAIChatNativePromptWithPendingPromptAndToSFlagOffThenTermsAcceptedOmitted() = runTest {
+        mockDuckChatFeature.nativeToSConsent().setRawStoredState(Toggle.State(enable = false))
+        whenever(mockPendingNativePromptStore.consume()).thenReturn(PendingNativePrompt("test prompt", null, null))
+
+        val result = testee.processJsCallbackMessage("aiChat", "getAIChatNativePrompt", "123", null, pageContext = viewModel.updatedPageContext)
+
+        assertFalse(result!!.params.has("termsAccepted"))
+        verify(mockTermsRepository, never()).markTermsAccepted(any())
+    }
+
+    @Test
+    fun whenGetAIChatNativePromptWithNoPendingPromptAndToSFlagOnThenTermsAcceptedOmitted() = runTest {
+        mockDuckChatFeature.nativeToSConsent().setRawStoredState(Toggle.State(enable = true))
+        whenever(mockPendingNativePromptStore.consume()).thenReturn(null)
+
+        val result = testee.processJsCallbackMessage("aiChat", "getAIChatNativePrompt", "123", null, pageContext = viewModel.updatedPageContext)
+
+        assertFalse(result!!.params.has("termsAccepted"))
+        verify(mockTermsRepository, never()).markTermsAccepted(any())
+    }
+
     @Test
     fun whenGetAIChatNativePromptWithPendingPromptButNoModelIdThenModelIdOmitted() = runTest {
         val pending = PendingNativePrompt("test prompt", null, null)