@codemation/core 0.12.0 → 0.13.1

package/dist/{runtime-BPZgnZ9G.js → runtime-iHBN1jyD.js}

@@ -1,4 +1,4 @@
-import { C as resolveItemExprsForExecution, T as ConnectionNodeIdFactory, _ as NodeIterationIdFactory, a as injectable, b as AgentConfigInspector, g as CredentialUnboundError, h as SuspensionRequest, v as AgentConnectionNodeCollector } from "./di-DdsgWfVy.js";
+import { E as ConnectionNodeIdFactory, _ as CredentialUnboundError, a as injectable, g as SuspensionRequest, h as BINARY_DEFAULT_MAX_BYTES, v as NodeIterationIdFactory, w as resolveItemExprsForExecution, x as AgentConfigInspector, y as AgentConnectionNodeCollector } from "./di-D9Mv3kF3.js";
 import { ZodError, z } from "zod";
 import { ReadableStream } from "node:stream/web";
 import { createHash } from "node:crypto";
@@ -890,6 +890,58 @@ var NodeEventPublisher = class {
 	}
 };
 
+//#endregion
+//#region src/binaries/boundedReadBinary.types.ts
+/**
+* Reads all bytes from an already-opened binary stream into a contiguous `Uint8Array`.
+*
+* Safety contract:
+*  - `attachment.size` is checked against `maxBytes` *before* any allocation (no OOM).
+*  - A single buffer of exactly `attachment.size` is pre-allocated; the stream fills it
+*    directly — no chunks array, no doubling.
+*  - A byte-count mismatch between the declared size and actual stream content is an error.
+*
+* This is the single canonical implementation; `ExecutionBinaryService.getBytes`,
+* `getText`, and `getJson` all delegate here. The per-package `readBinaryBody` helpers
+* in `core-nodes` and `core-nodes-ocr` have been removed in favour of this function.
+*/
+async function boundedReadBinary(result, attachment, maxBytes = BINARY_DEFAULT_MAX_BYTES) {
+	if (attachment.size > maxBytes) throw new Error(`Binary attachment size ${attachment.size} bytes exceeds maxBytes ${maxBytes}. Raise the node's maxBytes setting if this document is expected to be larger.`);
+	const out = new Uint8Array(attachment.size);
+	const reader = result.body.getReader();
+	let offset = 0;
+	while (true) {
+		const { done, value } = await reader.read();
+		if (done) break;
+		if (!value) continue;
+		if (offset + value.byteLength > out.byteLength) throw new Error(`Binary stream produced more bytes than the attachment's declared size (${attachment.size}).`);
+		out.set(value, offset);
+		offset += value.byteLength;
+	}
+	if (offset !== out.byteLength) throw new Error(`Binary stream produced ${offset} bytes but attachment declared size ${attachment.size}.`);
+	return out;
+}
+/** Shared implementation of `getBytes` used by both binary-service classes. */
+async function readBinaryAsBytes(storage, attachment, maxBytes) {
+	const result = await storage.openReadStream(attachment.storageKey);
+	if (!result) throw new Error("Binary attachment stream is unavailable.");
+	return boundedReadBinary(result, attachment, maxBytes);
+}
+/** Shared implementation of `getText` used by both binary-service classes. */
+async function readBinaryAsText(storage, attachment, maxBytes) {
+	const bytes = await readBinaryAsBytes(storage, attachment, maxBytes);
+	return new TextDecoder().decode(bytes);
+}
+/** Shared implementation of `getJson` used by both binary-service classes. */
+async function readBinaryAsJson(storage, attachment, maxBytes) {
+	const text = await readBinaryAsText(storage, attachment, maxBytes);
+	try {
+		return JSON.parse(text);
+	} catch (cause) {
+		throw new SyntaxError(`Binary attachment at storage key "${attachment.storageKey}" is not valid JSON: ${cause instanceof Error ? cause.message : String(cause)}`, { cause });
+	}
+}
+
 //#endregion
 //#region src/binaries/DefaultNodeBinaryAttachmentServiceFactory.ts
 var DefaultNodeBinaryAttachmentService = class DefaultNodeBinaryAttachmentService {
@@ -940,6 +992,15 @@ var DefaultNodeBinaryAttachmentService = class DefaultNodeBinaryAttachmentServic
 	async openReadStream(attachment) {
 		return await this.storage.openReadStream(attachment.storageKey);
 	}
+	async getBytes(attachment, maxBytes) {
+		return readBinaryAsBytes(this.storage, attachment, maxBytes);
+	}
+	async getText(attachment, maxBytes) {
+		return readBinaryAsText(this.storage, attachment, maxBytes);
+	}
+	async getJson(attachment, maxBytes) {
+		return readBinaryAsJson(this.storage, attachment, maxBytes);
+	}
 	createAttachmentId() {
 		return DefaultNodeBinaryAttachmentService.createAttachmentIdValue(`${this.activationId}-${this.now().getTime()}`);
 	}
@@ -1003,8 +1064,24 @@ var DefaultExecutionBinaryService = class {
 	forNode(args) {
 		return new DefaultNodeBinaryAttachmentService(this.storage, this.workflowId, this.runId, args.nodeId, args.activationId, this.now);
 	}
-	async openReadStream(attachment) {
-		return await this.storage.openReadStream(attachment.storageKey);
+	openReadStream(attachment) {
+		return this.storage.openReadStream(attachment.storageKey);
+	}
+	async getBytes(attachment, maxBytes) {
+		const stream = await this.openReadStream(attachment);
+		if (!stream) throw new Error("Binary attachment stream is unavailable.");
+		return boundedReadBinary(stream, attachment, maxBytes);
+	}
+	async getText(attachment, maxBytes) {
+		return new TextDecoder().decode(await this.getBytes(attachment, maxBytes));
+	}
+	async getJson(attachment, maxBytes) {
+		const text = await this.getText(attachment, maxBytes);
+		try {
+			return JSON.parse(text);
+		} catch (cause) {
+			throw new SyntaxError(`Binary attachment at storage key "${attachment.storageKey}" is not valid JSON: ${cause instanceof Error ? cause.message : String(cause)}`, { cause });
+		}
 	}
 };
 
@@ -4921,12 +4998,26 @@ var InlineDrivingScheduler = class {
 	drainingRuns = /* @__PURE__ */ new Set();
 	queuesByRunId = /* @__PURE__ */ new Map();
 	scheduledRuns = /* @__PURE__ */ new Set();
+	stopped = false;
+	activeDrainPromises = /* @__PURE__ */ new Set();
 	constructor(nodeExecutor) {
 		this.nodeExecutor = nodeExecutor;
 	}
 	setContinuation(continuation) {
 		this.continuation = continuation;
 	}
+	/**
+	* Prevents new drain cycles from being scheduled and waits for all currently-running
+	* drains to complete. Call before closing the DB connection or rolling back test transactions
+	* to ensure no in-flight writes hit a closed/rolled-back connection.
+	*
+	* Bounded: only the drains already in-flight at call time are awaited. Runs that keep
+	* re-scheduling will not schedule new work once stopped is set.
+	*/
+	async stop() {
+		this.stopped = true;
+		if (this.activeDrainPromises.size > 0) await Promise.allSettled(this.activeDrainPromises);
+	}
 	async prepareDispatch(request) {
 		const receipt = {
 			receiptId: request.activationId,
@@ -4976,11 +5067,15 @@ var InlineDrivingScheduler = class {
 		}
 	}
 	scheduleDrain(runId) {
-		if (this.drainingRuns.has(runId) || this.scheduledRuns.has(runId)) return;
+		if (this.stopped || this.drainingRuns.has(runId) || this.scheduledRuns.has(runId)) return;
 		this.scheduledRuns.add(runId);
 		setImmediate(() => {
 			this.scheduledRuns.delete(runId);
-			this.drainRun(runId);
+			if (this.stopped) return;
+			const p = this.drainRun(runId).finally(() => {
+				this.activeDrainPromises.delete(p);
+			});
+			this.activeDrainPromises.add(p);
 		});
 	}
 	async resumeAfterExecutionResult(continuation, request, outputs) {
@@ -6488,4 +6583,4 @@ var WorkflowRepositoryWebhookTriggerMatcherFactory = class {
 
 //#endregion
 export { CostTrackingTelemetryAttributeNames as $, PersistedWorkflowTokenRegistry as A, node as At, CatalogBackedCostTrackingTelemetryFactory as B, HumanTaskStoreToken as Bt, DefaultDrivingScheduler as C, WhenBuilder as Ct, NodeInstanceFactoryFactory as D, defineNode as Dt, StaticCostCatalog as E, defineBatchNode as Et, InProcessRetryRunnerFactory as F, PersistedRuntimeTypeNameResolver as Ft, AllWorkflowsActiveWorkflowActivationPolicy as G, CodemationTelemetryMetricNames as H, RunnableOutputBehaviorResolver as I, DefinedNodeRegistry as It, NoOpNodeExecutionTelemetry as J, NoOpExecutionTelemetryFactory as K, NodeOutputNormalizer as L, HitlResumeTokenSignerToken as Lt, NodeExecutorFactory as M, InjectableRuntimeDecoratorComposer as Mt, NodeExecutor as N, PersistedRuntimeTypeMetadataStore as Nt, NodeInstanceFactory as O, chatModel as Ot, RunSuspendedError as P, StackTraceCallSitePathResolver as Pt, NoOpCostTrackingTelemetry as Q, ItemExprResolver as R, HitlTimeoutJobSchedulerToken as Rt, HintOnlyOffloadPolicy as S, ChainCursor as St, RunPolicySnapshotFactory as T, isHumanApprovalNode as Tt, GenAiTelemetryAttributeNames as U, DefaultExecutionContextFactory as V, CodemationTelemetryAttributeNames as W, NoOpTelemetryArtifactReference as X, NoOpTelemetrySpanScope as Y, NoOpCostTrackingTelemetryFactory as Z, RunTerminalPersistenceCoordinator as _, WorkflowExecutableNodeClassifier as _t, InMemoryLiveWorkflowRepository as a, isPortsEmission as at, LocalOnlyScheduler as b, WorkflowDefinitionError as bt, EngineFactory as c, CredentialResolverFactory as ct, PollingTriggerRuntime as d, DefaultExecutionBinaryService as dt, CostTrackingTelemetryMetricNames as et, PollingTriggerDedupWindow as f, UnavailableBinaryStorage as ft, WorkflowPolicyErrorServices as g, WorkflowExecutableNodeClassifierFactory as gt, WorkflowStoragePolicyEvaluator as h, DefaultWorkflowGraphFactory as ht, RunIntentService as i, emitPorts as it, MissingRuntimeTriggerToken as j, tool as jt, WorkflowSnapshotCodec as k, getPersistedRuntimeTypeMetadata as kt, Engine as l, getOriginIndexFromItem as lt, InMemoryBinaryStorage as m, ConnectionInvocationEventPublisher as mt, WorkflowRepositoryWebhookTriggerMatcher as n, RetryPolicy as nt, EngineWorkflowRunnerServiceFactory as o, isUnbrandedPortsEmissionShape as ot, InMemoryRunDataFactory as p, NodeEventPublisher as pt, NoOpExecutionTelemetry as q, RunIntentServiceFactory as r, NoRetryPolicy as rt, EngineWorkflowRunnerService as s, DefaultAsyncSleeper as st, WorkflowRepositoryWebhookTriggerMatcherFactory as t, ExpRetryPolicy as tt, NoOpPollingTriggerLogger as u, ChildExecutionScopeFactory as ut, ENGINE_EXECUTION_LIMITS_DEFAULTS as v, ConnectionInvocationIdFactory as vt, ConfigDrivenOffloadPolicy as w, defineHumanApprovalNode as wt, InlineDrivingScheduler as x, NodeIdSlugifier as xt, EngineExecutionLimitsPolicy as y, WorkflowBuilder as yt, InProcessRetryRunner as z, HitlWorkspaceIdToken as zt };
-//# sourceMappingURL=runtime-BPZgnZ9G.js.map
+//# sourceMappingURL=runtime-iHBN1jyD.js.map