@restatedev/restate-sdk 1.0.0 → 1.1.0

package/README.md

@@ -53,6 +53,6 @@ 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                           | ✅   |
+| Restate Server\sdk-typescript | 1.0 | 1.1 |
+|-------------------------------|-----|-----|
+| 1.0                           | ✅   | ✅   |

package/dist/cjs/package.json

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

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

@@ -0,0 +1,9 @@
+export type { Context, ObjectContext, ObjectSharedContext, WorkflowContext, WorkflowSharedContext, Rand, } from "./context.js";
+export { CombineablePromise } 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 type { ServiceBundle, RestateEndpoint } from "./endpoint.js";
+export { RestateError, TerminalError, TimeoutError } from "./types/errors.js";
+export type { Logger, LogParams, RestateLogLevel, LoggerContext, LogSource, } from "./logger.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,YAAY,EACV,OAAO,EACP,aAAa,EACb,mBAAmB,EACnB,eAAe,EACf,qBAAqB,EACrB,IAAI,GACL,MAAM,cAAc,CAAC;AACtB,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAElD,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,YAAY,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AACpE,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAC9E,YAAY,EACV,MAAM,EACN,SAAS,EACT,eAAe,EACf,aAAa,EACb,SAAS,GACV,MAAM,aAAa,CAAC"}

package/dist/{src/public_api.js → cjs/src/common_api.js}

@@ -10,18 +10,16 @@
  * https://github.com/restatedev/sdk-typescript/blob/main/LICENSE
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.TimeoutError = exports.TerminalError = exports.RestateError = exports.endpoint = exports.handlers = exports.workflow = exports.object = exports.service = exports.CombineablePromise = void 0;
-var context_1 = require("./context");
-Object.defineProperty(exports, "CombineablePromise", { enumerable: true, get: function () { return context_1.CombineablePromise; } });
-var rpc_1 = require("./types/rpc");
-Object.defineProperty(exports, "service", { enumerable: true, get: function () { return rpc_1.service; } });
-Object.defineProperty(exports, "object", { enumerable: true, get: function () { return rpc_1.object; } });
-Object.defineProperty(exports, "workflow", { enumerable: true, get: function () { return rpc_1.workflow; } });
-Object.defineProperty(exports, "handlers", { enumerable: true, get: function () { return rpc_1.handlers; } });
-var endpoint_1 = require("./endpoint");
-Object.defineProperty(exports, "endpoint", { enumerable: true, get: function () { return endpoint_1.endpoint; } });
-var errors_1 = require("./types/errors");
-Object.defineProperty(exports, "RestateError", { enumerable: true, get: function () { return errors_1.RestateError; } });
-Object.defineProperty(exports, "TerminalError", { enumerable: true, get: function () { return errors_1.TerminalError; } });
-Object.defineProperty(exports, "TimeoutError", { enumerable: true, get: function () { return errors_1.TimeoutError; } });
-//# sourceMappingURL=public_api.js.map
+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;;;AAUH,2CAAkD;AAAzC,gHAAA,kBAAkB,OAAA;AAG3B,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/connection/http_connection.d.ts → cjs/src/connection/connection.d.ts}

@@ -1,45 +1,52 @@
 /// <reference types="node" />
 /// <reference types="node" />
-import stream from "node:stream";
-import { Connection, RestateStreamConsumer } from "./connection";
-import { Message } from "../types/types";
-import { Http2ServerRequest, IncomingHttpHeaders } from "node:http2";
+import type stream from "node:stream/web";
+import type { Message } from "../types/types.js";
 /**
- * A duplex stream with Restate Messages over HTTP2.
+ * A connection from the service/SDK to Restate.
+ * Accepts messages to be sent and committed to the journal.
+ */
+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.
+ * Messages include journal replay messages and completion messages.
+ */
+export interface RestateStreamConsumer {
+    handleMessage(m: Message): boolean;
+    handleStreamError(e: Error): void;
+    handleInputClosed(): void;
+}
+/**
+ * A stream with Restate Messages, potentially but not necessarily full duplex.
  *
  * This stream handles the following concerns:
  *
- * (1) encoding and decoding of messages from  and from raw bytes
- *
- * (2) buffering the outgoing messages, because the call sites that produce (potentially) large
- *     messages might not await their transfer. Aside from the fact that we achieve better pipelining
- *     that way, we also simply cannot guarantee that users of the Restate SDK actually await the
- *     relevant async API methods.
+ * (1) encoding and decoding of messages to and from raw bytes
  *
- *     This stream essentially buffers messages and, upon flush, sends them asynchronously, as the
- *     stream has availability. Flush requests queue up, if new data gets flushed while the previous
- *     data is still being sent.
  *
- * (3) Input messages can be pipelined to a sequence of consumers. For example, first to a journal,
+ * (2) Input messages can be pipelined to a sequence of consumers. For example, first to a journal,
  *     and afterwards to the state machine.
  *
- * (4) Handling the relevant stream events for errors and consolidating them to one error handler, plus
+ * (3) Handling the relevant stream events for errors and consolidating them to one error handler, plus
  *     notifications for cleanly closed input (to trigger suspension).
  */
