server.js

import { WebSocketServer } from "ws";
import { v4 as uuidv4 } from "uuid";

const wss = new WebSocketServer({ port: 3000 });
console.log("Server running on ws://localhost:3000");

const rooms = {}; // { roomId: { host: ws, viewers: [] } }

// ============================================================================
// FLIGHT DATA SIMULATION (NEW FEATURE)
// ============================================================================
let flightData = {
  flightNumber: "AA42",
  altitude: 32000,
  speed: 540,
  timeToDestination: "2h 15m"
};

// Function to simulate realistic flight data changes
function simulateFlightData() {
  // Altitude fluctuation: ±50 feet
  const altitudeChange = Math.floor(Math.random() * 101) - 50;
  flightData.altitude = Math.max(30000, Math.min(35000, flightData.altitude + altitudeChange));
  
  // Speed fluctuation: ±5 mph
  const speedChange = Math.floor(Math.random() * 11) - 5;
  flightData.speed = Math.max(520, Math.min(560, flightData.speed + speedChange));
  
  // Time to destination countdown (simplified)
  const timeParts = flightData.timeToDestination.split(/[hm\s]+/).filter(Boolean);
  let hours = parseInt(timeParts[0]) || 0;
  let minutes = parseInt(timeParts[1]) || 0;
  
  // Decrease by 1 minute every update (5 seconds in real time)
  minutes -= 1;
  if (minutes < 0) {
    hours -= 1;
    minutes = 59;
  }
  
  if (hours < 0) {
    hours = 0;
    minutes = 0;
  }
  
  flightData.timeToDestination = `${hours}h ${minutes}m`;
  
  return flightData;
}

// Broadcast flight data every 5 seconds to all connected clients
setInterval(() => {
  const updatedFlightData = simulateFlightData();
  const message = JSON.stringify({ type: "flight_data_update", data: updatedFlightData });
  
  wss.clients.forEach(client => {
    if (client.readyState === 1) { // WebSocket.OPEN
      client.send(message);
    }
  });
  
  console.log('[Flight Data]', updatedFlightData);
}, 5000);
// ============================================================================

wss.on("connection", (ws) => {
  // Send initial flight data to new connections
  ws.send(JSON.stringify({ type: "flight_data_update", data: flightData }));

  ws.on("message", (msg) => {
    const data = JSON.parse(msg);

    switch (data.type) {
      case "create-room": {
        const roomId = uuidv4();
        rooms[roomId] = { host: ws, viewers: [] };
        ws.roomId = roomId;
        ws.isHost = true;
        ws.send(JSON.stringify({ type: "created", room: roomId }));
        broadcastRoomList();
        break;
      }

      case "list-rooms": {
        const roomList = Object.keys(rooms);
        ws.send(JSON.stringify({ type: "room-list", rooms: roomList }));
        break;
      }

      case "join-room": {
        const room = rooms[data.roomId];
        if (room) {
          room.viewers.push(ws);
          ws.roomId = data.roomId;
          ws.isHost = false;

          // Notify host and viewer
          room.host.send(JSON.stringify({ type: "joined", room: data.roomId }));
          ws.send(JSON.stringify({ type: "joined", room: data.roomId }));
        }
        break;
      }

      case "offer":
        // send offer to all viewers
        const hostRoom = rooms[ws.roomId];
        if (hostRoom && ws.isHost) {
          hostRoom.viewers.forEach(v => v.send(JSON.stringify({ type: "offer", offer: data.offer })));
        }
        break;

      case "answer":
        // send answer to host
        const viewerRoom = rooms[ws.roomId];
        if (viewerRoom && !ws.isHost) {
          viewerRoom.host.send(JSON.stringify({ type: "answer", answer: data.answer }));
        }
        break;

      case "ice-candidate":
        // relay ICE candidate
        const roomRef = rooms[ws.roomId];
        if (!roomRef) return;
        if (ws.isHost) {
          // send to all viewers
          roomRef.viewers.forEach(v => v.send(JSON.stringify({ type: "ice-candidate", candidate: data.candidate })));
        } else {
          // send to host
          roomRef.host.send(JSON.stringify({ type: "ice-candidate", candidate: data.candidate }));
        }
        break;
    }
  });

  ws.on("close", () => {
    // remove from room
    if (ws.roomId && rooms[ws.roomId]) {
      const room = rooms[ws.roomId];
      if (ws.isHost) {
        // notify viewers room is closed
        room.viewers.forEach(v => v.send(JSON.stringify({ type: "room-closed", roomId: ws.roomId })));
        delete rooms[ws.roomId];
      } else {
        room.viewers = room.viewers.filter(v => v !== ws);
      }
      broadcastRoomList();
    }
  });
});

function broadcastRoomList() {
  const list = Object.keys(rooms);
  wss.clients.forEach(ws => {
    ws.send(JSON.stringify({ type: "room-list", rooms: list }));
  });
}

console.log('✈️  Flight data simulation enabled - broadcasting every 5 seconds');