chore: update draftMode() calls for async use

jakala-na · Nov 8, 2024 · e7f4520 · e7f4520
1 parent 2a6f245
commit e7f4520
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/app/[locale]/[[...slug]]/page.tsx b/app/[locale]/[[...slug]]/page.tsx
@@ -129,7 +129,7 @@ export default async function LandingPage(props: PageProps) {
   setStaticParamsLocale(locale);
   const slug = params.slug?.join('/') ?? 'home';
 
-  const { isEnabled: isDraftMode } = await draftMode();
+  const isDraftMode = (await draftMode()).isEnabled;
 
   const pageData = await getPage(slug, getLocaleFromPath(locale), isDraftMode);
 
@@ -159,7 +159,7 @@ export async function generateMetadata(props: PageProps): Promise<Metadata> {
   const params = await props.params;
   const { locale } = params;
   const slug = params.slug?.join('/') ?? 'home';
-  const { isEnabled: isDraftMode } = await draftMode();
+  const isDraftMode = (await draftMode()).isEnabled;
   return getPageMetadata(slug, getLocaleFromPath(locale), isDraftMode);
 }
 

diff --git a/app/[locale]/layout.tsx b/app/[locale]/layout.tsx
@@ -26,7 +26,7 @@ export default async function RootLayout(props: { children: React.ReactNode; par
 
   const shouldInjectToolbar = process.env.NODE_ENV === 'development';
   const { locale } = params;
-  const { isEnabled: isDraftMode } = await draftMode();
+  const isDraftMode = (await draftMode()).isEnabled;
 
   const layoutQuery = graphql(
     `

diff --git a/components/site-header/site-header.tsx b/components/site-header/site-header.tsx
@@ -1,11 +1,11 @@
-import { draftMode, type UnsafeUnwrappedDraftMode } from 'next/headers';
+import { draftMode } from 'next/headers';
 import Link from 'next/link';
 
 import { Navigation } from '#/components/navigation';
 
 // TODO: Fix fragment unmasking to type navigationData, if possible.
-export function SiteHeader(props: { navigationData: any }) {
-  const { isEnabled: isDraftMode } = draftMode() as unknown as UnsafeUnwrappedDraftMode;
+export async function SiteHeader(props: { navigationData: any }) {
+  const isDraftMode = (await draftMode()).isEnabled;
   return (
     <>
       {isDraftMode && (