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