@@ -3,7 +3,8 @@ import Head from "next/head";
33import type { NextComponentType } from "next" ;
44import DefaultApp from "next/app" ;
55import type { AppProps , AppContext } from "next/app" ;
6- import type { DocumentProps , DocumentContext } from "next/document" ;
6+ import type DefaultDocument from "next/document" ;
7+ import { DocumentProps , DocumentContext } from "next/document" ;
78import { rootColorSchemeStyleTagId , data_fr_scheme , data_fr_theme } from "./useIsDark/constants" ;
89import { getScriptToRunAsap } from "./useIsDark/scriptToRunAsap" ;
910import { SsrIsDarkProvider } from "./useIsDark/server" ;
@@ -14,7 +15,6 @@ import { fontUrlByFileBasename } from "./next-appdir/fontUrlByFileBasename";
1415import AppleTouchIcon from "./dsfr/favicon/apple-touch-icon.png" ;
1516import FaviconSvg from "./dsfr/favicon/favicon.svg" ;
1617import FaviconIco from "./dsfr/favicon/favicon.ico" ;
17- import DefaultDocument from "next/document" ;
1818import { getAssetUrl } from "./tools/getAssetUrl" ;
1919import { getColors } from "./fr/colors" ;
2020import { start } from "./start" ;
@@ -70,7 +70,10 @@ export type NextDsfrIntegrationApi = {
7070 App : AppComponent
7171 ) => AppComponent ;
7272 dsfrDocumentApi : {
73- augmentDocumentForDsfr : ( Document : NextComponentType < any , any , any > ) => void ;
73+ augmentDocumentForDsfr : ( params : {
74+ DefaultDocument : typeof DefaultDocument ;
75+ Document ?: NextComponentType < any , any , any > ;
76+ } ) => void ;
7477 getColorSchemeHtmlAttributes : (
7578 props : DocumentProps
7679 ) =>
@@ -232,7 +235,12 @@ export function createNextDsfrIntegrationApi(
232235 return AppWithDsfr as any ;
233236 }
234237
235- function augmentDocumentForDsfr ( Document : NextComponentType < any , any , any > ) : void {
238+ function augmentDocumentForDsfr ( params : {
239+ DefaultDocument : typeof DefaultDocument ;
240+ Document ?: NextComponentType < any , any , any > ;
241+ } ) : void {
242+ const { DefaultDocument, Document = DefaultDocument } = params ;
243+
236244 const super_getInitialProps =
237245 Document . getInitialProps ?. bind ( Document ) ??
238246 DefaultDocument . getInitialProps . bind ( DefaultDocument ) ;
0 commit comments