Update SupabaseClient.swift

Author: satishbabariya

Sources/Supabase/SupabaseClient.swift

@@ -5,16 +5,16 @@ import SupabaseStorage
 
 /**
  The main class for accessing Supabase functionality
- 
+
  Initialize this class using `.init(supabaseURL: String, supabaseKey: String)`
- 
+
  There are four main classes contained by the `Supabase` class.
  1.  `auth`
  2.  `database`
  3.  `realtime`
  4.  `storage`
  Each class listed is available under `Supabase.{name}`, eg: `Supabase.auth`
- 
+
  For more usage information read the README.md
  */
 public class SupabaseClient {
@@ -25,29 +25,29 @@ public class SupabaseClient {
     private var realtimeUrl: String
     private var authUrl: String
     private var storageUrl: String
-    
+
     /// Auth client for Supabase
     public var auth: GoTrueClient
-    
+
     /// Storage client for Supabase.
     public var storage: SupabaseStorageClient {
-        var headers: [String: String] = [:]
-        headers["apikey"] = supabaseKey
+        var headers: [String: String] = defaultHeaders
         headers["Authorization"] = "Bearer \(auth.session?.accessToken ?? supabaseKey)"
         return SupabaseStorageClient(url: storageUrl, headers: headers)
     }
-    
+
     /// Database client for Supabase.
     public var database: PostgrestClient {
-        var headers: [String: String] = [:]
-        headers["apikey"] = supabaseKey
+        var headers: [String: String] = defaultHeaders
         headers["Authorization"] = "Bearer \(auth.session?.accessToken ?? supabaseKey)"
         return PostgrestClient(url: restUrl, headers: headers, schema: schema)
     }
-    
+
     /// Realtime client for Supabase
     private var realtime: RealtimeClient
-    
+
+    private var defaultHeaders: [String: String]
+
     /// Init `Supabase` with the provided parameters.
     /// - Parameters:
     ///   - supabaseUrl: Unique Supabase project url
@@ -67,12 +67,14 @@ public class SupabaseClient {
         realtimeUrl = "\(supabaseUrl)/realtime/v1"
         authUrl = "\(supabaseUrl)/auth/v1"
         storageUrl = "\(supabaseUrl)/storage/v1"
-        
+
+        defaultHeaders = ["X-Client-Info": "supabase-swift/0.0.1", "apikey": supabaseKey]
+
         auth = GoTrueClient(
             url: authUrl,
-            headers: ["apikey": supabaseKey],
+            headers: defaultHeaders,
             autoRefreshToken: autoRefreshToken
         )
-        realtime = RealtimeClient(endPoint: realtimeUrl, params: ["apikey": supabaseKey])
+        realtime = RealtimeClient(endPoint: realtimeUrl, params: defaultHeaders)
     }
 }