From ce2f902c1f868df2680a96c531d091b9a53f80f7 Mon Sep 17 00:00:00 2001
From: Max Schmitt <max@schmitt.mx>
Date: Tue, 6 Jun 2023 20:09:39 +0200
Subject: [PATCH] chore: fix UI Mode on Codespaces (#23549)

---
 packages/trace-viewer/src/ui/uiModeView.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/trace-viewer/src/ui/uiModeView.tsx b/packages/trace-viewer/src/ui/uiModeView.tsx
index d302df1b95..9201da88c9 100644
--- a/packages/trace-viewer/src/ui/uiModeView.tsx
+++ b/packages/trace-viewer/src/ui/uiModeView.tsx
@@ -644,7 +644,7 @@ const callbacks = new Map<number, { resolve: (arg: any) => void, reject: (arg: E
 
 const initWebSocket = async (onClose: () => void) => {
   const guid = new URLSearchParams(window.location.search).get('ws');
-  const ws = new WebSocket(`ws://${window.location.hostname}:${window.location.port}/${guid}`);
+  const ws = new WebSocket(`${window.location.protocol === 'https:' ? 'wss' : 'ws'}://${window.location.hostname}:${window.location.port}/${guid}`);
   await new Promise(f => ws.addEventListener('open', f));
   ws.addEventListener('close', onClose);
   ws.addEventListener('message', event => {