Set FCM token to user's profile

Author: MarkNjunge

app/src/main/java/com/marknkamau/justjava/JustJavaApp.kt

@@ -15,6 +15,7 @@ import io.fabric.sdk.android.Fabric
 import com.crashlytics.android.Crashlytics
 import com.marknjunge.core.data.firebase.ClientDatabaseImpl
 import com.marknjunge.core.data.firebase.ClientDatabaseService
+import com.marknjunge.core.data.firebase.WriteListener
 import com.marknjunge.core.mpesa.MpesaInteractor
 import com.marknjunge.core.mpesa.MpesaInteractorImpl
 import com.squareup.leakcanary.LeakCanary
@@ -56,6 +57,21 @@ class JustJavaApp : Application() {
         notificationHelper = NotificationHelper(this)
 
         broadcastManager = androidx.localbroadcastmanager.content.LocalBroadcastManager.getInstance(this)
+
+        if (authService.isSignedIn()) {
+            val user = authService.getCurrentUser()
+
+            databaseService.updateUserFcmToken(user.userId, object : WriteListener {
+                override fun onError(reason: String) {
+                    Timber.e(reason)
+                }
+
+                override fun onSuccess() {
+                    Timber.i("FCM token saved")
+                }
+
+            })
+        }
     }
 
 }

app/src/main/java/com/marknkamau/justjava/data/network/MyFirebaseMessagingService.kt

@@ -3,6 +3,7 @@ package com.marknkamau.justjava.data.network
 import android.content.Intent
 import com.google.firebase.messaging.FirebaseMessagingService
 import com.google.firebase.messaging.RemoteMessage
+import com.marknjunge.core.data.firebase.WriteListener
 import com.marknkamau.justjava.JustJavaApp
 import timber.log.Timber
 
@@ -15,6 +16,8 @@ import timber.log.Timber
 class MyFirebaseMessagingService : FirebaseMessagingService() {
     private val notificationHelper by lazy { (application as JustJavaApp).notificationHelper }
     private val broadcastManager by lazy { (application as JustJavaApp).broadcastManager }
+    private val authService by lazy { (application as JustJavaApp).authService }
+    private val databaseService by lazy { (application as JustJavaApp).databaseService }
 
     companion object {
         const val MPESA_ORDER_PAID_ACTION = "mpesa"
@@ -54,4 +57,21 @@ class MyFirebaseMessagingService : FirebaseMessagingService() {
             }
         }
     }
+
+    override fun onNewToken(token: String) {
+        if (authService.isSignedIn()){
+            val user = authService.getCurrentUser()
+
+            databaseService.updateUserFcmToken(user.userId, object : WriteListener {
+                override fun onError(reason: String) {
+                    Timber.e(reason)
+                }
+
+                override fun onSuccess() {
+                    Timber.i("FCM token saved")
+                }
+
+            })
+        }
+    }
 }

app/src/main/java/com/marknkamau/justjava/ui/signup/SignUpPresenter.kt

@@ -5,6 +5,7 @@ import com.marknjunge.core.data.firebase.ClientDatabaseService
 import com.marknjunge.core.data.firebase.WriteListener
 import com.marknkamau.justjava.data.local.PreferencesRepository
 import com.marknjunge.core.model.UserDetails
+import timber.log.Timber
 
 internal class SignUpPresenter(private val activityView: SignUpView,
                                private val preferences: PreferencesRepository,
@@ -50,6 +51,7 @@ internal class SignUpPresenter(private val activityView: SignUpView,
                         preferences.saveUserDetails(userDetails)
                         activityView.displayMessage("Sign up successfully")
                         activityView.finishActivity()
+                        setFcmToken()
                     }
 
                     override fun onError(reason: String) {
@@ -66,4 +68,16 @@ internal class SignUpPresenter(private val activityView: SignUpView,
             }
         })
     }
