@snakechat/commskit-node 0.0.0-stage → 0.1.1

package/dist/redis.js

@@ -0,0 +1,296 @@
+import {
+  RateLimiter
+} from "./chunk-BBFVG7QU.js";
+import "./chunk-KLKRK6OG.js";
+import {
+  logger
+} from "./chunk-5BJ2SNM5.js";
+
+// src/core/redis.ts
+var USER_CHANNEL_PREFIX = "user_channel:";
+var USER_CHANNEL_PATTERN = "user_channel:*";
+var PRESENCE_KEY_PREFIX = "presence:";
+var AWAY_KEY_PREFIX = "presence_away:";
+var ROOM_CHANNEL_PREFIX = "room_channel:";
+var ROOM_CHANNEL_PATTERN = "room_channel:*";
+var CAME_ONLINE_KEY_PREFIX = "came_online:";
+async function loadIoRedis() {
+  const name = "ioredis";
+  try {
+    const mod = await import(
+      /* @vite-ignore */
+      name
+    );
+    return mod.default ?? mod.Redis ?? mod;
+  } catch {
+    throw new Error("RedisBridge needs ioredis: npm install ioredis");
+  }
+}
+var RedisBridge = class {
+  client;
+  heartbeatSeconds;
+  staleSeconds;
+  presenceTtlSeconds;
+  awayTtlSeconds;
+  url;
+  sub = null;
+  deliver = null;
+  roomDeliver = null;
+  constructor(opts = {}) {
+    if (!opts.url && !opts.client) throw new Error("RedisBridge needs a url or a client");
+    this.url = opts.url;
+    this.client = opts.client ?? null;
+    this.heartbeatSeconds = opts.heartbeatSeconds ?? 30;
+    this.staleSeconds = opts.staleSeconds ?? 90;
+    this.presenceTtlSeconds = opts.presenceTtlSeconds ?? 120;
+    this.awayTtlSeconds = opts.awayTtlSeconds ?? 12 * 3600;
+  }
+  setRoomDeliver(fn) {
+    this.roomDeliver = fn;
+  }
+  async start(deliver, _serverId) {
+    this.deliver = deliver;
+    let ready = false;
+    try {
+      if (!this.client) {
+        const Redis = await loadIoRedis();
+        const c = new Redis(this.url, { lazyConnect: true, maxRetriesPerRequest: 1, connectTimeout: 5e3, retryStrategy: (n) => ready ? Math.min(n * 200, 2e3) : null });
+        c.on("error", () => void 0);
+        this.client = c;
+        await c.connect();
+      }
+      await this.client.ping();
+      ready = true;
+      const sub = this.client.duplicate();
+      sub.on("error", () => void 0);
+      if (sub.connect) {
+        try {
+          await sub.connect();
+        } catch (err) {
+          if (!/already/i.test(String(err?.message))) throw err;
+        }
+      }
+      sub.on("pmessage", (_pattern, channel, message) => {
+        void this.consume(channel, message);
+      });
+      await sub.psubscribe(USER_CHANNEL_PATTERN, ROOM_CHANNEL_PATTERN);
+      this.sub = sub;
+    } catch (err) {
+      logger.warn(`Redis unavailable (${err?.message ?? err}); running single-worker`);
+      const c = this.client;
+      this.client = null;
+      this.sub = null;
+      try {
+        c?.disconnect?.();
+      } catch {
+      }
+    }
+  }
+  async stop() {
+    for (const obj of [this.sub, this.client]) {
+      if (!obj) continue;
+      try {
+        if (obj.quit) await obj.quit();
+        else obj.disconnect?.();
+      } catch {
+        try {
+          obj.disconnect?.();
+        } catch {
+        }
+      }
+    }
+    this.sub = null;
+    this.client = null;
+  }
+  async consume(channel, raw) {
+    if (typeof channel !== "string" || !raw) return;
+    if (channel.startsWith(ROOM_CHANNEL_PREFIX)) {
+      const room = channel.slice(ROOM_CHANNEL_PREFIX.length);
+      if (!this.roomDeliver) return;
+      try {
+        await this.roomDeliver(room, JSON.parse(raw));
+      } catch (err) {
+        logger.warn(`Cross-worker room delivery to ${room} failed: ${err?.message ?? err}`);
+      }
+      return;
+    }
+    if (!this.deliver) return;
+    const userId = channel.slice(channel.lastIndexOf(":") + 1);
+    try {
+      await this.deliver(userId, JSON.parse(raw));
+    } catch (err) {
+      logger.warn(`Cross-worker delivery to ${userId} failed: ${err?.message ?? err}`);
+    }
+  }
+  // ------------------------------------------------------------ presence
+  async setOnline(userId, serverId) {
+    const c = this.client;
+    if (!c) return;
+    const key = PRESENCE_KEY_PREFIX + userId;
+    try {
+      await c.hset(key, serverId, Math.floor(Date.now() / 1e3));
+      await c.expire(key, this.presenceTtlSeconds);
+    } catch (err) {
+      logger.warn(`setOnline(${userId}) failed: ${err?.message ?? err}`);
+    }
+  }
+  async setOffline(userId, serverId) {
+    const c = this.client;
+    if (!c) return;
+    try {
+      await c.hdel(PRESENCE_KEY_PREFIX + userId, serverId);
+    } catch (err) {
+      logger.warn(`setOffline(${userId}) failed: ${err?.message ?? err}`);
+    }
+  }
+  async setAway(userId, away) {
+    const c = this.client;
+    if (!c) return;
+    const key = AWAY_KEY_PREFIX + userId;
+    try {
+      if (away) await c.set(key, 1, "EX", this.awayTtlSeconds);
+      else await c.del(key);
+    } catch (err) {
+      logger.warn(`setAway(${userId}) failed: ${err?.message ?? err}`);
+    }
+  }
+  async getUserServers(userId) {
+    const c = this.client;
+    if (!c) return [];
+    let entries;
+    try {
+      entries = await c.hgetall(PRESENCE_KEY_PREFIX + userId);
+    } catch (err) {
+      logger.warn(`getUserServers(${userId}) failed: ${err?.message ?? err}`);
+      return [];
+    }
+    const cutoff = Date.now() / 1e3 - this.staleSeconds;
+    return Object.entries(entries ?? {}).filter(([, beat]) => Number(beat) >= cutoff).map(([server]) => server);
+  }
+  async isOnline(userId) {
+    if (!(await this.getUserServers(userId)).length) return false;
+    try {
+      return !await this.client.exists(AWAY_KEY_PREFIX + userId);
+    } catch {
+      return true;
+    }
+  }
+  async getOnlineUsers(userIds) {
+    const c = this.client;
+    if (!c || !userIds.length) return /* @__PURE__ */ new Set();
+    let results;
+    try {
+      results = await Promise.all(
+        userIds.map(async (uid) => [await c.hgetall(PRESENCE_KEY_PREFIX + uid), await c.exists(AWAY_KEY_PREFIX + uid)])
+      );
+    } catch (err) {
+      logger.warn(`getOnlineUsers failed: ${err?.message ?? err}`);
+      return /* @__PURE__ */ new Set();
+    }
+    const cutoff = Date.now() / 1e3 - this.staleSeconds;
+    const online = /* @__PURE__ */ new Set();
+    userIds.forEach((uid, i) => {
+      const [servers, away] = results[i];
+      if (!away && Object.values(servers ?? {}).some((b) => Number(b) >= cutoff)) online.add(uid);
+    });
+    return online;
+  }
+  async publish(userId, payload, origin) {
+    const c = this.client;
+    if (!c) return;
+    try {
+      await c.publish(USER_CHANNEL_PREFIX + userId, JSON.stringify({ origin, data: payload }));
+    } catch (err) {
+      logger.warn(`publish(${userId}) failed: ${err?.message ?? err}`);
+    }
+  }
+  async publishRoom(room, payload, origin) {
+    const c = this.client;
+    if (!c) return;
+    try {
+      await c.publish(ROOM_CHANNEL_PREFIX + room, JSON.stringify({ origin, data: payload }));
+    } catch (err) {
+      logger.warn(`publishRoom(${room}) failed: ${err?.message ?? err}`);
+    }
+  }
+  // ------------------------------------------------------- user.came_online
+  /** True for exactly one caller until the key expires or is released (`SET NX PX`). Without Redis every caller wins. */
+  async claimCameOnline(userId, ttlSeconds) {
+    const c = this.client;
+    if (!c) return true;
+    try {
+      return Boolean(await c.set(CAME_ONLINE_KEY_PREFIX + userId, 1, "PX", Math.max(1, Math.floor(ttlSeconds * 1e3)), "NX"));
+    } catch (err) {
+      logger.warn(`claimCameOnline(${userId}) failed: ${err?.message ?? err}`);
+      return true;
+    }
+  }
+  async releaseCameOnline(userId) {
+    const c = this.client;
+    if (!c) return;
+    try {
+      await c.del(CAME_ONLINE_KEY_PREFIX + userId);
+    } catch (err) {
+      logger.warn(`releaseCameOnline(${userId}) failed: ${err?.message ?? err}`);
+    }
+  }
+};
+var TOKEN_BUCKET_LUA = `
+local cap = tonumber(ARGV[1])
+local rate = tonumber(ARGV[2])
+local now = tonumber(ARGV[3])
+local cost = tonumber(ARGV[4])
+local v = redis.call('HMGET', KEYS[1], 't', 'ts')
+local tokens = tonumber(v[1])
+local ts = tonumber(v[2])
+if tokens == nil or ts == nil then tokens = cap; ts = now end
+if now > ts then tokens = math.min(cap, tokens + (now - ts) * rate) end
+local wait = 0
+if tokens >= cost then tokens = tokens - cost else wait = (cost - tokens) / rate end
+redis.call('HSET', KEYS[1], 't', tostring(tokens), 'ts', tostring(math.max(now, ts)))
+redis.call('PEXPIRE', KEYS[1], tonumber(ARGV[5]))
+return tostring(wait)
+`;
+var RedisRateLimiter = class {
+  constructor(source, prefix = "ratelimit:", fallback) {
+    this.source = source;
+    this.prefix = prefix;
+    this.fallback = fallback ?? new RateLimiter();
+  }
+  source;
+  prefix;
+  fallback;
+  client() {
+    const s = this.source;
+    if (typeof s.eval === "function") return s;
+    return s.client ?? null;
+  }
+  async hit(key, limit, cost = 1) {
+    if (!limit) return 0;
+    const client = this.client();
+    if (!client) return this.fallback.hit(key, limit, cost);
+    const ttlMs = Math.floor(limit.seconds * 1e3 * 2) + 1e3;
+    try {
+      const raw = await client.eval(TOKEN_BUCKET_LUA, 1, this.prefix + key, limit.count, String(limit.rate), String(Date.now() / 1e3), String(cost), ttlMs);
+      const wait = Number(Buffer.isBuffer(raw) ? raw.toString() : raw);
+      if (Number.isNaN(wait)) throw new Error(`unexpected reply ${String(raw)}`);
+      return Math.max(0, wait);
+    } catch (err) {
+      logger.warn(`Redis rate limit failed (${err?.message ?? err}); using the in-memory limiter`);
+      return this.fallback.hit(key, limit, cost);
+    }
+  }
+};
+export {
+  AWAY_KEY_PREFIX,
+  CAME_ONLINE_KEY_PREFIX,
+  PRESENCE_KEY_PREFIX,
+  ROOM_CHANNEL_PATTERN,
+  ROOM_CHANNEL_PREFIX,
+  RedisBridge,
+  RedisRateLimiter,
+  TOKEN_BUCKET_LUA,
+  USER_CHANNEL_PATTERN,
+  USER_CHANNEL_PREFIX
+};
+//# sourceMappingURL=redis.js.map

