Some work on eventEmitter

android/app/src/main/java/com/reactnativenavigation/NavigationApplication.java

 import android.support.annotation.NonNull;
 
 import com.facebook.react.ReactPackage;
+import com.facebook.react.bridge.ReactContext;
 import com.facebook.react.shell.MainReactPackage;
+import com.reactnativenavigation.bridge.NavigationReactEventEmitter;
 import com.reactnativenavigation.bridge.NavigationReactPackage;
 import com.reactnativenavigation.react.NavigationReactInstance;
 
 
     public static NavigationApplication instance;
     private NavigationReactInstance navigationReactInstance;
+    private NavigationReactEventEmitter reactEventEmitter;
     private Handler handler;
 
     @Override
         navigationReactInstance.startReactContextOnceInBackgroundAndExecuteJS();
     }
 
+    public NavigationReactEventEmitter getReactEventEmitter() {
+        if (reactEventEmitter == null) {
+            reactEventEmitter = new NavigationReactEventEmitter(getCurrentReactContext());
+        }
+
+        return reactEventEmitter;
+    }
+
+    private ReactContext getCurrentReactContext() {
+        return navigationReactInstance.getReactInstanceManager().getCurrentReactContext();
+    }
+
     public Handler getMainHandler() {
         return handler;
     }

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

 package com.reactnativenavigation.bridge;
 
+import com.facebook.react.bridge.Arguments;
+import com.facebook.react.bridge.ReactContext;
 import com.facebook.react.bridge.WritableMap;
 import com.facebook.react.modules.core.DeviceEventManagerModule.RCTDeviceEventEmitter;
 
     private static final String KEY_EVENT_ID = "id";
     private static final String KEY_EVENT_TYPE = "type";
     private static final String EVENT_TYPE = "NavBarButtonPress";
+    private RCTDeviceEventEmitter eventEmitter;
 
-    public void sendEvent(RCTDeviceEventEmitter eventEmitter, String eventName, String navigatorEventId, WritableMap params) {
-        if (eventEmitter == null) {
-            return;
-        }
+    public NavigationReactEventEmitter(ReactContext reactContext) {
+        this.eventEmitter = reactContext.getJSModule(RCTDeviceEventEmitter.class);
+    }
 
+    public void sendEvent(String eventId, String screenInstanceId) {
+        WritableMap params = Arguments.createMap();
         params.putString(KEY_EVENT_TYPE, EVENT_TYPE);
-        params.putString(KEY_EVENT_ID, eventName);
-//        params.putString(_Screen.KEY_NAVIGATOR_EVENT_ID, navigatorEventId);
-        eventEmitter.emit(navigatorEventId, params);
+        params.putString(KEY_EVENT_ID, eventId);
+        params.putString("navigatorEventID", screenInstanceId);
+        eventEmitter.emit(screenInstanceId, params);
     }
 }

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

     }
 
     private void addTitleBar() {
-        topBar.addTitleBarAndSetButtons(screenParams.buttons, this);
+        topBar.addTitleBarAndSetButtons(screenParams.buttons, screenParams.screenInstanceId);
         topBar.setTitle(screenParams.title);
     }
 

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

 
 import android.content.Context;
 import android.support.design.widget.AppBarLayout;
-import android.view.MenuItem;
 
 import com.reactnativenavigation.params.TitleBarButtonParams;
 import com.reactnativenavigation.views.TitleBar;
         setFitsSystemWindows(true);
     }
 
-    public void addTitleBarAndSetButtons(List<TitleBarButtonParams> buttons, MenuItem.OnMenuItemClickListener onTitleBarButtonClickListener) {
+    public void addTitleBarAndSetButtons(List<TitleBarButtonParams> buttons, String screenInstanceId) {
         titleBar = new TitleBar(getContext());
         addView(titleBar);
-        titleBar.setButtons(buttons, onTitleBarButtonClickListener);
+        titleBar.setButtons(buttons, screenInstanceId);
     }
 
     public void setTitleBarVisibility(boolean isHidden) {

android/app/src/main/java/com/reactnativenavigation/params/TitleBarButtonParams.java

         }
     }
 
