Merge pull request #29 from Trendyol/fix/RecalculateContent-LengthHeader

fix/RecalculateContent-LengthHeader
Trendyol · Jul 11, 2024 · c5d11a5 · c5d11a5
2 parents 5911ce9 + e25e31f
commit c5d11a5
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 5 deletions.
diff --git a/...Kit/Extensions/URLRequest+Extension.swift → ...Kit/Extensions/URLRequest+Extension.swift b/...Kit/Extensions/URLRequest+Extension.swift → ...Kit/Extensions/URLRequest+Extension.swift
@@ -10,8 +10,8 @@ import Foundation
 import FoundationNetworking
 #endif
 
-extension URLRequest {
-    public func cURL(pretty: Bool = false) -> String {
+public extension URLRequest {
+    func cURL(pretty: Bool = false) -> String {
         let newLine = pretty ? "\\\n" : ""
         let method = (pretty ? "--request " : "-X ") + "\(self.httpMethod ?? "GET") \(newLine)"
         let url: String = (pretty ? "--url " : "") + "\'\(self.url?.absoluteString ?? "")\' \(newLine)"
@@ -34,4 +34,14 @@ extension URLRequest {
 
         return cURL
     }
+
+    func recalculateContentLength() -> URLRequest {
+        var request = self
+
+        if !request.value(forHTTPHeaderField: "Content-Length").isNilOrEmpty {
+            request.setValue(request.httpBody?.count.description, forHTTPHeaderField: "Content-Length")
+        }
+
+        return request
+    }
 }
diff --git a/Core/MockingStarCore/Sources/MockingStarCore/MockingStar.swift b/Core/MockingStarCore/Sources/MockingStarCore/MockingStar.swift
@@ -89,9 +89,9 @@ public final class MockingStarCore {
         let liveRequest: URLRequest
 
         if let mockDomain {
-            liveRequest = try await updateProxyRequestWithPlugin(request: request, mockDomain: mockDomain)
+            liveRequest = try await updateProxyRequestWithPlugin(request: request, mockDomain: mockDomain).recalculateContentLength()
         } else {
-            liveRequest = request
+            liveRequest = request.recalculateContentLength()
         }
 
         let (data, response) = try await URLSession.shared.data(for: liveRequest)

diff --git a/Features/MockDetail/Sources/MockDetail/Views/MockReload/MockReloadViewModel.swift b/Features/MockDetail/Sources/MockDetail/Views/MockReload/MockReloadViewModel.swift
@@ -91,10 +91,11 @@ final class MockReloadViewModel {
     func reloadMock() async {
         isMockReloadingProgress = true
 
+        let request = (showUpdatedRequest ? updatedRequest() : mockModel.asURLRequest).recalculateContentLength()
         let result: (Data, URLResponse)
 
         do {
-            result = try await urlSession.data(for: showUpdatedRequest ? updatedRequest() : mockModel.asURLRequest)
+            result = try await urlSession.data(for: request)
         } catch {
             print("Error: \(error)")
             isMockReloadingProgress = false