@schematichq/schematic-js 1.4.0 → 1.6.0

package/dist/schematic.cjs.js

@@ -6,7 +6,11 @@ var __getOwnPropNames = Object.getOwnPropertyNames;
 var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
 var __commonJS = (cb, mod) => function __require() {
-  return mod || (0, cb[__getOwnPropNames(cb)[0]])((mod = { exports: {} }).exports, mod), mod.exports;
+  try {
+    return mod || (0, cb[__getOwnPropNames(cb)[0]])((mod = { exports: {} }).exports, mod), mod.exports;
+  } catch (e) {
+    throw mod = 0, e;
+  }
 };
 var __export = (target, all) => {
   for (var name in all)
@@ -30,9 +34,9 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 ));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
-// node_modules/cross-fetch/dist/browser-polyfill.js
+// node_modules/.pnpm/cross-fetch@4.1.0/node_modules/cross-fetch/dist/browser-polyfill.js
 var require_browser_polyfill = __commonJS({
-  "node_modules/cross-fetch/dist/browser-polyfill.js"(exports) {
+  "node_modules/.pnpm/cross-fetch@4.1.0/node_modules/cross-fetch/dist/browser-polyfill.js"(exports) {
     (function(self2) {
       var irrelevant = (function(exports2) {
         var g = typeof globalThis !== "undefined" && globalThis || typeof self2 !== "undefined" && self2 || // eslint-disable-next-line no-undef
@@ -573,16 +577,29 @@ __export(index_exports, {
   CheckFlagReturnFromJSON: () => CheckFlagReturnFromJSON,
   CheckFlagsResponseFromJSON: () => CheckFlagsResponseFromJSON,
   CheckPlanReturnFromJSON: () => CheckPlanReturnFromJSON,
+  CreditBalancesFromJSON: () => CreditBalancesFromJSON,
+  DEFAULT_INVOICE_QUERY: () => DEFAULT_INVOICE_QUERY,
   DatastreamCompanyPlanFromJSON: () => DatastreamCompanyPlanFromJSON,
   EventBodyFlagCheckToJSON: () => EventBodyFlagCheckToJSON,
+  INVOICE_MAX_PAGE_SIZE: () => INVOICE_MAX_PAGE_SIZE,
+  INVOICE_PAGE_SIZE: () => INVOICE_PAGE_SIZE,
+  InvoiceStatus: () => InvoiceStatus,
   RuleType: () => RuleType,
   Schematic: () => Schematic,
+  SchematicApiError: () => SchematicApiError,
+  SchematicBillingClient: () => SchematicBillingClient,
+  SchematicSession: () => SchematicSession,
+  TOKEN_CACHE_SIZE: () => TOKEN_CACHE_SIZE,
   TrialStatus: () => TrialStatus,
-  UsagePeriod: () => UsagePeriod
+  UsagePeriod: () => UsagePeriod,
+  billingApi: () => models_exports,
+  fetchBillingData: () => fetchBillingData,
+  normalizeInvoiceQuery: () => normalizeInvoiceQuery,
+  sessionKey: () => sessionKey
 });
 module.exports = __toCommonJS(index_exports);
 
-// node_modules/uuid/dist/stringify.js
+// node_modules/.pnpm/uuid@14.0.2/node_modules/uuid/dist/stringify.js
 var byteToHex = [];
 for (let i = 0; i < 256; ++i) {
   byteToHex.push((i + 256).toString(16).slice(1));
@@ -591,24 +608,19 @@ function unsafeStringify(arr, offset = 0) {
   return (byteToHex[arr[offset + 0]] + byteToHex[arr[offset + 1]] + byteToHex[arr[offset + 2]] + byteToHex[arr[offset + 3]] + "-" + byteToHex[arr[offset + 4]] + byteToHex[arr[offset + 5]] + "-" + byteToHex[arr[offset + 6]] + byteToHex[arr[offset + 7]] + "-" + byteToHex[arr[offset + 8]] + byteToHex[arr[offset + 9]] + "-" + byteToHex[arr[offset + 10]] + byteToHex[arr[offset + 11]] + byteToHex[arr[offset + 12]] + byteToHex[arr[offset + 13]] + byteToHex[arr[offset + 14]] + byteToHex[arr[offset + 15]]).toLowerCase();
 }
 
-// node_modules/uuid/dist/rng.js
-var getRandomValues;
+// node_modules/.pnpm/uuid@14.0.2/node_modules/uuid/dist/rng.js
 var rnds8 = new Uint8Array(16);
 function rng() {
-  if (!getRandomValues) {
-    if (typeof crypto === "undefined" || !crypto.getRandomValues) {
-      throw new Error("crypto.getRandomValues() not supported. See https://github.com/uuidjs/uuid#getrandomvalues-not-supported");
-    }
-    getRandomValues = crypto.getRandomValues.bind(crypto);
-  }
-  return getRandomValues(rnds8);
+  return crypto.getRandomValues(rnds8);
 }
 
-// node_modules/uuid/dist/native.js
-var randomUUID = typeof crypto !== "undefined" && crypto.randomUUID && crypto.randomUUID.bind(crypto);
-var native_default = { randomUUID };
-
-// node_modules/uuid/dist/v4.js
+// node_modules/.pnpm/uuid@14.0.2/node_modules/uuid/dist/v4.js
+function v4(options, buf, offset) {
+  if (!buf && !options && crypto.randomUUID) {
+    return crypto.randomUUID();
+  }
+  return _v4(options, buf, offset);
+}
 function _v4(options, buf, offset) {
   options = options || {};
   const rnds = options.random ?? options.rng?.() ?? rng();
@@ -629,12 +641,6 @@ function _v4(options, buf, offset) {
   }
   return unsafeStringify(rnds);
 }
-function v4(options, buf, offset) {
-  if (native_default.randomUUID && !buf && !options) {
-    return native_default.randomUUID();
-  }
-  return _v4(options, buf, offset);
-}
 var v4_default = v4;
 
 // src/index.ts
@@ -648,6 +654,36 @@ function EntitlementValueTypeFromJSONTyped(json, ignoreDiscriminator) {
   return json;
 }
 
+// src/types/api/models/MetricPeriodMonthReset.ts
+function MetricPeriodMonthResetFromJSON(json) {
+  return MetricPeriodMonthResetFromJSONTyped(json, false);
+}
+function MetricPeriodMonthResetFromJSONTyped(json, ignoreDiscriminator) {
+  return json;
+}
+
+// src/types/api/models/WarningTier.ts
+function WarningTierFromJSON(json) {
+  return WarningTierFromJSONTyped(json, false);
+}
+function WarningTierFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    key: json["key"],
+    value: json["value"]
+  };
+}
+
+// src/types/api/models/MetricPeriod.ts
+function MetricPeriodFromJSON(json) {
+  return MetricPeriodFromJSONTyped(json, false);
+}
+function MetricPeriodFromJSONTyped(json, ignoreDiscriminator) {
+  return json;
+}
+
 // src/types/api/models/FeatureEntitlement.ts
 function FeatureEntitlementFromJSON(json) {
   return FeatureEntitlementFromJSONTyped(json, false);
@@ -658,22 +694,35 @@ function FeatureEntitlementFromJSONTyped(json, ignoreDiscriminator) {
   }
   return {
     allocation: json["allocation"] == null ? void 0 : json["allocation"],
+    consumptionRate: json["consumption_rate"] == null ? void 0 : json["consumption_rate"],
     creditId: json["credit_id"] == null ? void 0 : json["credit_id"],
     creditRemaining: json["credit_remaining"] == null ? void 0 : json["credit_remaining"],
+    creditReserved: json["credit_reserved"] == null ? void 0 : json["credit_reserved"],
+    creditSettled: json["credit_settled"] == null ? void 0 : json["credit_settled"],
     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"],
+    eventSubtype: json["event_subtype"] == null ? void 0 : json["event_subtype"],
     featureId: json["feature_id"],
     featureKey: json["feature_key"],
-    metricPeriod: json["metric_period"] == null ? void 0 : json["metric_period"],
+    metricPeriod: json["metric_period"] == null ? void 0 : MetricPeriodFromJSON(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"],
+    monthReset: json["month_reset"] == null ? void 0 : MetricPeriodMonthResetFromJSON(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"])
+    valueType: EntitlementValueTypeFromJSON(json["value_type"]),
+    warningTiers: json["warning_tiers"] == null ? void 0 : json["warning_tiers"].map(WarningTierFromJSON)
   };
 }
 
+// src/types/api/models/RuleType.ts
+function RuleTypeFromJSON(json) {
+  return RuleTypeFromJSONTyped(json, false);
+}
+function RuleTypeFromJSONTyped(json, ignoreDiscriminator) {
+  return json;
+}
+
 // src/types/api/models/CheckFlagResponseData.ts
 function CheckFlagResponseDataFromJSON(json) {
   return CheckFlagResponseDataFromJSONTyped(json, false);
@@ -689,13 +738,13 @@ function CheckFlagResponseDataFromJSONTyped(json, ignoreDiscriminator) {
     featureAllocation: json["feature_allocation"] == null ? void 0 : json["feature_allocation"],
     featureUsage: json["feature_usage"] == null ? void 0 : json["feature_usage"],
     featureUsageEvent: json["feature_usage_event"] == null ? void 0 : json["feature_usage_event"],
-    featureUsagePeriod: json["feature_usage_period"] == null ? void 0 : json["feature_usage_period"],
+    featureUsagePeriod: json["feature_usage_period"] == null ? void 0 : MetricPeriodFromJSON(json["feature_usage_period"]),
     featureUsageResetAt: json["feature_usage_reset_at"] == null ? void 0 : new Date(json["feature_usage_reset_at"]),
     flag: json["flag"],
     flagId: json["flag_id"] == null ? void 0 : json["flag_id"],
     reason: json["reason"],
     ruleId: json["rule_id"] == null ? void 0 : json["rule_id"],
-    ruleType: json["rule_type"] == null ? void 0 : json["rule_type"],
+    ruleType: json["rule_type"] == null ? void 0 : RuleTypeFromJSON(json["rule_type"]),
     userId: json["user_id"] == null ? void 0 : json["user_id"],
     value: json["value"]
   };
@@ -715,6 +764,272 @@ function CheckFlagResponseFromJSONTyped(json, ignoreDiscriminator) {
   };
 }
 
+// src/types/api/runtime.ts
+var BASE_PATH = "https://api.schematichq.com".replace(/\/+$/, "");
+var Configuration = class {
+  constructor(configuration = {}) {
+    this.configuration = configuration;
+  }
+  configuration;
+  set config(configuration) {
+    this.configuration = configuration;
+  }
+  get basePath() {
+    return this.configuration.basePath != null ? this.configuration.basePath : BASE_PATH;
+  }
+  get fetchApi() {
+    return this.configuration.fetchApi;
+  }
+  get middleware() {
+    return this.configuration.middleware || [];
+  }
+  get queryParamsStringify() {
+    return this.configuration.queryParamsStringify || querystring;
+  }
+  get username() {
+    return this.configuration.username;
+  }
+  get password() {
+    return this.configuration.password;
+  }
+  get apiKey() {
+    const apiKey = this.configuration.apiKey;
+    if (apiKey) {
+      return typeof apiKey === "function" ? apiKey : () => apiKey;
+    }
+    return void 0;
+  }
+  get accessToken() {
+    const accessToken = this.configuration.accessToken;
+    if (accessToken) {
+      return typeof accessToken === "function" ? accessToken : async () => accessToken;
+    }
+    return void 0;
+  }
+  get headers() {
+    return this.configuration.headers;
+  }
+  get credentials() {
+    return this.configuration.credentials;
+  }
+};
+var DefaultConfig = new Configuration();
+var BaseAPI = class _BaseAPI {
+  constructor(configuration = DefaultConfig) {
+    this.configuration = configuration;
+    this.middleware = configuration.middleware;
+  }
+  configuration;
+  static jsonRegex = new RegExp(
+    "^(:?application/json|[^;/ 	]+/[^;/ 	]+[+]json)[ 	]*(:?;.*)?$",
+    "i"
+  );
+  middleware;
+  withMiddleware(...middlewares) {
+    const next = this.clone();
+    next.middleware = next.middleware.concat(...middlewares);
+    return next;
+  }
+  withPreMiddleware(...preMiddlewares) {
+    const middlewares = preMiddlewares.map((pre) => ({ pre }));
+    return this.withMiddleware(...middlewares);
+  }
+  withPostMiddleware(...postMiddlewares) {
+    const middlewares = postMiddlewares.map((post) => ({ post }));
+    return this.withMiddleware(...middlewares);
+  }
+  /**
+   * Check if the given MIME is a JSON MIME.
+   * JSON MIME examples:
+   *   application/json
+   *   application/json; charset=UTF8
+   *   APPLICATION/JSON
+   *   application/vnd.company+json
+   * @param mime - MIME (Multipurpose Internet Mail Extensions)
+   * @return True if the given MIME is JSON, false otherwise.
+   */
+  isJsonMime(mime) {
+    if (!mime) {
+      return false;
+    }
+    return _BaseAPI.jsonRegex.test(mime);
+  }
+  async request(context, initOverrides) {
+    const { url, init } = await this.createFetchParams(context, initOverrides);
+    const response = await this.fetchApi(url, init);
+    if (response && response.status >= 200 && response.status < 300) {
+      return response;
+    }
+    throw new ResponseError(response, "Response returned an error code");
+  }
+  async createFetchParams(context, initOverrides) {
+    let url = this.configuration.basePath + context.path;
+    if (context.query !== void 0 && Object.keys(context.query).length !== 0) {
+      url += "?" + this.configuration.queryParamsStringify(context.query);
+    }
+    const headers = Object.assign(
+      {},
+      this.configuration.headers,
+      context.headers
+    );
+    Object.keys(headers).forEach(
+      (key) => headers[key] === void 0 ? delete headers[key] : {}
+    );
+    const initOverrideFn = typeof initOverrides === "function" ? initOverrides : async () => initOverrides;
+    const initParams = {
+      method: context.method,
+      headers,
+      body: context.body,
+      credentials: this.configuration.credentials
+    };
+    const overriddenInit = {
+      ...initParams,
+      ...await initOverrideFn({
+        init: initParams,
+        context
+      })
+    };
+    let body;
+    if (isFormData(overriddenInit.body) || overriddenInit.body instanceof URLSearchParams || isBlob(overriddenInit.body)) {
+      body = overriddenInit.body;
+    } else if (this.isJsonMime(headers["Content-Type"])) {
+      body = JSON.stringify(overriddenInit.body);
+    } else {
+      body = overriddenInit.body;
+    }
+    const init = {
+      ...overriddenInit,
+      body
+    };
+    return { url, init };
+  }
+  fetchApi = async (url, init) => {
+    let fetchParams = { url, init };
+    for (const middleware of this.middleware) {
+      if (middleware.pre) {
+        fetchParams = await middleware.pre({
+          fetch: this.fetchApi,
+          ...fetchParams
+        }) || fetchParams;
+      }
+    }
+    let response = void 0;
+    try {
+      response = await (this.configuration.fetchApi || fetch)(
+        fetchParams.url,
+        fetchParams.init
+      );
+    } catch (e) {
+      for (const middleware of this.middleware) {
+        if (middleware.onError) {
+          response = await middleware.onError({
+            fetch: this.fetchApi,
+            url: fetchParams.url,
+            init: fetchParams.init,
+            error: e,
+            response: response ? response.clone() : void 0
+          }) || response;
+        }
+      }
+      if (response === void 0) {
+        if (e instanceof Error) {
+          throw new FetchError(
+            e,
+            "The request failed and the interceptors did not return an alternative response"
+          );
+        } else {
+          throw e;
+        }
+      }
+    }
+    for (const middleware of this.middleware) {
+      if (middleware.post) {
+        response = await middleware.post({
+          fetch: this.fetchApi,
+          url: fetchParams.url,
+          init: fetchParams.init,
+          response: response.clone()
+        }) || response;
+      }
+    }
+    return response;
+  };
+  /**
+   * Create a shallow clone of `this` by constructing a new instance
+   * and then shallow cloning data members.
+   */
+  clone() {
+    const constructor = this.constructor;
+    const next = new constructor(this.configuration);
+    next.middleware = this.middleware.slice();
+    return next;
+  }
+};
+function isBlob(value) {
+  return typeof Blob !== "undefined" && value instanceof Blob;
+}
+function isFormData(value) {
+  return typeof FormData !== "undefined" && value instanceof FormData;
+}
+var ResponseError = class extends Error {
+  constructor(response, msg) {
+    super(msg);
+    this.response = response;
+  }
+  response;
+  name = "ResponseError";
+};
+var FetchError = class extends Error {
+  constructor(cause, msg) {
+    super(msg);
+    this.cause = cause;
+  }
+  cause;
+  name = "FetchError";
+};
+function querystring(params, prefix = "") {
+  return Object.keys(params).map((key) => querystringSingleKey(key, params[key], prefix)).filter((part) => part.length > 0).join("&");
+}
+function querystringSingleKey(key, value, keyPrefix = "") {
+  const fullKey = keyPrefix + (keyPrefix.length ? `[${key}]` : key);
+  if (value instanceof Array) {
+    const multiValue = value.map((singleValue) => encodeURIComponent(String(singleValue))).join(`&${encodeURIComponent(fullKey)}=`);
+    return `${encodeURIComponent(fullKey)}=${multiValue}`;
+  }
+  if (value instanceof Set) {
+    const valueAsArray = Array.from(value);
+    return querystringSingleKey(key, valueAsArray, keyPrefix);
+  }
+  if (value instanceof Date) {
+    return `${encodeURIComponent(fullKey)}=${encodeURIComponent(value.toISOString())}`;
+  }
+  if (value instanceof Object) {
+    return querystring(value, fullKey);
+  }
+  return `${encodeURIComponent(fullKey)}=${encodeURIComponent(String(value))}`;
+}
+function mapValues(data, fn) {
+  return Object.keys(data).reduce(
+    (acc, key) => ({ ...acc, [key]: fn(data[key]) }),
+    {}
+  );
+}
+
+// src/types/api/models/CompanyCreditBalance.ts
+function CompanyCreditBalanceFromJSON(json) {
+  return CompanyCreditBalanceFromJSONTyped(json, false);
+}
+function CompanyCreditBalanceFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    remaining: json["remaining"],
+    reserved: json["reserved"],
+    settled: json["settled"]
+  };
+}
+
 // src/types/api/models/TrialStatus.ts
 var TrialStatus = {
   Active: "active",
@@ -753,6 +1068,7 @@ function CheckFlagsResponseDataFromJSONTyped(json, ignoreDiscriminator) {
     return json;
   }
   return {
+    creditBalances: json["credit_balances"] == null ? void 0 : mapValues(json["credit_balances"], CompanyCreditBalanceFromJSON),
     flags: json["flags"].map(CheckFlagResponseDataFromJSON),
     plan: json["plan"] == null ? void 0 : DatastreamCompanyPlanFromJSON(json["plan"])
   };
@@ -785,6 +1101,7 @@ function EventBodyFlagCheckToJSONTyped(value, ignoreDiscriminator = false) {
     error: value["error"],
     flag_id: value["flagId"],
     flag_key: value["flagKey"],
+    preflight: value["preflight"],
     reason: value["reason"],
     req_company: value["reqCompany"],
     req_user: value["reqUser"],
@@ -841,7 +1158,10 @@ var CheckFlagReturnFromJSON = (json) => {
   return {
     featureUsageExceeded,
     companyId: companyId == null ? void 0 : companyId,
+    creditId: entitlement?.creditId == null ? void 0 : entitlement.creditId,
     creditRemaining: entitlement?.creditRemaining == null ? void 0 : entitlement.creditRemaining,
+    creditReserved: entitlement?.creditReserved == null ? void 0 : entitlement.creditReserved,
+    creditSettled: entitlement?.creditSettled == null ? void 0 : entitlement.creditSettled,
     error: error == null ? void 0 : error,
     featureAllocation: resolvedAllocation == null ? void 0 : resolvedAllocation,
     featureUsage: resolvedUsage == null ? void 0 : resolvedUsage,
@@ -867,6 +1187,24 @@ var CheckPlanReturnFromJSON = (json) => {
     trialStatus: trialStatus == null ? void 0 : trialStatus
   };
 };
+var CreditBalancesFromJSON = (json) => {
+  if (json == null || typeof json !== "object") {
+    return {};
+  }
+  const balances = {};
+  for (const [creditId, raw] of Object.entries(json)) {
+    if (raw == null || typeof raw !== "object") continue;
+    const { remaining, reserved, settled } = CompanyCreditBalanceFromJSON(raw);
+    if (typeof remaining !== "number" || typeof reserved !== "number" || typeof settled !== "number") {
+      continue;
+    }
+    balances[creditId] = { remaining, reserved, settled };
+  }
+  return balances;
+};
+
+// src/cacheVersion.ts
+var cacheVersion = "bb56e75d";
 
 // src/utils.ts
 function contextString(context) {
@@ -885,10 +1223,597 @@ function contextString(context) {
 }
 
 // src/version.ts
-var version = "1.4.0";
+var version = "1.6.0";
+
+// src/billing/api/generated/models/index.ts
+var models_exports = {};
+__export(models_exports, {
+  ApiErrorFromJSON: () => ApiErrorFromJSON,
+  ApiErrorFromJSONTyped: () => ApiErrorFromJSONTyped,
+  ApiErrorToJSON: () => ApiErrorToJSON,
+  ApiErrorToJSONTyped: () => ApiErrorToJSONTyped,
+  CompanyInvoiceResponseDataFromJSON: () => CompanyInvoiceResponseDataFromJSON,
+  CompanyInvoiceResponseDataFromJSONTyped: () => CompanyInvoiceResponseDataFromJSONTyped,
+  CompanyInvoiceResponseDataToJSON: () => CompanyInvoiceResponseDataToJSON,
+  CompanyInvoiceResponseDataToJSONTyped: () => CompanyInvoiceResponseDataToJSONTyped,
+  CompanyInvoicesResponseDataFromJSON: () => CompanyInvoicesResponseDataFromJSON,
+  CompanyInvoicesResponseDataFromJSONTyped: () => CompanyInvoicesResponseDataFromJSONTyped,
+  CompanyInvoicesResponseDataToJSON: () => CompanyInvoicesResponseDataToJSON,
+  CompanyInvoicesResponseDataToJSONTyped: () => CompanyInvoicesResponseDataToJSONTyped,
+  GetCompanyInvoicesParamsFromJSON: () => GetCompanyInvoicesParamsFromJSON,
+  GetCompanyInvoicesParamsFromJSONTyped: () => GetCompanyInvoicesParamsFromJSONTyped,
+  GetCompanyInvoicesParamsToJSON: () => GetCompanyInvoicesParamsToJSON,
+  GetCompanyInvoicesParamsToJSONTyped: () => GetCompanyInvoicesParamsToJSONTyped,
+  GetCompanyInvoicesResponseFromJSON: () => GetCompanyInvoicesResponseFromJSON,
+  GetCompanyInvoicesResponseFromJSONTyped: () => GetCompanyInvoicesResponseFromJSONTyped,
+  GetCompanyInvoicesResponseToJSON: () => GetCompanyInvoicesResponseToJSON,
+  GetCompanyInvoicesResponseToJSONTyped: () => GetCompanyInvoicesResponseToJSONTyped,
+  InvoiceStatus: () => InvoiceStatus,
+  InvoiceStatusFromJSON: () => InvoiceStatusFromJSON,
+  InvoiceStatusFromJSONTyped: () => InvoiceStatusFromJSONTyped,
+  InvoiceStatusToJSON: () => InvoiceStatusToJSON,
+  InvoiceStatusToJSONTyped: () => InvoiceStatusToJSONTyped,
+  instanceOfApiError: () => instanceOfApiError,
+  instanceOfCompanyInvoiceResponseData: () => instanceOfCompanyInvoiceResponseData,
+  instanceOfCompanyInvoicesResponseData: () => instanceOfCompanyInvoicesResponseData,
+  instanceOfGetCompanyInvoicesParams: () => instanceOfGetCompanyInvoicesParams,
+  instanceOfGetCompanyInvoicesResponse: () => instanceOfGetCompanyInvoicesResponse,
+  instanceOfInvoiceStatus: () => instanceOfInvoiceStatus
+});
+
+// src/billing/api/generated/models/ApiError.ts
+function instanceOfApiError(value) {
+  if (!("error" in value) || value["error"] === void 0) return false;
+  return true;
+}
+function ApiErrorFromJSON(json) {
+  return ApiErrorFromJSONTyped(json, false);
+}
+function ApiErrorFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    error: json["error"]
+  };
+}
+function ApiErrorToJSON(json) {
+  return ApiErrorToJSONTyped(json, false);
+}
+function ApiErrorToJSONTyped(value, ignoreDiscriminator = false) {
+  if (value == null) {
+    return value;
+  }
+  return {
+    error: value["error"]
+  };
+}
+
+// src/billing/api/generated/models/InvoiceStatus.ts
+var InvoiceStatus = {
+  Draft: "draft",
+  Open: "open",
+  Paid: "paid",
+  Uncollectible: "uncollectible",
+  Void: "void"
+};
+function instanceOfInvoiceStatus(value) {
+  for (const key in InvoiceStatus) {
+    if (Object.prototype.hasOwnProperty.call(InvoiceStatus, key)) {
+      if (InvoiceStatus[key] === value) {
+        return true;
+      }
+    }
+  }
+  return false;
+}
+function InvoiceStatusFromJSON(json) {
+  return InvoiceStatusFromJSONTyped(json, false);
+}
+function InvoiceStatusFromJSONTyped(json, ignoreDiscriminator) {
+  return json;
+}
+function InvoiceStatusToJSON(value) {
+  return value;
+}
+function InvoiceStatusToJSONTyped(value, ignoreDiscriminator) {
+  return value;
+}
+
+// src/billing/api/generated/models/CompanyInvoiceResponseData.ts
+function instanceOfCompanyInvoiceResponseData(value) {
+  if (!("amountDue" in value) || value["amountDue"] === void 0) return false;
+  if (!("createdAt" in value) || value["createdAt"] === void 0) return false;
+  if (!("currency" in value) || value["currency"] === void 0) return false;
+  if (!("id" in value) || value["id"] === void 0) return false;
+  return true;
+}
+function CompanyInvoiceResponseDataFromJSON(json) {
+  return CompanyInvoiceResponseDataFromJSONTyped(json, false);
+}
+function CompanyInvoiceResponseDataFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    amountDue: json["amount_due"],
+    createdAt: new Date(json["created_at"]),
+    currency: json["currency"],
+    dueDate: json["due_date"] == null ? void 0 : new Date(json["due_date"]),
+    id: json["id"],
+    status: json["status"] == null ? void 0 : InvoiceStatusFromJSON(json["status"]),
+    url: json["url"] == null ? void 0 : json["url"]
+  };
+}
+function CompanyInvoiceResponseDataToJSON(json) {
+  return CompanyInvoiceResponseDataToJSONTyped(json, false);
+}
+function CompanyInvoiceResponseDataToJSONTyped(value, ignoreDiscriminator = false) {
+  if (value == null) {
+    return value;
+  }
+  return {
+    amount_due: value["amountDue"],
+    created_at: value["createdAt"].toISOString(),
+    currency: value["currency"],
+    due_date: value["dueDate"] == null ? void 0 : value["dueDate"].toISOString(),
+    id: value["id"],
+    status: InvoiceStatusToJSON(value["status"]),
+    url: value["url"]
+  };
+}
+
+// src/billing/api/generated/models/CompanyInvoicesResponseData.ts
+function instanceOfCompanyInvoicesResponseData(value) {
+  if (!("count" in value) || value["count"] === void 0) return false;
+  if (!("invoices" in value) || value["invoices"] === void 0) return false;
+  return true;
+}
+function CompanyInvoicesResponseDataFromJSON(json) {
+  return CompanyInvoicesResponseDataFromJSONTyped(json, false);
+}
+function CompanyInvoicesResponseDataFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    count: json["count"],
+    invoices: json["invoices"].map(
+      CompanyInvoiceResponseDataFromJSON
+    )
+  };
+}
+function CompanyInvoicesResponseDataToJSON(json) {
+  return CompanyInvoicesResponseDataToJSONTyped(json, false);
+}
+function CompanyInvoicesResponseDataToJSONTyped(value, ignoreDiscriminator = false) {
+  if (value == null) {
+    return value;
+  }
+  return {
+    count: value["count"],
+    invoices: value["invoices"].map(
+      CompanyInvoiceResponseDataToJSON
+    )
+  };
+}
+
+// src/billing/api/generated/models/GetCompanyInvoicesParams.ts
+function instanceOfGetCompanyInvoicesParams(value) {
+  return true;
+}
+function GetCompanyInvoicesParamsFromJSON(json) {
+  return GetCompanyInvoicesParamsFromJSONTyped(json, false);
+}
+function GetCompanyInvoicesParamsFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    includePending: json["include_pending"] == null ? void 0 : json["include_pending"],
+    limit: json["limit"] == null ? void 0 : json["limit"],
+    offset: json["offset"] == null ? void 0 : json["offset"]
+  };
+}
+function GetCompanyInvoicesParamsToJSON(json) {
+  return GetCompanyInvoicesParamsToJSONTyped(json, false);
+}
+function GetCompanyInvoicesParamsToJSONTyped(value, ignoreDiscriminator = false) {
+  if (value == null) {
+    return value;
+  }
+  return {
+    include_pending: value["includePending"],
+    limit: value["limit"],
+    offset: value["offset"]
+  };
+}
+
+// src/billing/api/generated/models/GetCompanyInvoicesResponse.ts
+function instanceOfGetCompanyInvoicesResponse(value) {
+  if (!("data" in value) || value["data"] === void 0) return false;
+  if (!("params" in value) || value["params"] === void 0) return false;
+  return true;
+}
+function GetCompanyInvoicesResponseFromJSON(json) {
+  return GetCompanyInvoicesResponseFromJSONTyped(json, false);
+}
+function GetCompanyInvoicesResponseFromJSONTyped(json, ignoreDiscriminator) {
+  if (json == null) {
+    return json;
+  }
+  return {
+    data: CompanyInvoicesResponseDataFromJSON(json["data"]),
+    params: GetCompanyInvoicesParamsFromJSON(json["params"])
+  };
+}
+function GetCompanyInvoicesResponseToJSON(json) {
+  return GetCompanyInvoicesResponseToJSONTyped(json, false);
+}
+function GetCompanyInvoicesResponseToJSONTyped(value, ignoreDiscriminator = false) {
+  if (value == null) {
+    return value;
+  }
+  return {
+    data: CompanyInvoicesResponseDataToJSON(value["data"]),
+    params: GetCompanyInvoicesParamsToJSON(value["params"])
+  };
+}
+
+// src/billing/contract/invoices.ts
+var DEFAULT_INVOICE_QUERY = {};
+function normalizeInvoiceQuery(query) {
+  const { includePending, ...rest } = query;
+  return includePending === true ? { ...rest, includePending: true } : rest;
+}
+
+// src/billing/session.ts
+var SchematicApiError = class extends Error {
+  constructor(status, path, body) {
+    super(
+      typeof body === "object" && body !== null && "error" in body ? String(body.error) : `Request to ${path} failed with status ${status}`
+    );
+    this.status = status;
+    this.path = path;
+    this.body = body;
+    this.name = "SchematicApiError";
+  }
+  status;
+  path;
+  body;
+};
+var DEFAULT_API_URL = "https://api.schematichq.com";
+var TOKEN_CACHE_SIZE = 10;
+function sessionKey(session) {
+  return JSON.stringify([session.company, session.user ?? null]);
+}
+var SchematicSession = class {
+  _apiUrl;
+  _headers;
+  _fetch;
+  _session;
+  _tokens = /* @__PURE__ */ new Map();
+  _resolving = /* @__PURE__ */ new Map();
+  /**
+   * Bumped when the session's identity changes, and only then — the object
+   * itself is replaced on every render. Anything asking "is this still the
+   * session I started under?" counts generations rather than comparing.
+   */
+  _generation = 0;
+  _listeners = /* @__PURE__ */ new Set();
+  constructor(options = {}) {
+    this._session = options.session;
+    this._apiUrl = (options.apiUrl ?? DEFAULT_API_URL).replace(/\/$/, "");
+    this._headers = options.additionalHeaders ?? {};
+    this._fetch = options.fetch ?? ((input, init) => globalThis.fetch(input, init));
+  }
+  get key() {
+    return this._session === null || this._session === void 0 ? void 0 : sessionKey(this._session);
+  }
+  get status() {
+    if (this._session === void 0) {
+      return "pending";
+    }
+    return this._session === null ? "ended" : "active";
+  }
+  /** Only a statement that changes which pair is read counts as a change. */
+  set(session) {
+    if (session === void 0) {
+      return;
+    }
+    const previous = this._session;
+    this._session = session;
+    if (session === null) {
+      if (previous === null) {
+        return;
+      }
+      this._tokens.clear();
+      this._resolving.clear();
+      this._generation += 1;
+      this._emit({ type: "ended" });
+      return;
+    }
+    if (previous !== null && previous !== void 0) {
+      if (previous.company === session.company && (previous.user ?? null) === (session.user ?? null)) {
+        return;
+      }
+      this._generation += 1;
+      this._emit({ type: "changed" });
+      return;
+    }
+    this._generation += 1;
+    this._emit({ type: "started" });
+  }
+  onChange(listener) {
+    this._listeners.add(listener);
+    return () => {
+      this._listeners.delete(listener);
+    };
+  }
+  _emit(event) {
+    for (const listener of this._listeners) {
+      listener(event);
+    }
+  }
+  _cached(key) {
+    const cached = this._tokens.get(key);
+    if (cached === void 0) {
+      return void 0;
+    }
+    if (cached.expiresAt !== void 0 && cached.expiresAt.getTime() <= Date.now()) {
+      this._tokens.delete(key);
+      return void 0;
+    }
+    this._tokens.delete(key);
+    this._tokens.set(key, cached);
+    return cached;
+  }
+  _cache(key, resolved) {
+    this._tokens.delete(key);
+    this._tokens.set(key, resolved);
+    while (this._tokens.size > TOKEN_CACHE_SIZE) {
+      const oldest = this._tokens.keys().next();
+      if (oldest.done === true) {
+        break;
+      }
+      this._tokens.delete(oldest.value);
+    }
+  }
+  /**
+   * The credential for a session, minted if this one is not held. `force`
+   * retires what is held first: the caller is a request the API has just
+   * rejected.
+   */
+  resolveToken(session, force) {
+    const { token } = session;
+    if (typeof token === "string") {
+      return Promise.resolve(token);
+    }
+    const key = sessionKey(session);
+    if (!force) {
+      const cached = this._cached(key);
+      if (cached !== void 0) {
+        return Promise.resolve(cached.token);
+      }
+    } else {
+      this._tokens.delete(key);
+    }
+    const inflight = this._resolving.get(key);
+    if (inflight !== void 0 && inflight.sessionGeneration === this._generation && (!force || inflight.forced)) {
+      return inflight.promise;
+    }
+    const provider = token;
+    const sessionGeneration = this._generation;
+    const owns = () => this._resolving.get(key)?.promise === promise;
+    const promise = Promise.resolve().then(() => provider()).then((result) => {
+      const { resolved, stamped } = asResolvedToken(result, session);
+      if (owns() && (this._generation === sessionGeneration || stamped)) {
+        this._cache(key, resolved);
+      }
+      return resolved.token;
+    }).finally(() => {
+      if (owns()) {
+        this._resolving.delete(key);
+      }
+    });
+    this._resolving.set(key, { promise, forced: force, sessionGeneration });
+    return promise;
+  }
+  /**
+   * Sends a request under the active session's credential, refreshing once
+   * and retrying if the API rejects it. Rejects rather than answering when
+   * the session changes while the request is in flight: the rows would
+   * otherwise come back as the answer to a question asked for another pair.
+   */
+  async request(path, options = {}) {
+    const { method = "GET", body } = options;
+    const send = async (credential2) => this._fetch(`${this._apiUrl}${path}`, {
+      method,
+      headers: {
+        "Accept": "application/json",
+        ...body === void 0 ? {} : { "Content-Type": "application/json" },
+        ...this._headers,
+        "X-Schematic-Api-Key": credential2
+      },
+      ...body === void 0 ? {} : { body: JSON.stringify(body) }
+    });
+    const session = this._session;
+    if (session === null) {
+      throw new Error("There is no session to read data for.");
+    }
+    if (session === void 0) {
+      throw new Error("A session is required to read data.");
+    }
+    const provider = session.token;
+    const generation = this._generation;
+    const stillOurs = async (response2) => {
+      if (this._generation === generation) {
+        return;
+      }
+      if (response2 !== void 0) {
+        await discardBody(response2);
+      }
+      throw new Error("The session changed while this request was in flight.");
+    };
+    const credential = await this.resolveToken(session, false);
+    if (this._generation !== generation) {
+      throw new Error("The session changed before this request was sent.");
+    }
+    let response = await send(credential);
+    await stillOurs(response);
+    if (response.status === 401 && typeof provider === "function" && this._generation === generation) {
+      const status = response.status;
+      const rejected = await readBody(response);
+      let retried = false;
+      const current = this._session;
+      if (this._generation === generation && current !== null && current !== void 0) {
+        const refreshed = await this.resolveToken(current, true);
+        if (this._generation === generation) {
+          response = await send(refreshed);
+          await stillOurs(response);
+          retried = true;
+        }
+      }
+      if (!retried) {
+        throw new SchematicApiError(status, path, rejected);
+      }
+    }
+    const parsed = await readBody(response);
+    await stillOurs();
+    if (!response.ok) {
+      throw new SchematicApiError(response.status, path, parsed);
+    }
+    return parsed;
+  }
+};
+function asResolvedToken(result, session) {
+  if (typeof result === "string") {
+    return { resolved: { token: result }, stamped: false };
+  }
+  const token = result?.token;
+  if (typeof token !== "string" || token === "") {
+    throw new Error("The access token provider did not return a token.");
+  }
+  if (result.company !== void 0 && result.company !== session.company || result.user !== void 0 && session.user !== void 0 && result.user !== session.user) {
+    throw new Error(
+      "The access token provider returned a token for a different session."
+    );
+  }
+  const stamped = result.company !== void 0 && (session.user === void 0 || result.user !== void 0);
+  return { resolved: { token, expiresAt: asDate(result.expiresAt) }, stamped };
+}
+function asDate(value) {
+  if (value === void 0 || value === null) {
+    return void 0;
+  }
+  const date = value instanceof Date ? value : new Date(value);
+  return Number.isNaN(date.getTime()) ? void 0 : date;
+}
+async function discardBody(response) {
+  try {
+    await response.body?.cancel();
+  } catch {
+  }
+}
+async function readBody(response) {
+  const text = await response.text();
+  if (text === "") {
+    return null;
+  }
+  try {
+    return JSON.parse(text);
+  } catch {
+    return text;
+  }
+}
+
+// src/billing/client.ts
+var INVOICE_PAGE_SIZE = 12;
+var INVOICE_MAX_PAGE_SIZE = 250;
+var SchematicBillingClient = class {
+  /**
+   * Public so a second client over the same surface can be constructed on
+   * this one rather than minting tokens of its own.
+   */
+  session;
+  constructor(options = {}) {
+    this.session = options instanceof SchematicSession ? options : new SchematicSession(options);
+  }
+  get sessionKey() {
+    return this.session.key;
+  }
+  get sessionStatus() {
+    return this.session.status;
+  }
+  setSession(session) {
+    this.session.set(session);
+  }
+  onSessionChange(listener) {
+    return this.session.onChange(listener);
+  }
+  fetchInvoices(params) {
+    const query = new URLSearchParams({
+      limit: String(Math.min(params.limit, INVOICE_MAX_PAGE_SIZE)),
+      offset: String(params.offset)
+    });
+    if (params.includePending !== void 0) {
+      query.set("include_pending", String(params.includePending));
+    }
+    const path = `/company/invoices?${query}`;
+    return this.session.request(path).then((body) => {
+      if (body === null || typeof body !== "object" || !("data" in body)) {
+        throw new Error(`Malformed response from ${path}`);
+      }
+      const data = body.data;
+      if (data == null || data.invoices == null) {
+        return { invoices: [], count: 0 };
+      }
+      const decoded = GetCompanyInvoicesResponseFromJSON(body).data;
+      return { invoices: decoded.invoices, count: decoded.count };
+    });
+  }
+};
+async function fetchBillingData(client, options = {}) {
+  const wanted = options.names ?? ["invoices"];
+  const data = {};
+  const invoices = options.invoices === void 0 ? void 0 : normalizeInvoiceQuery(options.invoices);
+  if (invoices !== void 0) {
+    data.params = { invoices };
+  }
+  data.sessionKey = client.sessionKey;
+  await Promise.all(
+    wanted.map(async (name) => {
+      try {
+        switch (name) {
+          case "invoices": {
+            const page = await client.fetchInvoices({
+              ...invoices ?? {},
+              limit: INVOICE_PAGE_SIZE,
+              offset: 0
+            });
+            data.invoices = {
+              invoices: page.invoices,
+              count: page.count,
+              // Two queries on the server: a count that outran the rows
+              // would seed a page that pages nothing.
+              hasMore: page.invoices.length > 0 && page.invoices.length < page.count
+            };
+            break;
+          }
+        }
+      } catch {
+      }
+    })
+  );
+  return data;
+}
 
 // src/index.ts
 var anonymousIdKey = "schematicId";
+var flagStateCachePrefix = "schematicCache";
+var flagStateCacheMaxContexts = 10;
+var flagStateCacheDefaultMaxAgeMs = 7 * 24 * 60 * 60 * 1e3;
+var emptyCreditBalances = Object.freeze({});
 var Schematic = class {
   additionalHeaders = {};
   apiKey;
@@ -905,11 +1830,17 @@ var Schematic = class {
   isPending = true;
   isPendingListeners = /* @__PURE__ */ new Set();
   planListeners = /* @__PURE__ */ new Set();
+  creditBalanceListeners = /* @__PURE__ */ new Set();
   storage;
+  persistFlagState = true;
+  flagStateCacheKey;
+  flagStateCacheMaxAgeMs = flagStateCacheDefaultMaxAgeMs;
+  cachedFlagState = null;
   useWebSocket = false;
   checks = {};
   featureUsageEventMap = {};
   planChecks = {};
+  creditBalances = {};
   webSocketUrl = "wss://api.schematichq.com";
   webSocketConnectionTimeout = 1e4;
   webSocketReconnect = true;
@@ -938,11 +1869,16 @@ var Schematic = class {
   fallbackCheckCache = {};
   constructor(apiKey, options) {
     this.apiKey = apiKey;
+    this.flagStateCacheKey = `${flagStateCachePrefix}:${apiKey}`;
     this.eventQueue = [];
     this.contextDependentEventQueue = [];
     this.useWebSocket = options?.useWebSocket ?? false;
     this.debugEnabled = options?.debug ?? false;
     this.offlineEnabled = options?.offline ?? false;
+    this.persistFlagState = options?.persistFlagState ?? true;
+    if (typeof options?.flagStateCacheMaxAgeMs === "number" && options.flagStateCacheMaxAgeMs > 0) {
+      this.flagStateCacheMaxAgeMs = options.flagStateCacheMaxAgeMs;
+    }
     if (typeof window !== "undefined" && typeof window.location !== "undefined") {
       const params = new URLSearchParams(window.location.search);
       const debugParam = params.get("schematic_debug");
@@ -975,6 +1911,7 @@ var Schematic = class {
       } catch {
       }
     }
+    this.hydrateFlagStateFromCache();
     if (options?.apiUrl !== void 0) {
       this.apiUrl = options.apiUrl;
     }
@@ -1268,6 +2205,29 @@ var Schematic = class {
         });
         this.notifyPlanListeners(plan);
       }
+      if (message.credit_balances !== void 0 && message.credit_balances !== null) {
+        const contextStr = contextString(context);
+        const updates = CreditBalancesFromJSON(message.credit_balances);
+        const previous = this.creditBalances[contextStr] ?? {};
+        let changed = false;
+        const merged = { ...previous };
+        for (const [creditId, next] of Object.entries(updates)) {
+          const current = previous[creditId];
+          if (current === void 0 || current.remaining !== next.remaining || current.reserved !== next.reserved || current.settled !== next.settled) {
+            merged[creditId] = next;
+            changed = true;
+          }
+        }
+        if (changed) {
+          this.creditBalances[contextStr] = merged;
+          this.debug(
+            `WebSocket credit balance update received. Notifying listeners`,
+            { creditBalances: merged }
+          );
+          this.notifyCreditBalanceListeners(merged);
+        }
+      }
+      this.persistContextToCache(contextString(context));
       this.flushContextDependentEventQueue();
       this.setIsPending(false);
     };
@@ -1375,7 +2335,32 @@ var Schematic = class {
       return;
     }
     try {
-      this.setIsPending(true);
+      const newContextStr = contextString(context);
+      const cacheHit = this.hasCachedValuesForContext(newContextStr);
+      if (cacheHit) {
+        this.debug(
+          `setContext: cache hit for context, pre-resolving from cached values`
+        );
+        this.context = context;
+        this.flushContextDependentEventQueue();
+        this.setIsPending(false);
+        const cachedChecks = this.checks[newContextStr] ?? {};
+        for (const [flagKey, check] of Object.entries(cachedChecks)) {
+          if (check === void 0) continue;
+          this.notifyFlagCheckListeners(flagKey, check);
+          this.notifyFlagValueListeners(flagKey, check.value);
+        }
+        const cachedPlan = this.planChecks[newContextStr];
+        if (cachedPlan !== void 0) {
+          this.notifyPlanListeners(cachedPlan);
+        }
+        const cachedCreditBalances = this.creditBalances[newContextStr];
+        if (cachedCreditBalances !== void 0) {
+          this.notifyCreditBalanceListeners(cachedCreditBalances);
+        }
+      } else {
+        this.setIsPending(true);
+      }
       if (!this.conn) {
         if (this.isConnecting) {
           this.debug(
@@ -1386,7 +2371,7 @@ var Schematic = class {
           }
           if (this.conn !== null) {
             const socket2 = await this.conn;
-            await this.wsSendMessage(socket2, context);
+            await this.wsSendMessage(socket2, context, cacheHit);
             return;
           }
         }
@@ -1402,7 +2387,7 @@ var Schematic = class {
           this.conn = this.wsConnect();
           const socket2 = await this.conn;
           this.isConnecting = false;
-          await this.wsSendMessage(socket2, context);
+          await this.wsSendMessage(socket2, context, cacheHit);
           return;
         } catch (error) {
           this.isConnecting = false;
@@ -1410,7 +2395,7 @@ var Schematic = class {
         }
       }
       const socket = await this.conn;
-      await this.wsSendMessage(socket, context);
+      await this.wsSendMessage(socket, context, cacheHit);
     } catch (error) {
       console.warn("Failed to establish WebSocket connection:", error);
       this.context = context;
@@ -1470,10 +2455,12 @@ var Schematic = class {
       `Optimistically updating feature usage for event: ${eventName}`,
       { quantity }
     );
+    let dirty = false;
     Object.entries(flagsForEvent).forEach(([flagKey, check]) => {
       if (check === void 0) return;
       const updatedCheck = { ...check };
       if (typeof updatedCheck.featureUsage === "number") {
+        dirty = true;
         updatedCheck.featureUsage += quantity;
         if (typeof updatedCheck.featureAllocation === "number") {
           const wasExceeded = updatedCheck.featureUsageExceeded === true;
@@ -1503,6 +2490,9 @@ var Schematic = class {
         this.notifyFlagValueListeners(flagKey, updatedCheck.value);
       }
     });
+    if (dirty) {
+      this.persistContextToCache(contextString(this.context));
+    }
   };
   /**
    * Event processing
@@ -1605,6 +2595,206 @@ var Schematic = class {
     this.storage.setItem(anonymousIdKey, generatedAnonymousId);
     return generatedAnonymousId;
   };
+  hasCachedValuesForContext = (contextStr) => {
+    const cachedChecks = this.checks[contextStr];
+    if (cachedChecks !== void 0 && Object.keys(cachedChecks).length > 0) {
+      return true;
+    }
+    if (this.planChecks[contextStr] !== void 0) {
+      return true;
+    }
+    const cachedCreditBalances = this.creditBalances[contextStr];
+    return cachedCreditBalances !== void 0 && Object.keys(cachedCreditBalances).length > 0;
+  };
+  readFlagStateCache = () => {
+    if (!this.persistFlagState || this.storage === void 0) {
+      return null;
+    }
+    let raw;
+    try {
+      raw = this.storage.getItem(this.flagStateCacheKey);
+    } catch (error) {
+      this.debug("Failed to read flag state cache:", error);
+      return null;
+    }
+    if (raw === null || raw === void 0) {
+      return null;
+    }
+    let parsed;
+    try {
+      parsed = typeof raw === "string" ? JSON.parse(raw) : raw;
+    } catch (error) {
+      this.debug("Flag state cache is not valid JSON, ignoring:", error);
+      return null;
+    }
+    if (parsed === null || typeof parsed !== "object" || parsed.version !== cacheVersion || typeof parsed.contexts !== "object" || parsed.contexts === null) {
+      this.debug(
+        `Flag state cache invalid or version mismatch (expected ${cacheVersion}), ignoring`,
+        {
+          parsedType: typeof parsed,
+          version: parsed?.version
+        }
+      );
+      return null;
+    }
+    return parsed;
+  };
+  reviveCachedCheck = (raw) => {
+    if (raw === null || typeof raw !== "object") return null;
+    const obj = raw;
+    if (typeof obj.flag !== "string" || typeof obj.value !== "boolean" || typeof obj.reason !== "string") {
+      return null;
+    }
+    const revived = { ...obj };
+    if (typeof revived.featureUsageResetAt === "string") {
+      const d = new Date(revived.featureUsageResetAt);
+      revived.featureUsageResetAt = isNaN(d.getTime()) ? void 0 : d;
+    }
+    return revived;
+  };
+  reviveCachedPlan = (raw) => {
+    if (raw === null || typeof raw !== "object") return null;
+    const obj = raw;
+    if (typeof obj.id !== "string" || typeof obj.name !== "string") {
+      return null;
+    }
+    const revived = { ...obj };
+    if (typeof revived.trialEndDate === "string") {
+      const d = new Date(revived.trialEndDate);
+      revived.trialEndDate = isNaN(d.getTime()) ? void 0 : d;
+    }
+    return revived;
+  };
+  reviveCachedCreditBalances = (raw) => {
+    if (raw === null || typeof raw !== "object") return null;
+    const revived = {};
+    for (const [creditId, value] of Object.entries(
+      raw
+    )) {
+      if (value === null || typeof value !== "object") continue;
+      const obj = value;
+      if (typeof obj.remaining !== "number" || typeof obj.reserved !== "number" || typeof obj.settled !== "number") {
+        continue;
+      }
+      revived[creditId] = {
+        remaining: obj.remaining,
+        reserved: obj.reserved,
+        settled: obj.settled
+      };
+    }
+    return Object.keys(revived).length > 0 ? revived : null;
+  };
+  hydrateFlagStateFromCache = () => {
+    const parsed = this.readFlagStateCache();
+    if (parsed === null) return;
+    const cutoff = Date.now() - this.flagStateCacheMaxAgeMs;
+    const fresh = {};
+    let contextsLoaded = 0;
+    let contextsExpired = 0;
+    for (const [contextStr, entry] of Object.entries(parsed.contexts)) {
+      if (entry === null || typeof entry !== "object") continue;
+      if (typeof entry.updatedAt !== "number") continue;
+      if (entry.updatedAt < cutoff) {
+        contextsExpired += 1;
+        continue;
+      }
+      const revivedChecks = {};
+      if (entry.checks !== null && typeof entry.checks === "object") {
+        for (const [flagKey, rawCheck] of Object.entries(entry.checks)) {
+          const revived = this.reviveCachedCheck(rawCheck);
+          if (revived === null || revived.flag !== flagKey) continue;
+          revivedChecks[flagKey] = revived;
+          if (typeof revived.featureUsageEvent === "string") {
+            this.updateFeatureUsageEventMap(revived);
+          }
+        }
+        if (Object.keys(revivedChecks).length > 0) {
+          this.checks[contextStr] = revivedChecks;
+        }
+      }
+      let revivedPlan;
+      if (entry.plan !== void 0 && entry.plan !== null) {
+        const p = this.reviveCachedPlan(entry.plan);
+        if (p !== null) {
+          revivedPlan = p;
+          this.planChecks[contextStr] = p;
+        }
+      }
+      let revivedCreditBalances;
+      if (entry.creditBalances !== void 0 && entry.creditBalances !== null) {
+        const balances = this.reviveCachedCreditBalances(entry.creditBalances);
+        if (balances !== null) {
+          revivedCreditBalances = balances;
+          this.creditBalances[contextStr] = balances;
+        }
+      }
+      fresh[contextStr] = {
+        checks: revivedChecks,
+        plan: revivedPlan,
+        creditBalances: revivedCreditBalances,
+        updatedAt: entry.updatedAt
+      };
+      contextsLoaded += 1;
+    }
+    this.cachedFlagState = { version: cacheVersion, contexts: fresh };
+    this.debug(
+      `Hydrated flag state cache: ${contextsLoaded} loaded, ${contextsExpired} expired`
+    );
+  };
+  persistContextToCache = (contextStr) => {
+    const checksForContext = this.checks[contextStr];
+    const planForContext = this.planChecks[contextStr];
+    const creditBalancesForContext = this.creditBalances[contextStr];
+    if (checksForContext === void 0 && planForContext === void 0 && creditBalancesForContext === void 0) {
+      return;
+    }
+    if (this.cachedFlagState === null) {
+      this.cachedFlagState = (this.persistFlagState ? this.readFlagStateCache() : null) ?? {
+        version: cacheVersion,
+        contexts: {}
+      };
+    }
+    const cleanChecks = {};
+    if (checksForContext !== void 0) {
+      for (const [flagKey, check] of Object.entries(checksForContext)) {
+        if (check !== void 0) cleanChecks[flagKey] = check;
+      }
+    }
+    this.cachedFlagState.contexts[contextStr] = {
+      checks: cleanChecks,
+      plan: planForContext,
+      creditBalances: creditBalancesForContext,
+      updatedAt: Date.now()
+    };
+    const entries = Object.entries(this.cachedFlagState.contexts);
+    if (entries.length > flagStateCacheMaxContexts) {
+      entries.sort((a, b) => b[1].updatedAt - a[1].updatedAt);
+      const survivors = entries.slice(0, flagStateCacheMaxContexts);
+      const survivorKeys = new Set(survivors.map(([k]) => k));
+      for (const key of Object.keys(this.checks)) {
+        if (!survivorKeys.has(key)) delete this.checks[key];
+      }
+      for (const key of Object.keys(this.planChecks)) {
+        if (!survivorKeys.has(key)) delete this.planChecks[key];
+      }
+      for (const key of Object.keys(this.creditBalances)) {
+        if (!survivorKeys.has(key)) delete this.creditBalances[key];
+      }
+      this.cachedFlagState = {
+        version: cacheVersion,
+        contexts: Object.fromEntries(survivors)
+      };
+    }
+    if (!this.persistFlagState || this.storage === void 0) return;
+    try {
+      this.storage.setItem(
+        this.flagStateCacheKey,
+        JSON.stringify(this.cachedFlagState)
+      );
+    } catch (error) {
+      this.debug("Failed to write flag state cache:", error);
+    }
+  };
   handleEvent = (eventType, eventBody) => {
     const event = {
       api_key: this.apiKey,
@@ -1865,6 +3055,9 @@ var Schematic = class {
     if (this.conn !== null) {
       try {
         const socket = await this.conn;
+        if (this.currentWebSocket === socket) {
+          this.currentWebSocket = null;
+        }
         if (socket.readyState === WebSocket.OPEN || socket.readyState === WebSocket.CONNECTING) {
           socket.close();
         }
@@ -2030,6 +3223,7 @@ var Schematic = class {
       this.debug(`Creating WebSocket connection ${connectionId} to ${wsUrl}`);
       let timeoutId = null;
       let isResolved = false;
+      let didOpen = false;
       timeoutId = setTimeout(() => {
         if (!isResolved) {
           isResolved = true;
@@ -2043,6 +3237,7 @@ var Schematic = class {
       webSocket.onopen = () => {
         if (isResolved) return;
         isResolved = true;
+        didOpen = true;
         if (timeoutId !== null) {
           clearTimeout(timeoutId);
         }
@@ -2068,11 +3263,12 @@ var Schematic = class {
         }
         this.debug(`WebSocket connection ${connectionId} closed`);
         this.conn = null;
+        const wasCurrent = this.currentWebSocket === webSocket;
         if (this.currentWebSocket === webSocket) {
           this.currentWebSocket = null;
           this.isConnecting = false;
         }
-        if (!isResolved && !this.wsIntentionalDisconnect && this.webSocketReconnect) {
+        if (wasCurrent && didOpen && !this.wsIntentionalDisconnect && this.webSocketReconnect) {
           this.attemptReconnect();
         }
       };
@@ -2168,6 +3364,16 @@ var Schematic = class {
     const plan = this.planChecks[contextStr];
     return plan;
   };
+  /** Get the company's lease-aware credit balances for the current context, keyed by credit ID */
+  getCreditBalances = () => {
+    const contextStr = contextString(this.context);
+    return this.creditBalances[contextStr] ?? emptyCreditBalances;
+  };
+  /** Get the company's lease-aware balance for a single credit type in the current context */
+  getCreditBalance = (creditId) => {
+    const contextStr = contextString(this.context);
+    return this.creditBalances[contextStr]?.[creditId];
+  };
   // flag checks state
   getFlagCheck = (flagKey) => {
     const contextStr = contextString(this.context);
@@ -2227,6 +3433,13 @@ var Schematic = class {
       this.planListeners.delete(listener);
     };
   };
+  /** Register an event listener that will be notified with the company's credit balances (keyed by credit ID) whenever they change */
+  addCreditBalanceListener = (listener) => {
+    this.creditBalanceListeners.add(listener);
+    return () => {
+      this.creditBalanceListeners.delete(listener);
+    };
+  };
   notifyFlagCheckListeners = (flagKey, check) => {
     const listeners = this.flagCheckListeners?.[flagKey] ?? [];
     if (listeners.size > 0) {
@@ -2290,6 +3503,17 @@ var Schematic = class {
       });
     });
   };
+  notifyCreditBalanceListeners = (value) => {
+    const listeners = this.creditBalanceListeners ?? [];
+    if (listeners.size > 0) {
+      this.debug(`Notifying ${listeners.size} credit balance listeners`, {
+        value
+      });
+    }
+    listeners.forEach(
+      (listener) => notifyCreditBalanceListener(listener, value)
+    );
+  };
 };
 var notifyPendingListener = (listener, value) => {
   if (listener.length > 0) {
@@ -2319,4 +3543,11 @@ var notifyPlanListener = (listener, value) => {
     listener();
   }
 };
+var notifyCreditBalanceListener = (listener, value) => {
+  if (listener.length > 0) {
+    listener(value);
+  } else {
+    listener();
+  }
+};
 /* @preserve */