vivid-planet · thomasdax98 · Sep 5, 2024 · Sep 3, 2024 · Sep 4, 2024 · Sep 4, 2024
diff --git a/site/package-lock.json b/site/package-lock.json
diff --git a/site/package.json b/site/package.json
@@ -28,7 +28,6 @@
         "cache-manager": "^5.6.1",
         "graphql": "^15.0.0",
         "next": "^14.2.0",
-        "next-runtime-env": "^3.2.2",
         "react": "^18.2.0",
         "react-dom": "^18.2.0",
         "react-intl": "^6.0.0",

diff --git a/site/src/app/[domain]/[language]/[[...path]]/page.tsx b/site/src/app/[domain]/[language]/[[...path]]/page.tsx
@@ -1,3 +1,5 @@
+export const dynamic = "error";
+
 import { gql, previewParams } from "@comet/cms-site";
 import { documentTypes } from "@src/documents";
 import { GQLPageTreeNodeScopeInput } from "@src/graphql.generated";

diff --git a/site/src/app/layout.tsx → site/src/app/[domain]/layout.tsx b/site/src/app/layout.tsx → site/src/app/[domain]/layout.tsx
@@ -1,33 +1,34 @@
 import { SitePreviewProvider } from "@comet/cms-site";
 import { GlobalStyle } from "@src/layout/GlobalStyle";
-import { getSiteConfig } from "@src/middleware";
+import { getSiteConfigForDomain } from "@src/middleware";
 import { ResponsiveSpacingStyle } from "@src/util/ResponsiveSpacingStyle";
+import { SiteConfigProvider } from "@src/util/SiteConfigProvider";
 import StyledComponentsRegistry from "@src/util/StyledComponentsRegistry";
 import type { Metadata } from "next";
 import { draftMode } from "next/headers";
