@netlify/plugin-nextjs 4.41.4 → 4.41.6

package/src/templates/edge-shared/utils.ts

@@ -1,5 +1,5 @@
 import type { Context } from 'https://edge.netlify.com'
-import { ElementHandlers, HTMLRewriter } from '../vendor/deno.land/x/html_rewriter@v0.1.0-pre.17/index.ts'
+import { ElementHandlers, HTMLRewriter } from '../vendor/deno.land/x/htmlrewriter@v1.0.0/src/index.ts'
 
 export interface FetchEventResult {
   response: Response
@@ -209,41 +209,49 @@ export const buildResponse = async ({
       headers.set('content-length', String(body.length))
       return new Response(body, { ...response, headers })
     }
-    // This var will hold the contents of the script tag
-    let buffer = ''
-    // Create an HTMLRewriter that matches the Next data script tag
-    const rewriter = new HTMLRewriter()
-
-    if (response.dataTransforms.length > 0) {
-      rewriter.on('script[id="__NEXT_DATA__"]', {
-        text(textChunk) {
-          // Grab all the chunks in the Next data script tag
-          buffer += textChunk.text
-          if (textChunk.lastInTextNode) {
-            try {
-              // When we have all the data, try to parse it as JSON
-              const data = JSON.parse(buffer.trim())
-              // Apply all of the transforms to the props
-              const props = response.dataTransforms.reduce((prev, transform) => transform(prev), data.props)
-              // Replace the data with the transformed props
-              // With `html: true` the input is treated as raw HTML
-              // @see https://developers.cloudflare.com/workers/runtime-apis/html-rewriter/#global-types
-              textChunk.replace(JSON.stringify({ ...data, props }), { html: true })
-            } catch (err) {
-              console.log('Could not parse', err)
-            }
-          } else {
-            // Remove the chunk after we've appended it to the buffer
-            textChunk.remove()
-          }
-        },
-      })
-    }
 
-    if (response.elementHandlers.length > 0) {
-      response.elementHandlers.forEach(([selector, handlers]) => rewriter.on(selector, handlers))
+    if (response.dataTransforms.length > 0 && response.elementHandlers.length > 0) {
+      const { initHtmlRewriter } = await import('./html-rewriter-wasm.ts')
+      await initHtmlRewriter()
+
+      // This var will hold the contents of the script tag
+      let buffer = ''
+      // Create an HTMLRewriter that matches the Next data script tag
+      const rewriter = new HTMLRewriter()
+
+      if (response.dataTransforms.length > 0) {
+        rewriter.on('script[id="__NEXT_DATA__"]', {
+          text(textChunk) {
+            // Grab all the chunks in the Next data script tag
+            buffer += textChunk.text
+            if (textChunk.lastInTextNode) {
+              try {
+                // When we have all the data, try to parse it as JSON
+                const data = JSON.parse(buffer.trim())
+                // Apply all of the transforms to the props
+                const props = response.dataTransforms.reduce((prev, transform) => transform(prev), data.props)
+                // Replace the data with the transformed props
+                // With `html: true` the input is treated as raw HTML
+                // @see https://developers.cloudflare.com/workers/runtime-apis/html-rewriter/#global-types
+                textChunk.replace(JSON.stringify({ ...data, props }), { html: true })
+              } catch (err) {
+                console.log('Could not parse', err)
+              }
+            } else {
+              // Remove the chunk after we've appended it to the buffer
+              textChunk.remove()
+            }
+          },
+        })
+      }
+
+      if (response.elementHandlers.length > 0) {
+        response.elementHandlers.forEach(([selector, handlers]) => rewriter.on(selector, handlers))
+      }
+      return rewriter.transform(response.originResponse)
+    } else {
+      return response.originResponse
     }
-    return rewriter.transform(response.originResponse)
   }
   const res = new Response(result.response.body, result.response)
   request.headers.set('x-nf-next-middleware', 'skip')