feat: expose page and assets on build hooks TransformContext

2 years ago · 468c049ccd
parent b86988333b
commit 468c049ccd
3 changed files with 36 additions and 22 deletions
--- a/src/node/build/build.ts
+++ b/src/node/build/build.ts
@ -59,6 +59,12 @@ export async function build(
        (chunk) => chunk.type === 'asset' && chunk.fileName.endsWith('.css')
      ) as OutputAsset

+      const assets = (siteConfig.mpa ? serverResult : clientResult).output
+        .filter(
+          (chunk) => chunk.type === 'asset' && !chunk.fileName.endsWith('.css')
+        )
+        .map((asset) => siteConfig.site.base + asset.fileName)
+
      // We embed the hash map and site config strings into each page directly
      // so that it doesn't alter the main chunk's hash on every build.
      // It's also embedded as a string and JSON.parsed from the client because
@ -79,6 +85,7 @@ export async function build(
              clientResult,
              appChunk,
              cssChunk,
+              assets,
              pageToHashMap,
              hashMapString,
              siteDataString
--- a/src/node/build/render.ts
+++ b/src/node/build/render.ts
@ -26,6 +26,7 @@ export async function renderPage(
  result: RollupOutput | null,
  appChunk: OutputChunk | undefined,
  cssChunk: OutputAsset | undefined,
+  assets: string[],
  pageToHashMap: Record<string, string>,
  hashMapString: string,
  siteDataString: string
@ -86,21 +87,18 @@ export async function renderPage(
    preloadLinks = preloadLinks.filter((link) => shouldPreload(link, page))
  }

-  const preloadLinksString = preloadLinks
-    .map((file) => {
-      return `<link rel="modulepreload" href="${
-        EXTERNAL_URL_RE.test(file) ? '' : siteData.base // don't add base to external urls
-      }${file}">`
-    })
-    .join('\n    ')
+  const toHeadTags = (files: string[], rel: string): HeadConfig[] =>
+    files.map((file) => [
+      'link',
+      {
+        rel,
+        // don't add base to external urls
+        href: (EXTERNAL_URL_RE.test(file) ? '' : siteData.base) + file
+      }
+    ])

-  const prefetchLinkString = prefetchLinks
-    .map((file) => {
-      return `<link rel="prefetch" href="${
-        EXTERNAL_URL_RE.test(file) ? '' : siteData.base // don't add base to external urls
-      }${file}">`
-    })
-    .join('\n    ')
+  const preloadHeadTags = toHeadTags(preloadLinks, 'modulepreload')
+  const prefetchHeadTags = toHeadTags(prefetchLinks, 'prefetch')

  const stylesheetLink = cssChunk
    ? `<link rel="preload stylesheet" href="${siteData.base}${cssChunk.fileName}" as="style">`
@ -109,21 +107,27 @@ export async function renderPage(
  const title: string = createTitle(siteData, pageData)
  const description: string = pageData.description || siteData.description

-  const headBeforeTransform = mergeHead(
-    siteData.head,
-    filterOutHeadDescription(pageData.frontmatter.head)
-  )
+  const headBeforeTransform = [
+    ...preloadHeadTags,
+    ...prefetchHeadTags,
+    ...mergeHead(
+      siteData.head,
+      filterOutHeadDescription(pageData.frontmatter.head)
+    )
+  ]

  const head = mergeHead(
    headBeforeTransform,
    (await config.transformHead?.({
+      page,
      siteConfig: config,
      siteData,
      pageData,
      title,
      description,
      head: headBeforeTransform,
-      content
+      content,
+      assets
    })) || []
  )

@ -165,8 +169,6 @@ export async function renderPage(
        ? `<script type="module" src="${siteData.base}${appChunk.fileName}"></script>`
        : ``
    }
-    ${preloadLinksString}
-    ${prefetchLinkString}
    ${await renderHead(head)}
  </head>
  <body>${teleports?.body || ''}
@ -179,13 +181,15 @@ export async function renderPage(

  await fs.ensureDir(path.dirname(htmlFileName))
  const transformedHtml = await config.transformHtml?.(html, htmlFileName, {
+    page,
    siteConfig: config,
    siteData,
    pageData,
    title,
    description,
    head,
-    content
+    content,
+    assets
  })
  await fs.writeFile(htmlFileName, transformedHtml || html)
 }
--- a/src/node/config.ts
+++ b/src/node/config.ts
@ -43,6 +43,7 @@ export interface UserConfig<ThemeConfig = any>
  srcExclude?: string[]
  outDir?: string
  cacheDir?: string
+
  shouldPreload?: (link: string, page: string) => boolean

  locales?: LocaleConfig<ThemeConfig>
@ -142,6 +143,7 @@ export interface UserConfig<ThemeConfig = any>
 }

 export interface TransformContext {
+  page: string
  siteConfig: SiteConfig
  siteData: SiteData
  pageData: PageData
@ -149,6 +151,7 @@ export interface TransformContext {
  description: string
  head: HeadConfig[]
  content: string
+  assets: string[]
 }

 export type RawConfigExports<ThemeConfig = any> =