package/dist/redis.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/core/redis.ts"],"sourcesContent":["/**\n * Optional Redis presence + cross-worker delivery (port of commskit.core.redis_bridge) and the shared\n * token-bucket limiter (port of commskit.core.ratelimit.RedisRateLimiter). Needs `ioredis` (optional peer\n * dependency, imported lazily) unless a client is injected.\n *\n * Key names match the Python kit and matrimonial-backend-v2, so Python and Node workers interoperate:\n * `presence:{uid}` (hash serverId -> heartbeat epoch seconds), `presence_away:{uid}`, pub/sub `user_channel:{uid}`\n * carrying `{\"origin\",\"data\"}`, `room_channel:{room}` (same envelope) and `came_online:{uid}` (SET NX PX).\n *\n * Every Redis failure is logged and degrades to \"not connected\" (local-only); it never throws.\n */\nimport type { Deliver, PresenceBridge } from \"./connections.js\";\nimport { logger } from \"./logger.js\";\nimport { type Limiter, RateLimit, RateLimiter } from \"./ratelimit.js\";\n\nexport const USER_CHANNEL_PREFIX = \"user_channel:\";\nexport const USER_CHANNEL_PATTERN = \"user_channel:*\";\nexport const PRESENCE_KEY_PREFIX = \"presence:\";\nexport const AWAY_KEY_PREFIX = \"presence_away:\";\nexport const ROOM_CHANNEL_PREFIX = \"room_channel:\";\nexport const ROOM_CHANNEL_PATTERN = \"room_channel:*\";\nexport const CAME_ONLINE_KEY_PREFIX = \"came_online:\";\n\n/** The slice of an `ioredis` client the bridge uses. */\nexport interface RedisClientLike {\n  ping(): Promise<unknown>;\n  hset(key: string, field: string, value: string | number): Promise<unknown>;\n  expire(key: string, seconds: number): Promise<unknown>;\n  hdel(key: string, field: string): Promise<unknown>;\n  hgetall(key: string): Promise<Record<string, string>>;\n  set(key: string, value: string | number, ...args: Array<string | number>): Promise<unknown>;\n  del(key: string): Promise<unknown>;\n  exists(key: string): Promise<number>;\n  publish(channel: string, message: string): Promise<unknown>;\n  eval(script: string, numKeys: number, ...args: Array<string | number>): Promise<unknown>;\n  /** A second connection for the subscriber. */\n  duplicate(): RedisSubscriberLike;\n  connect?(): Promise<unknown>;\n  quit?(): Promise<unknown>;\n  disconnect?(): void;\n  on?(event: string, listener: (...args: any[]) => void): unknown;\n}\n\nexport interface RedisSubscriberLike {\n  psubscribe(...patterns: string[]): Promise<unknown>;\n  on(event: \"pmessage\", listener: (pattern: string, channel: string, message: string) => void): unknown;\n  on(event: string, listener: (...args: any[]) => void): unknown;\n  connect?(): Promise<unknown>;\n  quit?(): Promise<unknown>;\n  disconnect?(): void;\n}\n\nexport type RoomDeliver = (room: string, envelope: any) => Promise<void>;\n\nexport interface RedisBridgeOptions {\n  url?: string;\n  /** An already-created ioredis-compatible client (tests, shared connections). */\n  client?: RedisClientLike;\n  heartbeatSeconds?: number;\n  staleSeconds?: number;\n  presenceTtlSeconds?: number;\n  awayTtlSeconds?: number;\n}\n\nasync function loadIoRedis(): Promise<any> {\n  const name = \"ioredis\"; // variable specifier: no compile-time dependency\n  try {\n    const mod: any = await import(/* @vite-ignore */ name);\n    return mod.default ?? mod.Redis ?? mod;\n  } catch {\n    throw new Error('RedisBridge needs ioredis: npm install ioredis');\n  }\n}\n\nexport class RedisBridge implements PresenceBridge {\n  client: RedisClientLike | null;\n  heartbeatSeconds: number;\n  staleSeconds: number;\n  presenceTtlSeconds: number;\n  awayTtlSeconds: number;\n  private url?: string;\n  private sub: RedisSubscriberLike | null = null;\n  private deliver: Deliver | null = null;\n  private roomDeliver: RoomDeliver | null = null;\n\n  constructor(opts: RedisBridgeOptions = {}) {\n    if (!opts.url && !opts.client) throw new Error(\"RedisBridge needs a url or a client\");\n    this.url = opts.url;\n    this.client = opts.client ?? null;\n    this.heartbeatSeconds = opts.heartbeatSeconds ?? 30;\n    this.staleSeconds = opts.staleSeconds ?? 90;\n    this.presenceTtlSeconds = opts.presenceTtlSeconds ?? 120;\n    this.awayTtlSeconds = opts.awayTtlSeconds ?? 12 * 3600;\n  }\n\n  setRoomDeliver(fn: RoomDeliver): void {\n    this.roomDeliver = fn;\n  }\n\n  async start(deliver: Deliver, _serverId: string): Promise<void> {\n    this.deliver = deliver;\n    let ready = false; // fail fast on the first connect; reconnect with backoff once it worked\n    try {\n      if (!this.client) {\n        const Redis = await loadIoRedis();\n        const c = new Redis(this.url, { lazyConnect: true, maxRetriesPerRequest: 1, connectTimeout: 5000, retryStrategy: (n: number) => (ready ? Math.min(n * 200, 2000) : null) });\n        c.on(\"error\", () => undefined);\n        this.client = c as RedisClientLike;\n        await c.connect();\n      }\n      await this.client.ping();\n      ready = true;\n      const sub = this.client.duplicate();\n      sub.on(\"error\", () => undefined);\n      if (sub.connect) {\n        try {\n          await sub.connect();\n        } catch (err) {\n          if (!/already/i.test(String((err as Error)?.message))) throw err;\n        }\n      }\n      sub.on(\"pmessage\", (_pattern: string, channel: string, message: string) => {\n        void this.consume(channel, message);\n      });\n      await sub.psubscribe(USER_CHANNEL_PATTERN, ROOM_CHANNEL_PATTERN);\n      this.sub = sub;\n    } catch (err) {\n      logger.warn(`Redis unavailable (${(err as Error)?.message ?? err}); running single-worker`);\n      const c = this.client;\n      this.client = null;\n      this.sub = null;\n      try {\n        c?.disconnect?.();\n      } catch {\n        /* ignore */\n      }\n    }\n  }\n\n  async stop(): Promise<void> {\n    for (const obj of [this.sub, this.client]) {\n      if (!obj) continue;\n      try {\n        if (obj.quit) await obj.quit();\n        else obj.disconnect?.();\n      } catch {\n        try {\n          obj.disconnect?.();\n        } catch {\n          /* ignore */\n        }\n      }\n    }\n    this.sub = null;\n    this.client = null;\n  }\n\n  private async consume(channel: string, raw: string): Promise<void> {\n    if (typeof channel !== \"string\" || !raw) return;\n    if (channel.startsWith(ROOM_CHANNEL_PREFIX)) {\n      const room = channel.slice(ROOM_CHANNEL_PREFIX.length);\n      if (!this.roomDeliver) return;\n      try {\n        await this.roomDeliver(room, JSON.parse(raw));\n      } catch (err) {\n        logger.warn(`Cross-worker room delivery to ${room} failed: ${(err as Error)?.message ?? err}`);\n      }\n      return;\n    }\n    if (!this.deliver) return;\n    const userId = channel.slice(channel.lastIndexOf(\":\") + 1);\n    try {\n      await this.deliver(userId, JSON.parse(raw));\n    } catch (err) {\n      logger.warn(`Cross-worker delivery to ${userId} failed: ${(err as Error)?.message ?? err}`);\n    }\n  }\n\n  // ------------------------------------------------------------ presence\n\n  async setOnline(userId: string, serverId: string): Promise<void> {\n    const c = this.client;\n    if (!c) return;\n    const key = PRESENCE_KEY_PREFIX + userId;\n    try {\n      await c.hset(key, serverId, Math.floor(Date.now() / 1000));\n      await c.expire(key, this.presenceTtlSeconds);\n    } catch (err) {\n      logger.warn(`setOnline(${userId}) failed: ${(err as Error)?.message ?? err}`);\n    }\n  }\n\n  async setOffline(userId: string, serverId: string): Promise<void> {\n    const c = this.client;\n    if (!c) return;\n    try {\n      await c.hdel(PRESENCE_KEY_PREFIX + userId, serverId);\n    } catch (err) {\n      logger.warn(`setOffline(${userId}) failed: ${(err as Error)?.message ?? err}`);\n    }\n  }\n\n  async setAway(userId: string, away: boolean): Promise<void> {\n    const c = this.client;\n    if (!c) return;\n    const key = AWAY_KEY_PREFIX + userId;\n    try {\n      if (away) await c.set(key, 1, \"EX\", this.awayTtlSeconds);\n      else await c.del(key);\n    } catch (err) {\n      logger.warn(`setAway(${userId}) failed: ${(err as Error)?.message ?? err}`);\n    }\n  }\n\n  async getUserServers(userId: string): Promise<string[]> {\n    const c = this.client;\n    if (!c) return [];\n    let entries: Record<string, string>;\n    try {\n      entries = await c.hgetall(PRESENCE_KEY_PREFIX + userId);\n    } catch (err) {\n      logger.warn(`getUserServers(${userId}) failed: ${(err as Error)?.message ?? err}`);\n      return [];\n    }\n    const cutoff = Date.now() / 1000 - this.staleSeconds;\n    return Object.entries(entries ?? {}).filter(([, beat]) => Number(beat) >= cutoff).map(([server]) => server);\n  }\n\n  async isOnline(userId: string): Promise<boolean> {\n    if (!(await this.getUserServers(userId)).length) return false;\n    try {\n      return !(await this.client!.exists(AWAY_KEY_PREFIX + userId));\n    } catch {\n      return true;\n    }\n  }\n\n  async getOnlineUsers(userIds: string[]): Promise<Set<string>> {\n    const c = this.client;\n    if (!c || !userIds.length) return new Set();\n    let results: Array<[Record<string, string>, number]>;\n    try {\n      results = await Promise.all(\n        userIds.map(async (uid) => [await c.hgetall(PRESENCE_KEY_PREFIX + uid), await c.exists(AWAY_KEY_PREFIX + uid)] as [Record<string, string>, number]),\n      );\n    } catch (err) {\n      logger.warn(`getOnlineUsers failed: ${(err as Error)?.message ?? err}`);\n      return new Set();\n    }\n    const cutoff = Date.now() / 1000 - this.staleSeconds;\n    const online = new Set<string>();\n    userIds.forEach((uid, i) => {\n      const [servers, away] = results[i]!;\n      if (!away && Object.values(servers ?? {}).some((b) => Number(b) >= cutoff)) online.add(uid);\n    });\n    return online;\n  }\n\n  async publish(userId: string, payload: unknown, origin: string): Promise<void> {\n    const c = this.client;\n    if (!c) return;\n    try {\n      await c.publish(USER_CHANNEL_PREFIX + userId, JSON.stringify({ origin, data: payload }));\n    } catch (err) {\n      logger.warn(`publish(${userId}) failed: ${(err as Error)?.message ?? err}`);\n    }\n  }\n\n  async publishRoom(room: string, payload: unknown, origin: string): Promise<void> {\n    const c = this.client;\n    if (!c) return;\n    try {\n      await c.publish(ROOM_CHANNEL_PREFIX + room, JSON.stringify({ origin, data: payload }));\n    } catch (err) {\n      logger.warn(`publishRoom(${room}) failed: ${(err as Error)?.message ?? err}`);\n    }\n  }\n\n  // ------------------------------------------------------- user.came_online\n\n  /** True for exactly one caller until the key expires or is released (`SET NX PX`). Without Redis every caller wins. */\n  async claimCameOnline(userId: string, ttlSeconds: number): Promise<boolean> {\n    const c = this.client;\n    if (!c) return true;\n    try {\n      return Boolean(await c.set(CAME_ONLINE_KEY_PREFIX + userId, 1, \"PX\", Math.max(1, Math.floor(ttlSeconds * 1000)), \"NX\"));\n    } catch (err) {\n      logger.warn(`claimCameOnline(${userId}) failed: ${(err as Error)?.message ?? err}`);\n      return true;\n    }\n  }\n\n  async releaseCameOnline(userId: string): Promise<void> {\n    const c = this.client;\n    if (!c) return;\n    try {\n      await c.del(CAME_ONLINE_KEY_PREFIX + userId);\n    } catch (err) {\n      logger.warn(`releaseCameOnline(${userId}) failed: ${(err as Error)?.message ?? err}`);\n    }\n  }\n}\n\n// ------------------------------------------------------------ rate limiter\n\n// KEYS[1] bucket; ARGV: capacity, rate (tokens/s), now (s), cost, ttl (ms). Returns the wait as a string.\n// Verbatim from commskit.core.ratelimit.\nexport const TOKEN_BUCKET_LUA = `\nlocal cap = tonumber(ARGV[1])\nlocal rate = tonumber(ARGV[2])\nlocal now = tonumber(ARGV[3])\nlocal cost = tonumber(ARGV[4])\nlocal v = redis.call('HMGET', KEYS[1], 't', 'ts')\nlocal tokens = tonumber(v[1])\nlocal ts = tonumber(v[2])\nif tokens == nil or ts == nil then tokens = cap; ts = now end\nif now > ts then tokens = math.min(cap, tokens + (now - ts) * rate) end\nlocal wait = 0\nif tokens >= cost then tokens = tokens - cost else wait = (cost - tokens) / rate end\nredis.call('HSET', KEYS[1], 't', tostring(tokens), 'ts', tostring(math.max(now, ts)))\nredis.call('PEXPIRE', KEYS[1], tonumber(ARGV[5]))\nreturn tostring(wait)\n`;\n\n/**\n * Token buckets in Redis, shared by all workers. `source` is a `RedisBridge` (its `client` is read on every call, so the\n * limiter can be built before `hub.connections.start()`) or an ioredis-compatible client. Any Redis error, or no\n * client yet, falls back to `fallback` (in-memory by default).\n */\nexport class RedisRateLimiter implements Limiter {\n  fallback: Limiter;\n  constructor(\n    private source: { client?: RedisClientLike | null } | RedisClientLike,\n    public prefix = \"ratelimit:\",\n    fallback?: Limiter,\n  ) {\n    this.fallback = fallback ?? new RateLimiter();\n  }\n\n  private client(): RedisClientLike | null {\n    const s: any = this.source;\n    if (typeof s.eval === \"function\") return s;\n    return s.client ?? null;\n  }\n\n  async hit(key: string, limit: RateLimit | null | undefined, cost = 1): Promise<number> {\n    if (!limit) return 0;\n    const client = this.client();\n    if (!client) return this.fallback.hit(key, limit, cost);\n    const ttlMs = Math.floor(limit.seconds * 1000 * 2) + 1000;\n    try {\n      const raw = await client.eval(TOKEN_BUCKET_LUA, 1, this.prefix + key, limit.count, String(limit.rate), String(Date.now() / 1000), String(cost), ttlMs);\n      const wait = Number(Buffer.isBuffer(raw) ? raw.toString() : raw);\n      if (Number.isNaN(wait)) throw new Error(`unexpected reply ${String(raw)}`);\n      return Math.max(0, wait);\n    } catch (err) {\n      logger.warn(`Redis rate limit failed (${(err as Error)?.message ?? err}); using the in-memory limiter`);\n      return this.fallback.hit(key, limit, cost);\n    }\n  }\n}\n"],"mappings":";;;;;;;;;AAeO,IAAM,sBAAsB;AAC5B,IAAM,uBAAuB;AAC7B,IAAM,sBAAsB;AAC5B,IAAM,kBAAkB;AACxB,IAAM,sBAAsB;AAC5B,IAAM,uBAAuB;AAC7B,IAAM,yBAAyB;AA2CtC,eAAe,cAA4B;AACzC,QAAM,OAAO;AACb,MAAI;AACF,UAAM,MAAW,MAAM;AAAA;AAAA,MAA0B;AAAA;AACjD,WAAO,IAAI,WAAW,IAAI,SAAS;AAAA,EACrC,QAAQ;AACN,UAAM,IAAI,MAAM,gDAAgD;AAAA,EAClE;AACF;AAEO,IAAM,cAAN,MAA4C;AAAA,EACjD;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACQ;AAAA,EACA,MAAkC;AAAA,EAClC,UAA0B;AAAA,EAC1B,cAAkC;AAAA,EAE1C,YAAY,OAA2B,CAAC,GAAG;AACzC,QAAI,CAAC,KAAK,OAAO,CAAC,KAAK,OAAQ,OAAM,IAAI,MAAM,qCAAqC;AACpF,SAAK,MAAM,KAAK;AAChB,SAAK,SAAS,KAAK,UAAU;AAC7B,SAAK,mBAAmB,KAAK,oBAAoB;AACjD,SAAK,eAAe,KAAK,gBAAgB;AACzC,SAAK,qBAAqB,KAAK,sBAAsB;AACrD,SAAK,iBAAiB,KAAK,kBAAkB,KAAK;AAAA,EACpD;AAAA,EAEA,eAAe,IAAuB;AACpC,SAAK,cAAc;AAAA,EACrB;AAAA,EAEA,MAAM,MAAM,SAAkB,WAAkC;AAC9D,SAAK,UAAU;AACf,QAAI,QAAQ;AACZ,QAAI;AACF,UAAI,CAAC,KAAK,QAAQ;AAChB,cAAM,QAAQ,MAAM,YAAY;AAChC,cAAM,IAAI,IAAI,MAAM,KAAK,KAAK,EAAE,aAAa,MAAM,sBAAsB,GAAG,gBAAgB,KAAM,eAAe,CAAC,MAAe,QAAQ,KAAK,IAAI,IAAI,KAAK,GAAI,IAAI,KAAM,CAAC;AAC1K,UAAE,GAAG,SAAS,MAAM,MAAS;AAC7B,aAAK,SAAS;AACd,cAAM,EAAE,QAAQ;AAAA,MAClB;AACA,YAAM,KAAK,OAAO,KAAK;AACvB,cAAQ;AACR,YAAM,MAAM,KAAK,OAAO,UAAU;AAClC,UAAI,GAAG,SAAS,MAAM,MAAS;AAC/B,UAAI,IAAI,SAAS;AACf,YAAI;AACF,gBAAM,IAAI,QAAQ;AAAA,QACpB,SAAS,KAAK;AACZ,cAAI,CAAC,WAAW,KAAK,OAAQ,KAAe,OAAO,CAAC,EAAG,OAAM;AAAA,QAC/D;AAAA,MACF;AACA,UAAI,GAAG,YAAY,CAAC,UAAkB,SAAiB,YAAoB;AACzE,aAAK,KAAK,QAAQ,SAAS,OAAO;AAAA,MACpC,CAAC;AACD,YAAM,IAAI,WAAW,sBAAsB,oBAAoB;AAC/D,WAAK,MAAM;AAAA,IACb,SAAS,KAAK;AACZ,aAAO,KAAK,sBAAuB,KAAe,WAAW,GAAG,0BAA0B;AAC1F,YAAM,IAAI,KAAK;AACf,WAAK,SAAS;AACd,WAAK,MAAM;AACX,UAAI;AACF,WAAG,aAAa;AAAA,MAClB,QAAQ;AAAA,MAER;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,OAAsB;AAC1B,eAAW,OAAO,CAAC,KAAK,KAAK,KAAK,MAAM,GAAG;AACzC,UAAI,CAAC,IAAK;AACV,UAAI;AACF,YAAI,IAAI,KAAM,OAAM,IAAI,KAAK;AAAA,YACxB,KAAI,aAAa;AAAA,MACxB,QAAQ;AACN,YAAI;AACF,cAAI,aAAa;AAAA,QACnB,QAAQ;AAAA,QAER;AAAA,MACF;AAAA,IACF;AACA,SAAK,MAAM;AACX,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,MAAc,QAAQ,SAAiB,KAA4B;AACjE,QAAI,OAAO,YAAY,YAAY,CAAC,IAAK;AACzC,QAAI,QAAQ,WAAW,mBAAmB,GAAG;AAC3C,YAAM,OAAO,QAAQ,MAAM,oBAAoB,MAAM;AACrD,UAAI,CAAC,KAAK,YAAa;AACvB,UAAI;AACF,cAAM,KAAK,YAAY,MAAM,KAAK,MAAM,GAAG,CAAC;AAAA,MAC9C,SAAS,KAAK;AACZ,eAAO,KAAK,iCAAiC,IAAI,YAAa,KAAe,WAAW,GAAG,EAAE;AAAA,MAC/F;AACA;AAAA,IACF;AACA,QAAI,CAAC,KAAK,QAAS;AACnB,UAAM,SAAS,QAAQ,MAAM,QAAQ,YAAY,GAAG,IAAI,CAAC;AACzD,QAAI;AACF,YAAM,KAAK,QAAQ,QAAQ,KAAK,MAAM,GAAG,CAAC;AAAA,IAC5C,SAAS,KAAK;AACZ,aAAO,KAAK,4BAA4B,MAAM,YAAa,KAAe,WAAW,GAAG,EAAE;AAAA,IAC5F;AAAA,EACF;AAAA;AAAA,EAIA,MAAM,UAAU,QAAgB,UAAiC;AAC/D,UAAM,IAAI,KAAK;AACf,QAAI,CAAC,EAAG;AACR,UAAM,MAAM,sBAAsB;AAClC,QAAI;AACF,YAAM,EAAE,KAAK,KAAK,UAAU,KAAK,MAAM,KAAK,IAAI,IAAI,GAAI,CAAC;AACzD,YAAM,EAAE,OAAO,KAAK,KAAK,kBAAkB;AAAA,IAC7C,SAAS,KAAK;AACZ,aAAO,KAAK,aAAa,MAAM,aAAc,KAAe,WAAW,GAAG,EAAE;AAAA,IAC9E;AAAA,EACF;AAAA,EAEA,MAAM,WAAW,QAAgB,UAAiC;AAChE,UAAM,IAAI,KAAK;AACf,QAAI,CAAC,EAAG;AACR,QAAI;AACF,YAAM,EAAE,KAAK,sBAAsB,QAAQ,QAAQ;AAAA,IACrD,SAAS,KAAK;AACZ,aAAO,KAAK,cAAc,MAAM,aAAc,KAAe,WAAW,GAAG,EAAE;AAAA,IAC/E;AAAA,EACF;AAAA,EAEA,MAAM,QAAQ,QAAgB,MAA8B;AAC1D,UAAM,IAAI,KAAK;AACf,QAAI,CAAC,EAAG;AACR,UAAM,MAAM,kBAAkB;AAC9B,QAAI;AACF,UAAI,KAAM,OAAM,EAAE,IAAI,KAAK,GAAG,MAAM,KAAK,cAAc;AAAA,UAClD,OAAM,EAAE,IAAI,GAAG;AAAA,IACtB,SAAS,KAAK;AACZ,aAAO,KAAK,WAAW,MAAM,aAAc,KAAe,WAAW,GAAG,EAAE;AAAA,IAC5E;AAAA,EACF;AAAA,EAEA,MAAM,eAAe,QAAmC;AACtD,UAAM,IAAI,KAAK;AACf,QAAI,CAAC,EAAG,QAAO,CAAC;AAChB,QAAI;AACJ,QAAI;AACF,gBAAU,MAAM,EAAE,QAAQ,sBAAsB,MAAM;AAAA,IACxD,SAAS,KAAK;AACZ,aAAO,KAAK,kBAAkB,MAAM,aAAc,KAAe,WAAW,GAAG,EAAE;AACjF,aAAO,CAAC;AAAA,IACV;AACA,UAAM,SAAS,KAAK,IAAI,IAAI,MAAO,KAAK;AACxC,WAAO,OAAO,QAAQ,WAAW,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,EAAE,IAAI,MAAM,OAAO,IAAI,KAAK,MAAM,EAAE,IAAI,CAAC,CAAC,MAAM,MAAM,MAAM;AAAA,EAC5G;AAAA,EAEA,MAAM,SAAS,QAAkC;AAC/C,QAAI,EAAE,MAAM,KAAK,eAAe,MAAM,GAAG,OAAQ,QAAO;AACxD,QAAI;AACF,aAAO,CAAE,MAAM,KAAK,OAAQ,OAAO,kBAAkB,MAAM;AAAA,IAC7D,QAAQ;AACN,aAAO;AAAA,IACT;AAAA,EACF;AAAA,EAEA,MAAM,eAAe,SAAyC;AAC5D,UAAM,IAAI,KAAK;AACf,QAAI,CAAC,KAAK,CAAC,QAAQ,OAAQ,QAAO,oBAAI,IAAI;AAC1C,QAAI;AACJ,QAAI;AACF,gBAAU,MAAM,QAAQ;AAAA,QACtB,QAAQ,IAAI,OAAO,QAAQ,CAAC,MAAM,EAAE,QAAQ,sBAAsB,GAAG,GAAG,MAAM,EAAE,OAAO,kBAAkB,GAAG,CAAC,CAAqC;AAAA,MACpJ;AAAA,IACF,SAAS,KAAK;AACZ,aAAO,KAAK,0BAA2B,KAAe,WAAW,GAAG,EAAE;AACtE,aAAO,oBAAI,IAAI;AAAA,IACjB;AACA,UAAM,SAAS,KAAK,IAAI,IAAI,MAAO,KAAK;AACxC,UAAM,SAAS,oBAAI,IAAY;AAC/B,YAAQ,QAAQ,CAAC,KAAK,MAAM;AAC1B,YAAM,CAAC,SAAS,IAAI,IAAI,QAAQ,CAAC;AACjC,UAAI,CAAC,QAAQ,OAAO,OAAO,WAAW,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,OAAO,CAAC,KAAK,MAAM,EAAG,QAAO,IAAI,GAAG;AAAA,IAC5F,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,QAAQ,QAAgB,SAAkB,QAA+B;AAC7E,UAAM,IAAI,KAAK;AACf,QAAI,CAAC,EAAG;AACR,QAAI;AACF,YAAM,EAAE,QAAQ,sBAAsB,QAAQ,KAAK,UAAU,EAAE,QAAQ,MAAM,QAAQ,CAAC,CAAC;AAAA,IACzF,SAAS,KAAK;AACZ,aAAO,KAAK,WAAW,MAAM,aAAc,KAAe,WAAW,GAAG,EAAE;AAAA,IAC5E;AAAA,EACF;AAAA,EAEA,MAAM,YAAY,MAAc,SAAkB,QAA+B;AAC/E,UAAM,IAAI,KAAK;AACf,QAAI,CAAC,EAAG;AACR,QAAI;AACF,YAAM,EAAE,QAAQ,sBAAsB,MAAM,KAAK,UAAU,EAAE,QAAQ,MAAM,QAAQ,CAAC,CAAC;AAAA,IACvF,SAAS,KAAK;AACZ,aAAO,KAAK,eAAe,IAAI,aAAc,KAAe,WAAW,GAAG,EAAE;AAAA,IAC9E;AAAA,EACF;AAAA;AAAA;AAAA,EAKA,MAAM,gBAAgB,QAAgB,YAAsC;AAC1E,UAAM,IAAI,KAAK;AACf,QAAI,CAAC,EAAG,QAAO;AACf,QAAI;AACF,aAAO,QAAQ,MAAM,EAAE,IAAI,yBAAyB,QAAQ,GAAG,MAAM,KAAK,IAAI,GAAG,KAAK,MAAM,aAAa,GAAI,CAAC,GAAG,IAAI,CAAC;AAAA,IACxH,SAAS,KAAK;AACZ,aAAO,KAAK,mBAAmB,MAAM,aAAc,KAAe,WAAW,GAAG,EAAE;AAClF,aAAO;AAAA,IACT;AAAA,EACF;AAAA,EAEA,MAAM,kBAAkB,QAA+B;AACrD,UAAM,IAAI,KAAK;AACf,QAAI,CAAC,EAAG;AACR,QAAI;AACF,YAAM,EAAE,IAAI,yBAAyB,MAAM;AAAA,IAC7C,SAAS,KAAK;AACZ,aAAO,KAAK,qBAAqB,MAAM,aAAc,KAAe,WAAW,GAAG,EAAE;AAAA,IACtF;AAAA,EACF;AACF;AAMO,IAAM,mBAAmB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAsBzB,IAAM,mBAAN,MAA0C;AAAA,EAE/C,YACU,QACD,SAAS,cAChB,UACA;AAHQ;AACD;AAGP,SAAK,WAAW,YAAY,IAAI,YAAY;AAAA,EAC9C;AAAA,EALU;AAAA,EACD;AAAA,EAHT;AAAA,EASQ,SAAiC;AACvC,UAAM,IAAS,KAAK;AACpB,QAAI,OAAO,EAAE,SAAS,WAAY,QAAO;AACzC,WAAO,EAAE,UAAU;AAAA,EACrB;AAAA,EAEA,MAAM,IAAI,KAAa,OAAqC,OAAO,GAAoB;AACrF,QAAI,CAAC,MAAO,QAAO;AACnB,UAAM,SAAS,KAAK,OAAO;AAC3B,QAAI,CAAC,OAAQ,QAAO,KAAK,SAAS,IAAI,KAAK,OAAO,IAAI;AACtD,UAAM,QAAQ,KAAK,MAAM,MAAM,UAAU,MAAO,CAAC,IAAI;AACrD,QAAI;AACF,YAAM,MAAM,MAAM,OAAO,KAAK,kBAAkB,GAAG,KAAK,SAAS,KAAK,MAAM,OAAO,OAAO,MAAM,IAAI,GAAG,OAAO,KAAK,IAAI,IAAI,GAAI,GAAG,OAAO,IAAI,GAAG,KAAK;AACrJ,YAAM,OAAO,OAAO,OAAO,SAAS,GAAG,IAAI,IAAI,SAAS,IAAI,GAAG;AAC/D,UAAI,OAAO,MAAM,IAAI,EAAG,OAAM,IAAI,MAAM,oBAAoB,OAAO,GAAG,CAAC,EAAE;AACzE,aAAO,KAAK,IAAI,GAAG,IAAI;AAAA,IACzB,SAAS,KAAK;AACZ,aAAO,KAAK,4BAA6B,KAAe,WAAW,GAAG,gCAAgC;AACtG,aAAO,KAAK,SAAS,IAAI,KAAK,OAAO,IAAI;AAAA,IAC3C;AAAA,EACF;AACF;","names":[]}

