Support assetPrefix specific protocol

packages/next/client/dev/error-overlay/websocket.ts

@@ -2,6 +2,19 @@ let source: WebSocket
 const eventCallbacks: ((event: any) => void)[] = []
 let lastActivity = Date.now()
 
+function getSocketProtocol(assetPrefix: string): string {
+  let protocol = location.protocol
+
+  try {
+    // assetPrefix is a url
+    protocol = new URL(assetPrefix).protocol
+
+    console.log('is a url!', protocol, new URL(assetPrefix))
+  } catch (_) {}
+
+  return protocol === 'http:' ? 'ws' : 'wss'
+}
+
 export function addMessageListener(cb: (event: any) => void) {
   eventCallbacks.push(cb)
 }
@@ -32,7 +45,7 @@ export function connectHMR(options: {
   function init() {
     if (source) source.close()
     const { hostname, port } = location
-    const protocol = location.protocol === 'http:' ? 'ws' : 'wss'
+    const protocol = getSocketProtocol(options.assetPrefix || '')
     const assetPrefix = options.assetPrefix.replace(/^\/+/, '')
 
     let url = `${protocol}://${hostname}:${port}${