import React from 'react'
import { QueryRenderer, fetchQuery } from 'react-relay'
import NextApp, { Container } from 'next/app'

import {
  initEnvironment,
  createEnvironment
} from '../lib/createEnvironment'

export default class App extends NextApp {
  static getInitialProps = async ({ Component, router, ctx }) => {
    const { variables } = Component.getInitialProps ? await Component.getInitialProps(ctx) : {}

    try {
      if (initEnvironment) {
        const { environment, relaySSR } = initEnvironment()

        await fetchQuery(environment, Component.query, variables)

        return {
          variables,
          relayData: await relaySSR.getCache()
        }
      }
    } catch (e) {
      console.log(e)
    }

    return {
      variables
    }
  };

  render () {
    const { Component, variables = {}, relayData } = this.props
    const environment = createEnvironment(
      relayData,
      JSON.stringify({
        queryID: Component.query ? Component.query().name : undefined,
        variables
      })
    )

    return (
      <Container>
        <QueryRenderer
          environment={environment}
          query={Component.query}
          variables={variables}
          render={({ error, props }) => {
            if (error) return <div>{error.message}</div>
            else if (props) return <Component {...props} />
            return <div>Loading</div>
          }}
        />
      </Container>
    )
  }
}