@restatedev/restate-sdk 1.12.0 → 1.14.0

package/dist/endpoint/handlers/preview.js

@@ -0,0 +1,98 @@
+import { ensureError } from "../../types/errors.js";
+import { X_RESTATE_SERVER } from "../../user_agent.js";
+import { errorResponse } from "./utils.js";
+
+//#region src/endpoint/handlers/preview.ts
+function handlePreview(endpoint, previewPathComponent) {
+	const service = endpoint.components.get(previewPathComponent.componentName);
+	if (!service) {
+		const msg = `No service found for name: ${previewPathComponent.componentName}`;
+		endpoint.rlog.error(msg);
+		return errorResponse(404, msg);
+	}
+	const resolvedSerde = service.serdeMatching(previewPathComponent.serdeName);
+	if (!resolvedSerde?.preview) {
+		const msg = `No previewable serde found for URL: ${JSON.stringify(previewPathComponent)}`;
+		endpoint.rlog.error(msg);
+		return errorResponse(404, msg);
+	}
+	return new RestatePreviewResponse(resolvedSerde, previewPathComponent.operation);
+}
+var RestatePreviewResponse = class {
+	constructor(serde, operation) {
+		this.serde = serde;
+		this.operation = operation;
+	}
+	async process({ inputReader, outputWriter, writeHead }) {
+		const response = await this.computeResponse(inputReader);
+		writeHead(response.statusCode, response.headers);
+		try {
+			await outputWriter.write(response.body);
+		} finally {
+			await outputWriter.close();
+		}
+	}
+	async computeResponse(inputReader) {
+		try {
+			const body = await readAllInput(inputReader);
+			return this.operation === "decode" ? await this.decode(body) : await this.encode(body);
+		} catch (e) {
+			return previewErrorResponse(ensureError(e).message);
+		}
+	}
+	async decode(requestBody) {
+		const value = this.serde.deserialize(requestBody);
+		const json = await this.serde.preview.toJsonString(value);
+		return {
+			statusCode: 200,
+			headers: {
+				"content-type": "application/json",
+				"x-restate-server": X_RESTATE_SERVER
+			},
+			body: new TextEncoder().encode(json)
+		};
+	}
+	async encode(requestBody) {
+		const json = new TextDecoder().decode(requestBody);
+		const value = await this.serde.preview.fromJsonString(json);
+		return {
+			statusCode: 200,
+			headers: {
+				"content-type": this.serde.contentType ?? "application/octet-stream",
+				"x-restate-server": X_RESTATE_SERVER
+			},
+			body: this.serde.serialize(value)
+		};
+	}
+};
+function previewErrorResponse(message) {
+	return {
+		statusCode: 422,
+		headers: {
+			"content-type": "application/json",
+			"x-restate-server": X_RESTATE_SERVER
+		},
+		body: new TextEncoder().encode(JSON.stringify({ message }))
+	};
+}
+async function readAllInput(inputReader) {
+	const chunks = [];
+	let totalLength = 0;
+	while (true) {
+		const chunk = await inputReader.next();
+		if (chunk.done) break;
+		chunks.push(chunk.value);
+		totalLength += chunk.value.length;
+	}
+	const combined = new Uint8Array(totalLength);
+	let offset = 0;
+	for (const chunk of chunks) {
+		combined.set(chunk, offset);
+		offset += chunk.length;
+	}
+	return combined;
+}
+
+//#endregion
+export { handlePreview };
+//# sourceMappingURL=preview.js.map

package/dist/endpoint/handlers/preview.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"preview.js","names":["serde: Serde<unknown>","operation: \"decode\" | \"encode\"","chunks: Uint8Array[]"],"sources":["../../../src/endpoint/handlers/preview.ts"],"sourcesContent":["/*\n * Copyright (c) 2023-2024 - Restate Software, Inc., Restate GmbH\n *\n * This file is part of the Restate SDK for Node.js/TypeScript,\n * which is released under the MIT license.\n *\n * You can find a copy of the license in file LICENSE in the root\n * directory of this repository or package, or at\n * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE\n */\n\nimport type { Serde } from \"@restatedev/restate-sdk-core\";\nimport type { Endpoint } from \"../endpoint.js\";\nimport type { PreviewPathComponents } from \"../components.js\";\nimport { X_RESTATE_SERVER } from \"../../user_agent.js\";\nimport { ensureError } from \"../../types/errors.js\";\nimport type {\n  InputReader,\n  OutputWriter,\n  ResponseHeaders,\n  RestateResponse,\n} from \"./types.js\";\nimport { errorResponse } from \"./utils.js\";\n\nexport function handlePreview(\n  endpoint: Endpoint,\n  previewPathComponent: PreviewPathComponents\n): RestateResponse {\n  const service = endpoint.components.get(previewPathComponent.componentName);\n  if (!service) {\n    const msg = `No service found for name: ${previewPathComponent.componentName}`;\n    endpoint.rlog.error(msg);\n    return errorResponse(404, msg);\n  }\n\n  const resolvedSerde = service.serdeMatching(previewPathComponent.serdeName);\n  if (!resolvedSerde?.preview) {\n    const msg = `No previewable serde found for URL: ${JSON.stringify(previewPathComponent)}`;\n    endpoint.rlog.error(msg);\n    return errorResponse(404, msg);\n  }\n\n  return new RestatePreviewResponse(\n    resolvedSerde,\n    previewPathComponent.operation\n  );\n}\n\ntype PreviewResponse = {\n  body: Uint8Array;\n  headers: ResponseHeaders;\n  statusCode: number;\n};\n\nclass RestatePreviewResponse implements RestateResponse {\n  constructor(\n    private readonly serde: Serde<unknown>,\n    private readonly operation: \"decode\" | \"encode\"\n  ) {}\n\n  async process({\n    inputReader,\n    outputWriter,\n    writeHead,\n  }: {\n    inputReader: InputReader;\n    outputWriter: OutputWriter;\n    writeHead: (statusCode: number, headers: ResponseHeaders) => void;\n    abortSignal: AbortSignal;\n  }): Promise<void> {\n    const response = await this.computeResponse(inputReader);\n\n    writeHead(response.statusCode, response.headers);\n    try {\n      await outputWriter.write(response.body);\n    } finally {\n      await outputWriter.close();\n    }\n  }\n\n  private async computeResponse(\n    inputReader: InputReader\n  ): Promise<PreviewResponse> {\n    try {\n      const body = await readAllInput(inputReader);\n      return this.operation === \"decode\"\n        ? await this.decode(body)\n        : await this.encode(body);\n    } catch (e) {\n      return previewErrorResponse(ensureError(e).message);\n    }\n  }\n\n  private async decode(requestBody: Uint8Array): Promise<PreviewResponse> {\n    const value = this.serde.deserialize(requestBody);\n    const json = await this.serde.preview!.toJsonString(value);\n    return {\n      statusCode: 200,\n      headers: {\n        \"content-type\": \"application/json\",\n        \"x-restate-server\": X_RESTATE_SERVER,\n      },\n      body: new TextEncoder().encode(json),\n    };\n  }\n\n  private async encode(requestBody: Uint8Array): Promise<PreviewResponse> {\n    const json = new TextDecoder().decode(requestBody);\n    const value = await this.serde.preview!.fromJsonString(json);\n    return {\n      statusCode: 200,\n      headers: {\n        \"content-type\": this.serde.contentType ?? \"application/octet-stream\",\n        \"x-restate-server\": X_RESTATE_SERVER,\n      },\n      body: this.serde.serialize(value),\n    };\n  }\n}\n\nfunction previewErrorResponse(message: string): PreviewResponse {\n  return {\n    statusCode: 422,\n    headers: {\n      \"content-type\": \"application/json\",\n      \"x-restate-server\": X_RESTATE_SERVER,\n    },\n    body: new TextEncoder().encode(JSON.stringify({ message })),\n  };\n}\n\nasync function readAllInput(inputReader: InputReader): Promise<Uint8Array> {\n  const chunks: Uint8Array[] = [];\n  let totalLength = 0;\n\n  while (true) {\n    const chunk = await inputReader.next();\n    if (chunk.done) {\n      break;\n    }\n    chunks.push(chunk.value);\n    totalLength += chunk.value.length;\n  }\n\n  const combined = new Uint8Array(totalLength);\n  let offset = 0;\n  for (const chunk of chunks) {\n    combined.set(chunk, offset);\n    offset += chunk.length;\n  }\n  return combined;\n}\n"],"mappings":";;;;;AAwBA,SAAgB,cACd,UACA,sBACiB;CACjB,MAAM,UAAU,SAAS,WAAW,IAAI,qBAAqB,cAAc;AAC3E,KAAI,CAAC,SAAS;EACZ,MAAM,MAAM,8BAA8B,qBAAqB;AAC/D,WAAS,KAAK,MAAM,IAAI;AACxB,SAAO,cAAc,KAAK,IAAI;;CAGhC,MAAM,gBAAgB,QAAQ,cAAc,qBAAqB,UAAU;AAC3E,KAAI,CAAC,eAAe,SAAS;EAC3B,MAAM,MAAM,uCAAuC,KAAK,UAAU,qBAAqB;AACvF,WAAS,KAAK,MAAM,IAAI;AACxB,SAAO,cAAc,KAAK,IAAI;;AAGhC,QAAO,IAAI,uBACT,eACA,qBAAqB,UACtB;;AASH,IAAM,yBAAN,MAAwD;CACtD,YACE,AAAiBA,OACjB,AAAiBC,WACjB;EAFiB;EACA;;CAGnB,MAAM,QAAQ,EACZ,aACA,cACA,aAMgB;EAChB,MAAM,WAAW,MAAM,KAAK,gBAAgB,YAAY;AAExD,YAAU,SAAS,YAAY,SAAS,QAAQ;AAChD,MAAI;AACF,SAAM,aAAa,MAAM,SAAS,KAAK;YAC/B;AACR,SAAM,aAAa,OAAO;;;CAI9B,MAAc,gBACZ,aAC0B;AAC1B,MAAI;GACF,MAAM,OAAO,MAAM,aAAa,YAAY;AAC5C,UAAO,KAAK,cAAc,WACtB,MAAM,KAAK,OAAO,KAAK,GACvB,MAAM,KAAK,OAAO,KAAK;WACpB,GAAG;AACV,UAAO,qBAAqB,YAAY,EAAE,CAAC,QAAQ;;;CAIvD,MAAc,OAAO,aAAmD;EACtE,MAAM,QAAQ,KAAK,MAAM,YAAY,YAAY;EACjD,MAAM,OAAO,MAAM,KAAK,MAAM,QAAS,aAAa,MAAM;AAC1D,SAAO;GACL,YAAY;GACZ,SAAS;IACP,gBAAgB;IAChB,oBAAoB;IACrB;GACD,MAAM,IAAI,aAAa,CAAC,OAAO,KAAK;GACrC;;CAGH,MAAc,OAAO,aAAmD;EACtE,MAAM,OAAO,IAAI,aAAa,CAAC,OAAO,YAAY;EAClD,MAAM,QAAQ,MAAM,KAAK,MAAM,QAAS,eAAe,KAAK;AAC5D,SAAO;GACL,YAAY;GACZ,SAAS;IACP,gBAAgB,KAAK,MAAM,eAAe;IAC1C,oBAAoB;IACrB;GACD,MAAM,KAAK,MAAM,UAAU,MAAM;GAClC;;;AAIL,SAAS,qBAAqB,SAAkC;AAC9D,QAAO;EACL,YAAY;EACZ,SAAS;GACP,gBAAgB;GAChB,oBAAoB;GACrB;EACD,MAAM,IAAI,aAAa,CAAC,OAAO,KAAK,UAAU,EAAE,SAAS,CAAC,CAAC;EAC5D;;AAGH,eAAe,aAAa,aAA+C;CACzE,MAAMC,SAAuB,EAAE;CAC/B,IAAI,cAAc;AAElB,QAAO,MAAM;EACX,MAAM,QAAQ,MAAM,YAAY,MAAM;AACtC,MAAI,MAAM,KACR;AAEF,SAAO,KAAK,MAAM,MAAM;AACxB,iBAAe,MAAM,MAAM;;CAG7B,MAAM,WAAW,IAAI,WAAW,YAAY;CAC5C,IAAI,SAAS;AACb,MAAK,MAAM,SAAS,QAAQ;AAC1B,WAAS,IAAI,OAAO,OAAO;AAC3B,YAAU,MAAM;;AAElB,QAAO"}

