commets, reuse domain from context value

Signed-off-by: Will Chou <[email protected]>

Author: wichopy

packages/react/src/provider/context.ts

@@ -7,7 +7,7 @@ import { NormalizedOptions, ReactFlagEvaluationOptions, normalizeOptions } from
  * DO NOT EXPORT PUBLICLY
  * @internal
  */
-export const Context = React.createContext<{ client: Client; options: ReactFlagEvaluationOptions } | undefined>(undefined);
+export const Context = React.createContext<{ client: Client; domain?: string; options: ReactFlagEvaluationOptions } | undefined>(undefined);
 
 /**
  * Get a normalized copy of the options used for this OpenFeatureProvider, see {@link normalizeOptions}.

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

@@ -1,23 +1,24 @@
+import { useContext } from 'react';
 import { OpenFeature, EvaluationContext } from '@openfeature/web-sdk';
-
-type DomainContextMutator = (domain: string, updatedContext: EvaluationContext) => Promise<void>; 
+import { Context } from './context';
 
 /**
- * 
+ *
  * A hook for accessing context mutating functions.
- * 
+ *
  */
 export function useContextMutator() {
+  async function mutateContext(updatedContext: EvaluationContext): Promise<void> {
+    const { domain } = useContext(Context) || {};
 
-  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);
+    if (!domain) {
+      throw new Error('No domain set for your context');
     }
+
+    OpenFeature.setContext(domain, updatedContext);
   }
-  
+
   return {
-    mutateContext, 
+    mutateContext,
   };
-}
+}