Set custom HTTP request header for reporting (#67)

Author: andrewjl-mux

Sources/MuxUploadSDK/Extensions/NSMutableURLRequest+Reporting.swift

@@ -7,7 +7,8 @@ import Foundation
 extension NSMutableURLRequest {
     static func makeJSONPost(
         url: URL,
-        httpBody: Data
+        httpBody: Data,
+        additionalHTTPHeaders: [String: String]
     ) -> NSMutableURLRequest {
         let request = NSMutableURLRequest(
             url: url,
@@ -18,6 +19,10 @@ extension NSMutableURLRequest {
         request.httpMethod = "POST"
         request.setValue("application/json", forHTTPHeaderField: "Accept")
         request.setValue("application/json", forHTTPHeaderField: "Content-Type")
+        for keypair in additionalHTTPHeaders {
+            request.setValue(keypair.value, forHTTPHeaderField: keypair.key)
+        }
+
         request.httpBody = httpBody
 
         return request

Sources/MuxUploadSDK/InternalUtilities/Reporting/Reporter.swift

@@ -20,6 +20,9 @@ class Reporter: NSObject {
 
     var sessionID: String = UUID().uuidString
     var url: URL
+    var additionalHTTPHeaders: [String: String] {
+        ["x-litix-sdk": "swift-upload-sdk"]
+    }
 
     // TODO: Set these using dependency Injection
     var locale: Locale {
@@ -65,7 +68,8 @@ class Reporter: NSObject {
 
         let request = NSMutableURLRequest.makeJSONPost(
             url: url,
-            httpBody: httpBody
+            httpBody: httpBody,
+            additionalHTTPHeaders: additionalHTTPHeaders
         )
 
         guard let dataTask = session?.dataTask(
@@ -268,7 +272,8 @@ extension Reporter: URLSessionDelegate, URLSessionTaskDelegate {
             // for any weirdness
             let request = NSMutableURLRequest.makeJSONPost(
                 url: redirectURL,
-                httpBody: httpBody
+                httpBody: httpBody,
+                additionalHTTPHeaders: additionalHTTPHeaders
             )
 
             completionHandler(request as URLRequest)

Sources/MuxUploadSDK/PublicAPI/MuxUpload.swift

@@ -430,7 +430,9 @@ public final class MuxUpload : Hashable, Equatable {
 
         if self.manageBySDK {
             // See if there's anything in progress already
-            fileWorker = uploadManager.findChunkedFileUploader(inputFileURL: videoFile)
+            fileWorker = uploadManager.findChunkedFileUploader(
+                inputFileURL: videoFile
+            )
         }
         if fileWorker != nil && !forceRestart {
             MuxUploadSDK.logger?.warning("start() called but upload is already in progress")
@@ -613,9 +615,8 @@ public final class MuxUpload : Hashable, Equatable {
         )
         fileWorker.start()
         self.fileWorker = fileWorker
-<<<<<<< HEAD
         uploadManager.registerUpload(self)
-=======
+
         let transportStatus = TransportStatus(
             progress: fileWorker.currentState.progress ?? Progress(),
             updatedTime: Date().timeIntervalSince1970,
@@ -626,7 +627,6 @@ public final class MuxUpload : Hashable, Equatable {
             startingTransportStatus: transportStatus
         )
         inputStatusHandler?(inputStatus)
->>>>>>> 071c77b (Input standardization 2 (#41))
     }
 
     func startNetworkTransport(

Sources/MuxUploadSDK/PublicAPI/UploadManager.swift

@@ -117,6 +117,7 @@ public final class UploadManager {
     }
 
     internal func registerUpload(_ upload: MuxUpload) {
+
         guard let fileWorker = upload.fileWorker else {
             // Only started uploads, aka uploads with a file
             // worker can be registered.
@@ -126,10 +127,6 @@ public final class UploadManager {
         }
 
         uploadsByID.updateValue(upload, forKey: upload.id)
-    }
-    
-    internal func registerUploader(_ fileWorker: ChunkedFileUploader, withId id: String) {
-        uploadsByID.updateValue(fileWorker, forKey: fileWorker.uploadInfo.id)
         fileWorker.addDelegate(withToken: UUID().uuidString, uploaderDelegate)
         Task.detached {
             await self.uploadActor.updateUpload(