From a917d0d2aa2b455aa02a15e409278158a8b48b17 Mon Sep 17 00:00:00 2001
From: Simon Oberzier <simon.oberzier@outlook.de>
Date: Mon, 9 Feb 2026 22:13:02 +0100
Subject: [PATCH] Persistant rooms

---
 public/index.html | 48 ++++++++++++++++++++-------------
 server.js         | 69 +++++++++++++++++++----------------------------
 2 files changed, 57 insertions(+), 60 deletions(-)

diff --git a/public/index.html b/public/index.html
index ecffeeb..8cfb09b 100644
--- a/public/index.html
+++ b/public/index.html
@@ -75,6 +75,14 @@
         button:hover {
             background: #34495e;
         }
+
+        #exitBtn {
+            background: #e74c3c;
+        }
+
+        #exitBtn:hover {
+            background: #c0392b;
+        }
     </style>
 </head>
 
@@ -86,8 +94,16 @@
     <br>
     <button id="restartBtn">🔄 Neustart</button>
     <button id="copyRoomBtn">📋 Link kopieren</button>
+    <button id="exitBtn">🚪 Raum verlassen</button>
 
     <script>
+        // Session Management
+        let sessionId = localStorage.getItem("game_session_id");
+        if (!sessionId) {
+            sessionId = Math.random().toString(36).substring(2) + Date.now().toString(36);
+            localStorage.setItem("game_session_id", sessionId);
+        }
+
         const params = new URLSearchParams(window.location.search);
         let roomId = params.get("room");
 
@@ -96,7 +112,7 @@
         }
 
         const protocol = location.protocol === "https:" ? "wss" : "ws";
-        const ws = new WebSocket(`${protocol}://${location.host}?room=${roomId}`);
+        const ws = new WebSocket(`${protocol}://${location.host}?room=${roomId}&sessionId=${sessionId}`);
         const svg = document.querySelector("svg");
 
         let myRole = null, state = null, selected = null, currentRoomId = null;
@@ -112,19 +128,17 @@
                 myRole = msg.role;
                 currentRoomId = msg.roomId;
                 document.getElementById("roleHeader").innerText = `Du bist: ${myRole.toUpperCase()} | Raum: ${currentRoomId}`;
+                if (!params.has("room")) {
+                    window.history.replaceState({}, '', `?room=${currentRoomId}`);
+                }
             }
