rsnext/examples/with-atlaskit/pages/_document.js
Alejandro d46a0aef0f
Integrate with-atlaskit (#11269)
* Add with-atlaskit

* Fix

* Fix

* Add babel
2020-04-05 13:27:49 +02:00

29 lines
713 B
JavaScript

import Document from 'next/document'
import { ServerStyleSheet } from 'styled-components'
export default class MyDocument extends Document {
static async getInitialProps(ctx) {
const sheet = new ServerStyleSheet()
const originalRenderPage = ctx.renderPage
try {
ctx.renderPage = () =>
originalRenderPage({
enhanceApp: App => props => sheet.collectStyles(<App {...props} />),
})
const initialProps = await Document.getInitialProps(ctx)
return {
...initialProps,
styles: (
<>
{initialProps.styles}
{sheet.getStyleElement()}
</>
),
}
} finally {
sheet.seal()
}
}
}