[HEAP-9449] Perform ref forwarding for react nav HOC

examples/TestDriver/App.js

@@ -1,13 +1,32 @@
 import React from 'react';
 import { Provider } from 'react-redux';
+import { NavigationActions } from 'react-navigation';
 import { store } from './src/reduxElements';
 import TopNavigator from './src/topNavigator';
 
+let _navigator;
+
+function setTopLevelNavigator(navigatorRef) {
+  _navigator = navigatorRef;
+
+  navigate('Initial');
+}
+
+function navigate(routeName, params) {
+  console.log(_navigator);
+  _navigator.dispatch(
+    NavigationActions.navigate({
+      routeName,
+      params,
+    })
+  );
+}
+
 export default class App extends React.Component {
   render() {
     return (
       <Provider store={store}>
-        <TopNavigator />
+        <TopNavigator ref={setTopLevelNavigator} />
       </Provider>
     );
   }

js/autotrack/reactNavigation.js

@@ -9,9 +9,17 @@ const EVENT_TYPE = 'reactNavigationScreenview';
 const INITIAL_ROUTE_TYPE = 'Heap_Navigation/INITIAL';
 
 export const withReactNavigationAutotrack = track => AppContainer => {
-  return class HeapNavigationWrapper extends React.Component {
+  class HeapNavigationWrapper extends React.Component {
     topLevelNavigator = null;
 
+    setRef(ref, value) {
+      if (typeof ref === 'function') {
+        ref(value);
+      } else if (ref !== null) {
+        ref.current = value;
+      }
+    }
+
     trackInitialRoute() {
       const initialPageviewPath = getActiveRouteName(
         this.topLevelNavigator.state.nav
@@ -28,14 +36,17 @@ export const withReactNavigationAutotrack = track => AppContainer => {
         return this._render();
       } catch (e) {
         bail(e);
-        return <AppContainer />;
+        const { forwardedRef, ...rest } = this.props;
+        return <AppContainer ref={forwardedRef} {...rest} />;
       }
     }
 
     _render() {
+      const { forwardedRef, ...rest } = this.props;
       return (
         <AppContainer
           ref={bailOnError(navigatorRef => {
+            this.setRef(forwardedRef, navigatorRef);
             if (this.topLevelNavigator !== navigatorRef) {
               console.log(
                 'Heap: React Navigation is instrumented for autocapture.'
@@ -55,13 +66,17 @@ export const withReactNavigationAutotrack = track => AppContainer => {
               });
             }
           })}
-          {...this.props}
+          {...rest}
         >
           {this.props.children}
         </AppContainer>
       );
     }
-  };
+  }
+
+  return React.forwardRef((props, ref) => {
+    return <HeapNavigationWrapper {...props} forwardedRef={ref} />;
+  });
 };
 
 const getActiveRouteName = navigationState => {