halt-rate 0.3.0 → 0.5.0

package/dist/index.mjs

@@ -28,6 +28,7 @@ function normalizePolicy(policy) {
     algorithm: policy.algorithm ?? "token_bucket" /* TOKEN_BUCKET */,
     keyStrategy: policy.keyStrategy ?? "ip" /* IP */,
     burst: policy.burst ?? Math.floor(policy.limit * 1.2),
+    slidingPrecision: policy.slidingPrecision ?? 10,
     cost: policy.cost ?? 1,
     blockDuration: policy.blockDuration ?? void 0,
     keyExtractor: policy.keyExtractor ?? void 0,
@@ -43,6 +44,9 @@ function normalizePolicy(policy) {
   if (normalized.cost <= 0) {
     throw new Error("cost must be positive");
   }
+  if (!Number.isInteger(normalized.slidingPrecision) || normalized.slidingPrecision <= 0) {
+    throw new Error("slidingPrecision must be a positive integer");
+  }
   if (normalized.burst < normalized.limit) {
     throw new Error("burst must be >= limit");
   }
@@ -484,21 +488,22 @@ var RateLimiter = class {
       return decision2;
     }
     const store = this.store;
-    let algorithm = this.algorithmCache.get(policy.name);
+    const cacheKey = `${policy.name}|${policy.algorithm}|${policy.limit}|${policy.window}|${policy.burst}|${policy.slidingPrecision}`;
+    let algorithm = this.algorithmCache.get(cacheKey);
     if (!algorithm) {
       if (policy.algorithm === "token_bucket" /* TOKEN_BUCKET */) {
         algorithm = new TokenBucket(policy.burst, policy.limit, policy.window);
       } else if (policy.algorithm === "fixed_window" /* FIXED_WINDOW */) {
         algorithm = new FixedWindow(policy.limit, policy.window);
       } else if (policy.algorithm === "sliding_window" /* SLIDING_WINDOW */) {
-        algorithm = new SlidingWindow(policy.limit, policy.window);
+        algorithm = new SlidingWindow(policy.limit, policy.window, policy.slidingPrecision);
       } else if (policy.algorithm === "leaky_bucket" /* LEAKY_BUCKET */) {
         const leakRate = policy.limit / policy.window;
         algorithm = new LeakyBucket(policy.burst, leakRate, policy.window);
       } else {
         throw new Error(`Algorithm ${policy.algorithm} not implemented`);
       }
-      this.algorithmCache.set(policy.name, algorithm);
+      this.algorithmCache.set(cacheKey, algorithm);
     }
     const span = this.otelTracer?.startSpan?.("halt.check", { attributes: { policy: policy.name, key } });
     const state = store.get(storageKey);
@@ -1093,6 +1098,70 @@ function getPlanPolicy(planName) {
   return PLAN_TIERS[normalized];
 }
 
+// src/core/registry.ts
+var PolicyRegistry = class {
+  constructor(initial = []) {
+    this.policies = /* @__PURE__ */ new Map();
+    for (const p of initial) this.register(p);
+  }
+  /** Add or replace a policy (keyed by `policy.name`). */
+  register(policy) {
+    this.policies.set(policy.name, policy);
+    return this;
+  }
+  get(name) {
+    return this.policies.get(name);
+  }
+  has(name) {
+    return this.policies.has(name);
+  }
+  /** Mutate fields of an existing policy at runtime (e.g. `{ limit: 500 }`). */
+  update(name, patch) {
+    const existing = this.policies.get(name);
+    if (!existing) throw new Error(`Unknown policy: ${name}`);
+    const updated = { ...existing, ...patch, name: existing.name };
+    if (patch.limit !== void 0 && patch.burst === void 0) {
+      delete updated.burst;
+    }
+    this.policies.set(name, updated);
+    return updated;
+  }
+  remove(name) {
+    return this.policies.delete(name);
+  }
+  list() {
+    return [...this.policies.values()];
+  }
+  /**
+   * Build a resolver for the limiter's `policy` option.
+   * @param selector maps a request to a registered policy name.
+   */
+  resolver(selector) {
+    return (request) => {
+      const name = selector(request);
+      const policy = this.policies.get(name);
+      if (!policy) throw new Error(`Unknown policy: ${name}`);
+      return policy;
+    };
+  }
+};
+function cachedPolicyResolver(loader, options = {}) {
+  const ttlMs = options.ttlMs ?? 5e3;
+  const keyFn = options.key ?? (() => "__default__");
+  const cache = /* @__PURE__ */ new Map();
+  return async (request) => {
+    const key = keyFn(request);
+    const now = Date.now();
+    const hit = cache.get(key);
+    if (hit && hit.expires > now) {
+      return hit.policy;
+    }
+    const policy = await loader(request);
+    cache.set(key, { policy, expires: now + ttlMs });
+    return policy;
+  };
+}
+
 // src/core/quota.ts
 var QuotaPeriod = /* @__PURE__ */ ((QuotaPeriod2) => {
   QuotaPeriod2["HOURLY"] = "hourly";
@@ -1611,6 +1680,6 @@ var OpenTelemetryMetrics = class {
   }
 };
 
-export { Algorithm, CompositeTelemetry, InMemoryStore, KeyStrategy, LoggingTelemetry, MetricsTelemetry, OpenTelemetryMetrics, PENALTY_LENIENT, PENALTY_MODERATE, PENALTY_STRICT, PenaltyManager, QUOTA_ENTERPRISE_MONTHLY, QUOTA_FREE_DAILY, QUOTA_FREE_MONTHLY, QUOTA_PRO_DAILY, QUOTA_PRO_MONTHLY, QuotaManager, QuotaPeriod, RateLimiter, RedisStore, StatsCollector, extractors_exports as extractors, isAtomicStore, normalizePolicy, presets_exports as presets, toHeaders };
+export { Algorithm, CompositeTelemetry, InMemoryStore, KeyStrategy, LoggingTelemetry, MetricsTelemetry, OpenTelemetryMetrics, PENALTY_LENIENT, PENALTY_MODERATE, PENALTY_STRICT, PenaltyManager, PolicyRegistry, QUOTA_ENTERPRISE_MONTHLY, QUOTA_FREE_DAILY, QUOTA_FREE_MONTHLY, QUOTA_PRO_DAILY, QUOTA_PRO_MONTHLY, QuotaManager, QuotaPeriod, RateLimiter, RedisStore, StatsCollector, cachedPolicyResolver, extractors_exports as extractors, isAtomicStore, normalizePolicy, presets_exports as presets, toHeaders };
 //# sourceMappingURL=index.mjs.map
 //# sourceMappingURL=index.mjs.map