Dynamically detect ws or wss

cogment · Feb 29, 2024 · d3faa7c · d3faa7c
1 parent bb5af0f
commit d3faa7c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cogment_lab/humans/static/index.html b/cogment_lab/humans/static/index.html
@@ -17,7 +17,9 @@ <h1>RL Env Interface</h1>
 
         document.getElementById('start-button').addEventListener('click', function() {
             if (!isConnected) {
-                socket = new WebSocket("wss://" + location.host + "/ws");
+                const wsProtocol = location.protocol === 'https:' ? 'wss://' : 'ws://';
+
+                socket = new WebSocket(wsProtocol + location.host + "/ws");
                 socket.onmessage = function(event) {
                     let image = document.getElementById('env-render');
                     let json = JSON.parse(event.data);