zen-code 1.0.0

package/dist/index-BUGHAUbY.mjs

@@ -0,0 +1,117 @@
+import "bun:sqlite";
+import { B as f, c as k } from "./chunk-YWE62C55-DID9N9eS.mjs";
+import { aY as m } from "./sql-CJsUpKEQ.mjs";
+import { EventEmitter as p } from "node:events";
+var l = "0", h = "1", w = "2", c = "3", u = "4", y = (e, t) => e.onmessage = ({ data: r }) => t(r), E = class extends k {
+  worker;
+  mitt;
+  constructor(e, t) {
+    super(async () => {
+      const r = await e();
+      this.mitt = r.mitt, this.worker = r.worker, r.handle(
+        this.worker,
+        ([s, ...i]) => this.mitt.emit(s, ...i)
+      ), this.worker.postMessage([
+        l,
+        r.data || {}
+      ]), await new Promise((s, i) => {
+        this.mitt.once(l, (a, o) => o ? i(o) : s());
+      }), this.conn = new W(this.worker, this.mitt), await t?.(this.conn);
+    });
+  }
+  async destroy() {
+    if (this.worker)
+      return this.worker.postMessage([w]), new Promise(
+        (e, t) => this.mitt?.once(
+          w,
+          (r, s) => s ? t(s) : e()
+        )
+      ).finally(() => {
+        this.worker?.terminate(), this.mitt?.off(), this.mitt = this.worker = void 0;
+      });
+  }
+}, W = class {
+  constructor(e, t) {
+    this.worker = e, this.mitt = t;
+  }
+  async *streamQuery({ parameters: e, sql: t, query: r }) {
+    this.worker.postMessage([
+      c,
+      m.is(r),
+      t,
+      e
+    ]);
+    let s = !1, i, a;
+    for (this.mitt.on(c, (o, n) => {
+      n ? a(n) : i([{ rows: [o] }, !1]);
+    }), this.mitt.on(u, (o, n) => {
+      n ? a(n) : i([void 0, !0]);
+    }); !s; ) {
+      const [o, n] = await new Promise((v, d) => {
+        i = v, a = d;
+      });
+      n ? (s = !0, this.mitt?.off(c), this.mitt?.off(u)) : yield o;
+    }
+  }
+  async executeQuery(e) {
+    const { parameters: t, sql: r, query: s } = e;
+    return this.worker.postMessage([
+      h,
+      m.is(s),
+      r,
+      t
+    ]), await new Promise((i, a) => {
+      this.mitt || a(new Error("kysely instance has been destroyed")), this.mitt.once(h, (o, n) => !n && o ? i(o) : a(n));
+    });
+  }
+}, q = class extends f {
+  /**
+   * Dialect for generic SQLite that run SQLs in worker thread
+   *
+   * @param executor function to create {@link IGenericSqliteWorkerExecutor}
+   * @param onCreateConnection optional callback after connection created
+   */
+  constructor(e, t) {
+    super(() => new E(e, t));
+  }
+}, x = class extends p {
+  off(e) {
+    return this.removeAllListeners(e);
+  }
+};
+function S() {
+  return new x();
+}
+var B = class extends q {
+  /**
+   * dialect for `bun:sqlite`, run sql in worker
+   */
+  constructor(e) {
+    const {
+      url: t = ":memory:",
+      cacheStatment: r = !1,
+      onCreateConnection: s,
+      worker: i = new Worker(
+        new URL(
+          /* @vite-ignore */
+          "/assets/worker-I4QjJYba.js",
+          import.meta.url
+        ),
+        { type: "module" }
+      ),
+      dbOptions: a = { create: !0 }
+    } = e || {};
+    super(
+      () => ({
+        data: { cache: r, fileName: t, opt: a },
+        mitt: S(),
+        handle: y,
+        worker: i
+      }),
+      s
+    );
+  }
+};
+export {
+  B as BunWorkerDialect
+};

package/dist/index-Butw72lT.mjs

