rivetkit 2.3.0-rc.8 → 2.3.0

package/dist/tsup/workflow/mod.cjs

@@ -1,23 +1,23 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } var _class; var _class2;
 
-var _chunkAWTPTUQ7cjs = require('../chunk-AWTPTUQ7.cjs');
+var _chunkB6VUNZUDcjs = require('../chunk-B6VUNZUD.cjs');
+require('../chunk-47HHIEXH.cjs');
 
 
 
 
-var _chunkSRNOPUC6cjs = require('../chunk-SRNOPUC6.cjs');
-require('../chunk-OT7FF6GB.cjs');
+var _chunkNIY3RSPXcjs = require('../chunk-NIY3RSPX.cjs');
 
 
 
-var _chunkLD5YASJUcjs = require('../chunk-LD5YASJU.cjs');
-require('../chunk-HGW6PBWR.cjs');
+var _chunkVE2X4KMGcjs = require('../chunk-VE2X4KMG.cjs');
+require('../chunk-3677IIOV.cjs');
 
 
-var _chunkBATTOVHFcjs = require('../chunk-BATTOVHF.cjs');
+var _chunkDXXJPH55cjs = require('../chunk-DXXJPH55.cjs');
 
 
-var _chunk63WNTDRCcjs = require('../chunk-63WNTDRC.cjs');
+var _chunkKORQB2IRcjs = require('../chunk-KORQB2IR.cjs');
 
 // src/workflow/mod.ts
 
@@ -34,269 +34,6 @@ var _chunk63WNTDRCcjs = require('../chunk-63WNTDRC.cjs');
 var _workflowengine = require('@rivetkit/workflow-engine');
 var _invariant = require('invariant'); var _invariant2 = _interopRequireDefault(_invariant);
 
