import Document, { Html, Head, Main, NextScript } from 'next/document';
import * as React from 'react';

import createEmotionServer from '@emotion/server/create-instance';
import { cache } from '@emotion/css';
import { IPageExtraInfo } from '@tempi/core-renderer';

const renderStatic = async (html) => {
  if (html === undefined) {
    throw new Error('did you forget to return html from renderToString?');
  }
  const { extractCritical } = createEmotionServer(cache);
  const { ids, css } = extractCritical(html);

  return { html, ids, css };
};

interface NextData {
  props?: {
    pageProps?: {
      page?: {
        extraInfo?: IPageExtraInfo;
        languageCode?: string;
      };
    };
  };
}

export default class AppDocument extends Document {
  static async getInitialProps(ctx) {
    const page = await ctx.renderPage();
    const { css, ids } = await renderStatic(page.html);
    const initialProps = await Document.getInitialProps(ctx);
    return {
      ...initialProps,
      styles: (
        <React.Fragment>
          {initialProps.styles}
          <style
            data-emotion={`css ${ids.join(' ')}`}
            dangerouslySetInnerHTML={{ __html: css }}
          />
        </React.Fragment>
      ),
    };
  }

  render() {
    const nextData = this.props.__NEXT_DATA__ as NextData;
    const page = nextData?.props?.pageProps?.page || {};
    return (
      <Html lang={page.languageCode || page.extraInfo?.languageCode || 'vi'}>
        <Head />
        <body>
          <Main />
          <NextScript />
        </body>
      </Html>
    );
  }
}