@@ -0,0 +1,462 @@
+import { B as C, u as v, c as x, T as $, m as I } from "./app-CV_FJyjI.mjs";
+import { createClient as O, createCluster as N } from "redis";
+const W = [
+  {
+    index: "checkpoints",
+    prefix: "checkpoint:",
+    schema: {
+      "$.thread_id": {
+        type: "TAG",
+        AS: "thread_id"
+      },
+      "$.checkpoint_ns": {
+        type: "TAG",
+        AS: "checkpoint_ns"
+      },
+      "$.checkpoint_id": {
+        type: "TAG",
+        AS: "checkpoint_id"
+      },
+      "$.parent_checkpoint_id": {
+        type: "TAG",
+        AS: "parent_checkpoint_id"
+      },
+      "$.checkpoint_ts": {
+        type: "NUMERIC",
+        AS: "checkpoint_ts"
+      },
+      "$.has_writes": {
+        type: "TAG",
+        AS: "has_writes"
+      },
+      "$.source": {
+        type: "TAG",
+        AS: "source"
+      },
+      "$.step": {
+        type: "NUMERIC",
+        AS: "step"
+      }
+    }
+  },
+  {
+    index: "checkpoint_blobs",
+    prefix: "checkpoint_blob:",
+    schema: {
+      "$.thread_id": {
+        type: "TAG",
+        AS: "thread_id"
+      },
+      "$.checkpoint_ns": {
+        type: "TAG",
+        AS: "checkpoint_ns"
+      },
+      "$.checkpoint_id": {
+        type: "TAG",
+        AS: "checkpoint_id"
+      },
+      "$.channel": {
+        type: "TAG",
+        AS: "channel"
+      },
+      "$.version": {
+        type: "TAG",
+        AS: "version"
+      },
+      "$.type": {
+        type: "TAG",
+        AS: "type"
+      }
+    }
+  },
+  {
+    index: "checkpoint_writes",
+    prefix: "checkpoint_write:",
+    schema: {
+      "$.thread_id": {
+        type: "TAG",
+        AS: "thread_id"
+      },
+      "$.checkpoint_ns": {
+        type: "TAG",
+        AS: "checkpoint_ns"
+      },
+      "$.checkpoint_id": {
+        type: "TAG",
+        AS: "checkpoint_id"
+      },
+      "$.task_id": {
+        type: "TAG",
+        AS: "task_id"
+      },
+      "$.idx": {
+        type: "NUMERIC",
+        AS: "idx"
+      },
+      "$.channel": {
+        type: "TAG",
+        AS: "channel"
+      },
+      "$.type": {
+        type: "TAG",
+        AS: "type"
+      }
+    }
+  }
+];
+var j = class T extends C {
+  client;
+  ttlConfig;
+  constructor(e, t) {
+    super(), this.client = e, this.ttlConfig = t;
+  }
+  static async fromUrl(e, t) {
+    const s = O({ url: e });
+    await s.connect();
+    const i = new T(s, t);
+    return await i.ensureIndexes(), i;
+  }
+  static async fromCluster(e, t) {
+    const s = N({ rootNodes: e });
+    await s.connect();
+    const i = new T(s, t);
+    return await i.ensureIndexes(), i;
+  }
+  async get(e) {
+    return (await this.getTuple(e))?.checkpoint;
+  }
+  async getTuple(e) {
+    const t = e.configurable?.thread_id, s = e.configurable?.checkpoint_ns ?? "", i = e.configurable?.checkpoint_id;
+    if (!t) return;
+    let c, o;
+    if (i)
+      c = `checkpoint:${t}:${s}:${i}`, o = await this.client.json.get(c);
+    else {
+      const r = `checkpoint:${t}:${s}:*`, p = await this.client.keys(r);
+      if (p.length === 0) return;
+      p.sort(), c = p[p.length - 1], o = await this.client.json.get(c);
+    }
+    if (!o) return;
+    this.ttlConfig?.refreshOnRead && this.ttlConfig?.defaultTTL && await this.applyTTL(c);
+    const { checkpoint: f, pendingWrites: n } = await this.loadCheckpointWithWrites(o);
+    return await this.createCheckpointTuple(o, f, n);
+  }
+  async put(e, t, s, i) {
+    await this.ensureIndexes();
+    const c = e.configurable?.thread_id, o = e.configurable?.checkpoint_ns ?? "", f = e.configurable?.checkpoint_id;
+    if (!c) throw new Error("thread_id is required");
+    const n = t.id || v(0), r = `checkpoint:${c}:${o}:${n}`, p = x(t);
+    if (p.channel_values && i !== void 0) if (Object.keys(i).length === 0) p.channel_values = {};
+    else {
+      const u = {};
+      for (const a of Object.keys(i)) a in p.channel_values && (u[a] = p.channel_values[a]);
+      p.channel_values = u;
+    }
+    const l = {
+      thread_id: c,
+      checkpoint_ns: o === "" ? "__empty__" : o,
+      checkpoint_id: n,
+      parent_checkpoint_id: f || null,
+      checkpoint: p,
+      metadata: s,
+      checkpoint_ts: Date.now(),
+      has_writes: "false"
+    };
+    return this.addSearchableMetadataFields(l, s), await this.client.json.set(r, "$", l), this.ttlConfig?.defaultTTL && await this.applyTTL(r), { configurable: {
+      thread_id: c,
+      checkpoint_ns: o,
+      checkpoint_id: n
+    } };
+  }
+  async *list(e, t) {
+    if (await this.ensureIndexes(), t?.filter !== void 0) {
+      const i = Object.values(t.filter).some((n) => n === null), c = [];
+      if (e?.configurable?.thread_id) {
+        const n = e.configurable.thread_id.replace(/[-.@]/g, "\\$&");
+        c.push(`(@thread_id:{${n}})`);
+      }
+      if (e?.configurable?.checkpoint_ns !== void 0) {
+        const n = e.configurable.checkpoint_ns;
+        if (n === "") c.push("(@checkpoint_ns:{__empty__})");
+        else {
+          const r = n.replace(/[-.@]/g, "\\$&");
+          c.push(`(@checkpoint_ns:{${r}})`);
+        }
+      }
+      if (!t?.before && t?.filter)
+        for (const [n, r] of Object.entries(t.filter)) r === void 0 || r === null || (typeof r == "string" ? c.push(`(@${n}:{${r}})`) : typeof r == "number" ? c.push(`(@${n}:[${r} ${r}])`) : typeof r == "object" && Object.keys(r).length);
+      c.length === 0 && c.push("*");
+      const o = c.join(" "), f = t?.limit ?? 10;
+      try {
+        const n = t?.before && !e?.configurable?.thread_id ? 1e3 : t?.before ? f * 10 : f;
+        let p = (await this.client.ft.search("checkpoints", o, {
+          LIMIT: {
+            from: 0,
+            size: n
+          },
+          SORTBY: {
+            BY: "checkpoint_ts",
+            DIRECTION: "DESC"
+          }
+        })).documents, l = 0;
+        for (const u of p) {
+          if (l >= f) break;
+          if (t?.before?.configurable?.checkpoint_id) {
+            const k = u.value.checkpoint_id, d = t.before.configurable.checkpoint_id;
+            if (k >= d) continue;
+          }
+          const a = u.value;
+          let h = !0;
+          if ((i || t?.before) && t?.filter) {
+            for (const [k, d] of Object.entries(t.filter)) if (d === null) {
+              if (a.metadata?.[k] !== null) {
+                h = !1;
+                break;
+              }
+            } else if (d !== void 0) {
+              const b = a.metadata?.[k];
+              if (typeof d == "object" && d !== null) {
+                if (g(b) !== g(d)) {
+                  h = !1;
+                  break;
+                }
+              } else if (b !== d) {
+                h = !1;
+                break;
+              }
+            }
+            if (!h) continue;
+          }
+          const { checkpoint: y, pendingWrites: _ } = await this.loadCheckpointWithWrites(a);
+          yield await this.createCheckpointTuple(a, y, _), l++;
+        }
+        return;
+      } catch (n) {
+        if (!n.message?.includes("no such index")) throw n;
+      }
+      if (e?.configurable?.thread_id) {
+        const n = e.configurable.thread_id, r = e.configurable.checkpoint_ns ?? "", p = `checkpoint:${n}:${r}:*`, l = await this.client.keys(p);
+        l.sort().reverse();
+        let u = l;
+        if (t?.before?.configurable?.checkpoint_id) {
+          const y = t.before.configurable.thread_id || n, _ = t.before.configurable.checkpoint_ns ?? r, k = `checkpoint:${y}:${_}:${t.before.configurable.checkpoint_id}`, d = l.indexOf(k);
+          d > 0 ? u = l.slice(d + 1) : d === 0 && (u = []);
+        }
+        const a = t?.limit ?? 10, h = u.slice(0, a);
+        for (const y of h) {
+          const _ = await this.client.json.get(y);
+          if (_) {
+            let k = !0;
+            for (const [S, m] of Object.entries(t.filter)) {
+              const A = _.metadata?.[S];
+              if (m === null) {
+                if (A !== null) {
+                  k = !1;
+                  break;
+                }
+              } else if (A !== m) {
+                k = !1;
+                break;
+              }
+            }
+            if (!k) continue;
+            const { checkpoint: d, pendingWrites: b } = await this.loadCheckpointWithWrites(_);
+            yield await this.createCheckpointTuple(_, d, b);
+          }
+        }
+      } else {
+        const n = e?.configurable?.checkpoint_ns !== void 0 ? `checkpoint:*:${e.configurable.checkpoint_ns === "" ? "__empty__" : e.configurable.checkpoint_ns}:*` : "checkpoint:*", r = await this.client.keys(n), p = [];
+        for (const a of r) {
+          const h = await this.client.json.get(a);
+          h && p.push({
+            key: a,
+            doc: h
+          });
+        }
+        p.sort((a, h) => h.doc.checkpoint_ts - a.doc.checkpoint_ts);
+        let l = 0;
+        const u = t?.limit ?? 10;
+        for (const { doc: a } of p) {
+          if (l >= u) break;
+          if (t?.before?.configurable?.checkpoint_id) {
+            const k = a.checkpoint_id, d = t.before.configurable.checkpoint_id;
+            if (k >= d) continue;
+          }
+          let h = !0;
+          if (t?.filter) {
+            for (const [k, d] of Object.entries(t.filter)) if (d === null) {
+              if (a.metadata?.[k] !== null) {
+                h = !1;
+                break;
+              }
+            } else if (d !== void 0) {
+              const b = a.metadata?.[k];
+              if (typeof d == "object" && d !== null) {
+                if (g(b) !== g(d)) {
+                  h = !1;
+                  break;
+                }
+              } else if (b !== d) {
+                h = !1;
+                break;
+              }
+            }
+            if (!h) continue;
+          }
+          const { checkpoint: y, pendingWrites: _ } = await this.loadCheckpointWithWrites(a);
+          yield await this.createCheckpointTuple(a, y, _), l++;
+        }
+      }
+      return;
+    }
+    const s = {
+      ...t,
+      filter: {}
+    };
+    yield* this.list(e, s);
+  }
+  async putWrites(e, t, s) {
+    await this.ensureIndexes();
+    const i = e.configurable?.thread_id, c = e.configurable?.checkpoint_ns ?? "", o = e.configurable?.checkpoint_id;
+    if (!i || !o) throw new Error("thread_id and checkpoint_id are required");
+    const f = [], n = performance.now() * 1e3;
+    for (let l = 0; l < t.length; l++) {
+      const [u, a] = t[l], h = `checkpoint_write:${i}:${c}:${o}:${s}:${l}`;
+      f.push(h);
+      const y = {
+        thread_id: i,
+        checkpoint_ns: c,
+        checkpoint_id: o,
+        task_id: s,
+        idx: l,
+        channel: u,
+        type: typeof a == "object" ? "json" : "string",
+        value: a,
+        timestamp: n,
+        global_idx: n + l
+      };
+      await this.client.json.set(h, "$", y);
+    }
+    if (f.length > 0) {
+      const l = `write_keys_zset:${i}:${c}:${o}`, u = {};
+      f.forEach((a, h) => {
+        u[a] = n + h;
+      }), await this.client.zAdd(l, Object.entries(u).map(([a, h]) => ({
+        score: h,
+        value: a
+      }))), this.ttlConfig?.defaultTTL && await this.applyTTL(...f, l);
+    }
+    const r = `checkpoint:${i}:${c}:${o}`;
+    if (await this.client.exists(r)) {
+      const l = await this.client.json.get(r);
+      l && (l.has_writes = "true", await this.client.json.set(r, "$", l));
+    }
+  }
+  async deleteThread(e) {
+    const t = `checkpoint:${e}:*`, s = await this.client.keys(t);
+    s.length > 0 && await this.client.del(s);
+    const i = `writes:${e}:*`, c = await this.client.keys(i);
+    c.length > 0 && await this.client.del(c);
+  }
+  async end() {
+    await this.client.quit();
+  }
+  async loadPendingWrites(e, t, s) {
+    const i = `checkpoint_write:${e}:${t}:${s}:*`, c = await this.client.keys(i);
+    if (c.length === 0) return;
+    const o = [];
+    for (const n of c) {
+      const r = await this.client.json.get(n);
+      r && o.push(r);
+    }
+    o.sort((n, r) => (n.global_idx || 0) - (r.global_idx || 0));
+    const f = [];
+    for (const n of o) {
+      const r = await this.serde.loadsTyped("json", JSON.stringify(n.value));
+      f.push([
+        n.task_id,
+        n.channel,
+        r
+      ]);
+    }
+    return f;
+  }
+  async loadCheckpointWithWrites(e) {
+    const t = await this.serde.loadsTyped("json", JSON.stringify(e.checkpoint));
+    if (t.v < 4 && e.parent_checkpoint_id != null) {
+      const i = e.checkpoint_ns === "__empty__" ? "" : e.checkpoint_ns;
+      await this.migratePendingSends(t, e.thread_id, i, e.parent_checkpoint_id);
+    }
+    let s;
+    if (e.has_writes === "true") {
+      const i = e.checkpoint_ns === "__empty__" ? "" : e.checkpoint_ns;
+      s = await this.loadPendingWrites(e.thread_id, i, e.checkpoint_id);
+    }
+    return {
+      checkpoint: t,
+      pendingWrites: s
+    };
+  }
+  async migratePendingSends(e, t, s, i) {
+    const c = await this.loadPendingWrites(t, s, i);
+    if (!c || c.length === 0) return;
+    const o = c.filter(([, n]) => n === $);
+    if (o.length === 0) return;
+    const f = [];
+    for (const [, , n] of o) f.push(n);
+    e.channel_values ??= {}, e.channel_values[$] = f, e.channel_versions[$] = Object.keys(e.channel_versions).length > 0 ? I(...Object.values(e.channel_versions)) : 1;
+  }
+  async createCheckpointTuple(e, t, s) {
+    const i = e.checkpoint_ns === "__empty__" ? "" : e.checkpoint_ns, c = await this.serde.loadsTyped("json", JSON.stringify(e.metadata));
+    return {
+      config: { configurable: {
+        thread_id: e.thread_id,
+        checkpoint_ns: i,
+        checkpoint_id: e.checkpoint_id
+      } },
+      checkpoint: t,
+      metadata: c,
+      parentConfig: e.parent_checkpoint_id ? { configurable: {
+        thread_id: e.thread_id,
+        checkpoint_ns: i,
+        checkpoint_id: e.parent_checkpoint_id
+      } } : void 0,
+      pendingWrites: s
+    };
+  }
+  addSearchableMetadataFields(e, t) {
+    t && ("source" in t && (e.source = t.source), "step" in t && (e.step = t.step), "writes" in t && (e.writes = typeof t.writes == "object" ? JSON.stringify(t.writes) : t.writes), "score" in t && (e.score = t.score));
+  }
+  async applyTTL(...e) {
+    if (!this.ttlConfig?.defaultTTL) return;
+    const t = Math.floor(this.ttlConfig.defaultTTL * 60), s = await Promise.allSettled(e.map((i) => this.client.expire(i, t)));
+    for (let i = 0; i < s.length; i++) s[i].status === "rejected" && console.warn(`Failed to set TTL for key ${e[i]}:`, s[i].reason);
+  }
+  async ensureIndexes() {
+    for (const e of W) try {
+      await this.client.ft.create(e.index, e.schema, {
+        ON: "JSON",
+        PREFIX: e.prefix
+      });
+    } catch (t) {
+      t.message?.includes("Index already exists") || console.error(`Failed to create index ${e.index}:`, t.message);
+    }
+  }
+};
+function g(w) {
+  if (w === null || typeof w != "object") return JSON.stringify(w);
+  if (Array.isArray(w)) return JSON.stringify(w.map((s) => g(s)));
+  const e = {}, t = Object.keys(w).sort();
+  for (const s of t) e[s] = w[s];
+  return JSON.stringify(e, (s, i) => {
+    if (i !== null && typeof i == "object" && !Array.isArray(i)) {
+      const c = {}, o = Object.keys(i).sort();
+      for (const f of o) c[f] = i[f];
+      return c;
+    }
+    return i;
+  });
+}
+export {
+  j as RedisSaver
+};

