Add headers to apolloClient

If withData receives ApolloClient instance, pass headers to ApolloClient

Author: alyavasilyeva

package-lock.json

@@ -39,8 +39,8 @@
     "apollo-boost": "^0.4.4",
     "graphql": "^14.3.1",
     "next": "^9.0.6",
-    "react-dom": "^16.8.6",
-    "react": "^16.8.6"
+    "react": "^16.8.6",
+    "react-dom": "^16.8.6"
   },
   "devDependencies": {
     "@apollo/react-hooks": "^3.0.1",
@@ -65,6 +65,7 @@
   "dependencies": {
     "@babel/runtime": "^7.4.5",
     "isomorphic-unfetch": "^3.0.0",
+    "lodash": "^4.17.15",
     "prop-types": "15.7.2",
     "prop-types-exact": "1.2.0",
     "url": "0.11.0"

package.json

@@ -3,6 +3,7 @@ import Head from 'next/head'
 import { ApolloProvider } from '@apollo/react-hooks'
 import { ApolloClient, InMemoryCache, HttpLink } from 'apollo-boost'
 import fetch from 'isomorphic-unfetch'
+import { isFunction } from 'lodash'
 
 let apolloClient = null
 
@@ -40,7 +41,11 @@ export default apolloConfig => {
 
         // Initialize ApolloClient, add it to the ctx object so
         // we can use it in `PageComponent.getInitialProp`.
-        const apolloClient = (ctx.apolloClient = initApolloClient(apolloConfig, null))
+        const apolloClient = (ctx.apolloClient = initApolloClient(
+          apolloConfig,
+          null,
+          ctx
+        ))
 
         // Run wrapped getInitialProps methods
         let pageProps = {}
@@ -101,7 +106,10 @@ export default apolloConfig => {
  * Creates or reuses apollo client in the browser.
  * @param  {Object} initialState
  */
-function initApolloClient(apolloConfig, initialState = {}) {
+function initApolloClient(apolloConfig, initialState = {}, ctx) {
+  if (isFunction(apolloConfig)) {
+    apolloConfig = apolloConfig(ctx)
+  }
   // Make sure to create a new client for every server-side request so that data
   // isn't shared between connections (which would be bad)
   if (typeof window === 'undefined') {