Removed getIdentityFlags method and added traits variable to getFeatu…

…reFlags method
Flagsmith · Jun 20, 2024 · 7046559 · 7046559
1 parent f6a4e64
commit 7046559
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 33 deletions.
diff --git a/FlagsmithClient/Classes/Flagsmith.swift b/FlagsmithClient/Classes/Flagsmith.swift
@@ -83,36 +83,52 @@ public final class Flagsmith: @unchecked Sendable {
     ///   - identity: ID of the user (optional)
     ///   - completion: Closure with Result which contains array of Flag objects in case of success or Error in case of failure
     public func getFeatureFlags(forIdentity identity: String? = nil,
+                                traits: [Trait]? = nil,
                                 completion: @Sendable @escaping (Result<[Flag], any Error>) -> Void)
     {
         if let identity = identity {
-            getIdentity(identity) { result in
-                switch result {
-                case let .success(thisIdentity):
-                    completion(.success(thisIdentity.flags))
-                case let .failure(error):
-                    if self.defaultFlags.isEmpty {
-                        completion(.failure(error))
-                    } else {
-                        completion(.success(self.defaultFlags))
+            if let traits = traits {
+                apiManager.request(.postTraits(identity: identity, traits: traits)) { (result: Result<Traits, Error>) in
+                    switch result {
+                    case let .success(result):
+                        completion(.success(result.flags))
+                    case let .failure(error):
+                        self.handleFlagsError(error, completion: completion)
+                    }
+                }
+            } else {
+                getIdentity(identity) { result in
+                    switch result {
+                    case let .success(thisIdentity):
+                        completion(.success(thisIdentity.flags))
+                    case let .failure(error):
+                        self.handleFlagsError(error, completion: completion)
                     }
                 }
             }
         } else {
-            apiManager.request(.getFlags) { (result: Result<[Flag], Error>) in
-                switch result {
-                case let .success(flags):
-                    completion(.success(flags))
-                case let .failure(error):
-                    if self.defaultFlags.isEmpty {
-                        completion(.failure(error))
-                    } else {
-                        completion(.success(self.defaultFlags))
+            if let _ = traits {
+                completion(.failure(FlagsmithError.invalidArgument("You must provide an identity to set traits")))
+            } else {
+                apiManager.request(.getFlags) { (result: Result<[Flag], Error>) in
+                    switch result {
+                    case let .success(flags):
+                        completion(.success(flags))
+                    case let .failure(error):
+                        self.handleFlagsError(error, completion: completion)
                     }
                 }
             }
         }
     }
+
+    private func handleFlagsError(_ error: any Error, completion: @Sendable @escaping (Result<[Flag], any Error>) -> Void) {
+        if self.defaultFlags.isEmpty {
+            completion(.failure(error))
+        } else {
+            completion(.success(self.defaultFlags))
+        }
+    }
 
     /// Check feature exists and is enabled optionally for a specific identity
     ///
@@ -268,21 +284,6 @@ public final class Flagsmith: @unchecked Sendable {
             completion(result.map(\.traits))
         }
     }
-
-    /// Get flags for an identity and set traits in the same call
-    ///
-    /// - Parameters:
-    ///   - traits: Traits to be created or updated
-    ///   - identity: ID of the user
-    ///   - completion: Closure with Result which contains a list of Flags in case of success or Error in case of failure
-    public func getIdentityFlags(_ traits: [Trait],
-                                 forIdentity identity: String,
-                                 completion: @Sendable @escaping (Result<[Flag], any Error>) -> Void)
-    {
-        apiManager.request(.postTraits(identity: identity, traits: traits)) { (result: Result<Traits, Error>) in
-            completion(result.map(\.flags))
-        }
-    }
 
     /// Get both feature flags and user traits for the provided identity
     ///

diff --git a/FlagsmithClient/Classes/FlagsmithError.swift b/FlagsmithClient/Classes/FlagsmithError.swift
@@ -21,6 +21,8 @@ public enum FlagsmithError: LocalizedError, Sendable {
     case decoding(DecodingError)
     /// Unknown or unhandled error was encountered.
     case unhandled(any Error)
+    /// Invalid argument error
+    case invalidArgument(String)
 
     public var errorDescription: String? {
         switch self {
@@ -36,6 +38,8 @@ public enum FlagsmithError: LocalizedError, Sendable {
             return "API Response could not be decoded: \(error.localizedDescription)"
         case let .unhandled(error):
             return "An unknown or unhandled error was encountered: \(error.localizedDescription)"
+        case let .invalidArgument(error):
+            return "Invalid argument error: \(error)"
         }
     }