let socket = null;
let msgCount = 0;

function setupSocket() {
  socket = new WebSocket("ws://localhost:5155/ws");

  socket.addEventListener("message", (event) => {
    msgCount += 1;
    const msgCountElement = document.getElementById("msgCount");
    if (msgCountElement) {
      msgCountElement.textContent = msgCount;
    }
    console.log("message: ", event.data);
  });

  socket.addEventListener("close", (event) => {
    const header = document.getElementById("msg");
    header.textContent = "Connection Closed";
    console.log("connection closed");
  });
}

function closeConn() {
  if (socket) {
    socket.close();
    socket = null;
    const header = document.getElementById("msg");
    header.textContent = "Connection Closed";
    console.log("connection closed");
  }
}

function openConn() {
  if (!socket || socket.readyState === WebSocket.CLOSED) {
    const header = document.getElementById("msg");
    header.textContent = "Connection Open";
    console.log("connection open");
    setupSocket();
  } else {
    console.log("Socket is already open or connecting.");
  }
}

// Start the first connection
setupSocket();