From 751e81923198c9e0a840540ff7f2d2d205f3500e Mon Sep 17 00:00:00 2001
From: Hayleigh Thompson <me@hayleigh.dev>
Date: Wed, 12 Jun 2024 17:03:11 +0100
Subject: :recycle: Use wss protocol if page is served over https.

---
 src/server-component.mjs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/server-component.mjs b/src/server-component.mjs
index ab08801..0700939 100644
--- a/src/server-component.mjs
+++ b/src/server-component.mjs
@@ -60,9 +60,12 @@ export class LustreServerComponent extends HTMLElement {
         } else if (prev !== next) {
           const id = this.getAttribute("id");
           const route = next + (id ? `?id=${id}` : "");
+          const protocol = window.location.protocol === "https:" ? "wss" : "ws";
 
           this.#socket?.close();
-          this.#socket = new WebSocket(`ws://${window.location.host}${route}`);
+          this.#socket = new WebSocket(
+            `${protocol}://${window.location.host}${route}`,
+          );
           this.#socket.addEventListener("message", (message) =>
             this.messageReceivedCallback(message),
           );
-- 
cgit v1.2.3