Added TLS support

1 files changed, 5 insertions, 1 deletions

diff --git a/pages.go b/pages.go
index ebe284a..61ee0ae 100644
--- a/pages.go
+++ b/pages.go
@@ -44,5 +44,9 @@ func Home(rw http.ResponseWriter, req *http.Request) {
 
 func Chatpage(rw http.ResponseWriter, req *http.Request) {
 	vars := mux.Vars(req)
-	TplChat.Execute(rw, ChatpageData{"ws://" + req.Host + strings.Replace(req.URL.Path+"/socket", "//", "/", -1), vars["chatroom"]})
+	wsproto := "ws://"
+	if *usetls {
+		wsproto = "wss://"
+	}
+	TplChat.Execute(rw, ChatpageData{wsproto + req.Host + strings.Replace(req.URL.Path+"/socket", "//", "/", -1), vars["chatroom"]})
 }