Merge pull request #15 from TheMagoo73/add-traits-support

Add traits support

Author: TheMagoo73

index.d.ts

@@ -61,8 +61,31 @@ export interface Flagsmith {
      * Force a re-evaluation and fetch of flags
      */
     getFlags:() => Promise<IFlags>
+
+    /**
+     * Get the value of a specific trait for the currently identified user
+     */
+    getTrait:(key:string) => string|number|boolean
+
+    /**
+     * Set the value of a trait for the current user. Triggers a fetch of the flags 
+     */
+    setTrait:(key: string, value: string|number|boolean) => Promise<IFlags>
+
+    /**
+     * Increment the value of a trait for the current user. Triggers a fetch of the flags
+     */
+    incrementTrait:(key: string, incrementBy: number) => Promise<IFlags>
+
+    /**
+     * Set a key value set of traits for a given user, triggers a fetch of the flags 
+     */
+    setTraits:(traits: Record<string, string|number|boolean>) => Promise<IFlags>
+
+
 }
 
+
 /**
  * Hook function to access Flagsmith functionality
  */

package.json

@@ -1,6 +1,6 @@
 {
   "name": "flagsmith-react",
-  "version": "0.1.1-alpha.3",
+  "version": "0.1.2-alpha.1",
   "preview": true,
   "description": "Flagsmith integration for React Single Page Applications (SPA)",
   "repository": {

src/flagsmith-provider.js

@@ -94,6 +94,30 @@ const FlagsmithProvider = ({ environmentId, children, flagsmith = reactFlagsmith
     }, [flagsmith]
   )
 
+  const getTrait = useCallback(
+    (key) => {
+      return flagsmith.getTrait(key)
+    }, [flagsmith]
+  )
+
+  const setTrait = useCallback(
+    async (key, value) => {
+      return flagsmith.setTrait(key, value)
+    }, [flagsmith]
+  )
+
+  const incrementTrait = useCallback(
+    async (key, incrementBy) => {
+      return flagsmith.incrementTrait(key, incrementBy)
+    }, [flagsmith]
+  )
+
+  const setTraits = useCallback(
+    async (traits) => {
+      return flagsmith.setTraits(traits)
+    }, [flagsmith]
+  )
+
   return (
     <FlagsmithContext.Provider value={{
       ...state, 
@@ -104,7 +128,11 @@ const FlagsmithProvider = ({ environmentId, children, flagsmith = reactFlagsmith
       logout,
       startListening,
       stopListening,
-      getFlags}}>
+      getFlags,
+      getTrait,
+      setTrait,
+      setTraits,
+      incrementTrait}}>
       {children}
     </FlagsmithContext.Provider>
   )