@aeroswarm/client 0.1.11 → 0.1.12

package/dist/react/index.js

@@ -23,11 +23,13 @@ var AeroWsClient = class {
   lastSeq = -1;
   lastHeartbeat = Date.now();
   heartbeatTimer = null;
+  reconnectTimer = null;
   status = "idle";
   connect(opts, callbacks) {
     this.lastSubscribe = opts;
     this.callbacks = callbacks;
     this.stopped = false;
+    this._clearReconnectTimer();
     this._setStatus("connecting");
     this._connect();
   }
@@ -38,10 +40,27 @@ var AeroWsClient = class {
   disconnect() {
     this.stopped = true;
     this._clearHeartbeatTimer();
-    this.ws?.close();
-    this.ws = null;
+    this._clearReconnectTimer();
+    this._dropSocket();
     this._setStatus("offline");
   }
+  /**
+   * Force a fresh connection now, skipping any backoff wait. Re-subscribes with
+   * the last `connect()` options (the server then sends a new snapshot).
+   * Also resumes after `disconnect()`. No-op if `connect()` was never called.
+   * Returns false in that no-op case.
+   */
+  reconnect() {
+    if (!this.lastSubscribe) return false;
+    this.stopped = false;
+    this.attempt = 0;
+    this._clearHeartbeatTimer();
+    this._clearReconnectTimer();
+    this._dropSocket();
+    this._setStatus("reconnecting");
+    this._connect();
+    return true;
+  }
   getStatus() {
     return this.status;
   }
@@ -50,8 +69,17 @@ var AeroWsClient = class {
     this.status = s;
     this.callbacks.onStatusChange?.(s);
   }
+  /** Close the current socket without letting its onclose schedule a reconnect. */
+  _dropSocket() {
+    const ws = this.ws;
+    this.ws = null;
+    if (!ws) return;
+    ws.onopen = ws.onmessage = ws.onclose = ws.onerror = null;
+    ws.close();
+  }
   _connect() {
     const wsUrl = `${this.url}?token=${encodeURIComponent(this.token)}`;
+    this._dropSocket();
     try {
       this.ws = new WebSocket(wsUrl);
       this.ws.binaryType = "arraybuffer";
@@ -167,10 +195,18 @@ var AeroWsClient = class {
   _scheduleReconnect() {
     const delay = Math.min(1e3 * 2 ** this.attempt, this.maxReconnectDelay);
     this.attempt++;
-    setTimeout(() => {
+    this._clearReconnectTimer();
+    this.reconnectTimer = setTimeout(() => {
+      this.reconnectTimer = null;
       if (!this.stopped) this._connect();
     }, delay);
   }
+  _clearReconnectTimer() {
+    if (this.reconnectTimer) {
+      clearTimeout(this.reconnectTimer);
+      this.reconnectTimer = null;
+    }
+  }
 };
 
 // src/sdk/websocket/replay.ts
@@ -320,6 +356,28 @@ var AeroRestClient = class {
     const qs = search.size ? `?${search.toString()}` : "";
     return this.req("GET", `/v1/tracks/current${qs}`);
   }
+  /** Flight plans received from AFTN feeds (FPL/CHG/DLA/DEP/ARR/CNL folded per flight). */
+  listFlightPlans(params) {
+    const search = new URLSearchParams();
+    if (params?.airportId) search.set("airportId", params.airportId);
+    if (params?.status) search.set("status", params.status);
+    if (params?.callsign) search.set("callsign", params.callsign);
+    if (params?.limit != null) search.set("limit", String(params.limit));
+    const qs = search.size ? `?${search.toString()}` : "";
+    return this.req("GET", `/v1/flight-plans${qs}`);
+  }
+  getFlightPlan(id) {
+    return this.req("GET", `/v1/flight-plans/${encodeURIComponent(id)}`);
+  }
+  /** Raw AFTN messages (every parsed ATS message, newest first — the input
+   *  side of the flight-plan pipeline). */
+  listAftnMessages(params) {
+    const search = new URLSearchParams();
+    if (params?.airportId) search.set("airportId", params.airportId);
+    if (params?.limit != null) search.set("limit", String(params.limit));
+    const qs = search.size ? `?${search.toString()}` : "";
+    return this.req("GET", `/v1/aftn/messages${qs}`);
+  }
   health() {
     return this.req("GET", "/health");
   }
@@ -351,6 +409,38 @@ var AeroGraphQLClient = class {
 
 // src/sdk/cache/index.ts
 import { isRenderableTrack } from "@aeroswarm/core/utils";
+
+// src/shared/label/index.ts
+var LABEL_COLORS = {
+  ARRIVAL: "#FACC15",
+  // yellow
+  DEPARTURE: "#22C55E",
+  // green
+  GROUND_VEHICLE: "#3B82F6",
+  // blue
+  LOST: "#FFFFFF",
+  // white
+  UNKNOWN: "#FFFFFF"
+  // white — no flight plan
+};
+function getLabelCategory(track) {
+  if (track.status === "lost") return "LOST";
+  if (track.targetType === "vehicle") return "GROUND_VEHICLE";
+  switch (track.flightPlan?.phase) {
+    case "ARRIVAL":
+      return "ARRIVAL";
+    case "DEPARTURE":
+      return "DEPARTURE";
+    default:
+      return "UNKNOWN";
+  }
+}
+function withLabel(track) {
+  const labelCategory = getLabelCategory(track);
+  return { ...track, labelCategory, labelColor: LABEL_COLORS[labelCategory] };
+}
+
+// src/sdk/cache/index.ts
 var TypedEmitter = class {
   listeners = /* @__PURE__ */ new Map();
   on(event, fn) {
@@ -364,15 +454,18 @@ var TypedEmitter = class {
 };
 var TrackCache = class extends TypedEmitter {
   tracks = /* @__PURE__ */ new Map();
+  /** Merges into the cached track and returns the labeled result. A "lost"
+   *  track is evicted from the cache but still returned (labelCategory LOST). */
   applyUpdate(track) {
+    const merged = withLabel({ ...this.tracks.get(track.trackId), ...track });
     if (track.status === "lost") {
       this.tracks.delete(track.trackId);
       this.emit("removed", { trackId: track.trackId });
-      return;
+      return merged;
     }
-    const merged = { ...this.tracks.get(track.trackId), ...track };
     this.tracks.set(track.trackId, merged);
     this.emit("update", merged);
+    return merged;
   }
   applyRemoved(ev) {
     this.tracks.delete(ev.trackId);
@@ -381,9 +474,11 @@ var TrackCache = class extends TypedEmitter {
   applySnapshot(tracks) {
     this.tracks.clear();
     for (const t of tracks) {
-      if (isRenderableTrack(t)) this.tracks.set(t.trackId, t);
+      if (isRenderableTrack(t)) this.tracks.set(t.trackId, withLabel(t));
     }
-    this.emit("snapshot", Array.from(this.tracks.values()));
+    const all = Array.from(this.tracks.values());
+    this.emit("snapshot", all);
+    return all;
   }
   getAll() {
     return Array.from(this.tracks.values());
@@ -477,14 +572,14 @@ var AeroSwarm = class _AeroSwarm {
             callbacks.onStatusChange?.(s);
           },
           onSnapshot: (tracks) => {
-            this.cache.applySnapshot(tracks);
+            const labeled = this.cache.applySnapshot(tracks);
             this._lastUpdateAt = Date.now();
-            callbacks.onSnapshot?.(tracks);
+            callbacks.onSnapshot?.(labeled);
           },
           onTrackUpdate: (track) => {
-            this.cache.applyUpdate(track);
+            const labeled = this.cache.applyUpdate(track);
             this._lastUpdateAt = Date.now();
-            callbacks.onTrackUpdate?.(track);
+            callbacks.onTrackUpdate?.(labeled);
           },
           onTrackRemoved: (ev) => {
             this.cache.applyRemoved(ev);
@@ -503,12 +598,20 @@ var AeroSwarm = class _AeroSwarm {
       return () => this.wsClient.disconnect();
     },
     updateFilter: (opts) => this.wsClient.updateFilter(opts),
+    /** Drop the live socket and reconnect immediately (no backoff wait),
+     *  re-subscribing with the last options. Returns false if nothing has
+     *  been subscribed yet. */
+    reconnect: () => this.wsClient.reconnect(),
     stream: (opts) => {
       const queue = [];
       let resolve = null;
       let done = false;
       this.wsClient.connect({ encoding: "json", ...opts }, {
-        onTrackUpdate: (t) => {
+        onSnapshot: (tracks) => {
+          this.cache.applySnapshot(tracks);
+        },
+        onTrackUpdate: (raw) => {
+          const t = this.cache.applyUpdate(raw);
           if (resolve) {
             resolve({ value: t, done: false });
             resolve = null;
@@ -549,6 +652,17 @@ var AeroSwarm = class _AeroSwarm {
       return this.trpc.tracks.history.query(input);
     }
   };
+  /** Flight plans received from AFTN feeds. Live tracks that correlate with a
+   *  plan (Mode-S address, else callsign) also carry `track.flightPlan`. */
+  flightPlans = {
+    list: (input) => this.rest.listFlightPlans(input).then((r) => r.data.flightPlans),
+    get: (id) => this.rest.getFlightPlan(id).then((r) => r.data.flightPlan)
+  };
+  /** Raw AFTN/ATS messages received by AFTN feeds — the input side of the
+   *  flight-plan pipeline (newest first, capped by the service). */
+  aftn = {
+    messages: (input) => this.rest.listAftnMessages(input).then((r) => r.data.messages)
+  };
   sources = {
     /** Registers a source and returns its token, tied to `airportId` — a
      *  source is a sensor physically at one airport, so this is required.
@@ -669,6 +783,9 @@ var AeroSwarm = class _AeroSwarm {
   disconnect() {
     this.wsClient.disconnect();
   }
+  reconnect() {
+    return this.wsClient.reconnect();
+  }
 };
 
 // src/react/context/index.tsx
@@ -917,7 +1034,7 @@ function useTrackHistory(options) {
 }
 
 // src/react/hooks/index.ts
-import { useEffect as useEffect2, useMemo as useMemo2, useRef as useRef2 } from "react";
+import { useCallback as useCallback2, useEffect as useEffect2, useMemo as useMemo2, useRef as useRef2 } from "react";
 import { useQuery } from "@tanstack/react-query";
 import { useShallow } from "zustand/react/shallow";
 
@@ -938,14 +1055,14 @@ var useTrackStore = create()((set, get) => ({
   degradedReason: null,
   applyUpdate(track) {
     const next = new Map(get().tracks);
-    next.set(track.trackId, { ...next.get(track.trackId), ...track });
+    next.set(track.trackId, withLabel({ ...next.get(track.trackId), ...track }));
     set({ tracks: next, lastUpdateAt: Date.now() });
   },
   applyBatch(updates, removedIds) {
     if (updates.length === 0 && removedIds.length === 0) return;
     const next = new Map(get().tracks);
     for (const track of updates) {
-      next.set(track.trackId, { ...next.get(track.trackId), ...track });
+      next.set(track.trackId, withLabel({ ...next.get(track.trackId), ...track }));
     }
     for (const id of removedIds) next.delete(id);
     set({ tracks: next, lastUpdateAt: Date.now() });
@@ -958,7 +1075,7 @@ var useTrackStore = create()((set, get) => ({
   applySnapshot(tracks) {
     const next = /* @__PURE__ */ new Map();
     for (const t of tracks) {
-      if (isRenderableTrack2(t)) next.set(t.trackId, t);
+      if (isRenderableTrack2(t)) next.set(t.trackId, withLabel(t));
     }
     set({ tracks: next, lastUpdateAt: Date.now() });
   },
@@ -1057,6 +1174,7 @@ function useAeroTracks(opts = {}) {
       useTrackStore.getState().setStatus("idle");
     };
   }, [opts.bbox?.join(","), opts.source, opts.minConfidence, opts.maxUpdateHz, opts.enabled]);
+  const reconnect = useCallback2(() => client.tracks.reconnect(), [client]);
   return {
     tracks,
     allTracks,
@@ -1066,7 +1184,8 @@ function useAeroTracks(opts = {}) {
     isDegraded,
     lastUpdateAt,
     error,
-    trackCount: tracks.length
+    trackCount: tracks.length,
+    reconnect
   };
 }
 var useTracks = useAeroTracks;