using wss if on https

NivEz · Aug 19, 2023 · eec9d4f · eec9d4f
1 parent 18b0d62
commit eec9d4f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/client/src/hooks/useWebsocket.js b/client/src/hooks/useWebsocket.js
@@ -5,7 +5,10 @@ export const useWebsocket = ({
 }) => {
     const [reMemo, setReMemo] = useState(null);
 
-    const ws = useMemo(() => new WebSocket(`ws://${host}`), [reMemo]);
+    const ws = useMemo(() => {
+        const protocol = location.protocol === "https:" ? "wss" : "ws";
+        return new WebSocket(`${protocol}://${host}`), [reMemo];
+    });
 
     const sendMessage = useCallback((commandType, commandValue = '') => {
         ws.send(`${commandType}!${commandValue}`);