@opengeni/events 0.2.2 → 0.2.4

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.d.ts CHANGED
@@ -2,6 +2,8 @@ import { SessionEvent } from '@opengeni/contracts';
2
2
  import { Database, AppendEventInput } from '@opengeni/db';
3
3
  export { NatsConnection, connect, nkeys } from 'nats';
4
4
 
5
+ declare function coalesceSessionEventDeltas(events: SessionEvent[]): SessionEvent[];
6
+
5
7
  /** A NATS permission set: subject allow/deny lists (ADR-26 `pub`/`sub` →
6
8
  * `allow`/`deny`). An empty/undefined list means "no explicit grant" — combined
7
9
  * with the agent scope below, the connection can ONLY reach what `allow` lists. */
@@ -229,4 +231,4 @@ declare function createResponderConnection(natsUrl: string, auth: NatsConnectAut
229
231
  declare function appendAndPublishEvents(db: Database, bus: EventBus, workspaceId: string, sessionId: string, events: AppendEventInput[]): Promise<SessionEvent[]>;
230
232
  declare function formatSse(event: SessionEvent): string;
231
233
 
232
- export { type DecodedAuthRequest, type EventBus, type MintAuthResponseInput, type MintUserJwtInput, type NatsConnectAuth, type NatsPermission, type NatsPermissions, type RequestConnection, type RequestHandler, type RequestReply, type ResponderConnection, appendAndPublishEvents, createNatsEventBus, createResponderConnection, decodeAuthRequest, formatSse, mintAuthResponse, mintUserJwt, workspaceAgentPermissions };
234
+ export { type DecodedAuthRequest, type EventBus, type MintAuthResponseInput, type MintUserJwtInput, type NatsConnectAuth, type NatsPermission, type NatsPermissions, type RequestConnection, type RequestHandler, type RequestReply, type ResponderConnection, appendAndPublishEvents, coalesceSessionEventDeltas, createNatsEventBus, createResponderConnection, decodeAuthRequest, formatSse, mintAuthResponse, mintUserJwt, workspaceAgentPermissions };
package/dist/index.js CHANGED
@@ -2,6 +2,96 @@
2
2
  import { appendSessionEvents, sessionSubject } from "@opengeni/db";
3
3
  import { connect, JSONCodec } from "nats";
4
4
 
5
+ // src/coalesce.ts
6
+ var COALESCIBLE_DELTA_TYPES = /* @__PURE__ */ new Set([
7
+ "agent.message.delta",
8
+ "agent.reasoning.delta",
9
+ "sandbox.command.output.delta"
10
+ ]);
11
+ function coalesceSessionEventDeltas(events) {
12
+ const coalesced = [];
13
+ let run = null;
14
+ const flush = () => {
15
+ if (!run) {
16
+ return;
17
+ }
18
+ coalesced.push({
19
+ ...run.first,
20
+ payload: {
21
+ text: run.text,
22
+ coalescedUntil: run.lastSequence,
23
+ ...run.first.type === "sandbox.command.output.delta" && run.sandboxName !== void 0 ? { name: run.sandboxName } : {}
24
+ }
25
+ });
26
+ run = null;
27
+ };
28
+ for (const event of events) {
29
+ if (!isCoalescibleDelta(event)) {
30
+ flush();
31
+ coalesced.push(event);
32
+ continue;
33
+ }
34
+ const sandboxName = event.type === "sandbox.command.output.delta" ? sandboxDeltaName(event.payload) : void 0;
35
+ if (run && sameDeltaRun(run.first, event, run.sandboxName, sandboxName)) {
36
+ run.text += deltaText(event);
37
+ run.lastSequence = event.sequence;
38
+ continue;
39
+ }
40
+ flush();
41
+ run = {
42
+ first: event,
43
+ lastSequence: event.sequence,
44
+ text: deltaText(event),
45
+ sandboxName
46
+ };
47
+ }
48
+ flush();
49
+ return coalesced;
50
+ }
51
+ function isCoalescibleDelta(event) {
52
+ return COALESCIBLE_DELTA_TYPES.has(event.type);
53
+ }
54
+ function sameDeltaRun(first, next, firstSandboxName, nextSandboxName) {
55
+ if (first.type !== next.type) {
56
+ return false;
57
+ }
58
+ if ((first.turnId ?? null) !== (next.turnId ?? null)) {
59
+ return false;
60
+ }
61
+ return first.type !== "sandbox.command.output.delta" || firstSandboxName === nextSandboxName;
62
+ }
63
+ function deltaText(event) {
64
+ if (event.type === "agent.reasoning.delta") {
65
+ return reasoningText(event.payload);
66
+ }
67
+ const payload = asRecord(event.payload);
68
+ if (event.type === "sandbox.command.output.delta") {
69
+ return typeof payload.text === "string" ? payload.text : typeof payload.output === "string" ? payload.output : "";
70
+ }
71
+ return typeof payload.text === "string" ? payload.text : "";
72
+ }
73
+ function reasoningText(payload) {
74
+ const record = asRecord(payload);
75
+ if (typeof record.text === "string") {
76
+ return record.text;
77
+ }
78
+ const content = asRecord(asRecord(record.item).rawItem).content;
79
+ if (!Array.isArray(content)) {
80
+ return "";
81
+ }
82
+ return content.map((part) => {
83
+ const text = asRecord(part).text;
84
+ return typeof text === "string" ? text : "";
85
+ }).join("");
86
+ }
87
+ function sandboxDeltaName(payload) {
88
+ const name = asRecord(payload).name;
89
+ return typeof name === "string" ? name : void 0;
90
+ }
91
+ function asRecord(value) {
92
+ return value && typeof value === "object" ? value : {};
93
+ }
94
+
5
95
  // src/nats-jwt.ts
6
96
  import { createHash } from "crypto";
7
97
  import { nkeys } from "nats";
@@ -330,6 +420,7 @@ function formatSse(event) {
330
420
  }
331
421
  export {
332
422
  appendAndPublishEvents,
423
+ coalesceSessionEventDeltas,
333
424
  connect2 as connect,
334
425
  createNatsEventBus,
335
426
  createResponderConnection,
package/dist/index.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"sources":["../src/index.ts","../src/nats-jwt.ts"],"sourcesContent":["import type { SessionBusMessage, SessionEvent } from \"@opengeni/contracts\";\nimport { appendSessionEvents, sessionSubject, type AppendEventInput, type Database } from \"@opengeni/db\";\nimport { connect, JSONCodec, type ConnectionOptions, type Msg, type NatsConnection, type Subscription } from \"nats\";\n\nconst codec = JSONCodec<SessionBusMessage | SessionEvent>();\n\n/**\n * Reconnect + keepalive defaults applied to EVERY long-lived NATS connection\n * this package opens (the event bus AND the standalone auth-callout responder).\n *\n * The production outage these guard against: an in-cluster NATS broker pod\n * restart. nats.js's stock policy gives up after ~10 attempts (~20s) and the\n * client goes permanently CONNECTION_CLOSED — which takes the whole control\n * plane down with it: every session-create publishes events to NATS, and the\n * API-hosted auth-callout responder dies so BYO agents get \"authorization\n * violation\". Recovery then required a MANUAL api+worker restart. With these\n * options the client retries forever and auto-recovers the moment the broker\n * returns. Factored into one source of truth so the call sites never drift.\n *\n * - `reconnect` + `maxReconnectAttempts: -1` — never give up (infinite retry).\n * - `reconnectTimeWait` (2s base) + `reconnectJitter`/`reconnectJitterTLS`\n * (up to 1s) — a fleet of api/worker pods doesn't thundering-herd the broker\n * on recovery.\n * - `waitOnFirstConnect` — a broker briefly unavailable at boot must not\n * hard-fail the process; the client keeps trying instead of throwing.\n * - `pingInterval`/`maxPingOut` — promptly detect a silently-dead socket so the\n * reconnect machinery actually engages instead of hanging on a zombie.\n */\nconst RECONNECT_OPTIONS = {\n reconnect: true,\n maxReconnectAttempts: -1,\n reconnectTimeWait: 2_000,\n reconnectJitter: 1_000,\n reconnectJitterTLS: 1_000,\n waitOnFirstConnect: true,\n pingInterval: 20_000,\n maxPingOut: 3,\n} satisfies ConnectionOptions;\n\n/**\n * The single source of truth for a long-lived connection's resilience: merge the\n * reconnect/keepalive defaults UNDER the caller's connection options (servers +\n * optional auth/name). Every long-lived `connect()` in this package goes through\n * here so the two call sites can never diverge.\n */\nfunction withReconnectDefaults(options: ConnectionOptions): ConnectionOptions {\n return { ...RECONNECT_OPTIONS, ...options };\n}\n\n/** How long a best-effort publish waits on `flush()` before giving up (see `publish`). */\nconst PUBLISH_FLUSH_TIMEOUT_MS = 2_000;\n\n/**\n * Await `nc.flush()` but never longer than `timeoutMs`. With infinite reconnect a\n * `flush()` issued while the broker is down does NOT reject — it pends until the\n * broker returns, which can be minutes. Racing it against a timer keeps a long\n * outage from stalling an in-flight turn; the published message stays buffered\n * and is delivered on reconnect regardless. A flush rejection (connection fully\n * CLOSED) is swallowed here so the timeout race never leaks an unhandled\n * rejection — the caller's publish path is what logs the drop.\n */\nasync function flushWithTimeout(nc: NatsConnection, timeoutMs: number): Promise<void> {\n let timer: ReturnType<typeof setTimeout> | undefined;\n const timeout = new Promise<void>((resolve) => {\n timer = setTimeout(resolve, timeoutMs);\n });\n try {\n await Promise.race([nc.flush().catch(() => undefined), timeout]);\n } finally {\n if (timer) {\n clearTimeout(timer);\n }\n }\n}\n\n/**\n * Drain a long-lived connection's status async-iterator to the log so a future\n * broker outage is OBSERVABLE (disconnect → reconnecting → reconnect → update).\n * Fire-and-forget for the connection's lifetime; the loop ends when the\n * connection closes. `label` distinguishes the event-bus connection from the\n * auth-callout responder in the logs.\n */\nfunction logConnectionStatus(nc: NatsConnection, label: string): void {\n void (async () => {\n try {\n for await (const status of nc.status()) {\n console.warn(`[nats:${label}] ${status.type}`, status.data);\n }\n } catch {\n // The status iterator simply ends when the connection closes; never let it\n // throw out of this background loop.\n }\n })();\n}\n\nexport {\n decodeAuthRequest,\n mintAuthResponse,\n mintUserJwt,\n workspaceAgentPermissions,\n type DecodedAuthRequest,\n type MintAuthResponseInput,\n type MintUserJwtInput,\n type NatsPermission,\n type NatsPermissions,\n} from \"./nats-jwt\";\n\n// Re-export the raw NATS primitives a consumer needs to open a direct connection or\n// generate nkeys (the auth-callout responder's standalone connection, the\n// agent-simulating integration tests). This keeps `nats` an internal dependency of\n// this leaf — callers in the bun workspace reach it through @opengeni/events rather\n// than depending on `nats` directly.\nexport { connect, nkeys, type NatsConnection } from \"nats\";\n\n/**\n * A raw request/reply reply — just the response bytes. Mirrors the subset of the\n * NATS `Msg` shape a binary request/reply caller needs (`NatsControlRpc` consumes\n * exactly this). Kept minimal so the events package does not leak the `nats` `Msg`\n * type into the agent-loop-free runtime leaf.\n */\nexport type RequestReply = { data: Uint8Array };\n\n/**\n * The minimal request/reply connection the selfhosted control plane consumes\n * (structurally identical to `@opengeni/runtime`'s `NatsRequestConnection`). The\n * API/worker hand this accessor to `NatsControlRpc` so the control transport rides\n * the SAME managed NATS connection the event bus already owns — a NATS connection\n * natively supports both pub/sub and request/reply, so there is NEVER a second\n * connection.\n */\nexport interface RequestConnection {\n request(subject: string, payload: Uint8Array, opts: { timeout: number }): Promise<RequestReply>;\n}\n\n/**\n * A handler answering a request/reply on a subscribed subject: given the request\n * bytes (+ the concrete subject the message landed on, for `agent.<ws>.<id>.rpc`\n * style wildcard routing), return the response bytes to reply with. A thrown error\n * leaves the request unanswered (the caller's request times out / sees no\n * responder), which the control plane maps to `agent_offline` / reconnecting.\n */\nexport type RequestHandler = (request: Uint8Array, subject: string) => Promise<Uint8Array> | Uint8Array;\n\nexport type EventBus = {\n publish: (workspaceId: string, sessionId: string, events: SessionEvent[]) => Promise<void>;\n subscribe: (workspaceId: string, sessionId: string, onEvents: (events: SessionEvent[]) => void | Promise<void>) => Promise<() => void>;\n /**\n * Issue a binary request/reply on a subject over the bus's NATS connection\n * (the selfhosted control plane: `agent.<ws>.<id>.rpc`). A new usage of what was\n * a one-way bus — same connection, native NATS request/reply. Rejects on a\n * no-responder (NATS 503) or a request timeout; the caller (`NatsControlRpc`)\n * maps those to `agent_offline` / `agent_reconnecting`, never a NotFound.\n */\n request: (subject: string, payload: Uint8Array, opts: { timeoutMs: number }) => Promise<RequestReply>;\n /**\n * Subscribe-and-reply on a subject (the responder side — the enrolled agent, or\n * a test stand-in for it): for every request on `subject`, call `handler` and\n * `respond` with its bytes over the SAME connection. Returns an unsubscribe fn.\n * A subject may be a NATS wildcard (e.g. `agent.*.*.rpc`).\n */\n subscribeRequests: (subject: string, handler: RequestHandler) => () => void;\n /**\n * Subscribe to the agent EVENT plane (the one-way fire-and-forget heartbeats +\n * going-offline the agent PUBLISHES on `agent.<ws>.<id>.events`, NOT a\n * request/reply). The M10 metrics-ingestion consumer subscribes the wildcard\n * `agent.*.*.events` and gets each raw payload plus its concrete subject (so it\n * can extract `<ws>`/`<id>` for the per-enrollment upsert). Returns an\n * unsubscribe fn. Decoding the AgentEvent is the caller's concern (this leaf\n * does not depend on `@opengeni/agent-proto`).\n */\n subscribeAgentEvents: (\n subject: string,\n handler: (payload: Uint8Array, subject: string) => void | Promise<void>,\n ) => () => void;\n /**\n * The `RequestConnection` accessor the selfhosted `NatsControlRpc` consumes —\n * the SAME managed connection (pub/sub + request/reply share it). The control\n * plane injects this so the transport never opens a second connection.\n */\n getRequestConnection: () => RequestConnection;\n close: () => Promise<void>;\n};\n\n/**\n * Connect the event bus + control-plane request/reply over ONE managed NATS\n * connection. `auth` is the PRIVILEGED control-plane login (M-AUTH): when the\n * server runs with auth_callout, the api/worker authenticates as a static account\n * user permitted to request `agent.*.rpc` + receive its inbox replies. When `auth`\n * is omitted the connection is anonymous (local dev / a NATS without auth_callout)\n * — the existing behavior, unchanged.\n */\nexport async function createNatsEventBus(\n natsUrl: string,\n auth?: { user: string; pass: string },\n): Promise<EventBus> {\n const connectOptions: ConnectionOptions = { servers: natsUrl };\n if (auth) {\n connectOptions.user = auth.user;\n connectOptions.pass = auth.pass;\n }\n const nc = await connect(withReconnectDefaults(connectOptions));\n logConnectionStatus(nc, \"event-bus\");\n const requestConnection: RequestConnection = {\n request: async (subject, payload, opts) => requestReply(nc, subject, payload, opts.timeout),\n };\n return {\n publish: async (workspaceId, sessionId, events) => {\n if (events.length === 0) {\n return;\n }\n // Best-effort LIVE fan-out. These events are ALREADY durably appended to\n // the DB before we get here (they carry a DB-assigned `sequence`), and\n // every consumer reconciles from that durable log — the server SSE stream\n // replays + gap-backfills via `listSessionEvents`, and the SDK client\n // reconnects and replays from the durable events endpoint. So a publish\n // that fails during a broker blip only delays LIVE delivery (healed by the\n // next successful publish's gap-backfill, or a stream reconnect); it must\n // never throw the in-flight turn to death.\n try {\n nc.publish(sessionSubject(workspaceId, sessionId), codec.encode({ workspaceId, sessionId, events }));\n } catch (error) {\n // `publish()` throws synchronously only when the connection is fully\n // CLOSED (with infinite reconnect, effectively never outside shutdown).\n console.warn(\n `[nats:event-bus] dropped live publish for ${workspaceId}/${sessionId}; events are durable in the DB and reconcile on stream replay`,\n error,\n );\n return;\n }\n await flushWithTimeout(nc, PUBLISH_FLUSH_TIMEOUT_MS);\n },\n subscribe: async (workspaceId, sessionId, onEvents) => subscribeSession(nc, workspaceId, sessionId, onEvents),\n request: async (subject, payload, opts) => requestReply(nc, subject, payload, opts.timeoutMs),\n subscribeRequests: (subject, handler) => subscribeRequests(nc, subject, handler),\n subscribeAgentEvents: (subject, handler) => subscribeAgentEvents(nc, subject, handler),\n getRequestConnection: () => requestConnection,\n close: async () => {\n await nc.drain();\n },\n };\n}\n\n/**\n * A standalone NATS connection answering request/reply on ONE subject — the\n * transport primitive the auth-callout responder uses. It is DELIBERATELY a\n * SEPARATE connection from the event bus: the callout responder authenticates as\n * the callout account's `auth_users` user (a username/password or token in the\n * `AUTH` account), which is a DIFFERENT identity from the control-plane's\n * privileged account that the event bus + `NatsControlRpc` ride. One connection\n * per identity; never multiplex the two.\n *\n * `request`/`reply` here is the RAW NATS request/reply (`$SYS.REQ.USER.AUTH`): the\n * server publishes an authorization request with a reply inbox; the handler returns\n * the signed authorization-response bytes which we `respond` on that inbox.\n */\nexport interface ResponderConnection {\n /** Subscribe-and-reply on `subject`; returns an async close that drains. */\n close: () => Promise<void>;\n}\n\n/** Connection auth for a standalone NATS connection (the callout responder). */\nexport type NatsConnectAuth =\n | { kind: \"user-password\"; user: string; pass: string }\n | { kind: \"token\"; token: string }\n | { kind: \"anonymous\" };\n\n/**\n * Open a standalone NATS connection and subscribe `subject`, replying to every\n * request with `handler(requestBytes, subject)`. Used by the auth-callout\n * responder to serve `$SYS.REQ.USER.AUTH` as the callout auth user. Returns a\n * handle whose `close()` drains the connection. A handler that throws leaves the\n * request UNANSWERED — for auth-callout that means the server denies the\n * connection on its own timeout, which is the correct fail-closed behavior (a\n * responder bug must never accidentally grant access).\n */\nexport async function createResponderConnection(\n natsUrl: string,\n auth: NatsConnectAuth,\n subject: string,\n handler: RequestHandler,\n options: { name?: string } = {},\n): Promise<ResponderConnection> {\n const connectOptions: ConnectionOptions = { servers: natsUrl };\n if (options.name) {\n connectOptions.name = options.name;\n }\n if (auth.kind === \"user-password\") {\n connectOptions.user = auth.user;\n connectOptions.pass = auth.pass;\n } else if (auth.kind === \"token\") {\n connectOptions.token = auth.token;\n }\n const nc = await connect(withReconnectDefaults(connectOptions));\n logConnectionStatus(nc, options.name ? `auth-callout:${options.name}` : \"auth-callout\");\n const sub: Subscription = nc.subscribe(subject);\n void (async () => {\n for await (const msg of sub) {\n if (!msg.reply) {\n continue;\n }\n try {\n const reply = await handler(msg.data, msg.subject);\n msg.respond(reply);\n } catch {\n // Leave UNANSWERED — fail-closed. The server denies the connect attempt\n // on its callout timeout; a responder error never grants access.\n }\n }\n })();\n return {\n close: async () => {\n sub.unsubscribe();\n await nc.drain();\n },\n };\n}\n\nexport async function appendAndPublishEvents(db: Database, bus: EventBus, workspaceId: string, sessionId: string, events: AppendEventInput[]): Promise<SessionEvent[]> {\n const appended = await appendSessionEvents(db, workspaceId, sessionId, events);\n // The DB append above is the durable system of record; the publish is only a\n // best-effort LIVE fan-out. Guard it so NO EventBus implementation can throw an\n // in-flight agent turn to death on a transient NATS disconnect — consumers\n // reconcile any missed live events from the durable log via the events/stream\n // endpoint (DB replay + gap-backfill). The managed `createNatsEventBus` bus\n // already swallows internally, so this catch is the belt-and-suspenders guard\n // for any other bus impl (and a fully CLOSED connection during shutdown).\n try {\n await bus.publish(workspaceId, sessionId, appended);\n } catch (error) {\n console.warn(\n `[events] live publish failed for ${workspaceId}/${sessionId}; ${appended.length} event(s) are durable and reconcile on stream replay`,\n error,\n );\n }\n return appended;\n}\n\nfunction subscribeSession(nc: NatsConnection, workspaceId: string, sessionId: string, onEvents: (events: SessionEvent[]) => void | Promise<void>): () => void {\n const sub: Subscription = nc.subscribe(sessionSubject(workspaceId, sessionId));\n void (async () => {\n for await (const msg of sub) {\n const decoded = codec.decode(msg.data) as SessionBusMessage | SessionEvent;\n const events = \"events\" in decoded ? decoded.events : [decoded];\n await onEvents(events);\n }\n })();\n return () => {\n sub.unsubscribe();\n };\n}\n\n/**\n * A binary request/reply over the managed connection. Returns ONLY the reply\n * bytes (the `RequestReply` shape) — the request/reply error semantics (a\n * no-responder NATS 503, a request timeout) propagate as the rejected promise so\n * the caller owns the mapping. The reply is delivered via the connection's\n * built-in mux inbox; no extra subscription is created here.\n */\nasync function requestReply(nc: NatsConnection, subject: string, payload: Uint8Array, timeout: number): Promise<RequestReply> {\n const msg: Msg = await nc.request(subject, payload, { timeout });\n return { data: msg.data };\n}\n\n/**\n * Subscribe to `subject` and reply to every request with the handler's bytes,\n * over the SAME connection. The responder side of request/reply: each delivered\n * `Msg` carries a `reply` inbox; `msg.respond(bytes)` publishes the answer there.\n * A handler that throws (or a message with no `reply` subject) is left unanswered\n * — the requester then sees a timeout, never a malformed reply.\n */\nfunction subscribeRequests(nc: NatsConnection, subject: string, handler: RequestHandler): () => void {\n const sub: Subscription = nc.subscribe(subject);\n void (async () => {\n for await (const msg of sub) {\n // A request always carries a reply inbox; a plain publish to this subject\n // (no reply) is ignored — request/reply is the only contract here.\n if (!msg.reply) {\n continue;\n }\n try {\n const reply = await handler(msg.data, msg.subject);\n msg.respond(reply);\n } catch {\n // Leave the request unanswered: the requester's request times out, which\n // the selfhosted control plane reads as a transient blip (reconnecting),\n // never a malformed reply. The responder stays subscribed for the next op.\n }\n }\n })();\n return () => {\n sub.unsubscribe();\n };\n}\n\n/**\n * Subscribe to the one-way agent event plane: deliver each published payload (the\n * agent's `AgentEvent` heartbeat / going-offline, NOT a request/reply) to the\n * handler with its concrete subject. A plain `nc.subscribe` (no reply); a handler\n * that throws is swallowed so one bad event never tears down the subscription\n * (ingestion is best-effort — a metrics gap is never fatal).\n */\nfunction subscribeAgentEvents(\n nc: NatsConnection,\n subject: string,\n handler: (payload: Uint8Array, subject: string) => void | Promise<void>,\n): () => void {\n const sub: Subscription = nc.subscribe(subject);\n void (async () => {\n for await (const msg of sub) {\n try {\n await handler(msg.data, msg.subject);\n } catch {\n // Swallow: best-effort ingestion. The subscription stays live for the\n // next event.\n }\n }\n })();\n return () => {\n sub.unsubscribe();\n };\n}\n\nexport function formatSse(event: SessionEvent): string {\n return [\n `id: ${event.sequence}`,\n `event: ${event.type}`,\n `data: ${JSON.stringify(event)}`,\n \"\",\n \"\",\n ].join(\"\\n\");\n}\n","// packages/events/src/nats-jwt.ts — NATS JWT v2 signing for the auth-callout\n// responder (bring-your-own-compute M-AUTH; dossier §10.1 NATS Accounts per\n// workspace + §17 the isolation smoke).\n//\n// This is the cryptographic core of the auth-callout tenancy boundary. When an\n// external agent connects to NATS presenting its `oge_` enrollment bearer as the\n// connect auth-token, nats-server (configured with `auth_callout`) issues an\n// authorization request on `$SYS.REQ.USER.AUTH`. Our responder (auth-callout.ts)\n// validates the bearer and answers with a SIGNED authorization-response JWT that\n// embeds a SIGNED user JWT scoping the connection to publish/subscribe ONLY\n// `agent.<workspaceId>.>` (+ the reply `_INBOX.>`). That per-subject permission\n// set IS the per-workspace isolation: workspace A's agent literally cannot\n// pub/sub workspace B's subjects (§19 the NATS-Accounts-misconfig leak risk is\n// closed at the JWT-permission layer, not just by subject naming).\n//\n// WHY HAND-ROLL THE JWT ENCODING (vs a dep): the NATS JWT v2 wire format is small,\n// stable, and fully specified (ADR-26 + nats-io/jwt): a base64url header\n// `{\"typ\":\"JWT\",\"alg\":\"ed25519-nkey\"}`, base64url JSON claims whose `jti` is the\n// base32(SHA-512/256(claims-with-blank-jti)), and an ed25519 nkey signature over\n// `header.payload`. nkeys (re-exported by the `nats` package we already depend on)\n// gives us the ed25519 sign primitive; Node `crypto` gives SHA-512/256. So we own\n// the encoding in a few well-tested functions rather than pull an alpha\n// `@nats-io/jwt` (0.0.x) whose nkeys-version compat is uncertain. No `xkey`\n// encryption is used (the bearer is already an authenticated identity claim and\n// the wire is TLS — encryption is an optional ADR-26 hardening, off here).\n//\n// SECURITY: the account SIGNING SEED never leaves this process and is NEVER logged.\n// Callers pass it as a `string` seed; we `fromSeed` it once per sign. The bearer\n// the responder validates is HMAC-verified elsewhere (verifyEnrollmentBearer); this\n// module only mints the scoped NATS credential once identity is proven.\n\nimport { createHash } from \"node:crypto\";\nimport { nkeys } from \"nats\";\n\n/** The NATS JWT v2 header — constant for every token we mint (ADR-26 / nats-io/jwt:\n * `TokenTypeJwt=\"JWT\"`, `AlgorithmNkey=\"ed25519-nkey\"`). */\nconst JWT_HEADER = { typ: \"JWT\", alg: \"ed25519-nkey\" } as const;\n\n/** NATS user-claim `nats.type` discriminator + `nats.version` for v2 claims. */\nconst USER_CLAIM_TYPE = \"user\";\nconst AUTH_RESPONSE_CLAIM_TYPE = \"authorization_response\";\nconst NATS_CLAIM_VERSION = 2;\n\n/** A NATS permission set: subject allow/deny lists (ADR-26 `pub`/`sub` →\n * `allow`/`deny`). An empty/undefined list means \"no explicit grant\" — combined\n * with the agent scope below, the connection can ONLY reach what `allow` lists. */\nexport interface NatsPermission {\n allow?: string[];\n deny?: string[];\n}\n\n/** The pub/sub permissions embedded in a user JWT. */\nexport interface NatsPermissions {\n pub: NatsPermission;\n sub: NatsPermission;\n}\n\n/**\n * The minimal nkey keypair surface this module needs — exactly what\n * `nkeys.fromSeed(seed)` returns. Declared structurally so the module does not\n * leak the `nats` nkeys type through its public signature.\n */\ninterface NkeyPair {\n getPublicKey(): string;\n sign(input: Uint8Array): Uint8Array;\n}\n\n/** base64url (RawURLEncoding — no padding), matching nats-io/jwt's `serialize`. */\nfunction base64UrlEncode(bytes: Uint8Array): string {\n return Buffer.from(bytes).toString(\"base64url\");\n}\n\n/** RFC 4648 base32 (standard alphabet, NO padding) — the encoding nats-io/jwt\n * uses for the `jti` hash. Node has no built-in base32, so a tiny encoder. */\nconst BASE32_ALPHABET = \"ABCDEFGHIJKLMNOPQRSTUVWXYZ234567\";\nfunction base32NoPadding(bytes: Uint8Array): string {\n let bits = 0;\n let value = 0;\n let out = \"\";\n for (const byte of bytes) {\n value = (value << 8) | byte;\n bits += 8;\n while (bits >= 5) {\n bits -= 5;\n out += BASE32_ALPHABET[(value >>> bits) & 31];\n }\n }\n if (bits > 0) {\n out += BASE32_ALPHABET[(value << (5 - bits)) & 31];\n }\n return out;\n}\n\n/**\n * Compute the canonical NATS `jti`: base32(NoPadding, std-alphabet) of the\n * SHA-512/256 of the claims object SERIALIZED WITH AN EMPTY `jti` (nats-io/jwt's\n * `hash`). nats-server recomputes + verifies this on decode, so it must match\n * byte-for-byte. We serialize the SAME object we will sign, only with `jti:\"\"`.\n */\nfunction computeJti(claimsWithBlankJti: object): string {\n const json = JSON.stringify(claimsWithBlankJti);\n const digest = createHash(\"sha512-256\").update(json, \"utf8\").digest();\n return base32NoPadding(digest);\n}\n\n/**\n * Encode + sign a NATS v2 JWT. The `claims` MUST already carry `iss`/`sub`/`iat`\n * (+ optional `aud`/`exp`) and a `nats` block; this function fills `jti` (the\n * canonical hash), serializes `header.payload`, signs that with `signingKey`, and\n * appends the base64url signature. Returns the compact `header.payload.signature`.\n */\nfunction encodeJwt(claims: Record<string, unknown>, signingKey: NkeyPair): string {\n // jti is the hash of the claims with jti blanked — set it blank, hash, then set.\n const withBlankJti = { ...claims, jti: \"\" };\n const jti = computeJti(withBlankJti);\n const finalClaims = { ...claims, jti };\n\n const header = base64UrlEncode(Buffer.from(JSON.stringify(JWT_HEADER), \"utf8\"));\n const payload = base64UrlEncode(Buffer.from(JSON.stringify(finalClaims), \"utf8\"));\n const signingInput = `${header}.${payload}`;\n const signature = signingKey.sign(Buffer.from(signingInput, \"utf8\"));\n return `${signingInput}.${base64UrlEncode(signature)}`;\n}\n\n/**\n * Input to mint a workspace-scoped NATS user JWT for an enrolled agent.\n * - `userPublicKey` — the `user_nkey` from the authorization request; it MUST be\n * the `sub` of the user JWT (nats-server rejects a mismatch).\n * - `accountSeed` — the callout account SIGNING seed (`SA...`); both the user JWT\n * `iss` (its public key) and the signature come from it. NEVER logged.\n * - `name` — a human label for the user (the agent id), for server logs.\n * - `permissions` — the pub/sub allow/deny lists (the workspace scope).\n * - `expiresAtSeconds` — optional absolute `exp` (unix seconds). When set the\n * server will expire the connection's credential; we tie it to the bearer's\n * remaining life so a revoked/expired enrollment cannot outlive its bearer.\n */\nexport interface MintUserJwtInput {\n userPublicKey: string;\n accountSeed: string;\n name: string;\n permissions: NatsPermissions;\n /** The target account NAME (the `auth_callout.account`) the user binds to; the\n * embedded user JWT's `aud` in server-config mode. */\n audienceAccount: string;\n expiresAtSeconds?: number;\n}\n\n/**\n * Mint a signed NATS user JWT scoped by `permissions`. In auth-callout SERVER\n * mode the user JWT is signed by the callout ISSUER ACCOUNT key, and its `iss` is\n * that account's public key. The returned JWT is embedded as `nats.jwt` in the\n * authorization response.\n */\nexport function mintUserJwt(input: MintUserJwtInput): string {\n const accountKey = nkeys.fromSeed(Buffer.from(input.accountSeed)) as unknown as NkeyPair;\n const accountPublicKey = accountKey.getPublicKey();\n const nowSeconds = Math.floor(Date.now() / 1000);\n\n const natsBlock: Record<string, unknown> = {\n type: USER_CLAIM_TYPE,\n version: NATS_CLAIM_VERSION,\n pub: input.permissions.pub,\n sub: input.permissions.sub,\n // Unlimited subscriptions / data / payload (the workspace subject scope, NOT\n // a connection-resource quota, is the boundary here).\n subs: -1,\n data: -1,\n payload: -1,\n };\n\n const claims: Record<string, unknown> = {\n jti: \"\",\n iat: nowSeconds,\n iss: accountPublicKey,\n name: input.name,\n sub: input.userPublicKey,\n // SERVER-config-mode placement: nats-server reads the embedded user JWT's `aud`\n // as the target account NAME (the configured `auth_callout.account`). This is\n // how the authenticated user binds to that account; the workspace isolation is\n // then carried by the pub/sub permissions below.\n aud: input.audienceAccount,\n nats: natsBlock,\n };\n if (typeof input.expiresAtSeconds === \"number\") {\n claims.exp = input.expiresAtSeconds;\n }\n return encodeJwt(claims, accountKey);\n}\n\n/**\n * Input to mint the authorization RESPONSE JWT the responder publishes back on the\n * request's reply subject (ADR-26 §3).\n * - `userPublicKey` — the request's `user_nkey`; the response `sub`.\n * - `serverId` — the request's `nats.server_id.id` (the server's public key); the\n * response `aud`.\n * - `accountSeed` — the callout account signing seed; signs the response and is\n * its `iss` (public key). NEVER logged.\n * - `userJwt` — the embedded signed user JWT (omit on a denial).\n * - `error` — a human-readable denial message (omit on success). When present the\n * server denies the connection.\n */\nexport interface MintAuthResponseInput {\n userPublicKey: string;\n serverId: string;\n accountSeed: string;\n userJwt?: string;\n error?: string;\n}\n\n/**\n * Mint the signed authorization-response JWT. On success it carries the embedded\n * user JWT (`nats.jwt`); on denial it carries `nats.error` and NO user JWT, which\n * makes nats-server refuse the connection. Signed by the callout account key (its\n * public key is `iss`); `sub` is the user_nkey, `aud` is the server id.\n */\nexport function mintAuthResponse(input: MintAuthResponseInput): string {\n const accountKey = nkeys.fromSeed(Buffer.from(input.accountSeed)) as unknown as NkeyPair;\n const accountPublicKey = accountKey.getPublicKey();\n const nowSeconds = Math.floor(Date.now() / 1000);\n\n const natsBlock: Record<string, unknown> = {\n type: AUTH_RESPONSE_CLAIM_TYPE,\n version: NATS_CLAIM_VERSION,\n };\n if (input.userJwt) {\n natsBlock.jwt = input.userJwt;\n }\n if (input.error) {\n natsBlock.error = input.error;\n }\n\n const claims: Record<string, unknown> = {\n jti: \"\",\n iat: nowSeconds,\n iss: accountPublicKey,\n // The response `aud` MUST be the SERVER public key in server-config mode\n // (nats-server validates \"Audience must be a server public key\"). The\n // authenticated user is placed into the configured `auth_callout.account` (the\n // SAME account the responder + the privileged control plane connect into), so\n // `agent.<ws>.<id>.rpc` request/reply routes; the workspace isolation is carried\n // entirely by the user JWT's pub/sub subject permissions (NOT by cross-account\n // placement, which server-config-mode nats does not support — nats-io#4335).\n aud: input.serverId,\n sub: input.userPublicKey,\n nats: natsBlock,\n };\n return encodeJwt(claims, accountKey);\n}\n\n/**\n * The fields the responder needs out of the authorization REQUEST JWT (ADR-26 §2).\n * The request is itself a NATS JWT (`header.payload.signature`) the server signs;\n * we only DECODE it (the server proves its own identity by the connection, and the\n * embedded `auth_token` is independently HMAC-verified), so we read the payload\n * without re-verifying the server signature.\n */\nexport interface DecodedAuthRequest {\n /** The public user nkey the response user JWT MUST be `sub`-scoped to. */\n userNkey: string;\n /** The server's public id — the response `aud`. */\n serverId: string;\n /** The connect `auth_token` the client presented (our `oge_` bearer), if any. */\n authToken: string | undefined;\n /** The connect username, if any (unused today; present for completeness). */\n user: string | undefined;\n}\n\n/**\n * Decode the authorization-request JWT payload (the middle base64url segment). The\n * request shape (ADR-26 §2): `nats.user_nkey`, `nats.server_id.id`, and the\n * presented connect options under `nats.connect_opts` (`auth_token` / `user`).\n * Returns null on a malformed token so the caller can deny cleanly.\n */\nexport function decodeAuthRequest(token: string): DecodedAuthRequest | null {\n const parts = token.split(\".\");\n if (parts.length !== 3) {\n return null;\n }\n let payload: unknown;\n try {\n payload = JSON.parse(Buffer.from(parts[1]!, \"base64url\").toString(\"utf8\"));\n } catch {\n return null;\n }\n if (typeof payload !== \"object\" || payload === null) {\n return null;\n }\n const nats = (payload as { nats?: unknown }).nats;\n if (typeof nats !== \"object\" || nats === null) {\n return null;\n }\n const natsObj = nats as {\n user_nkey?: unknown;\n server_id?: { id?: unknown } | unknown;\n connect_opts?: { auth_token?: unknown; user?: unknown } | unknown;\n };\n const userNkey = typeof natsObj.user_nkey === \"string\" ? natsObj.user_nkey : null;\n if (!userNkey) {\n return null;\n }\n const serverIdRaw =\n typeof natsObj.server_id === \"object\" && natsObj.server_id !== null\n ? (natsObj.server_id as { id?: unknown }).id\n : undefined;\n const serverId = typeof serverIdRaw === \"string\" ? serverIdRaw : \"\";\n const connectOpts =\n typeof natsObj.connect_opts === \"object\" && natsObj.connect_opts !== null\n ? (natsObj.connect_opts as { auth_token?: unknown; user?: unknown })\n : {};\n const authToken = typeof connectOpts.auth_token === \"string\" ? connectOpts.auth_token : undefined;\n const user = typeof connectOpts.user === \"string\" ? connectOpts.user : undefined;\n return { userNkey, serverId, authToken, user };\n}\n\n/**\n * Build the workspace-scoped permission set for an agent: it may publish + subscribe\n * ONLY `agent.<workspaceId>.>` (its own RPC/event/hello subtree) and the reply\n * `_INBOX.>` subtree (so request/reply round-trips work). Everything else is\n * implicitly denied (an allow-list with no other entries IS the deny-all-else).\n *\n * THE isolation assertion (§17): with `workspaceId=A`, the returned allow lists name\n * only `agent.A.>` — so a connection bearing this credential is rejected by\n * nats-server the instant it tries to pub/sub `agent.B.>`. This is the per-workspace\n * tenancy boundary, enforced cryptographically by the signed JWT, not by naming.\n */\nexport function workspaceAgentPermissions(workspaceId: string): NatsPermissions {\n const agentScope = `agent.${workspaceId}.>`;\n // The reply-inbox subtree must be reachable for request/reply (the control plane\n // requests on agent.<ws>.<id>.rpc with a reply inbox; the agent responds there).\n const inboxScope = \"_INBOX.>\";\n return {\n pub: { allow: [agentScope, inboxScope] },\n sub: { allow: [agentScope, inboxScope] },\n };\n}\n"],"mappings":";AACA,SAAS,qBAAqB,sBAA4D;AAC1F,SAAS,SAAS,iBAA2F;;;AC6B7G,SAAS,kBAAkB;AAC3B,SAAS,aAAa;AAItB,IAAM,aAAa,EAAE,KAAK,OAAO,KAAK,eAAe;AAGrD,IAAM,kBAAkB;AACxB,IAAM,2BAA2B;AACjC,IAAM,qBAAqB;AA2B3B,SAAS,gBAAgB,OAA2B;AAClD,SAAO,OAAO,KAAK,KAAK,EAAE,SAAS,WAAW;AAChD;AAIA,IAAM,kBAAkB;AACxB,SAAS,gBAAgB,OAA2B;AAClD,MAAI,OAAO;AACX,MAAI,QAAQ;AACZ,MAAI,MAAM;AACV,aAAW,QAAQ,OAAO;AACxB,YAAS,SAAS,IAAK;AACvB,YAAQ;AACR,WAAO,QAAQ,GAAG;AAChB,cAAQ;AACR,aAAO,gBAAiB,UAAU,OAAQ,EAAE;AAAA,IAC9C;AAAA,EACF;AACA,MAAI,OAAO,GAAG;AACZ,WAAO,gBAAiB,SAAU,IAAI,OAAS,EAAE;AAAA,EACnD;AACA,SAAO;AACT;AAQA,SAAS,WAAW,oBAAoC;AACtD,QAAM,OAAO,KAAK,UAAU,kBAAkB;AAC9C,QAAM,SAAS,WAAW,YAAY,EAAE,OAAO,MAAM,MAAM,EAAE,OAAO;AACpE,SAAO,gBAAgB,MAAM;AAC/B;AAQA,SAAS,UAAU,QAAiC,YAA8B;AAEhF,QAAM,eAAe,EAAE,GAAG,QAAQ,KAAK,GAAG;AAC1C,QAAM,MAAM,WAAW,YAAY;AACnC,QAAM,cAAc,EAAE,GAAG,QAAQ,IAAI;AAErC,QAAM,SAAS,gBAAgB,OAAO,KAAK,KAAK,UAAU,UAAU,GAAG,MAAM,CAAC;AAC9E,QAAM,UAAU,gBAAgB,OAAO,KAAK,KAAK,UAAU,WAAW,GAAG,MAAM,CAAC;AAChF,QAAM,eAAe,GAAG,MAAM,IAAI,OAAO;AACzC,QAAM,YAAY,WAAW,KAAK,OAAO,KAAK,cAAc,MAAM,CAAC;AACnE,SAAO,GAAG,YAAY,IAAI,gBAAgB,SAAS,CAAC;AACtD;AA+BO,SAAS,YAAY,OAAiC;AAC3D,QAAM,aAAa,MAAM,SAAS,OAAO,KAAK,MAAM,WAAW,CAAC;AAChE,QAAM,mBAAmB,WAAW,aAAa;AACjD,QAAM,aAAa,KAAK,MAAM,KAAK,IAAI,IAAI,GAAI;AAE/C,QAAM,YAAqC;AAAA,IACzC,MAAM;AAAA,IACN,SAAS;AAAA,IACT,KAAK,MAAM,YAAY;AAAA,IACvB,KAAK,MAAM,YAAY;AAAA;AAAA;AAAA,IAGvB,MAAM;AAAA,IACN,MAAM;AAAA,IACN,SAAS;AAAA,EACX;AAEA,QAAM,SAAkC;AAAA,IACtC,KAAK;AAAA,IACL,KAAK;AAAA,IACL,KAAK;AAAA,IACL,MAAM,MAAM;AAAA,IACZ,KAAK,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA,IAKX,KAAK,MAAM;AAAA,IACX,MAAM;AAAA,EACR;AACA,MAAI,OAAO,MAAM,qBAAqB,UAAU;AAC9C,WAAO,MAAM,MAAM;AAAA,EACrB;AACA,SAAO,UAAU,QAAQ,UAAU;AACrC;AA4BO,SAAS,iBAAiB,OAAsC;AACrE,QAAM,aAAa,MAAM,SAAS,OAAO,KAAK,MAAM,WAAW,CAAC;AAChE,QAAM,mBAAmB,WAAW,aAAa;AACjD,QAAM,aAAa,KAAK,MAAM,KAAK,IAAI,IAAI,GAAI;AAE/C,QAAM,YAAqC;AAAA,IACzC,MAAM;AAAA,IACN,SAAS;AAAA,EACX;AACA,MAAI,MAAM,SAAS;AACjB,cAAU,MAAM,MAAM;AAAA,EACxB;AACA,MAAI,MAAM,OAAO;AACf,cAAU,QAAQ,MAAM;AAAA,EAC1B;AAEA,QAAM,SAAkC;AAAA,IACtC,KAAK;AAAA,IACL,KAAK;AAAA,IACL,KAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,IAQL,KAAK,MAAM;AAAA,IACX,KAAK,MAAM;AAAA,IACX,MAAM;AAAA,EACR;AACA,SAAO,UAAU,QAAQ,UAAU;AACrC;AA0BO,SAAS,kBAAkB,OAA0C;AAC1E,QAAM,QAAQ,MAAM,MAAM,GAAG;AAC7B,MAAI,MAAM,WAAW,GAAG;AACtB,WAAO;AAAA,EACT;AACA,MAAI;AACJ,MAAI;AACF,cAAU,KAAK,MAAM,OAAO,KAAK,MAAM,CAAC,GAAI,WAAW,EAAE,SAAS,MAAM,CAAC;AAAA,EAC3E,QAAQ;AACN,WAAO;AAAA,EACT;AACA,MAAI,OAAO,YAAY,YAAY,YAAY,MAAM;AACnD,WAAO;AAAA,EACT;AACA,QAAM,OAAQ,QAA+B;AAC7C,MAAI,OAAO,SAAS,YAAY,SAAS,MAAM;AAC7C,WAAO;AAAA,EACT;AACA,QAAM,UAAU;AAKhB,QAAM,WAAW,OAAO,QAAQ,cAAc,WAAW,QAAQ,YAAY;AAC7E,MAAI,CAAC,UAAU;AACb,WAAO;AAAA,EACT;AACA,QAAM,cACJ,OAAO,QAAQ,cAAc,YAAY,QAAQ,cAAc,OAC1D,QAAQ,UAA+B,KACxC;AACN,QAAM,WAAW,OAAO,gBAAgB,WAAW,cAAc;AACjE,QAAM,cACJ,OAAO,QAAQ,iBAAiB,YAAY,QAAQ,iBAAiB,OAChE,QAAQ,eACT,CAAC;AACP,QAAM,YAAY,OAAO,YAAY,eAAe,WAAW,YAAY,aAAa;AACxF,QAAM,OAAO,OAAO,YAAY,SAAS,WAAW,YAAY,OAAO;AACvE,SAAO,EAAE,UAAU,UAAU,WAAW,KAAK;AAC/C;AAaO,SAAS,0BAA0B,aAAsC;AAC9E,QAAM,aAAa,SAAS,WAAW;AAGvC,QAAM,aAAa;AACnB,SAAO;AAAA,IACL,KAAK,EAAE,OAAO,CAAC,YAAY,UAAU,EAAE;AAAA,IACvC,KAAK,EAAE,OAAO,CAAC,YAAY,UAAU,EAAE;AAAA,EACzC;AACF;;;AD9NA,SAAS,WAAAA,UAAS,SAAAC,cAAkC;AA5GpD,IAAM,QAAQ,UAA4C;AAwB1D,IAAM,oBAAoB;AAAA,EACxB,WAAW;AAAA,EACX,sBAAsB;AAAA,EACtB,mBAAmB;AAAA,EACnB,iBAAiB;AAAA,EACjB,oBAAoB;AAAA,EACpB,oBAAoB;AAAA,EACpB,cAAc;AAAA,EACd,YAAY;AACd;AAQA,SAAS,sBAAsB,SAA+C;AAC5E,SAAO,EAAE,GAAG,mBAAmB,GAAG,QAAQ;AAC5C;AAGA,IAAM,2BAA2B;AAWjC,eAAe,iBAAiB,IAAoB,WAAkC;AACpF,MAAI;AACJ,QAAM,UAAU,IAAI,QAAc,CAAC,YAAY;AAC7C,YAAQ,WAAW,SAAS,SAAS;AAAA,EACvC,CAAC;AACD,MAAI;AACF,UAAM,QAAQ,KAAK,CAAC,GAAG,MAAM,EAAE,MAAM,MAAM,MAAS,GAAG,OAAO,CAAC;AAAA,EACjE,UAAE;AACA,QAAI,OAAO;AACT,mBAAa,KAAK;AAAA,IACpB;AAAA,EACF;AACF;AASA,SAAS,oBAAoB,IAAoB,OAAqB;AACpE,QAAM,YAAY;AAChB,QAAI;AACF,uBAAiB,UAAU,GAAG,OAAO,GAAG;AACtC,gBAAQ,KAAK,SAAS,KAAK,KAAK,OAAO,IAAI,IAAI,OAAO,IAAI;AAAA,MAC5D;AAAA,IACF,QAAQ;AAAA,IAGR;AAAA,EACF,GAAG;AACL;AAkGA,eAAsB,mBACpB,SACA,MACmB;AACnB,QAAM,iBAAoC,EAAE,SAAS,QAAQ;AAC7D,MAAI,MAAM;AACR,mBAAe,OAAO,KAAK;AAC3B,mBAAe,OAAO,KAAK;AAAA,EAC7B;AACA,QAAM,KAAK,MAAM,QAAQ,sBAAsB,cAAc,CAAC;AAC9D,sBAAoB,IAAI,WAAW;AACnC,QAAM,oBAAuC;AAAA,IAC3C,SAAS,OAAO,SAAS,SAAS,SAAS,aAAa,IAAI,SAAS,SAAS,KAAK,OAAO;AAAA,EAC5F;AACA,SAAO;AAAA,IACL,SAAS,OAAO,aAAa,WAAW,WAAW;AACjD,UAAI,OAAO,WAAW,GAAG;AACvB;AAAA,MACF;AASA,UAAI;AACF,WAAG,QAAQ,eAAe,aAAa,SAAS,GAAG,MAAM,OAAO,EAAE,aAAa,WAAW,OAAO,CAAC,CAAC;AAAA,MACrG,SAAS,OAAO;AAGd,gBAAQ;AAAA,UACN,6CAA6C,WAAW,IAAI,SAAS;AAAA,UACrE;AAAA,QACF;AACA;AAAA,MACF;AACA,YAAM,iBAAiB,IAAI,wBAAwB;AAAA,IACrD;AAAA,IACA,WAAW,OAAO,aAAa,WAAW,aAAa,iBAAiB,IAAI,aAAa,WAAW,QAAQ;AAAA,IAC5G,SAAS,OAAO,SAAS,SAAS,SAAS,aAAa,IAAI,SAAS,SAAS,KAAK,SAAS;AAAA,IAC5F,mBAAmB,CAAC,SAAS,YAAY,kBAAkB,IAAI,SAAS,OAAO;AAAA,IAC/E,sBAAsB,CAAC,SAAS,YAAY,qBAAqB,IAAI,SAAS,OAAO;AAAA,IACrF,sBAAsB,MAAM;AAAA,IAC5B,OAAO,YAAY;AACjB,YAAM,GAAG,MAAM;AAAA,IACjB;AAAA,EACF;AACF;AAmCA,eAAsB,0BACpB,SACA,MACA,SACA,SACA,UAA6B,CAAC,GACA;AAC9B,QAAM,iBAAoC,EAAE,SAAS,QAAQ;AAC7D,MAAI,QAAQ,MAAM;AAChB,mBAAe,OAAO,QAAQ;AAAA,EAChC;AACA,MAAI,KAAK,SAAS,iBAAiB;AACjC,mBAAe,OAAO,KAAK;AAC3B,mBAAe,OAAO,KAAK;AAAA,EAC7B,WAAW,KAAK,SAAS,SAAS;AAChC,mBAAe,QAAQ,KAAK;AAAA,EAC9B;AACA,QAAM,KAAK,MAAM,QAAQ,sBAAsB,cAAc,CAAC;AAC9D,sBAAoB,IAAI,QAAQ,OAAO,gBAAgB,QAAQ,IAAI,KAAK,cAAc;AACtF,QAAM,MAAoB,GAAG,UAAU,OAAO;AAC9C,QAAM,YAAY;AAChB,qBAAiB,OAAO,KAAK;AAC3B,UAAI,CAAC,IAAI,OAAO;AACd;AAAA,MACF;AACA,UAAI;AACF,cAAM,QAAQ,MAAM,QAAQ,IAAI,MAAM,IAAI,OAAO;AACjD,YAAI,QAAQ,KAAK;AAAA,MACnB,QAAQ;AAAA,MAGR;AAAA,IACF;AAAA,EACF,GAAG;AACH,SAAO;AAAA,IACL,OAAO,YAAY;AACjB,UAAI,YAAY;AAChB,YAAM,GAAG,MAAM;AAAA,IACjB;AAAA,EACF;AACF;AAEA,eAAsB,uBAAuB,IAAc,KAAe,aAAqB,WAAmB,QAAqD;AACrK,QAAM,WAAW,MAAM,oBAAoB,IAAI,aAAa,WAAW,MAAM;AAQ7E,MAAI;AACF,UAAM,IAAI,QAAQ,aAAa,WAAW,QAAQ;AAAA,EACpD,SAAS,OAAO;AACd,YAAQ;AAAA,MACN,oCAAoC,WAAW,IAAI,SAAS,KAAK,SAAS,MAAM;AAAA,MAChF;AAAA,IACF;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,iBAAiB,IAAoB,aAAqB,WAAmB,UAAwE;AAC5J,QAAM,MAAoB,GAAG,UAAU,eAAe,aAAa,SAAS,CAAC;AAC7E,QAAM,YAAY;AAChB,qBAAiB,OAAO,KAAK;AAC3B,YAAM,UAAU,MAAM,OAAO,IAAI,IAAI;AACrC,YAAM,SAAS,YAAY,UAAU,QAAQ,SAAS,CAAC,OAAO;AAC9D,YAAM,SAAS,MAAM;AAAA,IACvB;AAAA,EACF,GAAG;AACH,SAAO,MAAM;AACX,QAAI,YAAY;AAAA,EAClB;AACF;AASA,eAAe,aAAa,IAAoB,SAAiB,SAAqB,SAAwC;AAC5H,QAAM,MAAW,MAAM,GAAG,QAAQ,SAAS,SAAS,EAAE,QAAQ,CAAC;AAC/D,SAAO,EAAE,MAAM,IAAI,KAAK;AAC1B;AASA,SAAS,kBAAkB,IAAoB,SAAiB,SAAqC;AACnG,QAAM,MAAoB,GAAG,UAAU,OAAO;AAC9C,QAAM,YAAY;AAChB,qBAAiB,OAAO,KAAK;AAG3B,UAAI,CAAC,IAAI,OAAO;AACd;AAAA,MACF;AACA,UAAI;AACF,cAAM,QAAQ,MAAM,QAAQ,IAAI,MAAM,IAAI,OAAO;AACjD,YAAI,QAAQ,KAAK;AAAA,MACnB,QAAQ;AAAA,MAIR;AAAA,IACF;AAAA,EACF,GAAG;AACH,SAAO,MAAM;AACX,QAAI,YAAY;AAAA,EAClB;AACF;AASA,SAAS,qBACP,IACA,SACA,SACY;AACZ,QAAM,MAAoB,GAAG,UAAU,OAAO;AAC9C,QAAM,YAAY;AAChB,qBAAiB,OAAO,KAAK;AAC3B,UAAI;AACF,cAAM,QAAQ,IAAI,MAAM,IAAI,OAAO;AAAA,MACrC,QAAQ;AAAA,MAGR;AAAA,IACF;AAAA,EACF,GAAG;AACH,SAAO,MAAM;AACX,QAAI,YAAY;AAAA,EAClB;AACF;AAEO,SAAS,UAAU,OAA6B;AACrD,SAAO;AAAA,IACL,OAAO,MAAM,QAAQ;AAAA,IACrB,UAAU,MAAM,IAAI;AAAA,IACpB,SAAS,KAAK,UAAU,KAAK,CAAC;AAAA,IAC9B;AAAA,IACA;AAAA,EACF,EAAE,KAAK,IAAI;AACb;","names":["connect","nkeys"]}
1
+ {"version":3,"sources":["../src/index.ts","../src/coalesce.ts","../src/nats-jwt.ts"],"sourcesContent":["import type { SessionBusMessage, SessionEvent } from \"@opengeni/contracts\";\nimport { appendSessionEvents, sessionSubject, type AppendEventInput, type Database } from \"@opengeni/db\";\nimport { connect, JSONCodec, type ConnectionOptions, type Msg, type NatsConnection, type Subscription } from \"nats\";\n\nconst codec = JSONCodec<SessionBusMessage | SessionEvent>();\n\nexport { coalesceSessionEventDeltas } from \"./coalesce\";\n\n/**\n * Reconnect + keepalive defaults applied to EVERY long-lived NATS connection\n * this package opens (the event bus AND the standalone auth-callout responder).\n *\n * The production outage these guard against: an in-cluster NATS broker pod\n * restart. nats.js's stock policy gives up after ~10 attempts (~20s) and the\n * client goes permanently CONNECTION_CLOSED — which takes the whole control\n * plane down with it: every session-create publishes events to NATS, and the\n * API-hosted auth-callout responder dies so BYO agents get \"authorization\n * violation\". Recovery then required a MANUAL api+worker restart. With these\n * options the client retries forever and auto-recovers the moment the broker\n * returns. Factored into one source of truth so the call sites never drift.\n *\n * - `reconnect` + `maxReconnectAttempts: -1` — never give up (infinite retry).\n * - `reconnectTimeWait` (2s base) + `reconnectJitter`/`reconnectJitterTLS`\n * (up to 1s) — a fleet of api/worker pods doesn't thundering-herd the broker\n * on recovery.\n * - `waitOnFirstConnect` — a broker briefly unavailable at boot must not\n * hard-fail the process; the client keeps trying instead of throwing.\n * - `pingInterval`/`maxPingOut` — promptly detect a silently-dead socket so the\n * reconnect machinery actually engages instead of hanging on a zombie.\n */\nconst RECONNECT_OPTIONS = {\n reconnect: true,\n maxReconnectAttempts: -1,\n reconnectTimeWait: 2_000,\n reconnectJitter: 1_000,\n reconnectJitterTLS: 1_000,\n waitOnFirstConnect: true,\n pingInterval: 20_000,\n maxPingOut: 3,\n} satisfies ConnectionOptions;\n\n/**\n * The single source of truth for a long-lived connection's resilience: merge the\n * reconnect/keepalive defaults UNDER the caller's connection options (servers +\n * optional auth/name). Every long-lived `connect()` in this package goes through\n * here so the two call sites can never diverge.\n */\nfunction withReconnectDefaults(options: ConnectionOptions): ConnectionOptions {\n return { ...RECONNECT_OPTIONS, ...options };\n}\n\n/** How long a best-effort publish waits on `flush()` before giving up (see `publish`). */\nconst PUBLISH_FLUSH_TIMEOUT_MS = 2_000;\n\n/**\n * Await `nc.flush()` but never longer than `timeoutMs`. With infinite reconnect a\n * `flush()` issued while the broker is down does NOT reject — it pends until the\n * broker returns, which can be minutes. Racing it against a timer keeps a long\n * outage from stalling an in-flight turn; the published message stays buffered\n * and is delivered on reconnect regardless. A flush rejection (connection fully\n * CLOSED) is swallowed here so the timeout race never leaks an unhandled\n * rejection — the caller's publish path is what logs the drop.\n */\nasync function flushWithTimeout(nc: NatsConnection, timeoutMs: number): Promise<void> {\n let timer: ReturnType<typeof setTimeout> | undefined;\n const timeout = new Promise<void>((resolve) => {\n timer = setTimeout(resolve, timeoutMs);\n });\n try {\n await Promise.race([nc.flush().catch(() => undefined), timeout]);\n } finally {\n if (timer) {\n clearTimeout(timer);\n }\n }\n}\n\n/**\n * Drain a long-lived connection's status async-iterator to the log so a future\n * broker outage is OBSERVABLE (disconnect → reconnecting → reconnect → update).\n * Fire-and-forget for the connection's lifetime; the loop ends when the\n * connection closes. `label` distinguishes the event-bus connection from the\n * auth-callout responder in the logs.\n */\nfunction logConnectionStatus(nc: NatsConnection, label: string): void {\n void (async () => {\n try {\n for await (const status of nc.status()) {\n console.warn(`[nats:${label}] ${status.type}`, status.data);\n }\n } catch {\n // The status iterator simply ends when the connection closes; never let it\n // throw out of this background loop.\n }\n })();\n}\n\nexport {\n decodeAuthRequest,\n mintAuthResponse,\n mintUserJwt,\n workspaceAgentPermissions,\n type DecodedAuthRequest,\n type MintAuthResponseInput,\n type MintUserJwtInput,\n type NatsPermission,\n type NatsPermissions,\n} from \"./nats-jwt\";\n\n// Re-export the raw NATS primitives a consumer needs to open a direct connection or\n// generate nkeys (the auth-callout responder's standalone connection, the\n// agent-simulating integration tests). This keeps `nats` an internal dependency of\n// this leaf — callers in the bun workspace reach it through @opengeni/events rather\n// than depending on `nats` directly.\nexport { connect, nkeys, type NatsConnection } from \"nats\";\n\n/**\n * A raw request/reply reply — just the response bytes. Mirrors the subset of the\n * NATS `Msg` shape a binary request/reply caller needs (`NatsControlRpc` consumes\n * exactly this). Kept minimal so the events package does not leak the `nats` `Msg`\n * type into the agent-loop-free runtime leaf.\n */\nexport type RequestReply = { data: Uint8Array };\n\n/**\n * The minimal request/reply connection the selfhosted control plane consumes\n * (structurally identical to `@opengeni/runtime`'s `NatsRequestConnection`). The\n * API/worker hand this accessor to `NatsControlRpc` so the control transport rides\n * the SAME managed NATS connection the event bus already owns — a NATS connection\n * natively supports both pub/sub and request/reply, so there is NEVER a second\n * connection.\n */\nexport interface RequestConnection {\n request(subject: string, payload: Uint8Array, opts: { timeout: number }): Promise<RequestReply>;\n}\n\n/**\n * A handler answering a request/reply on a subscribed subject: given the request\n * bytes (+ the concrete subject the message landed on, for `agent.<ws>.<id>.rpc`\n * style wildcard routing), return the response bytes to reply with. A thrown error\n * leaves the request unanswered (the caller's request times out / sees no\n * responder), which the control plane maps to `agent_offline` / reconnecting.\n */\nexport type RequestHandler = (request: Uint8Array, subject: string) => Promise<Uint8Array> | Uint8Array;\n\nexport type EventBus = {\n publish: (workspaceId: string, sessionId: string, events: SessionEvent[]) => Promise<void>;\n subscribe: (workspaceId: string, sessionId: string, onEvents: (events: SessionEvent[]) => void | Promise<void>) => Promise<() => void>;\n /**\n * Issue a binary request/reply on a subject over the bus's NATS connection\n * (the selfhosted control plane: `agent.<ws>.<id>.rpc`). A new usage of what was\n * a one-way bus — same connection, native NATS request/reply. Rejects on a\n * no-responder (NATS 503) or a request timeout; the caller (`NatsControlRpc`)\n * maps those to `agent_offline` / `agent_reconnecting`, never a NotFound.\n */\n request: (subject: string, payload: Uint8Array, opts: { timeoutMs: number }) => Promise<RequestReply>;\n /**\n * Subscribe-and-reply on a subject (the responder side — the enrolled agent, or\n * a test stand-in for it): for every request on `subject`, call `handler` and\n * `respond` with its bytes over the SAME connection. Returns an unsubscribe fn.\n * A subject may be a NATS wildcard (e.g. `agent.*.*.rpc`).\n */\n subscribeRequests: (subject: string, handler: RequestHandler) => () => void;\n /**\n * Subscribe to the agent EVENT plane (the one-way fire-and-forget heartbeats +\n * going-offline the agent PUBLISHES on `agent.<ws>.<id>.events`, NOT a\n * request/reply). The M10 metrics-ingestion consumer subscribes the wildcard\n * `agent.*.*.events` and gets each raw payload plus its concrete subject (so it\n * can extract `<ws>`/`<id>` for the per-enrollment upsert). Returns an\n * unsubscribe fn. Decoding the AgentEvent is the caller's concern (this leaf\n * does not depend on `@opengeni/agent-proto`).\n */\n subscribeAgentEvents: (\n subject: string,\n handler: (payload: Uint8Array, subject: string) => void | Promise<void>,\n ) => () => void;\n /**\n * The `RequestConnection` accessor the selfhosted `NatsControlRpc` consumes —\n * the SAME managed connection (pub/sub + request/reply share it). The control\n * plane injects this so the transport never opens a second connection.\n */\n getRequestConnection: () => RequestConnection;\n close: () => Promise<void>;\n};\n\n/**\n * Connect the event bus + control-plane request/reply over ONE managed NATS\n * connection. `auth` is the PRIVILEGED control-plane login (M-AUTH): when the\n * server runs with auth_callout, the api/worker authenticates as a static account\n * user permitted to request `agent.*.rpc` + receive its inbox replies. When `auth`\n * is omitted the connection is anonymous (local dev / a NATS without auth_callout)\n * — the existing behavior, unchanged.\n */\nexport async function createNatsEventBus(\n natsUrl: string,\n auth?: { user: string; pass: string },\n): Promise<EventBus> {\n const connectOptions: ConnectionOptions = { servers: natsUrl };\n if (auth) {\n connectOptions.user = auth.user;\n connectOptions.pass = auth.pass;\n }\n const nc = await connect(withReconnectDefaults(connectOptions));\n logConnectionStatus(nc, \"event-bus\");\n const requestConnection: RequestConnection = {\n request: async (subject, payload, opts) => requestReply(nc, subject, payload, opts.timeout),\n };\n return {\n publish: async (workspaceId, sessionId, events) => {\n if (events.length === 0) {\n return;\n }\n // Best-effort LIVE fan-out. These events are ALREADY durably appended to\n // the DB before we get here (they carry a DB-assigned `sequence`), and\n // every consumer reconciles from that durable log — the server SSE stream\n // replays + gap-backfills via `listSessionEvents`, and the SDK client\n // reconnects and replays from the durable events endpoint. So a publish\n // that fails during a broker blip only delays LIVE delivery (healed by the\n // next successful publish's gap-backfill, or a stream reconnect); it must\n // never throw the in-flight turn to death.\n try {\n nc.publish(sessionSubject(workspaceId, sessionId), codec.encode({ workspaceId, sessionId, events }));\n } catch (error) {\n // `publish()` throws synchronously only when the connection is fully\n // CLOSED (with infinite reconnect, effectively never outside shutdown).\n console.warn(\n `[nats:event-bus] dropped live publish for ${workspaceId}/${sessionId}; events are durable in the DB and reconcile on stream replay`,\n error,\n );\n return;\n }\n await flushWithTimeout(nc, PUBLISH_FLUSH_TIMEOUT_MS);\n },\n subscribe: async (workspaceId, sessionId, onEvents) => subscribeSession(nc, workspaceId, sessionId, onEvents),\n request: async (subject, payload, opts) => requestReply(nc, subject, payload, opts.timeoutMs),\n subscribeRequests: (subject, handler) => subscribeRequests(nc, subject, handler),\n subscribeAgentEvents: (subject, handler) => subscribeAgentEvents(nc, subject, handler),\n getRequestConnection: () => requestConnection,\n close: async () => {\n await nc.drain();\n },\n };\n}\n\n/**\n * A standalone NATS connection answering request/reply on ONE subject — the\n * transport primitive the auth-callout responder uses. It is DELIBERATELY a\n * SEPARATE connection from the event bus: the callout responder authenticates as\n * the callout account's `auth_users` user (a username/password or token in the\n * `AUTH` account), which is a DIFFERENT identity from the control-plane's\n * privileged account that the event bus + `NatsControlRpc` ride. One connection\n * per identity; never multiplex the two.\n *\n * `request`/`reply` here is the RAW NATS request/reply (`$SYS.REQ.USER.AUTH`): the\n * server publishes an authorization request with a reply inbox; the handler returns\n * the signed authorization-response bytes which we `respond` on that inbox.\n */\nexport interface ResponderConnection {\n /** Subscribe-and-reply on `subject`; returns an async close that drains. */\n close: () => Promise<void>;\n}\n\n/** Connection auth for a standalone NATS connection (the callout responder). */\nexport type NatsConnectAuth =\n | { kind: \"user-password\"; user: string; pass: string }\n | { kind: \"token\"; token: string }\n | { kind: \"anonymous\" };\n\n/**\n * Open a standalone NATS connection and subscribe `subject`, replying to every\n * request with `handler(requestBytes, subject)`. Used by the auth-callout\n * responder to serve `$SYS.REQ.USER.AUTH` as the callout auth user. Returns a\n * handle whose `close()` drains the connection. A handler that throws leaves the\n * request UNANSWERED — for auth-callout that means the server denies the\n * connection on its own timeout, which is the correct fail-closed behavior (a\n * responder bug must never accidentally grant access).\n */\nexport async function createResponderConnection(\n natsUrl: string,\n auth: NatsConnectAuth,\n subject: string,\n handler: RequestHandler,\n options: { name?: string } = {},\n): Promise<ResponderConnection> {\n const connectOptions: ConnectionOptions = { servers: natsUrl };\n if (options.name) {\n connectOptions.name = options.name;\n }\n if (auth.kind === \"user-password\") {\n connectOptions.user = auth.user;\n connectOptions.pass = auth.pass;\n } else if (auth.kind === \"token\") {\n connectOptions.token = auth.token;\n }\n const nc = await connect(withReconnectDefaults(connectOptions));\n logConnectionStatus(nc, options.name ? `auth-callout:${options.name}` : \"auth-callout\");\n const sub: Subscription = nc.subscribe(subject);\n void (async () => {\n for await (const msg of sub) {\n if (!msg.reply) {\n continue;\n }\n try {\n const reply = await handler(msg.data, msg.subject);\n msg.respond(reply);\n } catch {\n // Leave UNANSWERED — fail-closed. The server denies the connect attempt\n // on its callout timeout; a responder error never grants access.\n }\n }\n })();\n return {\n close: async () => {\n sub.unsubscribe();\n await nc.drain();\n },\n };\n}\n\nexport async function appendAndPublishEvents(db: Database, bus: EventBus, workspaceId: string, sessionId: string, events: AppendEventInput[]): Promise<SessionEvent[]> {\n const appended = await appendSessionEvents(db, workspaceId, sessionId, events);\n // The DB append above is the durable system of record; the publish is only a\n // best-effort LIVE fan-out. Guard it so NO EventBus implementation can throw an\n // in-flight agent turn to death on a transient NATS disconnect — consumers\n // reconcile any missed live events from the durable log via the events/stream\n // endpoint (DB replay + gap-backfill). The managed `createNatsEventBus` bus\n // already swallows internally, so this catch is the belt-and-suspenders guard\n // for any other bus impl (and a fully CLOSED connection during shutdown).\n try {\n await bus.publish(workspaceId, sessionId, appended);\n } catch (error) {\n console.warn(\n `[events] live publish failed for ${workspaceId}/${sessionId}; ${appended.length} event(s) are durable and reconcile on stream replay`,\n error,\n );\n }\n return appended;\n}\n\nfunction subscribeSession(nc: NatsConnection, workspaceId: string, sessionId: string, onEvents: (events: SessionEvent[]) => void | Promise<void>): () => void {\n const sub: Subscription = nc.subscribe(sessionSubject(workspaceId, sessionId));\n void (async () => {\n for await (const msg of sub) {\n const decoded = codec.decode(msg.data) as SessionBusMessage | SessionEvent;\n const events = \"events\" in decoded ? decoded.events : [decoded];\n await onEvents(events);\n }\n })();\n return () => {\n sub.unsubscribe();\n };\n}\n\n/**\n * A binary request/reply over the managed connection. Returns ONLY the reply\n * bytes (the `RequestReply` shape) — the request/reply error semantics (a\n * no-responder NATS 503, a request timeout) propagate as the rejected promise so\n * the caller owns the mapping. The reply is delivered via the connection's\n * built-in mux inbox; no extra subscription is created here.\n */\nasync function requestReply(nc: NatsConnection, subject: string, payload: Uint8Array, timeout: number): Promise<RequestReply> {\n const msg: Msg = await nc.request(subject, payload, { timeout });\n return { data: msg.data };\n}\n\n/**\n * Subscribe to `subject` and reply to every request with the handler's bytes,\n * over the SAME connection. The responder side of request/reply: each delivered\n * `Msg` carries a `reply` inbox; `msg.respond(bytes)` publishes the answer there.\n * A handler that throws (or a message with no `reply` subject) is left unanswered\n * — the requester then sees a timeout, never a malformed reply.\n */\nfunction subscribeRequests(nc: NatsConnection, subject: string, handler: RequestHandler): () => void {\n const sub: Subscription = nc.subscribe(subject);\n void (async () => {\n for await (const msg of sub) {\n // A request always carries a reply inbox; a plain publish to this subject\n // (no reply) is ignored — request/reply is the only contract here.\n if (!msg.reply) {\n continue;\n }\n try {\n const reply = await handler(msg.data, msg.subject);\n msg.respond(reply);\n } catch {\n // Leave the request unanswered: the requester's request times out, which\n // the selfhosted control plane reads as a transient blip (reconnecting),\n // never a malformed reply. The responder stays subscribed for the next op.\n }\n }\n })();\n return () => {\n sub.unsubscribe();\n };\n}\n\n/**\n * Subscribe to the one-way agent event plane: deliver each published payload (the\n * agent's `AgentEvent` heartbeat / going-offline, NOT a request/reply) to the\n * handler with its concrete subject. A plain `nc.subscribe` (no reply); a handler\n * that throws is swallowed so one bad event never tears down the subscription\n * (ingestion is best-effort — a metrics gap is never fatal).\n */\nfunction subscribeAgentEvents(\n nc: NatsConnection,\n subject: string,\n handler: (payload: Uint8Array, subject: string) => void | Promise<void>,\n): () => void {\n const sub: Subscription = nc.subscribe(subject);\n void (async () => {\n for await (const msg of sub) {\n try {\n await handler(msg.data, msg.subject);\n } catch {\n // Swallow: best-effort ingestion. The subscription stays live for the\n // next event.\n }\n }\n })();\n return () => {\n sub.unsubscribe();\n };\n}\n\nexport function formatSse(event: SessionEvent): string {\n return [\n `id: ${event.sequence}`,\n `event: ${event.type}`,\n `data: ${JSON.stringify(event)}`,\n \"\",\n \"\",\n ].join(\"\\n\");\n}\n","import type { SessionEvent } from \"@opengeni/contracts\";\n\nconst COALESCIBLE_DELTA_TYPES = new Set([\n \"agent.message.delta\",\n \"agent.reasoning.delta\",\n \"sandbox.command.output.delta\",\n]);\n\ntype DeltaRun = {\n first: SessionEvent;\n lastSequence: number;\n text: string;\n sandboxName: string | undefined;\n};\n\nexport function coalesceSessionEventDeltas(events: SessionEvent[]): SessionEvent[] {\n const coalesced: SessionEvent[] = [];\n let run: DeltaRun | null = null;\n\n const flush = () => {\n if (!run) {\n return;\n }\n coalesced.push({\n ...run.first,\n payload: {\n text: run.text,\n coalescedUntil: run.lastSequence,\n ...(run.first.type === \"sandbox.command.output.delta\" && run.sandboxName !== undefined\n ? { name: run.sandboxName }\n : {}),\n },\n });\n run = null;\n };\n\n for (const event of events) {\n if (!isCoalescibleDelta(event)) {\n flush();\n coalesced.push(event);\n continue;\n }\n\n const sandboxName = event.type === \"sandbox.command.output.delta\" ? sandboxDeltaName(event.payload) : undefined;\n if (run && sameDeltaRun(run.first, event, run.sandboxName, sandboxName)) {\n run.text += deltaText(event);\n run.lastSequence = event.sequence;\n continue;\n }\n\n flush();\n run = {\n first: event,\n lastSequence: event.sequence,\n text: deltaText(event),\n sandboxName,\n };\n }\n\n flush();\n return coalesced;\n}\n\nfunction isCoalescibleDelta(event: SessionEvent): boolean {\n return COALESCIBLE_DELTA_TYPES.has(event.type);\n}\n\nfunction sameDeltaRun(\n first: SessionEvent,\n next: SessionEvent,\n firstSandboxName: string | undefined,\n nextSandboxName: string | undefined,\n): boolean {\n if (first.type !== next.type) {\n return false;\n }\n if ((first.turnId ?? null) !== (next.turnId ?? null)) {\n return false;\n }\n return first.type !== \"sandbox.command.output.delta\" || firstSandboxName === nextSandboxName;\n}\n\nfunction deltaText(event: SessionEvent): string {\n if (event.type === \"agent.reasoning.delta\") {\n return reasoningText(event.payload);\n }\n const payload = asRecord(event.payload);\n if (event.type === \"sandbox.command.output.delta\") {\n return typeof payload.text === \"string\" ? payload.text : typeof payload.output === \"string\" ? payload.output : \"\";\n }\n return typeof payload.text === \"string\" ? payload.text : \"\";\n}\n\nfunction reasoningText(payload: unknown): string {\n const record = asRecord(payload);\n if (typeof record.text === \"string\") {\n return record.text;\n }\n const content = asRecord(asRecord(record.item).rawItem).content;\n if (!Array.isArray(content)) {\n return \"\";\n }\n return content\n .map((part) => {\n const text = asRecord(part).text;\n return typeof text === \"string\" ? text : \"\";\n })\n .join(\"\");\n}\n\nfunction sandboxDeltaName(payload: unknown): string | undefined {\n const name = asRecord(payload).name;\n return typeof name === \"string\" ? name : undefined;\n}\n\nfunction asRecord(value: unknown): Record<string, unknown> {\n return value && typeof value === \"object\" ? value as Record<string, unknown> : {};\n}\n","// packages/events/src/nats-jwt.ts — NATS JWT v2 signing for the auth-callout\n// responder (bring-your-own-compute M-AUTH; dossier §10.1 NATS Accounts per\n// workspace + §17 the isolation smoke).\n//\n// This is the cryptographic core of the auth-callout tenancy boundary. When an\n// external agent connects to NATS presenting its `oge_` enrollment bearer as the\n// connect auth-token, nats-server (configured with `auth_callout`) issues an\n// authorization request on `$SYS.REQ.USER.AUTH`. Our responder (auth-callout.ts)\n// validates the bearer and answers with a SIGNED authorization-response JWT that\n// embeds a SIGNED user JWT scoping the connection to publish/subscribe ONLY\n// `agent.<workspaceId>.>` (+ the reply `_INBOX.>`). That per-subject permission\n// set IS the per-workspace isolation: workspace A's agent literally cannot\n// pub/sub workspace B's subjects (§19 the NATS-Accounts-misconfig leak risk is\n// closed at the JWT-permission layer, not just by subject naming).\n//\n// WHY HAND-ROLL THE JWT ENCODING (vs a dep): the NATS JWT v2 wire format is small,\n// stable, and fully specified (ADR-26 + nats-io/jwt): a base64url header\n// `{\"typ\":\"JWT\",\"alg\":\"ed25519-nkey\"}`, base64url JSON claims whose `jti` is the\n// base32(SHA-512/256(claims-with-blank-jti)), and an ed25519 nkey signature over\n// `header.payload`. nkeys (re-exported by the `nats` package we already depend on)\n// gives us the ed25519 sign primitive; Node `crypto` gives SHA-512/256. So we own\n// the encoding in a few well-tested functions rather than pull an alpha\n// `@nats-io/jwt` (0.0.x) whose nkeys-version compat is uncertain. No `xkey`\n// encryption is used (the bearer is already an authenticated identity claim and\n// the wire is TLS — encryption is an optional ADR-26 hardening, off here).\n//\n// SECURITY: the account SIGNING SEED never leaves this process and is NEVER logged.\n// Callers pass it as a `string` seed; we `fromSeed` it once per sign. The bearer\n// the responder validates is HMAC-verified elsewhere (verifyEnrollmentBearer); this\n// module only mints the scoped NATS credential once identity is proven.\n\nimport { createHash } from \"node:crypto\";\nimport { nkeys } from \"nats\";\n\n/** The NATS JWT v2 header — constant for every token we mint (ADR-26 / nats-io/jwt:\n * `TokenTypeJwt=\"JWT\"`, `AlgorithmNkey=\"ed25519-nkey\"`). */\nconst JWT_HEADER = { typ: \"JWT\", alg: \"ed25519-nkey\" } as const;\n\n/** NATS user-claim `nats.type` discriminator + `nats.version` for v2 claims. */\nconst USER_CLAIM_TYPE = \"user\";\nconst AUTH_RESPONSE_CLAIM_TYPE = \"authorization_response\";\nconst NATS_CLAIM_VERSION = 2;\n\n/** A NATS permission set: subject allow/deny lists (ADR-26 `pub`/`sub` →\n * `allow`/`deny`). An empty/undefined list means \"no explicit grant\" — combined\n * with the agent scope below, the connection can ONLY reach what `allow` lists. */\nexport interface NatsPermission {\n allow?: string[];\n deny?: string[];\n}\n\n/** The pub/sub permissions embedded in a user JWT. */\nexport interface NatsPermissions {\n pub: NatsPermission;\n sub: NatsPermission;\n}\n\n/**\n * The minimal nkey keypair surface this module needs — exactly what\n * `nkeys.fromSeed(seed)` returns. Declared structurally so the module does not\n * leak the `nats` nkeys type through its public signature.\n */\ninterface NkeyPair {\n getPublicKey(): string;\n sign(input: Uint8Array): Uint8Array;\n}\n\n/** base64url (RawURLEncoding — no padding), matching nats-io/jwt's `serialize`. */\nfunction base64UrlEncode(bytes: Uint8Array): string {\n return Buffer.from(bytes).toString(\"base64url\");\n}\n\n/** RFC 4648 base32 (standard alphabet, NO padding) — the encoding nats-io/jwt\n * uses for the `jti` hash. Node has no built-in base32, so a tiny encoder. */\nconst BASE32_ALPHABET = \"ABCDEFGHIJKLMNOPQRSTUVWXYZ234567\";\nfunction base32NoPadding(bytes: Uint8Array): string {\n let bits = 0;\n let value = 0;\n let out = \"\";\n for (const byte of bytes) {\n value = (value << 8) | byte;\n bits += 8;\n while (bits >= 5) {\n bits -= 5;\n out += BASE32_ALPHABET[(value >>> bits) & 31];\n }\n }\n if (bits > 0) {\n out += BASE32_ALPHABET[(value << (5 - bits)) & 31];\n }\n return out;\n}\n\n/**\n * Compute the canonical NATS `jti`: base32(NoPadding, std-alphabet) of the\n * SHA-512/256 of the claims object SERIALIZED WITH AN EMPTY `jti` (nats-io/jwt's\n * `hash`). nats-server recomputes + verifies this on decode, so it must match\n * byte-for-byte. We serialize the SAME object we will sign, only with `jti:\"\"`.\n */\nfunction computeJti(claimsWithBlankJti: object): string {\n const json = JSON.stringify(claimsWithBlankJti);\n const digest = createHash(\"sha512-256\").update(json, \"utf8\").digest();\n return base32NoPadding(digest);\n}\n\n/**\n * Encode + sign a NATS v2 JWT. The `claims` MUST already carry `iss`/`sub`/`iat`\n * (+ optional `aud`/`exp`) and a `nats` block; this function fills `jti` (the\n * canonical hash), serializes `header.payload`, signs that with `signingKey`, and\n * appends the base64url signature. Returns the compact `header.payload.signature`.\n */\nfunction encodeJwt(claims: Record<string, unknown>, signingKey: NkeyPair): string {\n // jti is the hash of the claims with jti blanked — set it blank, hash, then set.\n const withBlankJti = { ...claims, jti: \"\" };\n const jti = computeJti(withBlankJti);\n const finalClaims = { ...claims, jti };\n\n const header = base64UrlEncode(Buffer.from(JSON.stringify(JWT_HEADER), \"utf8\"));\n const payload = base64UrlEncode(Buffer.from(JSON.stringify(finalClaims), \"utf8\"));\n const signingInput = `${header}.${payload}`;\n const signature = signingKey.sign(Buffer.from(signingInput, \"utf8\"));\n return `${signingInput}.${base64UrlEncode(signature)}`;\n}\n\n/**\n * Input to mint a workspace-scoped NATS user JWT for an enrolled agent.\n * - `userPublicKey` — the `user_nkey` from the authorization request; it MUST be\n * the `sub` of the user JWT (nats-server rejects a mismatch).\n * - `accountSeed` — the callout account SIGNING seed (`SA...`); both the user JWT\n * `iss` (its public key) and the signature come from it. NEVER logged.\n * - `name` — a human label for the user (the agent id), for server logs.\n * - `permissions` — the pub/sub allow/deny lists (the workspace scope).\n * - `expiresAtSeconds` — optional absolute `exp` (unix seconds). When set the\n * server will expire the connection's credential; we tie it to the bearer's\n * remaining life so a revoked/expired enrollment cannot outlive its bearer.\n */\nexport interface MintUserJwtInput {\n userPublicKey: string;\n accountSeed: string;\n name: string;\n permissions: NatsPermissions;\n /** The target account NAME (the `auth_callout.account`) the user binds to; the\n * embedded user JWT's `aud` in server-config mode. */\n audienceAccount: string;\n expiresAtSeconds?: number;\n}\n\n/**\n * Mint a signed NATS user JWT scoped by `permissions`. In auth-callout SERVER\n * mode the user JWT is signed by the callout ISSUER ACCOUNT key, and its `iss` is\n * that account's public key. The returned JWT is embedded as `nats.jwt` in the\n * authorization response.\n */\nexport function mintUserJwt(input: MintUserJwtInput): string {\n const accountKey = nkeys.fromSeed(Buffer.from(input.accountSeed)) as unknown as NkeyPair;\n const accountPublicKey = accountKey.getPublicKey();\n const nowSeconds = Math.floor(Date.now() / 1000);\n\n const natsBlock: Record<string, unknown> = {\n type: USER_CLAIM_TYPE,\n version: NATS_CLAIM_VERSION,\n pub: input.permissions.pub,\n sub: input.permissions.sub,\n // Unlimited subscriptions / data / payload (the workspace subject scope, NOT\n // a connection-resource quota, is the boundary here).\n subs: -1,\n data: -1,\n payload: -1,\n };\n\n const claims: Record<string, unknown> = {\n jti: \"\",\n iat: nowSeconds,\n iss: accountPublicKey,\n name: input.name,\n sub: input.userPublicKey,\n // SERVER-config-mode placement: nats-server reads the embedded user JWT's `aud`\n // as the target account NAME (the configured `auth_callout.account`). This is\n // how the authenticated user binds to that account; the workspace isolation is\n // then carried by the pub/sub permissions below.\n aud: input.audienceAccount,\n nats: natsBlock,\n };\n if (typeof input.expiresAtSeconds === \"number\") {\n claims.exp = input.expiresAtSeconds;\n }\n return encodeJwt(claims, accountKey);\n}\n\n/**\n * Input to mint the authorization RESPONSE JWT the responder publishes back on the\n * request's reply subject (ADR-26 §3).\n * - `userPublicKey` — the request's `user_nkey`; the response `sub`.\n * - `serverId` — the request's `nats.server_id.id` (the server's public key); the\n * response `aud`.\n * - `accountSeed` — the callout account signing seed; signs the response and is\n * its `iss` (public key). NEVER logged.\n * - `userJwt` — the embedded signed user JWT (omit on a denial).\n * - `error` — a human-readable denial message (omit on success). When present the\n * server denies the connection.\n */\nexport interface MintAuthResponseInput {\n userPublicKey: string;\n serverId: string;\n accountSeed: string;\n userJwt?: string;\n error?: string;\n}\n\n/**\n * Mint the signed authorization-response JWT. On success it carries the embedded\n * user JWT (`nats.jwt`); on denial it carries `nats.error` and NO user JWT, which\n * makes nats-server refuse the connection. Signed by the callout account key (its\n * public key is `iss`); `sub` is the user_nkey, `aud` is the server id.\n */\nexport function mintAuthResponse(input: MintAuthResponseInput): string {\n const accountKey = nkeys.fromSeed(Buffer.from(input.accountSeed)) as unknown as NkeyPair;\n const accountPublicKey = accountKey.getPublicKey();\n const nowSeconds = Math.floor(Date.now() / 1000);\n\n const natsBlock: Record<string, unknown> = {\n type: AUTH_RESPONSE_CLAIM_TYPE,\n version: NATS_CLAIM_VERSION,\n };\n if (input.userJwt) {\n natsBlock.jwt = input.userJwt;\n }\n if (input.error) {\n natsBlock.error = input.error;\n }\n\n const claims: Record<string, unknown> = {\n jti: \"\",\n iat: nowSeconds,\n iss: accountPublicKey,\n // The response `aud` MUST be the SERVER public key in server-config mode\n // (nats-server validates \"Audience must be a server public key\"). The\n // authenticated user is placed into the configured `auth_callout.account` (the\n // SAME account the responder + the privileged control plane connect into), so\n // `agent.<ws>.<id>.rpc` request/reply routes; the workspace isolation is carried\n // entirely by the user JWT's pub/sub subject permissions (NOT by cross-account\n // placement, which server-config-mode nats does not support — nats-io#4335).\n aud: input.serverId,\n sub: input.userPublicKey,\n nats: natsBlock,\n };\n return encodeJwt(claims, accountKey);\n}\n\n/**\n * The fields the responder needs out of the authorization REQUEST JWT (ADR-26 §2).\n * The request is itself a NATS JWT (`header.payload.signature`) the server signs;\n * we only DECODE it (the server proves its own identity by the connection, and the\n * embedded `auth_token` is independently HMAC-verified), so we read the payload\n * without re-verifying the server signature.\n */\nexport interface DecodedAuthRequest {\n /** The public user nkey the response user JWT MUST be `sub`-scoped to. */\n userNkey: string;\n /** The server's public id — the response `aud`. */\n serverId: string;\n /** The connect `auth_token` the client presented (our `oge_` bearer), if any. */\n authToken: string | undefined;\n /** The connect username, if any (unused today; present for completeness). */\n user: string | undefined;\n}\n\n/**\n * Decode the authorization-request JWT payload (the middle base64url segment). The\n * request shape (ADR-26 §2): `nats.user_nkey`, `nats.server_id.id`, and the\n * presented connect options under `nats.connect_opts` (`auth_token` / `user`).\n * Returns null on a malformed token so the caller can deny cleanly.\n */\nexport function decodeAuthRequest(token: string): DecodedAuthRequest | null {\n const parts = token.split(\".\");\n if (parts.length !== 3) {\n return null;\n }\n let payload: unknown;\n try {\n payload = JSON.parse(Buffer.from(parts[1]!, \"base64url\").toString(\"utf8\"));\n } catch {\n return null;\n }\n if (typeof payload !== \"object\" || payload === null) {\n return null;\n }\n const nats = (payload as { nats?: unknown }).nats;\n if (typeof nats !== \"object\" || nats === null) {\n return null;\n }\n const natsObj = nats as {\n user_nkey?: unknown;\n server_id?: { id?: unknown } | unknown;\n connect_opts?: { auth_token?: unknown; user?: unknown } | unknown;\n };\n const userNkey = typeof natsObj.user_nkey === \"string\" ? natsObj.user_nkey : null;\n if (!userNkey) {\n return null;\n }\n const serverIdRaw =\n typeof natsObj.server_id === \"object\" && natsObj.server_id !== null\n ? (natsObj.server_id as { id?: unknown }).id\n : undefined;\n const serverId = typeof serverIdRaw === \"string\" ? serverIdRaw : \"\";\n const connectOpts =\n typeof natsObj.connect_opts === \"object\" && natsObj.connect_opts !== null\n ? (natsObj.connect_opts as { auth_token?: unknown; user?: unknown })\n : {};\n const authToken = typeof connectOpts.auth_token === \"string\" ? connectOpts.auth_token : undefined;\n const user = typeof connectOpts.user === \"string\" ? connectOpts.user : undefined;\n return { userNkey, serverId, authToken, user };\n}\n\n/**\n * Build the workspace-scoped permission set for an agent: it may publish + subscribe\n * ONLY `agent.<workspaceId>.>` (its own RPC/event/hello subtree) and the reply\n * `_INBOX.>` subtree (so request/reply round-trips work). Everything else is\n * implicitly denied (an allow-list with no other entries IS the deny-all-else).\n *\n * THE isolation assertion (§17): with `workspaceId=A`, the returned allow lists name\n * only `agent.A.>` — so a connection bearing this credential is rejected by\n * nats-server the instant it tries to pub/sub `agent.B.>`. This is the per-workspace\n * tenancy boundary, enforced cryptographically by the signed JWT, not by naming.\n */\nexport function workspaceAgentPermissions(workspaceId: string): NatsPermissions {\n const agentScope = `agent.${workspaceId}.>`;\n // The reply-inbox subtree must be reachable for request/reply (the control plane\n // requests on agent.<ws>.<id>.rpc with a reply inbox; the agent responds there).\n const inboxScope = \"_INBOX.>\";\n return {\n pub: { allow: [agentScope, inboxScope] },\n sub: { allow: [agentScope, inboxScope] },\n };\n}\n"],"mappings":";AACA,SAAS,qBAAqB,sBAA4D;AAC1F,SAAS,SAAS,iBAA2F;;;ACA7G,IAAM,0BAA0B,oBAAI,IAAI;AAAA,EACtC;AAAA,EACA;AAAA,EACA;AACF,CAAC;AASM,SAAS,2BAA2B,QAAwC;AACjF,QAAM,YAA4B,CAAC;AACnC,MAAI,MAAuB;AAE3B,QAAM,QAAQ,MAAM;AAClB,QAAI,CAAC,KAAK;AACR;AAAA,IACF;AACA,cAAU,KAAK;AAAA,MACb,GAAG,IAAI;AAAA,MACP,SAAS;AAAA,QACP,MAAM,IAAI;AAAA,QACV,gBAAgB,IAAI;AAAA,QACpB,GAAI,IAAI,MAAM,SAAS,kCAAkC,IAAI,gBAAgB,SACzE,EAAE,MAAM,IAAI,YAAY,IACxB,CAAC;AAAA,MACP;AAAA,IACF,CAAC;AACD,UAAM;AAAA,EACR;AAEA,aAAW,SAAS,QAAQ;AAC1B,QAAI,CAAC,mBAAmB,KAAK,GAAG;AAC9B,YAAM;AACN,gBAAU,KAAK,KAAK;AACpB;AAAA,IACF;AAEA,UAAM,cAAc,MAAM,SAAS,iCAAiC,iBAAiB,MAAM,OAAO,IAAI;AACtG,QAAI,OAAO,aAAa,IAAI,OAAO,OAAO,IAAI,aAAa,WAAW,GAAG;AACvE,UAAI,QAAQ,UAAU,KAAK;AAC3B,UAAI,eAAe,MAAM;AACzB;AAAA,IACF;AAEA,UAAM;AACN,UAAM;AAAA,MACJ,OAAO;AAAA,MACP,cAAc,MAAM;AAAA,MACpB,MAAM,UAAU,KAAK;AAAA,MACrB;AAAA,IACF;AAAA,EACF;AAEA,QAAM;AACN,SAAO;AACT;AAEA,SAAS,mBAAmB,OAA8B;AACxD,SAAO,wBAAwB,IAAI,MAAM,IAAI;AAC/C;AAEA,SAAS,aACP,OACA,MACA,kBACA,iBACS;AACT,MAAI,MAAM,SAAS,KAAK,MAAM;AAC5B,WAAO;AAAA,EACT;AACA,OAAK,MAAM,UAAU,WAAW,KAAK,UAAU,OAAO;AACpD,WAAO;AAAA,EACT;AACA,SAAO,MAAM,SAAS,kCAAkC,qBAAqB;AAC/E;AAEA,SAAS,UAAU,OAA6B;AAC9C,MAAI,MAAM,SAAS,yBAAyB;AAC1C,WAAO,cAAc,MAAM,OAAO;AAAA,EACpC;AACA,QAAM,UAAU,SAAS,MAAM,OAAO;AACtC,MAAI,MAAM,SAAS,gCAAgC;AACjD,WAAO,OAAO,QAAQ,SAAS,WAAW,QAAQ,OAAO,OAAO,QAAQ,WAAW,WAAW,QAAQ,SAAS;AAAA,EACjH;AACA,SAAO,OAAO,QAAQ,SAAS,WAAW,QAAQ,OAAO;AAC3D;AAEA,SAAS,cAAc,SAA0B;AAC/C,QAAM,SAAS,SAAS,OAAO;AAC/B,MAAI,OAAO,OAAO,SAAS,UAAU;AACnC,WAAO,OAAO;AAAA,EAChB;AACA,QAAM,UAAU,SAAS,SAAS,OAAO,IAAI,EAAE,OAAO,EAAE;AACxD,MAAI,CAAC,MAAM,QAAQ,OAAO,GAAG;AAC3B,WAAO;AAAA,EACT;AACA,SAAO,QACJ,IAAI,CAAC,SAAS;AACb,UAAM,OAAO,SAAS,IAAI,EAAE;AAC5B,WAAO,OAAO,SAAS,WAAW,OAAO;AAAA,EAC3C,CAAC,EACA,KAAK,EAAE;AACZ;AAEA,SAAS,iBAAiB,SAAsC;AAC9D,QAAM,OAAO,SAAS,OAAO,EAAE;AAC/B,SAAO,OAAO,SAAS,WAAW,OAAO;AAC3C;AAEA,SAAS,SAAS,OAAyC;AACzD,SAAO,SAAS,OAAO,UAAU,WAAW,QAAmC,CAAC;AAClF;;;ACtFA,SAAS,kBAAkB;AAC3B,SAAS,aAAa;AAItB,IAAM,aAAa,EAAE,KAAK,OAAO,KAAK,eAAe;AAGrD,IAAM,kBAAkB;AACxB,IAAM,2BAA2B;AACjC,IAAM,qBAAqB;AA2B3B,SAAS,gBAAgB,OAA2B;AAClD,SAAO,OAAO,KAAK,KAAK,EAAE,SAAS,WAAW;AAChD;AAIA,IAAM,kBAAkB;AACxB,SAAS,gBAAgB,OAA2B;AAClD,MAAI,OAAO;AACX,MAAI,QAAQ;AACZ,MAAI,MAAM;AACV,aAAW,QAAQ,OAAO;AACxB,YAAS,SAAS,IAAK;AACvB,YAAQ;AACR,WAAO,QAAQ,GAAG;AAChB,cAAQ;AACR,aAAO,gBAAiB,UAAU,OAAQ,EAAE;AAAA,IAC9C;AAAA,EACF;AACA,MAAI,OAAO,GAAG;AACZ,WAAO,gBAAiB,SAAU,IAAI,OAAS,EAAE;AAAA,EACnD;AACA,SAAO;AACT;AAQA,SAAS,WAAW,oBAAoC;AACtD,QAAM,OAAO,KAAK,UAAU,kBAAkB;AAC9C,QAAM,SAAS,WAAW,YAAY,EAAE,OAAO,MAAM,MAAM,EAAE,OAAO;AACpE,SAAO,gBAAgB,MAAM;AAC/B;AAQA,SAAS,UAAU,QAAiC,YAA8B;AAEhF,QAAM,eAAe,EAAE,GAAG,QAAQ,KAAK,GAAG;AAC1C,QAAM,MAAM,WAAW,YAAY;AACnC,QAAM,cAAc,EAAE,GAAG,QAAQ,IAAI;AAErC,QAAM,SAAS,gBAAgB,OAAO,KAAK,KAAK,UAAU,UAAU,GAAG,MAAM,CAAC;AAC9E,QAAM,UAAU,gBAAgB,OAAO,KAAK,KAAK,UAAU,WAAW,GAAG,MAAM,CAAC;AAChF,QAAM,eAAe,GAAG,MAAM,IAAI,OAAO;AACzC,QAAM,YAAY,WAAW,KAAK,OAAO,KAAK,cAAc,MAAM,CAAC;AACnE,SAAO,GAAG,YAAY,IAAI,gBAAgB,SAAS,CAAC;AACtD;AA+BO,SAAS,YAAY,OAAiC;AAC3D,QAAM,aAAa,MAAM,SAAS,OAAO,KAAK,MAAM,WAAW,CAAC;AAChE,QAAM,mBAAmB,WAAW,aAAa;AACjD,QAAM,aAAa,KAAK,MAAM,KAAK,IAAI,IAAI,GAAI;AAE/C,QAAM,YAAqC;AAAA,IACzC,MAAM;AAAA,IACN,SAAS;AAAA,IACT,KAAK,MAAM,YAAY;AAAA,IACvB,KAAK,MAAM,YAAY;AAAA;AAAA;AAAA,IAGvB,MAAM;AAAA,IACN,MAAM;AAAA,IACN,SAAS;AAAA,EACX;AAEA,QAAM,SAAkC;AAAA,IACtC,KAAK;AAAA,IACL,KAAK;AAAA,IACL,KAAK;AAAA,IACL,MAAM,MAAM;AAAA,IACZ,KAAK,MAAM;AAAA;AAAA;AAAA;AAAA;AAAA,IAKX,KAAK,MAAM;AAAA,IACX,MAAM;AAAA,EACR;AACA,MAAI,OAAO,MAAM,qBAAqB,UAAU;AAC9C,WAAO,MAAM,MAAM;AAAA,EACrB;AACA,SAAO,UAAU,QAAQ,UAAU;AACrC;AA4BO,SAAS,iBAAiB,OAAsC;AACrE,QAAM,aAAa,MAAM,SAAS,OAAO,KAAK,MAAM,WAAW,CAAC;AAChE,QAAM,mBAAmB,WAAW,aAAa;AACjD,QAAM,aAAa,KAAK,MAAM,KAAK,IAAI,IAAI,GAAI;AAE/C,QAAM,YAAqC;AAAA,IACzC,MAAM;AAAA,IACN,SAAS;AAAA,EACX;AACA,MAAI,MAAM,SAAS;AACjB,cAAU,MAAM,MAAM;AAAA,EACxB;AACA,MAAI,MAAM,OAAO;AACf,cAAU,QAAQ,MAAM;AAAA,EAC1B;AAEA,QAAM,SAAkC;AAAA,IACtC,KAAK;AAAA,IACL,KAAK;AAAA,IACL,KAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,IAQL,KAAK,MAAM;AAAA,IACX,KAAK,MAAM;AAAA,IACX,MAAM;AAAA,EACR;AACA,SAAO,UAAU,QAAQ,UAAU;AACrC;AA0BO,SAAS,kBAAkB,OAA0C;AAC1E,QAAM,QAAQ,MAAM,MAAM,GAAG;AAC7B,MAAI,MAAM,WAAW,GAAG;AACtB,WAAO;AAAA,EACT;AACA,MAAI;AACJ,MAAI;AACF,cAAU,KAAK,MAAM,OAAO,KAAK,MAAM,CAAC,GAAI,WAAW,EAAE,SAAS,MAAM,CAAC;AAAA,EAC3E,QAAQ;AACN,WAAO;AAAA,EACT;AACA,MAAI,OAAO,YAAY,YAAY,YAAY,MAAM;AACnD,WAAO;AAAA,EACT;AACA,QAAM,OAAQ,QAA+B;AAC7C,MAAI,OAAO,SAAS,YAAY,SAAS,MAAM;AAC7C,WAAO;AAAA,EACT;AACA,QAAM,UAAU;AAKhB,QAAM,WAAW,OAAO,QAAQ,cAAc,WAAW,QAAQ,YAAY;AAC7E,MAAI,CAAC,UAAU;AACb,WAAO;AAAA,EACT;AACA,QAAM,cACJ,OAAO,QAAQ,cAAc,YAAY,QAAQ,cAAc,OAC1D,QAAQ,UAA+B,KACxC;AACN,QAAM,WAAW,OAAO,gBAAgB,WAAW,cAAc;AACjE,QAAM,cACJ,OAAO,QAAQ,iBAAiB,YAAY,QAAQ,iBAAiB,OAChE,QAAQ,eACT,CAAC;AACP,QAAM,YAAY,OAAO,YAAY,eAAe,WAAW,YAAY,aAAa;AACxF,QAAM,OAAO,OAAO,YAAY,SAAS,WAAW,YAAY,OAAO;AACvE,SAAO,EAAE,UAAU,UAAU,WAAW,KAAK;AAC/C;AAaO,SAAS,0BAA0B,aAAsC;AAC9E,QAAM,aAAa,SAAS,WAAW;AAGvC,QAAM,aAAa;AACnB,SAAO;AAAA,IACL,KAAK,EAAE,OAAO,CAAC,YAAY,UAAU,EAAE;AAAA,IACvC,KAAK,EAAE,OAAO,CAAC,YAAY,UAAU,EAAE;AAAA,EACzC;AACF;;;AF5NA,SAAS,WAAAA,UAAS,SAAAC,cAAkC;AA9GpD,IAAM,QAAQ,UAA4C;AA0B1D,IAAM,oBAAoB;AAAA,EACxB,WAAW;AAAA,EACX,sBAAsB;AAAA,EACtB,mBAAmB;AAAA,EACnB,iBAAiB;AAAA,EACjB,oBAAoB;AAAA,EACpB,oBAAoB;AAAA,EACpB,cAAc;AAAA,EACd,YAAY;AACd;AAQA,SAAS,sBAAsB,SAA+C;AAC5E,SAAO,EAAE,GAAG,mBAAmB,GAAG,QAAQ;AAC5C;AAGA,IAAM,2BAA2B;AAWjC,eAAe,iBAAiB,IAAoB,WAAkC;AACpF,MAAI;AACJ,QAAM,UAAU,IAAI,QAAc,CAAC,YAAY;AAC7C,YAAQ,WAAW,SAAS,SAAS;AAAA,EACvC,CAAC;AACD,MAAI;AACF,UAAM,QAAQ,KAAK,CAAC,GAAG,MAAM,EAAE,MAAM,MAAM,MAAS,GAAG,OAAO,CAAC;AAAA,EACjE,UAAE;AACA,QAAI,OAAO;AACT,mBAAa,KAAK;AAAA,IACpB;AAAA,EACF;AACF;AASA,SAAS,oBAAoB,IAAoB,OAAqB;AACpE,QAAM,YAAY;AAChB,QAAI;AACF,uBAAiB,UAAU,GAAG,OAAO,GAAG;AACtC,gBAAQ,KAAK,SAAS,KAAK,KAAK,OAAO,IAAI,IAAI,OAAO,IAAI;AAAA,MAC5D;AAAA,IACF,QAAQ;AAAA,IAGR;AAAA,EACF,GAAG;AACL;AAkGA,eAAsB,mBACpB,SACA,MACmB;AACnB,QAAM,iBAAoC,EAAE,SAAS,QAAQ;AAC7D,MAAI,MAAM;AACR,mBAAe,OAAO,KAAK;AAC3B,mBAAe,OAAO,KAAK;AAAA,EAC7B;AACA,QAAM,KAAK,MAAM,QAAQ,sBAAsB,cAAc,CAAC;AAC9D,sBAAoB,IAAI,WAAW;AACnC,QAAM,oBAAuC;AAAA,IAC3C,SAAS,OAAO,SAAS,SAAS,SAAS,aAAa,IAAI,SAAS,SAAS,KAAK,OAAO;AAAA,EAC5F;AACA,SAAO;AAAA,IACL,SAAS,OAAO,aAAa,WAAW,WAAW;AACjD,UAAI,OAAO,WAAW,GAAG;AACvB;AAAA,MACF;AASA,UAAI;AACF,WAAG,QAAQ,eAAe,aAAa,SAAS,GAAG,MAAM,OAAO,EAAE,aAAa,WAAW,OAAO,CAAC,CAAC;AAAA,MACrG,SAAS,OAAO;AAGd,gBAAQ;AAAA,UACN,6CAA6C,WAAW,IAAI,SAAS;AAAA,UACrE;AAAA,QACF;AACA;AAAA,MACF;AACA,YAAM,iBAAiB,IAAI,wBAAwB;AAAA,IACrD;AAAA,IACA,WAAW,OAAO,aAAa,WAAW,aAAa,iBAAiB,IAAI,aAAa,WAAW,QAAQ;AAAA,IAC5G,SAAS,OAAO,SAAS,SAAS,SAAS,aAAa,IAAI,SAAS,SAAS,KAAK,SAAS;AAAA,IAC5F,mBAAmB,CAAC,SAAS,YAAY,kBAAkB,IAAI,SAAS,OAAO;AAAA,IAC/E,sBAAsB,CAAC,SAAS,YAAY,qBAAqB,IAAI,SAAS,OAAO;AAAA,IACrF,sBAAsB,MAAM;AAAA,IAC5B,OAAO,YAAY;AACjB,YAAM,GAAG,MAAM;AAAA,IACjB;AAAA,EACF;AACF;AAmCA,eAAsB,0BACpB,SACA,MACA,SACA,SACA,UAA6B,CAAC,GACA;AAC9B,QAAM,iBAAoC,EAAE,SAAS,QAAQ;AAC7D,MAAI,QAAQ,MAAM;AAChB,mBAAe,OAAO,QAAQ;AAAA,EAChC;AACA,MAAI,KAAK,SAAS,iBAAiB;AACjC,mBAAe,OAAO,KAAK;AAC3B,mBAAe,OAAO,KAAK;AAAA,EAC7B,WAAW,KAAK,SAAS,SAAS;AAChC,mBAAe,QAAQ,KAAK;AAAA,EAC9B;AACA,QAAM,KAAK,MAAM,QAAQ,sBAAsB,cAAc,CAAC;AAC9D,sBAAoB,IAAI,QAAQ,OAAO,gBAAgB,QAAQ,IAAI,KAAK,cAAc;AACtF,QAAM,MAAoB,GAAG,UAAU,OAAO;AAC9C,QAAM,YAAY;AAChB,qBAAiB,OAAO,KAAK;AAC3B,UAAI,CAAC,IAAI,OAAO;AACd;AAAA,MACF;AACA,UAAI;AACF,cAAM,QAAQ,MAAM,QAAQ,IAAI,MAAM,IAAI,OAAO;AACjD,YAAI,QAAQ,KAAK;AAAA,MACnB,QAAQ;AAAA,MAGR;AAAA,IACF;AAAA,EACF,GAAG;AACH,SAAO;AAAA,IACL,OAAO,YAAY;AACjB,UAAI,YAAY;AAChB,YAAM,GAAG,MAAM;AAAA,IACjB;AAAA,EACF;AACF;AAEA,eAAsB,uBAAuB,IAAc,KAAe,aAAqB,WAAmB,QAAqD;AACrK,QAAM,WAAW,MAAM,oBAAoB,IAAI,aAAa,WAAW,MAAM;AAQ7E,MAAI;AACF,UAAM,IAAI,QAAQ,aAAa,WAAW,QAAQ;AAAA,EACpD,SAAS,OAAO;AACd,YAAQ;AAAA,MACN,oCAAoC,WAAW,IAAI,SAAS,KAAK,SAAS,MAAM;AAAA,MAChF;AAAA,IACF;AAAA,EACF;AACA,SAAO;AACT;AAEA,SAAS,iBAAiB,IAAoB,aAAqB,WAAmB,UAAwE;AAC5J,QAAM,MAAoB,GAAG,UAAU,eAAe,aAAa,SAAS,CAAC;AAC7E,QAAM,YAAY;AAChB,qBAAiB,OAAO,KAAK;AAC3B,YAAM,UAAU,MAAM,OAAO,IAAI,IAAI;AACrC,YAAM,SAAS,YAAY,UAAU,QAAQ,SAAS,CAAC,OAAO;AAC9D,YAAM,SAAS,MAAM;AAAA,IACvB;AAAA,EACF,GAAG;AACH,SAAO,MAAM;AACX,QAAI,YAAY;AAAA,EAClB;AACF;AASA,eAAe,aAAa,IAAoB,SAAiB,SAAqB,SAAwC;AAC5H,QAAM,MAAW,MAAM,GAAG,QAAQ,SAAS,SAAS,EAAE,QAAQ,CAAC;AAC/D,SAAO,EAAE,MAAM,IAAI,KAAK;AAC1B;AASA,SAAS,kBAAkB,IAAoB,SAAiB,SAAqC;AACnG,QAAM,MAAoB,GAAG,UAAU,OAAO;AAC9C,QAAM,YAAY;AAChB,qBAAiB,OAAO,KAAK;AAG3B,UAAI,CAAC,IAAI,OAAO;AACd;AAAA,MACF;AACA,UAAI;AACF,cAAM,QAAQ,MAAM,QAAQ,IAAI,MAAM,IAAI,OAAO;AACjD,YAAI,QAAQ,KAAK;AAAA,MACnB,QAAQ;AAAA,MAIR;AAAA,IACF;AAAA,EACF,GAAG;AACH,SAAO,MAAM;AACX,QAAI,YAAY;AAAA,EAClB;AACF;AASA,SAAS,qBACP,IACA,SACA,SACY;AACZ,QAAM,MAAoB,GAAG,UAAU,OAAO;AAC9C,QAAM,YAAY;AAChB,qBAAiB,OAAO,KAAK;AAC3B,UAAI;AACF,cAAM,QAAQ,IAAI,MAAM,IAAI,OAAO;AAAA,MACrC,QAAQ;AAAA,MAGR;AAAA,IACF;AAAA,EACF,GAAG;AACH,SAAO,MAAM;AACX,QAAI,YAAY;AAAA,EAClB;AACF;AAEO,SAAS,UAAU,OAA6B;AACrD,SAAO;AAAA,IACL,OAAO,MAAM,QAAQ;AAAA,IACrB,UAAU,MAAM,IAAI;AAAA,IACpB,SAAS,KAAK,UAAU,KAAK,CAAC;AAAA,IAC9B;AAAA,IACA;AAAA,EACF,EAAE,KAAK,IAAI;AACb;","names":["connect","nkeys"]}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@opengeni/events",
3
- "version": "0.2.2",
3
+ "version": "0.2.4",
4
4
  "type": "module",
5
5
  "main": "./dist/index.js",
6
6
  "module": "./dist/index.js",
@@ -24,8 +24,8 @@
24
24
  "typecheck": "tsc --noEmit"
25
25
  },