-    // Todo: add id for click listener
+    public String eventId;
     public String label;
     public Drawable icon;
     public ScreenStyleParams.Color color;

android/app/src/main/java/com/reactnativenavigation/react/NavigationReactInstance.java

 import com.facebook.react.LifecycleState;
 import com.facebook.react.ReactInstanceManager;
 import com.facebook.react.ReactPackage;
-import com.facebook.react.bridge.ReactContext;
 import com.facebook.react.modules.core.DefaultHardwareBackBtnHandler;
-import com.facebook.react.modules.core.DeviceEventManagerModule.RCTDeviceEventEmitter;
 import com.reactnativenavigation.NavigationApplication;
 import com.reactnativenavigation.controllers.NavigationActivity;
 
         }
     }
 
-    public RCTDeviceEventEmitter getEventEmitter() {
-        ReactContext currentReactContext = reactInstanceManager.getCurrentReactContext();
-        if (currentReactContext == null) {
-            return null;
-        }
-
-        return currentReactContext.getJSModule(RCTDeviceEventEmitter.class);
-    }
-
     public void onActivityResult(int requestCode, int resultCode, Intent data) {
         reactInstanceManager.onActivityResult(requestCode, resultCode, data);
     }

android/app/src/main/java/com/reactnativenavigation/views/TitleBar.java

 import android.content.Context;
 import android.support.v7.widget.Toolbar;
 import android.view.Menu;
-import android.view.MenuItem;
 
 import com.reactnativenavigation.animation.HideOnScrollAnimator;
 import com.reactnativenavigation.params.TitleBarButtonParams;
         super(context);
     }
 
-    public void setButtons(List<TitleBarButtonParams> buttons, MenuItem.OnMenuItemClickListener onTitleBarButtonClickListener) {
+    public void setButtons(List<TitleBarButtonParams> buttons, String screenInstanceId) {
         Menu menu = getMenu();
         menu.clear();
 
         for (int i = 0; i < buttons.size(); i++) {
-            final TitleBarButton button = new TitleBarButton(menu, this, buttons.get(i), onTitleBarButtonClickListener);
+            final TitleBarButton button = new TitleBarButton(menu, this, buttons.get(i), screenInstanceId);
             // Add button in reverse order because in iOS index 0 starts at right
             final int index = buttons.size() - i - 1;
             button.addToMenu(index);

android/app/src/main/java/com/reactnativenavigation/views/TitleBarButton.java

 import android.view.View;
 import android.widget.TextView;
 
+import com.reactnativenavigation.NavigationApplication;
 import com.reactnativenavigation.params.TitleBarButtonParams;
 import com.reactnativenavigation.utils.ImageUtils;
 import com.reactnativenavigation.utils.ViewUtils;
 
 import java.util.ArrayList;
 
-public class TitleBarButton {
+public class TitleBarButton implements MenuItem.OnMenuItemClickListener {
 
     private final Menu menu;
     private final View parent;
     private TitleBarButtonParams buttonParams;
-    private MenuItem.OnMenuItemClickListener onMenuItemClickListener;
+    private String screenInstanceId;
 
-    public TitleBarButton(Menu menu, View parent, TitleBarButtonParams buttonParams, MenuItem.OnMenuItemClickListener onMenuItemClickListener) {
+    public TitleBarButton(Menu menu, View parent, TitleBarButtonParams buttonParams, String screenInstanceId) {
         this.menu = menu;
         this.parent = parent;
         this.buttonParams = buttonParams;
-        this.onMenuItemClickListener = onMenuItemClickListener;
+        this.screenInstanceId = screenInstanceId;
     }
 
     public MenuItem addToMenu(int index) {
         item.setEnabled(buttonParams.enabled);
         setIcon(item);
         setColor();
-        item.setOnMenuItemClickListener(onMenuItemClickListener);
+        item.setOnMenuItemClickListener(this);
         return item;
     }
 
     private boolean hasColor() {
         return buttonParams.color.hasColor();
     }
+
+    @Override
+    public boolean onMenuItemClick(MenuItem item) {
+        NavigationApplication.instance.getReactEventEmitter().sendEvent(buttonParams.eventId, screenInstanceId);
+        return true;
+    }
 }