picoclaw/web/frontend/src/features/chat/websocket.ts

export function normalizeWsUrlForBrowser(wsUrl: string): string {
  let finalWsUrl = wsUrl

  try {
    const parsedUrl = new URL(wsUrl)
    const isLocalHost =
      parsedUrl.hostname === "localhost" ||
      parsedUrl.hostname === "127.0.0.1" ||
      parsedUrl.hostname === "0.0.0.0"
    const isBrowserLocal =
      window.location.hostname === "localhost" ||
      window.location.hostname === "127.0.0.1"

    if (isLocalHost && !isBrowserLocal) {
      parsedUrl.hostname = window.location.hostname
      finalWsUrl = parsedUrl.toString()
    } else if (
      isLocalHost &&
      isBrowserLocal &&
      parsedUrl.hostname !== window.location.hostname &&
      (parsedUrl.hostname === "127.0.0.1" ||
        parsedUrl.hostname === "localhost") &&
      (window.location.hostname === "127.0.0.1" ||
        window.location.hostname === "localhost")
    ) {
      // Same machine, but cookies are host-specific; match the page origin.
      parsedUrl.hostname = window.location.hostname
      finalWsUrl = parsedUrl.toString()
    }
  } catch (error) {
    console.warn("Could not parse ws_url:", error)
  }

  return finalWsUrl
}

export function invalidateSocket(socket: WebSocket | null) {
  if (!socket) {
    return
  }

  socket.onopen = null
  socket.onmessage = null
  socket.onclose = null
  socket.onerror = null
  socket.close()
}

export function isCurrentSocket({
  socket,
  currentSocket,
  generation,
  currentGeneration,
  sessionId,
  currentSessionId,
}: {
  socket: WebSocket
  currentSocket: WebSocket | null
  generation: number
  currentGeneration: number
  sessionId: string
  currentSessionId: string
}): boolean {
  return (
    currentSocket === socket &&
    generation === currentGeneration &&
    sessionId === currentSessionId
  )
}