package/dist/s3.d.ts

@@ -0,0 +1,75 @@
+import { b as MediaStorage, M as MediaInfo } from './adapters-8ztjEYtz.js';
+
+declare const PRESIGN_MAX_SECONDS: number;
+/** The S3 operations the storage needs; the default implementation wraps @aws-sdk/client-s3 (inject a fake in tests). */
+interface S3Operations {
+    putObject(params: {
+        Bucket: string;
+        Key: string;
+        Body: Buffer;
+        ContentType: string;
+        [k: string]: unknown;
+    }): Promise<unknown>;
+    deleteObject(params: {
+        Bucket: string;
+        Key: string;
+    }): Promise<unknown>;
+    presignGet(params: {
+        Bucket: string;
+        Key: string;
+    }, expiresIn: number): Promise<string>;
+}
+declare function thumbnailKey(key: string): string;
+interface S3StorageOptions {
+    bucket: string;
+    region?: string;
+    /** An S3Operations implementation (default: built lazily from @aws-sdk/client-s3). */
+    client?: S3Operations;
+    prefix?: string;
+    publicBaseUrl?: string;
+    private?: boolean;
+    presignExpires?: number;
+    thumbnails?: boolean;
+    durations?: boolean;
+    /** Extra PutObject parameters (e.g. ServerSideEncryption, CacheControl). */
+    extraPutArgs?: Record<string, unknown>;
+}
+declare class S3Storage implements MediaStorage {
+    readonly bucket: string;
+    readonly region?: string;
+    readonly prefix: string;
+    readonly publicBaseUrl: string;
+    readonly private: boolean;
+    readonly presignExpires: number;
+    readonly thumbnails: boolean;
+    readonly durations: boolean;
+    readonly extraPutArgs: Record<string, unknown>;
+    private injected?;
+    private lazyOps;
+    constructor(opts: S3StorageOptions);
+    private ops;
+    makeKey(chatId: number, mediaType: string, fileName: string): string;
+    urlFor(key: string): Promise<string>;
+    /** Object key of a URL this storage produced (public or presigned); null otherwise. */
+    keyFor(fileUrl: string): string | null;
+    /** Fresh presigned GET URL for a stored file; null for URLs this storage did not produce. */
+    presign(fileUrl: string, expires?: number): Promise<string | null>;
+    private put;
+    upload(chatId: number, fileName: string, data: Buffer, mediaType: string): Promise<MediaInfo>;
+    /** Delete the object (and its thumbnail). Unknown URLs are ignored; S3 deletes are idempotent. */
+    delete(fileUrl: string): Promise<void>;
+}
+
+declare const THUMBNAIL_SIZE: readonly [number, number];
+declare const DURATION_MEDIA_TYPES: ReadonlySet<string>;
+declare const VIDEO_MEDIA_TYPES: ReadonlySet<string>;
+/** `<NAME>_PATH` env var if it points at a file, else the first match on PATH; null when not found. */
+declare function findTool(name: string, env?: NodeJS.ProcessEnv): string | null;
+/** JPEG thumbnail that fits in `size` (aspect kept, never enlarged); null without sharp or for a bad image. */
+declare function makeImageThumbnail(data: Buffer, size?: readonly [number, number], quality?: number): Promise<Buffer | null>;
+/** Duration in whole seconds via `ffprobe`; null when ffprobe is missing or fails. */
+declare function probeDuration(data: Buffer, suffix?: string, timeoutSeconds?: number): Promise<number | null>;
+/** JPEG of a frame near the start via `ffmpeg`; null when ffmpeg is missing or fails. */
+declare function makeVideoThumbnail(data: Buffer, suffix?: string, width?: number, timeoutSeconds?: number): Promise<Buffer | null>;
+
+export { DURATION_MEDIA_TYPES, PRESIGN_MAX_SECONDS, type S3Operations, S3Storage, type S3StorageOptions, THUMBNAIL_SIZE, VIDEO_MEDIA_TYPES, findTool, makeImageThumbnail, makeVideoThumbnail, probeDuration, thumbnailKey };