fix(HOCs): missing props (firebase + dispatch) added - #606

prescottprue · Jan 16, 2019 · 4c9c0c0 · 4c9c0c0
1 parent d322124
commit 4c9c0c0
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/src/withFirestore.js b/src/withFirestore.js
@@ -1,6 +1,7 @@
 import React, { Component } from 'react'
 import hoistStatics from 'hoist-non-react-statics'
 import { wrapDisplayName } from './utils'
+import ReactReduxFirebaseContext from './ReactReduxFirebaseContext'
 import ReduxFirestoreContext from './ReduxFirestoreContext'
 
 /**
@@ -35,9 +36,20 @@ export const createWithFirestore = (storeKey = 'store') => WrappedComponent => {
   const HoistedComp = hoistStatics(WithFirestore, WrappedComponent)
 
   const withFirestore = props => (
-    <ReduxFirestoreContext.Consumer>
-      {firestore => <HoistedComp firestore={firestore} {...props} />}
-    </ReduxFirestoreContext.Consumer>
+    <ReactReduxFirebaseContext.Consumer>
+      {firebase => (
+        <ReduxFirestoreContext.Consumer>
+          {firestore => (
+            <HoistedComp
+              firestore={firestore}
+              firebase={firebase}
+              dispatch={firebase.dispatch}
+              {...props}
+            />
+          )}
+        </ReduxFirestoreContext.Consumer>
+      )}
+    </ReactReduxFirebaseContext.Consumer>
   )
 
   withFirestore.displayName = wrapDisplayName(WrappedComponent, 'withFirestore')