@mentra/bluetooth-sdk 3.2.0-dev.195 → 3.2.0-dev.203

package/ios/Source/MentraBluetoothSDK.swift

@@ -8,9 +8,9 @@ private final class ActiveStreamKeepAlive {
     var pendingAckId: String?
     var missedAckCount = 0
     var task: Task<Void, Never>?
-    // Missed-ACK counting only begins once the stream is confirmed live/coming up, so a slow
-    // startup (glasses can't ACK until they reach starting/streaming) can't trip a false
-    // keep-alive timeout before the stream is ever up.
+    /// Missed-ACK counting only begins once the stream is confirmed live/coming up, so a slow
+    /// startup (glasses can't ACK until they reach starting/streaming) can't trip a false
+    /// keep-alive timeout before the stream is ever up.
     var armed = false
 
     init(streamId: String, intervalSeconds: Int) {
@@ -47,8 +47,8 @@ private final class PendingWifiScan {
     let pending: PendingResponse<[WifiScanResult]>
     let scanId: String
     var latestResults: [WifiScanResult] = []
-    // Chunks accumulated from scanId-echoing glasses, deduplicated by SSID;
-    // resolved only when the glasses flag the scan complete.
+    /// Chunks accumulated from scanId-echoing glasses, deduplicated by SSID;
+    /// resolved only when the glasses flag the scan complete.
     var accumulated: [WifiScanResult] = []
 
     init(pending: PendingResponse<[WifiScanResult]>, scanId: String) {
@@ -57,21 +57,68 @@ private final class PendingWifiScan {
     }
 }
 
-private enum WifiStatusOperation {
-    case connect
-    case forget
-}
-
 @MainActor
 private final class PendingWifiStatusRequest {
-    let operation: WifiStatusOperation
     let ssid: String
     let pending: PendingResponse<WifiStatusEvent>
 
-    init(operation: WifiStatusOperation, ssid: String, pending: PendingResponse<WifiStatusEvent>) {
-        self.operation = operation
+    init(
+        ssid: String,
+        pending: PendingResponse<WifiStatusEvent>
+    ) {
+        self.ssid = ssid
+        self.pending = pending
+    }
+}
+
+@MainActor
+private final class PendingWifiForgetRequest {
+    let ssid: String
+    let requestId: String
+    var sid: String
+    let epoch: UInt64
+    let pending: PendingResponse<WifiForgetResult>
+    var mode: WifiRequestMode
+    var commandSent = false
+
+    init(
+        ssid: String,
+        requestId: String,
+        sid: String,
+        epoch: UInt64,
+        pending: PendingResponse<WifiForgetResult>,
+        mode: WifiRequestMode
+    ) {
         self.ssid = ssid
+        self.requestId = requestId
+        self.sid = sid
+        self.epoch = epoch
+        self.pending = pending
+        self.mode = mode
+    }
+}
+
+@MainActor
+private final class PendingSavedWifiNetworks {
+    let requestId: String
+    var sid: String
+    let epoch: UInt64
+    let pending: PendingResponse<SavedWifiNetworksResult>
+    var mode: WifiRequestMode
+    var commandSent = false
+
+    init(
+        requestId: String,
+        sid: String,
+        epoch: UInt64,
+        pending: PendingResponse<SavedWifiNetworksResult>,
+        mode: WifiRequestMode
+    ) {
+        self.requestId = requestId
+        self.sid = sid
+        self.epoch = epoch
         self.pending = pending
+        self.mode = mode
     }
 }
 
@@ -101,17 +148,28 @@ private final class PendingVideoRecordingRequest {
     }
 }
 
-// seq records send order (assigned and handed to the BLE queue in a single
-// MainActor turn) so that an id-carrying status for a newer start can
-// identify which older in-flight starts it preempted.
+@MainActor
+private final class PendingVersionInfoRequest {
+    let pending: PendingResponse<VersionInfoResult>
+    let accumulator: VersionInfoResponseAccumulator
+
+    init(pending: PendingResponse<VersionInfoResult>, requestId: String) {
+        self.pending = pending
+        accumulator = VersionInfoResponseAccumulator(expectedRequestId: requestId)
+    }
+}
+
+/// seq records send order (assigned and handed to the BLE queue in a single
+/// MainActor turn) so that an id-carrying status for a newer start can
+/// identify which older in-flight starts it preempted.
 @MainActor
 private final class PendingStreamStart {
     let seq: Int
     let pending: PendingResponse<StreamStatusEvent>
-    // Set when an id-carrying error arrives: a fatal publisher error never
-    // reaches the reconnect machinery and winds down with a streamId-less
-    // stopped, so the stash both attributes that stopped to this start and
-    // preserves the real error details for the rejection.
+    /// Set when an id-carrying error arrives: a fatal publisher error never
+    /// reaches the reconnect machinery and winds down with a streamId-less
+    /// stopped, so the stash both attributes that stopped to this start and
+    /// preserves the real error details for the rejection.
     var lastError: StreamStatusEvent?
 
     init(seq: Int, pending: PendingResponse<StreamStatusEvent>) {
@@ -121,7 +179,7 @@ private final class PendingStreamStart {
 }
 
 @MainActor
-private final class PendingResponse<T> {
+final class PendingResponse<T> {
     private let operation: String
     private var continuation: CheckedContinuation<T, Error>?
     private var timeoutTask: Task<Void, Never>?
@@ -150,22 +208,39 @@ private final class PendingResponse<T> {
         continuation = nil
     }
 
-    func wait(timeoutMs: Int = 15_000) async throws -> T {
+    func wait(timeoutMs: Int? = 15000) async throws -> T {
+        if Task.isCancelled {
+            throw BluetoothSdkError(code: "request_cancelled", message: "\(operation) was cancelled.")
+        }
         if let result {
             return try result.get()
         }
-        return try await withCheckedThrowingContinuation { continuation in
-            self.continuation = continuation
-            timeoutTask = Task { @MainActor [weak self] in
-                do {
-                    try await Task.sleep(nanoseconds: UInt64(timeoutMs) * 1_000_000)
-                } catch {
-                    return
+        return try await withTaskCancellationHandler {
+            try await withCheckedThrowingContinuation { continuation in
+                self.continuation = continuation
+                if let timeoutMs {
+                    timeoutTask = Task { @MainActor [weak self] in
+                        do {
+                            try await Task.sleep(nanoseconds: UInt64(timeoutMs) * 1_000_000)
+                        } catch {
+                            return
+                        }
+                        self?.reject(
+                            BluetoothSdkError(
+                                code: "request_timeout",
+                                message: "\(self?.operation ?? "Request") timed out waiting for glasses response."
+                            )
+                        )
+                    }
                 }
-                self?.reject(
+            }
+        } onCancel: {
+            Task { @MainActor [weak self] in
+                guard let self else { return }
+                reject(
                     BluetoothSdkError(
-                        code: "request_timeout",
-                        message: "\(self?.operation ?? "Request") timed out waiting for glasses response."
+                        code: "request_cancelled",
+                        message: "\(operation) was cancelled."
                     )
                 )
             }
@@ -175,12 +250,12 @@ private final class PendingResponse<T> {
 
 @MainActor
 public final class MentraBluetoothSDK {
-    private static let wifiScanTimeoutMs = 20_000
+    private static let wifiScanTimeoutMs = 20000
     // A photo response is terminal only after capture, encoding, transport, and upload.
     // Max-quality BLE fallback can legitimately exceed the generic command deadline.
-    private static let photoRequestTimeoutMs = 30_000
-    private static let otaBesVersionWaitMs = 5_000
-    private static let otaMtkVersionWaitMs = 2_000
+    private static let photoRequestTimeoutMs = 30000
+    private static let otaBesVersionWaitMs = 5000
+    private static let otaMtkVersionWaitMs = 2000
     private static let otaVersionPollMs = 100
     private static let defaultStreamKeepAliveIntervalSeconds = 5
 
@@ -216,10 +291,13 @@ public final class MentraBluetoothSDK {
     private var pendingOtaQuery: PendingResponse<OtaQueryResult>?
     private var pendingOtaStart: PendingResponse<OtaStartAckEvent>?
     private var pendingWifiScan: PendingWifiScan?
+    private var pendingSavedWifiNetworks: PendingSavedWifiNetworks?
     private var wifiScanTask: Task<[WifiScanResult], Error>?
     private var pendingWifiStatus: PendingWifiStatusRequest?
+    private var pendingWifiForget: PendingWifiForgetRequest?
     private var pendingHotspotStatus: PendingHotspotStatusRequest?
-    private var pendingVersionInfo: PendingResponse<VersionInfoResult>?
+    private var pendingVersionInfo: PendingVersionInfoRequest?
+    private let wifiSessionCapabilities = WifiSessionCapabilities()
     private var configuredOtaVersionUrl: String?
 
     public init(configuration: MentraBluetoothSDKConfiguration = .default) {
@@ -609,7 +687,8 @@ public final class MentraBluetoothSDK {
                     for key in ["button_photo_zsl_mfnr", "button_photo_mfnr", "button_photo_zsl", "button_photo_noise_reduction",
                                 "button_photo_edge_enhancement", "button_photo_isp_digital_gain",
                                 "button_photo_isp_analog_gain", "button_photo_ae_exposure_divisor",
-                                "button_photo_iso_cap", "button_photo_compress", "button_photo_sound"] {
+                                "button_photo_iso_cap", "button_photo_compress", "button_photo_sound"]
+                    {
                         DeviceStore.shared.remove(cat, key)
                     }
                 }
@@ -863,7 +942,8 @@ public final class MentraBluetoothSDK {
                 return try await pending.wait(timeoutMs: MentraBluetoothSDK.wifiScanTimeoutMs)
             } catch {
                 if (error as? BluetoothSdkError)?.code == "request_timeout",
-                   !request.latestResults.isEmpty {
+                   !request.latestResults.isEmpty
+                {
                     return request.latestResults
                 }
                 throw error
@@ -874,14 +954,14 @@ public final class MentraBluetoothSDK {
     }
 
     public func sendWifiCredentials(ssid: String, password: String) async throws -> WifiStatusEvent {
-        guard pendingWifiStatus == nil else {
+        guard pendingWifiStatus == nil, pendingWifiForget == nil else {
             throw BluetoothSdkError(
                 code: "request_in_flight",
                 message: "A WiFi status command is already waiting for a glasses response."
             )
         }
         let pending = PendingResponse<WifiStatusEvent>(operation: "WiFi connect request")
-        pendingWifiStatus = PendingWifiStatusRequest(operation: .connect, ssid: ssid, pending: pending)
+        pendingWifiStatus = PendingWifiStatusRequest(ssid: ssid, pending: pending)
         DeviceManager.shared.sendWifiCredentials(ssid, password)
         do {
             let event = try await pending.wait()
@@ -897,25 +977,83 @@ public final class MentraBluetoothSDK {
         }
     }
 
-    public func forgetWifiNetwork(ssid: String) async throws -> WifiStatusEvent {
-        guard pendingWifiStatus == nil else {
+    public func forgetWifiNetwork(ssid: String) async throws -> WifiForgetResult {
+        guard wifiSsidIsValid(ssid) else {
+            throw BluetoothSdkError(code: "invalid_ssid", message: "WiFi SSID cannot be empty.")
+        }
+        guard pendingWifiStatus == nil, pendingWifiForget == nil else {
             throw BluetoothSdkError(
                 code: "request_in_flight",
                 message: "A WiFi status command is already waiting for a glasses response."
             )
         }
-        let pending = PendingResponse<WifiStatusEvent>(operation: "WiFi forget request")
-        pendingWifiStatus = PendingWifiStatusRequest(operation: .forget, ssid: ssid, pending: pending)
-        DeviceManager.shared.forgetWifiNetwork(ssid)
+        let pending = PendingResponse<WifiForgetResult>(operation: "WiFi forget request")
+        let requestId = "forget-\(UUID().uuidString)"
+        let request = PendingWifiForgetRequest(
+            ssid: ssid,
+            requestId: requestId,
+            sid: wifiSessionCapabilities.sessionId,
+            epoch: wifiSessionCapabilities.epoch,
+            pending: pending,
+            mode: wifiSessionCapabilities.forgetMode()
+        )
+        pendingWifiForget = request
+        dispatchWifiForgetIfReady(request)
         do {
             let event = try await pending.wait()
-            if pendingWifiStatus?.pending === pending {
-                pendingWifiStatus = nil
+            if pendingWifiForget === request {
+                pendingWifiForget = nil
             }
             return event
         } catch {
-            if pendingWifiStatus?.pending === pending {
-                pendingWifiStatus = nil
+            if pendingWifiForget === request {
+                pendingWifiForget = nil
+            }
+            if let sdkError = error as? BluetoothSdkError, sdkError.code == "request_timeout", request.mode == .discovering {
+                throw BluetoothSdkError(code: wifiCapabilityNegotiationTimeoutCode, message: "WiFi capability negotiation timed out.")
+            }
+            throw error
+        }
+    }
+
+    public func getSavedWifiNetworks() async throws -> SavedWifiNetworksResult {
+        guard pendingSavedWifiNetworks == nil else {
+            throw BluetoothSdkError(
+                code: "request_in_flight",
+                message: "A saved WiFi networks request is already waiting for a glasses response."
+            )
+        }
+        let requestId = "saved-\(UUID().uuidString)"
+        let mode = wifiSessionCapabilities.savedNetworksMode()
+        if mode == .legacy || mode == .unsupported {
+            return SavedWifiNetworksResult(
+                outcome: .unsupported,
+                networks: [],
+                error: "saved_wifi_networks_unsupported"
+            )
+        }
+        let pending = PendingResponse<SavedWifiNetworksResult>(operation: "Saved WiFi networks request")
+        let request = PendingSavedWifiNetworks(
+            requestId: requestId,
+            sid: wifiSessionCapabilities.sessionId,
+            epoch: wifiSessionCapabilities.epoch,
+            pending: pending,
+            mode: mode
+        )
+        pendingSavedWifiNetworks = request
+        dispatchSavedWifiNetworksIfReady(request)
+        do {
+            let networks = try await pending.wait()
+            if pendingSavedWifiNetworks === request {
+                pendingSavedWifiNetworks = nil
+            }
+            return networks
+        } catch {
+            if pendingSavedWifiNetworks === request {
+                pendingSavedWifiNetworks = nil
+            }
+            if let sdkError = error as? BluetoothSdkError, sdkError.code == "request_timeout", request.mode == .discovering {
+                throw BluetoothSdkError(code: wifiCapabilityNegotiationTimeoutCode, message: "WiFi capability negotiation timed out.")
             }
             throw error
         }
@@ -1064,7 +1202,7 @@ public final class MentraBluetoothSDK {
         stopStreamKeepAliveMonitor()
         DeviceManager.shared.startStream(values)
         do {
-            let event = try await pending.wait(timeoutMs: 30_000)
+            let event = try await pending.wait(timeoutMs: 30000)
             pendingStreamStarts.removeValue(forKey: streamId)
             if startSdkKeepAlive {
                 startStreamKeepAliveMonitor(
@@ -1123,7 +1261,7 @@ public final class MentraBluetoothSDK {
         stopStreamKeepAliveMonitor()
         DeviceManager.shared.stopStream()
         do {
-            let event = try await pending.wait(timeoutMs: 15_000)
+            let event = try await pending.wait(timeoutMs: 15000)
             if pendingStreamStop?.pending === pending {
                 pendingStreamStop = nil
             }
@@ -1195,7 +1333,7 @@ public final class MentraBluetoothSDK {
         )
         DeviceManager.shared.stopVideoRecording(requestId, webhookUrl, authToken)
         do {
-            let timeoutMs = waitForUpload ? videoUploadStopTimeoutMs : 15_000
+            let timeoutMs = waitForUpload ? videoUploadStopTimeoutMs : 15000
             let event = try await pending.wait(timeoutMs: timeoutMs)
             pendingVideoRecordingRequests.removeValue(forKey: requestId)
             return event
@@ -1242,17 +1380,19 @@ public final class MentraBluetoothSDK {
                 message: "A version info request is already waiting for a glasses response."
             )
         }
+        let requestId = UUID().uuidString
         let pending = PendingResponse<VersionInfoResult>(operation: "version info request")
-        pendingVersionInfo = pending
-        DeviceManager.shared.requestVersionInfo()
+        let request = PendingVersionInfoRequest(pending: pending, requestId: requestId)
+        pendingVersionInfo = request
+        DeviceManager.shared.requestVersionInfo(requestId: requestId)
         do {
             let status = try await pending.wait()
-            if pendingVersionInfo === pending {
+            if pendingVersionInfo === request {
                 pendingVersionInfo = nil
             }
             return status
         } catch {
-            if pendingVersionInfo === pending {
+            if pendingVersionInfo === request {
                 pendingVersionInfo = nil
             }
             throw error
@@ -1384,7 +1524,9 @@ public final class MentraBluetoothSDK {
         try await startOtaCommand(otaVersionUrl: otaVersionUrl)
     }
 
-    func sendOtaQueryStatus() async throws -> OtaQueryResult { try await queryOtaStatus() }
+    func sendOtaQueryStatus() async throws -> OtaQueryResult {
+        try await queryOtaStatus()
+    }
 
     func startAr99OtaFromFile(_ path: String) throws -> Bool {
         try requireGlassesConnected(operation: "start AR99 OTA")
@@ -1448,7 +1590,7 @@ public final class MentraBluetoothSDK {
         timeoutMs: Int,
         isReady: (GlassesStatus) -> Bool
     ) async -> GlassesStatus {
-        let deadline = Date().addingTimeInterval(Double(timeoutMs) / 1_000)
+        let deadline = Date().addingTimeInterval(Double(timeoutMs) / 1000)
         var status = initialStatus
         while Date() < deadline {
             status = glassesStatus
@@ -1456,7 +1598,7 @@ public final class MentraBluetoothSDK {
                 return status
             }
 
-            let remainingMs = max(0, Int(deadline.timeIntervalSinceNow * 1_000))
+            let remainingMs = max(0, Int(deadline.timeIntervalSinceNow * 1000))
             let sleepMs = min(Self.otaVersionPollMs, remainingMs)
             if sleepMs <= 0 {
                 break
@@ -1498,6 +1640,7 @@ public final class MentraBluetoothSDK {
 
     public func invalidate() {
         stopStreamKeepAliveMonitor()
+        resetWifiProtocolSession(sessionId: "", code: "sdk_closed")
         if let bluetoothAvailabilityListenerId {
             BluetoothAvailability.shared.removeStateListener(bluetoothAvailabilityListenerId)
             self.bluetoothAvailabilityListenerId = nil
@@ -1743,13 +1886,13 @@ public final class MentraBluetoothSDK {
         }
     }
 
-    // The glasses process BLE commands FIFO and every start_stream begins by
-    // stopping whatever runs, so an id-carrying status proving start X's
-    // start path ran on the glasses also proves every lower-seq start has
-    // already been preempted. Their only verdict on current firmware is a
-    // streamId-less stopped, which the id-less heuristic deliberately
-    // ignores — without this they would run out the 30s timeout instead of
-    // failing fast.
+    /// The glasses process BLE commands FIFO and every start_stream begins by
+    /// stopping whatever runs, so an id-carrying status proving start X's
+    /// start path ran on the glasses also proves every lower-seq start has
+    /// already been preempted. Their only verdict on current firmware is a
+    /// streamId-less stopped, which the id-less heuristic deliberately
+    /// ignores — without this they would run out the 30s timeout instead of
+    /// failing fast.
     private func rejectPreemptedStreamStarts(winnerSeq: Int) {
         for (streamId, start) in pendingStreamStarts where start.seq < winnerSeq {
             pendingStreamStarts.removeValue(forKey: streamId)
@@ -1763,11 +1906,11 @@ public final class MentraBluetoothSDK {
         }
     }
 
-    // The glasses process BLE commands FIFO, so this stop's ack also settles
-    // every start sent before it: whatever those starts brought up (or would
-    // have brought up) has been stopped, and no further status will arrive
-    // for them. Without this they would run out the 30s start timeout.
-    // Starts sent after the stop keep waiting for their own statuses.
+    /// The glasses process BLE commands FIFO, so this stop's ack also settles
+    /// every start sent before it: whatever those starts brought up (or would
+    /// have brought up) has been stopped, and no further status will arrive
+    /// for them. Without this they would run out the 30s start timeout.
+    /// Starts sent after the stop keep waiting for their own statuses.
     private func rejectStreamStartsSuperseded(byStopSeq stopSeq: Int) {
         for (streamId, start) in pendingStreamStarts where start.seq < stopSeq {
             pendingStreamStarts.removeValue(forKey: streamId)
@@ -1969,9 +2112,9 @@ public final class MentraBluetoothSDK {
         request.pending.resolve(results)
     }
 
-    // scanId-echoing glasses: results for another scan are ignored instead of
-    // resolving the pending request, and matching chunks accumulate until the
-    // glasses flag the scan complete.
+    /// scanId-echoing glasses: results for another scan are ignored instead of
+    /// resolving the pending request, and matching chunks accumulate until the
+    /// glasses flag the scan complete.
     private func handleCorrelatedWifiScanChunk(
         scanId: String,
         results: [WifiScanResult],
@@ -1997,49 +2140,167 @@ public final class MentraBluetoothSDK {
     }
 
     private func handleWifiStatusForRequests(_ event: WifiStatusEvent) {
-        guard let request = pendingWifiStatus else { return }
+        let connectRequest = pendingWifiStatus
         // A wifi_status carrying the explicit error field is the glasses' failure
         // verdict for the in-flight connect: reject now instead of running out the
         // request timeout. Only the error field counts as failure — the glasses'
         // connect sequence emits a debounced bare connected=false ~1-2s after
         // credentials while association is still in progress, and rejecting on that
         // would kill every connect attempt early.
-        if request.operation == .connect, let error = event.error {
-            if pendingWifiStatus === request {
+        if let connectRequest, let error = event.error {
+            if pendingWifiStatus === connectRequest {
                 pendingWifiStatus = nil
             }
-            request.pending.reject(
+            connectRequest.pending.reject(
                 BluetoothSdkError(
                     code: error,
-                    message: "Glasses failed to join \"\(request.ssid)\": \(error)"
+                    message: "Glasses failed to join \"\(connectRequest.ssid)\": \(error)"
                 )
             )
             return
         }
-        guard wifiStatusMatches(event.status, request: request) else { return }
-        if pendingWifiStatus === request {
-            pendingWifiStatus = nil
+        if let connectRequest, wifiStatusMatchesConnect(event.status, ssid: connectRequest.ssid) {
+            if pendingWifiStatus === connectRequest {
+                pendingWifiStatus = nil
+            }
+            connectRequest.pending.resolve(event)
+            return
+        }
+
+        // Forget never settles from link state: modern requests require a correlated result,
+        // while legacy requests resolve as unverified at accepted dispatch.
+    }
+
+    private func handleWifiForgetResultForRequests(_ data: [String: Any]) {
+        guard let request = pendingWifiForget,
+              request.mode == .modern,
+              request.epoch == wifiSessionCapabilities.epoch,
+              case let .supported(version) = wifiSessionCapabilities.forgetResult,
+              let result = parseWifiForgetResult(
+                  expectedRequestId: request.requestId,
+                  expectedSid: request.sid,
+                  expectedSsid: request.ssid,
+                  capabilityVersion: version,
+                  data: data
+              )
+        else { return }
+        if pendingWifiForget === request, request.epoch == wifiSessionCapabilities.epoch {
+            pendingWifiForget = nil
+        }
+        request.pending.resolve(result)
+    }
+
+    private func handleSavedWifiNetworksForRequests(_ data: [String: Any]) {
+        guard let request = pendingSavedWifiNetworks,
+              request.mode == .modern,
+              request.epoch == wifiSessionCapabilities.epoch,
+              case let .supported(version) = wifiSessionCapabilities.savedNetworks,
+              let result = parseSavedWifiNetworks(
+                  expectedRequestId: request.requestId,
+                  expectedSid: request.sid,
+                  capabilityVersion: version,
+                  data: data
+              )
+        else { return }
+        if pendingSavedWifiNetworks === request, request.epoch == wifiSessionCapabilities.epoch {
+            pendingSavedWifiNetworks = nil
+        }
+        request.pending.resolve(result)
+    }
+
+    private func wifiStatusMatchesConnect(_ status: WifiStatus, ssid: String) -> Bool {
+        if case let .connected(currentSsid, _) = status {
+            return currentSsid == ssid
+        }
+        return false
+    }
+
+    private func dispatchWifiForgetIfReady(_ request: PendingWifiForgetRequest) {
+        guard pendingWifiForget === request,
+              request.epoch == wifiSessionCapabilities.epoch,
+              request.mode != .discovering,
+              !request.commandSent
+        else { return }
+        if request.mode == .unsupported {
+            request.pending.reject(BluetoothSdkError(code: "wifi_protocol_unsupported", message: "Unsupported WiFi forget protocol version."))
+            return
+        }
+        request.sid = wifiSessionCapabilities.sessionId
+        request.commandSent = DeviceManager.shared.forgetWifiNetwork(
+            request.ssid,
+            requestId: request.mode == .modern ? request.requestId : nil,
+            sid: request.mode == .modern ? request.sid : nil
+        )
+        if !request.commandSent {
+            request.pending.reject(BluetoothSdkError(code: "dispatch_failed", message: "No active glasses transport accepted WiFi forget."))
+        } else if request.mode == .legacy {
+            pendingWifiForget = nil
+            request.pending.resolve(legacyWifiForgetResult(ssid: request.ssid))
         }
-        request.pending.resolve(event)
     }
 
-    private func wifiStatusMatches(_ status: WifiStatus, request: PendingWifiStatusRequest) -> Bool {
-        switch request.operation {
-        case .connect:
-            if case let .connected(ssid, _) = status {
-                return ssid == request.ssid
-            }
-            return false
-        case .forget:
-            switch status {
-            case .disconnected:
-                return true
-            case let .connected(ssid, _):
-                return ssid != request.ssid
+    private func dispatchSavedWifiNetworksIfReady(_ request: PendingSavedWifiNetworks) {
+        guard pendingSavedWifiNetworks === request,
+              request.epoch == wifiSessionCapabilities.epoch,
+              request.mode == .modern,
+              !request.commandSent
+        else { return }
+        request.sid = wifiSessionCapabilities.sessionId
+        request.commandSent = DeviceManager.shared.requestSavedWifiNetworks(requestId: request.requestId, sid: request.sid)
+        if !request.commandSent {
+            request.pending.reject(BluetoothSdkError(code: "dispatch_failed", message: "No active glasses transport accepted saved WiFi request."))
+        }
+    }
+
+    private func applyWifiProtocolCapabilities(_ data: [String: Any]) {
+        wifiSessionCapabilities.applyVersionInfo1(data)
+        if let request = pendingWifiForget,
+           request.epoch == wifiSessionCapabilities.epoch,
+           request.mode == .discovering
+        {
+            request.mode = wifiSessionCapabilities.forgetMode()
+            dispatchWifiForgetIfReady(request)
+        }
+        if let request = pendingSavedWifiNetworks,
+           request.epoch == wifiSessionCapabilities.epoch,
+           request.mode == .discovering
+        {
+            request.mode = wifiSessionCapabilities.savedNetworksMode()
+            if request.mode == .legacy || request.mode == .unsupported {
+                pendingSavedWifiNetworks = nil
+                request.pending.resolve(
+                    SavedWifiNetworksResult(
+                        outcome: .unsupported,
+                        networks: [],
+                        error: "saved_wifi_networks_unsupported"
+                    )
+                )
+            } else {
+                dispatchSavedWifiNetworksIfReady(request)
             }
         }
     }
 
+    private func resetWifiProtocolSession(sessionId: String, code: String) {
+        let error = BluetoothSdkError(code: code, message: "The glasses WiFi protocol session changed.")
+        let wifiStatus = pendingWifiStatus?.pending
+        let wifiForget = pendingWifiForget?.pending
+        let savedNetworks = pendingSavedWifiNetworks?.pending
+        let wifiScan = pendingWifiScan?.pending
+        let hotspot = pendingHotspotStatus?.pending
+        wifiSessionCapabilities.reset(sessionId: sessionId)
+        pendingWifiStatus = nil
+        pendingWifiForget = nil
+        pendingSavedWifiNetworks = nil
+        pendingWifiScan = nil
+        pendingHotspotStatus = nil
+        wifiStatus?.reject(error)
+        wifiForget?.reject(error)
+        savedNetworks?.reject(error)
+        wifiScan?.reject(error)
+        hotspot?.reject(error)
+    }
+
     private func handleHotspotStatusForRequests(_ event: HotspotStatusEvent) {
         guard let request = pendingHotspotStatus else { return }
         guard hotspotStatusMatches(event.status, enabled: request.enabled) else { return }
@@ -2072,6 +2333,9 @@ public final class MentraBluetoothSDK {
     private func dispatchStoreUpdate(_ category: String, _ changes: [String: Any]) {
         switch ObservableStore.normalizeCategory(category) {
         case "glasses":
+            if changes["connected"] as? Bool == false {
+                resetWifiProtocolSession(sessionId: "", code: "wifi_session_disconnected")
+            }
             analytics.observeGlassesStatus(glassesStatus)
             let nextState = state
             delegate?.mentraBluetoothSDK(self, didUpdate: nextState)
@@ -2116,7 +2380,8 @@ public final class MentraBluetoothSDK {
             projectName: projectName
         )
     }
-private func dispatchDiscoveredDevices(_ rawSearchResults: Any?) {
+
+    private func dispatchDiscoveredDevices(_ rawSearchResults: Any?) {
         guard let results = rawSearchResults as? [[String: Any]] else { return }
         for result in results {
             guard let name = result["name"] as? String else { continue }
@@ -2175,6 +2440,19 @@ private func dispatchDiscoveredDevices(_ rawSearchResults: Any?) {
         activeSession.onComplete(activeSession.latestResults)
     }
 
+    private func handleVersionInfoForRequest(_ data: [String: Any]) {
+        guard let request = pendingVersionInfo else { return }
+        switch request.accumulator.accept(data) {
+        case .ignored:
+            break
+        case .waiting:
+            break
+        case let .complete(result):
+            pendingVersionInfo = nil
+            request.pending.resolve(result)
+        }
+    }
+
     private func dispatchBridgeEvent(_ eventName: String, _ data: [String: Any]) {
         switch eventName {
         case "log":
@@ -2225,6 +2503,23 @@ private func dispatchDiscoveredDevices(_ rawSearchResults: Any?) {
             let event = WifiStatusEvent(values: data)
             handleWifiStatusForRequests(event)
             delegate?.mentraBluetoothSDK(self, didReceive: .wifiStatus(event))
+        case "wifi_forget_result":
+            handleWifiForgetResultForRequests(data)
+            delegate?.mentraBluetoothSDK(self, didReceive: .raw(name: eventName, values: data))
+        case "saved_wifi_networks":
+            handleSavedWifiNetworksForRequests(data)
+            delegate?.mentraBluetoothSDK(self, didReceive: .raw(name: eventName, values: data))
+        case "wifi_protocol_session_ready":
+            resetWifiProtocolSession(
+                sessionId: data["sid"] as? String ?? "",
+                code: "wifi_session_restarted"
+            )
+        case "glasses_session_changed":
+            resetWifiProtocolSession(
+                sessionId: data["sid"] as? String ?? "",
+                code: "wifi_session_changed"
+            )
+            delegate?.mentraBluetoothSDK(self, didReceive: .raw(name: eventName, values: data))
         case "wifi_scan_result":
             let networks = (data["networks"] as? [[String: Any]])?.map(WifiScanResult.init(values:)) ?? []
             let hasCompletionFlag = data.keys.contains("scanComplete") || data.keys.contains("scan_complete")
@@ -2306,8 +2601,11 @@ private func dispatchDiscoveredDevices(_ rawSearchResults: Any?) {
             handleSettingsAckForRequests(event)
             delegate?.mentraBluetoothSDK(self, didReceive: .settingsAck(event))
         case "version_info":
+            if let infoType = data["versionInfoType"] as? String, ["version_info_1", "version_info"].contains(infoType) {
+                applyWifiProtocolCapabilities(data)
+            }
             let event = VersionInfoResult(values: data)
-            pendingVersionInfo?.resolve(event)
+            handleVersionInfoForRequest(data)
             delegate?.mentraBluetoothSDK(self, didReceive: .versionInfo(event))
         case "compatible_glasses_search_stop":
             delegate?.mentraBluetoothSDK(self, didStopScan: .completed)