@restatedev/restate-sdk 0.9.2 → 1.0.1

package/README.md

@@ -46,3 +46,13 @@ For brand-new projects, we recommend using the [Restate Node Template](https://g
 ```shell
 npx -y @restatedev/create-app@latest
 ```
+
+## Versions
+
+This library follows [Semantic Versioning](https://semver.org/).
+
+The compatibility with Restate is described in the following table:
+
+| Restate Server\sdk-typescript | 1.0 |
+|-------------------------------|-----|
+| 1.0                           | ✅   |

package/dist/cjs/package.json

@@ -0,0 +1 @@
+{"type":"commonjs"}

package/dist/cjs/src/common_api.d.ts

@@ -0,0 +1,7 @@
+export { Context, ObjectContext, ObjectSharedContext, CombineablePromise, WorkflowContext, WorkflowSharedContext, Rand, } from "./context.js";
+export type { Client, SendClient } from "./types/rpc.js";
+export { service, object, workflow, handlers } from "./types/rpc.js";
+export type { ServiceDefinition, VirtualObjectDefinition, WorkflowDefinition, } from "@restatedev/restate-sdk-core";
+export { ServiceBundle, RestateEndpoint } from "./endpoint.js";
+export { RestateError, TerminalError, TimeoutError } from "./types/errors.js";
+//# sourceMappingURL=common_api.d.ts.map

package/dist/cjs/src/common_api.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"common_api.d.ts","sourceRoot":"","sources":["../../../src/common_api.ts"],"names":[],"mappings":"AAWA,OAAO,EACL,OAAO,EACP,aAAa,EACb,mBAAmB,EACnB,kBAAkB,EAClB,eAAe,EACf,qBAAqB,EACrB,IAAI,GACL,MAAM,cAAc,CAAC;AAEtB,YAAY,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACzD,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAErE,YAAY,EACV,iBAAiB,EACjB,uBAAuB,EACvB,kBAAkB,GACnB,MAAM,8BAA8B,CAAC;AAEtC,OAAO,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAC/D,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/cjs/src/common_api.js

@@ -0,0 +1,25 @@
+"use strict";
+/*
+ * Copyright (c) 2023-2024 - Restate Software, Inc., Restate GmbH
+ *
+ * This file is part of the Restate SDK for Node.js/TypeScript,
+ * which is released under the MIT license.
+ *
+ * You can find a copy of the license in file LICENSE in the root
+ * directory of this repository or package, or at
+ * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TimeoutError = exports.TerminalError = exports.RestateError = exports.handlers = exports.workflow = exports.object = exports.service = exports.CombineablePromise = void 0;
+var context_js_1 = require("./context.js");
+Object.defineProperty(exports, "CombineablePromise", { enumerable: true, get: function () { return context_js_1.CombineablePromise; } });
+var rpc_js_1 = require("./types/rpc.js");
+Object.defineProperty(exports, "service", { enumerable: true, get: function () { return rpc_js_1.service; } });
+Object.defineProperty(exports, "object", { enumerable: true, get: function () { return rpc_js_1.object; } });
+Object.defineProperty(exports, "workflow", { enumerable: true, get: function () { return rpc_js_1.workflow; } });
+Object.defineProperty(exports, "handlers", { enumerable: true, get: function () { return rpc_js_1.handlers; } });
+var errors_js_1 = require("./types/errors.js");
+Object.defineProperty(exports, "RestateError", { enumerable: true, get: function () { return errors_js_1.RestateError; } });
+Object.defineProperty(exports, "TerminalError", { enumerable: true, get: function () { return errors_js_1.TerminalError; } });
+Object.defineProperty(exports, "TimeoutError", { enumerable: true, get: function () { return errors_js_1.TimeoutError; } });
+//# sourceMappingURL=common_api.js.map

package/dist/cjs/src/common_api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"common_api.js","sourceRoot":"","sources":["../../../src/common_api.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAEH,2CAQsB;AAJpB,gHAAA,kBAAkB,OAAA;AAOpB,yCAAqE;AAA5D,iGAAA,OAAO,OAAA;AAAE,gGAAA,MAAM,OAAA;AAAE,kGAAA,QAAQ,OAAA;AAAE,kGAAA,QAAQ,OAAA;AAS5C,+CAA8E;AAArE,yGAAA,YAAY,OAAA;AAAE,0GAAA,aAAa,OAAA;AAAE,yGAAA,YAAY,OAAA"}

package/dist/{src → cjs/src}/connection/buffered_connection.d.ts

@@ -1,8 +1,7 @@
 /// <reference types="node" />
-import { Message } from "../types/types";
-import { Connection } from "./connection";
-import { Buffer } from "node:buffer";
-export declare class BufferedConnection implements Connection {
+import type { Message } from "../types/types.js";
+import type { Buffer } from "node:buffer";
+export declare class BufferedConnection {
     private readonly flushFn;
     private queue;
     private flushing;

package/dist/cjs/src/connection/buffered_connection.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"buffered_connection.d.ts","sourceRoot":"","sources":["../../../../src/connection/buffered_connection.ts"],"names":[],"mappings":";AACA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAE1C,qBAAa,kBAAkB;IAIjB,OAAO,CAAC,QAAQ,CAAC,OAAO;IAHpC,OAAO,CAAC,KAAK,CAAiB;IAC9B,OAAO,CAAC,QAAQ,CAAoC;gBAEvB,OAAO,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC;IAEvE,IAAI,CAAC,GAAG,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAajC,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;IAKpB,OAAO,CAAC,aAAa;YASP,KAAK;CASpB"}

package/dist/{src → cjs/src}/connection/buffered_connection.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.BufferedConnection = void 0;
-const encoder_1 = require("../io/encoder");
+const encoder_js_1 = require("../io/encoder.js");
 class BufferedConnection {
     flushFn;
     queue = [];
@@ -35,7 +35,7 @@ class BufferedConnection {
         if (this.queue.length === 0) {
             return Promise.resolve();
         }
-        const buffer = (0, encoder_1.encodeMessages)(this.queue);
+        const buffer = (0, encoder_js_1.encodeMessages)(this.queue);
         this.queue = [];
         return this.flushFn(buffer);
     }

package/dist/cjs/src/connection/buffered_connection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"buffered_connection.js","sourceRoot":"","sources":["../../../../src/connection/buffered_connection.ts"],"names":[],"mappings":";;;AAAA,iDAAkD;AAIlD,MAAa,kBAAkB;IAIA;IAHrB,KAAK,GAAc,EAAE,CAAC;IACtB,QAAQ,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;IAEpD,YAA6B,OAA0C;QAA1C,YAAO,GAAP,OAAO,CAAmC;IAAG,CAAC;IAE3E,IAAI,CAAC,GAAY;QACf,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,GAAG,KAAK,CAAC,EAAE,CAAC;YACd,2DAA2D;YAC3D,kDAAkD;YAClD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;QACjE,CAAC;QACD,gDAAgD;QAChD,mFAAmF;QACnF,+CAA+C;QAC/C,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,GAAG;QACD,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAEO,aAAa;QACnB,mEAAmE;QACnE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,CACrC,YAAY,CAAC,GAAG,EAAE;YAChB,IAAI,CAAC,KAAK,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAC3C,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,KAAK;QACjB,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;QAC3B,CAAC;QACD,MAAM,MAAM,GAAG,IAAA,2BAAc,EAAC,IAAI,CAAC,KAAK,CAAW,CAAC;QACpD,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;QAEhB,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC;CACF;AA1CD,gDA0CC"}

package/dist/{src → cjs/src}/connection/connection.d.ts

@@ -1,4 +1,4 @@
-import { Message } from "../types/types";
+import type { Message } from "../types/types.js";
 /**
  * A connection from the service/SDK to Restate.
  * Accepts messages to be sent and committed to the journal.
@@ -6,6 +6,7 @@ import { Message } from "../types/types";
 export interface Connection {
     send(msg: Message): Promise<void>;
     end(): Promise<void>;
+    headers(): ReadonlyMap<string, string | string[] | undefined>;
 }
 /**
  * A consumer of a message stream from Restate.

package/dist/cjs/src/connection/connection.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"connection.d.ts","sourceRoot":"","sources":["../../../../src/connection/connection.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAEjD;;;GAGG;AACH,MAAM,WAAW,UAAU;IACzB,IAAI,CAAC,GAAG,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAElC,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAErB,OAAO,IAAI,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC,CAAC;CAC/D;AAED;;;GAGG;AACH,MAAM,WAAW,qBAAqB;IACpC,aAAa,CAAC,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC;IAEnC,iBAAiB,CAAC,CAAC,EAAE,KAAK,GAAG,IAAI,CAAC;IAElC,iBAAiB,IAAI,IAAI,CAAC;CAC3B"}

package/dist/cjs/src/connection/connection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connection.js","sourceRoot":"","sources":["../../../../src/connection/connection.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG"}

package/dist/{src → cjs/src}/connection/http_connection.d.ts

@@ -1,7 +1,9 @@
 /// <reference types="node" />
-import stream from "node:stream";
-import { Connection, RestateStreamConsumer } from "./connection";
-import { Message } from "../types/types";
+/// <reference types="node" />
+import type stream from "node:stream";
+import type { Connection, RestateStreamConsumer } from "./connection.js";
+import type { Message } from "../types/types.js";
+import type { Http2ServerRequest, IncomingHttpHeaders } from "node:http2";
 /**
  * A duplex stream with Restate Messages over HTTP2.
  *
@@ -25,18 +27,20 @@ import { Message } from "../types/types";
  *     notifications for cleanly closed input (to trigger suspension).
  */
 export declare class RestateHttp2Connection implements Connection {
+    private readonly attemptHeaders;
     private readonly rawStream;
     /**
      * create a RestateDuplex stream from an http2 (duplex) stream.
      */
-    static from(http2stream: stream.Duplex): RestateHttp2Connection;
+    static from(request: Http2ServerRequest, http2stream: stream.Duplex): RestateHttp2Connection;
     private readonly sdkInput;
     private currentConsumer;
     private inputBuffer;
     private consumerError?;
     private consumerInputClosed;
     private outputBuffer;
-    constructor(rawStream: stream.Duplex);
+    constructor(attemptHeaders: IncomingHttpHeaders, rawStream: stream.Duplex);
+    headers(): ReadonlyMap<string, string | string[] | undefined>;
     /**
      * Pipes the messages from this connection to the given consumer. The consumer
      * will also receive error and stream closing notifications.

package/dist/cjs/src/connection/http_connection.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"http_connection.d.ts","sourceRoot":"","sources":["../../../../src/connection/http_connection.ts"],"names":[],"mappings":";;AAWA,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AAEtC,OAAO,KAAK,EAAE,UAAU,EAAE,qBAAqB,EAAE,MAAM,iBAAiB,CAAC;AACzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAIjD,OAAO,KAAK,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAK1E;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,qBAAa,sBAAuB,YAAW,UAAU;IAyBrD,OAAO,CAAC,QAAQ,CAAC,cAAc;IAC/B,OAAO,CAAC,QAAQ,CAAC,SAAS;IAzB5B;;OAEG;WACW,IAAI,CAChB,OAAO,EAAE,kBAAkB,EAC3B,WAAW,EAAE,MAAM,CAAC,MAAM,GACzB,sBAAsB;IAOzB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAkB;IAG3C,OAAO,CAAC,eAAe,CAAsC;IAC7D,OAAO,CAAC,WAAW,CAAiB;IACpC,OAAO,CAAC,aAAa,CAAC,CAAQ;IAC9B,OAAO,CAAC,mBAAmB,CAAS;IAEpC,OAAO,CAAC,YAAY,CAAqB;gBAGtB,cAAc,EAAE,mBAAmB,EACnC,SAAS,EAAE,MAAM,CAAC,MAAM;IA4E3C,OAAO,IAAI,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC;IAQ7D;;;;;;;;OAQG;IACI,cAAc,CAAC,QAAQ,EAAE,qBAAqB,GAAG,IAAI;IA+B5D;;OAEG;IACI,qBAAqB,IAAI,IAAI;IAQpC;;;;;;;;;;;;OAYG;IACI,IAAI,CAAC,GAAG,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAIxC;;OAEG;IACU,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAYlC"}

package/dist/{src → cjs/src}/connection/http_connection.js

@@ -11,10 +11,10 @@
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.RestateHttp2Connection = void 0;
-const decoder_1 = require("../io/decoder");
-const logger_1 = require("../logger");
+const decoder_js_1 = require("../io/decoder.js");
+const logger_js_1 = require("../logger.js");
 const promises_1 = require("node:stream/promises");
-const buffered_connection_1 = require("./buffered_connection");
+const buffered_connection_js_1 = require("./buffered_connection.js");
 // utility promise, for cases where we want to save allocation of an extra promise
 const RESOLVED = Promise.resolve();
 /**
@@ -40,12 +40,13 @@ const RESOLVED = Promise.resolve();
  *     notifications for cleanly closed input (to trigger suspension).
  */
 class RestateHttp2Connection {
+    attemptHeaders;
     rawStream;
     /**
      * create a RestateDuplex stream from an http2 (duplex) stream.
      */
-    static from(http2stream) {
-        return new RestateHttp2Connection(http2stream);
+    static from(request, http2stream) {
+        return new RestateHttp2Connection(request.headers, http2stream);
     }
     // --------------------------------------------------------------------------
     // input as decoded messages
@@ -56,10 +57,11 @@ class RestateHttp2Connection {
     consumerError;
     consumerInputClosed = false;
     outputBuffer;
-    constructor(rawStream) {
+    constructor(attemptHeaders, rawStream) {
+        this.attemptHeaders = attemptHeaders;
         this.rawStream = rawStream;
-        this.sdkInput = rawStream.pipe((0, decoder_1.streamDecoder)());
-        this.outputBuffer = new buffered_connection_1.BufferedConnection((buffer) => {
+        this.sdkInput = rawStream.pipe((0, decoder_js_1.streamDecoder)());
+        this.outputBuffer = new buffered_connection_js_1.BufferedConnection((buffer) => {
             const hasMoreCapacity = rawStream.write(buffer);
             if (hasMoreCapacity) {
                 return RESOLVED;
@@ -102,17 +104,17 @@ class RestateHttp2Connection {
         };
         // those two event types should cover all types of connection losses
         rawStream.on("aborted", () => {
-            logger_1.rlog.error("Connection to Restate was lost");
+            logger_js_1.rlog.error("Connection to Restate was lost");
             errorHandler(new Error("Connection to Restate was lost"));
         });
         // this is both the raw http2 stream and the output SDK->Restate
         rawStream.on("error", (e) => {
-            logger_1.rlog.error("Error in http2 stream to Restate: " + (e.stack ?? e.message));
+            logger_js_1.rlog.error("Error in http2 stream to Restate: " + (e.stack ?? e.message));
             errorHandler(e);
         });
         // these events notify of errors in the decoding pipeline
         this.sdkInput.on("error", (e) => {
-            logger_1.rlog.error("Error in input stream (Restate to Service): " + (e.stack ?? e.message));
+            logger_js_1.rlog.error("Error in input stream (Restate to Service): " + (e.stack ?? e.message));
             errorHandler(e);
         });
         // see if streams get torn down before they end cleanly
@@ -122,6 +124,9 @@ class RestateHttp2Connection {
             }
         });
     }
+    headers() {
+        return new Map(Object.entries(this.attemptHeaders));
+    }
     // --------------------------------------------------------------------------
     //  input stream handling
     // --------------------------------------------------------------------------

package/dist/cjs/src/connection/http_connection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"http_connection.js","sourceRoot":"","sources":["../../../../src/connection/http_connection.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAGH,iDAAiD;AAGjD,4CAAoC;AACpC,mDAAgD;AAChD,qEAA8D;AAG9D,kFAAkF;AAClF,MAAM,QAAQ,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;AAElD;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAa,sBAAsB;IAyBd;IACA;IAzBnB;;OAEG;IACI,MAAM,CAAC,IAAI,CAChB,OAA2B,EAC3B,WAA0B;QAE1B,OAAO,IAAI,sBAAsB,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAClE,CAAC;IAED,6EAA6E;IAE7E,4BAA4B;IACX,QAAQ,CAAkB;IAE3C,oBAAoB;IACZ,eAAe,GAAiC,IAAI,CAAC;IACrD,WAAW,GAAc,EAAE,CAAC;IAC5B,aAAa,CAAS;IACtB,mBAAmB,GAAG,KAAK,CAAC;IAE5B,YAAY,CAAqB;IAEzC,YACmB,cAAmC,EACnC,SAAwB;QADxB,mBAAc,GAAd,cAAc,CAAqB;QACnC,cAAS,GAAT,SAAS,CAAe;QAEzC,IAAI,CAAC,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,IAAA,0BAAa,GAAE,CAAC,CAAC;QAEhD,IAAI,CAAC,YAAY,GAAG,IAAI,2CAAkB,CAAC,CAAC,MAAM,EAAE,EAAE;YACpD,MAAM,eAAe,GAAG,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,eAAe,EAAE,CAAC;gBACpB,OAAO,QAAQ,CAAC;YAClB,CAAC;iBAAM,CAAC;gBACN,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YACpE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,gCAAgC;QAChC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,CAAU,EAAE,EAAE;YACtC,wEAAwE;YACxE,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;gBACzB,IAAI,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;oBAC1C,IAAI,CAAC,qBAAqB,EAAE,CAAC;gBAC/B,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,oCAAoC;QACpC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;YAC3B,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC;YAChC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;gBACzB,IAAI,CAAC,eAAe,CAAC,iBAAiB,EAAE,CAAC;YAC3C,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,oCAAoC;QACpC,mCAAmC;QAEnC,gEAAgE;QAChE,MAAM,YAAY,GAAG,CAAC,CAAQ,EAAE,EAAE;YAChC,iDAAiD;YACjD,IAAI,IAAI,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;gBACrC,OAAO;YACT,CAAC;YACD,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC;YACvB,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;gBACzB,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC;YAC5C,CAAC;QACH,CAAC,CAAC;QAEF,oEAAoE;QACpE,SAAS,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;YAC3B,gBAAI,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAC;YAC7C,YAAY,CAAC,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,gEAAgE;QAChE,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,CAAQ,EAAE,EAAE;YACjC,gBAAI,CAAC,KAAK,CAAC,oCAAoC,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;YAC1E,YAAY,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,yDAAyD;QACzD,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,CAAQ,EAAE,EAAE;YACrC,gBAAI,CAAC,KAAK,CACR,8CAA8C,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,CACxE,CAAC;YACF,YAAY,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,uDAAuD;QACvD,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAC7B,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC9B,YAAY,CAAC,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC,CAAC;YAC7D,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,OAAO;QACL,OAAO,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;IACtD,CAAC;IAED,6EAA6E;IAC7E,yBAAyB;IACzB,6EAA6E;IAE7E;;;;;;;;OAQG;IACI,cAAc,CAAC,QAA+B;QACnD,IAAI,IAAI,CAAC,eAAe,KAAK,IAAI,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;QAClD,CAAC;QAED,IAAI,CAAC,eAAe,GAAG,QAAQ,CAAC;QAEhC,qCAAqC;QACrC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,QAAQ,CAAC,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACjD,CAAC;QACD,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC7B,QAAQ,CAAC,iBAAiB,EAAE,CAAC;QAC/B,CAAC;QAED,2FAA2F;QAC3F,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC;QAC/B,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,IAAI,CAAC,GAAG,CAAC,CAAC;YACV,OAAO,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;gBACxB,MAAM,IAAI,GAAG,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC9C,CAAC,EAAE,CAAC;gBACJ,IAAI,IAAI,EAAE,CAAC;oBACT,IAAI,CAAC,qBAAqB,EAAE,CAAC;oBAC7B,MAAM;gBACR,CAAC;YACH,CAAC;YACD,IAAI,CAAC,WAAW,GAAG,CAAC,KAAK,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;IAED;;OAEG;IACI,qBAAqB;QAC1B,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;IAC9B,CAAC;IAED,6EAA6E;IAC7E,0BAA0B;IAC1B,6EAA6E;IAE7E;;;;;;;;;;;;OAYG;IACI,IAAI,CAAC,GAAY;QACtB,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,GAAG;QACd,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,CAAC;QAE9B,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC;QAErB,MAAM,OAAO,GAAG;YACd,KAAK,EAAE,IAAI;YACX,OAAO,EAAE,IAAI;SACd,CAAC;QAEF,MAAM,IAAA,mBAAQ,EAAC,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC1C,CAAC;CACF;AAjMD,wDAiMC"}

package/dist/{src → cjs/src}/connection/lambda_connection.d.ts

@@ -1,13 +1,15 @@
 /// <reference types="node" />
-import { Connection } from "./connection";
-import { Message } from "../types/types";
+import type { Connection } from "./connection.js";
+import type { Message } from "../types/types.js";
 import { Buffer } from "node:buffer";
 export declare class LambdaConnection implements Connection {
+    private readonly attemptHeaders;
     private suspendedOrCompleted;
     private outputBuffer;
     private readonly completionPromise;
     private resolveOnCompleted;
-    constructor(suspendedOrCompleted?: boolean);
+    constructor(attemptHeaders: Record<string, string | string[] | undefined>, suspendedOrCompleted?: boolean);
+    headers(): ReadonlyMap<string, string | string[] | undefined>;
     send(msg: Message): Promise<void>;
     getResult(): Promise<Buffer>;
     end(): Promise<void>;

package/dist/cjs/src/connection/lambda_connection.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"lambda_connection.d.ts","sourceRoot":"","sources":["../../../../src/connection/lambda_connection.ts"],"names":[],"mappings":";AAWA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAOlD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAEjD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAIrC,qBAAa,gBAAiB,YAAW,UAAU;IAS/C,OAAO,CAAC,QAAQ,CAAC,cAAc;IAI/B,OAAO,CAAC,oBAAoB;IAX9B,OAAO,CAAC,YAAY,CAA2B;IAG/C,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAkB;IACpD,OAAO,CAAC,kBAAkB,CAAiD;gBAGxD,cAAc,EAAE,MAAM,CACrC,MAAM,EACN,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAC9B,EACO,oBAAoB,UAAQ;IAQtC,OAAO,IAAI,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC;IAK7D,IAAI,CAAC,GAAG,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAiBjC,SAAS,IAAI,OAAO,CAAC,MAAM,CAAC;IAI5B,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAQrB"}

package/dist/{src → cjs/src}/connection/lambda_connection.js

@@ -11,34 +11,39 @@
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.LambdaConnection = void 0;
-const encoder_1 = require("../io/encoder");
-const protocol_1 = require("../types/protocol");
-const logger_1 = require("../logger");
+const encoder_js_1 = require("../io/encoder.js");
+const protocol_js_1 = require("../types/protocol.js");
+const logger_js_1 = require("../logger.js");
 const node_buffer_1 = require("node:buffer");
 const RESOLVED = Promise.resolve();
 class LambdaConnection {
+    attemptHeaders;
     suspendedOrCompleted;
     // Empty buffer to store journal output messages
     outputBuffer = node_buffer_1.Buffer.alloc(0);
     // Callback to resolve the invocation promise of the Lambda handler when the response is ready
     completionPromise;
     resolveOnCompleted;
-    constructor(suspendedOrCompleted = false) {
+    constructor(attemptHeaders, suspendedOrCompleted = false) {
+        this.attemptHeaders = attemptHeaders;
         this.suspendedOrCompleted = suspendedOrCompleted;
         // Promise that signals when the invocation is over, to then flush the messages
         this.completionPromise = new Promise((resolve) => {
             this.resolveOnCompleted = resolve;
         });
     }
+    headers() {
+        return new Map(Object.entries(this.attemptHeaders));
+    }
     // Send a message back to the runtime
     send(msg) {
         // Add the header and the body to buffer and add to the output buffer
-        const msgBuffer = (0, encoder_1.encodeMessage)(msg);
+        const msgBuffer = (0, encoder_js_1.encodeMessage)(msg);
         this.outputBuffer = node_buffer_1.Buffer.concat([this.outputBuffer, msgBuffer]);
         // An output message, suspension message or error message is the end of a Lambda invocation
-        if (msg.messageType === protocol_1.OUTPUT_ENTRY_MESSAGE_TYPE ||
-            msg.messageType === protocol_1.SUSPENSION_MESSAGE_TYPE ||
-            msg.messageType === protocol_1.ERROR_MESSAGE_TYPE) {
+        if (msg.messageType === protocol_js_1.OUTPUT_ENTRY_MESSAGE_TYPE ||
+            msg.messageType === protocol_js_1.SUSPENSION_MESSAGE_TYPE ||
+            msg.messageType === protocol_js_1.ERROR_MESSAGE_TYPE) {
             this.suspendedOrCompleted = true;
         }
         return RESOLVED;
@@ -48,7 +53,7 @@ class LambdaConnection {
     }
     end() {
         if (this.suspendedOrCompleted) {
-            logger_1.rlog.debug("Flushing output buffer...");
+            logger_js_1.rlog.debug("Flushing output buffer...");
             this.resolveOnCompleted(this.outputBuffer);
         }
         this.outputBuffer = node_buffer_1.Buffer.alloc(0);

package/dist/cjs/src/connection/lambda_connection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"lambda_connection.js","sourceRoot":"","sources":["../../../../src/connection/lambda_connection.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAGH,iDAAiD;AACjD,sDAI8B;AAE9B,4CAAoC;AACpC,6CAAqC;AAErC,MAAM,QAAQ,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;AAElD,MAAa,gBAAgB;IASR;IAIT;IAZV,gDAAgD;IACxC,YAAY,GAAW,oBAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAE/C,8FAA8F;IAC7E,iBAAiB,CAAkB;IAC5C,kBAAkB,CAAiD;IAE3E,YACmB,cAGhB,EACO,uBAAuB,KAAK;QAJnB,mBAAc,GAAd,cAAc,CAG9B;QACO,yBAAoB,GAApB,oBAAoB,CAAQ;QAEpC,+EAA+E;QAC/E,IAAI,CAAC,iBAAiB,GAAG,IAAI,OAAO,CAAS,CAAC,OAAO,EAAE,EAAE;YACvD,IAAI,CAAC,kBAAkB,GAAG,OAAO,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC;IAED,OAAO;QACL,OAAO,IAAI,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;IACtD,CAAC;IAED,qCAAqC;IACrC,IAAI,CAAC,GAAY;QACf,qEAAqE;QACrE,MAAM,SAAS,GAAG,IAAA,0BAAa,EAAC,GAAG,CAAC,CAAC;QACrC,IAAI,CAAC,YAAY,GAAG,oBAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC,CAAC;QAElE,2FAA2F;QAC3F,IACE,GAAG,CAAC,WAAW,KAAK,uCAAyB;YAC7C,GAAG,CAAC,WAAW,KAAK,qCAAuB;YAC3C,GAAG,CAAC,WAAW,KAAK,gCAAkB,EACtC,CAAC;YACD,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC;QACnC,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED,GAAG;QACD,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9B,gBAAI,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;YACxC,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,CAAC,YAAY,GAAG,oBAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACpC,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF;AAvDD,4CAuDC"}