@restatedev/restate-sdk 0.0.0-SNAPSHOT-20231025205407

package/dist/invocation.js

@@ -0,0 +1,138 @@
+"use strict";
+/*
+ * Copyright (c) 2023 - 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.Invocation = exports.InvocationBuilder = void 0;
+const utils_1 = require("./utils/utils");
+const protocol_1 = require("./types/protocol");
+const local_state_store_1 = require("./local_state_store");
+const errors_1 = require("./types/errors");
+var State;
+(function (State) {
+    State[State["ExpectingStart"] = 0] = "ExpectingStart";
+    State[State["ExpectingInput"] = 1] = "ExpectingInput";
+    State[State["ExpectingFurtherReplay"] = 2] = "ExpectingFurtherReplay";
+    State[State["Complete"] = 3] = "Complete";
+})(State || (State = {}));
+class InvocationBuilder {
+    method;
+    complete = new utils_1.CompletablePromise();
+    state = State.ExpectingStart;
+    runtimeReplayIndex = 0;
+    replayEntries = new Map();
+    id = undefined;
+    debugId = undefined;
+    invocationValue = undefined;
+    nbEntriesToReplay = undefined;
+    localStateStore;
+    constructor(method) {
+        this.method = method;
+    }
+    handleMessage(m) {
+        try {
+            switch (this.state) {
+                case State.ExpectingStart:
+                    checkState(State.ExpectingStart, protocol_1.START_MESSAGE_TYPE, m);
+                    this.handleStartMessage(m.message);
+                    this.state = State.ExpectingInput;
+                    return false;
+                case State.ExpectingInput:
+                    checkState(State.ExpectingInput, protocol_1.POLL_INPUT_STREAM_ENTRY_MESSAGE_TYPE, m);
+                    this.addReplayEntry(m);
+                    break;
+                case State.ExpectingFurtherReplay:
+                    this.addReplayEntry(m);
+                    break;
+                case State.Complete:
+                    throw new Error(`Journal builder is getting a message after the journal was complete. entries-to-replay: ${this.nbEntriesToReplay}, message: ${(0, utils_1.printMessageAsJson)(m)}`);
+            }
+            this.state =
+                this.replayEntries.size === this.nbEntriesToReplay
+                    ? State.Complete
+                    : State.ExpectingFurtherReplay;
+            if (this.state === State.Complete) {
+                this.complete.resolve();
+                return true;
+            }
+            return false;
+        }
+        catch (e) {
+            const error = (0, errors_1.ensureError)(e);
+            this.complete.reject(error);
+            return true; // we want no further messages
+        }
+    }
+    handleStreamError(e) {
+        this.complete.reject(e);
+    }
+    handleInputClosed() {
+        this.complete.reject(new Error("Input closed before journal is complete"));
+    }
+    completion() {
+        return this.complete.promise;
+    }
+    handleStartMessage(m) {
+        this.nbEntriesToReplay = m.knownEntries;
+        this.id = m.id;
+        this.debugId = m.debugId;
+        this.localStateStore = new local_state_store_1.LocalStateStore(m.partialState, m.stateMap);
+        return this;
+    }
+    addReplayEntry(m) {
+        if (m.messageType === protocol_1.POLL_INPUT_STREAM_ENTRY_MESSAGE_TYPE) {
+            this.invocationValue = m.message.value;
+        }
+        // Will be retrieved when the user code reaches this point
+        this.replayEntries.set(this.runtimeReplayIndex, m);
+        this.incrementRuntimeReplayIndex();
+        return this;
+    }
+    incrementRuntimeReplayIndex() {
+        this.runtimeReplayIndex++;
+    }
+    isComplete() {
+        return this.state === State.Complete;
+    }
+    build() {
+        if (!this.isComplete()) {
+            throw new Error(`Cannot build invocation. Not all data present: ${JSON.stringify(this)}`);
+        }
+        return new Invocation(this.method, this.id, this.debugId, this.nbEntriesToReplay, this.replayEntries, this.invocationValue, this.localStateStore);
+    }
+}
+exports.InvocationBuilder = InvocationBuilder;
+class Invocation {
+    method;
+    id;
+    debugId;
+    nbEntriesToReplay;
+    replayEntries;
+    invocationValue;
+    localStateStore;
+    logPrefix;
+    constructor(method, id, debugId, nbEntriesToReplay, replayEntries, invocationValue, localStateStore) {
+        this.method = method;
+        this.id = id;
+        this.debugId = debugId;
+        this.nbEntriesToReplay = nbEntriesToReplay;
+        this.replayEntries = replayEntries;
+        this.invocationValue = invocationValue;
+        this.localStateStore = localStateStore;
+        this.logPrefix = `[${(0, utils_1.makeFqServiceName)(this.method.packge, this.method.service)}/${this.method.method.name}] [${this.debugId}]`;
+    }
+}
+exports.Invocation = Invocation;
+function checkState(state, expected, m) {
+    if (m.messageType !== expected) {
+        throw new Error(`Unexpected message in state ${state}: ${(0, utils_1.printMessageAsJson)(m)}`);
+    }
+}
+//# sourceMappingURL=invocation.js.map

package/dist/invocation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"invocation.js","sourceRoot":"","sources":["../src/invocation.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAUH,yCAIuB;AACvB,+CAG0B;AAE1B,2DAAsD;AACtD,2CAA6C;AAE7C,IAAK,KAKJ;AALD,WAAK,KAAK;IACR,qDAAkB,CAAA;IAClB,qDAAkB,CAAA;IAClB,qEAA0B,CAAA;IAC1B,yCAAY,CAAA;AACd,CAAC,EALI,KAAK,KAAL,KAAK,QAKT;AAED,MAAa,iBAAiB;IAaC;IAZZ,QAAQ,GAAG,IAAI,0BAAkB,EAAQ,CAAC;IAEnD,KAAK,GAAU,KAAK,CAAC,cAAc,CAAC;IAEpC,kBAAkB,GAAG,CAAC,CAAC;IACvB,aAAa,GAAG,IAAI,GAAG,EAAmB,CAAC;IAC3C,EAAE,GAAY,SAAS,CAAC;IACxB,OAAO,GAAY,SAAS,CAAC;IAC7B,eAAe,GAAY,SAAS,CAAC;IACrC,iBAAiB,GAAY,SAAS,CAAC;IACvC,eAAe,CAAmB;IAE1C,YAA6B,MAAqC;QAArC,WAAM,GAAN,MAAM,CAA+B;IAAG,CAAC;IAE/D,aAAa,CAAC,CAAU;QAC7B,IAAI;YACF,QAAQ,IAAI,CAAC,KAAK,EAAE;gBAClB,KAAK,KAAK,CAAC,cAAc;oBACvB,UAAU,CAAC,KAAK,CAAC,cAAc,EAAE,6BAAkB,EAAE,CAAC,CAAC,CAAC;oBACxD,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,OAAuB,CAAC,CAAC;oBACnD,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,cAAc,CAAC;oBAClC,OAAO,KAAK,CAAC;gBAEf,KAAK,KAAK,CAAC,cAAc;oBACvB,UAAU,CACR,KAAK,CAAC,cAAc,EACpB,+CAAoC,EACpC,CAAC,CACF,CAAC;oBACF,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;oBACvB,MAAM;gBAER,KAAK,KAAK,CAAC,sBAAsB;oBAC/B,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;oBACvB,MAAM;gBAER,KAAK,KAAK,CAAC,QAAQ;oBACjB,MAAM,IAAI,KAAK,CACb,2FACE,IAAI,CAAC,iBACP,cAAc,IAAA,0BAAkB,EAAC,CAAC,CAAC,EAAE,CACtC,CAAC;aACL;YAED,IAAI,CAAC,KAAK;gBACR,IAAI,CAAC,aAAa,CAAC,IAAI,KAAK,IAAI,CAAC,iBAAiB;oBAChD,CAAC,CAAC,KAAK,CAAC,QAAQ;oBAChB,CAAC,CAAC,KAAK,CAAC,sBAAsB,CAAC;YAEnC,IAAI,IAAI,CAAC,KAAK,KAAK,KAAK,CAAC,QAAQ,EAAE;gBACjC,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACxB,OAAO,IAAI,CAAC;aACb;YAED,OAAO,KAAK,CAAC;SACd;QAAC,OAAO,CAAC,EAAE;YACV,MAAM,KAAK,GAAG,IAAA,oBAAW,EAAC,CAAC,CAAC,CAAC;YAC7B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5B,OAAO,IAAI,CAAC,CAAC,8BAA8B;SAC5C;IACH,CAAC;IAEM,iBAAiB,CAAC,CAAQ;QAC/B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;IAC1B,CAAC;IACM,iBAAiB;QACtB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC,CAAC;IAC7E,CAAC;IAEM,UAAU;QACf,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;IAC/B,CAAC;IAEO,kBAAkB,CAAC,CAAe;QACxC,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC,YAAY,CAAC;QACxC,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,EAAE,CAAC;QACf,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;QACzB,IAAI,CAAC,eAAe,GAAG,IAAI,mCAAe,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC;QACvE,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,cAAc,CAAC,CAAU;QAC/B,IAAI,CAAC,CAAC,WAAW,KAAK,+CAAoC,EAAE;YAC1D,IAAI,CAAC,eAAe,GAAI,CAAC,CAAC,OAAuC,CAAC,KAAK,CAAC;SACzE;QAED,0DAA0D;QAC1D,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC;QACnD,IAAI,CAAC,2BAA2B,EAAE,CAAC;QACnC,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,2BAA2B;QACjC,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;IAEM,UAAU;QACf,OAAO,IAAI,CAAC,KAAK,KAAK,KAAK,CAAC,QAAQ,CAAC;IACvC,CAAC;IAEM,KAAK;QACV,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE;YACtB,MAAM,IAAI,KAAK,CACb,kDAAkD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CACzE,CAAC;SACH;QACD,OAAO,IAAI,UAAU,CACnB,IAAI,CAAC,MAAO,EACZ,IAAI,CAAC,EAAG,EACR,IAAI,CAAC,OAAQ,EACb,IAAI,CAAC,iBAAkB,EACvB,IAAI,CAAC,aAAc,EACnB,IAAI,CAAC,eAAgB,EACrB,IAAI,CAAC,eAAgB,CACtB,CAAC;IACJ,CAAC;CACF;AArHD,8CAqHC;AAED,MAAa,UAAU;IAGH;IACA;IACA;IACA;IACA;IACA;IACA;IARF,SAAS,CAAC;IAC1B,YACkB,MAAqC,EACrC,EAAU,EACV,OAAe,EACf,iBAAyB,EACzB,aAAmC,EACnC,eAAuB,EACvB,eAAgC;QANhC,WAAM,GAAN,MAAM,CAA+B;QACrC,OAAE,GAAF,EAAE,CAAQ;QACV,YAAO,GAAP,OAAO,CAAQ;QACf,sBAAiB,GAAjB,iBAAiB,CAAQ;QACzB,kBAAa,GAAb,aAAa,CAAsB;QACnC,oBAAe,GAAf,eAAe,CAAQ;QACvB,oBAAe,GAAf,eAAe,CAAiB;QAEhD,IAAI,CAAC,SAAS,GAAG,IAAI,IAAA,yBAAiB,EACpC,IAAI,CAAC,MAAM,CAAC,MAAM,EAClB,IAAI,CAAC,MAAM,CAAC,OAAO,CACpB,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,MAAM,IAAI,CAAC,OAAO,GAAG,CAAC;IACpD,CAAC;CACF;AAhBD,gCAgBC;AAED,SAAS,UAAU,CAAC,KAAY,EAAE,QAAgB,EAAE,CAAU;IAC5D,IAAI,CAAC,CAAC,WAAW,KAAK,QAAQ,EAAE;QAC9B,MAAM,IAAI,KAAK,CACb,+BAA+B,KAAK,KAAK,IAAA,0BAAkB,EAAC,CAAC,CAAC,EAAE,CACjE,CAAC;KACH;AACH,CAAC"}

package/dist/io/decoder.d.ts

@@ -0,0 +1,8 @@
+/// <reference types="node" />
+/// <reference types="node" />
+import stream from "stream";
+import { Message } from "../types/types";
+export declare function streamDecoder(): stream.Transform;
+export declare function decodeLambdaBody(msgBase64: string): Message[];
+export declare function decodeMessagesBuffer(buffer: Buffer): Message[];
+//# sourceMappingURL=decoder.d.ts.map

package/dist/io/decoder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"decoder.d.ts","sourceRoot":"","sources":["../../src/io/decoder.ts"],"names":[],"mappings":";;AAsBA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAU,OAAO,EAAE,MAAM,gBAAgB,CAAC;AA4EjD,wBAAgB,aAAa,IAAI,MAAM,CAAC,SAAS,CAiBhD;AAMD,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,EAAE,CAS7D;AAED,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,EAAE,CAsB9D"}

package/dist/io/decoder.js

@@ -0,0 +1,133 @@
+"use strict";
+/*
+ * Copyright (c) 2023 - 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
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.decodeMessagesBuffer = exports.decodeLambdaBody = exports.streamDecoder = void 0;
+// This is a NodeJs stream transformer. It is used to convert a chunked stream of bytes to
+// a stream of JavaScript objects of the form { header: .. , message: ..} where:
+// * header has some information about the frame like, the message type, and some flags.
+// * message is the Protobuf decoded message.
+//
+// To use this one would need to do the following:
+//
+// let decodedStream = stream.pipe(streamDecoder());
+//
+// at this point the decodedStream is a high level stream of objects {header, message}
+const stream_1 = __importDefault(require("stream"));
+const protocol_1 = require("../types/protocol");
+const types_1 = require("../types/types");
+const assert_1 = __importDefault(require("assert"));
+const errors_1 = require("../types/errors");
+const WAITING_FOR_HEADER = 0;
+const WAITING_FOR_BODY = 1;
+function initalDecoderState(buf) {
+    return {
+        state: WAITING_FOR_HEADER,
+        header: undefined,
+        buf,
+    };
+}
+function appendBufferToDecoder(state, chunk) {
+    state.buf = Buffer.concat([state.buf, chunk]);
+}
+function decodeMessages(decoderState, out) {
+    let buf = decoderState.buf;
+    while (buf.length > 0 || decoderState.state === WAITING_FOR_BODY) {
+        switch (decoderState.state) {
+            case WAITING_FOR_HEADER: {
+                (0, assert_1.default)(decoderState.header === undefined);
+                if (buf.length < 8) {
+                    decoderState.buf = buf;
+                    return decoderState;
+                }
+                const h = buf.readBigUInt64BE();
+                buf = buf.subarray(8);
+                decoderState.header = types_1.Header.fromU64be(h);
+                decoderState.state = WAITING_FOR_BODY;
+                break;
+            }
+            case WAITING_FOR_BODY: {
+                const header = decoderState.header;
+                (0, assert_1.default)(header !== undefined);
+                if (buf.length < header.frameLength) {
+                    decoderState.buf = buf;
+                    return decoderState;
+                }
+                const frame = buf.subarray(0, header.frameLength);
+                buf = buf.subarray(header.frameLength);
+                decoderState.state = WAITING_FOR_HEADER;
+                decoderState.header = undefined;
+                const pbType = protocol_1.PROTOBUF_MESSAGE_BY_TYPE.get(header.messageType);
+                if (pbType === undefined) {
+                    throw new Error("Got unknown message type " + header.messageType);
+                }
+                else {
+                    const message = pbType.decode(frame);
+                    out.push(new types_1.Message(header.messageType, message, header.completedFlag, header.protocolVersion, header.requiresAckFlag));
+                }
+                break;
+            }
+        }
+    }
+    decoderState.buf = buf;
+    return decoderState;
+}
+function streamDecoder() {
+    let decoderState = initalDecoderState(Buffer.alloc(0));
+    return new stream_1.default.Transform({
+        writableObjectMode: true,
+        objectMode: true,
+        transform(chunk, _encoding, cb) {
+            try {
+                appendBufferToDecoder(decoderState, chunk);
+                decoderState = decodeMessages(decoderState, this);
+                cb();
+            }
+            catch (e) {
+                cb((0, errors_1.ensureError)(e), null);
+            }
+        },
+    });
+}
+exports.streamDecoder = streamDecoder;
+// Decodes messages from Lambda requests to an array of headers + protocol messages
+const base64regex = /^([0-9a-zA-Z+/]{4})*(([0-9a-zA-Z+/]{2}==)|([0-9a-zA-Z+/]{3}=))?$/;
+function decodeLambdaBody(msgBase64) {
+    if (!base64regex.test(msgBase64)) {
+        throw new Error("Cannot parse the lambda request body, body was not valid base64 encoded: " +
+            msgBase64);
+    }
+    const buffer = Buffer.from(msgBase64, "base64");
+    return decodeMessagesBuffer(buffer);
+}
+exports.decodeLambdaBody = decodeLambdaBody;
+function decodeMessagesBuffer(buffer) {
+    const decodedEntries = [];
+    let finalState;
+    try {
+        finalState = decodeMessages(initalDecoderState(buffer), decodedEntries);
+    }
+    catch (e) {
+        const err = (0, errors_1.ensureError)(e);
+        throw new Error("Cannot parse the lambda request body, message was not a valid sequence of Restate messages: " +
+            err.message, { cause: err });
+    }
+    if (finalState.buf.length > 0) {
+        throw new Error("Cannot parse the request body: Trailing data (incomplete message) in request body: " +
+            finalState.buf.toString("hex"));
+    }
+    return decodedEntries;
+}
+exports.decodeMessagesBuffer = decodeMessagesBuffer;
+//# sourceMappingURL=decoder.js.map

package/dist/io/decoder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"decoder.js","sourceRoot":"","sources":["../../src/io/decoder.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;;;;AAEH,0FAA0F;AAC1F,gFAAgF;AAChF,wFAAwF;AACxF,6CAA6C;AAC7C,EAAE;AACF,kDAAkD;AAClD,EAAE;AACF,oDAAoD;AACpD,EAAE;AACF,sFAAsF;AAEtF,oDAA4B;AAC5B,gDAA6D;AAC7D,0CAAiD;AACjD,oDAA4B;AAC5B,4CAA8C;AAK9C,MAAM,kBAAkB,GAAG,CAAC,CAAC;AAC7B,MAAM,gBAAgB,GAAG,CAAC,CAAC;AAE3B,SAAS,kBAAkB,CAAC,GAAW;IACrC,OAAO;QACL,KAAK,EAAE,kBAAkB;QACzB,MAAM,EAAE,SAAS;QACjB,GAAG;KACJ,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAmB,EAAE,KAAa;IAC/D,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC;AAChD,CAAC;AAED,SAAS,cAAc,CAAC,YAA0B,EAAE,GAAW;IAC7D,IAAI,GAAG,GAAG,YAAY,CAAC,GAAG,CAAC;IAE3B,OAAO,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,YAAY,CAAC,KAAK,KAAK,gBAAgB,EAAE;QAChE,QAAQ,YAAY,CAAC,KAAK,EAAE;YAC1B,KAAK,kBAAkB,CAAC,CAAC;gBACvB,IAAA,gBAAM,EAAC,YAAY,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;gBAC1C,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE;oBAClB,YAAY,CAAC,GAAG,GAAG,GAAG,CAAC;oBACvB,OAAO,YAAY,CAAC;iBACrB;gBACD,MAAM,CAAC,GAAG,GAAG,CAAC,eAAe,EAAE,CAAC;gBAChC,GAAG,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;gBACtB,YAAY,CAAC,MAAM,GAAG,cAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBAC1C,YAAY,CAAC,KAAK,GAAG,gBAAgB,CAAC;gBACtC,MAAM;aACP;YACD,KAAK,gBAAgB,CAAC,CAAC;gBACrB,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;gBACnC,IAAA,gBAAM,EAAC,MAAM,KAAK,SAAS,CAAC,CAAC;gBAE7B,IAAI,GAAG,CAAC,MAAM,GAAG,MAAM,CAAC,WAAW,EAAE;oBACnC,YAAY,CAAC,GAAG,GAAG,GAAG,CAAC;oBACvB,OAAO,YAAY,CAAC;iBACrB;gBACD,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;gBAClD,GAAG,GAAG,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;gBACvC,YAAY,CAAC,KAAK,GAAG,kBAAkB,CAAC;gBACxC,YAAY,CAAC,MAAM,GAAG,SAAS,CAAC;gBAEhC,MAAM,MAAM,GAAG,mCAAwB,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;gBAChE,IAAI,MAAM,KAAK,SAAS,EAAE;oBACxB,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC;iBACnE;qBAAM;oBACL,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBACrC,GAAG,CAAC,IAAI,CACN,IAAI,eAAO,CACT,MAAM,CAAC,WAAW,EAClB,OAAO,EACP,MAAM,CAAC,aAAa,EACpB,MAAM,CAAC,eAAe,EACtB,MAAM,CAAC,eAAe,CACvB,CACF,CAAC;iBACH;gBACD,MAAM;aACP;SACF;KACF;IAED,YAAY,CAAC,GAAG,GAAG,GAAG,CAAC;IACvB,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,SAAgB,aAAa;IAC3B,IAAI,YAAY,GAAG,kBAAkB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAEvD,OAAO,IAAI,gBAAM,CAAC,SAAS,CAAC;QAC1B,kBAAkB,EAAE,IAAI;QACxB,UAAU,EAAE,IAAI;QAEhB,SAAS,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE;YAC5B,IAAI;gBACF,qBAAqB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;gBAC3C,YAAY,GAAG,cAAc,CAAC,YAAY,EAAE,IAAI,CAAC,CAAC;gBAClD,EAAE,EAAE,CAAC;aACN;YAAC,OAAO,CAAC,EAAE;gBACV,EAAE,CAAC,IAAA,oBAAW,EAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;aAC1B;QACH,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAjBD,sCAiBC;AAED,mFAAmF;AACnF,MAAM,WAAW,GACf,kEAAkE,CAAC;AAErE,SAAgB,gBAAgB,CAAC,SAAiB;IAChD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE;QAChC,MAAM,IAAI,KAAK,CACb,2EAA2E;YACzE,SAAS,CACZ,CAAC;KACH;IACD,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAChD,OAAO,oBAAoB,CAAC,MAAM,CAAC,CAAC;AACtC,CAAC;AATD,4CASC;AAED,SAAgB,oBAAoB,CAAC,MAAc;IACjD,MAAM,cAAc,GAAc,EAAE,CAAC;IACrC,IAAI,UAAU,CAAC;IACf,IAAI;QACF,UAAU,GAAG,cAAc,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,cAAc,CAAC,CAAC;KACzE;IAAC,OAAO,CAAC,EAAE;QACV,MAAM,GAAG,GAAG,IAAA,oBAAW,EAAC,CAAC,CAAC,CAAC;QAC3B,MAAM,IAAI,KAAK,CACb,8FAA8F;YAC5F,GAAG,CAAC,OAAO,EACb,EAAE,KAAK,EAAE,GAAG,EAAE,CACf,CAAC;KACH;IAED,IAAI,UAAU,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE;QAC7B,MAAM,IAAI,KAAK,CACb,qFAAqF;YACnF,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CACjC,CAAC;KACH;IAED,OAAO,cAAc,CAAC;AACxB,CAAC;AAtBD,oDAsBC"}

package/dist/io/encoder.d.ts

@@ -0,0 +1,7 @@
+/// <reference types="node" />
+import stream from "stream";
+import { Message } from "../types/types";
+export declare function streamEncoder(): stream.Transform;
+export declare function encodeMessage(msg: Message): Uint8Array;
+export declare function encodeMessages(messages: Message[]): Uint8Array;
+//# sourceMappingURL=encoder.d.ts.map

package/dist/io/encoder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"encoder.d.ts","sourceRoot":"","sources":["../../src/io/encoder.ts"],"names":[],"mappings":";AAWA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAU,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAEjD,wBAAgB,aAAa,IAAI,MAAM,CAAC,SAAS,CAYhD;AAED,wBAAgB,aAAa,CAAC,GAAG,EAAE,OAAO,GAAG,UAAU,CAEtD;AAED,wBAAgB,cAAc,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,UAAU,CAyB9D"}

package/dist/io/encoder.js

@@ -0,0 +1,57 @@
+"use strict";
+/*
+ * Copyright (c) 2023 - 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
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.encodeMessages = exports.encodeMessage = exports.streamEncoder = void 0;
+const stream_1 = __importDefault(require("stream"));
+const protocol_1 = require("../types/protocol");
+const types_1 = require("../types/types");
+function streamEncoder() {
+    return new stream_1.default.Transform({
+        writableObjectMode: true,
+        objectMode: true,
+        transform(msg, _encoding, cb) {
+            // We do not catch errors here because we want them to be handled at the Connection level,
+            // so we can close the state machine.
+            const result = encodeMessage(msg);
+            cb(null, result);
+        },
+    });
+}
+exports.streamEncoder = streamEncoder;
+function encodeMessage(msg) {
+    return encodeMessages([msg]);
+}
+exports.encodeMessage = encodeMessage;
+function encodeMessages(messages) {
+    const chunks = [];
+    for (const message of messages) {
+        const pbType = protocol_1.PROTOBUF_MESSAGE_BY_TYPE.get(BigInt(message.messageType));
+        if (pbType === undefined) {
+            throw new Error("Trying to encode a message with unknown message type " +
+                message.messageType);
+        }
+        const bodyBuf = pbType.encode(message.message).finish();
+        const header = new types_1.Header(BigInt(message.messageType), bodyBuf.length, message.completed, message.protocolVersion, // only set for incoming start message
+        message.requiresAck);
+        const encoded = header.toU64be();
+        const headerBuf = Buffer.alloc(8);
+        headerBuf.writeBigUInt64BE(encoded);
+        chunks.push(headerBuf);
+        chunks.push(bodyBuf);
+    }
+    return Buffer.concat(chunks);
+}
+exports.encodeMessages = encodeMessages;
+//# sourceMappingURL=encoder.js.map

package/dist/io/encoder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"encoder.js","sourceRoot":"","sources":["../../src/io/encoder.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;;;;AAEH,oDAA4B;AAC5B,gDAA6D;AAC7D,0CAAiD;AAEjD,SAAgB,aAAa;IAC3B,OAAO,IAAI,gBAAM,CAAC,SAAS,CAAC;QAC1B,kBAAkB,EAAE,IAAI;QACxB,UAAU,EAAE,IAAI;QAEhB,SAAS,CAAC,GAAG,EAAE,SAAS,EAAE,EAAE;YAC1B,0FAA0F;YAC1F,qCAAqC;YACrC,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;YAClC,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACnB,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAZD,sCAYC;AAED,SAAgB,aAAa,CAAC,GAAY;IACxC,OAAO,cAAc,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;AAC/B,CAAC;AAFD,sCAEC;AAED,SAAgB,cAAc,CAAC,QAAmB;IAChD,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE;QAC9B,MAAM,MAAM,GAAG,mCAAwB,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;QACzE,IAAI,MAAM,KAAK,SAAS,EAAE;YACxB,MAAM,IAAI,KAAK,CACb,uDAAuD;gBACrD,OAAO,CAAC,WAAW,CACtB,CAAC;SACH;QACD,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;QACxD,MAAM,MAAM,GAAG,IAAI,cAAM,CACvB,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAC3B,OAAO,CAAC,MAAM,EACd,OAAO,CAAC,SAAS,EACjB,OAAO,CAAC,eAAe,EAAE,sCAAsC;QAC/D,OAAO,CAAC,WAAW,CACpB,CAAC;QACF,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;QACjC,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAClC,SAAS,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACpC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACvB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;KACtB;IACD,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;AAC/B,CAAC;AAzBD,wCAyBC"}

package/dist/journal.d.ts

@@ -0,0 +1,42 @@
+import * as p from "./types/protocol";
+import { Failure } from "./generated/proto/protocol";
+import { CompletionMessage, OutputStreamEntryMessage, SuspensionMessage } from "./types/protocol";
+import { Invocation } from "./invocation";
+export declare class Journal<I, O> {
+    readonly invocation: Invocation<I, O>;
+    private state;
+    private userCodeJournalIndex;
+    private pendingJournalEntries;
+    constructor(invocation: Invocation<I, O>);
+    handleInputMessage(m: p.PollInputStreamEntryMessage): void;
+    handleUserSideMessage(messageType: bigint, message: p.ProtocolMessage | Uint8Array): Promise<any | undefined>;
+    isUnResolved(index: number): boolean;
+    handleRuntimeCompletionMessage(m: CompletionMessage): void;
+    private handleReplay;
+    resolveResult<T>(journalIndex: number, journalEntry: JournalEntry, value: T | undefined, failure?: Failure | undefined, failureWouldBeTerminal?: boolean): void;
+    handleClosingMessage(messageType: bigint, message: OutputStreamEntryMessage | SuspensionMessage): void;
+    private checkJournalMatch;
+    getCompletableIndices(): number[];
+    private transitionState;
+    private incrementUserCodeIndex;
+    isClosed(): boolean;
+    isProcessing(): boolean;
+    getUserCodeJournalIndex(): number;
+    close(): void;
+    outputMsgWasReplayed(): boolean | undefined;
+    nextEntryWillBeReplayed(): boolean;
+}
+export declare class JournalEntry {
+    readonly messageType: bigint;
+    readonly message: p.ProtocolMessage | Uint8Array;
+    promise: Promise<any>;
+    resolve: (value: any) => void;
+    reject: (reason?: any) => void;
+    constructor(messageType: bigint, message: p.ProtocolMessage | Uint8Array);
+}
+export declare enum NewExecutionState {
+    REPLAYING = "REPLAYING",
+    PROCESSING = "PROCESSING",
+    CLOSED = "CLOSED"
+}
+//# sourceMappingURL=journal.d.ts.map

package/dist/journal.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"journal.d.ts","sourceRoot":"","sources":["../src/journal.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,CAAC,MAAM,kBAAkB,CAAC;AACtC,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AACrD,OAAO,EAML,iBAAiB,EAMjB,wBAAwB,EAQxB,iBAAiB,EAClB,MAAM,kBAAkB,CAAC;AAI1B,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAG1C,qBAAa,OAAO,CAAC,CAAC,EAAE,CAAC;IASX,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,EAAE,CAAC,CAAC;IARjD,OAAO,CAAC,KAAK,CAA+B;IAE5C,OAAO,CAAC,oBAAoB,CAAK;IAIjC,OAAO,CAAC,qBAAqB,CAAmC;gBAE3C,UAAU,EAAE,UAAU,CAAC,CAAC,EAAE,CAAC,CAAC;IAejD,kBAAkB,CAAC,CAAC,EAAE,CAAC,CAAC,2BAA2B;IAa5C,qBAAqB,CAC1B,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,CAAC,CAAC,eAAe,GAAG,UAAU,GAEtC,OAAO,CAAC,GAAG,GAAG,SAAS,CAAC;IAgFpB,YAAY,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO;IAKpC,8BAA8B,CAAC,CAAC,EAAE,iBAAiB;IAkC1D,OAAO,CAAC,YAAY;IAuHpB,aAAa,CAAC,CAAC,EACb,YAAY,EAAE,MAAM,EACpB,YAAY,EAAE,YAAY,EAC1B,KAAK,EAAE,CAAC,GAAG,SAAS,EACpB,OAAO,CAAC,EAAE,OAAO,GAAG,SAAS,EAC7B,sBAAsB,CAAC,EAAE,OAAO;IAclC,oBAAoB,CAClB,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,wBAAwB,GAAG,iBAAiB;IAiBvD,OAAO,CAAC,iBAAiB;IAmBlB,qBAAqB,IAAI,MAAM,EAAE;IAOxC,OAAO,CAAC,eAAe;IAcvB,OAAO,CAAC,sBAAsB;IAUvB,QAAQ,IAAI,OAAO;IAInB,YAAY,IAAI,OAAO;IAIvB,uBAAuB,IAAI,MAAM;IAIjC,KAAK;IAIL,oBAAoB;IAgB3B,uBAAuB;CAGxB;AAED,qBAAa,YAAY;IASrB,QAAQ,CAAC,WAAW,EAAE,MAAM;IAC5B,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC,eAAe,GAAG,UAAU;IAR3C,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC;IAEtB,OAAO,EAAG,CAAC,KAAK,EAAE,GAAG,KAAK,IAAI,CAAC;IAE/B,MAAM,EAAG,CAAC,MAAM,CAAC,EAAE,GAAG,KAAK,IAAI,CAAC;gBAG5B,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,CAAC,CAAC,eAAe,GAAG,UAAU;CAQnD;AAOD,oBAAY,iBAAiB;IAC3B,SAAS,cAAc;IACvB,UAAU,eAAe;IACzB,MAAM,WAAW;CAClB"}