Wire up Android setTabBadge

android/app/src/main/java/com/reactnativenavigation/activities/BaseReactActivity.java

@@ -46,6 +46,7 @@ import javax.annotation.Nullable;
 public abstract class BaseReactActivity extends AppCompatActivity implements DefaultHardwareBackBtnHandler {
 
     protected static final String KEY_ANIMATED = "animated";
+    protected static final String KEY_BADGE = "badge";
     protected static final String KEY_HIDDEN = "hidden";
     protected static final String KEY_TAB_INDEX = "tabIndex";
     protected static final String KEY_TITLE = "title";
@@ -370,6 +371,10 @@ public abstract class BaseReactActivity extends AppCompatActivity implements Def
         mToolbar.setTitle(title.getString(KEY_TITLE));
     }
 
+    public void setTabBadge(ReadableMap params) {
+        return;
+    }
+
     public void switchToTab(ReadableMap params) {
         return;
     }

android/app/src/main/java/com/reactnativenavigation/activities/BottomTabActivity.java

@@ -190,6 +190,21 @@ public class BottomTabActivity extends BaseReactActivity implements AHBottomNavi
         }
     }
 
+    @Override
+    public void setTabBadge(ReadableMap params) {
+        // Badge comes across as int, but if it's 0 clear the notification
+        int badgeCount = params.getInt(KEY_BADGE);
+        String badge = (badgeCount > 0) ? Integer.toString(badgeCount) : "";
+
+        // Tab index is optional, so default to current tab
+        int tabIndex = mBottomNavigation.getCurrentItem();
+        if (params.hasKey(KEY_TAB_INDEX)) {
+            tabIndex = params.getInt(KEY_TAB_INDEX);
+        }
+
+        mBottomNavigation.setNotification(badge, tabIndex);
+    }
+
     @Override
     public void switchToTab(ReadableMap params) {
         int tabIndex = params.getInt(KEY_TAB_INDEX);

android/app/src/main/java/com/reactnativenavigation/modules/RctActivityModule.java

@@ -120,6 +120,20 @@ public class RctActivityModule extends ReactContextBaseJavaModule {
         });
     }
 
+    @ReactMethod
+    public void setTabBadge(final ReadableMap params) {
+        final BaseReactActivity context = ContextProvider.getActivityContext();
+        if (context == null || context.isFinishing()) {
+            return;
+        }
+        context.runOnUiThread(new Runnable() {
+            @Override
+            public void run() {
+                context.setTabBadge(params);
+            }
+        });
+    }
+
     @ReactMethod
     public void switchToTab(final ReadableMap params) {
         final BaseReactActivity context = ContextProvider.getActivityContext();

src/platformSpecific.android.js

@@ -84,6 +84,13 @@ function navigatorResetTo(navigator, params) {
   RctActivity.navigatorResetTo(params);
 }
 
+function navigatorSetTabBadge(navigator, params) {
+  RctActivity.setTabBadge({
+    tabIndex: params.tabIndex,
+    badge: params.badge
+  });
+}
+
 function navigatorSetTitle(navigator, params) {
   RctActivity.setNavigatorTitle(params);
 }
@@ -157,6 +164,7 @@ export default {
   dismissModal,
   dismissAllModals,
   navigatorSetButtons,
+  navigatorSetTabBadge,
   navigatorSetTitle,
   navigatorSwitchToTab,
   navigatorToggleTabs