Instantly share code, notes, and snippets.
Created
July 17, 2020 09:04
-
Star
0
(0)
You must be signed in to star a gist -
Fork
0
(0)
You must be signed in to fork a gist
-
Save lionelyoung/6f9020de23f257599bdabfdb0bf40bff to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import React from "react" | |
import Head from "next/head" | |
import { ApolloProvider } from "@apollo/react-hooks" | |
import { ApolloClient } from "apollo-client" | |
import { InMemoryCache } from "apollo-cache-inmemory" | |
import { HttpLink } from "apollo-link-http" | |
import fetch from "isomorphic-unfetch" | |
let apolloClient = null | |
/** | |
* Creates and provides the apolloContext | |
* to a next.js PageTree. Use it by wrapping | |
* your PageComponent via HOC pattern. | |
* @param {Function|Class} PageComponent | |
* @param {Object} [config] | |
* @param {Boolean} [config.ssr=true] | |
*/ | |
export function withApollo(PageComponent, { ssr = true } = {}) { | |
const WithApollo = ({ apolloClient, apolloState, ...pageProps }) => { | |
const client = apolloClient || initApolloClient(apolloState) | |
return ( | |
<ApolloProvider client={client} > | |
<PageComponent {...pageProps} /> | |
</ApolloProvider> | |
) | |
} | |
// Set the correct displayName in development | |
if (process.env.NODE_ENV !== "production") { | |
const displayName = | |
PageComponent.displayName || PageComponent.name || "Component" | |
if (displayName === "App") { | |
console.warn("This withApollo HOC only works with PageComponents.") | |
} | |
WithApollo.displayName = `withApollo(${displayName})` | |
} | |
if (ssr || PageComponent.getInitialProps) { | |
WithApollo.getInitialProps = async ctx => { | |
const { AppTree } = ctx | |
// Initialize ApolloClient, add it to the ctx object so | |
// we can use it in `PageComponent.getInitialProp`. | |
const apolloClient = (ctx.apolloClient = initApolloClient({}, ctx.req.headers.cookie)) | |
// Run wrapped getInitialProps methods | |
let pageProps = {} | |
if (PageComponent.getInitialProps) { | |
pageProps = await PageComponent.getInitialProps(ctx) | |
} | |
// Only on the server: | |
if (typeof window === "undefined") { | |
// When redirecting, the response is finished. | |
// No point in continuing to render | |
if (ctx.res && ctx.res.finished) { | |
return pageProps | |
} | |
// Only if ssr is enabled | |
if (ssr) { | |
try { | |
// Run all GraphQL queries | |
const { getDataFromTree } = await import("@apollo/react-ssr") | |
await getDataFromTree( | |
<AppTree | |
pageProps={{ | |
...pageProps, | |
apolloClient | |
}} | |
/> | |
) | |
} catch (error) { | |
// Prevent Apollo Client GraphQL errors from crashing SSR. | |
// Handle them in components via the data.error prop: | |
// https://www.apollographql.com/docs/react/api/react-apollo.html#graphql-query-data-error | |
console.error("Error while running `getDataFromTree`", error) | |
} | |
// getDataFromTree does not call componentWillUnmount | |
// head side effect therefore need to be cleared manually | |
Head.rewind() | |
} | |
} | |
// Extract query data from the Apollo store | |
// @ts-ignore | |
const apolloState = apolloClient.cache.extract() | |
return { | |
...pageProps, | |
apolloState | |
} | |
} | |
} | |
return WithApollo | |
} | |
/** | |
* Always creates a new apollo client on the server | |
* Creates or reuses apollo client in the browser. | |
* @param {Object} initialState | |
*/ | |
function initApolloClient(initialState = {}, cookie = "") { | |
// 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") { | |
return createApolloClient(initialState, cookie) | |
} | |
// Reuse client on the client-side | |
if (!apolloClient) { | |
// @ts-ignore | |
apolloClient = createApolloClient(initialState) | |
} | |
return apolloClient | |
} | |
/** | |
* Creates and configures the ApolloClient | |
* @param {Object} [initialState={}] | |
*/ | |
function createApolloClient(initialState = {}, cookie) { | |
const enchancedFetch = (url, init) => { | |
return fetch(url, { | |
...init, | |
headers: { | |
...init.headers, | |
"Cookie": cookie | |
} | |
}).then(response => response) | |
} | |
// Check out https://github.com/zeit/next.js/pull/4611 if you want to use the AWSAppSyncClient | |
return new ApolloClient({ | |
ssrMode: typeof window === "undefined", // Disables forceFetch on the server (so queries are only run once) | |
link: new HttpLink({ | |
uri: "http://localhost:3000/api", // Server URL (must be absolute) | |
credentials: "same-origin", // Additional fetch() options like `credentials` or `headers` | |
fetch: enchancedFetch | |
}), | |
cache: new InMemoryCache().restore(initialState) | |
}) | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment