react-native-navigation/src/Screen.js

import { Component, NativeAppEventEmitter } from 'react-native';
import platformSpecific from './platformSpecific';
import Navigation from './Navigation';

const _allNavigatorEventHandlers = {};

class Navigator {
  constructor(navigatorID, navigatorEventID) {
    this.navigatorID = navigatorID;
    this.navigatorEventID = navigatorEventID;
    this.navigatorEventHandler = null;
    this.navigatorEventSubscription = null;
  }
  push(params = {}) {
    return platformSpecific.navigatorPush(this, params);
  }
  pop(params = {}) {
    return platformSpecific.navigatorPop(this, params);
  }
  popToRoot(params = {}) {
    return platformSpecific.navigatorPopToRoot(this, params);
  }
  resetTo(params = {}) {
    return platformSpecific.navigatorResetTo(this, params);
  }
  showModal(params = {}) {
    return Navigation.showModal(params);
  }
  dismissModal(params = {}) {
    return Navigation.dismissModal(params);
  }
  showLightBox(params = {}) {
    return Navigation.showLightBox(params);
  }
  dismissLightBox(params = {}) {
    return Navigation.dismissLightBox(params);
  }
  setButtons(params = {}) {
    return platformSpecific.navigatorSetButtons(this, this.navigatorEventID, params);
  }
  setTitle(params = {}) {
    return platformSpecific.navigatorSetTitle(this, params);
  }
  toggleDrawer(params = {}) {
    return platformSpecific.navigatorToggleDrawer(this, params);
  }
  toggleTabs(params = {}) {
    return platformSpecific.navigatorToggleTabs(this, params);
  }
  setTabBadge(params = {}) {
    return platformSpecific.navigatorSetTabBadge(this, params);
  }
  switchToTab(params = {}) {
    return platformSpecific.navigatorSwitchToTab(this, params);
  }
  setOnNavigatorEvent(callback) {
    this.navigatorEventHandler = callback;
    if (!this.navigatorEventSubscription) {
      this.navigatorEventSubscription = NativeAppEventEmitter.addListener(this.navigatorEventID, (event) => this.onNavigatorEvent(event));
      _allNavigatorEventHandlers[this.navigatorEventID] = (event) => this.onNavigatorEvent(event);
    }
  }
  handleDeepLink(params = {}) {
    if (!params.link) return;
    const event = {
      type: 'DeepLink',
      link: params.link
    };
    for (let i in _allNavigatorEventHandlers) {
      _allNavigatorEventHandlers[i](event);
    }
  }
  onNavigatorEvent(event) {
    if (this.navigatorEventHandler) {
      this.navigatorEventHandler(event);
    }
  }
  cleanup() {
    if (this.navigatorEventSubscription) {
      this.navigatorEventSubscription.remove();
      delete _allNavigatorEventHandlers[this.navigatorEventID];
    }
  }
}

export default class Screen extends Component {
  static navigatorStyle = {};
  static navigatorButtons = {};
  constructor(props) {
    super(props);
    if (props.navigatorID) {
      this.navigator = new Navigator(props.navigatorID, props.navigatorEventID);
    }
  }
  componentWillUnmount() {
    if (this.navigator) {
      this.navigator.cleanup();
      this.navigator = undefined;
    }
  }
}