@chat-adapter/state-redis 4.0.0

package/dist/index.d.ts

@@ -0,0 +1,44 @@
+import { StateAdapter, Lock } from 'chat';
+import { RedisClientType } from 'redis';
+
+interface RedisStateAdapterOptions {
+    /** Redis connection URL (e.g., redis://localhost:6379) */
+    url: string;
+    /** Key prefix for all Redis keys (default: "chat-sdk") */
+    keyPrefix?: string;
+}
+/**
+ * Redis state adapter for production use.
+ *
+ * Provides persistent subscriptions and distributed locking
+ * across multiple server instances.
+ */
+declare class RedisStateAdapter implements StateAdapter {
+    private client;
+    private keyPrefix;
+    private connected;
+    private connectPromise;
+    constructor(options: RedisStateAdapterOptions);
+    private key;
+    private subscriptionsSetKey;
+    connect(): Promise<void>;
+    disconnect(): Promise<void>;
+    subscribe(threadId: string): Promise<void>;
+    unsubscribe(threadId: string): Promise<void>;
+    isSubscribed(threadId: string): Promise<boolean>;
+    listSubscriptions(adapterName?: string): AsyncIterable<string>;
+    acquireLock(threadId: string, ttlMs: number): Promise<Lock | null>;
+    releaseLock(lock: Lock): Promise<void>;
+    extendLock(lock: Lock, ttlMs: number): Promise<boolean>;
+    get<T = unknown>(key: string): Promise<T | null>;
+    set<T = unknown>(key: string, value: T, ttlMs?: number): Promise<void>;
+    delete(key: string): Promise<void>;
+    private ensureConnected;
+    /**
+     * Get the underlying Redis client for advanced usage.
+     */
+    getClient(): RedisClientType;
+}
+declare function createRedisState(options: RedisStateAdapterOptions): RedisStateAdapter;
+
+export { RedisStateAdapter, type RedisStateAdapterOptions, createRedisState };

package/dist/index.js