package/dist/index-C5j-48Ft.mjs

@@ -0,0 +1,75 @@
+import { B as i, c as o, d as c } from "./chunk-YWE62C55-DID9N9eS.mjs";
+import { aY as a } from "./sql-CJsUpKEQ.mjs";
+var u = class extends o {
+  db;
+  constructor(t, e) {
+    super(async () => {
+      this.db = await t(), this.conn = new l(this.db), await e?.(this.conn);
+    });
+  }
+  async destroy() {
+    await this.db?.close();
+  }
+}, l = class {
+  constructor(t) {
+    this.db = t;
+  }
+  async *streamQuery({ parameters: t, query: e, sql: r }) {
+    if (!this.db.iterator)
+      throw new Error("streamQuery() is not supported.");
+    const s = this.db.iterator(a.is(e), r, t);
+    for await (const n of s)
+      yield { rows: [n] };
+  }
+  async executeQuery({ parameters: t, query: e, sql: r }) {
+    return await this.db.query(a.is(e), r, t);
+  }
+}, d = class extends i {
+  /**
+   * Dialect for generic SQLite that run SQLs in current thread
+   *
+   * @param executor function to create {@link IGenericSqlite}
+   * @param onCreateConnection optional callback after connection created
+   */
+  constructor(t, e) {
+    super(() => new u(t, e));
+  }
+};
+async function y(t) {
+  return typeof t == "function" ? await t() : t;
+}
+var h = class extends d {
+  /**
+   * dialect for [node sqlite3 wasm](https://github.com/tndrle/node-sqlite3-wasm)
+   */
+  constructor(t) {
+    super(
+      async () => {
+        const e = await y(t.database);
+        return {
+          db: e,
+          close: () => e.close(),
+          query: c({
+            all: (r, s) => e.all(r, s),
+            run: (r) => {
+              const { changes: s, lastInsertRowid: n } = e.run(r);
+              return {
+                insertId: BigInt(n),
+                numAffectedRows: BigInt(s)
+              };
+            }
+          }),
+          iterator: (r, s, n) => {
+            if (!r)
+              throw new Error("Only support select query");
+            return e.prepare(s).iterate(n);
+          }
+        };
+      },
+      t.onCreateConnection
+    );
+  }
+};
+export {
+  h as NodeWasmDialect
+};