1 files changed, 57 insertions, 1 deletions
diff --git a/Hotline/State/AppState.swift b/Hotline/State/AppState.swift
index 3917ad0..558af2a 100644
--- a/Hotline/State/AppState.swift
+++ b/Hotline/State/AppState.swift
@@ -12,8 +12,64 @@ final class AppState {
 
   }
 
-  var activeHotline: Hotline? = nil
+  var activeHotline: HotlineState? = nil
   var activeServerState: ServerState? = nil
 
   var cloudKitReady: Bool = false
+
+  // MARK: - Transfers
+
+  /// All active transfers across all servers
+  /// Transfers persist even if you disconnect from the server
+  var transfers: [TransferInfo] = []
+
+  /// Track download tasks by reference number for cancellation
+  @ObservationIgnored private var transferTasks: [UUID: Task<Void, Never>] = [:]
+  
+  /// Add a transfer to the transfer list
+  @MainActor
+  func addTransfer(_ transfer: TransferInfo) {
+    self.transfers.append(transfer)
+  }
+
+  /// Cancel a transfer by transfer ID
+  @MainActor
+  func cancelTransfer(id: UUID) {
+    guard let transferIndex = self.transfers.firstIndex(where: { $0.id == id }) else {
+      return
+    }
+    
+    // Cancel the task if it exists
+    if let task = self.transferTasks[id] {
+      task.cancel()
+      self.transferTasks.removeValue(forKey: id)
+    }
+
+    // Remove from transfers list
+    self.transfers.remove(at: transferIndex)
+  }
+
+  /// Cancel all active transfers
+  @MainActor
+  func cancelAllTransfers() {
+    for (_, task) in self.transferTasks {
+      task.cancel()
+    }
+    self.transferTasks.removeAll()
+
+    // Clear transfers
+    self.transfers.removeAll()
+  }
+
+  /// Register a transfer task
+  @MainActor
+  func registerTransferTask(_ task: Task<Void, Never>, transferID: UUID) {
+    self.transferTasks[transferID] = task
+  }
+  
+  /// Unregister a download task (called on completion/failure)
+  @MainActor
+  func unregisterTransferTask(for transferID: UUID) {
+    self.transferTasks.removeValue(forKey: transferID)
+  }
 }