server.js

// AFK WebSocket Server
// Bridges CLI agent ↔ Chrome Extension overlay
// Usage: node server.js

const { WebSocketServer } = require('ws');

const PORT = 38377;
const wss = new WebSocketServer({ port: PORT });
let lastBroadcast = null;

console.log(`AFK WebSocket server listening on ws://localhost:${PORT}`);

wss.on('connection', (ws, req) => {
  const origin = req.headers.origin || 'unknown';
  console.log(`[AFK] Client connected: ${origin}`);

  // Late joiner support: new extension tabs immediately get the latest overlay state.
  if (lastBroadcast && ws.readyState === 1) {
    ws.send(lastBroadcast);
  }

  ws.on('message', (data) => {
    // Broadcast to all other connected clients (CLI → Extension)
    const msg = data.toString();
    lastBroadcast = msg;
    console.log(`[AFK] Broadcast: ${msg.slice(0, 80)}`);
    wss.clients.forEach((client) => {
      if (client.readyState === 1) {
        client.send(msg);
      }
    });
  });

  ws.on('close', () => {
    console.log(`[AFK] Client disconnected`);
  });

  ws.on('error', (err) => {
    console.error(`[AFK] Error: ${err.message}`);
  });
});

wss.on('error', (err) => {
  if (err.code === 'EADDRINUSE') {
    console.error(`[AFK] Port ${PORT} already in use. Kill the existing process first.`);
  } else {
    console.error(`[AFK] Server error: ${err.message}`);
  }
  process.exit(1);
});