Download Management

// From DownloadManager.swift:6-14
@MainActor
class DownloadManager: ObservableObject {
    @Published var activeDownloads: [Download] = []
    @Published var recentDownloads: [Download] = []
    @Published var isDownloadsPopoverOpen = false

    let modelContainer: ModelContainer
    let modelContext: ModelContext
    private var activeDownloadTasks: [UUID: WKDownload] = [:]

// From DownloadManager.swift:39-66
func startDownload(
    from downloadTask: WKDownload,
    originalURL: URL,
    suggestedFilename: String,
    expectedSize: Int64 = 0
) -> Download {
    let download = Download(
        originalURL: originalURL,
        fileName: suggestedFilename,
        fileSize: expectedSize
    )

    download.status = .downloading
    download.isActive = true

    // Save to SwiftData
    modelContext.insert(download)
    do {
        try modelContext.save()
    } catch {
        // Failed to save download
    }

    activeDownloadTasks[download.id] = downloadTask
    activeDownloads.append(download)
    refreshRecentDownloads()
    return download
}

// From Download.swift:4-10
enum DownloadStatus: String, Codable {
    case pending
    case downloading
    case completed
    case failed
    case cancelled
}

// From DownloadManager.swift:68-78
func updateDownloadProgress(_ download: Download, downloadedBytes: Int64, totalBytes: Int64) {
    download.updateProgress(downloadedBytes: downloadedBytes, totalBytes: totalBytes)

    try? modelContext.save()

    // Trigger UI updates
    DispatchQueue.main.async {
        self.objectWillChange.send()
        download.objectWillChange.send()
    }
}

// From Download.swift:54-65
func updateProgress(downloadedBytes: Int64, totalBytes: Int64) {
    self.downloadedBytes = downloadedBytes
    self.fileSize = totalBytes
    self.progress = totalBytes > 0 ? Double(downloadedBytes) / Double(totalBytes) : 0.0

    // Update published properties for UI
    DispatchQueue.main.async {
        self.displayDownloadedBytes = downloadedBytes
        self.displayFileSize = totalBytes > 0 ? totalBytes : self.fileSize
        self.displayProgress = self.progress
    }
}

// From Download.swift:13-26
@Model
class Download: ObservableObject, Identifiable {
    var id: UUID
    var originalURL: URL
    var originalURLString: String
    var fileName: String
    var fileSize: Int64
    var downloadedBytes: Int64
    var status: DownloadStatus
    var progress: Double
    var destinationURL: URL?
    var createdAt: Date
    var completedAt: Date?
    var error: String?

// From DownloadManager.swift:80-90
func completeDownload(_ download: Download, destinationURL: URL) {
    download.markCompleted(destinationURL: destinationURL)

    try? modelContext.save()

    activeDownloadTasks.removeValue(forKey: download.id)
    activeDownloads.removeAll { $0.id == download.id }
    refreshRecentDownloads()

    // Show notification or update UI
}

// From Download.swift:67-73
func markCompleted(destinationURL: URL) {
    self.status = .completed
    self.progress = 1.0
    self.destinationURL = destinationURL
    self.completedAt = Date()
    self.isActive = false
}

// From DownloadManager.swift:102-114
func cancelDownload(_ download: Download) {
    if let downloadTask = activeDownloadTasks[download.id] {
        downloadTask.cancel()
    }

    download.markCancelled()

    try? modelContext.save()

    activeDownloadTasks.removeValue(forKey: download.id)
    activeDownloads.removeAll { $0.id == download.id }
    refreshRecentDownloads()
}

// From DownloadManager.swift:92-100
func failDownload(_ download: Download, error: String) {
    download.markFailed(error: error)

    try? modelContext.save()

    activeDownloadTasks.removeValue(forKey: download.id)
    activeDownloads.removeAll { $0.id == download.id }
    refreshRecentDownloads()
}

// From Download.swift:75-79
func markFailed(error: String) {
    self.status = .failed
    self.error = error
    self.isActive = false
}

// From DownloadManager.swift:25-37
private func loadRecentDownloads() {
    let descriptor = FetchDescriptor<Download>(
        sortBy: [SortDescriptor(\.createdAt, order: .reverse)]
    )

    do {
        let downloads = try modelContext.fetch(descriptor)
        self.recentDownloads = Array(downloads.prefix(20)) // Show last 20 downloads
        self.activeDownloads = downloads.filter { $0.status == .downloading }
    } catch {
        // Failed to load downloads
    }
}

// From DownloadManager.swift:116-124
func clearCompletedDownloads() {
    let completedDownloads = recentDownloads.filter { $0.status == .completed }
    for download in completedDownloads {
        modelContext.delete(download)
    }

    try? modelContext.save()
    refreshRecentDownloads()
}

// From DownloadManager.swift:137-140
func openDownloadInFinder(_ download: Download) {
    guard let destinationURL = download.destinationURL else { return }
    NSWorkspace.shared.selectFile(destinationURL.path, inFileViewerRootedAtPath: "")
}

// From DownloadManager.swift:126-135
func deleteDownload(_ download: Download) {
    // If it's an active download, cancel it first
    if download.status == .downloading {
        cancelDownload(download)
    }

    modelContext.delete(download)
    try? modelContext.save()
    refreshRecentDownloads()
}

// From DownloadManager.swift:147-150
func getDownloadsDirectory() -> URL {
    return FileManager.default.urls(for: .downloadsDirectory, in: .userDomainMask)
        .first ?? URL(fileURLWithPath: NSHomeDirectory())
}

// From DownloadManager.swift:153-168
func createUniqueFilename(for url: URL) -> URL {
    var finalURL = url
    var counter = 1

    while FileManager.default.fileExists(atPath: finalURL.path) {
        let filename = url.deletingPathExtension().lastPathComponent
        let ext = url.pathExtension
        let directory = url.deletingLastPathComponent()

        let newFilename = ext.isEmpty ? "\(filename) (\(counter))" : "\(filename) (\(counter)).\(ext)"
        finalURL = directory.appendingPathComponent(newFilename)
        counter += 1
    }

    return finalURL
}

// From Download.swift:86-96
var formattedFileSize: String {
    return ByteCountFormatter.string(
        fromByteCount: displayFileSize > 0 ? displayFileSize : fileSize,
        countStyle: .file
    )
}

var formattedDownloadedSize: String {
    return ByteCountFormatter.string(fromByteCount: displayDownloadedBytes, countStyle: .file)
}