-// src/workflow/driver.ts
-var WORKFLOW_STORAGE_PREFIX = _chunkLD5YASJUcjs.workflowStoragePrefix.call(void 0, );
-function stripWorkflowKey(prefixed) {
-  return prefixed.slice(WORKFLOW_STORAGE_PREFIX.length);
-}
-function computeUpperBound(prefix) {
-  const upperBound = prefix.slice();
-  for (let i = upperBound.length - 1; i >= 0; i--) {
-    if (upperBound[i] !== 255) {
-      upperBound[i]++;
-      return upperBound.slice(0, i + 1);
-    }
-  }
-  return null;
-}
-var ActorWorkflowMessageDriver = class {
-  #actor;
-  #runCtx;
-  constructor(actor, runCtx) {
-    this.#actor = actor;
-    this.#runCtx = runCtx;
-  }
-  async addMessage(message) {
-    await this.#runCtx.internalKeepAwake(
-      this.#actor.queueManager.enqueue(message.name, message.data)
-    );
-  }
-  async receiveMessages(opts) {
-    const messages = await this.#runCtx.internalKeepAwake(
-      this.#actor.queueManager.receive(
-        opts.names && opts.names.length > 0 ? [...opts.names] : void 0,
-        opts.count,
-        0,
-        void 0,
-        opts.completable
-      )
-    );
-    return messages.map((message) => ({
-      id: message.id.toString(),
-      name: message.name,
-      data: message.body,
-      sentAt: message.createdAt,
-      ...opts.completable ? {
-        complete: async (response) => {
-          await this.#runCtx.internalKeepAwake(
-            this.#actor.queueManager.completeMessage(
-              message,
-              response
-            )
-          );
-        }
-      } : {}
-    }));
-  }
-  async completeMessage(messageId, response) {
-    let parsedId;
-    try {
-      parsedId = BigInt(messageId);
-    } catch (e) {
-      return;
-    }
-    await this.#runCtx.internalKeepAwake(
-      this.#actor.queueManager.completeMessageById(parsedId, response)
-    );
-  }
-};
-var ActorWorkflowDriver = (_class = class {
-  __init() {this.workerPollInterval = 100}
-  
-  #actor;
-  #runCtx;
-  constructor(actor, runCtx) {;_class.prototype.__init.call(this);
-    this.#actor = actor;
-    this.#runCtx = runCtx;
-    this.messageDriver = new ActorWorkflowMessageDriver(actor, runCtx);
-  }
-  async get(key) {
-    const [value] = await this.#runCtx.internalKeepAwake(
-      this.#actor.driver.kvBatchGet(this.#actor.id, [
-        _chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, key)
-      ])
-    );
-    return _nullishCoalesce(value, () => ( null));
-  }
-  async set(key, value) {
-    await this.#runCtx.internalKeepAwake(
-      this.#actor.driver.kvBatchPut(this.#actor.id, [
-        [_chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, key), value]
-      ])
-    );
-  }
-  async delete(key) {
-    await this.#runCtx.internalKeepAwake(
-      this.#actor.driver.kvBatchDelete(this.#actor.id, [
-        _chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, key)
-      ])
-    );
-  }
-  async deletePrefix(prefix) {
-    const start = _chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, prefix);
-    const end = computeUpperBound(start);
-    if (end) {
-      await this.#runCtx.internalKeepAwake(
-        this.#actor.driver.kvDeleteRange(this.#actor.id, start, end)
-      );
-    } else {
-      const entries = await this.#runCtx.internalKeepAwake(
-        this.#actor.driver.kvListPrefix(this.#actor.id, start)
-      );
-      if (entries.length === 0) {
-        return;
-      }
-      await this.#runCtx.internalKeepAwake(
-        this.#actor.driver.kvBatchDelete(
-          this.#actor.id,
-          entries.map(([key]) => key)
-        )
-      );
-    }
-  }
-  async deleteRange(start, end) {
-    await this.#runCtx.internalKeepAwake(
-      this.#actor.driver.kvDeleteRange(
-        this.#actor.id,
-        _chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, start),
-        _chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, end)
-      )
-    );
-  }
-  async list(prefix) {
-    const entries = await this.#runCtx.internalKeepAwake(
-      this.#actor.driver.kvListPrefix(
-        this.#actor.id,
-        _chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, prefix)
-      )
-    );
-    return entries.map(([key, value]) => ({
-      key: stripWorkflowKey(key),
-      value
-    }));
-  }
-  async batch(writes) {
-    if (writes.length === 0) return;
-    await this.#runCtx.internalKeepAwake(
-      Promise.all([
-        this.#actor.driver.kvBatchPut(
-          this.#actor.id,
-          writes.map(({ key, value }) => [
-            _chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, key),
-            value
-          ])
-        ),
-        this.#actor.stateManager.saveState({
-          immediate: true
-        })
-      ])
-    );
-  }
-  async setAlarm(_workflowId, wakeAt) {
-    await this.#runCtx.internalKeepAwake(
-      this.#actor.driver.setAlarm(this.#actor, wakeAt)
-    );
-  }
-  async clearAlarm(_workflowId) {
-    return;
-  }
-  waitForMessages(messageNames, abortSignal) {
-    return this.#actor.queueManager.waitForNames(
-      messageNames.length > 0 ? messageNames : void 0,
-      abortSignal
-    );
-  }
-}, _class);
-var NoopWorkflowMessageDriver = class {
-  async addMessage(_message) {
-    throw new Error("Workflow control driver does not support messages");
-  }
-  async receiveMessages(_opts) {
-    throw new Error("Workflow control driver does not support messages");
-  }
-  async completeMessage(_messageId, _response) {
-    throw new Error("Workflow control driver does not support messages");
-  }
-};
-var ActorWorkflowControlDriver = (_class2 = class {
-  __init2() {this.workerPollInterval = 100}
-  __init3() {this.messageDriver = new NoopWorkflowMessageDriver()}
-  #actor;
-  constructor(actor) {;_class2.prototype.__init2.call(this);_class2.prototype.__init3.call(this);
-    this.#actor = actor;
-  }
-  async get(key) {
-    const [value] = await this.#actor.driver.kvBatchGet(this.#actor.id, [
-      _chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, key)
-    ]);
-    return _nullishCoalesce(value, () => ( null));
-  }
-  async set(key, value) {
-    await this.#actor.driver.kvBatchPut(this.#actor.id, [
-      [_chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, key), value]
-    ]);
-  }
-  async delete(key) {
-    await this.#actor.driver.kvBatchDelete(this.#actor.id, [
-      _chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, key)
-    ]);
-  }
-  async deletePrefix(prefix) {
-    const start = _chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, prefix);
-    const end = computeUpperBound(start);
-    if (end) {
-      await this.#actor.driver.kvDeleteRange(this.#actor.id, start, end);
-      return;
-    }
-    const entries = await this.#actor.driver.kvListPrefix(
-      this.#actor.id,
-      start
-    );
-    if (entries.length === 0) {
-      return;
-    }
-    await this.#actor.driver.kvBatchDelete(
-      this.#actor.id,
-      entries.map(([key]) => key)
-    );
-  }
-  async deleteRange(start, end) {
-    await this.#actor.driver.kvDeleteRange(
-      this.#actor.id,
-      _chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, start),
-      _chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, end)
-    );
-  }
-  async list(prefix) {
-    const entries = await this.#actor.driver.kvListPrefix(
-      this.#actor.id,
-      _chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, prefix)
-    );
-    return entries.map(([key, value]) => ({
-      key: stripWorkflowKey(key),
-      value
-    }));
-  }
-  async batch(writes) {
-    if (writes.length === 0) {
-      return;
-    }
-    await this.#actor.driver.kvBatchPut(
-      this.#actor.id,
-      writes.map(({ key, value }) => [_chunkLD5YASJUcjs.makeWorkflowKey.call(void 0, key), value])
-    );
-  }
-  async setAlarm(_workflowId, wakeAt) {
-    await this.#actor.driver.setAlarm(this.#actor, wakeAt);
-  }
-  async clearAlarm(_workflowId) {
-    return;
-  }
-  waitForMessages(_messageNames, _abortSignal) {
-    throw new Error("Workflow control driver does not support messages");
-  }
-}, _class2);
-
 // src/workflow/constants.ts
 var WORKFLOW_GUARD_KV_KEY = "__rivet_actor_workflow_guard_triggered";
 