@@ -0,0 +1,174 @@
+// src/index.ts
+import { createClient } from "redis";
+var RedisStateAdapter = class {
+  client;
+  keyPrefix;
+  connected = false;
+  connectPromise = null;
+  constructor(options) {
+    this.client = createClient({ url: options.url });
+    this.keyPrefix = options.keyPrefix || "chat-sdk";
+    this.client.on("error", (err) => {
+      console.error("[chat-sdk] Redis client error:", err);
+    });
+  }
+  key(type, id) {
+    return `${this.keyPrefix}:${type}:${id}`;
+  }
+  subscriptionsSetKey() {
+    return `${this.keyPrefix}:subscriptions`;
+  }
+  async connect() {
+    if (this.connected) {
+      return;
+    }
+    if (!this.connectPromise) {
+      this.connectPromise = this.client.connect().then(() => {
+        this.connected = true;
+      });
+    }
+    await this.connectPromise;
+  }
+  async disconnect() {
+    if (this.connected) {
+      await this.client.quit();
+      this.connected = false;
+      this.connectPromise = null;
+    }
+  }
+  async subscribe(threadId) {
+    this.ensureConnected();
+    await this.client.sAdd(this.subscriptionsSetKey(), threadId);
+  }
+  async unsubscribe(threadId) {
+    this.ensureConnected();
+    await this.client.sRem(this.subscriptionsSetKey(), threadId);
+  }
+  async isSubscribed(threadId) {
+    this.ensureConnected();
+    return this.client.sIsMember(this.subscriptionsSetKey(), threadId);
+  }
+  async *listSubscriptions(adapterName) {
+    this.ensureConnected();
+    let cursor = 0;
+    do {
+      const result = await this.client.sScan(
+        this.subscriptionsSetKey(),
+        cursor,
+        {
+          COUNT: 100
+        }
+      );
+      cursor = result.cursor;
+      for (const threadId of result.members) {
+        if (adapterName) {
+          if (threadId.startsWith(`${adapterName}:`)) {
+            yield threadId;
+          }
+        } else {
+          yield threadId;
+        }
+      }
+    } while (cursor !== 0);
+  }
+  async acquireLock(threadId, ttlMs) {
+    this.ensureConnected();
+    const token = generateToken();
+    const lockKey = this.key("lock", threadId);
+    const acquired = await this.client.set(lockKey, token, {
+      NX: true,
+      PX: ttlMs
+    });
+    if (acquired) {
+      return {
+        threadId,
+        token,
+        expiresAt: Date.now() + ttlMs
+      };
+    }
+    return null;
+  }
+  async releaseLock(lock) {
+    this.ensureConnected();
+    const lockKey = this.key("lock", lock.threadId);
+    const script = `
+      if redis.call("get", KEYS[1]) == ARGV[1] then
+        return redis.call("del", KEYS[1])
+      else
+        return 0
+      end
+    `;
+    await this.client.eval(script, {
+      keys: [lockKey],
+      arguments: [lock.token]
+    });
+  }
+  async extendLock(lock, ttlMs) {
+    this.ensureConnected();
+    const lockKey = this.key("lock", lock.threadId);
+    const script = `
+      if redis.call("get", KEYS[1]) == ARGV[1] then
+        return redis.call("pexpire", KEYS[1], ARGV[2])
+      else
+        return 0
+      end
+    `;
+    const result = await this.client.eval(script, {
+      keys: [lockKey],
+      arguments: [lock.token, ttlMs.toString()]
+    });
+    return result === 1;
+  }
+  async get(key) {
+    this.ensureConnected();
+    const cacheKey = this.key("cache", key);
+    const value = await this.client.get(cacheKey);
+    if (value === null) {
+      return null;
+    }
+    try {
+      return JSON.parse(value);
+    } catch {
+      return value;
+    }
+  }
+  async set(key, value, ttlMs) {
+    this.ensureConnected();
+    const cacheKey = this.key("cache", key);
+    const serialized = JSON.stringify(value);
+    if (ttlMs) {
+      await this.client.set(cacheKey, serialized, { PX: ttlMs });
+    } else {
+      await this.client.set(cacheKey, serialized);
+    }
+  }
+  async delete(key) {
+    this.ensureConnected();
+    const cacheKey = this.key("cache", key);
+    await this.client.del(cacheKey);
+  }
+  ensureConnected() {
+    if (!this.connected) {
+      throw new Error(
+        "RedisStateAdapter is not connected. Call connect() first."
+      );
+    }
+  }
+  /**
+   * Get the underlying Redis client for advanced usage.
+   */
+  getClient() {
+    return this.client;
+  }
+};
+function generateToken() {
+  return `redis_${Date.now()}_${Math.random().toString(36).substring(2, 15)}`;
+}
+function createRedisState(options) {
+  return new RedisStateAdapter(options);
+}
+export {
+  RedisStateAdapter,
+  createRedisState
+};
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import type { Lock, StateAdapter } from \"chat\";\nimport { createClient, type RedisClientType } from \"redis\";\n\nexport interface RedisStateAdapterOptions {\n  /** Redis connection URL (e.g., redis://localhost:6379) */\n  url: string;\n  /** Key prefix for all Redis keys (default: \"chat-sdk\") */\n  keyPrefix?: string;\n}\n\n/**\n * Redis state adapter for production use.\n *\n * Provides persistent subscriptions and distributed locking\n * across multiple server instances.\n */\nexport class RedisStateAdapter implements StateAdapter {\n  private client: RedisClientType;\n  private keyPrefix: string;\n  private connected = false;\n  private connectPromise: Promise<void> | null = null;\n\n  constructor(options: RedisStateAdapterOptions) {\n    this.client = createClient({ url: options.url });\n    this.keyPrefix = options.keyPrefix || \"chat-sdk\";\n\n    // Handle connection errors\n    this.client.on(\"error\", (err) => {\n      console.error(\"[chat-sdk] Redis client error:\", err);\n    });\n  }\n\n  private key(type: \"sub\" | \"lock\" | \"cache\", id: string): string {\n    return `${this.keyPrefix}:${type}:${id}`;\n  }\n\n  private subscriptionsSetKey(): string {\n    return `${this.keyPrefix}:subscriptions`;\n  }\n\n  async connect(): Promise<void> {\n    if (this.connected) {\n      return;\n    }\n\n    // Reuse existing connection attempt to avoid race conditions\n    if (!this.connectPromise) {\n      this.connectPromise = this.client.connect().then(() => {\n        this.connected = true;\n      });\n    }\n\n    await this.connectPromise;\n  }\n\n  async disconnect(): Promise<void> {\n    if (this.connected) {\n      await this.client.quit();\n      this.connected = false;\n      this.connectPromise = null;\n    }\n  }\n\n  async subscribe(threadId: string): Promise<void> {\n    this.ensureConnected();\n    await this.client.sAdd(this.subscriptionsSetKey(), threadId);\n  }\n\n  async unsubscribe(threadId: string): Promise<void> {\n    this.ensureConnected();\n    await this.client.sRem(this.subscriptionsSetKey(), threadId);\n  }\n\n  async isSubscribed(threadId: string): Promise<boolean> {\n    this.ensureConnected();\n    return this.client.sIsMember(this.subscriptionsSetKey(), threadId);\n  }\n\n  async *listSubscriptions(adapterName?: string): AsyncIterable<string> {\n    this.ensureConnected();\n\n    // Use SSCAN for large sets to avoid blocking\n    let cursor = 0;\n    do {\n      const result = await this.client.sScan(\n        this.subscriptionsSetKey(),\n        cursor,\n        {\n          COUNT: 100,\n        },\n      );\n      cursor = result.cursor;\n\n      for (const threadId of result.members) {\n        if (adapterName) {\n          if (threadId.startsWith(`${adapterName}:`)) {\n            yield threadId;\n          }\n        } else {\n          yield threadId;\n        }\n      }\n    } while (cursor !== 0);\n  }\n\n  async acquireLock(threadId: string, ttlMs: number): Promise<Lock | null> {\n    this.ensureConnected();\n\n    const token = generateToken();\n    const lockKey = this.key(\"lock\", threadId);\n\n    // Use SET NX EX for atomic lock acquisition\n    const acquired = await this.client.set(lockKey, token, {\n      NX: true,\n      PX: ttlMs,\n    });\n\n    if (acquired) {\n      return {\n        threadId,\n        token,\n        expiresAt: Date.now() + ttlMs,\n      };\n    }\n\n    return null;\n  }\n\n  async releaseLock(lock: Lock): Promise<void> {\n    this.ensureConnected();\n\n    const lockKey = this.key(\"lock\", lock.threadId);\n\n    // Use Lua script for atomic check-and-delete\n    const script = `\n      if redis.call(\"get\", KEYS[1]) == ARGV[1] then\n        return redis.call(\"del\", KEYS[1])\n      else\n        return 0\n      end\n    `;\n\n    await this.client.eval(script, {\n      keys: [lockKey],\n      arguments: [lock.token],\n    });\n  }\n\n  async extendLock(lock: Lock, ttlMs: number): Promise<boolean> {\n    this.ensureConnected();\n\n    const lockKey = this.key(\"lock\", lock.threadId);\n\n    // Use Lua script for atomic check-and-extend\n    const script = `\n      if redis.call(\"get\", KEYS[1]) == ARGV[1] then\n        return redis.call(\"pexpire\", KEYS[1], ARGV[2])\n      else\n        return 0\n      end\n    `;\n\n    const result = await this.client.eval(script, {\n      keys: [lockKey],\n      arguments: [lock.token, ttlMs.toString()],\n    });\n\n    return result === 1;\n  }\n\n  async get<T = unknown>(key: string): Promise<T | null> {\n    this.ensureConnected();\n\n    const cacheKey = this.key(\"cache\", key);\n    const value = await this.client.get(cacheKey);\n\n    if (value === null) {\n      return null;\n    }\n\n    try {\n      return JSON.parse(value) as T;\n    } catch {\n      // If parsing fails, return as string\n      return value as unknown as T;\n    }\n  }\n\n  async set<T = unknown>(key: string, value: T, ttlMs?: number): Promise<void> {\n    this.ensureConnected();\n\n    const cacheKey = this.key(\"cache\", key);\n    const serialized = JSON.stringify(value);\n\n    if (ttlMs) {\n      await this.client.set(cacheKey, serialized, { PX: ttlMs });\n    } else {\n      await this.client.set(cacheKey, serialized);\n    }\n  }\n\n  async delete(key: string): Promise<void> {\n    this.ensureConnected();\n\n    const cacheKey = this.key(\"cache\", key);\n    await this.client.del(cacheKey);\n  }\n\n  private ensureConnected(): void {\n    if (!this.connected) {\n      throw new Error(\n        \"RedisStateAdapter is not connected. Call connect() first.\",\n      );\n    }\n  }\n\n  /**\n   * Get the underlying Redis client for advanced usage.\n   */\n  getClient(): RedisClientType {\n    return this.client;\n  }\n}\n\nfunction generateToken(): string {\n  return `redis_${Date.now()}_${Math.random().toString(36).substring(2, 15)}`;\n}\n\nexport function createRedisState(\n  options: RedisStateAdapterOptions,\n): RedisStateAdapter {\n  return new RedisStateAdapter(options);\n}\n"],"mappings":";AACA,SAAS,oBAA0C;AAe5C,IAAM,oBAAN,MAAgD;AAAA,EAC7C;AAAA,EACA;AAAA,EACA,YAAY;AAAA,EACZ,iBAAuC;AAAA,EAE/C,YAAY,SAAmC;AAC7C,SAAK,SAAS,aAAa,EAAE,KAAK,QAAQ,IAAI,CAAC;AAC/C,SAAK,YAAY,QAAQ,aAAa;AAGtC,SAAK,OAAO,GAAG,SAAS,CAAC,QAAQ;AAC/B,cAAQ,MAAM,kCAAkC,GAAG;AAAA,IACrD,CAAC;AAAA,EACH;AAAA,EAEQ,IAAI,MAAgC,IAAoB;AAC9D,WAAO,GAAG,KAAK,SAAS,IAAI,IAAI,IAAI,EAAE;AAAA,EACxC;AAAA,EAEQ,sBAA8B;AACpC,WAAO,GAAG,KAAK,SAAS;AAAA,EAC1B;AAAA,EAEA,MAAM,UAAyB;AAC7B,QAAI,KAAK,WAAW;AAClB;AAAA,IACF;AAGA,QAAI,CAAC,KAAK,gBAAgB;AACxB,WAAK,iBAAiB,KAAK,OAAO,QAAQ,EAAE,KAAK,MAAM;AACrD,aAAK,YAAY;AAAA,MACnB,CAAC;AAAA,IACH;AAEA,UAAM,KAAK;AAAA,EACb;AAAA,EAEA,MAAM,aAA4B;AAChC,QAAI,KAAK,WAAW;AAClB,YAAM,KAAK,OAAO,KAAK;AACvB,WAAK,YAAY;AACjB,WAAK,iBAAiB;AAAA,IACxB;AAAA,EACF;AAAA,EAEA,MAAM,UAAU,UAAiC;AAC/C,SAAK,gBAAgB;AACrB,UAAM,KAAK,OAAO,KAAK,KAAK,oBAAoB,GAAG,QAAQ;AAAA,EAC7D;AAAA,EAEA,MAAM,YAAY,UAAiC;AACjD,SAAK,gBAAgB;AACrB,UAAM,KAAK,OAAO,KAAK,KAAK,oBAAoB,GAAG,QAAQ;AAAA,EAC7D;AAAA,EAEA,MAAM,aAAa,UAAoC;AACrD,SAAK,gBAAgB;AACrB,WAAO,KAAK,OAAO,UAAU,KAAK,oBAAoB,GAAG,QAAQ;AAAA,EACnE;AAAA,EAEA,OAAO,kBAAkB,aAA6C;AACpE,SAAK,gBAAgB;AAGrB,QAAI,SAAS;AACb,OAAG;AACD,YAAM,SAAS,MAAM,KAAK,OAAO;AAAA,QAC/B,KAAK,oBAAoB;AAAA,QACzB;AAAA,QACA;AAAA,UACE,OAAO;AAAA,QACT;AAAA,MACF;AACA,eAAS,OAAO;AAEhB,iBAAW,YAAY,OAAO,SAAS;AACrC,YAAI,aAAa;AACf,cAAI,SAAS,WAAW,GAAG,WAAW,GAAG,GAAG;AAC1C,kBAAM;AAAA,UACR;AAAA,QACF,OAAO;AACL,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,IACF,SAAS,WAAW;AAAA,EACtB;AAAA,EAEA,MAAM,YAAY,UAAkB,OAAqC;AACvE,SAAK,gBAAgB;AAErB,UAAM,QAAQ,cAAc;AAC5B,UAAM,UAAU,KAAK,IAAI,QAAQ,QAAQ;AAGzC,UAAM,WAAW,MAAM,KAAK,OAAO,IAAI,SAAS,OAAO;AAAA,MACrD,IAAI;AAAA,MACJ,IAAI;AAAA,IACN,CAAC;AAED,QAAI,UAAU;AACZ,aAAO;AAAA,QACL;AAAA,QACA;AAAA,QACA,WAAW,KAAK,IAAI,IAAI;AAAA,MAC1B;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,YAAY,MAA2B;AAC3C,SAAK,gBAAgB;AAErB,UAAM,UAAU,KAAK,IAAI,QAAQ,KAAK,QAAQ;AAG9C,UAAM,SAAS;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAQf,UAAM,KAAK,OAAO,KAAK,QAAQ;AAAA,MAC7B,MAAM,CAAC,OAAO;AAAA,MACd,WAAW,CAAC,KAAK,KAAK;AAAA,IACxB,CAAC;AAAA,EACH;AAAA,EAEA,MAAM,WAAW,MAAY,OAAiC;AAC5D,SAAK,gBAAgB;AAErB,UAAM,UAAU,KAAK,IAAI,QAAQ,KAAK,QAAQ;AAG9C,UAAM,SAAS;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAQf,UAAM,SAAS,MAAM,KAAK,OAAO,KAAK,QAAQ;AAAA,MAC5C,MAAM,CAAC,OAAO;AAAA,MACd,WAAW,CAAC,KAAK,OAAO,MAAM,SAAS,CAAC;AAAA,IAC1C,CAAC;AAED,WAAO,WAAW;AAAA,EACpB;AAAA,EAEA,MAAM,IAAiB,KAAgC;AACrD,SAAK,gBAAgB;AAErB,UAAM,WAAW,KAAK,IAAI,SAAS,GAAG;AACtC,UAAM,QAAQ,MAAM,KAAK,OAAO,IAAI,QAAQ;AAE5C,QAAI,UAAU,MAAM;AAClB,aAAO;AAAA,IACT;AAEA,QAAI;AACF,aAAO,KAAK,MAAM,KAAK;AAAA,IACzB,QAAQ;AAEN,aAAO;AAAA,IACT;AAAA,EACF;AAAA,EAEA,MAAM,IAAiB,KAAa,OAAU,OAA+B;AAC3E,SAAK,gBAAgB;AAErB,UAAM,WAAW,KAAK,IAAI,SAAS,GAAG;AACtC,UAAM,aAAa,KAAK,UAAU,KAAK;AAEvC,QAAI,OAAO;AACT,YAAM,KAAK,OAAO,IAAI,UAAU,YAAY,EAAE,IAAI,MAAM,CAAC;AAAA,IAC3D,OAAO;AACL,YAAM,KAAK,OAAO,IAAI,UAAU,UAAU;AAAA,IAC5C;AAAA,EACF;AAAA,EAEA,MAAM,OAAO,KAA4B;AACvC,SAAK,gBAAgB;AAErB,UAAM,WAAW,KAAK,IAAI,SAAS,GAAG;AACtC,UAAM,KAAK,OAAO,IAAI,QAAQ;AAAA,EAChC;AAAA,EAEQ,kBAAwB;AAC9B,QAAI,CAAC,KAAK,WAAW;AACnB,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,YAA6B;AAC3B,WAAO,KAAK;AAAA,EACd;AACF;AAEA,SAAS,gBAAwB;AAC/B,SAAO,SAAS,KAAK,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,SAAS,EAAE,EAAE,UAAU,GAAG,EAAE,CAAC;AAC3E;AAEO,SAAS,iBACd,SACmB;AACnB,SAAO,IAAI,kBAAkB,OAAO;AACtC;","names":[]}

package/package.json

@@ -0,0 +1,47 @@
+{
+  "name": "@chat-adapter/state-redis",
+  "version": "4.0.0",
+  "description": "Redis state adapter for chat (production)",
+  "type": "module",
+  "main": "./dist/index.js",
+  "module": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "dependencies": {
+    "redis": "^4.7.0",
+    "chat": "4.0.0"
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "devDependencies": {
+    "@types/node": "^22.10.2",
+    "tsup": "^8.3.5",
+    "typescript": "^5.7.2",
+    "vitest": "^2.1.8"
+  },
+  "keywords": [
+    "chat",
+    "state",
+    "redis",
+    "production"
+  ],
+  "license": "MIT",
+  "scripts": {
+    "build": "tsup",
+    "dev": "tsup --watch",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "typecheck": "tsc --noEmit",
+    "lint": "biome check src",
+    "clean": "rm -rf dist"
+  }
+}