allow for secure protocols

ms.go

@@ -21,6 +21,10 @@ func main() {
 	if os.Getenv("SERVER_NAME") != "" {
 		serverName = os.Getenv("SERVER_NAME")
 	}
+	secureProtocol := ""
+	if os.Getenv("SECURE_PROTOCOL") == "1" {
+		secureProtocol = "s"
+	}
 
 	rand.Seed(time.Now().Unix())
 
@@ -30,8 +34,8 @@ func main() {
 
 	router.GET("/", func(c *gin.Context) {
 		c.HTML(http.StatusOK, "index.html", gin.H{
-			"websocket": "ws://" + serverName + ":" + port + "/ws",
-			"sse":       "http://" + serverName + ":" + port + "/sse",
+			"websocket": "ws" + secureProtocol + "://" + serverName + ":" + port + "/ws",
+			"sse":       "http" + secureProtocol + "://" + serverName + ":" + port + "/sse",
 		})
 	})
 	router.GET("/ws", websocketHandler(websocket.Upgrader{