26
26
  "dependencies": {
27
- "@opengeni/contracts": "^0.4.0",
28
- "@opengeni/db": "^0.2.2",
27
+ "@opengeni/contracts": "^0.6.0",
28
+ "@opengeni/db": "^0.4.0",
29
29
  "nats": "^2.29.3"
30
30
  },
31
31
  "devDependencies": {
@@ -0,0 +1,118 @@
1
+ import type { SessionEvent } from "@opengeni/contracts";
2
+
3
+ const COALESCIBLE_DELTA_TYPES = new Set([
4
+ "agent.message.delta",
5
+ "agent.reasoning.delta",
6
+ "sandbox.command.output.delta",
7
+ ]);
8
+
9
+ type DeltaRun = {
10
+ first: SessionEvent;
11
+ lastSequence: number;
12
+ text: string;
13
+ sandboxName: string | undefined;
14
+ };
15
+
16
+ export function coalesceSessionEventDeltas(events: SessionEvent[]): SessionEvent[] {
17
+ const coalesced: SessionEvent[] = [];
18
+ let run: DeltaRun | null = null;
19
+
20
+ const flush = () => {
21
+ if (!run) {
22
+ return;
23
+ }
24
+ coalesced.push({
25
+ ...run.first,
26
+ payload: {
27
+ text: run.text,
28
+ coalescedUntil: run.lastSequence,
29
+ ...(run.first.type === "sandbox.command.output.delta" && run.sandboxName !== undefined
30
+ ? { name: run.sandboxName }
31
+ : {}),
32
+ },
33
+ });
34
+ run = null;
35
+ };
36
+
37
+ for (const event of events) {
38
+ if (!isCoalescibleDelta(event)) {
39
+ flush();
40
+ coalesced.push(event);
41
+ continue;
42
+ }
43
+
44
+ const sandboxName = event.type === "sandbox.command.output.delta" ? sandboxDeltaName(event.payload) : undefined;
45
+ if (run && sameDeltaRun(run.first, event, run.sandboxName, sandboxName)) {
46
+ run.text += deltaText(event);
47
+ run.lastSequence = event.sequence;
48
+ continue;
49
+ }
50
+
51
+ flush();
52
+ run = {
53
+ first: event,
54
+ lastSequence: event.sequence,
55
+ text: deltaText(event),
56
+ sandboxName,
57
+ };
58
+ }
59
+
60
+ flush();
61
+ return coalesced;
62
+ }
63
+
64
+ function isCoalescibleDelta(event: SessionEvent): boolean {
65
+ return COALESCIBLE_DELTA_TYPES.has(event.type);
66
+ }
67
+
68
+ function sameDeltaRun(
69
+ first: SessionEvent,
70
+ next: SessionEvent,
71
+ firstSandboxName: string | undefined,
72
+ nextSandboxName: string | undefined,
73
+ ): boolean {
74
+ if (first.type !== next.type) {
75
+ return false;
76
+ }
77
+ if ((first.turnId ?? null) !== (next.turnId ?? null)) {
78
+ return false;
79
+ }
80
+ return first.type !== "sandbox.command.output.delta" || firstSandboxName === nextSandboxName;
81
+ }
82
+
83
+ function deltaText(event: SessionEvent): string {
84
+ if (event.type === "agent.reasoning.delta") {
85
+ return reasoningText(event.payload);
86
+ }
87
+ const payload = asRecord(event.payload);
88
+ if (event.type === "sandbox.command.output.delta") {
89
+ return typeof payload.text === "string" ? payload.text : typeof payload.output === "string" ? payload.output : "";
90
+ }
91
+ return typeof payload.text === "string" ? payload.text : "";
92
+ }
93
+
94
+ function reasoningText(payload: unknown): string {
95
+ const record = asRecord(payload);
96
+ if (typeof record.text === "string") {
97
+ return record.text;
98
+ }
99
+ const content = asRecord(asRecord(record.item).rawItem).content;
100
+ if (!Array.isArray(content)) {
101
+ return "";
102
+ }
103
+ return content
104
+ .map((part) => {
105
+ const text = asRecord(part).text;
106
+ return typeof text === "string" ? text : "";
107
+ })
108
+ .join("");
109
+ }
110
+
111
+ function sandboxDeltaName(payload: unknown): string | undefined {
112
+ const name = asRecord(payload).name;
113
+ return typeof name === "string" ? name : undefined;
114
+ }
115
+
116
+ function asRecord(value: unknown): Record<string, unknown> {
117
+ return value && typeof value === "object" ? value as Record<string, unknown> : {};
118
+ }
package/src/index.ts CHANGED
@@ -4,6 +4,8 @@ import { connect, JSONCodec, type ConnectionOptions, type Msg, type NatsConnecti
4
4
 
5
5
  const codec = JSONCodec<SessionBusMessage | SessionEvent>();
6
6
 
7
+ export { coalesceSessionEventDeltas } from "./coalesce";
8
+
7
9
  /**
8
10
  * Reconnect + keepalive defaults applied to EVERY long-lived NATS connection
9
11
  * this package opens (the event bus AND the standalone auth-callout responder).