Add progress delegate (#30)

Co-authored-by: Pouya Yarandi <[email protected]>
divar-ir · Aug 5, 2023 · 1e94006 · 1e94006
1 parent 1a1d46d
commit 1e94006
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 0 deletions.
diff --git a/Sources/NetShears/NetShears.swift b/Sources/NetShears/NetShears.swift
@@ -18,10 +18,16 @@ public extension BodyExporterDelegate {
     func netShears(exportRequestBodyFor request: NetShearsRequestModel) -> BodyExportType { .default }
 }
 
+public protocol TaskProgressDelegate: AnyObject {
+    func task(_ url: URL, didRecieveProgress progress: Progress)
+}
+
 public final class NetShears: NSObject {
 
     public static let shared = NetShears()
     public weak var bodyExportDelegate: BodyExporterDelegate?
+    public weak var taskProgressDelegate: TaskProgressDelegate?
+
     internal var loggerEnable = false
     internal var interceptorEnable = false
     internal var listenerEnable = false

diff --git a/Sources/NetShears/URLProtocol/NetworkInterceptorUrlProtocol.swift b/Sources/NetShears/URLProtocol/NetworkInterceptorUrlProtocol.swift
@@ -121,5 +121,11 @@ extension NetworkInterceptorUrlProtocol: URLSessionDataDelegate {
     func urlSessionDidFinishEvents(forBackgroundURLSession session: URLSession) {
         client?.urlProtocolDidFinishLoading(self)
     }
+
+    func urlSession(_ session: URLSession, task: URLSessionTask, didSendBodyData bytesSent: Int64, totalBytesSent: Int64, totalBytesExpectedToSend: Int64) {
+        if let url = task.currentRequest?.url {
+            NetShears.shared.taskProgressDelegate?.task(url, didRecieveProgress: task.progress)
+        }
+    }
 }
 
diff --git a/Sources/NetShears/URLProtocol/NetworkLoggerUrlProtocol.swift b/Sources/NetShears/URLProtocol/NetworkLoggerUrlProtocol.swift
@@ -137,6 +137,12 @@ extension NetworkLoggerUrlProtocol: URLSessionDataDelegate {
     func urlSessionDidFinishEvents(forBackgroundURLSession session: URLSession) {
         client?.urlProtocolDidFinishLoading(self)
     }
+
+    func urlSession(_ session: URLSession, task: URLSessionTask, didSendBodyData bytesSent: Int64, totalBytesSent: Int64, totalBytesExpectedToSend: Int64) {
+        if let url = task.currentRequest?.url {
+            NetShears.shared.taskProgressDelegate?.task(url, didRecieveProgress: task.progress)
+        }
+    }
 }
 
 
diff --git a/Sources/NetShears/URLProtocol/NetwrokListenerUrlProtocol.swift b/Sources/NetShears/URLProtocol/NetwrokListenerUrlProtocol.swift
@@ -137,6 +137,12 @@ extension NetwrokListenerUrlProtocol: URLSessionDataDelegate {
     func urlSessionDidFinishEvents(forBackgroundURLSession session: URLSession) {
         client?.urlProtocolDidFinishLoading(self)
     }
+
+    func urlSession(_ session: URLSession, task: URLSessionTask, didSendBodyData bytesSent: Int64, totalBytesSent: Int64, totalBytesExpectedToSend: Int64) {
+        if let url = task.currentRequest?.url {
+            NetShears.shared.taskProgressDelegate?.task(url, didRecieveProgress: task.progress)
+        }
+    }
 }