daedalOS/pages/_document.tsx
2024-11-14 13:44:52 -08:00

56 lines
1.2 KiB
TypeScript

import NextDocument, {
type DocumentContext,
type DocumentInitialProps,
Head,
Html,
Main,
NextScript,
} from "next/document";
import { ServerStyleSheet } from "styled-components";
import { DEFAULT_LOCALE } from "utils/constants";
const withStyledComponents = async (
ctx: DocumentContext
): Promise<DocumentInitialProps> => {
const { renderPage } = ctx;
const sheet = new ServerStyleSheet();
try {
ctx.renderPage = () =>
renderPage({
enhanceApp: (App) => (props) => sheet.collectStyles(<App {...props} />),
});
const { styles, ...initialProps } = await NextDocument.getInitialProps(ctx);
return {
...initialProps,
styles: [styles, sheet.getStyleElement()],
};
} finally {
sheet.seal();
}
};
class Document extends NextDocument {
public static override async getInitialProps(
ctx: DocumentContext
): Promise<DocumentInitialProps> {
return withStyledComponents(ctx);
}
public override render(): React.JSX.Element {
return (
<Html lang={DEFAULT_LOCALE}>
<Head />
<body>
<Main />
<NextScript />
</body>
</Html>
);
}
}
export default Document;