@@ -516,7 +253,7 @@ var ActorWorkflowContext = class _ActorWorkflowContext {
     let id;
     try {
       id = BigInt(message.id);
-    } catch (e2) {
+    } catch (e) {
       throw new Error(`Invalid queue message id "${message.id}"`);
     }
     return {
@@ -547,7 +284,7 @@ var ActorWorkflowContext = class _ActorWorkflowContext {
   async #withActorAccessAndStateRollback(run) {
     let stateSnapshot = null;
     try {
-      stateSnapshot = { state: this.#runCtx[_chunkSRNOPUC6cjs.RAW_STATE_SYMBOL]() };
+      stateSnapshot = { state: this.#runCtx[_chunkNIY3RSPXcjs.RAW_STATE_SYMBOL]() };
     } catch (error) {
       this.#runCtx.log.debug({
         msg: "failed to get state, likely due to being stateless workflow",
@@ -588,7 +325,7 @@ var ActorWorkflowContext = class _ActorWorkflowContext {
       if (state && typeof state === "object" && "guardTriggered" in state) {
         state.guardTriggered = true;
       }
-    } catch (e3) {
+    } catch (e2) {
     }
     this.#runCtx.waitUntil(
       (async () => {
@@ -608,6 +345,269 @@ var ActorWorkflowContext = class _ActorWorkflowContext {
   }
 };
 
+// src/workflow/driver.ts
+var WORKFLOW_STORAGE_PREFIX = _chunkVE2X4KMGcjs.workflowStoragePrefix.call(void 0, );
+function stripWorkflowKey(prefixed) {
+  return prefixed.slice(WORKFLOW_STORAGE_PREFIX.length);
+}
+function computeUpperBound(prefix) {
+  const upperBound = prefix.slice();
+  for (let i = upperBound.length - 1; i >= 0; i--) {
+    if (upperBound[i] !== 255) {
+      upperBound[i]++;
+      return upperBound.slice(0, i + 1);
+    }
+  }
+  return null;
+}
+var ActorWorkflowMessageDriver = class {
+  #actor;
+  #runCtx;
+  constructor(actor, runCtx) {
+    this.#actor = actor;
+    this.#runCtx = runCtx;
+  }
+  async addMessage(message) {
+    await this.#runCtx.internalKeepAwake(
+      this.#actor.queueManager.enqueue(message.name, message.data)
+    );
+  }
+  async receiveMessages(opts) {
+    const messages = await this.#runCtx.internalKeepAwake(
+      this.#actor.queueManager.receive(
+        opts.names && opts.names.length > 0 ? [...opts.names] : void 0,
+        opts.count,
+        0,
+        void 0,
+        opts.completable
+      )
+    );
+    return messages.map((message) => ({
+      id: message.id.toString(),
+      name: message.name,
+      data: message.body,
+      sentAt: message.createdAt,
+      ...opts.completable ? {
+        complete: async (response) => {
+          await this.#runCtx.internalKeepAwake(
+            this.#actor.queueManager.completeMessage(
+              message,
+              response
+            )
+          );
+        }
+      } : {}
+    }));
+  }
+  async completeMessage(messageId, response) {
+    let parsedId;
+    try {
+      parsedId = BigInt(messageId);
+    } catch (e3) {
+      return;
+    }
+    await this.#runCtx.internalKeepAwake(
+      this.#actor.queueManager.completeMessageById(parsedId, response)
+    );
+  }
+};
+var ActorWorkflowDriver = (_class = class {
+  __init() {this.workerPollInterval = 100}
+  
+  #actor;
+  #runCtx;
+  constructor(actor, runCtx) {;_class.prototype.__init.call(this);
+    this.#actor = actor;
+    this.#runCtx = runCtx;
+    this.messageDriver = new ActorWorkflowMessageDriver(actor, runCtx);
+  }
+  async get(key) {
+    const [value] = await this.#runCtx.internalKeepAwake(
+      this.#actor.driver.kvBatchGet(this.#actor.id, [
+        _chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, key)
+      ])
+    );
+    return _nullishCoalesce(value, () => ( null));
+  }
+  async set(key, value) {
+    await this.#runCtx.internalKeepAwake(
+      this.#actor.driver.kvBatchPut(this.#actor.id, [
+        [_chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, key), value]
+      ])
+    );
+  }
+  async delete(key) {
+    await this.#runCtx.internalKeepAwake(
+      this.#actor.driver.kvBatchDelete(this.#actor.id, [
+        _chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, key)
+      ])
+    );
+  }
+  async deletePrefix(prefix) {
+    const start = _chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, prefix);
+    const end = computeUpperBound(start);
+    if (end) {
+      await this.#runCtx.internalKeepAwake(
+        this.#actor.driver.kvDeleteRange(this.#actor.id, start, end)
+      );
+    } else {
+      const entries = await this.#runCtx.internalKeepAwake(
+        this.#actor.driver.kvListPrefix(this.#actor.id, start)
+      );
+      if (entries.length === 0) {
+        return;
+      }
+      await this.#runCtx.internalKeepAwake(
+        this.#actor.driver.kvBatchDelete(
+          this.#actor.id,
+          entries.map(([key]) => key)
+        )
+      );
+    }
+  }
+  async deleteRange(start, end) {
+    await this.#runCtx.internalKeepAwake(
+      this.#actor.driver.kvDeleteRange(
+        this.#actor.id,
+        _chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, start),
+        _chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, end)
+      )
+    );
+  }
+  async list(prefix) {
+    const entries = await this.#runCtx.internalKeepAwake(
+      this.#actor.driver.kvListPrefix(
+        this.#actor.id,
+        _chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, prefix)
+      )
+    );
+    return entries.map(([key, value]) => ({
+      key: stripWorkflowKey(key),
+      value
+    }));
+  }
+  async batch(writes) {
+    if (writes.length === 0) return;
+    await this.#runCtx.internalKeepAwake(
+      Promise.all([
+        this.#actor.driver.kvBatchPut(
+          this.#actor.id,
+          writes.map(({ key, value }) => [
+            _chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, key),
+            value
+          ])
+        ),
+        this.#actor.stateManager.saveState({
+          immediate: true
+        })
+      ])
+    );
+  }
+  async setAlarm(_workflowId, wakeAt) {
+    await this.#runCtx.internalKeepAwake(
+      this.#actor.driver.setAlarm(this.#actor, wakeAt)
+    );
+  }
+  async clearAlarm(_workflowId) {
+    return;
+  }
+  waitForMessages(messageNames, abortSignal) {
+    return this.#actor.queueManager.waitForNames(
+      messageNames.length > 0 ? messageNames : void 0,
+      abortSignal
+    );
+  }
+}, _class);
+var NoopWorkflowMessageDriver = class {
+  async addMessage(_message) {
+    throw new Error("Workflow control driver does not support messages");
+  }
+  async receiveMessages(_opts) {
+    throw new Error("Workflow control driver does not support messages");
+  }
+  async completeMessage(_messageId, _response) {
+    throw new Error("Workflow control driver does not support messages");
+  }
+};
+var ActorWorkflowControlDriver = (_class2 = class {
+  __init2() {this.workerPollInterval = 100}
+  __init3() {this.messageDriver = new NoopWorkflowMessageDriver()}
+  #actor;
+  constructor(actor) {;_class2.prototype.__init2.call(this);_class2.prototype.__init3.call(this);
+    this.#actor = actor;
+  }
+  async get(key) {
+    const [value] = await this.#actor.driver.kvBatchGet(this.#actor.id, [
+      _chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, key)
+    ]);
+    return _nullishCoalesce(value, () => ( null));
+  }
+  async set(key, value) {
+    await this.#actor.driver.kvBatchPut(this.#actor.id, [
+      [_chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, key), value]
+    ]);
+  }
+  async delete(key) {
+    await this.#actor.driver.kvBatchDelete(this.#actor.id, [
+      _chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, key)
+    ]);
+  }
+  async deletePrefix(prefix) {
+    const start = _chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, prefix);
+    const end = computeUpperBound(start);
+    if (end) {
+      await this.#actor.driver.kvDeleteRange(this.#actor.id, start, end);
+      return;
+    }
+    const entries = await this.#actor.driver.kvListPrefix(
+      this.#actor.id,
+      start
+    );
+    if (entries.length === 0) {
+      return;
+    }
+    await this.#actor.driver.kvBatchDelete(
+      this.#actor.id,
+      entries.map(([key]) => key)
+    );
+  }
+  async deleteRange(start, end) {
+    await this.#actor.driver.kvDeleteRange(
+      this.#actor.id,
+      _chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, start),
+      _chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, end)
+    );
+  }
+  async list(prefix) {
+    const entries = await this.#actor.driver.kvListPrefix(
+      this.#actor.id,
+      _chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, prefix)
+    );
+    return entries.map(([key, value]) => ({
+      key: stripWorkflowKey(key),
+      value
+    }));
+  }
+  async batch(writes) {
+    if (writes.length === 0) {
+      return;
+    }
+    await this.#actor.driver.kvBatchPut(
+      this.#actor.id,
+      writes.map(({ key, value }) => [_chunkVE2X4KMGcjs.makeWorkflowKey.call(void 0, key), value])
+    );
+  }
+  async setAlarm(_workflowId, wakeAt) {
+    await this.#actor.driver.setAlarm(this.#actor, wakeAt);
+  }
+  async clearAlarm(_workflowId) {
+    return;
+  }
+  waitForMessages(_messageNames, _abortSignal) {
+    throw new Error("Workflow control driver does not support messages");
+  }
+}, _class2);
+
 // src/workflow/mod.ts
 
 function shouldRethrowWorkflowError(error) {
@@ -620,7 +620,7 @@ function shouldRethrowWorkflowError(error) {
   return true;
 }
 function workflowReplayInFlightError() {
-  return new (0, _chunk63WNTDRCcjs.RivetError)(
+  return new (0, _chunkKORQB2IRcjs.RivetError)(
     "actor",
     "workflow_in_flight",
     "Workflow replay is unavailable while the workflow is currently in flight.",
@@ -639,13 +639,13 @@ function workflow(fn, options = {}) {
   function getWorkflowInspector(actorId) {
     let workflowInspector = workflowInspectors.get(actorId);
     if (!workflowInspector) {
-      workflowInspector = _chunkAWTPTUQ7cjs.createWorkflowInspectorAdapter.call(void 0, );
+      workflowInspector = _chunkB6VUNZUDcjs.createWorkflowInspectorAdapter.call(void 0, );
       workflowInspectors.set(actorId, workflowInspector);
     }
     return workflowInspector;
   }
   async function run(runCtx) {
-    const actor = runCtx[_chunkSRNOPUC6cjs.ACTOR_CONTEXT_INTERNAL_SYMBOL];
+    const actor = runCtx[_chunkNIY3RSPXcjs.ACTOR_CONTEXT_INTERNAL_SYMBOL];
     _invariant2.default.call(void 0, actor, "workflow() requires an actor instance");
     const workflowInspector = getWorkflowInspector(actor.id);
     const driver = new ActorWorkflowDriver(actor, runCtx);
@@ -705,20 +705,20 @@ function workflow(fn, options = {}) {
       if (shouldRethrowWorkflowError(error)) {
         runCtx.log.error({
           msg: "workflow run failed",
-          error: _chunkBATTOVHFcjs.stringifyError.call(void 0, error)
+          error: _chunkDXXJPH55cjs.stringifyError.call(void 0, error)
         });
         throw error;
       }
       runCtx.log.warn({
         msg: "workflow failed and will sleep until woken",
-        error: _chunkBATTOVHFcjs.stringifyError.call(void 0, error)
+        error: _chunkDXXJPH55cjs.stringifyError.call(void 0, error)
       });
     } finally {
       runCtx.abortSignal.removeEventListener("abort", onAbort);
     }
   }
   const runWithConfig = run;
-  runWithConfig[_chunkSRNOPUC6cjs.RUN_FUNCTION_CONFIG_SYMBOL] = {
+  runWithConfig[_chunkNIY3RSPXcjs.RUN_FUNCTION_CONFIG_SYMBOL] = {
     icon: "diagram-project",
     inspectorFactory: (actor) => {
       const actorId = resolveWorkflowInspectorActorId(actor);