import React, { Component, PropTypes } from 'react' import htmlescape from 'htmlescape' import flush from 'styled-jsx/server' export default class Document extends Component { static getInitialProps ({ renderPage }) { const {html, head} = renderPage() const styles = flush() return { html, head, styles } } static childContextTypes = { _documentProps: PropTypes.any } getChildContext () { return { _documentProps: this.props } } render () { return

} } export class Head extends Component { static contextTypes = { _documentProps: PropTypes.any } render () { const { head, styles } = this.context._documentProps return {(head || []).map((h, i) => React.cloneElement(h, { key: i }))} {styles || null} {this.props.children} } } export class Main extends Component { static contextTypes = { _documentProps: PropTypes.any } render () { const { html } = this.context._documentProps return

} } export class NextScript extends Component { static contextTypes = { _documentProps: PropTypes.any } getChunkScript (filename, additionalProps = {}) { const { __NEXT_DATA__ } = this.context._documentProps let { buildStats } = __NEXT_DATA__ const hash = buildStats ? buildStats[filename].hash : '-' return (