@mentra/cloud-client 3.2.1-beta.502 → 3.2.1-beta.508

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mentra/cloud-client",
-  "version": "3.2.1-beta.502",
+  "version": "3.2.1-beta.508",
   "type": "module",
   "main": "./src/index.ts",
   "exports": {
@@ -12,7 +12,7 @@
     "test": "bun test"
   },
   "dependencies": {
-    "@mentra/cloud-protocol": "3.2.1-beta.502",
+    "@mentra/cloud-protocol": "3.2.1-beta.508",
     "tweetnacl": "^1.0.3"
   },
   "devDependencies": {

package/src/modules/runtime/audio-udp.ts

@@ -78,6 +78,13 @@ export class UdpAudio {
     return this.session?.sessionTag ?? null;
   }
 
+  /** Refresh a dead UDP route without changing the session key, tag or sequence. */
+  resetSocket(): void {
+    if (!this.session) return;
+    this.session.socket.close();
+    this.session.socket = this.udpFactory();
+  }
+
   /**
    * Open the UDP socket and load the session's routing + encryption material
    * from `connection.ack.audio`.

package/src/modules/runtime/runtime.ts

@@ -162,6 +162,7 @@ export class Runtime implements RuntimeModule {
   private udpProbeTimer: unknown | null = null;
   private udpProbeStartedAt = 0;
   private lastUdpAckAt = 0;
+  private readonly pendingUdpProbes = new Map<string, number>();
 
   /**
    * Whether the inbound-message routing has been wired to the connection yet.
@@ -393,7 +394,9 @@ export class Runtime implements RuntimeModule {
   private configureAudio(ack: ConnectionAck): void {
     if (ack.audio) {
       this.audio.configure(ack.audio);
-      this.updateStatus({ audioTransport: "udp" });
+      // Configured credentials do not prove DNS/connect or UDP delivery works.
+      // Use the already-open WS while probes establish the new UDP route.
+      this.updateStatus({ audioTransport: this.connection.isOpen ? "ws" : "none" });
       this.startUdpLiveness();
       return;
     }
@@ -536,8 +539,8 @@ export class Runtime implements RuntimeModule {
     this.lastUdpAckAt = 0;
     this.sendUdpProbe();
     this.udpProbeTimer = this.timers.setInterval(() => {
-      this.sendUdpProbe();
       this.checkUdpLiveness();
+      this.sendUdpProbe();
     }, UDP_PROBE_INTERVAL_MS);
   }
 
@@ -548,12 +551,18 @@ export class Runtime implements RuntimeModule {
     }
     this.udpProbeStartedAt = 0;
     this.lastUdpAckAt = 0;
+    this.pendingUdpProbes.clear();
   }
 
   private sendUdpProbe(): void {
     const tag = this.audio.sessionTag;
     if (tag === null) return;
-    const probeId = `${tag}-${Date.now()}-${Math.floor(Math.random() * 1_000_000)}`;
+    const now = Date.now();
+    for (const [id, sentAt] of this.pendingUdpProbes) {
+      if (now - sentAt >= UDP_LIVENESS_TIMEOUT_MS) this.pendingUdpProbes.delete(id);
+    }
+    const probeId = `${tag}-${now}-${Math.floor(Math.random() * 1_000_000)}`;
+    this.pendingUdpProbes.set(probeId, now);
     this.audio.sendProbe(probeId);
   }
 
@@ -562,6 +571,14 @@ export class Runtime implements RuntimeModule {
     const since = this.lastUdpAckAt || this.udpProbeStartedAt;
     if (since === 0 || Date.now() - since < UDP_LIVENESS_TIMEOUT_MS) return;
     this.updateStatus({ audioTransport: this.connection.isOpen ? "ws" : "none" });
+    // Connected UDP sockets retain their resolved peer and route. Recreate one
+    // after a missed liveness window so DNS/network changes can recover even
+    // when the WebSocket stayed open. Only this replacement's probes may restore
+    // UDP: an old socket's delayed ack does not establish the new route works.
+    this.pendingUdpProbes.clear();
+    this.audio.resetSocket();
+    this.lastUdpAckAt = 0;
+    this.udpProbeStartedAt = Date.now();
   }
 
   private handleUdpLivenessAck(payload: {
@@ -571,6 +588,9 @@ export class Runtime implements RuntimeModule {
     receivedAt: number;
   }): void {
     if (payload.sessionTag !== this.audio.sessionTag) return;
+    const sentAt = this.pendingUdpProbes.get(payload.probeId);
+    this.pendingUdpProbes.delete(payload.probeId);
+    if (sentAt === undefined || Date.now() - sentAt >= UDP_LIVENESS_TIMEOUT_MS) return;
     this.lastUdpAckAt = Date.now();
     if (this.status.status === "connected") {
       this.updateStatus({ audioTransport: "udp" });