-export declare class RestateHttp2Connection implements Connection {
+export declare class RestateConnection implements Connection {
     private readonly attemptHeaders;
-    private readonly rawStream;
     /**
-     * create a RestateDuplex stream from an http2 (duplex) stream.
+     * create a RestateBidiConnection stream from a duplex stream
      */
-    static from(request: Http2ServerRequest, http2stream: stream.Duplex): RestateHttp2Connection;
-    private readonly sdkInput;
+    static from(rlog: Console, headers: Record<string, string | string[] | undefined>, rawStream: stream.ReadableWritablePair<Uint8Array, Uint8Array>): RestateConnection;
+    private readonly sdkInputPipeline;
     private currentConsumer;
     private inputBuffer;
     private consumerError?;
     private consumerInputClosed;
-    private outputBuffer;
-    constructor(attemptHeaders: IncomingHttpHeaders, rawStream: stream.Duplex);
+    private readonly sdkOutput;
+    constructor(rlog: Console, attemptHeaders: Record<string, string | string[] | undefined>, rawStream: stream.ReadableWritablePair<Uint8Array, Uint8Array>);
     headers(): ReadonlyMap<string, string | string[] | undefined>;
     /**
      * Pipes the messages from this connection to the given consumer. The consumer
@@ -70,8 +77,8 @@ export declare class RestateHttp2Connection implements Connection {
      */
     send(msg: Message): Promise<void>;
     /**
-     * Ends the stream, awaiting pending writes.
+     * Ends the output stream, awaiting pending writes.
      */
     end(): Promise<void>;
 }
-//# sourceMappingURL=http_connection.d.ts.map
+//# sourceMappingURL=connection.d.ts.map

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,MAAM,MAAM,iBAAiB,CAAC;AAE1C,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAIjD;;;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;AAED;;;;;;;;;;;;;GAaG;AACH,qBAAa,iBAAkB,YAAW,UAAU;IA0BhD,OAAO,CAAC,QAAQ,CAAC,cAAc;IAzBjC;;OAEG;WACW,IAAI,CAChB,IAAI,EAAE,OAAO,EACb,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC,EACtD,SAAS,EAAE,MAAM,CAAC,oBAAoB,CAAC,UAAU,EAAE,UAAU,CAAC,GAC7D,iBAAiB;IAMpB,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAgB;IAGjD,OAAO,CAAC,eAAe,CAAsC;IAC7D,OAAO,CAAC,WAAW,CAAiB;IACpC,OAAO,CAAC,aAAa,CAAC,CAAQ;IAC9B,OAAO,CAAC,mBAAmB,CAAS;IAEpC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAiD;gBAGzE,IAAI,EAAE,OAAO,EACI,cAAc,EAAE,MAAM,CACrC,MAAM,EACN,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAC9B,EACD,SAAS,EAAE,MAAM,CAAC,oBAAoB,CAAC,UAAU,EAAE,UAAU,CAAC;IAsEhE,OAAO,IAAI,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC;IAI7D;;;;;;;;OAQG;IACI,cAAc,CAAC,QAAQ,EAAE,qBAAqB,GAAG,IAAI;IA+B5D;;OAEG;IACI,qBAAqB,IAAI,IAAI;IAQpC;;;;;;;;;;;;OAYG;IACU,IAAI,CAAC,GAAG,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ9C;;OAEG;IACU,GAAG,IAAI,OAAO,CAAC,IAAI,CAAC;CAIlC"}

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

@@ -0,0 +1,181 @@
+"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.RestateConnection = void 0;
+const decoder_js_1 = require("../io/decoder.js");
+const encoder_js_1 = require("../io/encoder.js");
+const web_1 = require("node:stream/web");
+/**
+ * A stream with Restate Messages, potentially but not necessarily full duplex.
+ *
+ * This stream handles the following concerns:
+ *
+ * (1) encoding and decoding of messages to and from raw bytes
+ *
+ *
+ * (2) Input messages can be pipelined to a sequence of consumers. For example, first to a journal,
+ *     and afterwards to the state machine.
+ *
+ * (3) Handling the relevant stream events for errors and consolidating them to one error handler, plus
+ *     notifications for cleanly closed input (to trigger suspension).
+ */
+class RestateConnection {
+    attemptHeaders;
+    /**
+     * create a RestateBidiConnection stream from a duplex stream
+     */
+    static from(rlog, headers, rawStream) {
+        return new RestateConnection(rlog, headers, rawStream);
+    }
+    // --------------------------------------------------------------------------
+    sdkInputPipeline;
+    // consumer handling
+    currentConsumer = null;
+    inputBuffer = [];
+    consumerError;
+    consumerInputClosed = false;
+    sdkOutput;
+    constructor(rlog, attemptHeaders, rawStream) {
+        // --------------------------------------------------------------------------
+        //  input stream handling
+        // --------------------------------------------------------------------------
+        this.attemptHeaders = attemptHeaders;
+        const sdkInputSink = new web_1.WritableStream({
+            write: (m) => {
+                // deliver message, if we have a consumer. otherwise buffer the message.
+                if (this.currentConsumer) {
+                    if (this.currentConsumer.handleMessage(m)) {
+                        this.removeCurrentConsumer();
+                    }
+                }
+                else {
+                    this.inputBuffer.push(m);
+                }
+            },
+            // this notifies of errors in the decoding pipeline as well as the incoming body stream
+            abort: (e) => {
+                rlog.error("Error in request stream from Restate: " + (e.stack ?? e.message));
+                errorHandler(e);
+            },
+            close: () => {
+                // remember and forward close events
+                this.consumerInputClosed = true;
+                if (this.currentConsumer) {
+                    this.currentConsumer.handleInputClosed();
+                }
+            },
+        });
+        this.sdkInputPipeline = rawStream.readable
+            .pipeThrough((0, decoder_js_1.streamDecoder)())
+            .pipeTo(sdkInputSink);
+        this.sdkOutput = rawStream.writable.getWriter();
+        // --------- error handling --------
+        // the error handler for all sorts of errors coming from streams
+        const errorHandler = (e) => {
+            // make sure we don't overwrite the initial error
+            if (this.consumerError !== undefined) {
+                return;
+            }
+            this.consumerError = e;
+            if (this.currentConsumer) {
+                this.currentConsumer.handleStreamError(e);
+            }
+        };
+        // this notifies of errors in the outgoing response stream
+        this.sdkOutput.closed.catch((e) => {
+            rlog.error("Error in response stream to Restate: " + (e.stack ?? e.message));
+            errorHandler(e);
+        });
+        // this notifies of errors in the decoding pipeline as well as the incoming body stream
+        this.sdkInputPipeline.catch((e) => {
+            rlog.error("Error in request stream from Restate: " + (e.stack ?? e.message));
+            errorHandler(e);
+        });
+    }
+    headers() {
+        return new Map(Object.entries(this.attemptHeaders));
+    }
+    /**
+     * Pipes the messages from this connection to the given consumer. The consumer
+     * will also receive error and stream closing notifications.
+     *
+     * Once the 'handleMessage()' method returns 'true', the consumer is immediately removed.
+     * That way, consumers can consume a bounded amount of messages (like just the initial journal).
+     *
+     * There can only be one consumer at a time.
+     */
+    pipeToConsumer(consumer) {
+        if (this.currentConsumer !== null) {
+            throw new Error("Already piping to a consumer");
+        }
+        this.currentConsumer = consumer;
+        // propagate pre-existing information
+        if (this.consumerError) {
+            consumer.handleStreamError(this.consumerError);
+        }
+        if (this.consumerInputClosed) {
+            consumer.handleInputClosed();
+        }
+        // pipe the buffered input messages, if we buffered some before the consumer was registered
+        const input = this.inputBuffer;
+        if (input.length > 0) {
+            let i = 0;
+            while (i < input.length) {
+                const done = consumer.handleMessage(input[i]);
+                i++;
+                if (done) {
+                    this.removeCurrentConsumer();
+                    break;
+                }
+            }
+            this.inputBuffer = i === input.length ? [] : this.inputBuffer.slice(i);
+        }
+    }
+    /**
+     * Removes the current consumer, if there is one.
+     */
+    removeCurrentConsumer() {
+        this.currentConsumer = null;
+    }
+    // --------------------------------------------------------------------------
+    //  output stream handling
+    // --------------------------------------------------------------------------
+    /**
+     * Adds a message to the output stream.
+     *
+     * This always puts the message into the node stream, but will return a promise that is resolved once
+     * further messages can be written.
+     *
+     * The reasoning is that some, but not all Restate operations return promises and are typically
+     * awaited. For example, rpc, sleep, side-effect have promises and are awaited, while one-way-sends and
+     * state updates don't return promises.
+     *
+     * As a pragmatic solution, we always accept messages, but return a promise for when the output has
+     * capacity again, so that at least the operations that await results will respect backpressure.
+     */
+    async send(msg) {
+        const bytes = (0, encoder_js_1.encodeMessage)(msg);
+        // don't await the write as it will not complete until data is completely flushed
+        this.sdkOutput.write(bytes).catch(() => { });
+        // however, do await until there is capacity for more, so there's backpressure
+        await this.sdkOutput.ready;
+    }
+    /**
+     * Ends the output stream, awaiting pending writes.
+     */
+    async end() {
+        // we don't care if the stream had any errors at this point
+        await this.sdkOutput.close().catch(() => { });
+    }
+}
+exports.RestateConnection = RestateConnection;
+//# sourceMappingURL=connection.js.map

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;;;AAGH,iDAAiD;AAEjD,iDAAiD;AACjD,yCAAiD;AA0BjD;;;;;;;;;;;;;GAaG;AACH,MAAa,iBAAiB;IA0BT;IAzBnB;;OAEG;IACI,MAAM,CAAC,IAAI,CAChB,IAAa,EACb,OAAsD,EACtD,SAA8D;QAE9D,OAAO,IAAI,iBAAiB,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC;IACzD,CAAC;IAED,6EAA6E;IAE5D,gBAAgB,CAAgB;IAEjD,oBAAoB;IACZ,eAAe,GAAiC,IAAI,CAAC;IACrD,WAAW,GAAc,EAAE,CAAC;IAC5B,aAAa,CAAS;IACtB,mBAAmB,GAAG,KAAK,CAAC;IAEnB,SAAS,CAAiD;IAE3E,YACE,IAAa,EACI,cAGhB,EACD,SAA8D;QAE9D,6EAA6E;QAC7E,yBAAyB;QACzB,6EAA6E;QAR5D,mBAAc,GAAd,cAAc,CAG9B;QAOD,MAAM,YAAY,GAAG,IAAI,oBAAc,CAAU;YAC/C,KAAK,EAAE,CAAC,CAAC,EAAE,EAAE;gBACX,wEAAwE;gBACxE,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;oBACzB,IAAI,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;wBAC1C,IAAI,CAAC,qBAAqB,EAAE,CAAC;oBAC/B,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;YACD,uFAAuF;YACvF,KAAK,EAAE,CAAC,CAAQ,EAAE,EAAE;gBAClB,IAAI,CAAC,KAAK,CACR,wCAAwC,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,CAClE,CAAC;gBACF,YAAY,CAAC,CAAC,CAAC,CAAC;YAClB,CAAC;YACD,KAAK,EAAE,GAAG,EAAE;gBACV,oCAAoC;gBACpC,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC;gBAChC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;oBACzB,IAAI,CAAC,eAAe,CAAC,iBAAiB,EAAE,CAAC;gBAC3C,CAAC;YACH,CAAC;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,gBAAgB,GAAG,SAAS,CAAC,QAAQ;aACvC,WAAW,CAAU,IAAA,0BAAa,GAAE,CAAC;aACrC,MAAM,CAAC,YAAY,CAAC,CAAC;QAExB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;QAEhD,oCAAoC;QAEpC,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,0DAA0D;QAC1D,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAQ,EAAE,EAAE;YACvC,IAAI,CAAC,KAAK,CACR,uCAAuC,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,CACjE,CAAC;YACF,YAAY,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;QAEH,uFAAuF;QACvF,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,CAAQ,EAAE,EAAE;YACvC,IAAI,CAAC,KAAK,CACR,wCAAwC,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,CAClE,CAAC;YACF,YAAY,CAAC,CAAC,CAAC,CAAC;QAClB,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;;;;;;;;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,KAAK,CAAC,IAAI,CAAC,GAAY;QAC5B,MAAM,KAAK,GAAG,IAAA,0BAAa,EAAC,GAAG,CAAC,CAAC;QACjC,iFAAiF;QACjF,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC5C,8EAA8E;QAC9E,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;IAC7B,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,GAAG;QACd,2DAA2D;QAC3D,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IAC/C,CAAC;CACF;AAvLD,8CAuLC"}

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

@@ -1,5 +1,5 @@
 /// <reference types="node" />
-import type { Client, SendClient } from "./types/rpc";
+import type { Client, SendClient } from "./types/rpc.js";
 import type { RestateContext, RestateObjectContext, RestateObjectSharedContext, RestateWorkflowContext, RestateWorkflowSharedContext, Service, ServiceDefinitionFrom, VirtualObject, VirtualObjectDefinitionFrom, Workflow, WorkflowDefinitionFrom } from "@restatedev/restate-sdk-core";
 /**
  * Represents the original request as sent to this handler.
@@ -29,10 +29,14 @@ export interface Request {
      */
     readonly body: Uint8Array;
 }
