@schematichq/schematic-react 1.2.17 → 1.3.0

package/dist/schematic-react.cjs.js

@@ -39,7 +39,8 @@ __export(index_exports, {
   useSchematicEntitlement: () => useSchematicEntitlement,
   useSchematicEvents: () => useSchematicEvents,
   useSchematicFlag: () => useSchematicFlag,
-  useSchematicIsPending: () => useSchematicIsPending
+  useSchematicIsPending: () => useSchematicIsPending,
+  useSchematicPlan: () => useSchematicPlan
 });
 module.exports = __toCommonJS(index_exports);
 
@@ -651,6 +652,36 @@ function v4(options, buf, offset) {
 }
 var v4_default = v4;
 var import_polyfill = __toESM2(require_browser_polyfill());
+function EntitlementValueTypeFromJSON(json) {
+  return EntitlementValueTypeFromJSONTyped(json, false);
+}
+function EntitlementValueTypeFromJSONTyped(json, ignoreDiscriminator) {
+  return json;
+}
+function FeatureEntitlementFromJSON(json) {
+  return FeatureEntitlementFromJSONTyped(json, false);
+}
+function FeatureEntitlementFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    allocation: json["allocation"] == null ? void 0 : json["allocation"],
+    creditId: json["credit_id"] == null ? void 0 : json["credit_id"],
+    creditRemaining: json["credit_remaining"] == null ? void 0 : json["credit_remaining"],
+    creditTotal: json["credit_total"] == null ? void 0 : json["credit_total"],
+    creditUsed: json["credit_used"] == null ? void 0 : json["credit_used"],
+    eventName: json["event_name"] == null ? void 0 : json["event_name"],
+    featureId: json["feature_id"],
+    featureKey: json["feature_key"],
+    metricPeriod: json["metric_period"] == null ? void 0 : json["metric_period"],
+    metricResetAt: json["metric_reset_at"] == null ? void 0 : new Date(json["metric_reset_at"]),
+    monthReset: json["month_reset"] == null ? void 0 : json["month_reset"],
+    softLimit: json["soft_limit"] == null ? void 0 : json["soft_limit"],
+    usage: json["usage"] == null ? void 0 : json["usage"],
+    valueType: EntitlementValueTypeFromJSON(json["value_type"])
+  };
+}
 function CheckFlagResponseDataFromJSON(json) {
   return CheckFlagResponseDataFromJSONTyped(json, false);
 }
@@ -660,6 +691,7 @@ function CheckFlagResponseDataFromJSONTyped(json, ignoreDiscriminator) {
   }
   return {
     companyId: json["company_id"] == null ? void 0 : json["company_id"],
+    entitlement: json["entitlement"] == null ? void 0 : FeatureEntitlementFromJSON(json["entitlement"]),
     error: json["error"] == null ? void 0 : json["error"],
     featureAllocation: json["feature_allocation"] == null ? void 0 : json["feature_allocation"],
     featureUsage: json["feature_usage"] == null ? void 0 : json["feature_usage"],
@@ -675,26 +707,6 @@ function CheckFlagResponseDataFromJSONTyped(json, ignoreDiscriminator) {
     value: json["value"]
   };
 }
-function EventBodyFlagCheckToJSON(json) {
-  return EventBodyFlagCheckToJSONTyped(json, false);
-}
-function EventBodyFlagCheckToJSONTyped(value, ignoreDiscriminator = false) {
-  if (value == null) {
-    return value;
-  }
-  return {
-    company_id: value["companyId"],
-    error: value["error"],
-    flag_id: value["flagId"],
-    flag_key: value["flagKey"],
-    reason: value["reason"],
-    req_company: value["reqCompany"],
-    req_user: value["reqUser"],
-    rule_id: value["ruleId"],
-    user_id: value["userId"],
-    value: value["value"]
-  };
-}
 function CheckFlagResponseFromJSON(json) {
   return CheckFlagResponseFromJSONTyped(json, false);
 }
@@ -707,6 +719,19 @@ function CheckFlagResponseFromJSONTyped(json, ignoreDiscriminator) {
     params: json["params"]
   };
 }
