From 5dc7946df2ed78fc9473d35d4eeb4ae6ee51c3dc Mon Sep 17 00:00:00 2001
From: iCrawl <buechler.noel@outlook.com>
Date: Mon, 15 Aug 2022 18:23:10 +0200
Subject: [PATCH] fix: properly render critical styles on the server

---
 packages/website/src/pages/_document.tsx | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/packages/website/src/pages/_document.tsx b/packages/website/src/pages/_document.tsx
index 14eb6c49f..534ad0ab2 100644
--- a/packages/website/src/pages/_document.tsx
+++ b/packages/website/src/pages/_document.tsx
@@ -1,9 +1,17 @@
-import { createGetInitialProps } from '@mantine/next';
-import Document, { Html, Head, Main, NextScript } from 'next/document';
+import { createStylesServer, ServerStyles } from '@mantine/next';
+import Document, { Html, Head, Main, NextScript, DocumentContext, DocumentInitialProps } from 'next/document';
+
+const stylesServer = createStylesServer();
 
-const getInitialProps = createGetInitialProps();
 export default class _Document extends Document {
-	public static override getInitialProps = getInitialProps;
+	public static override async getInitialProps(ctx: DocumentContext): Promise<DocumentInitialProps> {
+		const initialProps = await Document.getInitialProps(ctx);
+
+		return {
+			...initialProps,
+			styles: [initialProps.styles, <ServerStyles html={initialProps.html} server={stylesServer} key="styles" />],
+		};
+	}
 
 	public override render() {
 		return (