import App, { Container } from 'next/app'
import React from 'react'
import { setState } from '../shared-module'

setState(typeof window === 'undefined' ? 'UPDATED' : 'UPDATED CLIENT')

class Layout extends React.Component {
  state = {
    random: false
  }

  componentDidMount () {
    this.setState({ random: Math.random() })
  }

  render () {
    const { children } = this.props
    const { random } = this.state
    return (
      <div>
        <p id='hello-app'>Hello App</p>
        <p id='hello-hmr'>Hello HMR</p>
        <p id='random-number'>{random}</p>
        {children}
      </div>
    )
  }
}

export default class MyApp extends App {
  static async getInitialProps ({ Component, router, ctx }) {
    // throw _app GIP err here
    let pageProps = {}

    if (Component.getInitialProps) {
      pageProps = await Component.getInitialProps(ctx)
    }

    return { pageProps }
  }

  render () {
    const { Component, pageProps } = this.props
    return (
      <Container>
        <Layout>
          <Component {...pageProps} />
        </Layout>
      </Container>
    )
  }
}