@arbitro/client 0.2.0 → 0.4.1

package/dist/index.mjs

@@ -1,3 +1,18 @@
+import {
+  frame,
+  packDisconnect,
+  packHello,
+  packPublish,
+  packPublishBatch,
+  packPublishWithReply
+} from "./chunk-SKCXQO7R.mjs";
+import {
+  HEADER_SIZE,
+  OFF_ACTION,
+  OFF_MSG_LEN,
+  OFF_SEQ
+} from "./chunk-6BCX2E2R.mjs";
+
 // src/types/config.ts
 var DeliverPolicy = /* @__PURE__ */ ((DeliverPolicy2) => {
   DeliverPolicy2["All"] = "All";
@@ -177,101 +192,6 @@ function makeLazyMessage(raw, codec, fields, onAck, onNack, onNackDelay) {
   return msg;
 }
 
-// src/proto/constants.ts
-var MAGIC_V2 = 843207233;
-var HELLO_SIZE = 8;
-var CURRENT_VERSION = 2;
-var HEADER_SIZE = 16;
-var OFF_ACTION = 0;
-var OFF_FLAGS = 2;
-var OFF_ENTRY_FLAGS = 3;
-var OFF_MSG_LEN = 4;
-var OFF_SEQ = 8;
-
-// src/proto/frame.ts
-function frame(action, seq, bodyLen, flags = 0, entryFlags = 0) {
-  const buf = Buffer.allocUnsafe(HEADER_SIZE + bodyLen);
-  buf.writeUInt16LE(action, OFF_ACTION);
-  buf[OFF_FLAGS] = flags;
-  buf[OFF_ENTRY_FLAGS] = entryFlags;
-  buf.writeUInt32LE(bodyLen, OFF_MSG_LEN);
-  buf.writeBigUInt64LE(seq, OFF_SEQ);
-  return buf;
-}
-function packHello(caps = 2 /* Reply */) {
-  const buf = Buffer.allocUnsafe(HELLO_SIZE);
-  buf.writeUInt32LE(MAGIC_V2, 0);
-  buf[4] = CURRENT_VERSION;
-  buf[5] = 0 /* Client */;
-  buf.writeUInt16LE(caps, 6);
-  return buf;
-}
-function packDisconnect(seq) {
-  return frame(1541 /* Disconnect */, seq, 0);
-}
-
-// src/proto/publish.ts
-var EMPTY = Buffer.alloc(0);
-function packPublish(seq, streamId2, subject, payload, flags = 0, entryFlags = 0, msgId = EMPTY) {
-  const buf = frame(
-    257 /* Publish */,
-    seq,
-    8 + subject.length + msgId.length + payload.length,
-    flags,
-    entryFlags
-  );
-  buf.writeUInt32LE(streamId2, HEADER_SIZE);
-  buf.writeUInt16LE(subject.length, HEADER_SIZE + 4);
-  buf.writeUInt16LE(msgId.length, HEADER_SIZE + 6);
-  let off = HEADER_SIZE + 8;
-  subject.copy(buf, off);
-  off += subject.length;
-  msgId.copy(buf, off);
-  off += msgId.length;
-  payload.copy(buf, off);
-  return buf;
-}
-function packPublishWithReply(seq, streamId2, subject, replyTo, payload, flags = 0, entryFlags = 0) {
-  const tail = subject.length + replyTo.length + payload.length;
-  const buf = frame(260 /* PublishWithReply */, seq, 12 + tail, flags, entryFlags);
-  buf.writeUInt32LE(streamId2, HEADER_SIZE);
-  buf.writeUInt16LE(subject.length, HEADER_SIZE + 4);
-  buf.writeUInt16LE(replyTo.length, HEADER_SIZE + 6);
-  buf.writeUInt32LE(0, HEADER_SIZE + 8);
-  let off = HEADER_SIZE + 12;
-  subject.copy(buf, off);
-  off += subject.length;
-  replyTo.copy(buf, off);
-  off += replyTo.length;
-  payload.copy(buf, off);
-  return buf;
-}
-function packPublishBatch(seq, streamId2, entries, flags = 0, entryFlags = 0) {
-  let tail = 0;
-  for (const e of entries) {
-    const midLen = e.msgId ? e.msgId.length : 0;
-    tail += 8 + e.subject.length + midLen + e.payload.length;
-  }
-  const buf = frame(259 /* PublishBatch */, seq, 8 + tail, flags, entryFlags);
-  buf.writeUInt32LE(streamId2, HEADER_SIZE);
-  buf.writeUInt32LE(entries.length, HEADER_SIZE + 4);
-  let off = HEADER_SIZE + 8;
-  for (const e of entries) {
-    const mid = e.msgId ?? EMPTY;
-    buf.writeUInt16LE(e.subject.length, off);
-    buf.writeUInt16LE(mid.length, off + 2);
-    buf.writeUInt32LE(e.payload.length, off + 4);
-    off += 8;
-    buf.write(e.subject, off);
-    off += e.subject.length;
-    mid.copy(buf, off);
-    off += mid.length;
-    e.payload.copy(buf, off);
-    off += e.payload.length;
-  }
-  return buf;
-}
-
 // src/proto/delivery.ts
 function packCold(action, seq, body) {
   const utf8 = Buffer.from(JSON.stringify(body), "utf8");
@@ -572,6 +492,37 @@ function resolveLogger(logger) {
   return logger ?? NOOP;
 }
 
+// src/cron/cron-frame.ts
+function packCreateCron(seq, body) {
+  const json = Buffer.from(JSON.stringify(body), "utf8");
+  const buf = frame(1793 /* CreateCron */, seq, json.length);
+  json.copy(buf, HEADER_SIZE);
+  return buf;
+}
+function packDeleteCron(seq, name) {
+  const buf = frame(1794 /* DeleteCron */, seq, name.length);
+  name.copy(buf, HEADER_SIZE);
+  return buf;
+}
+var CRON_FIRE_FIXED = 2 + 8 + 8;
+function decodeCronFire(body) {
+  if (body.length < CRON_FIRE_FIXED) return void 0;
+  const nameLen = body.readUInt16LE(0);
+  if (body.length < CRON_FIRE_FIXED + nameLen) return void 0;
+  const fireTimeMs = body.readBigUInt64LE(2);
+  const fireCount = body.readBigUInt64LE(10);
+  const name = body.subarray(18, 18 + nameLen).toString();
+  return { name, fireTimeMs, fireCount };
+}
+function packCronAck(seq, name, ok) {
+  const bodyLen = 3 + name.length;
+  const buf = frame(1797 /* CronAck */, seq, bodyLen);
+  buf.writeUInt16LE(name.length, HEADER_SIZE);
+  buf[HEADER_SIZE + 2] = ok ? 0 : 1;
+  name.copy(buf, HEADER_SIZE + 3);
+  return buf;
+}
+
 // src/net/connection.ts
 function parseAddr(addr) {
   const i = addr.lastIndexOf(":");
@@ -595,6 +546,7 @@ var Connection = class _Connection {
   log;
   activeSubs = /* @__PURE__ */ new Map();
   metrics;
+  cronState;
   attachSocket(socket) {
     socket.setNoDelay(true);
     socket.on("data", (chunk) => this.framer.push(chunk, (f) => this.onFrame(f)));
@@ -649,6 +601,10 @@ var Connection = class _Connection {
   setMetrics(m) {
     this.metrics = m;
   }
+  /** Attach cron state so the connection can dispatch CronFire frames. */
+  setCronState(s) {
+    this.cronState = s;
+  }
   // ── Frame routing ─────────────────────────────────────────────────────────
   // Seq-based dispatch: match reply.header.seq → pending request. O(1).
   resolvePending(frame2) {
@@ -700,6 +656,10 @@ var Connection = class _Connection {
         this.handleBatchDeliver(frame2);
         return;
       }
+      case 1796 /* CronFire */: {
+        this.dispatchCronFire(frame2);
+        return;
+      }
       case 1538 /* Pong */:
         return;
       default: {
@@ -749,6 +709,19 @@ var Connection = class _Connection {
       off = tailEnd;
     }
   }
+  // ── Cron dispatch ──────────────────────────────────────────────────────────
+  dispatchCronFire(frame2) {
+    const body = frame2.subarray(HEADER_SIZE);
+    const view = decodeCronFire(body);
+    if (!view) return;
+    const handler = this.cronState?.getHandler(view.name);
+    const nameBuf = Buffer.from(view.name);
+    if (!handler) {
+      this.send(packCronAck(this.nextSeq(), nameBuf, false));
+      return;
+    }
+    handler({ name: view.name, fireTime: view.fireTimeMs, fireCount: view.fireCount }).then(() => this.send(packCronAck(this.nextSeq(), nameBuf, true))).catch(() => this.send(packCronAck(this.nextSeq(), nameBuf, false)));
+  }
   // ── Subscriptions ─────────────────────────────────────────────────────────
   async sendSubscribeV2(consumerId, filter, handler, onRenew) {
     return new Promise((resolve, reject) => {
@@ -791,6 +764,14 @@ var Connection = class _Connection {
       }).catch(() => {
       });
     }
+    this.replayCrons();
+  }
+  replayCrons() {
+    if (!this.cronState) return;
+    for (const { config } of this.cronState.allConfigs()) {
+      const seq = this.nextSeq();
+      this.socket.write(packCreateCron(seq, config));
+    }
   }
   // ── Routes (internal use) ─────────────────────────────────────────────────
   registerRoute(consumerId, handler) {
@@ -1074,9 +1055,9 @@ var ClientMetrics = class {
 };
 
 // src/stream/publish.ts
-var EMPTY2 = Buffer.alloc(0);
+var EMPTY = Buffer.alloc(0);
 function toMsgIdBuf(id) {
-  if (id == null) return EMPTY2;
+  if (id == null) return EMPTY;
   return typeof id === "string" ? Buffer.from(id) : id;
 }
 async function streamPublishAck(conn, sid, subject, data, opts) {
@@ -1105,6 +1086,86 @@ async function streamRequest(conn, sid, subject, data, timeoutMs) {
   return Buffer.alloc(0);
 }
 
+// src/cron/cron-builder.ts
+var CronBuilder = class {
+  constructor(conn, cronState, cronName) {
+    this.conn = conn;
+    this.cronState = cronState;
+    this.cronName = cronName;
+  }
+  expr;
+  timezone;
+  timeoutMs = 3e4;
+  allowOverlap = false;
+  every(expression) {
+    this.expr = expression;
+    return this;
+  }
+  tz(timezone) {
+    this.timezone = timezone;
+    return this;
+  }
+  timeout(ms) {
+    this.timeoutMs = ms;
+    return this;
+  }
+  overlap(allow) {
+    this.allowOverlap = allow;
+    return this;
+  }
+  async run(handler) {
+    if (!this.expr) throw new Error("cron expression required \u2014 call .every()");
+    const body = {
+      name: this.cronName,
+      every: this.expr,
+      tz: this.timezone,
+      timeout_ms: this.timeoutMs,
+      overlap: this.allowOverlap
+    };
+    const seq = this.conn.nextSeq();
+    await this.conn.sendExpectReply(packCreateCron(seq, body));
+    this.cronState.register(this.cronName, body, handler);
+    return new CronHandle(this.conn, this.cronState, this.cronName);
+  }
+};
+var CronHandle = class {
+  constructor(conn, cronState, cronName) {
+    this.conn = conn;
+    this.cronState = cronState;
+    this.cronName = cronName;
+  }
+  get name() {
+    return this.cronName;
+  }
+  async stop() {
+    const nameBuf = Buffer.from(this.cronName);
+    const seq = this.conn.nextSeq();
+    await this.conn.sendExpectReply(packDeleteCron(seq, nameBuf));
+    this.cronState.remove(this.cronName);
+  }
+};
+
+// src/cron/cron-state.ts
+var CronState = class {
+  handlers = /* @__PURE__ */ new Map();
+  register(name, config, handler) {
+    this.handlers.set(name, { handler, config });
+  }
+  remove(name) {
+    this.handlers.delete(name);
+  }
+  getHandler(name) {
+    return this.handlers.get(name)?.handler;
+  }
+  allConfigs() {
+    const out = [];
+    for (const [name, entry] of this.handlers) {
+      out.push({ name, config: entry.config });
+    }
+    return out;
+  }
+};
+
 // src/client/client.ts
 var DEFAULT_CONFIG = {
   servers: ["127.0.0.1:9898"],
@@ -1119,6 +1180,7 @@ var ArbitroClient = class {
   logger;
   sidCache = /* @__PURE__ */ new Map();
   _metrics = new ClientMetrics();
+  _cronState = new CronState();
   constructor(config) {
     this.cfg = { ...DEFAULT_CONFIG, ...config };
     this.tls = config.tls;
@@ -1135,6 +1197,7 @@ var ArbitroClient = class {
       this.logger
     );
     this.conn.setMetrics(this._metrics);
+    this.conn.setCronState(this._cronState);
     return this;
   }
   /**
@@ -1220,6 +1283,22 @@ var ArbitroClient = class {
     const sid = await this.resolveStreamId(streamName);
     return streamRequest(this.conn, sid, subject, data, timeoutMs);
   }
+  /**
+   * Publish a message with a delivery delay. The broker parks the message
+   * in its delayed journal and delivers it to consumers after `delayMs`
+   * milliseconds. Returns a Promise that resolves once the broker confirms
+   * receipt.
+   */
+  async publishDelayed(streamName, subject, data, delayMs) {
+    const sid = await this.resolveStreamId(streamName);
+    const { packPublishDelayed: packPublishDelayed2 } = await import("./publish-UA3YZGMK.mjs");
+    const { Flag: Flag3 } = await import("./constants-KF57DJ2L.mjs");
+    const subj = Buffer.from(this.prefixed(subject));
+    await this.conn.sendExpectReply(
+      packPublishDelayed2(this.conn.nextSeq(), sid, subj, data, BigInt(delayMs), Flag3.AckReq)
+    );
+    this._metrics.publishesSent++;
+  }
   async subscribe(streamName, configOrCb, callbackOrOpts, opts) {
     let config;
     let callback;
@@ -1339,7 +1418,7 @@ var ArbitroClient = class {
       maxInflight: config.maxAckPending ?? 0,
       ackPolicy: ackPolicyByte,
       deliverPolicy: deliverPolicyToU8(config.deliverPolicy),
-      deliverMode: config.fanout ? 1 : 0,
+      deliverMode: config.fanout ? 0 : 1,
       ackWaitMs: config.ackWaitMs ?? 0,
       startSeq: BigInt(config.startSeq ?? 0)
     };
@@ -1454,6 +1533,11 @@ var ArbitroClient = class {
   stream(name, config) {
     return new Stream(this, name, config);
   }
+  // ── Cron ──────────────────────────────────────────────────────────────────
+  /** Start building a cron job. Call `.every()` then `.run()` to register. */
+  cron(name) {
+    return new CronBuilder(this.conn, this._cronState, name);
+  }
   // ── Lifecycle ─────────────────────────────────────────────────────────────
   async close() {
     await this.conn.close();
@@ -1555,6 +1639,8 @@ export {
   ArbitroError,
   Codec,
   Consumer,
+  CronBuilder,
+  CronHandle,
   DeliverPolicy,
   ErrorCode,
   JournalType,