64 lines
1.4 KiB
TypeScript
64 lines
1.4 KiB
TypeScript
import React from 'react'
|
|
import Link from 'next/link'
|
|
import { render } from 'react-dom'
|
|
import App, { Container, AppContext } from 'next/app'
|
|
import { renderToString } from 'react-dom/server'
|
|
|
|
class MyApp<P = {}> extends App<P & { html: string }> {
|
|
static async getInitialProps({
|
|
Component,
|
|
AppTree,
|
|
router,
|
|
ctx,
|
|
}: AppContext) {
|
|
let pageProps = {}
|
|
|
|
if (Component.getInitialProps) {
|
|
pageProps = await Component.getInitialProps(ctx)
|
|
}
|
|
|
|
let html
|
|
const toRender = (
|
|
<AppTree
|
|
{...{
|
|
router,
|
|
Component,
|
|
pageProps,
|
|
another: 'hello',
|
|
}}
|
|
/>
|
|
)
|
|
|
|
if (typeof window !== 'undefined') {
|
|
const el = document.createElement('div')
|
|
document.querySelector('body').appendChild(el)
|
|
render(toRender, el)
|
|
html = el.innerHTML
|
|
el.remove()
|
|
} else {
|
|
html = renderToString(toRender)
|
|
}
|
|
|
|
return { pageProps, html }
|
|
}
|
|
|
|
render() {
|
|
const { Component, pageProps, html, router } = this.props
|
|
const href = router.pathname === '/' ? '/another' : '/'
|
|
|
|
return html && router.pathname !== '/hello' ? (
|
|
<>
|
|
<div dangerouslySetInnerHTML={{ __html: html }} />
|
|
<Link href={href}>
|
|
<a id={href === '/' ? 'home' : 'another'}>to {href}</a>
|
|
</Link>
|
|
</>
|
|
) : (
|
|
<Container>
|
|
<Component {...pageProps} />
|
|
</Container>
|
|
)
|
|
}
|
|
}
|
|
|
|
export default MyApp
|