package/dist/endpoint/handlers/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/endpoint/handlers/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE/C,MAAM,WAAW,OAAO;IACtB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC;CAC/C;AAED,MAAM,WAAW,eAAe;IAC9B,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,iBAAiB;IAChC,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;IAC1B,QAAQ,CAAC,SAAS,EAAE,OAAO,EAAE,CAAC;CAC/B;AAED,MAAM,MAAM,qBAAqB,GAC7B;IAAE,IAAI,EAAE,KAAK,GAAG,SAAS,CAAC;IAAC,KAAK,EAAE,UAAU,CAAA;CAAE,GAC9C;IAAE,IAAI,EAAE,IAAI,CAAC;IAAC,KAAK,EAAE,SAAS,CAAA;CAAE,CAAC;AAErC,MAAM,MAAM,WAAW,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;AAEpD,MAAM,WAAW,YAAY;IAE3B,KAAK,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACxC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACxB;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,OAAO,EAAE,eAAe,CAAC;IAClC,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAK5B,OAAO,CAAC,KAAK,EAAE;QACb,WAAW,EAAE,WAAW,CAAC;QACzB,YAAY,EAAE,YAAY,CAAC;QAC3B,WAAW,EAAE,WAAW,CAAC;KAC1B,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACnB;AAED,MAAM,WAAW,cAAc;IAE7B,QAAQ,EAAE,QAAQ,CAAC;IAEnB,MAAM,CAAC,OAAO,EAAE,cAAc,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,eAAe,CAAC;CAC/E"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/endpoint/handlers/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE/C,MAAM,WAAW,OAAO;IACtB,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC;CAC/C;AAED,MAAM,WAAW,eAAe;IAC9B,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,iBAAiB;IAChC,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;IAC1B,QAAQ,CAAC,SAAS,EAAE,OAAO,EAAE,CAAC;CAC/B;AAED,MAAM,MAAM,qBAAqB,GAC7B;IAAE,IAAI,EAAE,KAAK,GAAG,SAAS,CAAC;IAAC,KAAK,EAAE,UAAU,CAAA;CAAE,GAC9C;IAAE,IAAI,EAAE,IAAI,CAAC;IAAC,KAAK,EAAE,SAAS,CAAA;CAAE,CAAC;AAErC,MAAM,MAAM,WAAW,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;AAEpD,MAAM,WAAW,YAAY;IAE3B,KAAK,CAAC,KAAK,EAAE,UAAU,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACxC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACxB;AAED,MAAM,WAAW,eAAe;IAC9B;;;;;;;OAOG;IACH,OAAO,CAAC,KAAK,EAAE;QACb,WAAW,EAAE,WAAW,CAAC;QACzB,YAAY,EAAE,YAAY,CAAC;QAC3B,SAAS,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,KAAK,IAAI,CAAC;QAClE,WAAW,EAAE,WAAW,CAAC;KAC1B,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACnB;AAED,MAAM,WAAW,cAAc;IAE7B,QAAQ,EAAE,QAAQ,CAAC;IAEnB,MAAM,CAAC,OAAO,EAAE,cAAc,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,eAAe,CAAC;CAC/E"}

package/dist/endpoint/handlers/utils.cjs

@@ -26,24 +26,39 @@ function errorResponse(code, message) {
 	}, new TextEncoder().encode(JSON.stringify({ message })));
 }
 function simpleResponse(statusCode, headers, body) {
-	return {
-		headers,
-		statusCode,
-		async process({ inputReader, outputWriter }) {
-			if (inputReader !== void 0) while (true) {
-				const { done } = await inputReader.next();
-				if (done) break;
-			}
-			await outputWriter.write(body);
-			await outputWriter.close();
+	return { async process({ inputReader, outputWriter, writeHead }) {
+		writeHead(statusCode, headers);
+		while (true) {
+			const { done } = await inputReader.next();
+			if (done) break;
 		}
-	};
+		await outputWriter.write(body);
+		await outputWriter.close();
+	} };
 }
 function emptyInputReader() {
 	return (async function* () {})()[Symbol.asyncIterator]();
 }
+/**
+* Bundles a `writeHead` callback with a Promise that resolves once the head
+* is committed. Used by adapters (fetch, lambda) that need to observe the
+* head commit from outside the `process()` call.
+*/
+function captureHead() {
+	const { promise, resolve } = Promise.withResolvers();
+	return {
+		writeHead: (statusCode, headers) => {
+			resolve({
+				statusCode,
+				headers: { ...headers }
+			});
+		},
+		head: promise
+	};
+}
 
 //#endregion
+exports.captureHead = captureHead;
 exports.emptyInputReader = emptyInputReader;
 exports.errorResponse = errorResponse;
 exports.invocationIdFromHeaders = invocationIdFromHeaders;

package/dist/endpoint/handlers/utils.d.ts

@@ -8,4 +8,16 @@ export declare function invocationIdFromHeaders(headers: Headers): string;
 export declare function errorResponse(code: number, message: string): RestateResponse;
 export declare function simpleResponse(statusCode: number, headers: ResponseHeaders, body: Uint8Array): RestateResponse;
 export declare function emptyInputReader(): InputReader;
+/**
+ * Bundles a `writeHead` callback with a Promise that resolves once the head
+ * is committed. Used by adapters (fetch, lambda) that need to observe the
+ * head commit from outside the `process()` call.
+ */
+export declare function captureHead(): {
+    writeHead: (statusCode: number, headers: ResponseHeaders) => void;
+    head: Promise<{
+        statusCode: number;
+        headers: ResponseHeaders;
+    }>;
+};
 //# sourceMappingURL=utils.d.ts.map

package/dist/endpoint/handlers/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/endpoint/handlers/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,eAAe,EAChB,MAAM,mCAAmC,CAAC;AAC3C,OAAO,KAAK,EACV,OAAO,EACP,WAAW,EACX,eAAe,EACf,eAAe,EAChB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAgB,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAI/D,wBAAgB,yBAAyB,CACvC,eAAe,EAAE,eAAe,EAChC,GAAG,EAAE,MAAM,EACX,OAAO,EAAE,OAAO,EAChB,iBAAiB,CAAC,EAAE;IAAE,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAA;CAAE,GAC7C,MAAM,GAAG,SAAS,CAuBpB;AAED,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,OAAO,UASvD;AAED,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,eAAe,CAS5E;AAED,wBAAgB,cAAc,CAC5B,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,eAAe,EACxB,IAAI,EAAE,UAAU,GACf,eAAe,CAkBjB;AAED,wBAAgB,gBAAgB,IAAI,WAAW,CAE9C"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/endpoint/handlers/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAGL,eAAe,EAChB,MAAM,mCAAmC,CAAC;AAC3C,OAAO,KAAK,EACV,OAAO,EACP,WAAW,EACX,eAAe,EACf,eAAe,EAChB,MAAM,YAAY,CAAC;AACpB,OAAO,EAAgB,MAAM,EAAE,MAAM,yBAAyB,CAAC;AAI/D,wBAAgB,yBAAyB,CACvC,eAAe,EAAE,eAAe,EAChC,GAAG,EAAE,MAAM,EACX,OAAO,EAAE,OAAO,EAChB,iBAAiB,CAAC,EAAE;IAAE,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAA;CAAE,GAC7C,MAAM,GAAG,SAAS,CAuBpB;AAED,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,OAAO,UASvD;AAED,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,eAAe,CAS5E;AAED,wBAAgB,cAAc,CAC5B,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,eAAe,EACxB,IAAI,EAAE,UAAU,GACf,eAAe,CAgBjB;AAED,wBAAgB,gBAAgB,IAAI,WAAW,CAE9C;AAED;;;;GAIG;AACH,wBAAgB,WAAW,IAAI;IAC7B,SAAS,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,KAAK,IAAI,CAAC;IAClE,IAAI,EAAE,OAAO,CAAC;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,eAAe,CAAA;KAAE,CAAC,CAAC;CACjE,CAWA"}

package/dist/endpoint/handlers/utils.js

