Added Leave Button

gambling.js

@@ -90,25 +90,21 @@ app.post("/register", (request, response) => {
 app.get("/login", (request, response) => {
     response.sendFile(path.join(__dirname, "./public/login.html"));
 })
+
 app.post("/login", (request, response) => {
     let {username, password} = request.body;
     username = username.toLowerCase();
-
     if (!username || !password)
         return response.status(400).json({error: "Username and password are required"});
-
     const users = get_data();
     const user = users.find(u => u.username === username);
     if (!user) return response.status(401).json({error: "Invalid credentials"});
-
     if (user.password !== password) {
         return response.status(401).json({error: "Invalid password"});
     }
-
     const token = generateAccessToken({username});
     user.token = token;
     save_data(users);
-
     response.json({username, token});
 });
 
@@ -116,23 +112,18 @@ app.get("/", (req, res) => {
     const token =
         req.headers.authorization?.split(" ")[1] ||
         req.cookies.authorization;
-
     if (!token) {
         return res.sendFile(path.join(__dirname, "./public/unauthorized.html"));
     }
-
     jwt.verify(token, process.env.TOKEN_SECRET, (err, user) => {
         if (err) {
             return res.sendFile(path.join(__dirname, "./public/unauthorized.html"));
         }
-
         const users = get_data();
         const foundUser = users.find(u => u.username === user.username);
-
         if (!foundUser || foundUser.token !== token) {
             return res.sendFile(path.join(__dirname, "./public/unauthorized.html"));
         }
-
         req.user = user;
         res.sendFile(path.join(__dirname, "./public/home.html"));
     });
@@ -146,11 +137,9 @@ app.get("/balance", authenticateToken, (request, response) => {
     console.log(username);
     const allUsers = get_data();
     const me = allUsers.find(u => u.username === username);
-
     if (!me) {
         return response.status(404).json({error: "User not found"});
     }
-
     response.json({balance: me.money}).status(200);
 });
 
@@ -166,11 +155,9 @@ app.delete("/logout", authenticateToken, (request, response) => {
     const users = get_data();
     const user = users.find(u => u.username === username);
     let lobbyId = request.lobbyId;
-
     if (playerMap[lobbyId].includes(username)) {
         playerMap[lobbyId].splice(username, 1);
     }
-
     // 1) Clear the HttpOnly cookie in the browser
     response.clearCookie("authorization", {
         httpOnly: true,
@@ -213,17 +200,13 @@ async function joinLobby(lobbyId, player_name) {
 
 app.get("/lobby/create", authenticateToken, async (req, res) => {
     const username = req.query.username;
-
     if (!username) {
         return res.status(400).send("Username is required");
     }
-
     let lobbyId = await createLobby();
-
     if (!playerMap[lobbyId]) {
         playerMap[lobbyId] = [];
     }
-
     if (!playerMap[lobbyId].includes(username)) {
         playerMap[lobbyId].push(username);
     }
@@ -253,15 +236,21 @@ app.get("/lobby/join", authenticateToken, async (req, res) => {
 });
 
 app.delete("/lobby/leave", authenticateToken, async (req, res) => {
-    let lobbyId = req.query.lobbyId;
-    let username = req.query.username;
-
-    console.log(playerMap);
+    const lobbyId = req.query.lobbyId;
+    const username = req.query.username;
+    if (!lobbyId || !username) {
+        return res.status(400).json({error: "Missing lobbyId or username"});
+    }
     if (lobbyId in playerMap) {
-        playerMap[lobbyId].splice(username, 1);
-
-        res.send("Succesfully logged out.");
-    } else console.log("Couldn't leave");
+        const index = playerMap[lobbyId].indexOf(username);
+        if (index > -1) {
+            playerMap[lobbyId].splice(index, 1);
+            return res.json({message: "Successfully left lobby"});
+        } else {
+            return res.status(404).json({error: "User not in lobby"});
+        }
+    }
+    res.status(404).json({error: "Lobby not found"});
 });
 
 app.get("/lobby/players", authenticateToken, async (req, res) => {

public/frontend_home.js

@@ -6,6 +6,28 @@ document.addEventListener("DOMContentLoaded", async () => {
     const joinForm = document.getElementById("join-form");
     const joinCodeInput = document.getElementById("join-code");
     const joinSubmit = document.getElementById("submit-join");
+    const leaveButton = document.getElementById("leave-button");
+
+    leaveButton.addEventListener("click", async () => {
+        const username = await getUsernameFromToken();
+        const lobbyId = document.getElementById("lobby-id").textContent;
+        try {
+            const res = await fetch(`/lobby/leave?lobbyId=${lobbyId}&username=${username}`, {
+                method: "DELETE",
+            });
+            if (res.ok) {
+                document.getElementById("lobby-room").classList.add("hidden");
+                document.getElementById("player-list").innerHTML = "<p>Waiting for players...</p>";
+                document.getElementById("lobby-id").textContent = "N/A";
+            } else {
+                const err = await res.json();
+                alert("Failed to leave lobby: " + (err.error || "Unknown error"));
+            }
+        } catch (err) {
+            console.error("Error leaving lobby:", err);
+            alert("Could not leave lobby.");
+        }
+    });
 
     const getUsernameFromToken = async () => {
         try {
@@ -17,6 +39,7 @@ document.addEventListener("DOMContentLoaded", async () => {
             return null;
         }
     };
+
     try {
         const balanceRes = await fetch('/balance', {method: "GET"});
         if (balanceRes.ok) {

public/home.html

@@ -21,6 +21,7 @@
 <div id="join-error" class="error-message hidden"></div>
 <div id="lobby-room" class="lobby-room hidden">
     <h2>Lobby ID: <span id="lobby-id">N/A</span></h2>
+    <button id="leave-button">Leave Lobby</button>
     <div id="player-list">
         <p>Waiting for players...</p>
     </div>