-            if (msg.type === "state") {
-                state = msg.game;
-                render();
-            }
+            if (msg.type === "state") { state = msg.game; render(); }
             if (msg.type === "full") alert("Dieser Raum ist leider voll!");
         };
 
         function render() {
             if (!state) return;
             svg.innerHTML = "";
-
-            // Update Status UI
             const statusEl = document.getElementById("status");
             if (state.winner) {
                 statusEl.innerHTML = `<span style="color:${state.winner === myRole ? 'green' : 'red'}">SPIEL ENDE: ${state.winner.toUpperCase()} GEWINNT!</span>`;
@@ -133,17 +147,14 @@
                 statusEl.innerHTML = `${turnText}<br><small>Zug: ${state.moveCount} / 40</small>`;
             }
 
-            // Draw Lines
             edges.forEach(([a, b]) => {
                 const [x1, y1] = pos[a], [x2, y2] = pos[b];
                 const line = document.createElementNS("http://www.w3.org/2000/svg", "line");
-                line.setAttribute("x1", x1); line.setAttribute("y1", y1);
-                line.setAttribute("x2", x2); line.setAttribute("y2", y2);
+                line.setAttribute("x1", x1); line.setAttribute("y1", y1); line.setAttribute("x2", x2); line.setAttribute("y2", y2);
                 line.setAttribute("stroke", "#bdc3c7"); line.setAttribute("stroke-width", "3");
                 svg.appendChild(line);
             });
 
-            // Draw Nodes
             for (let i = 0; i <= 10; i++) {
                 const [x, y] = pos[i];
                 let type = "empty";
@@ -157,7 +168,6 @@
                 const circle = document.createElementNS("http://www.w3.org/2000/svg", "circle");
                 circle.setAttribute("cx", x); circle.setAttribute("cy", y); circle.setAttribute("r", 25);
                 circle.setAttribute("class", `node ${type} ${selected === i ? 'selected' : ''} ${isPossible ? 'possible' : ''}`);
-
                 circle.onclick = () => handleClick(i);
                 svg.appendChild(circle);
 
@@ -172,15 +182,12 @@
 
         function handleClick(i) {
             if (!state || state.turn !== myRole || state.winner) return;
-
-            const isMyPiece = (myRole === "wolf" && state.wolf === i) ||
-                (myRole === "sheep" && state.sheep.includes(i));
-
+            const isMyPiece = (myRole === "wolf" && state.wolf === i) || (myRole === "sheep" && state.sheep.includes(i));
             if (selected === null) {
                 if (isMyPiece) { selected = i; render(); }
             } else {
                 if (isMyPiece) {
-                    selected = (selected === i) ? null : i; // Toggle selection
+                    selected = (selected === i) ? null : i;
                 } else {
                     ws.send(JSON.stringify({ from: selected, to: i }));
                     selected = null;
@@ -190,11 +197,16 @@
         }
 
         document.getElementById("restartBtn").onclick = () => ws.send(JSON.stringify({ type: "restart" }));
-
         document.getElementById("copyRoomBtn").onclick = () => {
             const url = `${location.origin}${location.pathname}?room=${currentRoomId}`;
             navigator.clipboard.writeText(url).then(() => alert("Link kopiert!"));
         };
+        document.getElementById("exitBtn").onclick = () => {
+            if (confirm("Möchtest du den Raum verlassen?")) {
+                ws.send(JSON.stringify({ type: "leave" }));
+                window.location.href = window.location.pathname;
+            }
+        };
     </script>
 </body>
 
diff --git a/server.js b/server.js
index 13e34c8..02329b7 100644
--- a/server.js
+++ b/server.js
@@ -9,16 +9,15 @@ app.use(express.static("public"));
 const server = http.createServer(app);
 const wss = new WebSocket.Server({ server });
 
-// Movement logic: Wolf can move anywhere, Sheep only forward
 const board_wolf = {
-    0:[1,2,3], 1:[0,2,4,5], 2:[0,1,3,5], 3:[0,2,5,6], 4:[1,5,7],
-    5:[1,2,3,4,6,7,8,9], 6:[3,5,9], 7:[4,5,8,10], 8:[5,7,9,10],
-    9:[5,6,8,10], 10:[7,8,9]
+    0: [1, 2, 3], 1: [0, 2, 4, 5], 2: [0, 1, 3, 5], 3: [0, 2, 5, 6], 4: [1, 5, 7],
+    5: [1, 2, 3, 4, 6, 7, 8, 9], 6: [3, 5, 9], 7: [4, 5, 8, 10], 8: [5, 7, 9, 10],
+    9: [5, 6, 8, 10], 10: [7, 8, 9]
 };
 
 const board_sheep = {
-    0:[1,2,3], 1:[2,4,5], 2:[1,3,5], 3:[2,5,6], 4:[5,7],
-    5:[4,6,7,8,9], 6:[5,9], 7:[8,10], 8:[7,9,10], 9:[8,10], 10:[]
+    0: [1, 2, 3], 1: [2, 4, 5], 2: [1, 3, 5], 3: [2, 5, 6], 4: [5, 7],
+    5: [4, 6, 7, 8, 9], 6: [5, 9], 7: [8, 10], 8: [7, 9, 10], 9: [8, 10], 10: []
 };
 
 const rooms = {};
@@ -28,39 +27,27 @@ function broadcast(roomId) {
     if (!room) return;
     const msg = JSON.stringify({ type: "state", game: room.game });
     room.players.forEach(p => {
-        if (p.ws.readyState === WebSocket.OPEN) p.ws.send(msg);
+        if (p.ws && p.ws.readyState === WebSocket.OPEN) p.ws.send(msg);
     });
 }
 
 function checkWinConditions(game) {
-    // 1. Wolf reaches node 0
     if (game.wolf === 0) return "wolf";
-
-    // 2. Turn limit (40 moves)
     if (game.moveCount >= 40) return "wolf";
 
-    // Mobility Check helpers
-    const canWolfMove = board_wolf[game.wolf].some(to => 
-        !game.sheep.includes(to)
-    );
-    
-    const canSheepMove = game.sheep.some(from => 
+    const canWolfMove = board_wolf[game.wolf].some(to => !game.sheep.includes(to));
+    const canSheepMove = game.sheep.some(from =>
         board_sheep[from].some(to => to !== game.wolf && !game.sheep.includes(to))
     );
 
-    // 3. Wolf trapped
     if (!canWolfMove) return "sheep";
-
-    // 4. Sheep cannot move
     if (!canSheepMove && game.turn === "sheep") return "wolf";
-
     return null;
 }
 
 function handleMove(roomId, role, { from, to }) {
     const room = rooms[roomId];
     if (!room || room.game.winner) return;
-
     const game = room.game;
     if (game.turn !== role) return;
 
@@ -78,20 +65,16 @@ function handleMove(roomId, role, { from, to }) {
         game.sheep[i] = to;
         game.turn = "wolf";
     }
-
     game.moveCount++;
     game.winner = checkWinConditions(game);
     broadcast(roomId);
 }
 
 wss.on("connection", (ws, req) => {
-    ws.isAlive = true;
-    ws.on("pong", () => ws.isAlive = true);
-
     const urlParams = new URLSearchParams(req.url.split('?')[1]);
     let roomId = urlParams.get("room");
+    let sessionId = urlParams.get("sessionId");
 
-    // Room creation/joining logic
     if (!roomId || !rooms[roomId]) {
         if (!roomId) roomId = randomBytes(3).toString("hex");
         rooms[roomId] = {
@@ -101,16 +84,23 @@ wss.on("connection", (ws, req) => {
     }
 
     const room = rooms[roomId];
-    if (room.players.length >= 2) {
+    let player = room.players.find(p => p.sessionId === sessionId);
+
+    if (!player && room.players.length >= 2) {
         ws.send(JSON.stringify({ type: "full" }));
         ws.close();
         return;
     }
 
-    const role = room.players.length === 0 ? "sheep" : "wolf";
-    room.players.push({ ws, role });
+    if (player) {
+        player.ws = ws; // Re-attach new socket to existing session
+    } else {
+        const role = room.players.length === 0 ? "sheep" : "wolf";
+        player = { ws, role, sessionId };
+        room.players.push(player);
+    }
 
-    ws.send(JSON.stringify({ type: "role", role, roomId }));
+    ws.send(JSON.stringify({ type: "role", role: player.role, roomId }));
     broadcast(roomId);
 
     ws.on("message", msg => {
@@ -120,23 +110,18 @@ wss.on("connection", (ws, req) => {
         if (data.type === "restart") {
             room.game = { wolf: 5, sheep: [0, 1, 3], turn: "sheep", moveCount: 0, winner: null };
             broadcast(roomId);
-            return;
+        } else if (data.type === "leave") {
+            room.players = room.players.filter(p => p.sessionId !== sessionId);
+            if (room.players.length === 0) delete rooms[roomId];
+            ws.close();
+        } else {
+            handleMove(roomId, player.role, data);
         }
-        handleMove(roomId, role, data);
     });
 
     ws.on("close", () => {
-        room.players = room.players.filter(p => p.ws !== ws);
-        if (room.players.length === 0) delete rooms[roomId];
+        // We don't remove the player here so they can reconnect on refresh
     });
 });
 
-setInterval(() => {
-    wss.clients.forEach(ws => {
-        if (!ws.isAlive) return ws.terminate();
-        ws.isAlive = false;
-        ws.ping();
-    });
-}, 30000);
-
 server.listen(3030, () => console.log("Server läuft auf Port 3030"));
\ No newline at end of file