handle domain and global

Author: wichopy

packages/react/src/provider/use-context-mutator.ts

@@ -2,28 +2,22 @@ import { OpenFeature, EvaluationContext } from '@openfeature/web-sdk';
 
 type DomainContextMutator = (domain: string, updatedContext: EvaluationContext) => Promise<void>; 
 
-type ContextMutatorReturn = {
-  mutateContext: DomainContextMutator;
-}
 /**
  * 
  * A hook for accessing context mutating functions.
  * 
- * @returns {ContextMutatorReturn} 
  */
-export function useContextMutator(): {
-  mutateContext: DomainContextMutator;
-} {
-  return {
-    /**
-     * 
-     * Mutates the evaluation context for a given domain.
-     * 
-     * @param {string}domain
-     * @param {EvaluationContext} updatedContext
-     */
-    mutateContext: async (domain: string, updatedContext: EvaluationContext) => {
-      await OpenFeature.setContext(domain, updatedContext);
+export function useContextMutator() {
+
+  async function mutateContext(domainOrUpdatedContext: string | EvaluationContext, updatedContextOrUndefined?: EvaluationContext): Promise<void> {
+    if (typeof domainOrUpdatedContext === 'string' && updatedContextOrUndefined) {
+      await OpenFeature.setContext(domainOrUpdatedContext, updatedContextOrUndefined);
+    } else if (typeof domainOrUpdatedContext !== 'string') {
+      OpenFeature.setContext(domainOrUpdatedContext);
     }
+  }
+  
+  return {
+    mutateContext, 
   };
 }