From 282fe84b9fbf8e003fd0a4bc5d427fc49c28b881 Mon Sep 17 00:00:00 2001
From: Oleg Semenenko <146032743+oleg-livefront@users.noreply.github.com>
Date: Tue, 23 Jan 2024 21:55:08 -0600
Subject: [PATCH] Updating getAuthCodesFlow to update values on emission.
 (#741)

---
 .../bitwarden/data/vault/manager/TotpCodeManagerImpl.kt     | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/app/src/main/java/com/x8bit/bitwarden/data/vault/manager/TotpCodeManagerImpl.kt b/app/src/main/java/com/x8bit/bitwarden/data/vault/manager/TotpCodeManagerImpl.kt
index e28ad5fd33..dbc7c5c45f 100644
--- a/app/src/main/java/com/x8bit/bitwarden/data/vault/manager/TotpCodeManagerImpl.kt
+++ b/app/src/main/java/com/x8bit/bitwarden/data/vault/manager/TotpCodeManagerImpl.kt
@@ -32,7 +32,7 @@ class TotpCodeManagerImpl(
     private val unconfinedScope = CoroutineScope(dispatcherManager.unconfined)
 
     private val mutableVerificationCodeStateFlowMap =
-        mutableMapOf<String, StateFlow<DataState<VerificationCodeItem?>>>()
+        mutableMapOf<CipherView, StateFlow<DataState<VerificationCodeItem?>>>()
 
     override fun getTotpCodesStateFlow(
         userId: String,
@@ -78,7 +78,7 @@ class TotpCodeManagerImpl(
     ): StateFlow<DataState<VerificationCodeItem?>> {
         val cipherId = cipher.id ?: return MutableStateFlow(DataState.Loaded(null))
 
-        return mutableVerificationCodeStateFlowMap.getOrPut(cipherId) {
+        return mutableVerificationCodeStateFlowMap.getOrPut(cipher) {
             flow<DataState<VerificationCodeItem?>> {
 
                 val totpCode = cipher
@@ -134,7 +134,7 @@ class TotpCodeManagerImpl(
                 }
             }
                 .onCompletion {
-                    mutableVerificationCodeStateFlowMap.remove(cipherId)
+                    mutableVerificationCodeStateFlowMap.remove(cipher)
                 }
                 .stateIn(
                     scope = unconfinedScope,