+function DatastreamCompanyPlanFromJSON(json) {
+  return DatastreamCompanyPlanFromJSONTyped(json, false);
+}
+function DatastreamCompanyPlanFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    id: json["id"],
+    name: json["name"],
+    trialEndDate: json["trial_end_date"] == null ? void 0 : new Date(json["trial_end_date"])
+  };
+}
 function CheckFlagsResponseDataFromJSON(json) {
   return CheckFlagsResponseDataFromJSONTyped(json, false);
 }
@@ -715,7 +740,8 @@ function CheckFlagsResponseDataFromJSONTyped(json, ignoreDiscriminator) {
     return json;
   }
   return {
-    flags: json["flags"].map(CheckFlagResponseDataFromJSON)
+    flags: json["flags"].map(CheckFlagResponseDataFromJSON),
+    plan: json["plan"] == null ? void 0 : DatastreamCompanyPlanFromJSON(json["plan"])
   };
 }
 function CheckFlagsResponseFromJSON(json) {
@@ -730,6 +756,26 @@ function CheckFlagsResponseFromJSONTyped(json, ignoreDiscriminator) {
     params: json["params"]
   };
 }
+function EventBodyFlagCheckToJSON(json) {
+  return EventBodyFlagCheckToJSONTyped(json, false);
+}
+function EventBodyFlagCheckToJSONTyped(value, ignoreDiscriminator = false) {
+  if (value == null) {
+    return value;
+  }
+  return {
+    company_id: value["companyId"],
+    error: value["error"],
+    flag_id: value["flagId"],
+    flag_key: value["flagKey"],
+    reason: value["reason"],
+    req_company: value["reqCompany"],
+    req_user: value["reqUser"],
+    rule_id: value["ruleId"],
+    user_id: value["userId"],
+    value: value["value"]
+  };
+}
 var RuleType = /* @__PURE__ */ ((RuleType2) => {
   RuleType2["GLOBAL_OVERRIDE"] = "global_override";
   RuleType2["COMPANY_OVERRIDE"] = "company_override";
@@ -785,6 +831,14 @@ var CheckFlagReturnFromJSON = (json) => {
     value
   };
 };
+var CheckPlanReturnFromJSON = (json) => {
+  const { id, name, trialEndDate } = DatastreamCompanyPlanFromJSON(json);
+  return {
+    id,
+    name,
+    trialEndDate: trialEndDate == null ? void 0 : trialEndDate
+  };
+};
 function contextString(context) {
   const sortedContext = Object.keys(context).reduce((acc, key) => {
     const sortedKeys = Object.keys(
@@ -799,7 +853,7 @@ function contextString(context) {
   }, {});
   return JSON.stringify(sortedContext);
 }
-var version = "1.2.17";
+var version = "1.3.0";
 var anonymousIdKey = "schematicId";
 var Schematic = class {
   additionalHeaders = {};
@@ -816,10 +870,12 @@ var Schematic = class {
   flagValueListeners = {};
   isPending = true;
   isPendingListeners = /* @__PURE__ */ new Set();
+  planListeners = /* @__PURE__ */ new Set();
   storage;
   useWebSocket = false;
   checks = {};
   featureUsageEventMap = {};
+  planChecks = {};
   webSocketUrl = "wss://api.schematichq.com";
   webSocketConnectionTimeout = 1e4;
   webSocketReconnect = true;
@@ -845,6 +901,7 @@ var Schematic = class {
   retryTimer = null;
   flagValueDefaults = {};
   flagCheckDefaults = {};
+  fallbackCheckCache = {};
   constructor(apiKey, options) {
     this.apiKey = apiKey;
     this.eventQueue = [];
@@ -1104,6 +1161,13 @@ var Schematic = class {
       );
       if (typeof flagCheck !== "undefined") {
         this.submitFlagCheckEvent(key, flagCheck, context);
+      } else {
+        const fallbackResult = this.resolveFallbackCheckFlagReturn(
+          key,
+          fallback,
+          "No flag values available"
+        );
+        this.submitFlagCheckEvent(key, fallbackResult, context);
       }
       return result;
     } catch (error) {
@@ -1144,28 +1208,32 @@ var Schematic = class {
           this.checks[contextStr] = {};
         }
         this.checks[contextStr][flagCheck.flag] = flagCheck;
-        this.debug(`WebSocket flag update:`, {
-          flag: flagCheck.flag,
-          value: flagCheck.value,
-          flagCheck
-        });
         if (typeof flagCheck.featureUsageEvent === "string") {
           this.updateFeatureUsageEventMap(flagCheck);
         }
         if ((this.flagCheckListeners[flag.flag]?.size ?? 0) > 0 || (this.flagValueListeners[flag.flag]?.size ?? 0) > 0) {
           this.submitFlagCheckEvent(flagCheck.flag, flagCheck, context);
         }
-        this.debug(`About to notify listeners for flag ${flag.flag}`, {
-          flag: flag.flag,
-          value: flagCheck.value
-        });
+        this.debug(
+          `WebSocket flag update received. Notifying listeners for ${flag.flag}`,
+          {
+            flag: flag.flag,
+            value: flagCheck.value,
+            flagCheck
+          }
+        );
         this.notifyFlagCheckListeners(flag.flag, flagCheck);
         this.notifyFlagValueListeners(flag.flag, flagCheck.value);
-        this.debug(`Finished notifying listeners for flag ${flag.flag}`, {
-          flag: flag.flag,
-          value: flagCheck.value
-        });
       });
+      if (message.plan !== void 0 && message.plan !== null) {
+        const plan = CheckPlanReturnFromJSON(message.plan);
+        const contextStr = contextString(context);
+        this.planChecks[contextStr] = plan;
+        this.debug(`WebSocket plan update received. Notifying listeners`, {
+          plan
+        });
+        this.notifyPlanListeners(plan);
+      }
       this.flushContextDependentEventQueue();
       this.setIsPending(false);
     };
@@ -1269,6 +1337,9 @@ var Schematic = class {
       this.setIsPending(false);
       return Promise.resolve();
     }
+    if (contextString(context) === contextString(this.context) && this.conn !== null && !this.isPending) {
+      return;
+    }
     try {
       this.setIsPending(true);
       if (!this.conn) {
@@ -1308,7 +1379,10 @@ var Schematic = class {
       await this.wsSendMessage(socket, context);
     } catch (error) {
       console.warn("Failed to establish WebSocket connection:", error);
-      throw error;
+      this.context = context;
+      this.flushContextDependentEventQueue();
+      this.setIsPending(false);
+      this.attemptReconnect();
     }
   };
   /**
@@ -2049,11 +2123,17 @@ var Schematic = class {
     };
   };
   setIsPending = (isPending) => {
+    if (this.isPending === isPending) return;
     this.isPending = isPending;
     this.isPendingListeners.forEach(
       (listener) => notifyPendingListener(listener, isPending)
     );
   };
+  getPlan = () => {
+    const contextStr = contextString(this.context);
+    const plan = this.planChecks[contextStr];
+    return plan;
+  };
   // flag checks state
   getFlagCheck = (flagKey) => {
     const contextStr = contextString(this.context);
@@ -2063,11 +2143,14 @@ var Schematic = class {
       return check;
     }
     if (flagKey in this.flagCheckDefaults || flagKey in this.flagValueDefaults) {
-      return this.resolveFallbackCheckFlagReturn(
-        flagKey,
-        void 0,
-        "Default value used"
-      );
+      if (!(flagKey in this.fallbackCheckCache)) {
+        this.fallbackCheckCache[flagKey] = this.resolveFallbackCheckFlagReturn(
+          flagKey,
+          void 0,
+          "Default value used"
+        );
+      }
+      return this.fallbackCheckCache[flagKey];
     }
     return void 0;
   };
@@ -2104,6 +2187,12 @@ var Schematic = class {
       this.flagCheckListeners[flagKey].delete(listener);
     };
   };
+  addPlanListener = (listener) => {
+    this.planListeners.add(listener);
+    return () => {
+      this.planListeners.delete(listener);
+    };
+  };
   notifyFlagCheckListeners = (flagKey, check) => {
     const listeners = this.flagCheckListeners?.[flagKey] ?? [];
     if (listeners.size > 0) {
@@ -2152,6 +2241,21 @@ var Schematic = class {
       });
     });
   };
+  notifyPlanListeners = (value) => {
+    const listeners = this.planListeners ?? [];
+    if (listeners.size > 0) {
+      this.debug(`Notifying ${listeners.size} plan listeners`, { value });
+    }
+    listeners.forEach((listener, index) => {
+      this.debug(`Calling listener ${index} for plan`, {
+        value
+      });
+      notifyPlanListener(listener, value);
+      this.debug(`Listener ${index} for plan completed`, {
+        value
+      });
+    });
+  };
 };
 var notifyPendingListener = (listener, value) => {
   if (listener.length > 0) {
@@ -2174,12 +2278,19 @@ var notifyFlagValueListener = (listener, value) => {
     listener();
   }
 };
+var notifyPlanListener = (listener, value) => {
+  if (listener.length > 0) {
+    listener(value);
+  } else {
+    listener();
+  }
+};
 
 // src/context/schematic.tsx
 var import_react = __toESM(require("react"));
 
 // src/version.ts
-var version2 = "1.2.17";
+var version2 = "1.3.0";
 
 // src/context/schematic.tsx
 var import_jsx_runtime = require("react/jsx-runtime");
@@ -2300,6 +2411,23 @@ var useSchematicEntitlement = (key, opts) => {
   }, [client, key, fallbackCheck]);
   return (0, import_react2.useSyncExternalStore)(subscribe, getSnapshot, () => fallbackCheck);
 };
+var useSchematicPlan = (opts) => {
+  const client = useSchematicClient(opts);
+  const fallback = opts?.fallback;
+  const fallbackPlan = (0, import_react2.useMemo)(
+    () => fallback,
+    [fallback?.id, fallback?.name, fallback?.trialEndDate?.getTime()]
+  );
+  const subscribe = (0, import_react2.useCallback)(
+    (callback) => client.addPlanListener(callback),
+    [client]
+  );
+  const getSnapshot = (0, import_react2.useCallback)(() => {
+    const plan = client.getPlan();
+    return plan ?? fallbackPlan;
+  }, [client, fallbackPlan]);
+  return (0, import_react2.useSyncExternalStore)(subscribe, getSnapshot, () => fallbackPlan);
+};
 var useSchematicIsPending = (opts) => {
   const client = useSchematicClient(opts);
   const subscribe = (0, import_react2.useCallback)(

package/dist/schematic-react.d.ts

@@ -1,4 +1,5 @@
 import { CheckFlagReturn } from '@schematichq/schematic-js';
+import { CheckPlanReturn } from '@schematichq/schematic-js';
 import { Event as Event_2 } from '@schematichq/schematic-js';
 import { EventBody } from '@schematichq/schematic-js';
 import { EventBodyIdentify } from '@schematichq/schematic-js';
@@ -21,6 +22,8 @@ declare type BaseSchematicProviderProps = Omit<SchematicJS.SchematicOptions, "cl
 
 export { CheckFlagReturn }
 
+export { CheckPlanReturn }
+
 export { Event_2 as Event }
 
 export { EventBody }
@@ -90,4 +93,10 @@ export declare type UseSchematicFlagOpts = SchematicHookOpts & {
 
 export declare const useSchematicIsPending: (opts?: SchematicHookOpts) => boolean;
 
+export declare const useSchematicPlan: (opts?: UseSchematicPlanOpts) => SchematicJS.CheckPlanReturn | undefined;
+
+export declare type UseSchematicPlanOpts = SchematicHookOpts & {
+    fallback?: SchematicJS.CheckPlanReturn;
+};
+
 export { }

package/dist/schematic-react.esm.js

@@ -606,6 +606,36 @@ function v4(options, buf, offset) {
 }
 var v4_default = v4;
 var import_polyfill = __toESM(require_browser_polyfill());
+function EntitlementValueTypeFromJSON(json) {
+  return EntitlementValueTypeFromJSONTyped(json, false);
+}
+function EntitlementValueTypeFromJSONTyped(json, ignoreDiscriminator) {
+  return json;
+}
+function FeatureEntitlementFromJSON(json) {
+  return FeatureEntitlementFromJSONTyped(json, false);
+}
+function FeatureEntitlementFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    allocation: json["allocation"] == null ? void 0 : json["allocation"],
+    creditId: json["credit_id"] == null ? void 0 : json["credit_id"],
+    creditRemaining: json["credit_remaining"] == null ? void 0 : json["credit_remaining"],
+    creditTotal: json["credit_total"] == null ? void 0 : json["credit_total"],
+    creditUsed: json["credit_used"] == null ? void 0 : json["credit_used"],
+    eventName: json["event_name"] == null ? void 0 : json["event_name"],
+    featureId: json["feature_id"],
+    featureKey: json["feature_key"],
+    metricPeriod: json["metric_period"] == null ? void 0 : json["metric_period"],
+    metricResetAt: json["metric_reset_at"] == null ? void 0 : new Date(json["metric_reset_at"]),
+    monthReset: json["month_reset"] == null ? void 0 : json["month_reset"],
+    softLimit: json["soft_limit"] == null ? void 0 : json["soft_limit"],
+    usage: json["usage"] == null ? void 0 : json["usage"],
+    valueType: EntitlementValueTypeFromJSON(json["value_type"])
+  };
+}
 function CheckFlagResponseDataFromJSON(json) {
   return CheckFlagResponseDataFromJSONTyped(json, false);
 }
@@ -615,6 +645,7 @@ function CheckFlagResponseDataFromJSONTyped(json, ignoreDiscriminator) {
   }
   return {
     companyId: json["company_id"] == null ? void 0 : json["company_id"],
+    entitlement: json["entitlement"] == null ? void 0 : FeatureEntitlementFromJSON(json["entitlement"]),
     error: json["error"] == null ? void 0 : json["error"],
     featureAllocation: json["feature_allocation"] == null ? void 0 : json["feature_allocation"],
     featureUsage: json["feature_usage"] == null ? void 0 : json["feature_usage"],
@@ -630,26 +661,6 @@ function CheckFlagResponseDataFromJSONTyped(json, ignoreDiscriminator) {
     value: json["value"]
   };
 }
-function EventBodyFlagCheckToJSON(json) {
-  return EventBodyFlagCheckToJSONTyped(json, false);
-}
-function EventBodyFlagCheckToJSONTyped(value, ignoreDiscriminator = false) {
-  if (value == null) {
-    return value;
-  }
-  return {
-    company_id: value["companyId"],
-    error: value["error"],
-    flag_id: value["flagId"],
-    flag_key: value["flagKey"],
-    reason: value["reason"],
-    req_company: value["reqCompany"],
-    req_user: value["reqUser"],
-    rule_id: value["ruleId"],
-    user_id: value["userId"],
-    value: value["value"]
-  };
-}
 function CheckFlagResponseFromJSON(json) {
   return CheckFlagResponseFromJSONTyped(json, false);
 }
@@ -662,6 +673,19 @@ function CheckFlagResponseFromJSONTyped(json, ignoreDiscriminator) {
     params: json["params"]
   };
 }
+function DatastreamCompanyPlanFromJSON(json) {
+  return DatastreamCompanyPlanFromJSONTyped(json, false);
+}
+function DatastreamCompanyPlanFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    id: json["id"],
+    name: json["name"],
+    trialEndDate: json["trial_end_date"] == null ? void 0 : new Date(json["trial_end_date"])
+  };
+}
 function CheckFlagsResponseDataFromJSON(json) {
   return CheckFlagsResponseDataFromJSONTyped(json, false);
 }
@@ -670,7 +694,8 @@ function CheckFlagsResponseDataFromJSONTyped(json, ignoreDiscriminator) {
     return json;
   }
   return {
-    flags: json["flags"].map(CheckFlagResponseDataFromJSON)
+    flags: json["flags"].map(CheckFlagResponseDataFromJSON),
+    plan: json["plan"] == null ? void 0 : DatastreamCompanyPlanFromJSON(json["plan"])
   };
 }
 function CheckFlagsResponseFromJSON(json) {
@@ -685,6 +710,26 @@ function CheckFlagsResponseFromJSONTyped(json, ignoreDiscriminator) {
     params: json["params"]
   };
 }
+function EventBodyFlagCheckToJSON(json) {
+  return EventBodyFlagCheckToJSONTyped(json, false);
+}
+function EventBodyFlagCheckToJSONTyped(value, ignoreDiscriminator = false) {
+  if (value == null) {
+    return value;
+  }
+  return {
+    company_id: value["companyId"],
+    error: value["error"],
+    flag_id: value["flagId"],
+    flag_key: value["flagKey"],
+    reason: value["reason"],
+    req_company: value["reqCompany"],
+    req_user: value["reqUser"],
+    rule_id: value["ruleId"],
+    user_id: value["userId"],
+    value: value["value"]
+  };
+}
 var RuleType = /* @__PURE__ */ ((RuleType2) => {
   RuleType2["GLOBAL_OVERRIDE"] = "global_override";
   RuleType2["COMPANY_OVERRIDE"] = "company_override";
@@ -740,6 +785,14 @@ var CheckFlagReturnFromJSON = (json) => {
     value
   };
 };
+var CheckPlanReturnFromJSON = (json) => {
+  const { id, name, trialEndDate } = DatastreamCompanyPlanFromJSON(json);
+  return {
+    id,
+    name,
+    trialEndDate: trialEndDate == null ? void 0 : trialEndDate
+  };
+};
 function contextString(context) {
   const sortedContext = Object.keys(context).reduce((acc, key) => {
     const sortedKeys = Object.keys(
@@ -754,7 +807,7 @@ function contextString(context) {
   }, {});
   return JSON.stringify(sortedContext);
 }
-var version = "1.2.17";
+var version = "1.3.0";
 var anonymousIdKey = "schematicId";
 var Schematic = class {
   additionalHeaders = {};
@@ -771,10 +824,12 @@ var Schematic = class {
   flagValueListeners = {};
   isPending = true;
   isPendingListeners = /* @__PURE__ */ new Set();
+  planListeners = /* @__PURE__ */ new Set();
   storage;
   useWebSocket = false;
   checks = {};
   featureUsageEventMap = {};
+  planChecks = {};
   webSocketUrl = "wss://api.schematichq.com";
   webSocketConnectionTimeout = 1e4;
   webSocketReconnect = true;
@@ -800,6 +855,7 @@ var Schematic = class {
   retryTimer = null;
   flagValueDefaults = {};
   flagCheckDefaults = {};
+  fallbackCheckCache = {};
   constructor(apiKey, options) {
     this.apiKey = apiKey;
     this.eventQueue = [];
@@ -1059,6 +1115,13 @@ var Schematic = class {
       );
       if (typeof flagCheck !== "undefined") {
         this.submitFlagCheckEvent(key, flagCheck, context);
+      } else {
+        const fallbackResult = this.resolveFallbackCheckFlagReturn(
+          key,
+          fallback,
+          "No flag values available"
+        );
+        this.submitFlagCheckEvent(key, fallbackResult, context);
       }
       return result;
     } catch (error) {
@@ -1099,28 +1162,32 @@ var Schematic = class {
           this.checks[contextStr] = {};
         }
         this.checks[contextStr][flagCheck.flag] = flagCheck;
-        this.debug(`WebSocket flag update:`, {
-          flag: flagCheck.flag,
-          value: flagCheck.value,
-          flagCheck
-        });
         if (typeof flagCheck.featureUsageEvent === "string") {
           this.updateFeatureUsageEventMap(flagCheck);
         }
         if ((this.flagCheckListeners[flag.flag]?.size ?? 0) > 0 || (this.flagValueListeners[flag.flag]?.size ?? 0) > 0) {
           this.submitFlagCheckEvent(flagCheck.flag, flagCheck, context);
         }
-        this.debug(`About to notify listeners for flag ${flag.flag}`, {
-          flag: flag.flag,
-          value: flagCheck.value
-        });
+        this.debug(
+          `WebSocket flag update received. Notifying listeners for ${flag.flag}`,
+          {
+            flag: flag.flag,
+            value: flagCheck.value,
+            flagCheck
+          }
+        );
         this.notifyFlagCheckListeners(flag.flag, flagCheck);
         this.notifyFlagValueListeners(flag.flag, flagCheck.value);
-        this.debug(`Finished notifying listeners for flag ${flag.flag}`, {
-          flag: flag.flag,
-          value: flagCheck.value
-        });
       });
+      if (message.plan !== void 0 && message.plan !== null) {
+        const plan = CheckPlanReturnFromJSON(message.plan);
+        const contextStr = contextString(context);
+        this.planChecks[contextStr] = plan;
+        this.debug(`WebSocket plan update received. Notifying listeners`, {
+          plan
+        });
+        this.notifyPlanListeners(plan);
+      }
       this.flushContextDependentEventQueue();
       this.setIsPending(false);
     };
@@ -1224,6 +1291,9 @@ var Schematic = class {
       this.setIsPending(false);
       return Promise.resolve();
     }
+    if (contextString(context) === contextString(this.context) && this.conn !== null && !this.isPending) {
+      return;
+    }
     try {
       this.setIsPending(true);
       if (!this.conn) {
@@ -1263,7 +1333,10 @@ var Schematic = class {
       await this.wsSendMessage(socket, context);
     } catch (error) {
       console.warn("Failed to establish WebSocket connection:", error);
-      throw error;
+      this.context = context;
+      this.flushContextDependentEventQueue();
+      this.setIsPending(false);
+      this.attemptReconnect();
     }
   };
   /**
@@ -2004,11 +2077,17 @@ var Schematic = class {
     };
   };
   setIsPending = (isPending) => {
+    if (this.isPending === isPending) return;
     this.isPending = isPending;
     this.isPendingListeners.forEach(
       (listener) => notifyPendingListener(listener, isPending)
     );
   };
+  getPlan = () => {
+    const contextStr = contextString(this.context);
+    const plan = this.planChecks[contextStr];
+    return plan;
+  };
   // flag checks state
   getFlagCheck = (flagKey) => {
     const contextStr = contextString(this.context);
@@ -2018,11 +2097,14 @@ var Schematic = class {
       return check;
     }
     if (flagKey in this.flagCheckDefaults || flagKey in this.flagValueDefaults) {
-      return this.resolveFallbackCheckFlagReturn(
-        flagKey,
-        void 0,
-        "Default value used"
-      );
+      if (!(flagKey in this.fallbackCheckCache)) {
+        this.fallbackCheckCache[flagKey] = this.resolveFallbackCheckFlagReturn(
+          flagKey,
+          void 0,
+          "Default value used"
+        );
+      }
+      return this.fallbackCheckCache[flagKey];
     }
     return void 0;
   };
@@ -2059,6 +2141,12 @@ var Schematic = class {
       this.flagCheckListeners[flagKey].delete(listener);
     };
   };
+  addPlanListener = (listener) => {
+    this.planListeners.add(listener);
+    return () => {
+      this.planListeners.delete(listener);
+    };
+  };
   notifyFlagCheckListeners = (flagKey, check) => {
     const listeners = this.flagCheckListeners?.[flagKey] ?? [];
     if (listeners.size > 0) {
@@ -2107,6 +2195,21 @@ var Schematic = class {
       });
     });
   };
+  notifyPlanListeners = (value) => {
+    const listeners = this.planListeners ?? [];
+    if (listeners.size > 0) {
+      this.debug(`Notifying ${listeners.size} plan listeners`, { value });
+    }
+    listeners.forEach((listener, index) => {
+      this.debug(`Calling listener ${index} for plan`, {
+        value
+      });
+      notifyPlanListener(listener, value);
+      this.debug(`Listener ${index} for plan completed`, {
+        value
+      });
+    });
+  };
 };
 var notifyPendingListener = (listener, value) => {
   if (listener.length > 0) {
@@ -2129,12 +2232,19 @@ var notifyFlagValueListener = (listener, value) => {
     listener();
   }
 };
+var notifyPlanListener = (listener, value) => {
+  if (listener.length > 0) {
+    listener(value);
+  } else {
+    listener();
+  }
+};
 
 // src/context/schematic.tsx
 import React, { createContext, useEffect, useMemo, useRef } from "react";
 
 // src/version.ts
-var version2 = "1.2.17";
+var version2 = "1.3.0";
 
 // src/context/schematic.tsx
 import { jsx } from "react/jsx-runtime";
@@ -2255,6 +2365,23 @@ var useSchematicEntitlement = (key, opts) => {
   }, [client, key, fallbackCheck]);
   return useSyncExternalStore(subscribe, getSnapshot, () => fallbackCheck);
 };
+var useSchematicPlan = (opts) => {
+  const client = useSchematicClient(opts);
+  const fallback = opts?.fallback;
+  const fallbackPlan = useMemo2(
+    () => fallback,
+    [fallback?.id, fallback?.name, fallback?.trialEndDate?.getTime()]
+  );
+  const subscribe = useCallback(
+    (callback) => client.addPlanListener(callback),
+    [client]
+  );
+  const getSnapshot = useCallback(() => {
+    const plan = client.getPlan();
+    return plan ?? fallbackPlan;
+  }, [client, fallbackPlan]);
+  return useSyncExternalStore(subscribe, getSnapshot, () => fallbackPlan);
+};
 var useSchematicIsPending = (opts) => {
   const client = useSchematicClient(opts);
   const subscribe = useCallback(
@@ -2274,7 +2401,8 @@ export {
   useSchematicEntitlement,
   useSchematicEvents,
   useSchematicFlag,
-  useSchematicIsPending
+  useSchematicIsPending,
+  useSchematicPlan
 };
 /*! Bundled license information:
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@schematichq/schematic-react",
-  "version": "1.2.17",
+  "version": "1.3.0",
   "main": "dist/schematic-react.cjs.js",
   "module": "dist/schematic-react.esm.js",
   "types": "dist/schematic-react.d.ts",
@@ -31,34 +31,39 @@
     "prepare": "husky"
   },
   "dependencies": {
-    "@schematichq/schematic-js": "^1.2.17"
+    "@schematichq/schematic-js": "^1.3.0"
   },
   "devDependencies": {
     "@eslint/js": "^10.0.1",
-    "@microsoft/api-extractor": "^7.56.3",
+    "@microsoft/api-extractor": "^7.57.7",
     "@testing-library/dom": "^10.4.1",
     "@testing-library/jest-dom": "^6.9.1",
     "@testing-library/react": "^16.3.2",
     "@types/react": "^19.2.14",
     "@vitest/browser": "^4.0.18",
-    "esbuild": "^0.27.3",
-    "eslint": "^9.39.2",
+    "esbuild": "^0.27.4",
+    "eslint": "^10.0.3",
     "eslint-plugin-import": "^2.32.0",
     "eslint-plugin-react": "^7.37.5",
     "eslint-plugin-react-hooks": "^7.0.1",
-    "globals": "^17.3.0",
-    "happy-dom": "^20.6.1",
+    "globals": "^17.4.0",
+    "happy-dom": "^20.8.4",
     "husky": "^9.1.7",
-    "jsdom": "^28.0.0",
+    "jsdom": "^29.0.0",
     "prettier": "^3.8.1",
     "react": "^19.2.4",
     "react-dom": "^19.2.4",
     "typescript": "^5.9.3",
-    "typescript-eslint": "^8.55.0",
+    "typescript-eslint": "^8.57.1",
     "vitest": "^4.0.18"
   },
   "peerDependencies": {
     "react": ">=18"
   },
+  "resolutions": {
+    "minimatch": ">=3.1.3",
+    "rollup": ">=4.59.0",
+    "undici": ">=7.24.0"
+  },
   "packageManager": "yarn@1.22.22+sha512.a6b2f7906b721bba3d67d4aff083df04dad64c399707841b7acf00f6b133b7ac24255f2652fa22ae3534329dc6180534e98d17432037ff6fd140556e2bb3137e"
 }