+
+    private fun setFcmToken(){
+        database.updateUserFcmToken(auth.getCurrentUser().userId, object : WriteListener {
+            override fun onError(reason: String) {
+                Timber.e(reason)
+            }
+
+            override fun onSuccess() {
+                Timber.i("FCM token saved")
+            }
+        })
+    }
 }

core/src/main/java/com/marknjunge/core/data/firebase/ClientDatabaseImpl.kt

@@ -37,6 +37,25 @@ class ClientDatabaseImpl : ClientDatabaseService {
                 }
     }
 
+    override fun updateUserFcmToken(userId: String, listener: WriteListener) {
+        FirebaseInstanceId.getInstance().instanceId
+                .addOnSuccessListener { instanceIdResult ->
+                    val data = mapOf(
+                            DatabaseKeys.User.fcmToken to instanceIdResult.token
+                    )
+                    firestore.collection("users")
+                            .document(userId)
+                            .update(data)
+                            .addOnSuccessListener {
+                                listener.onSuccess()
+                            }
+                            .addOnFailureListener {
+                                listener.onError(it.message ?: "Unable tp update user token")
+                            }
+                }
+                .addOnFailureListener { listener.onError(it.message ?: "Unable to retrieve user token") }
+    }
+
     override fun getUserDefaults(id: String, listener: ClientDatabaseService.UserDetailsListener) {
         firestore.collection("users")
                 .document(id)

core/src/main/java/com/marknjunge/core/data/firebase/ClientDatabaseService.kt

@@ -9,6 +9,8 @@ interface ClientDatabaseService {
 
     fun updateUserDetails(id: String, name: String, phone: String, address: String, listener: WriteListener)
 
+    fun updateUserFcmToken(userId: String, listener: WriteListener)
+
     fun getUserDefaults(id: String, listener: UserDetailsListener)
 
     fun placeNewOrder(order: Order, orderItems: List<OrderItem>, listener: WriteListener)

core/src/main/java/com/marknjunge/core/model/DatabaseKeys.kt

@@ -13,6 +13,7 @@ object DatabaseKeys {
         const val email = "email"
         const val phone = "phone"
         const val address = "address"
+        const val fcmToken = "fcmToken"
     }
 
     object Order {

justjavastaff/src/main/java/com/marknkamau/justjavastaff/ui/login/LoginPresenter.kt

@@ -1,6 +1,9 @@
 package com.marknkamau.justjavastaff.ui.login
 
 import com.marknjunge.core.auth.AuthService
+import com.marknjunge.core.data.firebase.ClientDatabaseService
+import com.marknjunge.core.data.firebase.WriteListener
+import timber.log.Timber
 import java.util.regex.Pattern
 
 /**
@@ -9,7 +12,7 @@ import java.util.regex.Pattern
  * https://github.com/MarkNjunge
  */
 
-class LoginPresenter(private val auth: AuthService, private val view: LoginView) {
+class LoginPresenter(private val auth: AuthService, private val databaseService: ClientDatabaseService,private val view: LoginView) {
 
     fun signIn(email: String, password: String) {
         val pattern = Pattern.compile("^([a-zA-Z0-9_.-])+@justjava.com+")
@@ -23,6 +26,7 @@ class LoginPresenter(private val auth: AuthService, private val view: LoginView)
         auth.signIn(email, password, object : AuthService.AuthActionListener {
             override fun actionSuccessful(response: String) {
                 view.onSignedIn()
+                setFcmToken()
             }
 
             override fun actionFailed(response: String) {
@@ -31,4 +35,16 @@ class LoginPresenter(private val auth: AuthService, private val view: LoginView)
         })
     }
 
+    private fun setFcmToken(){
+        databaseService.updateUserFcmToken(auth.getCurrentUser().userId, object : WriteListener {
+            override fun onError(reason: String) {
+                Timber.e(reason)
+            }
+
+            override fun onSuccess() {
+                Timber.i("FCM token saved")
+            }
+        })
+    }
+
 }