const WebSocket = require('ws');

const wss = new WebSocket.Server({ port: 8080 });

// Connection handling
wss.on('connection', (ws) => {
  console.log('Client connected');

  // Handle incoming messages
  ws.on('message', (data) => {
    console.log('Received:', data.toString());
    
    // Broadcast to all clients
    wss.clients.forEach((client) => {
      if (client.readyState === WebSocket.OPEN) {
        client.send(data);
      }
    });
  });

  // Handle disconnection
  ws.on('close', () => {
    console.log('Client disconnected');
  });

  // Send heartbeat
  const interval = setInterval(() => {
    if (ws.readyState === WebSocket.OPEN) {
      ws.ping();
    }
  }, 30000);

  ws.on('close', () => clearInterval(interval));
});

class WebSocketClient {
  constructor(url) {
    this.url = url;
    this.reconnectInterval = 5000;
    this.shouldReconnect = true;
    this.connect();
  }

  connect() {
    this.ws = new WebSocket(this.url);

    this.ws.onopen = () => {
      console.log('Connected');
      this.reconnectInterval = 5000; // Reset backoff
    };

    this.ws.onmessage = (event) => {
      console.log('Message:', event.data);
      // Handle incoming messages
    };

    this.ws.onclose = () => {
      console.log('Disconnected');
      if (this.shouldReconnect) {
        this.reconnect();
      }
    };

    this.ws.onerror = (error) => {
      console.error('Error:', error);
    };
  }

  reconnect() {
    setTimeout(() => {
      this.reconnectInterval *= 2; // Exponential backoff
      this.connect();
    }, this.reconnectInterval);
  }

  send(data) {
    if (this.ws.readyState === WebSocket.OPEN) {
      this.ws.send(JSON.stringify(data));
    }
  }
}

// Usage
const client = new WebSocketClient('ws://localhost:8080');
client.send({ type: 'chat', message: 'Hello!' });