Implement setTabBadge in Android

android/app/src/main/java/com/reactnativenavigation/bridge/NavigationReactModule.java

@@ -72,7 +72,13 @@ public class NavigationReactModule extends ReactContextBaseJavaModule {
     }
 
     @ReactMethod
-    public void setTabBadge(final ReadableMap params) {
+    public void setBottomTabBadgeByIndex(Integer index, String badge) {
+        NavigationCommandsHandler.setBottomTabBadgeByIndex(index, badge);
+    }
+
+    @ReactMethod
+    public void setBottomTabBadgeByNavigatorId(String navigatorId, String badge) {
+        NavigationCommandsHandler.setBottomTabBadgeByNavigatorId(navigatorId, badge);
     }
 
     @ReactMethod

android/app/src/main/java/com/reactnativenavigation/controllers/NavigationActivity.java

@@ -224,4 +224,16 @@ public class NavigationActivity extends AppCompatActivity implements DefaultHard
             ((BottomTabsLayout) layout).selectBottomTabByNavigatorId(navigatorId);
         }
     }
+
+    public void setBottomTabBadgeByIndex(Integer index, String badge) {
+        if (layout instanceof BottomTabsLayout) {
+            ((BottomTabsLayout) layout).setBottomTabBadgeByIndex(index, badge);
+        }
+    }
+
+    public void setBottomTabBadgeByNavigatorId(String navigatorId, String badge) {
+        if (layout instanceof BottomTabsLayout) {
+            ((BottomTabsLayout) layout).setBottomTabBadgeByNavigatorId(navigatorId, badge);
+        }
+    }
 }

android/app/src/main/java/com/reactnativenavigation/controllers/NavigationCommandsHandler.java

@@ -266,4 +266,32 @@ public class NavigationCommandsHandler {
             }
         });
     }
+
+    public static void setBottomTabBadgeByIndex(final Integer index, final String badge) {
+        final NavigationActivity currentActivity = NavigationActivity.currentActivity;
+        if (currentActivity == null) {
+            return;
+        }
+
+        NavigationApplication.instance.runOnMainThread(new Runnable() {
+            @Override
+            public void run() {
+                currentActivity.setBottomTabBadgeByIndex(index, badge);
+            }
+        });
+    }
+
+    public static void setBottomTabBadgeByNavigatorId(final String navigatorId, final String badge) {
+        final NavigationActivity currentActivity = NavigationActivity.currentActivity;
+        if (currentActivity == null) {
+            return;
+        }
+
+        NavigationApplication.instance.runOnMainThread(new Runnable() {
+            @Override
+            public void run() {
+                currentActivity.setBottomTabBadgeByNavigatorId(navigatorId, badge);
+            }
+        });
+    }
 }

android/app/src/main/java/com/reactnativenavigation/layouts/BottomTabsLayout.java

@@ -252,6 +252,14 @@ public class BottomTabsLayout extends RelativeLayout implements Layout, AHBottom
         return screenStacks[index];
     }
 
+    public void setBottomTabBadgeByIndex(Integer index, String badge) {
+        bottomTabs.setNotification(badge, index);
+    }
+
+    public void setBottomTabBadgeByNavigatorId(String navigatorId, String badge) {
+        bottomTabs.setNotification(badge, getScreenStackIndex(navigatorId));
+    }
+
     private int getScreenStackIndex(String navigatorId) throws ScreenStackNotFoundException {
         for (int i = 0; i < screenStacks.length; i++) {
             if (screenStacks[i].getNavigatorId().equals(navigatorId)) {

src/deprecated/platformSpecificDeprecated.android.js

@@ -232,10 +232,12 @@ function navigatorSetButtons(navigator, navigatorEventID, params) {
 }
 
 function navigatorSetTabBadge(navigator, params) {
-  //RctActivity.setTabBadge({
-  //  tabIndex: params.tabIndex,
-  //  badge: params.badge
-  //});
+  const badge = params.badge.toString();
+  if (params.tabIndex >= 0) {
+    newPlatformSpecific.setBottomTabBadgeByIndex(params.tabIndex, badge);
+  } else {
+    newPlatformSpecific.setBottomTabBadgeByNavigatorId(navigator.navigatorID, badge);
+  }
 }
 
 function navigatorSetTitle(navigator, params) {

src/platformSpecific.android.js

@@ -100,6 +100,14 @@ function selectBottomTabByTabIndex(index) {
   NativeReactModule.selectBottomTabByTabIndex(index);
 }
 
+function setBottomTabBadgeByIndex(index, badge) {
+  NativeReactModule.setBottomTabBadgeByIndex(index, badge);
+}
+
+function setBottomTabBadgeByNavigatorId(navigatorId, badge) {
+  NativeReactModule.setBottomTabBadgeByNavigatorId(navigatorId, badge);
+}
+
 module.exports = {
   startApp,
   push,
@@ -116,5 +124,7 @@ module.exports = {
   toggleSideMenuVisible,
   setSideMenuVisible,
   selectBottomTabByNavigatorId,
-  selectBottomTabByTabIndex
+  selectBottomTabByTabIndex,
+  setBottomTabBadgeByNavigatorId,
+  setBottomTabBadgeByIndex
 };