Class WsUpstream

tracker/src/upstream.ts:33–122

}

export class WsUpstream implements Upstream {
  private ws: WebSocket | null = null;
  private connected = false;
  private closed = false;
  private reconnectTimer: ReturnType<typeof setTimeout> | null = null;
  private config: Config = DEFAULT_CONFIG;
  private aircraft = new Map<string, Aircraft>();

  constructor(
    private url: string,
    private events: UpstreamEvents = {},
  ) {}

  start(): void {
    this.closed = false;
    this.open();
  }

  stop(): void {
    this.closed = true;
    if (this.reconnectTimer) clearTimeout(this.reconnectTimer);
    this.ws?.close();
  }

  isConnected(): boolean {
    return this.connected;
  }
  getConfig(): Config {
    return this.config;
  }
  getAircraft(): Aircraft[] {
    return [...this.aircraft.values()];
  }
  find(hex: string): Aircraft | undefined {
    return this.aircraft.get(hex);
  }

  patchConfig(patch: Partial<Config>): void {
    const msg: ClientMessage = { type: "patchConfig", patch };
    if (this.ws?.readyState === WebSocket.OPEN) {
      this.ws.send(JSON.stringify(msg));
    }
    // Optimistic local merge so the tracker reacts immediately.
    this.config = mergeConfig(this.config, patch);
    this.events.onConfig?.(this.config);
  }

  private open(): void {
    const ws = new WebSocket(this.url);
    this.ws = ws;
    ws.on("open", () => {
      this.connected = true;
      ws.send(JSON.stringify({ type: "hello", role: "control" } satisfies ClientMessage));
    });
    ws.on("close", () => {
      this.connected = false;
      this.scheduleReconnect();
    });