mirror of
https://github.com/terribleplan/next.js.git
synced 2024-01-19 02:48:18 +00:00
dca2ca6f2b
As seen on `with-apollo-auth` there are some things that need to be addressed here too. * #4554 remove useless `apolloState` from App props on `getDataFromTree` * #4563 simplify `apolloState` prop Let me know if further changes/fixes are needed. Thank you 🎉
61 lines
1.7 KiB
JavaScript
61 lines
1.7 KiB
JavaScript
import initApollo from './init-apollo'
|
|
import Head from 'next/head'
|
|
import { getDataFromTree } from 'react-apollo'
|
|
|
|
export default (App) => {
|
|
return class Apollo extends React.Component {
|
|
static displayName = 'withApollo(App)'
|
|
static async getInitialProps (ctx) {
|
|
const { Component, router } = ctx
|
|
|
|
let appProps = {}
|
|
if (App.getInitialProps) {
|
|
appProps = await App.getInitialProps(ctx)
|
|
}
|
|
|
|
// Run all GraphQL queries in the component tree
|
|
// and extract the resulting data
|
|
const apollo = initApollo()
|
|
if (!process.browser) {
|
|
try {
|
|
// Run all GraphQL queries
|
|
await getDataFromTree(
|
|
<App
|
|
{...appProps}
|
|
Component={Component}
|
|
router={router}
|
|
apolloClient={apollo}
|
|
/>
|
|
)
|
|
} catch (error) {
|
|
// Prevent Apollo Client GraphQL errors from crashing SSR.
|
|
// Handle them in components via the data.error prop:
|
|
// http://dev.apollodata.com/react/api-queries.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
|
|
const apolloState = apollo.cache.extract()
|
|
|
|
return {
|
|
...appProps,
|
|
apolloState
|
|
}
|
|
}
|
|
|
|
constructor (props) {
|
|
super(props)
|
|
this.apolloClient = initApollo(props.apolloState)
|
|
}
|
|
|
|
render () {
|
|
return <App {...this.props} apolloClient={this.apolloClient} />
|
|
}
|
|
}
|
|
}
|