rsnext/test/integration/amphtml/pages/_document.js
JJ Kasper 11016221cd
Make sure AmpContext is available in _document (#7021)
* Make sure AmpContext is available during _document
render and update filtering of script tags in AMP mode

* Update amphtml test to make sure
AmpContext is set for _document render

* Fix stray comma in render.tsx
2019-04-12 19:04:52 -05:00

29 lines
598 B
JavaScript

import { useAmp } from 'next/amp'
import Document, { Html, Head, Main, NextScript } from 'next/document'
const AmpTst = () => {
const isAmp = useAmp()
return <p>{isAmp ? 'AMP Power!!!' : 'no AMP for you...'}</p>
}
class MyDocument extends Document {
static async getInitialProps (ctx) {
const initialProps = await Document.getInitialProps(ctx)
return { ...initialProps }
}
render () {
return (
<Html>
<Head />
<body>
<Main />
<AmpTst />
<NextScript />
</body>
</Html>
)
}
}
export default MyDocument