www: fix ws for visitors

1 files changed, 7 insertions, 1 deletions

diff --git a/juick-www/src/main/static/scripts.js b/juick-www/src/main/static/scripts.js
index 6a2ba370..588e4fdb 100644
--- a/juick-www/src/main/static/scripts.js
+++ b/juick-www/src/main/static/scripts.js
@@ -131,7 +131,13 @@ function initWS() {
     if (hash) {
         url += '?hash=' + hash;
     } else {
-        url += pageMID;
+        let content = document.getElementById('content');
+        if (content) {
+            let pageMID = content.getAttribute('data-mid');
+            if (pageMID) {
+                url += pageMID;
+            }
+        }
     }
 
     ws = new WebSocket(url);