Add support for a custom Firebase messaging service

Author: vbabenkoru

README.md

@@ -34,7 +34,29 @@ compile 'com.iterable:iterableapi:3.0.5'
 compile 'com.google.firebase:firebase-messaging:X.X.X' // Min version 9.0.0
 ```
 
-See [Bintray](https://bintray.com/davidtruong/maven/Iterable-SDK) for the latest version of the Iterable Android SDK. 
+See [Bintray](https://bintray.com/davidtruong/maven/Iterable-SDK) for the latest version of the Iterable Android SDK.
+
+#### Handling Firebase push messages and tokens
+
+The SDK adds a FirebaseMessagingService and FirebaseInstanceIdService to the app manifest automatically, so you don't have to do any extra setup to handle incoming push messages.
+If your application implements its own FirebaseMessagingService, make sure you forward `onMessageReceived` and `onNewToken` calls to `IterableFirebaseMessagingService.handleMessageReceived` and `IterableFirebaseInstanceIDService.handleTokenRefresh`, respectively:
+
+```java
+public class MyFirebaseMessagingService extends FirebaseMessagingService {
+
+    @Override
+    public void onMessageReceived(RemoteMessage remoteMessage) {
+        IterableFirebaseMessagingService.handleMessageReceived(this, remoteMessage);
+    }
+
+    @Override
+    public void onNewToken(String s) {
+        IterableFirebaseInstanceIDService.handleTokenRefresh();
+    }
+}
+```
+
+Note that `FirebaseInstanceIdService` is deprecated and replaced with `onNewToken` in recent versions of Firebase.
 
 # Using the SDK
 

iterableapi/src/main/java/com/iterable/iterableapi/IterableFirebaseInstanceIDService.java

@@ -18,6 +18,14 @@ public class IterableFirebaseInstanceIDService extends FirebaseInstanceIdService
 
     @Override
     public void onTokenRefresh() {
+        handleTokenRefresh();
+    }
+
+    /**
+     * Handles token refresh
+     * Call this from a custom {@link FirebaseInstanceIdService} to register the new token with Iterable
+     */
+    public static void handleTokenRefresh() {
         String registrationToken = FirebaseInstanceId.getInstance().getToken();
         IterableLogger.d(TAG, "New Firebase Token generated: " + registrationToken);
         IterableApi.getInstance().registerForPush();

iterableapi/src/main/java/com/iterable/iterableapi/IterableFirebaseMessagingService.java

@@ -1,9 +1,6 @@
 package com.iterable.iterableapi;
 
-import android.content.ComponentName;
 import android.content.Context;
-import android.content.Intent;
-import android.content.pm.PackageManager;
 import android.os.AsyncTask;
 import android.os.Bundle;
 
@@ -18,40 +15,54 @@ public class IterableFirebaseMessagingService extends FirebaseMessagingService {
 
     @Override
     public void onMessageReceived(RemoteMessage remoteMessage) {
-        if (remoteMessage.getData().size() > 0) {
-            Map<String,String> messageData = remoteMessage.getData();
-            handlePushReceived(messageData);
-            IterableLogger.d(TAG, "Message data payload: " + remoteMessage.getData());
+        handleMessageReceived(this, remoteMessage);
+    }
+
+    /**
+     * Handles receiving an incoming push notification from the intent.
+     *
+     * Call this from a custom {@link FirebaseMessagingService} to pass Iterable push messages to
+     * Iterable SDK for tracking and rendering
+     * @param remoteMessage Remote message received from Firebase in
+     *        {@link FirebaseMessagingService#onMessageReceived(RemoteMessage)}
+     * @return Boolean indicating whether it was an Iterable message or not
+     */
+    public static boolean handleMessageReceived(Context context, RemoteMessage remoteMessage) {
+        Map<String,String> messageData = remoteMessage.getData();
+
+        if (messageData == null || messageData.size() == 0) {
+            return false;
         }
 
+        IterableLogger.d(TAG, "Message data payload: " + remoteMessage.getData());
         // Check if message contains a notification payload.
         if (remoteMessage.getNotification() != null) {
             IterableLogger.d(TAG, "Message Notification Body: " + remoteMessage.getNotification().getBody());
         }
-    }
 
-    /**
-     * Handles receiving an incoming push notification from the intent.
-     * @param messageData
-     */
-    private void handlePushReceived(Map<String,String> messageData) {
         Bundle extras = new Bundle();
         for (Map.Entry<String, String> entry : messageData.entrySet()) {
             extras.putString(entry.getKey(), entry.getValue());
         }
 
+        if (!IterableNotificationBuilder.isIterablePush(extras)) {
+            IterableLogger.d(TAG, "Not an Iterable push message");
+            return false;
+        }
+
         if (!IterableNotificationBuilder.isGhostPush(extras)) {
             if (!IterableNotificationBuilder.isEmptyBody(extras)) {
                 IterableLogger.d(TAG, "Iterable push received " + messageData);
                 IterableNotificationBuilder notificationBuilder = IterableNotificationBuilder.createNotification(
-                        getApplicationContext(), extras);
+                        context.getApplicationContext(), extras);
                 new IterableNotificationManager().execute(notificationBuilder);
             } else {
                 IterableLogger.d(TAG, "Iterable OS notification push received");
             }
         } else {
             IterableLogger.d(TAG, "Iterable ghost silent push received");
         }
+        return true;
     }
 }
 

iterableapi/src/main/java/com/iterable/iterableapi/IterableNotificationBuilder.java

@@ -376,6 +376,10 @@ private static int getIconId(Context context) {
         return iconId;
     }
 
+    static boolean isIterablePush(Bundle extras) {
+        return extras != null && extras.containsKey(IterableConstants.ITERABLE_DATA_KEY);
+    }
+
     /**
      * Returns if the given notification is a ghost/silent push notification
      *