Bug 2021407 - Adds shared prefs implementation of summarization settings

Author: boek

mobile/android/android-components/.buildconfig.yml

@@ -1408,14 +1408,18 @@ projects:
     path: components/feature/summarize
     publish: true
     upstream_dependencies:
+    - components:browser-errorpages
     - components:compose-base
     - components:concept-base
+    - components:concept-engine
     - components:concept-fetch
     - components:concept-llm
+    - components:concept-storage
     - components:lib-publicsuffixlist
     - components:lib-state
     - components:support-base
     - components:support-ktx
+    - components:support-test-fakes
     - components:support-utils
     - components:tooling-lint
     - components:ui-colors

mobile/android/android-components/components/feature/summarize/build.gradle

@@ -24,9 +24,9 @@ android {
 }
 
 dependencies {
+    implementation project(':components:compose-base')
     implementation project(':components:concept-llm')
     implementation project(':components:lib-state')
-    implementation project(':components:compose-base')
     implementation project(':components:ui-icons')
     implementation project(':components:ui-richtext')
 
@@ -35,10 +35,13 @@ dependencies {
     implementation libs.androidx.compose.ui.graphics
     implementation libs.androidx.compose.ui.tooling.preview
     implementation libs.androidx.compose.material3
+    implementation libs.androidx.core.ktx
     debugImplementation libs.androidx.compose.ui.tooling
 
     testImplementation libs.androidx.test.junit
     testImplementation libs.kotlinx.coroutines.test
+
+    testImplementation project(':components:support-test-fakes')
 }
 
 apply from: '../../../common-config.gradle'

mobile/android/android-components/components/feature/summarize/src/main/java/mozilla/components/feature/summarize/SummarizationSettings.kt

@@ -4,6 +4,11 @@
 
 package mozilla.components.feature.summarize
 
+import android.content.Context
+import android.content.Context.MODE_PRIVATE
+import android.content.SharedPreferences
+import androidx.core.content.edit
+
 /**
  * Defines the contract for persisting and retrieving user settings related to
  * the summarization feature.
@@ -53,5 +58,34 @@ interface SummarizationSettings {
                 hasConsentedToShake = newValue
             }
         }
+
+        /**
+         * Creates am implementation of [SummarizationSettings] that is backed by [SharedPreferences].
+         *
+         * @param context required to create [SharedPreferences].
+         * @return An in-memory [SummarizationSettings] instance.
+         */
+        fun sharedPrefs(
+            context: Context,
+        ): SummarizationSettings {
+            val prefs = context.getSharedPreferences("summarization_prefs", MODE_PRIVATE)
+            return SharedPrefsBackedSummarizationSettings(prefs)
+        }
+    }
+}
+
+internal class SharedPrefsBackedSummarizationSettings(
+    private val prefs: SharedPreferences,
+) : SummarizationSettings {
+    override suspend fun getHasConsentedToShake(): Boolean {
+        return prefs.getBoolean(HAS_CONSENT_TO_SHAKE, false)
+    }
+
+    override suspend fun setHasConsentedToShake(newValue: Boolean) {
+        prefs.edit { putBoolean(HAS_CONSENT_TO_SHAKE, newValue) }
+    }
+
+    companion object {
+        const val HAS_CONSENT_TO_SHAKE = "has_consent_to_shake"
     }
 }

mobile/android/android-components/components/feature/summarize/src/test/java/mozilla/components/feature/summarize/SummarizationSettingsTest.kt

@@ -0,0 +1,37 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+package mozilla.components.feature.summarize
+
+import kotlinx.coroutines.flow.toList
+import kotlinx.coroutines.launch
+import kotlinx.coroutines.test.runTest
+import mozilla.components.feature.summarize.SummarizationState.Finished
+import mozilla.components.feature.summarize.SummarizationState.Inert
+import mozilla.components.feature.summarize.SummarizationState.ShakeConsentRequired
+import mozilla.components.feature.summarize.SummarizationState.Summarized
+import mozilla.components.feature.summarize.SummarizationState.Summarizing
+import mozilla.components.feature.summarize.fakes.FakeCloudProvider
+import mozilla.components.feature.summarize.fakes.FakeLlm
+import mozilla.components.support.test.fakes.android.FakeSharedPreferences
+import org.junit.Assert.assertEquals
+import org.junit.Assert.assertFalse
+import org.junit.Assert.assertTrue
+import org.junit.Test
+import kotlin.time.Duration.Companion.seconds
+
+class SummarizationSettingsTest {
+
+    @Test
+    fun `test that we can set and retrieve that the user has consented to shake`() = runTest {
+        val prefs = FakeSharedPreferences()
+        val settings = SharedPrefsBackedSummarizationSettings(prefs)
+
+        assertFalse(settings.getHasConsentedToShake())
+
+        settings.setHasConsentedToShake(true)
+
+        assertTrue(settings.getHasConsentedToShake())
+    }
+}

mobile/android/fenix/app/src/main/java/org/mozilla/fenix/summarization/SummarizationFragment.kt

@@ -55,7 +55,7 @@ class SummarizationFragment : BottomSheetDialogFragment() {
         SummarizationStoreViewModel.factory(
             initializedFromShake = args.fromShake,
             llmProvider = provider,
-            settings = SummarizationSettings.inMemory(),
+            settings = SummarizationSettings.sharedPrefs(requireContext()),
             pageContentExtractor = engineSession.asPageContentExtractor(),
         )
     }