fix(vite): compatible with vite4 for messaging

vuejs · Mar 22, 2024 · aec7f73 · aec7f73
1 parent d45ed17
commit aec7f73
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/packages/core/src/vite-bridge/client.ts b/packages/core/src/vite-bridge/client.ts
@@ -21,7 +21,13 @@ export function callViteServerAction<T>(name: string) {
     const uniqueEventKey = nanoid()
     return new Promise<T>((resolve) => {
       const cb = (e: T) => {
-        viteClient.off(uniqueEventKey, cb)
+        if (viteClient.off)
+          viteClient.off(uniqueEventKey, cb)
+
+        else
+          // compatible with vite 4.x
+          viteClient.dispose?.(cb)
+
         resolve(e)
       }
       viteClient.on(uniqueEventKey, cb)
@@ -38,7 +44,12 @@ export function defineViteClientListener(name: string) {
     const viteClient = getViteClientHotContext()
     viteClient.on(name, listener)
     return () => {
-      viteClient.off(name, listener)
+      if (viteClient.off)
+        viteClient.off(name, listener)
+
+      else
+        // compatible with vite 4.x
+        viteClient.dispose?.(listener)
     }
   }
 }