+export type TypedState = Record<string, any>;
+export type UntypedState = {
+    _: never;
+};
 /**
  * Key value store operations. Only keyed services have an attached key-value store.
  */
-export interface KeyValueStore {
+export interface KeyValueStore<TState extends TypedState> {
     /**
      * Get/retrieve state from the Restate runtime.
      * Note that state objects are serialized with `Buffer.from(JSON.stringify(theObject))`
@@ -44,7 +48,7 @@ export interface KeyValueStore {
      * @example
      * const state = await ctx.get<string>("STATE");
      */
-    get<T>(name: string): Promise<T | null>;
+    get<TValue, TKey extends keyof TState = string>(name: TState extends UntypedState ? string : TKey): Promise<(TState extends UntypedState ? TValue : TState[TKey]) | null>;
     stateKeys(): Promise<Array<string>>;
     /**
      * Set/store state in the Restate runtime.
@@ -57,7 +61,7 @@ export interface KeyValueStore {
      * @example
      * ctx.set("STATE", "Hello");
      */
-    set<T>(name: string, value: T): void;
+    set<TValue, TKey extends keyof TState = string>(name: TState extends UntypedState ? string : TKey, value: TState extends UntypedState ? TValue : TState[TKey]): void;
     /**
      * Clear/delete state in the Restate runtime.
      * @param name key of the state to delete
@@ -65,7 +69,7 @@ export interface KeyValueStore {
      * @example
      * ctx.clear("STATE");
      */
-    clear(name: string): void;
+    clear<TKey extends keyof TState>(name: TState extends UntypedState ? string : TKey): void;
     /**
      * Clear/delete all the state entries in the Restate runtime.
      *
@@ -383,7 +387,7 @@ export interface Context extends RestateContext {
  * This context can be used only within virtual objects.
  *
  */
-export interface ObjectContext extends Context, KeyValueStore, RestateObjectContext {
+export interface ObjectContext<TState extends TypedState = UntypedState> extends Context, KeyValueStore<TState>, RestateObjectContext {
     key: string;
 }
 /**
@@ -397,7 +401,7 @@ export interface ObjectContext extends Context, KeyValueStore, RestateObjectCont
  * This context can be used only within a shared virtual objects.
  *
  */
-export interface ObjectSharedContext extends Context, RestateObjectSharedContext {
+export interface ObjectSharedContext<TState extends TypedState = UntypedState> extends Context, RestateObjectSharedContext {
     key: string;
     /**
      * Get/retrieve state from the Restate runtime.
@@ -410,7 +414,7 @@ export interface ObjectSharedContext extends Context, RestateObjectSharedContext
      * @example
      * const state = await ctx.get<string>("STATE");
      */
-    get<T>(name: string): Promise<T | null>;
+    get<TValue, TKey extends keyof TState = string>(name: TState extends UntypedState ? string : TKey): Promise<(TState extends UntypedState ? TValue : TState[TKey]) | null>;
     /**
      * Retrieve all the state keys for this object.
      */
@@ -509,7 +513,7 @@ export type DurablePromise<T> = Promise<T> & {
      */
     get(): CombineablePromise<T>;
 };
-export interface WorkflowSharedContext extends ObjectSharedContext, RestateWorkflowSharedContext {
+export interface WorkflowSharedContext<TState extends TypedState = UntypedState> extends ObjectSharedContext<TState>, RestateWorkflowSharedContext {
     /**
      * Create a durable promise that can be resolved or rejected during the workflow execution.
      * The promise is bound to the workflow and will be persisted across suspensions and retries.
@@ -536,6 +540,6 @@ export interface WorkflowSharedContext extends ObjectSharedContext, RestateWorkf
      */
     promise<T = void>(name: string): DurablePromise<T>;
 }
-export interface WorkflowContext extends WorkflowSharedContext, ObjectContext, RestateWorkflowContext {
+export interface WorkflowContext<TState extends TypedState = UntypedState> extends WorkflowSharedContext<TState>, ObjectContext<TState>, RestateWorkflowContext {
 }
 //# sourceMappingURL=context.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"context.d.ts","sourceRoot":"","sources":["../../../src/context.ts"],"names":[],"mappings":";AAWA,OAAO,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACzD,OAAO,KAAK,EACV,cAAc,EACd,oBAAoB,EACpB,0BAA0B,EAC1B,sBAAsB,EACtB,4BAA4B,EAC5B,OAAO,EACP,qBAAqB,EACrB,aAAa,EACb,2BAA2B,EAC3B,QAAQ,EACR,sBAAsB,EACvB,MAAM,8BAA8B,CAAC;AAGtC;;;;;GAKG;AACH,MAAM,WAAW,OAAO;IACtB;;;OAGG;IACH,QAAQ,CAAC,EAAE,EAAE,UAAU,CAAC;IAExB;;;OAGG;IACH,QAAQ,CAAC,OAAO,EAAE,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAE9C;;;;OAIG;IACH,QAAQ,CAAC,cAAc,EAAE,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC,CAAC;IAE5E;;OAEG;IACH,QAAQ,CAAC,IAAI,EAAE,UAAU,CAAC;CAC3B;AAGD,MAAM,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;AAC7C,MAAM,MAAM,YAAY,GAAG;IAAE,CAAC,EAAE,KAAK,CAAA;CAAE,CAAC;AAExC;;GAEG;AACH,MAAM,WAAW,aAAa,CAAC,MAAM,SAAS,UAAU;IACtD;;;;;;;;;;OAUG;IACH,GAAG,CAAC,MAAM,EAAE,IAAI,SAAS,MAAM,MAAM,GAAG,MAAM,EAC5C,IAAI,EAAE,MAAM,SAAS,YAAY,GAAG,MAAM,GAAG,IAAI,GAChD,OAAO,CAAC,CAAC,MAAM,SAAS,YAAY,GAAG,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IAEzE,SAAS,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAEpC;;;;;;;;;;OAUG;IACH,GAAG,CAAC,MAAM,EAAE,IAAI,SAAS,MAAM,MAAM,GAAG,MAAM,EAC5C,IAAI,EAAE,MAAM,SAAS,YAAY,GAAG,MAAM,GAAG,IAAI,EACjD,KAAK,EAAE,MAAM,SAAS,YAAY,GAAG,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,GACzD,IAAI,CAAC;IAER;;;;;;OAMG;IACH,KAAK,CAAC,IAAI,SAAS,MAAM,MAAM,EAC7B,IAAI,EAAE,MAAM,SAAS,YAAY,GAAG,MAAM,GAAG,IAAI,GAChD,IAAI,CAAC;IAER;;;;;OAKG;IACH,QAAQ,IAAI,IAAI,CAAC;CAClB;AAED,MAAM,WAAW,WAAW;IAC1B;;;;;;;;;;;;;;;;;;;;;;;;OAwBG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,WAAW;IAC1B;;OAEG;IACH,GAAG,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;IACvB;;QAEI;IACJ,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;CAC3B;AAED;;GAEG;AACH,MAAM,MAAM,SAAS,CAAC,CAAC,IAAI,CAAC,MAAM,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;AAE1D;;;;;;;;;;GAUG;AACH,MAAM,WAAW,OAAQ,SAAQ,cAAc;IAC7C;;;;;;;;OAQG;IACH,IAAI,EAAE,IAAI,CAAC;IAEX;;;OAGG;IACH,OAAO,EAAE,OAAO,CAAC;IAEjB;;OAEG;IACH,IAAI,EAAE,WAAW,CAAC;IAElB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAyCG;IACH,GAAG,CAAC,CAAC,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IAEzC;;;;;;OAMG;IACH,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IAEvD;;;;;;;;;;;;;;;;;;OAkBG;IACH,SAAS,CAAC,CAAC,KAAK;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,kBAAkB,CAAC,CAAC,CAAC,CAAA;KAAE,CAAC;IAE/D;;;;;;;;;;;OAWG;IACH,gBAAgB,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;IAEnD;;;;;;;;;OASG;IACH,eAAe,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IAElD;;;;;;;OAOG;IACH,KAAK,CAAC,MAAM,EAAE,MAAM,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC;IAEhD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAsCG;IACH,aAAa,CAAC,CAAC,EAAE,IAAI,EAAE,qBAAqB,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAErE;;;;;OAKG;IACH,YAAY,CAAC,CAAC,EACZ,IAAI,EAAE,2BAA2B,CAAC,CAAC,CAAC,EACpC,GAAG,EAAE,MAAM,GACV,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;IAE5B;;;;;OAKG;IACH,cAAc,CAAC,CAAC,EACd,IAAI,EAAE,sBAAsB,CAAC,CAAC,CAAC,EAC/B,GAAG,EAAE,MAAM,GACV,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAEvB;;;;;OAKG;IACH,kBAAkB,CAAC,CAAC,EAClB,IAAI,EAAE,sBAAsB,CAAC,CAAC,CAAC,EAC/B,GAAG,EAAE,MAAM,GACV,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IAE3B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAwCG;IACH,iBAAiB,CAAC,CAAC,EACjB,OAAO,EAAE,qBAAqB,CAAC,CAAC,CAAC,EACjC,IAAI,CAAC,EAAE,WAAW,GACjB,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAE1B;;;;;;OAMG;IACH,gBAAgB,CAAC,CAAC,EAChB,GAAG,EAAE,2BAA2B,CAAC,CAAC,CAAC,EACnC,GAAG,EAAE,MAAM,EACX,IAAI,CAAC,EAAE,WAAW,GACjB,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;IAEhC;;;OAGG;IACH,OAAO,IAAI,OAAO,CAAC;CACpB;AAED;;;;;;;;;;;GAWG;AACH,MAAM,WAAW,aAAa,CAAC,MAAM,SAAS,UAAU,GAAG,YAAY,CACrE,SAAQ,OAAO,EACb,aAAa,CAAC,MAAM,CAAC,EACrB,oBAAoB;IACtB,GAAG,EAAE,MAAM,CAAC;CACb;AAED;;;;;;;;;;GAUG;AACH,MAAM,WAAW,mBAAmB,CAAC,MAAM,SAAS,UAAU,GAAG,YAAY,CAC3E,SAAQ,OAAO,EACb,0BAA0B;IAC5B,GAAG,EAAE,MAAM,CAAC;IAEZ;;;;;;;;;;OAUG;IACH,GAAG,CAAC,MAAM,EAAE,IAAI,SAAS,MAAM,MAAM,GAAG,MAAM,EAC5C,IAAI,EAAE,MAAM,SAAS,YAAY,GAAG,MAAM,GAAG,IAAI,GAChD,OAAO,CAAC,CAAC,MAAM,SAAS,YAAY,GAAG,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IAEzE;;OAEG;IACH,SAAS,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,IAAI;IACnB;;;OAGG;IACH,MAAM,IAAI,MAAM,CAAC;IAEjB;;;OAGG;IACH,MAAM,IAAI,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,MAAM,kBAAkB,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,GAAG;IAC/C;;;;;;OAMG;IACH,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;CACvC,CAAC;AAEF,eAAO,MAAM,kBAAkB;IAC7B;;;;;;;;OAQG;uEAEO,CAAC;IAaX;;;;;;;;OAQG;;IAaH;;;;;;;;;OASG;;IAaH;;;;;;;;OAQG;;CAgBJ,CAAC;AAEF;;;;GAIG;AACH,MAAM,MAAM,cAAc,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,GAAG;IAC3C;;OAEG;IACH,IAAI,IAAI,OAAO,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC;IAE/B;;;OAGG;IACH,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAElC;;;OAGG;IACH,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAExC;;OAEG;IACH,GAAG,IAAI,kBAAkB,CAAC,CAAC,CAAC,CAAC;CAC9B,CAAC;AAEF,MAAM,WAAW,qBAAqB,CAAC,MAAM,SAAS,UAAU,GAAG,YAAY,CAC7E,SAAQ,mBAAmB,CAAC,MAAM,CAAC,EACjC,4BAA4B;IAC9B;;;;;;;;;;;;;;;;;;;;;;;OAuBG;IACH,OAAO,CAAC,CAAC,GAAG,IAAI,EAAE,IAAI,EAAE,MAAM,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;CACpD;AAED,MAAM,WAAW,eAAe,CAAC,MAAM,SAAS,UAAU,GAAG,YAAY,CACvE,SAAQ,qBAAqB,CAAC,MAAM,CAAC,EACnC,aAAa,CAAC,MAAM,CAAC,EACrB,sBAAsB;CAAG"}

package/dist/{src → cjs/src}/context.js

@@ -11,7 +11,7 @@
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.CombineablePromise = void 0;
-const context_impl_1 = require("./context_impl");
+const context_impl_js_1 = require("./context_impl.js");
 exports.CombineablePromise = {
     /**
      * Creates a Promise that is resolved with an array of results when all of the provided Promises
@@ -26,7 +26,7 @@ exports.CombineablePromise = {
         if (values.length == 0) {
             return Promise.all(values);
         }
-        return context_impl_1.ContextImpl.createCombinator(Promise.all.bind(Promise), values);
+        return context_impl_js_1.ContextImpl.createCombinator(Promise.all.bind(Promise), values);
     },
     /**
      * Creates a Promise that is resolved or rejected when any of the provided Promises are resolved
@@ -41,7 +41,7 @@ exports.CombineablePromise = {
         if (values.length == 0) {
             return Promise.race(values);
         }
-        return context_impl_1.ContextImpl.createCombinator(Promise.race.bind(Promise), values);
+        return context_impl_js_1.ContextImpl.createCombinator(Promise.race.bind(Promise), values);
     },
     /**
      * Creates a promise that fulfills when any of the input's promises fulfills, with this first fulfillment value.
@@ -57,7 +57,7 @@ exports.CombineablePromise = {
         if (values.length == 0) {
             return Promise.any(values);
         }
-        return context_impl_1.ContextImpl.createCombinator(Promise.any.bind(Promise), values);
+        return context_impl_js_1.ContextImpl.createCombinator(Promise.any.bind(Promise), values);
     },
     /**
      * Creates a promise that fulfills when all the input's promises settle (including when an empty iterable is passed),
@@ -72,7 +72,7 @@ exports.CombineablePromise = {
         if (values.length == 0) {
             return Promise.allSettled(values);
         }
-        return context_impl_1.ContextImpl.createCombinator(Promise.allSettled.bind(Promise), values);
+        return context_impl_js_1.ContextImpl.createCombinator(Promise.allSettled.bind(Promise), values);
     },
 };
 //# sourceMappingURL=context.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"context.js","sourceRoot":"","sources":["../../../src/context.ts"],"names":[],"mappings":";AAAA;;;;;;;;;GASG;;;AAgBH,uDAAgD;AA4fnC,QAAA,kBAAkB,GAAG;IAChC;;;;;;;;OAQG;IACH,GAAG,CACD,MAAS;QAET,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACvB,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC;QACD,OAAO,6BAAW,CAAC,gBAAgB,CACjC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,EACzB,MAAM,CAGN,CAAC;IACL,CAAC;IAED;;;;;;;;OAQG;IACH,IAAI,CACF,MAAS;QAET,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACvB,OAAO,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC9B,CAAC;QACD,OAAO,6BAAW,CAAC,gBAAgB,CACjC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,EAC1B,MAAM,CACwB,CAAC;IACnC,CAAC;IAED;;;;;;;;;OASG;IACH,GAAG,CACD,MAAS;QAET,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACvB,OAAO,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC;QACD,OAAO,6BAAW,CAAC,gBAAgB,CACjC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,EACzB,MAAM,CACwB,CAAC;IACnC,CAAC;IAED;;;;;;;;OAQG;IACH,UAAU,CACR,MAAS;QAIT,IAAI,MAAM,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACvB,OAAO,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACpC,CAAC;QACD,OAAO,6BAAW,CAAC,gBAAgB,CACjC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAChC,MAAM,CAGN,CAAC;IACL,CAAC;CACF,CAAC"}

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

@@ -1,21 +1,12 @@
 /// <reference types="node" />
 /// <reference types="node" />
-/// <reference types="node" />
-import { CombineablePromise, ContextDate, DurablePromise, ObjectContext, Rand, Request, RunAction, SendOptions, WorkflowContext } from "./context";
-import { StateMachine } from "./state_machine";
-import { AsyncLocalStorage } from "node:async_hooks";
-import { Client, HandlerKind, SendClient } from "./types/rpc";
+import type { CombineablePromise, ContextDate, DurablePromise, ObjectContext, Rand, Request, RunAction, SendOptions, WorkflowContext } from "./context.js";
+import type { StateMachine } from "./state_machine.js";
+import type { Client, SendClient } from "./types/rpc.js";
+import { HandlerKind } from "./types/rpc.js";
 import type { Service, ServiceDefinitionFrom, VirtualObjectDefinitionFrom, VirtualObject, WorkflowDefinitionFrom, Workflow } from "@restatedev/restate-sdk-core";
-import { WrappedPromise } from "./utils/promises";
+import type { WrappedPromise } from "./utils/promises.js";
 import { Buffer } from "node:buffer";
-export declare enum CallContextType {
-    None = 0,
-    Run = 1
-}
-export interface CallContext {
-    type: CallContextType;
-    delay?: number;
-}
 export type InternalCombineablePromise<T> = CombineablePromise<T> & {
     journalIndex: number;
 };
@@ -24,12 +15,11 @@ export declare class ContextImpl implements ObjectContext, WorkflowContext {
     readonly handlerKind: HandlerKind;
     readonly keyedContextKey: string | undefined;
     readonly stateMachine: StateMachine;
-    readonly rand: Rand;
-    static callContext: AsyncLocalStorage<CallContext>;
     private executingRun;
     private readonly invocationRequest;
+    readonly rand: Rand;
     readonly date: ContextDate;
-    constructor(id: Buffer, console: Console, handlerKind: HandlerKind, keyedContextKey: string | undefined, invocationValue: Uint8Array, invocationHeaders: ReadonlyMap<string, string>, attemptHeaders: ReadonlyMap<string, string | string[] | undefined>, stateMachine: StateMachine, rand?: Rand);
+    constructor(id: Buffer, console: Console, handlerKind: HandlerKind, keyedContextKey: string | undefined, invocationValue: Uint8Array, invocationHeaders: ReadonlyMap<string, string>, attemptHeaders: ReadonlyMap<string, string | string[] | undefined>, stateMachine: StateMachine);
     workflowClient<D>(opts: WorkflowDefinitionFrom<D>, key: string): Client<Workflow<D>>;
     promise<T = void>(name: string): DurablePromise<T>;
     get key(): string;
@@ -57,7 +47,6 @@ export declare class ContextImpl implements ObjectContext, WorkflowContext {
     rejectAwakeable(id: string, reason: string): void;
     private completeAwakeable;
     static createCombinator<T extends readonly CombineablePromise<unknown>[]>(combinatorConstructor: (promises: PromiseLike<unknown>[]) => Promise<unknown>, promises: T): WrappedPromise<unknown>;
-    private isInRun;
     private checkNotExecutingRun;
     private checkState;
     markCombineablePromise<T>(p: WrappedPromise<T>): InternalCombineablePromise<T>;

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

@@ -0,0 +1 @@
+{"version":3,"file":"context_impl.d.ts","sourceRoot":"","sources":["../../../src/context_impl.ts"],"names":[],"mappings":";;AAaA,OAAO,KAAK,EACV,kBAAkB,EAClB,WAAW,EACX,cAAc,EACd,aAAa,EACb,IAAI,EACJ,OAAO,EACP,SAAS,EACT,WAAW,EACX,eAAe,EAChB,MAAM,cAAc,CAAC;AACtB,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AA6CvD,OAAO,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACzD,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,KAAK,EACV,OAAO,EACP,qBAAqB,EACrB,2BAA2B,EAC3B,aAAa,EACb,sBAAsB,EACtB,QAAQ,EACT,MAAM,8BAA8B,CAAC;AAGtC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAGrC,MAAM,MAAM,0BAA0B,CAAC,CAAC,IAAI,kBAAkB,CAAC,CAAC,CAAC,GAAG;IAClE,YAAY,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF,qBAAa,WAAY,YAAW,aAAa,EAAE,eAAe;aAmB9C,OAAO,EAAE,OAAO;aAChB,WAAW,EAAE,WAAW;aACxB,eAAe,EAAE,MAAM,GAAG,SAAS;IAKnD,QAAQ,CAAC,YAAY,EAAE,YAAY;IAvBrC,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAU;IAC5C,SAAgB,IAAI,EAAE,IAAI,CAAC;IAE3B,SAAgB,IAAI,EAAE,WAAW,CAQ/B;gBAGA,EAAE,EAAE,MAAM,EACM,OAAO,EAAE,OAAO,EAChB,WAAW,EAAE,WAAW,EACxB,eAAe,EAAE,MAAM,GAAG,SAAS,EACnD,eAAe,EAAE,UAAU,EAC3B,iBAAiB,EAAE,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,EAC9C,cAAc,EAAE,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC,EAEzD,YAAY,EAAE,YAAY;IAWrC,cAAc,CAAC,CAAC,EACd,IAAI,EAAE,sBAAsB,CAAC,CAAC,CAAC,EAC/B,GAAG,EAAE,MAAM,GACV,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IAkBf,OAAO,CAAC,CAAC,GAAG,IAAI,EAAE,IAAI,EAAE,MAAM,GAAG,cAAc,CAAC,CAAC,CAAC;IAIzD,IAAW,GAAG,IAAI,MAAM,CAevB;IAEM,OAAO,IAAI,OAAO;IAKlB,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAoCvC,SAAS,IAAI,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAuBnC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,GAAG,IAAI;IAQpC,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IASzB,QAAQ,IAAI,IAAI;IAYvB,OAAO,CAAC,MAAM;YAgCA,YAAY;IAyB1B,aAAa,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,qBAAqB,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IAuBxE,YAAY,CAAC,CAAC,EACZ,EAAE,IAAI,EAAE,EAAE,2BAA2B,CAAC,CAAC,CAAC,EACxC,GAAG,EAAE,MAAM,GACV,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;IAuBpB,iBAAiB,CAAC,CAAC,EACxB,OAAO,EAAE,qBAAqB,CAAC,CAAC,CAAC,EACjC,IAAI,CAAC,EAAE,WAAW,GACjB,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;IAwBlB,gBAAgB,CAAC,CAAC,EACvB,GAAG,EAAE,2BAA2B,CAAC,CAAC,CAAC,EACnC,GAAG,EAAE,MAAM,EACX,IAAI,CAAC,EAAE,WAAW,GACjB,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC;IAyB/B,kBAAkB,CAAC,CAAC,EAClB,GAAG,EAAE,sBAAsB,CAAC,CAAC,CAAC,EAC9B,GAAG,EAAE,MAAM,EACX,IAAI,CAAC,EAAE,WAAW,GACjB,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;IA6BnB,GAAG,CAAC,CAAC,EACV,YAAY,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,EACnC,qBAAqB,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,GACnC,OAAO,CAAC,CAAC,CAAC;IAoGN,KAAK,CAAC,MAAM,EAAE,MAAM,GAAG,kBAAkB,CAAC,IAAI,CAAC;IAKtD,OAAO,CAAC,aAAa;IAWd,SAAS,CAAC,CAAC,KAAK;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,kBAAkB,CAAC,CAAC,CAAC,CAAA;KAAE;IAkC9D,gBAAgB,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,GAAG,IAAI;IAalD,eAAe,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;IAUxD,OAAO,CAAC,iBAAiB;WAcX,gBAAgB,CAC5B,CAAC,SAAS,SAAS,kBAAkB,CAAC,OAAO,CAAC,EAAE,EAEhD,qBAAqB,EAAE,CACrB,QAAQ,EAAE,WAAW,CAAC,OAAO,CAAC,EAAE,KAC7B,OAAO,CAAC,OAAO,CAAC,EACrB,QAAQ,EAAE,CAAC,GACV,cAAc,CAAC,OAAO,CAAC;IA4B1B,OAAO,CAAC,oBAAoB;IAU5B,OAAO,CAAC,UAAU;IAIlB,sBAAsB,CAAC,CAAC,EACtB,CAAC,EAAE,cAAc,CAAC,CAAC,CAAC,GACnB,0BAA0B,CAAC,CAAC,CAAC;CA4BjC"}