Implemented getting flags by identity and setting traits at the same time

FlagsmithClient/Classes/Flagsmith.swift

@@ -268,6 +268,21 @@ 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
     ///

FlagsmithClient/Classes/Traits.swift

@@ -13,4 +13,11 @@ import Foundation
 public struct Traits: Codable, Sendable {
     public let traits: [Trait]
     public let identifier: String?
+    public let flags: [Flag]
+
+    init(traits: [Trait], identifier: String?, flags: [Flag] = []) {
+        self.traits = traits
+        self.identifier = identifier
+        self.flags = flags
+    }
 }

FlagsmithClient/Tests/RouterTests.swift

@@ -76,7 +76,8 @@ final class RouterTests: FlagsmithClientTestCase {
               "trait_value" : 42
             }
           ],
-          "identifier" : "A1B2C3D4"
+          "identifier" : "A1B2C3D4",
+          "flags": []
         }
         """.json(using: .utf8)
         let body = try request.httpBody.json()