fix: use listhen's publicUrl if available (#3500)

Author: farnabaz

src/runtime/internal/websocket.ts

@@ -1,5 +1,6 @@
 import { loadDatabaseAdapter } from './database.client'
 import { useRuntimeConfig, refreshNuxtData } from '#imports'
+import { joinURL } from 'ufo'
 
 const logger = {
   log: (...args: unknown[]) => console.log('[Content]', ...args),
@@ -79,9 +80,8 @@ export function useContentWebSocket() {
     }
 
     // WebSocket Base URL
-    const wsURL = new URL(`${(useRuntimeConfig().public.content as { wsUrl: string }).wsUrl}ws`)
+    const wsURL = new URL(joinURL((useRuntimeConfig().public.content as { wsUrl: string }).wsUrl, 'ws'))
     wsURL.protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:'
-    wsURL.hostname = window.location.hostname
 
     logger.log(`WS connect to ${wsURL}`)
 

src/utils/dev.ts

@@ -36,7 +36,8 @@ export async function startSocketServer(nuxt: Nuxt, options: ModuleOptions, mani
       listener = await listen(() => 'Nuxt Content', websocketOptions)
 
       // Register ws url
-      ;(nitro.options.runtimeConfig.public.content as Record<string, unknown>).wsUrl = listener.url.replace('http', 'ws')
+      const publicConfig = nitro.options.runtimeConfig.public.content as Record<string, unknown>
+      publicConfig.wsUrl = (websocketOptions.publicURL || listener.url).replace('http', 'ws')
 
       listener.server.on('upgrade', websocket.serve)
     })