app-ios: remove port forwards when client disconnects

Author: aitjcize

app-ios/app-ios/ViewModels/ClientViewModel.swift

@@ -92,6 +92,14 @@ class ClientViewModel: ObservableObject {
     private func handleAgentLeft(_ client: Client) {
         clients.removeAll { $0.mid == client.mid }
         sortClients()
+
+        // Post a notification that a client has left
+        // This will be observed by PortForwardViewModel to clean up port forwards
+        NotificationCenter.default.post(
+            name: .clientDisconnected,
+            object: nil,
+            userInfo: ["clientId": client.mid]
+        )
     }
 
     func setFilterPattern(_ pattern: String) {
@@ -151,6 +159,7 @@ class ClientViewModel: ObservableObject {
 
 extension Notification.Name {
     static let logoutRequested = Notification.Name("logoutRequested")
+    static let clientDisconnected = Notification.Name("clientDisconnected")
 }
 
 // Extension to convert Publisher to async/await

app-ios/app-ios/ViewModels/PortForwardViewModel.swift

@@ -52,6 +52,30 @@ class PortForwardViewModel: ObservableObject {
 
     init(webSocketService: WebSocketService = WebSocketService()) {
         self.webSocketService = webSocketService
+
+        // Set up notification observer for client disconnection
+        NotificationCenter.default.addObserver(
+            self,
+            selector: #selector(handleClientDisconnected(_:)),
+            name: .clientDisconnected,
+            object: nil
+        )
+    }
+
+    deinit {
+        NotificationCenter.default.removeObserver(self)
+    }
+
+    @objc private func handleClientDisconnected(_ notification: Notification) {
+        // Extract the client ID from the notification
+        guard let clientId = notification.userInfo?["clientId"] as? String else {
+            return
+        }
+
+        // Close all port forwards for this client
+        DispatchQueue.main.async { [weak self] in
+            self?.closeAllPortForwardsForClient(clientId: clientId)
+        }
     }
 
     // Helper method for logging errors
@@ -142,6 +166,17 @@ class PortForwardViewModel: ObservableObject {
         objectWillChange.send()
     }
 
+    // Close all port forwards for a specific client
+    func closeAllPortForwardsForClient(clientId: String) {
+        // Get all port forward IDs for this client
+        let portForwardIds = portForwards.values.filter { $0.clientId == clientId }.map { $0.id }
+
+        // Close each port forward
+        for id in portForwardIds {
+            closePortForward(id: id)
+        }
+    }
+
     private func startLocalServer(for portForward: PortForward) {
         // Create a TCP listener on localhost with the specified port
         let port = NWEndpoint.Port(integerLiteral: UInt16(portForward.localPort))