const WebSocket = require("ws");

const onlineUsers = new Map();

// 保存用户连接
function addConnection(username, ws) {
  onlineUsers.set(username, ws);

  console.log(`用户 ${username} 已连接`);
  console.log("当前在线用户：", [...onlineUsers.keys()]);
}

// 获取用户连接
function getConnection(username) {
  return onlineUsers.get(username);
}

// 删除用户连接
function removeConnection(username, ws) {
  const currentWs = onlineUsers.get(username);

  // 只有 Map 里面存的还是这个连接，才允许删除
  if (currentWs === ws) {
    onlineUsers.delete(username);
  }
  console.log(`用户 ${username} 已断开`);
  console.log("当前在线用户：", [...onlineUsers.keys()]);
}

// 获取所有在线用户
function getOnlineUsers() {
  return Array.from(onlineUsers.keys());
}

// 广播所有在线用户
function broadcast(data) {
  const message = JSON.stringify(data);

  for (const [username, ws] of onlineUsers) {
    if (ws.readyState === WebSocket.OPEN) {
      ws.send(message, (error) => {
        if (error) {
          console.error(`向用户 ${username} 广播失败：`, error);
        }
      });
    }
  }
}

module.exports = {
  addConnection,
  getConnection,
  removeConnection,
  getOnlineUsers,
  broadcast
};