-import { PublicEnvScript } from "next-runtime-env";
 
 export const metadata: Metadata = {
     title: "Comet Starter",
 };
 
 export default async function RootLayout({
     children,
+    params: { domain },
 }: Readonly<{
     children: React.ReactNode;
+    params: { domain: string };
 }>) {
-    const { gtmId } = await getSiteConfig();
+    const siteConfig = await getSiteConfigForDomain(decodeURIComponent(domain));
+    const isDraftModeEnabled = draftMode().isEnabled;
 
     return (
         <html>
-            <head>
-                <PublicEnvScript />
-            </head>
+            <head />
             <body>
-                {gtmId && (
+                {siteConfig.gtmId && (
                     <noscript>
                         <iframe
-                            src={`https://www.googletagmanager.com/ns.html?id=${gtmId}`}
+                            src={`https://www.googletagmanager.com/ns.html?id=${siteConfig.gtmId}`}
                             height="0"
                             width="0"
                             style={{ display: "none", visibility: "hidden" }}
@@ -37,7 +38,9 @@ export default async function RootLayout({
                 <StyledComponentsRegistry>
                     <GlobalStyle />
                     <ResponsiveSpacingStyle />
-                    {draftMode().isEnabled ? <SitePreviewProvider>{children}</SitePreviewProvider> : children}
+                    <SiteConfigProvider siteConfig={siteConfig}>
+                        {isDraftModeEnabled ? <SitePreviewProvider>{children}</SitePreviewProvider> : children}
+                    </SiteConfigProvider>
                 </StyledComponentsRegistry>
             </body>
         </html>

diff --git a/site/src/app/block-preview/SiteConfigsProvider.tsx b/site/src/app/block-preview/SiteConfigsProvider.tsx
@@ -0,0 +1,22 @@
+"use client";
+
+import { useIFrameBridge } from "@comet/cms-site";
+import type { ContentScope, PublicSiteConfig } from "@src/site-configs";
+import { createContext, PropsWithChildren, useContext } from "react";
+
+const SiteConfigsContext = createContext<PublicSiteConfig[]>([]);
+
+export function useBlockPreviewSiteConfig(
+    getSiteConfigFromContentScope: (siteConfigs: PublicSiteConfig[], contentScope: ContentScope) => PublicSiteConfig | undefined,
+) {
+    const iFrameBridge = useIFrameBridge();
+    if (!iFrameBridge.hasBridge) throw new Error("IFrameBridge not initialized.");
+    const contentScope = iFrameBridge.contentScope as ContentScope;
+    const siteConfigs = useContext(SiteConfigsContext);
+    if (!contentScope) return;
+    return getSiteConfigFromContentScope(siteConfigs, contentScope);
+}
+
+export function SiteConfigsProvider({ children, siteConfigs }: PropsWithChildren<{ siteConfigs: PublicSiteConfig[] }>) {
+    return <SiteConfigsContext.Provider value={siteConfigs}>{children}</SiteConfigsContext.Provider>;
+}
diff --git a/site/src/app/block-preview/page/page.tsx → site/src/app/block-preview/[type]/page.tsx b/site/src/app/block-preview/page/page.tsx → site/src/app/block-preview/[type]/page.tsx
@@ -2,11 +2,15 @@
 import { BlockPreviewProvider, IFrameBridgeProvider, useBlockPreviewFetch, useIFrameBridge } from "@comet/cms-site";
 import { PageContentBlockData } from "@src/blocks.generated";
 import { PageContentBlock } from "@src/documents/pages/blocks/PageContentBlock";
+import { StageBlock } from "@src/documents/pages/blocks/StageBlock";
 import { graphQLApiUrl } from "@src/util/graphQLClient";
 import { recursivelyLoadBlockData } from "@src/util/recursivelyLoadBlockData";
+import { SiteConfigProvider } from "@src/util/SiteConfigProvider";
 import { useEffect, useState } from "react";
 
-const PreviewPage: React.FunctionComponent = () => {
+import { useBlockPreviewSiteConfig } from "../SiteConfigsProvider";
+
+const PagePreview: React.FunctionComponent = () => {
     const iFrameBridge = useIFrameBridge();
 
     const { fetch, graphQLFetch } = useBlockPreviewFetch(graphQLApiUrl);
@@ -31,14 +35,37 @@ const PreviewPage: React.FunctionComponent = () => {
 
     return <div>{blockData && <PageContentBlock data={blockData} />}</div>;
 };
-const IFrameBridgePreviewPage = () => {
+
+const StagePreview = () => {
+    const { block: stage } = useIFrameBridge();
+    return stage ? <StageBlock data={stage} /> : null;
+};
+
+const previewComponents = {
+    page: PagePreview,
+    stage: StagePreview,
+};
+
+const PreviewWrapper = ({ type }: { type: string }) => {
+    const siteConfig = useBlockPreviewSiteConfig((siteConfigs, contentScope) =>
+        siteConfigs.find((siteConfig) => siteConfig.scope.domain === contentScope.domain),
+    );
+    const PreviewComponent = previewComponents[type];
+    return (
+        <SiteConfigProvider siteConfig={siteConfig}>
+            <PreviewComponent />
+        </SiteConfigProvider>
+    );
+};
+
+const PreviewPage = ({ params }: { params: { type: string } }) => {
     return (
         <IFrameBridgeProvider>
             <BlockPreviewProvider>
-                <PreviewPage />
+                <PreviewWrapper type={params.type} />
             </BlockPreviewProvider>
         </IFrameBridgeProvider>
     );
 };
 
-export default IFrameBridgePreviewPage;
+export default PreviewPage;
diff --git a/site/src/app/block-preview/layout.tsx b/site/src/app/block-preview/layout.tsx
@@ -0,0 +1,21 @@
+import { GlobalStyle } from "@src/layout/GlobalStyle";
+import { getSiteConfigs } from "@src/middleware";
+import { ResponsiveSpacingStyle } from "@src/util/ResponsiveSpacingStyle";
+import StyledComponentsRegistry from "@src/util/StyledComponentsRegistry";
+
+import { SiteConfigsProvider } from "./SiteConfigsProvider";
+
+export default async function BlockPreviewLayout({ children }: { children: React.ReactNode }) {
+    const siteConfigs = getSiteConfigs();
+    return (
+        <html>
+            <body>
+                <StyledComponentsRegistry>
+                    <GlobalStyle />
+                    <ResponsiveSpacingStyle />
+                    <SiteConfigsProvider siteConfigs={siteConfigs}>{children}</SiteConfigsProvider>
+                </StyledComponentsRegistry>
+            </body>
+        </html>
+    );
+}
diff --git a/site/src/app/block-preview/stage/page.tsx b/site/src/app/block-preview/stage/page.tsx
diff --git a/site/src/config.ts b/site/src/config.ts
diff --git a/site/src/middleware.ts b/site/src/middleware.ts
@@ -1,18 +1,24 @@
 import { previewParams } from "@comet/cms-site";
-import { getSiteConfigs } from "@src/config";
 import { Rewrite } from "next/dist/lib/load-custom-routes";
 import { headers } from "next/headers";
 import type { NextRequest } from "next/server";
 import { NextResponse } from "next/server";
 
 import { createRedirects } from "./redirects/redirects";
+import type { PublicSiteConfig } from "./site-configs";
 
 function getHost(headers: Headers) {
     const host = headers.get("x-forwarded-host") ?? headers.get("host");
     if (!host) throw new Error("Could not evaluate host");
     return host;
 }
 
+export function getSiteConfigForDomain(domain: string) {
+    const siteConfig = getSiteConfigs().find((siteConfig) => siteConfig.scope.domain === domain);
+    if (!siteConfig) throw new Error(`SiteConfig not found for domain ${domain}`);
+    return siteConfig;
+}
+
 async function getSiteConfigForHost(host: string) {
     const sitePreviewParams = await previewParams({ skipDraftModeCheck: true });
     if (sitePreviewParams?.scope) {
@@ -22,6 +28,16 @@ async function getSiteConfigForHost(host: string) {
     return getSiteConfigs().find((siteConfig) => siteConfig.domains.main === host || siteConfig.domains.preliminary === host);
 }
 
+let siteConfigs: PublicSiteConfig[];
+export function getSiteConfigs() {
+    if (!siteConfigs) {
+        const json = process.env.NEXT_PUBLIC_SITE_CONFIGS;
+        if (!json) throw new Error("process.env.NEXT_PUBLIC_SITE_CONFIGS must be set.");
+        siteConfigs = JSON.parse(json) as PublicSiteConfig[];
+    }
+    return siteConfigs;
+}
+
 // Used for getting SiteConfig in server-components where params is not available (e.g. sitemap, not-found - see https://github.com/vercel/next.js/discussions/43179)
 export async function getSiteConfig() {
     const host = getHost(headers());

diff --git a/site/src/util/SiteConfigProvider.tsx b/site/src/util/SiteConfigProvider.tsx
@@ -0,0 +1,15 @@
+"use client";
+
+import type { PublicSiteConfig } from "@src/site-configs.d";
+import { createContext, PropsWithChildren, useContext } from "react";
+
+export const SiteConfigContext = createContext<PublicSiteConfig | undefined>(undefined);
+export const useSiteConfig = () => {
+    const siteConfig = useContext(SiteConfigContext);
+    if (!siteConfig) throw new Error("SiteConfig not set in SiteConfigProvider");
+    return siteConfig;
+};
+
+export function SiteConfigProvider({ children, siteConfig }: PropsWithChildren<{ siteConfig?: PublicSiteConfig }>) {
+    return <SiteConfigContext.Provider value={siteConfig}>{children}</SiteConfigContext.Provider>;
+}