From 61383d1ff9d569e62c657765190303826f8459a7 Mon Sep 17 00:00:00 2001
From: Pooya Parsa <pooya@pi0.io>
Date: Thu, 14 Sep 2023 19:02:48 +0200
Subject: [PATCH] feat(dev): reuse main dev port for vite hmr (#184)

---
 src/utils/dev.ts | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/src/utils/dev.ts b/src/utils/dev.ts
index c8c59d64..1b83273b 100644
--- a/src/utils/dev.ts
+++ b/src/utils/dev.ts
@@ -147,6 +147,28 @@ class NuxtDevServer extends EventEmitter {
       },
     })
 
+    // Connect Vite HMR
+    this._currentNuxt.hooks.hookOnce(
+      'vite:extendConfig',
+      (config, { isClient }) => {
+        if (isClient && config.server) {
+          config.server.hmr = {
+            ...(config.server.hmr as Exclude<
+              typeof config.server.hmr,
+              boolean
+            >),
+            protocol: undefined,
+            port: undefined,
+            host: undefined,
+            server: this.listener.server,
+          }
+        }
+      },
+    )
+    this._currentNuxt.hooks.hookOnce('close', () => {
+      this.listener.server.removeAllListeners('upgrade')
+    })
+
     // Write manifest and also check if we need cache invalidation
     if (!reload) {
       const previousManifest = await loadNuxtManifest(