Add apolloClient to NextPageContext & NextPageContext

examples/with-apollo/lib/apollo.js

@@ -37,10 +37,18 @@ export function withApollo(PageComponent, { ssr = true } = {}) {
   if (ssr || PageComponent.getInitialProps) {
     WithApollo.getInitialProps = async ctx => {
       const { AppTree } = ctx
+      const inAppContext = Boolean(ctx.ctx)
 
-      // Initialize ApolloClient, add it to the ctx object so
-      // we can use it in `PageComponent.getInitialProp`.
-      const apolloClient = (ctx.apolloClient = initApolloClient())
+      // Initialize ApolloClient
+      const apolloClient = initApolloClient()
+
+      // Add apolloClient to NextPageContext & NextAppContext
+      // This allows us to consume the apolloClient inside our
+      // custom `getInitialProps({ apolloClient })`.
+      ctx.apolloClient = apolloClient
+      if (inAppContext) {
+        ctx.ctx.apolloClient = apolloClient
+      }
 
       // Run wrapped getInitialProps methods
       let pageProps = {}
@@ -64,7 +72,6 @@ export function withApollo(PageComponent, { ssr = true } = {}) {
 
             // Since AppComponents and PageComponents have different context types
             // we need to modify their props a little.
-            const inAppContext = Boolean(ctx.ctx)
             let props
             if (inAppContext) {
               props = { ...pageProps, apolloClient }