feat: support wss protocol

dev6699 · Aug 5, 2024 · 47b9bd1 · 47b9bd1
1 parent c9e5e38
commit 47b9bd1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/view/script.js b/view/script.js
@@ -1,7 +1,9 @@
 (() => {
   const BASE64_PREFIX = "data:image/png;base64,"
 
-  const socket = new WebSocket("ws://" + window.location.host + "/ws");
+  const wsProtocol = window.location.protocol === 'https:' ? 'wss://' : 'ws://';
+
+  const socket = new WebSocket(wsProtocol + window.location.host + "/ws");
   socket.onopen = () => {
     console.log("Successfully Connected");
     socket.send("Hi From the Client!");