Fix a screen capture bug that clears the setting when the app language changes (#3372)

app/src/main/java/com/x8bit/bitwarden/MainActivity.kt

@@ -66,6 +66,7 @@ class MainActivity : AppCompatActivity() {
         }
         setContent {
             val state by mainViewModel.stateFlow.collectAsStateWithLifecycle()
+            updateScreenCapture(isScreenCaptureAllowed = state.isScreenCaptureAllowed)
             LocalManagerProvider {
                 BitwardenTheme(theme = state.theme) {
                     RootNavScreen(
@@ -97,15 +98,8 @@ class MainActivity : AppCompatActivity() {
             .eventFlow
             .onEach { event ->
                 when (event) {
-                    is MainEvent.CompleteAutofill -> {
-                        handleCompleteAutofill(event)
-                    }
-
+                    is MainEvent.CompleteAutofill -> handleCompleteAutofill(event)
                     MainEvent.Recreate -> handleRecreate()
-
-                    is MainEvent.ScreenCaptureSettingChange -> {
-                        handleScreenCaptureSettingChange(event)
-                    }
                 }
             }
             .launchIn(lifecycleScope)
@@ -118,15 +112,15 @@ class MainActivity : AppCompatActivity() {
         )
     }
 
-    private fun handleScreenCaptureSettingChange(event: MainEvent.ScreenCaptureSettingChange) {
-        if (event.isAllowed) {
+    private fun handleRecreate() {
+        recreate()
+    }
+
+    private fun updateScreenCapture(isScreenCaptureAllowed: Boolean) {
+        if (isScreenCaptureAllowed) {
             window.clearFlags(WindowManager.LayoutParams.FLAG_SECURE)
         } else {
             window.addFlags(WindowManager.LayoutParams.FLAG_SECURE)
         }
     }
-
-    private fun handleRecreate() {
-        recreate()
-    }
 }

app/src/main/java/com/x8bit/bitwarden/MainViewModel.kt

@@ -50,8 +50,9 @@ class MainViewModel @Inject constructor(
     private val authRepository: AuthRepository,
     private val savedStateHandle: SavedStateHandle,
 ) : BaseViewModel<MainState, MainEvent, MainAction>(
-    MainState(
+    initialState = MainState(
         theme = settingsRepository.appTheme,
+        isScreenCaptureAllowed = settingsRepository.isScreenCaptureAllowed,
     ),
 ) {
     private var specialCircumstance: SpecialCircumstance?
@@ -81,9 +82,8 @@ class MainViewModel @Inject constructor(
 
         settingsRepository
             .isScreenCaptureAllowedStateFlow
-            .onEach { isAllowed ->
-                sendEvent(MainEvent.ScreenCaptureSettingChange(isAllowed))
-            }
+            .map { MainAction.Internal.ScreenCaptureUpdate(it) }
+            .onEach(::trySendAction)
             .launchIn(viewModelScope)
 
         authRepository
@@ -124,6 +124,7 @@ class MainViewModel @Inject constructor(
             }
 
             is MainAction.Internal.CurrentUserStateChange -> handleCurrentUserStateChange()
+            is MainAction.Internal.ScreenCaptureUpdate -> handleScreenCaptureUpdate(action)
             is MainAction.Internal.ThemeUpdate -> handleAppThemeUpdated(action)
             is MainAction.Internal.VaultUnlockStateChange -> handleVaultUnlockStateChange()
             is MainAction.ReceiveFirstIntent -> handleFirstIntentReceived(action)
@@ -141,6 +142,10 @@ class MainViewModel @Inject constructor(
         recreateUiAndGarbageCollect()
     }
 
+    private fun handleScreenCaptureUpdate(action: MainAction.Internal.ScreenCaptureUpdate) {
+        mutableStateFlow.update { it.copy(isScreenCaptureAllowed = action.isScreenCaptureEnabled) }
+    }
+
     private fun handleAppThemeUpdated(action: MainAction.Internal.ThemeUpdate) {
         mutableStateFlow.update { it.copy(theme = action.theme) }
     }
@@ -249,6 +254,7 @@ class MainViewModel @Inject constructor(
 @Parcelize
 data class MainState(
     val theme: AppTheme,
+    val isScreenCaptureAllowed: Boolean,
 ) : Parcelable
 
 /**
@@ -281,6 +287,13 @@ sealed class MainAction {
          */
         data object CurrentUserStateChange : Internal()
 
+        /**
+         * Indicates that the screen capture state has changed.
+         */
+        data class ScreenCaptureUpdate(
+            val isScreenCaptureEnabled: Boolean,
+        ) : Internal()
+
         /**
          * Indicates that the app theme has changed.
          */
@@ -305,11 +318,6 @@ sealed class MainEvent {
      */
     data class CompleteAutofill(val cipherView: CipherView) : MainEvent()
 
-    /**
-     * Event indicating a change in the screen capture setting.
-     */
-    data class ScreenCaptureSettingChange(val isAllowed: Boolean) : MainEvent()
-
     /**
      * Event indicating that the UI should recreate itself.
      */

app/src/test/java/com/x8bit/bitwarden/MainViewModelTest.kt

@@ -59,6 +59,7 @@ class MainViewModelTest : BaseViewModelTest() {
     private val settingsRepository = mockk<SettingsRepository> {
         every { appTheme } returns AppTheme.DEFAULT
         every { appThemeStateFlow } returns mutableAppThemeFlow
+        every { isScreenCaptureAllowed } returns true
         every { isScreenCaptureAllowedStateFlow } returns mutableScreenCaptureAllowedFlow
     }
     private val authRepository = mockk<AuthRepository> {
@@ -129,9 +130,6 @@ class MainViewModelTest : BaseViewModelTest() {
         val viewModel = createViewModel()
 
         viewModel.eventFlow.test {
-            // Ignore initial screen capture event
-            awaitItem()
-
             mutableUserStateFlow.value = UserState(
                 activeUserId = userId1,
                 accounts = listOf(
@@ -179,9 +177,6 @@ class MainViewModelTest : BaseViewModelTest() {
             val viewModel = createViewModel()
 
             viewModel.eventFlow.test {
-                // Ignore initial screen capture event
-                awaitItem()
-
                 mutableVaultStateEventFlow.tryEmit(VaultStateEvent.Unlocked(userId = "userId"))
                 expectNoEvents()
 
@@ -198,9 +193,6 @@ class MainViewModelTest : BaseViewModelTest() {
         val viewModel = createViewModel()
         val cipherView = mockk<CipherView>()
         viewModel.eventFlow.test {
-            // Ignore initial screen capture event
-            awaitItem()
-
             autofillSelectionManager.emitAutofillSelection(cipherView = cipherView)
             assertEquals(
                 MainEvent.CompleteAutofill(cipherView = cipherView),
@@ -229,9 +221,7 @@ class MainViewModelTest : BaseViewModelTest() {
         val viewModel = createViewModel()
 
         assertEquals(
-            MainState(
-                theme = AppTheme.DEFAULT,
-            ),
+            DEFAULT_STATE,
             viewModel.stateFlow.value,
         )
         viewModel.trySendAction(
@@ -240,7 +230,7 @@ class MainViewModelTest : BaseViewModelTest() {
             ),
         )
         assertEquals(
-            MainState(
+            DEFAULT_STATE.copy(
                 theme = AppTheme.DARK,
             ),
             viewModel.stateFlow.value,
@@ -623,25 +613,19 @@ class MainViewModelTest : BaseViewModelTest() {
         )
     }
 
-    @Suppress("MaxLineLength")
     @Test
-    fun `changes in the allowed screen capture value should result in emissions of ScreenCaptureSettingChange `() =
-        runTest {
-            val viewModel = createViewModel()
+    fun `changes in the allowed screen capture value should update the state`() {
+        val viewModel = createViewModel()
 
-            viewModel.eventFlow.test {
-                assertEquals(
-                    MainEvent.ScreenCaptureSettingChange(isAllowed = true),
-                    awaitItem(),
-                )
+        assertEquals(DEFAULT_STATE, viewModel.stateFlow.value)
 
-                mutableScreenCaptureAllowedFlow.value = false
-                assertEquals(
-                    MainEvent.ScreenCaptureSettingChange(isAllowed = false),
-                    awaitItem(),
-                )
-            }
-        }
+        mutableScreenCaptureAllowedFlow.value = false
+
+        assertEquals(
+            DEFAULT_STATE.copy(isScreenCaptureAllowed = false),
+            viewModel.stateFlow.value,
+        )
+    }
 
     private fun createViewModel(
         initialSpecialCircumstance: SpecialCircumstance? = null,
@@ -659,6 +643,11 @@ class MainViewModelTest : BaseViewModelTest() {
     )
 }
 
+private val DEFAULT_STATE: MainState = MainState(
+    theme = AppTheme.DEFAULT,
+    isScreenCaptureAllowed = true,
+)
+
 private const val SPECIAL_CIRCUMSTANCE_KEY: String = "special-circumstance"
 private val DEFAULT_ACCOUNT = UserState.Account(
     userId = "activeUserId",