fix: bundle ws types (#5340)

vitejs · Oct 18, 2021 · bc4a96c · bc4a96c
1 parent ffb57d6
commit bc4a96c
Show file tree

Hide file tree

Showing 5 changed files with 548 additions and 2 deletions.
diff --git a/packages/vite/package.json b/packages/vite/package.json
@@ -54,6 +54,8 @@
   },
   "devDependencies": {
     "@ampproject/remapping": "^1.0.1",
+    "@babel/parser": "^7.15.8",
+    "@babel/types": "^7.15.6",
     "@rollup/plugin-alias": "^3.1.5",
     "@rollup/plugin-commonjs": "^21.0.0",
     "@rollup/plugin-dynamic-import-vars": "^1.4.0",

diff --git a/packages/vite/src/node/index.ts b/packages/vite/src/node/index.ts
@@ -81,6 +81,7 @@ export type {
   ErrorPayload
 } from 'types/hmrPayload'
 export type { Connect } from 'types/connect'
+export type { WebSocket } from 'types/ws'
 export type { HttpProxy } from 'types/http-proxy'
 export type { FSWatcher, WatchOptions } from 'types/chokidar'
 export type { Terser } from 'types/terser'

diff --git a/packages/vite/src/node/server/ws.ts b/packages/vite/src/node/server/ws.ts
@@ -5,15 +5,16 @@ import {
   ServerOptions as HttpsServerOptions
 } from 'https'
 import WebSocket from 'ws'
+import { WebSocket as WebSocketTypes } from 'types/ws'
 import { ErrorPayload, HMRPayload } from 'types/hmrPayload'
 import { ResolvedConfig } from '..'
 import { isObject } from '../utils'
 import { Socket } from 'net'
 export const HMR_HEADER = 'vite-hmr'
 
 export interface WebSocketServer {
-  on: WebSocket.Server['on']
-  off: WebSocket.Server['off']
+  on: WebSocketTypes.Server['on']
+  off: WebSocketTypes.Server['off']
   send(payload: HMRPayload): void
   close(): Promise<void>
 }