import { Response } from 'express';

const sseClients = new Set<Response>();

export function addSSEClient(res: Response): void {
  sseClients.add(res);
}

export function removeSSEClient(res: Response): void {
  sseClients.delete(res);
}

export function sseBroadcast(data: Record<string, any>): void {
  const msg = `data: ${JSON.stringify(data)}\n\n`;
  for (const c of sseClients) {
    try { c.write(msg); } catch { sseClients.delete(c); }
  }
}

export function getSSEClientCount(): number {
  return sseClients.size;
}