feat(vite): support Vite's transformIndexHtml hook

nuxt · Nov 5, 2021 · dd65101 · dd65101
1 parent f461591
commit dd65101
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/packages/bridge/src/vite/client.ts b/packages/bridge/src/vite/client.ts
@@ -58,6 +58,12 @@ export async function buildClient (ctx: ViteBuildContext) {
 
   // Create development server
   const viteServer = await vite.createServer(clientConfig)
+
+  // Support Vite's transformIndexHtml hook
+  ctx.nuxt.hook('nitro:document', async (doc) => {
+    doc.contents = await viteServer.transformIndexHtml(doc.src, doc.contents) || doc.contents
+  })
+
   await ctx.nuxt.callHook('vite:serverCreated', viteServer)
 
   const viteMiddleware = (req, res, next) => {

diff --git a/packages/vite/src/client.ts b/packages/vite/src/client.ts
@@ -75,6 +75,12 @@ export async function buildClient (ctx: ViteBuildContext) {
   await ctx.nuxt.callHook('vite:extendConfig', clientConfig, { isClient: true, isServer: false })
 
   const viteServer = await vite.createServer(clientConfig)
+
+  // Support Vite's transformIndexHtml hook
+  ctx.nuxt.hook('nitro:document', async (doc) => {
+    doc.contents = await viteServer.transformIndexHtml(doc.src, doc.contents) || doc.contents
+  })
+
   await ctx.nuxt.callHook('vite:serverCreated', viteServer)
 
   const viteMiddleware: Connect.NextHandleFunction = (req, res, next) => {