Track data usage in DownloadManager

Records bytes transferred during episode downloads, distinguishing
between user-initiated and automatic downloads.

Author: bjtitus

podcasts/DownloadManager+URLSessionDelegate.swift

@@ -210,6 +210,39 @@ extension DownloadManager: URLSessionDelegate, URLSessionDownloadDelegate {
             return
         }
 
+        // Track cellular data usage
+        let bytesReceived = task.countOfBytesReceived
+        let isCellular = metrics.transactionMetrics.last?.isCellular ?? false
+
+        if bytesReceived > 0 {
+            let autoDownloadStatus = AutoDownloadStatus(rawValue: episode.autoDownloadStatus)
+            let operationType: CellularDataUsageManager.OperationType
+            switch autoDownloadStatus {
+            case .playerDownloadedForStreaming:
+                operationType = .stream
+            default:
+                operationType = .download
+            }
+
+            let sessionType: CellularDataUsageManager.SessionType =
+                session === wifiOnlyBackgroundSession ? .background :
+                (session === cellularBackgroundSession ? .background : .foreground)
+
+            let bytesDownloaded = operationType != .stream ? bytesReceived : 0
+            let bytesStreamed = operationType == .stream ? bytesReceived : 0
+            let connectionType: CellularDataUsageManager.ConnectionType = isCellular ? .cellular : .wifi
+
+            dataManager.cellularDataUsageManager.add(
+                episodeUuid: episode.uuid,
+                podcastUuid: episode.parentIdentifier(),
+                bytesDownloaded: bytesDownloaded,
+                bytesStreamed: bytesStreamed,
+                operationType: operationType,
+                connectionType: connectionType,
+                sessionType: sessionType
+            )
+        }
+
         if let failure = taskFailure[episode.uuid] {
             logDownload(episode, failure: failure, metrics: metrics, session: session)
             taskFailure.removeValue(forKey: episode.uuid)

podcasts/DownloadManager.swift

@@ -399,7 +399,7 @@ class DownloadManager: NSObject, FilePathProtocol {
         let exportPath = outputURL.pathComponents.joined(separator: "/")
         let exportStatus =  ExportStatus()
         let originalSizeInBytes = episode.sizeInBytes
-        let customLoaderDelegate = MediaExporterResourceLoaderDelegate(saveFilePath: exportPath) { [weak self, exportStatus] status, contentType, bytesDownloaded, bytesExpected in
+        let customLoaderDelegate = MediaExporterResourceLoaderDelegate(saveFilePath: exportPath, episodeUuid: episode.uuid, podcastUuid: episode.parentIdentifier()) { [weak self, exportStatus] status, contentType, bytesDownloaded, bytesExpected in
             guard let self else {
                 return
             }