Merge pull request #1 from zhigang1992/feat/headers

feat: add custom headers support

Author: devx-opensource

Sources/FalClient/Client+Config.swift

@@ -32,9 +32,11 @@ public enum ClientCredentials: CustomStringConvertible {
 public struct ClientConfig {
     public let credentials: ClientCredentials
     public let requestProxy: String?
+    public let customHeaders: [String: String]?
 
-    init(credentials: ClientCredentials = .fromEnv, requestProxy: String? = nil) {
+    init(credentials: ClientCredentials = .fromEnv, requestProxy: String? = nil, customHeaders: [String: String]? = nil) {
         self.credentials = credentials
         self.requestProxy = requestProxy
+        self.customHeaders = customHeaders
     }
 }

Sources/FalClient/Client+Request.swift

@@ -33,6 +33,11 @@ extension Client {
         }
 
         var request = URLRequest(url: url)
+        if let header = config.customHeaders {
+            for (key, value) in header {
+                request.setValue(value, forHTTPHeaderField: key)
+            }
+        }
         request.httpMethod = options.httpMethod.rawValue.uppercased()
         request.setValue("application/json", forHTTPHeaderField: "accept")
         request.setValue("application/json", forHTTPHeaderField: "content-type")

Sources/FalClient/FalClient.swift

@@ -76,8 +76,8 @@ public struct FalClient: Client {
 }
 
 public extension FalClient {
-    static func withProxy(_ url: String) -> Client {
-        FalClient(config: ClientConfig(requestProxy: url))
+    static func withProxy(_ url: String, headers: [String: String]? = nil) -> Client {
+        FalClient(config: ClientConfig(requestProxy: url, customHeaders: headers))
     }
 
     static func withCredentials(_ credentials: ClientCredentials) -> Client {