Run validateContext during any context change

Author: MattIPv4

packages/web/src/open-feature.ts

@@ -53,6 +53,8 @@ export class OpenFeatureAPI
   );
   protected _domainScopedProviders: Map<string, ProviderWrapper<Provider, ClientProviderStatus>> = new Map();
   protected _createEventEmitter = () => new OpenFeatureEventEmitter();
+  protected _defaultOptions: ProviderOptions = {};
+  protected _domainScopedOptions: Map<string, ProviderOptions> = new Map();
 
   private constructor() {
     super('client');
@@ -82,6 +84,14 @@ export class OpenFeatureAPI
     return this._domainScopedProviders.get(domain)?.status ?? this._defaultProvider.status;
   }
 
+  private getProviderOptions(domain?: string): ProviderOptions {
+    if (!domain) {
+      return this._defaultOptions;
+    }
+
+    return this._domainScopedOptions.get(domain) ?? this._defaultOptions;
+  }
+
   /**
    * Sets the default provider for flag evaluations and returns a promise that resolves when the provider is ready.
    * This provider will be used by domainless clients and clients associated with domains to which no provider is bound.
@@ -141,6 +151,12 @@ export class OpenFeatureAPI
       ? objectOrUndefined<ProviderOptions>(optionsOrUndefined)
       : objectOrUndefined<ProviderOptions>(contextOptionsOrUndefined);
 
+    if (domain) {
+      this._domainScopedOptions.set(domain, options ?? {});
+    } else {
+      this._defaultOptions = options ?? {};
+    }
+
     let skipInitialization = false;
     if (context) {
       // validate the context to decide if we should initialize the provider with it.
@@ -219,6 +235,12 @@ export class OpenFeatureAPI
       ? objectOrUndefined<ProviderOptions>(optionsOrUndefined)
       : objectOrUndefined<ProviderOptions>(contextOptionsOrUndefined);
 
+    if (domain) {
+      this._domainScopedOptions.set(domain, options ?? {});
+    } else {
+      this._defaultOptions = options ?? {};
+    }
+
     let skipInitialization = false;
     let validateContextError: unknown;
     if (context) {
@@ -332,8 +354,6 @@ export class OpenFeatureAPI
     const domain = stringOrUndefined(domainOrContext);
     const context = objectOrUndefined<T>(domainOrContext) ?? objectOrUndefined(contextOrUndefined) ?? {};
 
-    // TODO: We need to store and call `validateContext` here if provided in `setProvider` options
-
     if (domain) {
       const wrapper = this._domainScopedProviders.get(domain);
       if (wrapper) {
@@ -474,8 +494,19 @@ export class OpenFeatureAPI
     const providerName = wrapper.provider?.metadata?.name || 'unnamed-provider';
 
     try {
+      // validate the context to decide if we should run the context change handler.
+      const options = this.getProviderOptions(domain);
+      if (typeof options.validateContext === 'function') {
+        if (!options.validateContext(newContext)) {
+          this._logger.debug(
+            `Skipping context change for domain '${domain ?? 'default'}' due to validateContext returning false.`,
+          );
+          return;
+        }
+      }
+
       // if the provider hasn't initialized yet, and isn't actively initializing, initialize instead of running context change handler
-      // the provider will be in this state if the user requested delayed initialization until the first context change
+      // the provider will be in this state if validateContext was used during setProvider to skip initialization
       const initializationPromise = this.initializeProviderForDomain(wrapper, domain);
       if (initializationPromise) {
         await initializationPromise;