Set the headers from the APIRequest library.

duckduckgo · Aug 30, 2024 · 4d3e43b · 4d3e43b
1 parent 85b776f
commit 4d3e43b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/Package.swift b/Package.swift
@@ -323,6 +323,7 @@ let package = Package(
             dependencies: [
                 .target(name: "WireGuardC"),
                 "Common",
+                "Networking"
             ],
             swiftSettings: [
                 .define("DEBUG", .when(configuration: .debug))

diff --git a/Sources/NetworkProtection/Networking/NetworkProtectionClient.swift b/Sources/NetworkProtection/Networking/NetworkProtectionClient.swift
@@ -17,6 +17,8 @@
 //
 
 import Foundation
+import Networking
+import os.log
 
 protocol NetworkProtectionClient {
     func getLocations(authToken: String) async -> Result<[NetworkProtectionLocation], NetworkProtectionClientError>
@@ -244,6 +246,7 @@ final class NetworkProtectionBackendClient: NetworkProtectionClient {
 
     func getLocations(authToken: String) async -> Result<[NetworkProtectionLocation], NetworkProtectionClientError> {
         var request = URLRequest(url: locationsURL)
+        request.allHTTPHeaderFields = APIRequest.Headers().httpHeaders
         request.setValue("bearer \(authToken)", forHTTPHeaderField: "Authorization")
         let downloadedData: Data
 
@@ -287,6 +290,7 @@ final class NetworkProtectionBackendClient: NetworkProtectionClient {
 
     func getServers(authToken: String) async -> Result<[NetworkProtectionServer], NetworkProtectionClientError> {
         var request = URLRequest(url: serversURL)
+        request.allHTTPHeaderFields = APIRequest.Headers().httpHeaders
         request.setValue("bearer \(authToken)", forHTTPHeaderField: "Authorization")
         let downloadedData: Data
 
@@ -330,6 +334,7 @@ final class NetworkProtectionBackendClient: NetworkProtectionClient {
 
     func getServerStatus(authToken: String, serverName: String) async -> Result<NetworkProtectionServerStatus, NetworkProtectionClientError> {
         var request = URLRequest(url: serverStatusURL(serverName: serverName))
+        request.allHTTPHeaderFields = APIRequest.Headers().httpHeaders
         request.setValue("bearer \(authToken)", forHTTPHeaderField: "Authorization")
         let downloadedData: Data
 
@@ -382,6 +387,7 @@ final class NetworkProtectionBackendClient: NetworkProtectionClient {
         }
 
         var request = URLRequest(url: registerKeyURL)
+        request.allHTTPHeaderFields = APIRequest.Headers().httpHeaders
         request.setValue("bearer \(authToken)", forHTTPHeaderField: "Authorization")
         request.addValue("application/json", forHTTPHeaderField: "Content-Type")
         request.httpMethod = "POST"
@@ -444,6 +450,7 @@ final class NetworkProtectionBackendClient: NetworkProtectionClient {
         }
 
         var request = URLRequest(url: endpoint)
+        request.allHTTPHeaderFields = APIRequest.Headers().httpHeaders
         request.addValue("application/json", forHTTPHeaderField: "Content-Type")
         request.httpMethod = "POST"
         request.httpBody = requestBodyData