Отправка гиф

client-mobile/auth/presentation/AuthViewModel.kt

@@ -9,6 +9,7 @@ import kotlinx.coroutines.flow.StateFlow
 import kotlinx.coroutines.flow.asStateFlow
 import kotlinx.coroutines.flow.update
 import kotlinx.coroutines.launch
+import com.google.firebase.messaging.FirebaseMessaging
 import javax.inject.Inject
 
 data class AuthState(
@@ -35,6 +36,7 @@ class AuthViewModel @Inject constructor(
             repository.login(userName, password)
                 .onSuccess {
                     _state.update { it.copy(isLoading = false, isAuthenticated = true) }
+                    updatePushToken()
                 }
                 .onFailure { e ->
                     _state.update { it.copy(isLoading = false, error = e.message) }
@@ -48,10 +50,22 @@ class AuthViewModel @Inject constructor(
             repository.register(userName, password)
                 .onSuccess {
                     _state.update { it.copy(isLoading = false, isAuthenticated = true) }
+                    updatePushToken()
                 }
                 .onFailure { e ->
                     _state.update { it.copy(isLoading = false, error = e.message) }
                 }
         }
     }
+
+    private fun updatePushToken() {
+        FirebaseMessaging.getInstance().token.addOnCompleteListener { task ->
+            if (task.isSuccessful) {
+                val token = task.result
+                viewModelScope.launch {
+                    repository.updatePushToken(token)
+                }
+            }
+        }
+    }
 }