@@ -26,23 +26,37 @@ function errorResponse(code, message) {
 	}, new TextEncoder().encode(JSON.stringify({ message })));
 }
 function simpleResponse(statusCode, headers, body) {
-	return {
-		headers,
-		statusCode,
-		async process({ inputReader, outputWriter }) {
-			if (inputReader !== void 0) while (true) {
-				const { done } = await inputReader.next();
-				if (done) break;
-			}
-			await outputWriter.write(body);
-			await outputWriter.close();
+	return { async process({ inputReader, outputWriter, writeHead }) {
+		writeHead(statusCode, headers);
+		while (true) {
+			const { done } = await inputReader.next();
+			if (done) break;
 		}
-	};
+		await outputWriter.write(body);
+		await outputWriter.close();
+	} };
 }
 function emptyInputReader() {
 	return (async function* () {})()[Symbol.asyncIterator]();
 }
+/**
+* Bundles a `writeHead` callback with a Promise that resolves once the head
+* is committed. Used by adapters (fetch, lambda) that need to observe the
+* head commit from outside the `process()` call.
+*/
+function captureHead() {
+	const { promise, resolve } = Promise.withResolvers();
+	return {
+		writeHead: (statusCode, headers) => {
+			resolve({
+				statusCode,
+				headers: { ...headers }
+			});
+		},
+		head: promise
+	};
+}
 
 //#endregion
-export { emptyInputReader, errorResponse, invocationIdFromHeaders, simpleResponse, tryCreateContextualLogger };
+export { captureHead, emptyInputReader, errorResponse, invocationIdFromHeaders, simpleResponse, tryCreateContextualLogger };
 //# sourceMappingURL=utils.js.map

package/dist/endpoint/handlers/utils.js.map

@@ -1 +1 @@
-{"version":3,"file":"utils.js","names":[],"sources":["../../../src/endpoint/handlers/utils.ts"],"sourcesContent":["import {\n  LoggerContext,\n  LogSource,\n  LoggerTransport,\n} from \"../../logging/logger_transport.js\";\nimport type {\n  Headers,\n  InputReader,\n  ResponseHeaders,\n  RestateResponse,\n} from \"./types.js\";\nimport { createLogger, Logger } from \"../../logging/logger.js\";\nimport { parseUrlComponents } from \"../components.js\";\nimport { X_RESTATE_SERVER } from \"../../user_agent.js\";\n\nexport function tryCreateContextualLogger(\n  loggerTransport: LoggerTransport,\n  url: string,\n  headers: Headers,\n  additionalContext?: { [name: string]: string }\n): Logger | undefined {\n  try {\n    const path = new URL(url, \"https://example.com\").pathname;\n    const parsed = parseUrlComponents(path);\n    if (parsed.type !== \"invoke\") {\n      return undefined;\n    }\n    const invocationId = invocationIdFromHeaders(headers);\n    return createLogger(\n      loggerTransport,\n      LogSource.SYSTEM,\n      new LoggerContext(\n        invocationId,\n        parsed.componentName,\n        parsed.handlerName,\n        undefined,\n        undefined,\n        additionalContext\n      )\n    );\n  } catch {\n    return undefined;\n  }\n}\n\nexport function invocationIdFromHeaders(headers: Headers) {\n  const invocationIdHeader = headers[\"x-restate-invocation-id\"];\n  const invocationId =\n    typeof invocationIdHeader === \"string\"\n      ? invocationIdHeader\n      : Array.isArray(invocationIdHeader)\n        ? (invocationIdHeader[0] ?? \"unknown id\")\n        : \"unknown id\";\n  return invocationId;\n}\n\nexport function errorResponse(code: number, message: string): RestateResponse {\n  return simpleResponse(\n    code,\n    {\n      \"content-type\": \"application/json\",\n      \"x-restate-server\": X_RESTATE_SERVER,\n    },\n    new TextEncoder().encode(JSON.stringify({ message }))\n  );\n}\n\nexport function simpleResponse(\n  statusCode: number,\n  headers: ResponseHeaders,\n  body: Uint8Array\n): RestateResponse {\n  return {\n    headers,\n    statusCode,\n    async process({ inputReader, outputWriter }): Promise<void> {\n      if (inputReader !== undefined) {\n        // Drain the input stream\n        while (true) {\n          const { done } = await inputReader.next();\n          if (done) break;\n        }\n      }\n\n      await outputWriter.write(body);\n      // This closes both the writer and the stream!!!\n      await outputWriter.close();\n    },\n  };\n}\n\nexport function emptyInputReader(): InputReader {\n  return (async function* () {})()[Symbol.asyncIterator]();\n}\n"],"mappings":";;;;;;AAeA,SAAgB,0BACd,iBACA,KACA,SACA,mBACoB;AACpB,KAAI;EACF,MAAM,OAAO,IAAI,IAAI,KAAK,sBAAsB,CAAC;EACjD,MAAM,SAAS,mBAAmB,KAAK;AACvC,MAAI,OAAO,SAAS,SAClB;EAEF,MAAM,eAAe,wBAAwB,QAAQ;AACrD,SAAO,aACL,iBACA,UAAU,QACV,IAAI,cACF,cACA,OAAO,eACP,OAAO,aACP,QACA,QACA,kBACD,CACF;SACK;AACN;;;AAIJ,SAAgB,wBAAwB,SAAkB;CACxD,MAAM,qBAAqB,QAAQ;AAOnC,QALE,OAAO,uBAAuB,WAC1B,qBACA,MAAM,QAAQ,mBAAmB,GAC9B,mBAAmB,MAAM,eAC1B;;AAIV,SAAgB,cAAc,MAAc,SAAkC;AAC5E,QAAO,eACL,MACA;EACE,gBAAgB;EAChB,oBAAoB;EACrB,EACD,IAAI,aAAa,CAAC,OAAO,KAAK,UAAU,EAAE,SAAS,CAAC,CAAC,CACtD;;AAGH,SAAgB,eACd,YACA,SACA,MACiB;AACjB,QAAO;EACL;EACA;EACA,MAAM,QAAQ,EAAE,aAAa,gBAA+B;AAC1D,OAAI,gBAAgB,OAElB,QAAO,MAAM;IACX,MAAM,EAAE,SAAS,MAAM,YAAY,MAAM;AACzC,QAAI,KAAM;;AAId,SAAM,aAAa,MAAM,KAAK;AAE9B,SAAM,aAAa,OAAO;;EAE7B;;AAGH,SAAgB,mBAAgC;AAC9C,SAAQ,mBAAmB,KAAK,CAAC,OAAO,gBAAgB"}
+{"version":3,"file":"utils.js","names":[],"sources":["../../../src/endpoint/handlers/utils.ts"],"sourcesContent":["import {\n  LoggerContext,\n  LogSource,\n  LoggerTransport,\n} from \"../../logging/logger_transport.js\";\nimport type {\n  Headers,\n  InputReader,\n  ResponseHeaders,\n  RestateResponse,\n} from \"./types.js\";\nimport { createLogger, Logger } from \"../../logging/logger.js\";\nimport { parseUrlComponents } from \"../components.js\";\nimport { X_RESTATE_SERVER } from \"../../user_agent.js\";\n\nexport function tryCreateContextualLogger(\n  loggerTransport: LoggerTransport,\n  url: string,\n  headers: Headers,\n  additionalContext?: { [name: string]: string }\n): Logger | undefined {\n  try {\n    const path = new URL(url, \"https://example.com\").pathname;\n    const parsed = parseUrlComponents(path);\n    if (parsed.type !== \"invoke\") {\n      return undefined;\n    }\n    const invocationId = invocationIdFromHeaders(headers);\n    return createLogger(\n      loggerTransport,\n      LogSource.SYSTEM,\n      new LoggerContext(\n        invocationId,\n        parsed.componentName,\n        parsed.handlerName,\n        undefined,\n        undefined,\n        additionalContext\n      )\n    );\n  } catch {\n    return undefined;\n  }\n}\n\nexport function invocationIdFromHeaders(headers: Headers) {\n  const invocationIdHeader = headers[\"x-restate-invocation-id\"];\n  const invocationId =\n    typeof invocationIdHeader === \"string\"\n      ? invocationIdHeader\n      : Array.isArray(invocationIdHeader)\n        ? (invocationIdHeader[0] ?? \"unknown id\")\n        : \"unknown id\";\n  return invocationId;\n}\n\nexport function errorResponse(code: number, message: string): RestateResponse {\n  return simpleResponse(\n    code,\n    {\n      \"content-type\": \"application/json\",\n      \"x-restate-server\": X_RESTATE_SERVER,\n    },\n    new TextEncoder().encode(JSON.stringify({ message }))\n  );\n}\n\nexport function simpleResponse(\n  statusCode: number,\n  headers: ResponseHeaders,\n  body: Uint8Array\n): RestateResponse {\n  return {\n    async process({ inputReader, outputWriter, writeHead }): Promise<void> {\n      writeHead(statusCode, headers);\n\n      // Drain the input stream\n      while (true) {\n        const { done } = await inputReader.next();\n        if (done) break;\n      }\n\n      await outputWriter.write(body);\n      // This closes both the writer and the stream!!!\n      await outputWriter.close();\n    },\n  };\n}\n\nexport function emptyInputReader(): InputReader {\n  return (async function* () {})()[Symbol.asyncIterator]();\n}\n\n/**\n * Bundles a `writeHead` callback with a Promise that resolves once the head\n * is committed. Used by adapters (fetch, lambda) that need to observe the\n * head commit from outside the `process()` call.\n */\nexport function captureHead(): {\n  writeHead: (statusCode: number, headers: ResponseHeaders) => void;\n  head: Promise<{ statusCode: number; headers: ResponseHeaders }>;\n} {\n  const { promise, resolve } = Promise.withResolvers<{\n    statusCode: number;\n    headers: ResponseHeaders;\n  }>();\n  return {\n    writeHead: (statusCode, headers) => {\n      resolve({ statusCode, headers: { ...headers } });\n    },\n    head: promise,\n  };\n}\n"],"mappings":";;;;;;AAeA,SAAgB,0BACd,iBACA,KACA,SACA,mBACoB;AACpB,KAAI;EACF,MAAM,OAAO,IAAI,IAAI,KAAK,sBAAsB,CAAC;EACjD,MAAM,SAAS,mBAAmB,KAAK;AACvC,MAAI,OAAO,SAAS,SAClB;EAEF,MAAM,eAAe,wBAAwB,QAAQ;AACrD,SAAO,aACL,iBACA,UAAU,QACV,IAAI,cACF,cACA,OAAO,eACP,OAAO,aACP,QACA,QACA,kBACD,CACF;SACK;AACN;;;AAIJ,SAAgB,wBAAwB,SAAkB;CACxD,MAAM,qBAAqB,QAAQ;AAOnC,QALE,OAAO,uBAAuB,WAC1B,qBACA,MAAM,QAAQ,mBAAmB,GAC9B,mBAAmB,MAAM,eAC1B;;AAIV,SAAgB,cAAc,MAAc,SAAkC;AAC5E,QAAO,eACL,MACA;EACE,gBAAgB;EAChB,oBAAoB;EACrB,EACD,IAAI,aAAa,CAAC,OAAO,KAAK,UAAU,EAAE,SAAS,CAAC,CAAC,CACtD;;AAGH,SAAgB,eACd,YACA,SACA,MACiB;AACjB,QAAO,EACL,MAAM,QAAQ,EAAE,aAAa,cAAc,aAA4B;AACrE,YAAU,YAAY,QAAQ;AAG9B,SAAO,MAAM;GACX,MAAM,EAAE,SAAS,MAAM,YAAY,MAAM;AACzC,OAAI,KAAM;;AAGZ,QAAM,aAAa,MAAM,KAAK;AAE9B,QAAM,aAAa,OAAO;IAE7B;;AAGH,SAAgB,mBAAgC;AAC9C,SAAQ,mBAAmB,KAAK,CAAC,OAAO,gBAAgB;;;;;;;AAQ1D,SAAgB,cAGd;CACA,MAAM,EAAE,SAAS,YAAY,QAAQ,eAGjC;AACJ,QAAO;EACL,YAAY,YAAY,YAAY;AAClC,WAAQ;IAAE;IAAY,SAAS,EAAE,GAAG,SAAS;IAAE,CAAC;;EAElD,MAAM;EACP"}

package/dist/endpoint/node_endpoint.cjs

@@ -74,19 +74,22 @@ function nodeHandlerImpl(endpoint, protocolMode) {
 	const handler = require_generic.createRestateHandler(endpoint, protocolMode, {});
 	return (httpRequest, httpResponse) => {
 		const url = httpRequest.url;
+		const inputReader = inputReaderAdapter(httpRequest);
+		const outputWriter = outputWriterAdapter(httpResponse);
+		const res = httpResponse;
 		const abortController = new AbortController();
 		httpRequest.on("close", () => {
 			abortController.abort();
 		});
-		const restateResponse = handler.handle({
+		const writeHead = res.writeHead.bind(res);
+		handler.handle({
 			url,
 			headers: httpRequest.headers,
 			extraArgs: []
-		});
-		httpResponse.writeHead(restateResponse.statusCode, restateResponse.headers);
-		restateResponse.process({
-			inputReader: inputReaderAdapter(httpRequest),
-			outputWriter: outputWriterAdapter(httpResponse),
+		}).process({
+			inputReader,
+			outputWriter,
+			writeHead,
 			abortSignal: abortController.signal
 		}).catch((e) => {
 			const error = require_errors.ensureError(e);

package/dist/endpoint/node_endpoint.js

@@ -72,19 +72,22 @@ function nodeHandlerImpl(endpoint, protocolMode) {
 	const handler = createRestateHandler(endpoint, protocolMode, {});
 	return (httpRequest, httpResponse) => {
 		const url = httpRequest.url;
+		const inputReader = inputReaderAdapter(httpRequest);
+		const outputWriter = outputWriterAdapter(httpResponse);
+		const res = httpResponse;
 		const abortController = new AbortController();
 		httpRequest.on("close", () => {
 			abortController.abort();
 		});
-		const restateResponse = handler.handle({
+		const writeHead = res.writeHead.bind(res);
+		handler.handle({
 			url,
 			headers: httpRequest.headers,
 			extraArgs: []
-		});
-		httpResponse.writeHead(restateResponse.statusCode, restateResponse.headers);
-		restateResponse.process({
-			inputReader: inputReaderAdapter(httpRequest),
-			outputWriter: outputWriterAdapter(httpResponse),
+		}).process({
+			inputReader,
+			outputWriter,
+			writeHead,
 			abortSignal: abortController.signal
 		}).catch((e) => {
 			const error = ensureError(e);

package/dist/endpoint/node_endpoint.js.map

@@ -1 +1 @@
-{"version":3,"file":"node_endpoint.js","names":[],"sources":["../../src/endpoint/node_endpoint.ts"],"sourcesContent":["/*\n * Copyright (c) 2023-2024 - Restate Software, Inc., Restate GmbH\n *\n * This file is part of the Restate SDK for Node.js/TypeScript,\n * which is released under the MIT license.\n *\n * You can find a copy of the license in file LICENSE in the root\n * directory of this repository or package, or at\n * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE\n */\n\nimport type { RestateEndpoint } from \"../index.js\";\nimport type {\n  JournalValueCodec,\n  ServiceDefinition,\n  VirtualObjectDefinition,\n  WorkflowDefinition,\n} from \"@restatedev/restate-sdk-core\";\nimport { IncomingMessage, ServerResponse } from \"node:http\";\nimport { Http2ServerRequest, Http2ServerResponse } from \"node:http2\";\nimport * as http2 from \"node:http2\";\nimport type { Endpoint } from \"./endpoint.js\";\nimport { EndpointBuilder } from \"./endpoint.js\";\nimport { createRestateHandler } from \"./handlers/generic.js\";\nimport { ensureError } from \"../types/errors.js\";\nimport type { LoggerTransport } from \"../logging/logger_transport.js\";\nimport type { DefaultServiceOptions } from \"../endpoint.js\";\nimport { tryCreateContextualLogger } from \"./handlers/utils.js\";\nimport type { InputReader, OutputWriter } from \"./handlers/types.js\";\nimport type { ProtocolMode } from \"./discovery.js\";\n\nexport class NodeEndpoint implements RestateEndpoint {\n  private builder: EndpointBuilder = new EndpointBuilder();\n\n  public bind<P extends string, M>(\n    definition:\n      | ServiceDefinition<P, M>\n      | VirtualObjectDefinition<P, M>\n      | WorkflowDefinition<P, M>\n  ): RestateEndpoint {\n    this.builder.bind(definition);\n    return this;\n  }\n\n  public withIdentityV1(...keys: string[]): RestateEndpoint {\n    this.builder.addIdentityKeys(...keys);\n    return this;\n  }\n\n  public defaultServiceOptions(\n    options: DefaultServiceOptions\n  ): RestateEndpoint {\n    this.builder.setDefaultServiceOptions(options);\n    return this;\n  }\n\n  public setLogger(logger: LoggerTransport): RestateEndpoint {\n    this.builder.setLogger(logger);\n    return this;\n  }\n\n  public journalValueCodecProvider(\n    codecProvider: () => Promise<JournalValueCodec>\n  ): RestateEndpoint {\n    this.builder.setJournalValueCodecProvider(codecProvider);\n    return this;\n  }\n\n  http2Handler(options?: {\n    bidirectional?: boolean;\n  }): (request: Http2ServerRequest, response: Http2ServerResponse) => void {\n    return nodeHttp2Handler(\n      this.builder.build(),\n      options?.bidirectional === false ? \"REQUEST_RESPONSE\" : \"BIDI_STREAM\"\n    );\n  }\n\n  http1Handler(options?: {\n    bidirectional?: boolean;\n  }): (request: IncomingMessage, response: ServerResponse) => void {\n    return nodeHttp1Handler(\n      this.builder.build(),\n      options?.bidirectional ? \"BIDI_STREAM\" : \"REQUEST_RESPONSE\"\n    );\n  }\n\n  handler(options?: { bidirectional?: boolean }): {\n    (request: IncomingMessage, response: ServerResponse): void;\n    (request: Http2ServerRequest, response: Http2ServerResponse): void;\n  } {\n    const endpoint = this.builder.build();\n    const h2Handler = nodeHttp2Handler(\n      endpoint,\n      options?.bidirectional === false ? \"REQUEST_RESPONSE\" : \"BIDI_STREAM\"\n    );\n    const h1Handler = nodeHttp1Handler(\n      endpoint,\n      options?.bidirectional ? \"BIDI_STREAM\" : \"REQUEST_RESPONSE\"\n    );\n\n    return ((\n      request: IncomingMessage | Http2ServerRequest,\n      response: ServerResponse | Http2ServerResponse\n    ) => {\n      if (request.httpVersionMajor >= 2) {\n        h2Handler(\n          request as Http2ServerRequest,\n          response as Http2ServerResponse\n        );\n      } else {\n        h1Handler(request as IncomingMessage, response as ServerResponse);\n      }\n    }) as {\n      (request: IncomingMessage, response: ServerResponse): void;\n      (request: Http2ServerRequest, response: Http2ServerResponse): void;\n    };\n  }\n\n  listen(port?: number): Promise<number> {\n    const endpoint = this.builder.build();\n\n    const actualPort = port ?? parseInt(process.env.PORT ?? \"9080\");\n    endpoint.rlog.info(`Restate SDK started listening on ${actualPort}...`);\n\n    const server = http2.createServer(\n      nodeHttp2Handler(endpoint, \"BIDI_STREAM\")\n    );\n\n    return new Promise((resolve, reject) => {\n      let failed = false;\n      server.once(\"error\", (e: Error) => {\n        failed = true;\n        reject(e);\n      });\n      server.listen(actualPort, () => {\n        if (failed) {\n          return;\n        }\n        const address = server.address();\n        if (address === null || typeof address === \"string\") {\n          reject(\n            new TypeError(\n              \"endpoint.listen() currently supports only binding to a PORT\"\n            )\n          );\n        } else {\n          resolve(address.port);\n        }\n      });\n    });\n  }\n}\n\nfunction nodeHttp1Handler(\n  endpoint: Endpoint,\n  protocolMode: ProtocolMode\n): (request: IncomingMessage, response: ServerResponse) => void {\n  return nodeHandlerImpl(endpoint, protocolMode);\n}\n\nfunction nodeHttp2Handler(\n  endpoint: Endpoint,\n  protocolMode: ProtocolMode\n): (request: Http2ServerRequest, response: Http2ServerResponse) => void {\n  return nodeHandlerImpl(endpoint, protocolMode);\n}\n\nfunction nodeHandlerImpl(\n  endpoint: Endpoint,\n  protocolMode: ProtocolMode\n): (\n  request: Http2ServerRequest | IncomingMessage,\n  response: Http2ServerResponse | ServerResponse\n) => void {\n  const handler = createRestateHandler(endpoint, protocolMode, {});\n\n  return (httpRequest, httpResponse) => {\n    const url = httpRequest.url!;\n\n    // Abort controller used to cleanup resources at the end of this stream lifecycle\n    const abortController = new AbortController();\n    httpRequest.on(\"close\", () => {\n      abortController.abort();\n    });\n\n    const restateResponse = handler.handle({\n      url,\n      headers: httpRequest.headers,\n      extraArgs: [],\n    });\n    const res = httpResponse as NodeWritableResponse;\n    res.writeHead(restateResponse.statusCode, restateResponse.headers);\n\n    restateResponse\n      .process({\n        inputReader: inputReaderAdapter(httpRequest),\n        outputWriter: outputWriterAdapter(httpResponse),\n        abortSignal: abortController.signal,\n      })\n      .catch((e) => {\n        // handle should never throw\n        const error = ensureError(e);\n        const logger =\n          tryCreateContextualLogger(\n            endpoint.loggerTransport,\n            url,\n            httpRequest.headers\n          ) ?? endpoint.rlog;\n        logger.error(\"Unexpected error: \" + (error.stack ?? error.message));\n      });\n  };\n}\n\n// Both ServerResponse and Http2ServerResponse satisfy this interface.\n// We use it to avoid TS union overload incompatibilities.\ninterface NodeWritableResponse {\n  writeHead(statusCode: number, headers: Record<string, string>): void;\n  write(chunk: Uint8Array, callback: (err?: Error | null) => void): boolean;\n  end(callback: () => void): void;\n}\n\nfunction inputReaderAdapter(\n  request: Http2ServerRequest | IncomingMessage\n): InputReader {\n  return (request as AsyncIterable<Uint8Array>)[Symbol.asyncIterator]();\n}\n\nfunction outputWriterAdapter(\n  response: Http2ServerResponse | ServerResponse\n): OutputWriter {\n  const res = response as NodeWritableResponse;\n  return {\n    write: function (value: Uint8Array): Promise<void> {\n      return new Promise((resolve, reject) => {\n        res.write(value, (err) => (err ? reject(err) : resolve()));\n      });\n    },\n    close: function (): Promise<void> {\n      return new Promise((resolve) => {\n        res.end(() => resolve());\n      });\n    },\n  };\n}\n"],"mappings":";;;;;;;AA+BA,IAAa,eAAb,MAAqD;CACnD,AAAQ,UAA2B,IAAI,iBAAiB;CAExD,AAAO,KACL,YAIiB;AACjB,OAAK,QAAQ,KAAK,WAAW;AAC7B,SAAO;;CAGT,AAAO,eAAe,GAAG,MAAiC;AACxD,OAAK,QAAQ,gBAAgB,GAAG,KAAK;AACrC,SAAO;;CAGT,AAAO,sBACL,SACiB;AACjB,OAAK,QAAQ,yBAAyB,QAAQ;AAC9C,SAAO;;CAGT,AAAO,UAAU,QAA0C;AACzD,OAAK,QAAQ,UAAU,OAAO;AAC9B,SAAO;;CAGT,AAAO,0BACL,eACiB;AACjB,OAAK,QAAQ,6BAA6B,cAAc;AACxD,SAAO;;CAGT,aAAa,SAE4D;AACvE,SAAO,iBACL,KAAK,QAAQ,OAAO,EACpB,SAAS,kBAAkB,QAAQ,qBAAqB,cACzD;;CAGH,aAAa,SAEoD;AAC/D,SAAO,iBACL,KAAK,QAAQ,OAAO,EACpB,SAAS,gBAAgB,gBAAgB,mBAC1C;;CAGH,QAAQ,SAGN;EACA,MAAM,WAAW,KAAK,QAAQ,OAAO;EACrC,MAAM,YAAY,iBAChB,UACA,SAAS,kBAAkB,QAAQ,qBAAqB,cACzD;EACD,MAAM,YAAY,iBAChB,UACA,SAAS,gBAAgB,gBAAgB,mBAC1C;AAED,WACE,SACA,aACG;AACH,OAAI,QAAQ,oBAAoB,EAC9B,WACE,SACA,SACD;OAED,WAAU,SAA4B,SAA2B;;;CAQvE,OAAO,MAAgC;EACrC,MAAM,WAAW,KAAK,QAAQ,OAAO;EAErC,MAAM,aAAa,QAAQ,SAAS,QAAQ,IAAI,QAAQ,OAAO;AAC/D,WAAS,KAAK,KAAK,oCAAoC,WAAW,KAAK;EAEvE,MAAM,SAAS,MAAM,aACnB,iBAAiB,UAAU,cAAc,CAC1C;AAED,SAAO,IAAI,SAAS,SAAS,WAAW;GACtC,IAAI,SAAS;AACb,UAAO,KAAK,UAAU,MAAa;AACjC,aAAS;AACT,WAAO,EAAE;KACT;AACF,UAAO,OAAO,kBAAkB;AAC9B,QAAI,OACF;IAEF,MAAM,UAAU,OAAO,SAAS;AAChC,QAAI,YAAY,QAAQ,OAAO,YAAY,SACzC,wBACE,IAAI,UACF,8DACD,CACF;QAED,SAAQ,QAAQ,KAAK;KAEvB;IACF;;;AAIN,SAAS,iBACP,UACA,cAC8D;AAC9D,QAAO,gBAAgB,UAAU,aAAa;;AAGhD,SAAS,iBACP,UACA,cACsE;AACtE,QAAO,gBAAgB,UAAU,aAAa;;AAGhD,SAAS,gBACP,UACA,cAIQ;CACR,MAAM,UAAU,qBAAqB,UAAU,cAAc,EAAE,CAAC;AAEhE,SAAQ,aAAa,iBAAiB;EACpC,MAAM,MAAM,YAAY;EAGxB,MAAM,kBAAkB,IAAI,iBAAiB;AAC7C,cAAY,GAAG,eAAe;AAC5B,mBAAgB,OAAO;IACvB;EAEF,MAAM,kBAAkB,QAAQ,OAAO;GACrC;GACA,SAAS,YAAY;GACrB,WAAW,EAAE;GACd,CAAC;AAEF,EADY,aACR,UAAU,gBAAgB,YAAY,gBAAgB,QAAQ;AAElE,kBACG,QAAQ;GACP,aAAa,mBAAmB,YAAY;GAC5C,cAAc,oBAAoB,aAAa;GAC/C,aAAa,gBAAgB;GAC9B,CAAC,CACD,OAAO,MAAM;GAEZ,MAAM,QAAQ,YAAY,EAAE;AAO5B,IALE,0BACE,SAAS,iBACT,KACA,YAAY,QACb,IAAI,SAAS,MACT,MAAM,wBAAwB,MAAM,SAAS,MAAM,SAAS;IACnE;;;AAYR,SAAS,mBACP,SACa;AACb,QAAQ,QAAsC,OAAO,gBAAgB;;AAGvE,SAAS,oBACP,UACc;CACd,MAAM,MAAM;AACZ,QAAO;EACL,OAAO,SAAU,OAAkC;AACjD,UAAO,IAAI,SAAS,SAAS,WAAW;AACtC,QAAI,MAAM,QAAQ,QAAS,MAAM,OAAO,IAAI,GAAG,SAAS,CAAE;KAC1D;;EAEJ,OAAO,WAA2B;AAChC,UAAO,IAAI,SAAS,YAAY;AAC9B,QAAI,UAAU,SAAS,CAAC;KACxB;;EAEL"}
+{"version":3,"file":"node_endpoint.js","names":[],"sources":["../../src/endpoint/node_endpoint.ts"],"sourcesContent":["/*\n * Copyright (c) 2023-2024 - Restate Software, Inc., Restate GmbH\n *\n * This file is part of the Restate SDK for Node.js/TypeScript,\n * which is released under the MIT license.\n *\n * You can find a copy of the license in file LICENSE in the root\n * directory of this repository or package, or at\n * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE\n */\n\nimport type { RestateEndpoint } from \"../index.js\";\nimport type {\n  JournalValueCodec,\n  ServiceDefinition,\n  VirtualObjectDefinition,\n  WorkflowDefinition,\n} from \"@restatedev/restate-sdk-core\";\nimport { IncomingMessage, ServerResponse } from \"node:http\";\nimport { Http2ServerRequest, Http2ServerResponse } from \"node:http2\";\nimport * as http2 from \"node:http2\";\nimport type { Endpoint } from \"./endpoint.js\";\nimport { EndpointBuilder } from \"./endpoint.js\";\nimport { createRestateHandler } from \"./handlers/generic.js\";\nimport { ensureError } from \"../types/errors.js\";\nimport type { LoggerTransport } from \"../logging/logger_transport.js\";\nimport type { DefaultServiceOptions } from \"../endpoint.js\";\nimport { tryCreateContextualLogger } from \"./handlers/utils.js\";\nimport type { InputReader, OutputWriter } from \"./handlers/types.js\";\nimport type { ProtocolMode } from \"./discovery.js\";\n\nexport class NodeEndpoint implements RestateEndpoint {\n  private builder: EndpointBuilder = new EndpointBuilder();\n\n  public bind<P extends string, M>(\n    definition:\n      | ServiceDefinition<P, M>\n      | VirtualObjectDefinition<P, M>\n      | WorkflowDefinition<P, M>\n  ): RestateEndpoint {\n    this.builder.bind(definition);\n    return this;\n  }\n\n  public withIdentityV1(...keys: string[]): RestateEndpoint {\n    this.builder.addIdentityKeys(...keys);\n    return this;\n  }\n\n  public defaultServiceOptions(\n    options: DefaultServiceOptions\n  ): RestateEndpoint {\n    this.builder.setDefaultServiceOptions(options);\n    return this;\n  }\n\n  public setLogger(logger: LoggerTransport): RestateEndpoint {\n    this.builder.setLogger(logger);\n    return this;\n  }\n\n  public journalValueCodecProvider(\n    codecProvider: () => Promise<JournalValueCodec>\n  ): RestateEndpoint {\n    this.builder.setJournalValueCodecProvider(codecProvider);\n    return this;\n  }\n\n  http2Handler(options?: {\n    bidirectional?: boolean;\n  }): (request: Http2ServerRequest, response: Http2ServerResponse) => void {\n    return nodeHttp2Handler(\n      this.builder.build(),\n      options?.bidirectional === false ? \"REQUEST_RESPONSE\" : \"BIDI_STREAM\"\n    );\n  }\n\n  http1Handler(options?: {\n    bidirectional?: boolean;\n  }): (request: IncomingMessage, response: ServerResponse) => void {\n    return nodeHttp1Handler(\n      this.builder.build(),\n      options?.bidirectional ? \"BIDI_STREAM\" : \"REQUEST_RESPONSE\"\n    );\n  }\n\n  handler(options?: { bidirectional?: boolean }): {\n    (request: IncomingMessage, response: ServerResponse): void;\n    (request: Http2ServerRequest, response: Http2ServerResponse): void;\n  } {\n    const endpoint = this.builder.build();\n    const h2Handler = nodeHttp2Handler(\n      endpoint,\n      options?.bidirectional === false ? \"REQUEST_RESPONSE\" : \"BIDI_STREAM\"\n    );\n    const h1Handler = nodeHttp1Handler(\n      endpoint,\n      options?.bidirectional ? \"BIDI_STREAM\" : \"REQUEST_RESPONSE\"\n    );\n\n    return ((\n      request: IncomingMessage | Http2ServerRequest,\n      response: ServerResponse | Http2ServerResponse\n    ) => {\n      if (request.httpVersionMajor >= 2) {\n        h2Handler(\n          request as Http2ServerRequest,\n          response as Http2ServerResponse\n        );\n      } else {\n        h1Handler(request as IncomingMessage, response as ServerResponse);\n      }\n    }) as {\n      (request: IncomingMessage, response: ServerResponse): void;\n      (request: Http2ServerRequest, response: Http2ServerResponse): void;\n    };\n  }\n\n  listen(port?: number): Promise<number> {\n    const endpoint = this.builder.build();\n\n    const actualPort = port ?? parseInt(process.env.PORT ?? \"9080\");\n    endpoint.rlog.info(`Restate SDK started listening on ${actualPort}...`);\n\n    const server = http2.createServer(\n      nodeHttp2Handler(endpoint, \"BIDI_STREAM\")\n    );\n\n    return new Promise((resolve, reject) => {\n      let failed = false;\n      server.once(\"error\", (e: Error) => {\n        failed = true;\n        reject(e);\n      });\n      server.listen(actualPort, () => {\n        if (failed) {\n          return;\n        }\n        const address = server.address();\n        if (address === null || typeof address === \"string\") {\n          reject(\n            new TypeError(\n              \"endpoint.listen() currently supports only binding to a PORT\"\n            )\n          );\n        } else {\n          resolve(address.port);\n        }\n      });\n    });\n  }\n}\n\nfunction nodeHttp1Handler(\n  endpoint: Endpoint,\n  protocolMode: ProtocolMode\n): (request: IncomingMessage, response: ServerResponse) => void {\n  return nodeHandlerImpl(endpoint, protocolMode);\n}\n\nfunction nodeHttp2Handler(\n  endpoint: Endpoint,\n  protocolMode: ProtocolMode\n): (request: Http2ServerRequest, response: Http2ServerResponse) => void {\n  return nodeHandlerImpl(endpoint, protocolMode);\n}\n\nfunction nodeHandlerImpl(\n  endpoint: Endpoint,\n  protocolMode: ProtocolMode\n): (\n  request: Http2ServerRequest | IncomingMessage,\n  response: Http2ServerResponse | ServerResponse\n) => void {\n  const handler = createRestateHandler(endpoint, protocolMode, {});\n\n  return (httpRequest, httpResponse) => {\n    const url = httpRequest.url!;\n    const inputReader = inputReaderAdapter(httpRequest);\n    const outputWriter = outputWriterAdapter(httpResponse);\n    const res = httpResponse as NodeWritableResponse;\n\n    // Abort controller used to cleanup resources at the end of this stream lifecycle\n    const abortController = new AbortController();\n    httpRequest.on(\"close\", () => {\n      abortController.abort();\n    });\n\n    const writeHead = res.writeHead.bind(res);\n\n    // handle should never throw\n    const restateResponse = handler.handle({\n      url,\n      headers: httpRequest.headers,\n      extraArgs: [],\n    });\n\n    restateResponse\n      .process({\n        inputReader,\n        outputWriter,\n        writeHead,\n        abortSignal: abortController.signal,\n      })\n      .catch((e) => {\n        // Responses handle their own errors before rejecting; anything\n        // reaching here is an unexpected failure — just log.\n        const error = ensureError(e);\n        const logger =\n          tryCreateContextualLogger(\n            endpoint.loggerTransport,\n            url,\n            httpRequest.headers\n          ) ?? endpoint.rlog;\n        logger.error(\"Unexpected error: \" + (error.stack ?? error.message));\n      });\n  };\n}\n\n// Both ServerResponse and Http2ServerResponse satisfy this interface.\n// We use it to avoid TS union overload incompatibilities.\ninterface NodeWritableResponse {\n  writeHead(statusCode: number, headers: Record<string, string>): void;\n  write(chunk: Uint8Array, callback: (err?: Error | null) => void): boolean;\n  end(callback: () => void): void;\n}\n\nfunction inputReaderAdapter(\n  request: Http2ServerRequest | IncomingMessage\n): InputReader {\n  return (request as AsyncIterable<Uint8Array>)[Symbol.asyncIterator]();\n}\n\nfunction outputWriterAdapter(\n  response: Http2ServerResponse | ServerResponse\n): OutputWriter {\n  const res = response as NodeWritableResponse;\n  return {\n    write: function (value: Uint8Array): Promise<void> {\n      return new Promise((resolve, reject) => {\n        res.write(value, (err) => (err ? reject(err) : resolve()));\n      });\n    },\n    close: function (): Promise<void> {\n      return new Promise((resolve) => {\n        res.end(() => resolve());\n      });\n    },\n  };\n}\n"],"mappings":";;;;;;;AA+BA,IAAa,eAAb,MAAqD;CACnD,AAAQ,UAA2B,IAAI,iBAAiB;CAExD,AAAO,KACL,YAIiB;AACjB,OAAK,QAAQ,KAAK,WAAW;AAC7B,SAAO;;CAGT,AAAO,eAAe,GAAG,MAAiC;AACxD,OAAK,QAAQ,gBAAgB,GAAG,KAAK;AACrC,SAAO;;CAGT,AAAO,sBACL,SACiB;AACjB,OAAK,QAAQ,yBAAyB,QAAQ;AAC9C,SAAO;;CAGT,AAAO,UAAU,QAA0C;AACzD,OAAK,QAAQ,UAAU,OAAO;AAC9B,SAAO;;CAGT,AAAO,0BACL,eACiB;AACjB,OAAK,QAAQ,6BAA6B,cAAc;AACxD,SAAO;;CAGT,aAAa,SAE4D;AACvE,SAAO,iBACL,KAAK,QAAQ,OAAO,EACpB,SAAS,kBAAkB,QAAQ,qBAAqB,cACzD;;CAGH,aAAa,SAEoD;AAC/D,SAAO,iBACL,KAAK,QAAQ,OAAO,EACpB,SAAS,gBAAgB,gBAAgB,mBAC1C;;CAGH,QAAQ,SAGN;EACA,MAAM,WAAW,KAAK,QAAQ,OAAO;EACrC,MAAM,YAAY,iBAChB,UACA,SAAS,kBAAkB,QAAQ,qBAAqB,cACzD;EACD,MAAM,YAAY,iBAChB,UACA,SAAS,gBAAgB,gBAAgB,mBAC1C;AAED,WACE,SACA,aACG;AACH,OAAI,QAAQ,oBAAoB,EAC9B,WACE,SACA,SACD;OAED,WAAU,SAA4B,SAA2B;;;CAQvE,OAAO,MAAgC;EACrC,MAAM,WAAW,KAAK,QAAQ,OAAO;EAErC,MAAM,aAAa,QAAQ,SAAS,QAAQ,IAAI,QAAQ,OAAO;AAC/D,WAAS,KAAK,KAAK,oCAAoC,WAAW,KAAK;EAEvE,MAAM,SAAS,MAAM,aACnB,iBAAiB,UAAU,cAAc,CAC1C;AAED,SAAO,IAAI,SAAS,SAAS,WAAW;GACtC,IAAI,SAAS;AACb,UAAO,KAAK,UAAU,MAAa;AACjC,aAAS;AACT,WAAO,EAAE;KACT;AACF,UAAO,OAAO,kBAAkB;AAC9B,QAAI,OACF;IAEF,MAAM,UAAU,OAAO,SAAS;AAChC,QAAI,YAAY,QAAQ,OAAO,YAAY,SACzC,wBACE,IAAI,UACF,8DACD,CACF;QAED,SAAQ,QAAQ,KAAK;KAEvB;IACF;;;AAIN,SAAS,iBACP,UACA,cAC8D;AAC9D,QAAO,gBAAgB,UAAU,aAAa;;AAGhD,SAAS,iBACP,UACA,cACsE;AACtE,QAAO,gBAAgB,UAAU,aAAa;;AAGhD,SAAS,gBACP,UACA,cAIQ;CACR,MAAM,UAAU,qBAAqB,UAAU,cAAc,EAAE,CAAC;AAEhE,SAAQ,aAAa,iBAAiB;EACpC,MAAM,MAAM,YAAY;EACxB,MAAM,cAAc,mBAAmB,YAAY;EACnD,MAAM,eAAe,oBAAoB,aAAa;EACtD,MAAM,MAAM;EAGZ,MAAM,kBAAkB,IAAI,iBAAiB;AAC7C,cAAY,GAAG,eAAe;AAC5B,mBAAgB,OAAO;IACvB;EAEF,MAAM,YAAY,IAAI,UAAU,KAAK,IAAI;AASzC,EANwB,QAAQ,OAAO;GACrC;GACA,SAAS,YAAY;GACrB,WAAW,EAAE;GACd,CAAC,CAGC,QAAQ;GACP;GACA;GACA;GACA,aAAa,gBAAgB;GAC9B,CAAC,CACD,OAAO,MAAM;GAGZ,MAAM,QAAQ,YAAY,EAAE;AAO5B,IALE,0BACE,SAAS,iBACT,KACA,YAAY,QACb,IAAI,SAAS,MACT,MAAM,wBAAwB,MAAM,SAAS,MAAM,SAAS;IACnE;;;AAYR,SAAS,mBACP,SACa;AACb,QAAQ,QAAsC,OAAO,gBAAgB;;AAGvE,SAAS,oBACP,UACc;CACd,MAAM,MAAM;AACZ,QAAO;EACL,OAAO,SAAU,OAAkC;AACjD,UAAO,IAAI,SAAS,SAAS,WAAW;AACtC,QAAI,MAAM,QAAQ,QAAS,MAAM,OAAO,IAAI,GAAG,SAAS,CAAE;KAC1D;;EAEJ,OAAO,WAA2B;AAChC,UAAO,IAAI,SAAS,YAAY;AAC9B,QAAI,UAAU,SAAS,CAAC;KACxB;;EAEL"}

package/dist/io.cjs

@@ -6,10 +6,12 @@ const require_promises = require('./promises.cjs');
 */
 var InputPump = class {
 	currentRead;
+	closed;
 	constructor(coreVm, inputReader, errorCallback) {
 		this.coreVm = coreVm;
 		this.inputReader = inputReader;
 		this.errorCallback = errorCallback;
+		this.closed = false;
 	}
 	awaitNextProgress() {
 		if (this.currentRead === void 0) this.currentRead = this.readNext().finally(() => {
@@ -18,6 +20,7 @@ var InputPump = class {
 		return new Promise((resolve) => this.currentRead?.finally(resolve));
 	}
 	async readNext() {
+		if (this.closed) return require_promises.pendingPromise();
 		let nextValue;
 		try {
 			nextValue = await this.inputReader.next();
@@ -25,8 +28,10 @@ var InputPump = class {
 			this.errorCallback(e);
 			return require_promises.pendingPromise();
 		}
-		if (nextValue.done) this.coreVm.notify_input_closed();
-		else if (nextValue.value !== void 0) this.coreVm.notify_input(nextValue.value);
+		if (nextValue.done) {
+			this.closed = true;
+			this.coreVm.notify_input_closed();
+		} else if (nextValue.value !== void 0) this.coreVm.notify_input(nextValue.value);
 	}
 };
 /**

package/dist/io.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"io.d.ts","sourceRoot":"","sources":["../src/io.ts"],"names":[],"mappings":"AAaA,OAAO,KAAK,KAAK,EAAE,MAAM,yDAAyD,CAAC;AAEnF,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAEzE;;GAEG;AACH,qBAAa,SAAS;IAIlB,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,WAAW;IAC5B,OAAO,CAAC,QAAQ,CAAC,aAAa;IALhC,OAAO,CAAC,WAAW,CAAC,CAAgB;gBAGjB,MAAM,EAAE,EAAE,CAAC,MAAM,EACjB,WAAW,EAAE,WAAW,EACxB,aAAa,EAAE,CAAC,CAAC,EAAE,GAAG,KAAK,IAAI;IAMlD,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC;YAYpB,QAAQ;CAevB;AAED;;GAEG;AACH,qBAAa,UAAU;IAEnB,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,YAAY;gBADZ,MAAM,EAAE,EAAE,CAAC,MAAM,EACjB,YAAY,EAAE,YAAY;IAGvC,iBAAiB;CASxB"}
+{"version":3,"file":"io.d.ts","sourceRoot":"","sources":["../src/io.ts"],"names":[],"mappings":"AAaA,OAAO,KAAK,KAAK,EAAE,MAAM,yDAAyD,CAAC;AAEnF,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAEzE;;GAEG;AACH,qBAAa,SAAS;IAKlB,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,WAAW;IAC5B,OAAO,CAAC,QAAQ,CAAC,aAAa;IANhC,OAAO,CAAC,WAAW,CAAC,CAAgB;IACpC,OAAO,CAAC,MAAM,CAAU;gBAGL,MAAM,EAAE,EAAE,CAAC,MAAM,EACjB,WAAW,EAAE,WAAW,EACxB,aAAa,EAAE,CAAC,CAAC,EAAE,GAAG,KAAK,IAAI;IAQlD,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC;YAYpB,QAAQ;CAmBvB;AAED;;GAEG;AACH,qBAAa,UAAU;IAEnB,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,YAAY;gBADZ,MAAM,EAAE,EAAE,CAAC,MAAM,EACjB,YAAY,EAAE,YAAY;IAGvC,iBAAiB;CASxB"}

package/dist/io.js

@@ -6,10 +6,12 @@ import { pendingPromise } from "./promises.js";
 */
 var InputPump = class {
 	currentRead;
+	closed;
 	constructor(coreVm, inputReader, errorCallback) {
 		this.coreVm = coreVm;
 		this.inputReader = inputReader;
 		this.errorCallback = errorCallback;
+		this.closed = false;
 	}
 	awaitNextProgress() {
 		if (this.currentRead === void 0) this.currentRead = this.readNext().finally(() => {
@@ -18,6 +20,7 @@ var InputPump = class {
 		return new Promise((resolve) => this.currentRead?.finally(resolve));
 	}
 	async readNext() {
+		if (this.closed) return pendingPromise();
 		let nextValue;
 		try {
 			nextValue = await this.inputReader.next();
@@ -25,8 +28,10 @@ var InputPump = class {
 			this.errorCallback(e);
 			return pendingPromise();
 		}
-		if (nextValue.done) this.coreVm.notify_input_closed();
-		else if (nextValue.value !== void 0) this.coreVm.notify_input(nextValue.value);
+		if (nextValue.done) {
+			this.closed = true;
+			this.coreVm.notify_input_closed();
+		} else if (nextValue.value !== void 0) this.coreVm.notify_input(nextValue.value);
 	}
 };
 /**

package/dist/io.js.map

@@ -1 +1 @@
-{"version":3,"file":"io.js","names":["coreVm: vm.WasmVM","inputReader: InputReader","errorCallback: (e: any) => void","outputWriter: OutputWriter"],"sources":["../src/io.ts"],"sourcesContent":["/*\n * Copyright (c) 2023-2025 - Restate Software, Inc., Restate GmbH\n *\n * This file is part of the Restate SDK for Node.js/TypeScript,\n * which is released under the MIT license.\n *\n * You can find a copy of the license in file LICENSE in the root\n * directory of this repository or package, or at\n * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE\n */\n\n/* eslint-disable @typescript-eslint/no-explicit-any */\n\nimport type * as vm from \"./endpoint/handlers/vm/sdk_shared_core_wasm_bindings.js\";\nimport { pendingPromise } from \"./promises.js\";\nimport { InputReader, OutputWriter } from \"./endpoint/handlers/types.js\";\n\n/**\n * Adapter between input stream and vm. It moves forward when [awaitNextProgress] is invoked.\n */\nexport class InputPump {\n  private currentRead?: Promise<void>;\n\n  constructor(\n    private readonly coreVm: vm.WasmVM,\n    private readonly inputReader: InputReader,\n    private readonly errorCallback: (e: any) => void\n  ) {}\n\n  // This function triggers a read on the input reader,\n  // and will notify the caller that a read was executed\n  // and the result was piped in the state machine.\n  awaitNextProgress(): Promise<void> {\n    if (this.currentRead === undefined) {\n      // Register a new read\n      this.currentRead = this.readNext().finally(() => {\n        this.currentRead = undefined;\n      });\n    }\n\n    // eslint-disable-next-line @typescript-eslint/no-misused-promises\n    return new Promise<void>((resolve) => this.currentRead?.finally(resolve));\n  }\n\n  private async readNext(): Promise<void> {\n    // Take input, and notify it to the vm\n    let nextValue;\n    try {\n      nextValue = await this.inputReader.next();\n    } catch (e) {\n      this.errorCallback(e);\n      return pendingPromise<void>();\n    }\n    if (nextValue.done) {\n      this.coreVm.notify_input_closed();\n    } else if (nextValue.value !== undefined) {\n      this.coreVm.notify_input(nextValue.value);\n    }\n  }\n}\n\n/**\n * Adapter between output stream and vm. It moves forward when [awaitNextProgress] is invoked.\n */\nexport class OutputPump {\n  constructor(\n    private readonly coreVm: vm.WasmVM,\n    private readonly outputWriter: OutputWriter\n  ) {}\n\n  async awaitNextProgress() {\n    const nextOutput = this.coreVm.take_output() as\n      | Uint8Array\n      | null\n      | undefined;\n    if (nextOutput instanceof Uint8Array) {\n      await this.outputWriter.write(nextOutput);\n    }\n  }\n}\n"],"mappings":";;;;;;AAoBA,IAAa,YAAb,MAAuB;CACrB,AAAQ;CAER,YACE,AAAiBA,QACjB,AAAiBC,aACjB,AAAiBC,eACjB;EAHiB;EACA;EACA;;CAMnB,oBAAmC;AACjC,MAAI,KAAK,gBAAgB,OAEvB,MAAK,cAAc,KAAK,UAAU,CAAC,cAAc;AAC/C,QAAK,cAAc;IACnB;AAIJ,SAAO,IAAI,SAAe,YAAY,KAAK,aAAa,QAAQ,QAAQ,CAAC;;CAG3E,MAAc,WAA0B;EAEtC,IAAI;AACJ,MAAI;AACF,eAAY,MAAM,KAAK,YAAY,MAAM;WAClC,GAAG;AACV,QAAK,cAAc,EAAE;AACrB,UAAO,gBAAsB;;AAE/B,MAAI,UAAU,KACZ,MAAK,OAAO,qBAAqB;WACxB,UAAU,UAAU,OAC7B,MAAK,OAAO,aAAa,UAAU,MAAM;;;;;;AAQ/C,IAAa,aAAb,MAAwB;CACtB,YACE,AAAiBF,QACjB,AAAiBG,cACjB;EAFiB;EACA;;CAGnB,MAAM,oBAAoB;EACxB,MAAM,aAAa,KAAK,OAAO,aAAa;AAI5C,MAAI,sBAAsB,WACxB,OAAM,KAAK,aAAa,MAAM,WAAW"}
+{"version":3,"file":"io.js","names":["coreVm: vm.WasmVM","inputReader: InputReader","errorCallback: (e: any) => void","outputWriter: OutputWriter"],"sources":["../src/io.ts"],"sourcesContent":["/*\n * Copyright (c) 2023-2025 - Restate Software, Inc., Restate GmbH\n *\n * This file is part of the Restate SDK for Node.js/TypeScript,\n * which is released under the MIT license.\n *\n * You can find a copy of the license in file LICENSE in the root\n * directory of this repository or package, or at\n * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE\n */\n\n/* eslint-disable @typescript-eslint/no-explicit-any */\n\nimport type * as vm from \"./endpoint/handlers/vm/sdk_shared_core_wasm_bindings.js\";\nimport { pendingPromise } from \"./promises.js\";\nimport { InputReader, OutputWriter } from \"./endpoint/handlers/types.js\";\n\n/**\n * Adapter between input stream and vm. It moves forward when [awaitNextProgress] is invoked.\n */\nexport class InputPump {\n  private currentRead?: Promise<void>;\n  private closed: boolean;\n\n  constructor(\n    private readonly coreVm: vm.WasmVM,\n    private readonly inputReader: InputReader,\n    private readonly errorCallback: (e: any) => void\n  ) {\n    this.closed = false;\n  }\n\n  // This function triggers a read on the input reader,\n  // and will notify the caller that a read was executed\n  // and the result was piped in the state machine.\n  awaitNextProgress(): Promise<void> {\n    if (this.currentRead === undefined) {\n      // Register a new read\n      this.currentRead = this.readNext().finally(() => {\n        this.currentRead = undefined;\n      });\n    }\n\n    // eslint-disable-next-line @typescript-eslint/no-misused-promises\n    return new Promise<void>((resolve) => this.currentRead?.finally(resolve));\n  }\n\n  private async readNext(): Promise<void> {\n    if (this.closed) {\n      return pendingPromise<void>();\n    }\n    // Take input, and notify it to the vm\n    let nextValue;\n    try {\n      nextValue = await this.inputReader.next();\n    } catch (e) {\n      this.errorCallback(e);\n      return pendingPromise<void>();\n    }\n    if (nextValue.done) {\n      this.closed = true;\n      this.coreVm.notify_input_closed();\n    } else if (nextValue.value !== undefined) {\n      this.coreVm.notify_input(nextValue.value);\n    }\n  }\n}\n\n/**\n * Adapter between output stream and vm. It moves forward when [awaitNextProgress] is invoked.\n */\nexport class OutputPump {\n  constructor(\n    private readonly coreVm: vm.WasmVM,\n    private readonly outputWriter: OutputWriter\n  ) {}\n\n  async awaitNextProgress() {\n    const nextOutput = this.coreVm.take_output() as\n      | Uint8Array\n      | null\n      | undefined;\n    if (nextOutput instanceof Uint8Array) {\n      await this.outputWriter.write(nextOutput);\n    }\n  }\n}\n"],"mappings":";;;;;;AAoBA,IAAa,YAAb,MAAuB;CACrB,AAAQ;CACR,AAAQ;CAER,YACE,AAAiBA,QACjB,AAAiBC,aACjB,AAAiBC,eACjB;EAHiB;EACA;EACA;AAEjB,OAAK,SAAS;;CAMhB,oBAAmC;AACjC,MAAI,KAAK,gBAAgB,OAEvB,MAAK,cAAc,KAAK,UAAU,CAAC,cAAc;AAC/C,QAAK,cAAc;IACnB;AAIJ,SAAO,IAAI,SAAe,YAAY,KAAK,aAAa,QAAQ,QAAQ,CAAC;;CAG3E,MAAc,WAA0B;AACtC,MAAI,KAAK,OACP,QAAO,gBAAsB;EAG/B,IAAI;AACJ,MAAI;AACF,eAAY,MAAM,KAAK,YAAY,MAAM;WAClC,GAAG;AACV,QAAK,cAAc,EAAE;AACrB,UAAO,gBAAsB;;AAE/B,MAAI,UAAU,MAAM;AAClB,QAAK,SAAS;AACd,QAAK,OAAO,qBAAqB;aACxB,UAAU,UAAU,OAC7B,MAAK,OAAO,aAAa,UAAU,MAAM;;;;;;AAQ/C,IAAa,aAAb,MAAwB;CACtB,YACE,AAAiBF,QACjB,AAAiBG,cACjB;EAFiB;EACA;;CAGnB,MAAM,oBAAoB;EACxB,MAAM,aAAa,KAAK,OAAO,aAAa;AAI5C,MAAI,sBAAsB,WACxB,OAAM,KAAK,aAAa,MAAM,WAAW"}

package/dist/package.cjs

@@ -1,6 +1,6 @@
 
 //#region package.json
-var version = "1.12.0";
+var version = "1.14.0";
 
 //#endregion
 Object.defineProperty(exports, 'version', {

package/dist/package.js

@@ -1,5 +1,5 @@
 //#region package.json
-var version = "1.12.0";
+var version = "1.14.0";
 
 //#endregion
 export { version };

package/dist/package.js.map

@@ -1 +1 @@
-{"version":3,"file":"package.js","names":[],"sources":["../package.json"],"sourcesContent":["{\n  \"name\": \"@restatedev/restate-sdk\",\n  \"version\": \"1.12.0\",\n  \"description\": \"Typescript SDK for Restate\",\n  \"author\": \"Restate Developers\",\n  \"email\": \"code@restate.dev\",\n  \"license\": \"MIT\",\n  \"homepage\": \"https://github.com/restatedev/sdk-typescript#readme\",\n  \"repository\": {\n    \"type\": \"git\",\n    \"url\": \"git+https://github.com/restatedev/sdk-typescript.git\"\n  },\n  \"bugs\": {\n    \"url\": \"https://github.com/restatedev/sdk-typescript/issues\"\n  },\n  \"type\": \"module\",\n  \"main\": \"./dist/index.cjs\",\n  \"module\": \"./dist/index.js\",\n  \"types\": \"./dist/index.d.cts\",\n  \"exports\": {\n    \".\": {\n      \"import\": \"./dist/index.js\",\n      \"require\": \"./dist/index.cjs\"\n    },\n    \"./fetch\": {\n      \"import\": \"./dist/fetch.js\",\n      \"require\": \"./dist/fetch.cjs\"\n    },\n    \"./lambda\": {\n      \"import\": \"./dist/lambda.js\",\n      \"require\": \"./dist/lambda.cjs\"\n    },\n    \"./node\": {\n      \"import\": \"./dist/node.js\",\n      \"require\": \"./dist/node.cjs\"\n    },\n    \"./package.json\": \"./package.json\"\n  },\n  \"files\": [\n    \"dist\",\n    \"README.md\"\n  ],\n  \"publishConfig\": {\n    \"access\": \"public\"\n  },\n  \"scripts\": {\n    \"test\": \"turbo run _test --filter={.}...\",\n    \"_test\": \"vitest run\",\n    \"build\": \"turbo run _build --filter={.}...\",\n    \"_build\": \"tsc --noEmit && tsdown\",\n    \"dev\": \"tsc --noEmit --watch\",\n    \"clean\": \"rm -rf dist *.tsbuildinfo .turbo\",\n    \"check:types\": \"turbo run _check:types --filter={.}...\",\n    \"_check:types\": \"tsc --noEmit --project tsconfig.build.json\",\n    \"lint\": \"eslint .\",\n    \"check:exports\": \"turbo run _check:exports --filter={.}...\",\n    \"_check:exports\": \"attw --pack .\",\n    \"check:api\": \"turbo run _check:api --filter={.}...\",\n    \"_check:api\": \"api-extractor run --local && api-extractor run --local --config api-extractor.fetch.json && api-extractor run --local --config api-extractor.lambda.json && api-extractor run --local --config api-extractor.node.json\",\n    \"prepublishOnly\": \"pnpm -w verify\"\n  },\n  \"dependencies\": {\n    \"@restatedev/restate-sdk-core\": \"workspace:*\"\n  },\n  \"devDependencies\": {\n    \"@types/aws-lambda\": \"^8.10.115\"\n  },\n  \"typesVersions\": {\n    \"*\": {\n      \"fetch\": [\n        \"./dist/fetch.d.ts\"\n      ],\n      \"lambda\": [\n        \"./dist/lambda.d.ts\"\n      ],\n      \"node\": [\n        \"./dist/node.d.ts\"\n      ]\n    }\n  }\n}\n"],"mappings":";cAEa"}
+{"version":3,"file":"package.js","names":[],"sources":["../package.json"],"sourcesContent":["{\n  \"name\": \"@restatedev/restate-sdk\",\n  \"version\": \"1.14.0\",\n  \"description\": \"Typescript SDK for Restate\",\n  \"author\": \"Restate Developers\",\n  \"email\": \"code@restate.dev\",\n  \"license\": \"MIT\",\n  \"homepage\": \"https://github.com/restatedev/sdk-typescript#readme\",\n  \"repository\": {\n    \"type\": \"git\",\n    \"url\": \"git+https://github.com/restatedev/sdk-typescript.git\"\n  },\n  \"bugs\": {\n    \"url\": \"https://github.com/restatedev/sdk-typescript/issues\"\n  },\n  \"type\": \"module\",\n  \"main\": \"./dist/index.cjs\",\n  \"module\": \"./dist/index.js\",\n  \"types\": \"./dist/index.d.cts\",\n  \"exports\": {\n    \".\": {\n      \"import\": \"./dist/index.js\",\n      \"require\": \"./dist/index.cjs\"\n    },\n    \"./fetch\": {\n      \"import\": \"./dist/fetch.js\",\n      \"require\": \"./dist/fetch.cjs\"\n    },\n    \"./lambda\": {\n      \"import\": \"./dist/lambda.js\",\n      \"require\": \"./dist/lambda.cjs\"\n    },\n    \"./node\": {\n      \"import\": \"./dist/node.js\",\n      \"require\": \"./dist/node.cjs\"\n    },\n    \"./package.json\": \"./package.json\"\n  },\n  \"files\": [\n    \"dist\",\n    \"README.md\"\n  ],\n  \"publishConfig\": {\n    \"access\": \"public\"\n  },\n  \"scripts\": {\n    \"test\": \"turbo run _test --filter={.}...\",\n    \"_test\": \"vitest run\",\n    \"build\": \"turbo run _build --filter={.}...\",\n    \"_build\": \"tsc --noEmit && tsdown\",\n    \"dev\": \"tsc --noEmit --watch\",\n    \"clean\": \"rm -rf dist *.tsbuildinfo .turbo\",\n    \"check:types\": \"turbo run _check:types --filter={.}...\",\n    \"_check:types\": \"tsc --noEmit --project tsconfig.build.json\",\n    \"lint\": \"eslint .\",\n    \"check:exports\": \"turbo run _check:exports --filter={.}...\",\n    \"_check:exports\": \"attw --pack .\",\n    \"check:api\": \"turbo run _check:api --filter={.}...\",\n    \"_check:api\": \"api-extractor run --local && api-extractor run --local --config api-extractor.fetch.json && api-extractor run --local --config api-extractor.lambda.json && api-extractor run --local --config api-extractor.node.json\",\n    \"prepublishOnly\": \"pnpm -w verify\"\n  },\n  \"dependencies\": {\n    \"@restatedev/restate-sdk-core\": \"workspace:*\"\n  },\n  \"devDependencies\": {\n    \"@types/aws-lambda\": \"^8.10.115\"\n  },\n  \"typesVersions\": {\n    \"*\": {\n      \"fetch\": [\n        \"./dist/fetch.d.ts\"\n      ],\n      \"lambda\": [\n        \"./dist/lambda.d.ts\"\n      ],\n      \"node\": [\n        \"./dist/node.d.ts\"\n      ]\n    }\n  }\n}\n"],"mappings":";cAEa"}

package/dist/promises.cjs

@@ -140,6 +140,7 @@ var CombinatorRestatePromise = class CombinatorRestatePromise extends BaseRestat
 };
 var MappedRestatePromise = class extends BaseRestatePromise {
 	publicPromiseMapper;
+	_mappedPromise;
 	constructor(ctx, inner, mapper) {
 		super(ctx);
 		this.inner = inner;
@@ -162,6 +163,9 @@ var MappedRestatePromise = class extends BaseRestatePromise {
 		return this.inner.uncompletedLeaves();
 	}
 	publicPromise() {
+		return this._mappedPromise ??= this.buildMappedPromise();
+	}
+	buildMappedPromise() {
 		const promiseMapper = this.publicPromiseMapper;
 		return this.inner.publicPromise().then((t) => promiseMapper(t, void 0), (error) => {
 			if (error instanceof require_errors.RestateError) return promiseMapper(void 0, error);
@@ -171,22 +175,26 @@ var MappedRestatePromise = class extends BaseRestatePromise {
 	[Symbol.toStringTag] = "RestateMappedPromise";
 };
 var ConstRestatePromise = class ConstRestatePromise extends InternalRestatePromise {
-	constructor(constPromise, settled) {
+	_constPromise;
+	constructor(promiseFactory, settled) {
 		super();
-		this.constPromise = constPromise;
+		this.promiseFactory = promiseFactory;
 		this.settled = settled;
 	}
+	get constPromise() {
+		return this._constPromise ??= this.promiseFactory();
+	}
 	static resolve(value) {
-		return new ConstRestatePromise(Promise.resolve(value), true);
+		return new ConstRestatePromise(() => Promise.resolve(value), true);
 	}
 	static reject(reason) {
-		return new ConstRestatePromise(Promise.reject(reason), true);
+		return new ConstRestatePromise(() => Promise.reject(reason), true);
 	}
 	static pending() {
-		return new ConstRestatePromise(pendingPromise(), false);
+		return new ConstRestatePromise(() => pendingPromise(), false);
 	}
 	static fromPromise(promise, settled) {
-		return new ConstRestatePromise(promise, settled);
+		return new ConstRestatePromise(() => promise, settled);
 	}
 	then(onfulfilled, onrejected) {
 		return this.constPromise.then(onfulfilled, onrejected);
@@ -202,7 +210,9 @@ var ConstRestatePromise = class ConstRestatePromise extends InternalRestatePromi
 		return ConstRestatePromise.reject(new require_errors.TimeoutError());
 	}
 	map(mapper) {
-		return ConstRestatePromise.fromPromise(this.constPromise.then((value) => mapper(value, void 0), (reason) => mapper(void 0, reason)), this.settled);
+		if (!this.settled) return this;
+		const selfConstPromise = this.constPromise;
+		return new ConstRestatePromise(() => selfConstPromise.then((value) => mapper(value, void 0), (reason) => mapper(void 0, reason)), this.settled);
 	}
 	tryCancel() {}
 	publicPromise() {
@@ -243,9 +253,10 @@ var PromisesExecutor = class {
 				const handles = restatePromise.uncompletedLeaves();
 				if (handles.length === 0) return;
 				const doProgressResult = this.coreVm.do_progress(new Uint32Array(handles));
-				if (doProgressResult === "AnyCompleted") {} else if (doProgressResult === "ReadFromInput") await this.inputPump.awaitNextProgress();
-				else if (doProgressResult === "WaitingPendingRun") await this.runClosuresTracker.awaitNextCompletedRun();
-				else if (doProgressResult === "CancelSignalReceived") {
+				if (doProgressResult === "AnyCompleted") {} else if (doProgressResult === "ReadFromInput" || doProgressResult === "WaitingPendingRun") {
+					await this.outputPump.awaitNextProgress();
+					await Promise.race([this.inputPump.awaitNextProgress(), this.runClosuresTracker.awaitNextCompletedRun()]);
+				} else if (doProgressResult === "CancelSignalReceived") {
 					restatePromise.tryCancel();
 					return;
 				} else {