feat: pass router to init function so that e.g. locale can be passed …

…to apollo
lfades · Apr 18, 2021 · 1853249 · 1853249
1 parent a2b9606
commit 1853249
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/types.ts b/src/types.ts
@@ -2,6 +2,7 @@ import ApolloClient from 'apollo-client';
 import { IncomingHttpHeaders } from 'http';
 import { NextPage, NextPageContext } from 'next';
 import { AppContext } from 'next/app';
+import { NextRouter } from 'next/dist/client/router';
 import { ReactNode } from 'react';
 
 export interface WithApolloOptions {
@@ -25,6 +26,7 @@ export interface WithApolloProps<TCache> {
 export interface InitApolloOptions<TCache> {
   ctx?: NextPageContext;
   headers?: IncomingHttpHeaders;
+  router?: NextRouter;
   initialState?: TCache;
 }
 

diff --git a/src/withApollo.tsx b/src/withApollo.tsx
@@ -54,9 +54,14 @@ export default function withApollo<TCache = any>(
     if (getInitialProps || getDataFromTree) {
       WithApollo.getInitialProps = async (pageCtx: ApolloContext) => {
         const ctx = 'Component' in pageCtx ? pageCtx.ctx : pageCtx;
+        const router = 'Component' in pageCtx ? pageCtx.router : undefined;
         const { AppTree } = pageCtx;
         const headers = ctx.req ? ctx.req.headers : {};
-        const apollo = initApollo<TCache>(client, { ctx, headers });
+        const apollo = initApollo<TCache>(client, {
+          ctx,
+          headers,
+          router
+        });
         const apolloState: WithApolloState<TCache> = {};
 
         let pageProps = {};