import express from "express";
import { createServer } from "http";
import { WebSocketServer } from "ws";

const app = express();
const server = createServer(app);
const wss = new WebSocketServer({ server });

"color:#6a9955">// Configuration
const PORT = process.env.PORT || 3000;
const MAX_CONNECTIONS = 100;

interface Client {
  id: string;
  name: string;
  socket: WebSocket;
  joinedAt: Date;
}

const clients: Map<string, Client> = new Map();

"color:#6a9955">// Generate unique ID
function generateId(): string {
  return Math.random().toString(36).substring(2, 9);
}

"color:#6a9955">// Broadcast to all connected clients
function broadcast(message: object, exclude?: string) {
  const data = JSON.stringify(message);
  for (const [id, client] of clients) {
    if (id !== exclude) {
      client.socket.send(data);
    }
  }
}

wss.on("connection", (socket) => {
  if (clients.size >= MAX_CONNECTIONS) {
    socket.close(1013, "Server is full");
    return;
  }

  const id = generateId();
  const client: Client = {
    id,
    name: `User ${clients.size + 1}`,
    socket: socket as unknown as WebSocket,
    joinedAt: new Date(),
  };

  clients.set(id, client);
  console.log(`Client connected: ${id}`);

  "color:#6a9955">// Welcome message
  socket.send(JSON.stringify({
    type: "welcome",
    clientId: id,
    online: clients.size,
  }));

  broadcast({ type: "join", clientId: id }, id);

  socket.on("message", (raw) => {
    try {
      const msg = JSON.parse(raw.toString());
      broadcast({ ...msg, from: id }, id);
    } catch {
      "color:#6a9955">// Invalid JSON, ignore
    }
  });

  socket.on("close", () => {
    clients.delete(id);
    broadcast({ type: "leave", clientId: id });
  });
});

app.get("/health", (_, res) => {
  res.json({ status: "ok", connections: clients.size });
});

server.listen(PORT, () => {
  console.log(`Server running on port ${PORT}`);
});