rivetkit 2.3.23 → 2.3.24-rc.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/browser/client.d.ts +2 -0
- package/dist/browser/client.js +1 -1
- package/dist/browser/client.js.map +1 -1
- package/dist/browser/inspector/client.js +1 -1
- package/dist/browser/inspector/client.js.map +1 -1
- package/dist/tsup/agent-os/index.cjs +1 -1
- package/dist/tsup/agent-os/index.cjs.map +1 -1
- package/dist/tsup/agent-os/index.d.cts +2 -0
- package/dist/tsup/agent-os/index.d.ts +2 -0
- package/dist/tsup/agent-os/index.js +1 -1
- package/dist/tsup/agent-os/index.js.map +1 -1
- package/dist/tsup/{chunk-UV5AN6YV.js → chunk-64CRWLOX.js} +3 -3
- package/dist/tsup/{chunk-CIWSGFQG.js → chunk-6HW3TU7Q.js} +2 -2
- package/dist/tsup/{chunk-LKNGMRIT.js → chunk-B6WTPGOR.js} +3 -3
- package/dist/tsup/{chunk-JLM6TML2.cjs → chunk-CE6NBGOL.cjs} +3 -3
- package/dist/tsup/{chunk-JLM6TML2.cjs.map → chunk-CE6NBGOL.cjs.map} +1 -1
- package/dist/tsup/{chunk-FVVRY42Y.cjs → chunk-DAC73W53.cjs} +7 -7
- package/dist/tsup/{chunk-FVVRY42Y.cjs.map → chunk-DAC73W53.cjs.map} +1 -1
- package/dist/tsup/{chunk-VJNX5RCS.cjs → chunk-ERBNWARB.cjs} +4 -4
- package/dist/tsup/{chunk-VJNX5RCS.cjs.map → chunk-ERBNWARB.cjs.map} +1 -1
- package/dist/tsup/{chunk-674WU4L6.js → chunk-JJCUSUWO.js} +2 -2
- package/dist/tsup/chunk-JJCUSUWO.js.map +1 -0
- package/dist/tsup/{chunk-7ZDC4XEA.js → chunk-JKDAOE4T.js} +4 -4
- package/dist/tsup/{chunk-CRZJ74SA.cjs → chunk-JMO6AVUC.cjs} +3 -3
- package/dist/tsup/{chunk-CRZJ74SA.cjs.map → chunk-JMO6AVUC.cjs.map} +1 -1
- package/dist/tsup/{chunk-CSOHAAZK.js → chunk-LTGBHN5Z.js} +1 -1
- package/dist/tsup/chunk-LTGBHN5Z.js.map +1 -0
- package/dist/tsup/{chunk-Z3Z5EOSP.js → chunk-M5J3PXNU.js} +2 -2
- package/dist/tsup/{chunk-NWCAIWHL.cjs → chunk-NGJWQMUU.cjs} +1 -1
- package/dist/tsup/chunk-NGJWQMUU.cjs.map +1 -0
- package/dist/tsup/{chunk-KC3T3MMB.js → chunk-OB5JLDBM.js} +2 -2
- package/dist/tsup/{chunk-Z7QK4HTR.cjs → chunk-PFHGTLF2.cjs} +10 -10
- package/dist/tsup/{chunk-Z7QK4HTR.cjs.map → chunk-PFHGTLF2.cjs.map} +1 -1
- package/dist/tsup/{chunk-HO6UA2RJ.js → chunk-QQ454DVR.js} +2 -2
- package/dist/tsup/{chunk-GZKR7ORE.cjs → chunk-R3UI7MO3.cjs} +140 -140
- package/dist/tsup/{chunk-GZKR7ORE.cjs.map → chunk-R3UI7MO3.cjs.map} +1 -1
- package/dist/tsup/{chunk-XUCVB6BD.cjs → chunk-SWQCSORE.cjs} +3 -3
- package/dist/tsup/{chunk-XUCVB6BD.cjs.map → chunk-SWQCSORE.cjs.map} +1 -1
- package/dist/tsup/{chunk-3Y7MKDLZ.cjs → chunk-YVVFOKEP.cjs} +2 -2
- package/dist/tsup/chunk-YVVFOKEP.cjs.map +1 -0
- package/dist/tsup/client/mod.cjs +5 -5
- package/dist/tsup/client/mod.d.cts +2 -2
- package/dist/tsup/client/mod.d.ts +2 -2
- package/dist/tsup/client/mod.js +4 -4
- package/dist/tsup/common/log.cjs +2 -2
- package/dist/tsup/common/log.js +1 -1
- package/dist/tsup/common/websocket.cjs +3 -3
- package/dist/tsup/common/websocket.js +2 -2
- package/dist/tsup/{config-osdJqQwX.d.cts → config-CuKzfH58.d.cts} +4 -0
- package/dist/tsup/{config-Cv6pjCt-.d.ts → config-RV6gIa2L.d.ts} +4 -0
- package/dist/tsup/{context-DpjrM7n7.d.cts → context-DPCxyjax.d.cts} +1 -1
- package/dist/tsup/{context-CoVbKBuk.d.ts → context-DoMKZBQA.d.ts} +1 -1
- package/dist/tsup/db/drizzle.cjs +2 -2
- package/dist/tsup/db/drizzle.js +1 -1
- package/dist/tsup/db/mod.cjs +3 -3
- package/dist/tsup/db/mod.js +2 -2
- package/dist/tsup/dynamic/mod.cjs +2 -2
- package/dist/tsup/dynamic/mod.d.cts +1 -1
- package/dist/tsup/dynamic/mod.d.ts +1 -1
- package/dist/tsup/dynamic/mod.js +1 -1
- package/dist/tsup/inspector/mod.cjs +5 -5
- package/dist/tsup/inspector/mod.js +4 -4
- package/dist/tsup/inspector/workflow.cjs +4 -4
- package/dist/tsup/inspector/workflow.js +3 -3
- package/dist/tsup/inspector-tab/mod.cjs +2 -2
- package/dist/tsup/inspector-tab/mod.d.cts +2 -2
- package/dist/tsup/inspector-tab/mod.d.ts +2 -2
- package/dist/tsup/inspector-tab/mod.js +1 -1
- package/dist/tsup/mod.cjs +416 -188
- package/dist/tsup/mod.cjs.map +1 -1
- package/dist/tsup/mod.d.cts +3 -3
- package/dist/tsup/mod.d.ts +3 -3
- package/dist/tsup/mod.js +318 -90
- package/dist/tsup/mod.js.map +1 -1
- package/dist/tsup/test/mod.cjs +6 -6
- package/dist/tsup/test/mod.d.cts +1 -1
- package/dist/tsup/test/mod.d.ts +1 -1
- package/dist/tsup/test/mod.js +4 -4
- package/dist/tsup/utils.cjs +2 -2
- package/dist/tsup/utils.js +1 -1
- package/dist/tsup/workflow/mod.cjs +69 -21
- package/dist/tsup/workflow/mod.cjs.map +1 -1
- package/dist/tsup/workflow/mod.d.cts +3 -3
- package/dist/tsup/workflow/mod.d.ts +3 -3
- package/dist/tsup/workflow/mod.js +64 -16
- package/dist/tsup/workflow/mod.js.map +1 -1
- package/package.json +9 -9
- package/src/actor/config.ts +2 -0
- package/src/registry/lost-callbacks.test.ts +64 -0
- package/src/registry/napi-runtime.ts +8 -0
- package/src/registry/native.ts +332 -65
- package/src/registry/run-handler-coordinator.test.ts +102 -0
- package/src/registry/run-handler-coordinator.ts +84 -17
- package/src/registry/runtime.ts +2 -0
- package/src/registry/wasm-runtime.ts +8 -0
- package/src/workflow/context.ts +20 -0
- package/src/workflow/driver.ts +25 -1
- package/src/workflow/mod.ts +46 -12
- package/dist/tsup/chunk-3Y7MKDLZ.cjs.map +0 -1
- package/dist/tsup/chunk-674WU4L6.js.map +0 -1
- package/dist/tsup/chunk-CSOHAAZK.js.map +0 -1
- package/dist/tsup/chunk-NWCAIWHL.cjs.map +0 -1
- /package/dist/tsup/{chunk-UV5AN6YV.js.map → chunk-64CRWLOX.js.map} +0 -0
- /package/dist/tsup/{chunk-CIWSGFQG.js.map → chunk-6HW3TU7Q.js.map} +0 -0
- /package/dist/tsup/{chunk-LKNGMRIT.js.map → chunk-B6WTPGOR.js.map} +0 -0
- /package/dist/tsup/{chunk-7ZDC4XEA.js.map → chunk-JKDAOE4T.js.map} +0 -0
- /package/dist/tsup/{chunk-Z3Z5EOSP.js.map → chunk-M5J3PXNU.js.map} +0 -0
- /package/dist/tsup/{chunk-KC3T3MMB.js.map → chunk-OB5JLDBM.js.map} +0 -0
- /package/dist/tsup/{chunk-HO6UA2RJ.js.map → chunk-QQ454DVR.js.map} +0 -0
package/dist/tsup/test/mod.cjs
CHANGED
|
@@ -1,13 +1,13 @@
|
|
|
1
1
|
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
|
|
2
2
|
|
|
3
|
-
var
|
|
3
|
+
var _chunkDAC73W53cjs = require('../chunk-DAC73W53.cjs');
|
|
4
4
|
|
|
5
5
|
|
|
6
|
-
var
|
|
6
|
+
var _chunkR3UI7MO3cjs = require('../chunk-R3UI7MO3.cjs');
|
|
7
7
|
require('../chunk-MYWZCDML.cjs');
|
|
8
8
|
require('../chunk-KVFIONKQ.cjs');
|
|
9
|
-
require('../chunk-
|
|
10
|
-
require('../chunk-
|
|
9
|
+
require('../chunk-JMO6AVUC.cjs');
|
|
10
|
+
require('../chunk-YVVFOKEP.cjs');
|
|
11
11
|
require('../chunk-MGI2VVF6.cjs');
|
|
12
12
|
|
|
13
13
|
// src/test/ready.ts
|
|
@@ -36,8 +36,8 @@ async function setupTest(c, registry) {
|
|
|
36
36
|
registry.config.noWelcome = true;
|
|
37
37
|
registry.start();
|
|
38
38
|
await waitForRegistryReady(registry);
|
|
39
|
-
const client =
|
|
40
|
-
...
|
|
39
|
+
const client = _chunkDAC73W53cjs.createClient.call(void 0, {
|
|
40
|
+
..._chunkR3UI7MO3cjs.convertRegistryConfigToClientConfig.call(void 0, registry.parseConfig()),
|
|
41
41
|
disableMetadataLookup: false
|
|
42
42
|
});
|
|
43
43
|
c.onTestFinished(async () => {
|
package/dist/tsup/test/mod.d.cts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { TestContext } from 'vitest';
|
|
2
|
-
import { aI as Registry, a6 as Client } from '../config-
|
|
2
|
+
import { aI as Registry, a6 as Client } from '../config-CuKzfH58.cjs';
|
|
3
3
|
import 'zod/v4';
|
|
4
4
|
import '../config-ey2I2zz5.cjs';
|
|
5
5
|
import '@rivetkit/rivetkit-wasm';
|
package/dist/tsup/test/mod.d.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { TestContext } from 'vitest';
|
|
2
|
-
import { aI as Registry, a6 as Client } from '../config-
|
|
2
|
+
import { aI as Registry, a6 as Client } from '../config-RV6gIa2L.js';
|
|
3
3
|
import 'zod/v4';
|
|
4
4
|
import '../config-ey2I2zz5.js';
|
|
5
5
|
import '@rivetkit/rivetkit-wasm';
|
package/dist/tsup/test/mod.js
CHANGED
|
@@ -1,13 +1,13 @@
|
|
|
1
1
|
import {
|
|
2
2
|
createClient
|
|
3
|
-
} from "../chunk-
|
|
3
|
+
} from "../chunk-64CRWLOX.js";
|
|
4
4
|
import {
|
|
5
5
|
convertRegistryConfigToClientConfig
|
|
6
|
-
} from "../chunk-
|
|
6
|
+
} from "../chunk-B6WTPGOR.js";
|
|
7
7
|
import "../chunk-JI6GZ2C2.js";
|
|
8
8
|
import "../chunk-JTHHCZCZ.js";
|
|
9
|
-
import "../chunk-
|
|
10
|
-
import "../chunk-
|
|
9
|
+
import "../chunk-OB5JLDBM.js";
|
|
10
|
+
import "../chunk-JJCUSUWO.js";
|
|
11
11
|
import "../chunk-ZOVJDAOW.js";
|
|
12
12
|
|
|
13
13
|
// src/test/ready.ts
|
package/dist/tsup/utils.cjs
CHANGED
|
@@ -19,7 +19,7 @@
|
|
|
19
19
|
|
|
20
20
|
|
|
21
21
|
|
|
22
|
-
var
|
|
22
|
+
var _chunkYVVFOKEPcjs = require('./chunk-YVVFOKEP.cjs');
|
|
23
23
|
require('./chunk-MGI2VVF6.cjs');
|
|
24
24
|
|
|
25
25
|
|
|
@@ -42,5 +42,5 @@ require('./chunk-MGI2VVF6.cjs');
|
|
|
42
42
|
|
|
43
43
|
|
|
44
44
|
|
|
45
|
-
exports.EXTRA_ERROR_LOG =
|
|
45
|
+
exports.EXTRA_ERROR_LOG = _chunkYVVFOKEPcjs.EXTRA_ERROR_LOG; exports.SinglePromiseQueue = _chunkYVVFOKEPcjs.SinglePromiseQueue; exports.VERSION = _chunkYVVFOKEPcjs.VERSION; exports.arrayBuffersEqual = _chunkYVVFOKEPcjs.arrayBuffersEqual; exports.assertUnreachable = _chunkYVVFOKEPcjs.assertUnreachable; exports.bufferToArrayBuffer = _chunkYVVFOKEPcjs.bufferToArrayBuffer; exports.combineUrlPath = _chunkYVVFOKEPcjs.combineUrlPath; exports.dbg = _chunkYVVFOKEPcjs.dbg; exports.detectRuntime = _chunkYVVFOKEPcjs.detectRuntime; exports.getEnvUniversal = _chunkYVVFOKEPcjs.getEnvUniversal; exports.httpUserAgent = _chunkYVVFOKEPcjs.httpUserAgent; exports.interval = _chunkYVVFOKEPcjs.interval; exports.joinSignals = _chunkYVVFOKEPcjs.joinSignals; exports.jsonParseCompat = _chunkYVVFOKEPcjs.jsonParseCompat; exports.jsonStringifyCompat = _chunkYVVFOKEPcjs.jsonStringifyCompat; exports.promiseWithResolvers = _chunkYVVFOKEPcjs.promiseWithResolvers; exports.setLongTimeout = _chunkYVVFOKEPcjs.setLongTimeout; exports.sleep = _chunkYVVFOKEPcjs.sleep; exports.stringifyError = _chunkYVVFOKEPcjs.stringifyError; exports.toUint8Array = _chunkYVVFOKEPcjs.toUint8Array;
|
|
46
46
|
//# sourceMappingURL=utils.cjs.map
|
package/dist/tsup/utils.js
CHANGED
|
@@ -1,19 +1,19 @@
|
|
|
1
1
|
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } async function _asyncNullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return await rhsFn(); } } var _class; var _class2;
|
|
2
2
|
|
|
3
|
-
var
|
|
4
|
-
require('../chunk-
|
|
5
|
-
require('../chunk-
|
|
3
|
+
var _chunkPFHGTLF2cjs = require('../chunk-PFHGTLF2.cjs');
|
|
4
|
+
require('../chunk-SWQCSORE.cjs');
|
|
5
|
+
require('../chunk-ERBNWARB.cjs');
|
|
6
6
|
|
|
7
7
|
|
|
8
8
|
|
|
9
|
-
var
|
|
9
|
+
var _chunkNGJWQMUUcjs = require('../chunk-NGJWQMUU.cjs');
|
|
10
10
|
|
|
11
11
|
|
|
12
12
|
|
|
13
13
|
var _chunkMYWZCDMLcjs = require('../chunk-MYWZCDML.cjs');
|
|
14
14
|
|
|
15
15
|
|
|
16
|
-
var
|
|
16
|
+
var _chunkYVVFOKEPcjs = require('../chunk-YVVFOKEP.cjs');
|
|
17
17
|
|
|
18
18
|
|
|
19
19
|
|
|
@@ -34,11 +34,20 @@ var _chunkMGI2VVF6cjs = require('../chunk-MGI2VVF6.cjs');
|
|
|
34
34
|
var _workflowengine = require('@rivetkit/workflow-engine');
|
|
35
35
|
var _invariant = require('invariant'); var _invariant2 = _interopRequireDefault(_invariant);
|
|
36
36
|
|
|
37
|
+
// src/workflow/context.ts
|
|
38
|
+
|
|
39
|
+
|
|
37
40
|
// src/workflow/constants.ts
|
|
38
41
|
var WORKFLOW_GUARD_KV_KEY = "__rivet_actor_workflow_guard_triggered";
|
|
39
42
|
|
|
40
43
|
// src/workflow/context.ts
|
|
41
44
|
var DEACTIVATE_STEP = /* @__PURE__ */ Symbol("workflow.step.deactivate");
|
|
45
|
+
function throwIfGenerationLost(runCtx) {
|
|
46
|
+
const actor = runCtx[_chunkNGJWQMUUcjs.ACTOR_CONTEXT_INTERNAL_SYMBOL];
|
|
47
|
+
if ((actor == null ? void 0 : actor.isLost) === true) {
|
|
48
|
+
throw new (0, _workflowengine.EvictedError)();
|
|
49
|
+
}
|
|
50
|
+
}
|
|
42
51
|
var WorkflowStepContext = class {
|
|
43
52
|
#runCtx;
|
|
44
53
|
#active = true;
|
|
@@ -293,6 +302,7 @@ var WorkflowContext = class _WorkflowContext {
|
|
|
293
302
|
// vars mutations made before a failure always remain visible.
|
|
294
303
|
async #runStep(run) {
|
|
295
304
|
const stepCtx = new WorkflowStepContext(this.#runCtx, () => this.#markGuardTriggered());
|
|
305
|
+
throwIfGenerationLost(this.#runCtx);
|
|
296
306
|
try {
|
|
297
307
|
return await run(stepCtx);
|
|
298
308
|
} finally {
|
|
@@ -317,6 +327,7 @@ var WorkflowContext = class _WorkflowContext {
|
|
|
317
327
|
// Runs a step rollback compensation with an active step context.
|
|
318
328
|
async #runRollback(rollback, output) {
|
|
319
329
|
const stepCtx = new WorkflowStepContext(this.#runCtx, () => this.#markGuardTriggered());
|
|
330
|
+
throwIfGenerationLost(this.#runCtx);
|
|
320
331
|
try {
|
|
321
332
|
await rollback(stepCtx, output);
|
|
322
333
|
} finally {
|
|
@@ -350,6 +361,7 @@ var WorkflowContext = class _WorkflowContext {
|
|
|
350
361
|
);
|
|
351
362
|
}
|
|
352
363
|
#createChildContext(ctx) {
|
|
364
|
+
throwIfGenerationLost(this.#runCtx);
|
|
353
365
|
return new _WorkflowContext(ctx, this.#runCtx);
|
|
354
366
|
}
|
|
355
367
|
};
|
|
@@ -393,10 +405,26 @@ function normalizeSqlBlob(value) {
|
|
|
393
405
|
}
|
|
394
406
|
throw new Error("workflow sqlite value was not a blob");
|
|
395
407
|
}
|
|
408
|
+
function guardRawAccessAgainstLost(access, runCtx) {
|
|
409
|
+
return {
|
|
410
|
+
...access,
|
|
411
|
+
execute: ((...args) => {
|
|
412
|
+
throwIfGenerationLost(runCtx);
|
|
413
|
+
return access.execute(...args);
|
|
414
|
+
}),
|
|
415
|
+
transaction: (callback, options) => {
|
|
416
|
+
throwIfGenerationLost(runCtx);
|
|
417
|
+
return access.transaction(
|
|
418
|
+
(tx) => callback(guardRawAccessAgainstLost(tx, runCtx)),
|
|
419
|
+
options
|
|
420
|
+
);
|
|
421
|
+
}
|
|
422
|
+
};
|
|
423
|
+
}
|
|
396
424
|
function runtimeDbFromContext(runCtx) {
|
|
397
425
|
const db = runCtx == null ? void 0 : runCtx.db;
|
|
398
426
|
if (db && typeof db === "object" && "execute" in db && "transaction" in db) {
|
|
399
|
-
return db;
|
|
427
|
+
return guardRawAccessAgainstLost(db, runCtx);
|
|
400
428
|
}
|
|
401
429
|
return void 0;
|
|
402
430
|
}
|
|
@@ -740,18 +768,26 @@ function isRunHandlerUnavailable(error) {
|
|
|
740
768
|
function workflow(fn, options = {}) {
|
|
741
769
|
const onError = options.onError;
|
|
742
770
|
const workflowInspectors = /* @__PURE__ */ new Map();
|
|
743
|
-
function
|
|
744
|
-
|
|
771
|
+
function workflowInspectorKey(actorId, actorGeneration) {
|
|
772
|
+
return actorGeneration === void 0 ? actorId : `${actorId}#${actorGeneration}`;
|
|
773
|
+
}
|
|
774
|
+
function getWorkflowInspector(actorId, actorGeneration) {
|
|
775
|
+
const key = workflowInspectorKey(actorId, actorGeneration);
|
|
776
|
+
let workflowInspector = workflowInspectors.get(key);
|
|
745
777
|
if (!workflowInspector) {
|
|
746
|
-
workflowInspector =
|
|
747
|
-
workflowInspectors.set(
|
|
778
|
+
workflowInspector = _chunkPFHGTLF2cjs.createWorkflowInspectorAdapter.call(void 0, );
|
|
779
|
+
workflowInspectors.set(key, workflowInspector);
|
|
748
780
|
}
|
|
749
781
|
return workflowInspector;
|
|
750
782
|
}
|
|
751
783
|
async function run(runCtx) {
|
|
752
|
-
const actor = runCtx[
|
|
784
|
+
const actor = runCtx[_chunkNGJWQMUUcjs.ACTOR_CONTEXT_INTERNAL_SYMBOL];
|
|
753
785
|
_invariant2.default.call(void 0, actor, "workflow() requires an actor instance");
|
|
754
|
-
const
|
|
786
|
+
const actorGeneration = actor.generation;
|
|
787
|
+
const workflowInspector = getWorkflowInspector(
|
|
788
|
+
actor.id,
|
|
789
|
+
actorGeneration
|
|
790
|
+
);
|
|
755
791
|
const driver = new ActorWorkflowDriver(actor, runCtx);
|
|
756
792
|
const controlDriver = new ActorWorkflowControlDriver(actor, runCtx);
|
|
757
793
|
workflowInspector.setReplayFromStep(async (entryId) => {
|
|
@@ -784,7 +820,10 @@ function workflow(fn, options = {}) {
|
|
|
784
820
|
});
|
|
785
821
|
const handle = _workflowengine.runWorkflow.call(void 0,
|
|
786
822
|
actor.id,
|
|
787
|
-
async (ctx) =>
|
|
823
|
+
async (ctx) => {
|
|
824
|
+
throwIfGenerationLost(runCtx);
|
|
825
|
+
return await fn(new WorkflowContext(ctx, runCtx));
|
|
826
|
+
},
|
|
788
827
|
void 0,
|
|
789
828
|
driver,
|
|
790
829
|
{
|
|
@@ -793,7 +832,12 @@ function workflow(fn, options = {}) {
|
|
|
793
832
|
// compatible but not structurally assignable.
|
|
794
833
|
logger: runCtx.log,
|
|
795
834
|
onHistoryUpdated: workflowInspector.update,
|
|
796
|
-
onError: onError ? async (event) =>
|
|
835
|
+
onError: onError ? async (event) => {
|
|
836
|
+
if (actor.isLost === true) {
|
|
837
|
+
return;
|
|
838
|
+
}
|
|
839
|
+
await onError(runCtx, event);
|
|
840
|
+
} : void 0
|
|
797
841
|
}
|
|
798
842
|
);
|
|
799
843
|
workflowInspector.setGetState(async () => await handle.getState());
|
|
@@ -816,29 +860,33 @@ function workflow(fn, options = {}) {
|
|
|
816
860
|
if (shouldRethrowWorkflowError(error)) {
|
|
817
861
|
runCtx.log.error({
|
|
818
862
|
msg: "workflow run failed",
|
|
819
|
-
error:
|
|
863
|
+
error: _chunkYVVFOKEPcjs.stringifyError.call(void 0, error)
|
|
820
864
|
});
|
|
821
865
|
throw error;
|
|
822
866
|
}
|
|
823
867
|
runCtx.log.warn({
|
|
824
868
|
msg: "workflow failed and will sleep until woken",
|
|
825
|
-
error:
|
|
869
|
+
error: _chunkYVVFOKEPcjs.stringifyError.call(void 0, error)
|
|
826
870
|
});
|
|
827
871
|
} finally {
|
|
828
872
|
runCtx.abortSignal.removeEventListener("abort", onAbort);
|
|
829
873
|
}
|
|
830
874
|
}
|
|
831
|
-
return
|
|
875
|
+
return _chunkNGJWQMUUcjs.defineRunHandler.call(void 0, run, {
|
|
832
876
|
icon: "diagram-project",
|
|
833
877
|
inspectorKind: "workflow",
|
|
834
|
-
createInspector: ({ actorId, control }) => {
|
|
835
|
-
const
|
|
878
|
+
createInspector: ({ actorId, actorGeneration, control }) => {
|
|
879
|
+
const key = workflowInspectorKey(actorId, actorGeneration);
|
|
880
|
+
const workflowInspector = getWorkflowInspector(
|
|
881
|
+
actorId,
|
|
882
|
+
actorGeneration
|
|
883
|
+
);
|
|
836
884
|
workflowInspector.control = control;
|
|
837
885
|
return {
|
|
838
886
|
inspector: { workflow: workflowInspector.adapter },
|
|
839
887
|
dispose: () => {
|
|
840
|
-
if (workflowInspectors.get(
|
|
841
|
-
workflowInspectors.delete(
|
|
888
|
+
if (workflowInspectors.get(key) === workflowInspector) {
|
|
889
|
+
workflowInspectors.delete(key);
|
|
842
890
|
}
|
|
843
891
|
}
|
|
844
892
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/workflow/mod.cjs","../../../src/workflow/mod.ts","../../../src/workflow/constants.ts","../../../src/workflow/context.ts","../../../src/workflow/driver.ts"],"names":[],"mappings":"AAAA;AACE;AACF,yDAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B;AACE;AACA;AACF,yDAA8B;AAC9B;AACE;AACA;AACF,yDAA8B;AAC9B;AACE;AACF,yDAA8B;AAC9B;AACE;AACA;AACF,yDAA8B;AAC9B;AACA;ACrBA;AACC;AACA;AACA;AACA;AACA;AACA;AACA;AAEA;AACA;AACA;AAAA,2DAEM;AACP,4FAAsB;ADqBtB;AACA;AEpCO,IAAM,sBAAA,EAAwB,wCAAA;AFsCrC;AACA;AGmPA,IAAM,gBAAA,kBAAkB,MAAA,CAAO,0BAA0B,CAAA;AAQlD,IAAM,oBAAA,EAAN,MASL;AAAA,EACD,CAAA,MAAA;AAAA,EAUA,CAAA,OAAA,EAAU,IAAA;AAAA,EACV,CAAA,gBAAA;AAAA,EAEA,WAAA,CACC,MAAA,EAUA,gBAAA,EACC;AACD,IAAA,IAAA,CAAK,CAAA,OAAA,EAAU,MAAA;AACf,IAAA,IAAA,CAAK,CAAA,iBAAA,EAAoB,gBAAA;AAAA,EAC1B;AAAA,EAEA,CAAC,eAAe,CAAA,CAAA,EAAU;AACzB,IAAA,IAAA,CAAK,CAAA,OAAA,EAAU,KAAA;AAAA,EAChB;AAAA,EAEA,CAAA,YAAA,CAAc,OAAA,EAAuB;AACpC,IAAA,GAAA,CAAI,CAAC,IAAA,CAAK,CAAA,MAAA,EAAS;AAClB,MAAA,IAAA,CAAK,CAAA,gBAAA,CAAkB,CAAA;AACvB,MAAA,MAAM,IAAI,KAAA;AAAA,QACT,CAAA,EAAA;AACD,MAAA;AACD,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACE,IAAA;AACL,IAAA;AACD,EAAA;AAEI,EAAA;AACE,IAAA;AACL,IAAA;AACD,EAAA;AAEI,EAAA;AACE,IAAA;AACL,IAAA;AAGD,EAAA;AAEA,EAAA;AACM,IAAA;AACL,IAAA;AACD,EAAA;AAUA,EAAA;AACM,IAAA;AACA,IAAA;AACJ,MAAA;AACI,MAAA;AACL,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AASA,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AACA,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAOA,EAAA;AACM,IAAA;AACL,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAMA,EAAA;AACM,IAAA;AACA,IAAA;AACN,EAAA;AAEA,EAAA;AACM,IAAA;AACA,IAAA;AACN,EAAA;AACD;AASa;AAUZ,EAAA;AACA,EAAA;AAUA,EAAA;AAaM,IAAA;AACA,IAAA;AACN,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AAoBA,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AAuBA,IAAA;AACC,MAAA;AACA,MAAA;AAAqB,QAAA;AAErB,MAAA;AACD,IAAA;AAEA,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AACD,EAAA;AA6BM,EAAA;AA0BD,IAAA;AACH,MAAA;AACC,QAAA;AACD,MAAA;AACA,MAAA;AACA,MAAA;AAA8B,QAAA;AAE9B,MAAA;AACD,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACI,MAAA;AACH,MAAA;AACA,MAAA;AAGD,IAAA;AACA,IAAA;AACD,EAAA;AA6BM,EAAA;AA0BD,IAAA;AACH,MAAA;AACC,QAAA;AACD,MAAA;AACA,MAAA;AACA,MAAA;AAA8B,QAAA;AAE9B,MAAA;AACD,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACI,MAAA;AACH,MAAA;AACA,MAAA;AAGD,IAAA;AACA,IAAA;AACD,EAAA;AA6BM,EAAA;AA0BD,IAAA;AACH,MAAA;AACC,QAAA;AACD,MAAA;AACA,MAAA;AACA,MAAA;AAA8B,QAAA;AACjB,UAAA;AAAI,UAAA;AAEhB,QAAA;AACD,MAAA;AACD,IAAA;AACA,IAAA;AACA,IAAA;AACI,MAAA;AACH,MAAA;AACD,IAAA;AACA,IAAA;AACD,EAAA;AA+BM,EAAA;AA4BD,IAAA;AACH,MAAA;AACC,QAAA;AACD,MAAA;AACA,MAAA;AACA,MAAA;AAA8B,QAAA;AACjB,UAAA;AACX;AAAA;AAAA,UAAA;AAQD,QAAA;AACD,MAAA;AACD,IAAA;AACA,IAAA;AACA,IAAA;AACI,MAAA;AACH,MAAA;AAGD,IAAA;AACA,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEA,EAAA;AACC,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAyBM,EAAA;AAIL,IAAA;AACC,MAAA;AACC,QAAA;AACA,QAAA;AAAA,UAAA;AAGA,QAAA;AACA,MAAA;AACF,IAAA;AACA,IAAA;AAA8B,MAAA;AAE9B,IAAA;AACD,EAAA;AAEM,EAAA;AAiBL,IAAA;AACC,MAAA;AACA,MAAA;AAEC,IAAA;AACF,IAAA;AAA+B,MAAA;AAE/B,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEA,EAAA;AACC,IAAA;AACD,EAAA;AAEA,EAAA;AACC,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAKM,EAAA;AAaL,IAAA;AAWI,IAAA;AACH,MAAA;AACD,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AAEA,EAAA;AAGK,IAAA;AACA,IAAA;AACH,MAAA;AACD,IAAA;AACC,MAAA;AACD,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACI,MAAA;AACL,IAAA;AACD,EAAA;AAAA;AAGM,EAAA;AAcL,IAAA;AAUI,IAAA;AACH,MAAA;AACD,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAAA;AAAA;AAIA,EAAA;AACK,IAAA;AACH,MAAA;AACA,MAAA;AAKE,QAAA;AACF,MAAA;AACD,IAAA;AAEA,IAAA;AAEK,IAAA;AACH,MAAA;AACA,QAAA;AACC,UAAA;AACD,QAAA;AACC,UAAA;AAAuB,YAAA;AACjB,YAAA;AACL,UAAA;AAEF,QAAA;AACE,MAAA;AACJ,IAAA;AACD,EAAA;AAEA,EAAA;AAYC,IAAA;AACD,EAAA;AACD;AH1uBS;AACA;AIvVH;AAIA;AACA;AAeG;AACD,EAAA;AACR;AAES;AACF,EAAA;AACN,EAAA;AACK,IAAA;AACH,MAAA;AACA,MAAA;AACD,IAAA;AACD,EAAA;AACO,EAAA;AACR;AAES;AACJ,EAAA;AACH,IAAA;AACD,EAAA;AACI,EAAA;AACH,IAAA;AACD,EAAA;AACI,EAAA;AACH,IAAA;AACD,EAAA;AACI,EAAA;AACH,IAAA;AACA,IAAA;AACC,MAAA;AACC,QAAA;AACD,MAAA;AACA,MAAA;AACD,IAAA;AACA,IAAA;AACD,EAAA;AACM,EAAA;AACP;AAES;AAGF,EAAA;AAEL,EAAA;AAKA,IAAA;AACD,EAAA;AACO,EAAA;AACR;AAEM;AACL,EAAA;AAEA,EAAA;AACM,IAAA;AACJ,MAAA;AACC,QAAA;AACD,MAAA;AACD,IAAA;AACK,IAAA;AACN,EAAA;AAEM,EAAA;AAlGP,IAAA;AAmGE,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AACA,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AACD,EAAA;AAEM,EAAA;AACD,IAAA;AACJ,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACA,IAAA;AACI,IAAA;AACH,MAAA;AACC,QAAA;AACA,QAAA;AACA,QAAA;AACD,MAAA;AACA,MAAA;AACD,IAAA;AACA,IAAA;AACA,IAAA;AACC,MAAA;AAGD,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACA,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACA,MAAA;AACD,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACA,IAAA;AACA,IAAA;AAGE,MAAA;AACA,MAAA;AACC,IAAA;AACJ,EAAA;AAEM,EAAA;AACD,IAAA;AACJ,IAAA;AACC,MAAA;AACC,QAAA;AACC,UAAA;AAAS,YAAA;AACR,YAAA;AACmB,YAAA;AACnB,UAAA;AAEF,QAAA;AACA,MAAA;AACF,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACA,IAAA;AAEG,MAAA;AACA,MAAA;AACA,MAAA;AAEA,IAAA;AACA,MAAA;AACA,MAAA;AACD,IAAA;AACF,IAAA;AACC,MAAA;AACA,MAAA;AACA,IAAA;AACF,EAAA;AAEM,EAAA;AACL,IAAA;AAKC,MAAA;AACC,QAAA;AAAuB,UAAA;AACtB,UAAA;AAED,QAAA;AACC,UAAA;AAAS,YAAA;AACR,YAAA;AACA,UAAA;AAEF,QAAA;AACA,MAAA;AACF,IAAA;AACD,EAAA;AACD;AAEgB;AACT,EAAA;AACF,EAAA;AACA,EAAA;AACJ,EAAA;AACC,IAAA;AAKC,IAAA;AAIA,MAAA;AACA,MAAA;AACA,MAAA;AACD,IAAA;AACA,IAAA;AACA,IAAA;AACD,EAAA;AACI,EAAA;AACG,EAAA;AACR;AAES;AACJ,EAAA;AACJ,EAAA;AACK,IAAA;AACL,EAAA;AACO,EAAA;AACR;AAEM;AACL,EAAA;AACA,EAAA;AAEA,EAAA;AAIM,IAAA;AACA,IAAA;AACN,EAAA;AAEM,EAAA;AACL,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AAEM,EAAA;AAKL,IAAA;AACC,MAAA;AACC,QAAA;AAGA,QAAA;AACA,QAAA;AACA,QAAA;AACA,QAAA;AACD,MAAA;AACD,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACI,MAAA;AAED,QAAA;AACC,UAAA;AAAmB,YAAA;AACO,cAAA;AACxB,cAAA;AACA,YAAA;AACD,UAAA;AAEF,QAAA;AACD,MAAA;AAED,IAAA;AACH,EAAA;AAEM,EAAA;AAID,IAAA;AACA,IAAA;AACH,MAAA;AACD,IAAA;AACC,MAAA;AACD,IAAA;AAEA,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AACD;AAEa;AACH,iBAAA;AACA,kBAAA;AACA,EAAA;AACA,EAAA;AACT,EAAA;AACA,EAAA;AACA,EAAA;AAEA,EAAA;AAIM,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACJ,MAAA;AACD,IAAA;AACK,IAAA;AACN,EAAA;AAAA;AAGA,EAAA;AACC,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACD,IAAA;AAEJ,IAAA;AAAqB,MAAA;AAErB,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AAEM,EAAA;AAEL,IAAA;AACD,EAAA;AAEA,EAAA;AAIC,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AACD,EAAA;AACD;AAEM;AACC,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AAKL,IAAA;AACD,EAAA;AAEM,EAAA;AAIL,IAAA;AACD,EAAA;AACD;AAEa;AACH,kBAAA;AACA,kBAAA;AAET,EAAA;AACA,EAAA;AAEA,EAAA;AAIM,IAAA;AACA,IAAA;AACN,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACD,IAAA;AACH,MAAA;AACD,IAAA;AAEA,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEA,EAAA;AAIC,IAAA;AACD,EAAA;AACD;AJuNS;AACA;AC7pBA;AAeA;AAEP,EAAA;AAMA,IAAA;AACD,EAAA;AAGC,EAAA;AAIA,IAAA;AACD,EAAA;AAEO,EAAA;AACR;AAES;AACD,EAAA;AACN,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AACD,EAAA;AACD;AAES;AAEP,EAAA;AAIF;AAES;AAEP,EAAA;AAIF;AA2BgB;AA4CT,EAAA;AAIA,EAAA;AAEN,EAAA;AACK,IAAA;AACC,IAAA;AACJ,MAAA;AACA,MAAA;AACD,IAAA;AACA,IAAA;AACD,EAAA;AAEA,EAAA;AAYC,IAAA;AAKA,IAAA;AACA,IAAA;AAEA,IAAA;AACA,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACC,QAAA;AACD,MAAA;AAEA,MAAA;AACA,MAAA;AACA,MAAA;AACC,QAAA;AAAyB,UAAA;AACC,UAAA;AAExB,YAAA;AAAuB,cAAA;AAChB,cAAA;AACN,cAAA;AACA,cAAA;AACuB,YAAA;AAExB,YAAA;AACA,YAAA;AAA4C,UAAA;AAE9C,QAAA;AACD,MAAA;AACC,QAAA;AAIC,UAAA;AACD,QAAA;AACA,QAAA;AACD,MAAA;AACA,IAAA;AAED,IAAA;AACC,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACC,QAAA;AAAM;AAAA;AAGN,QAAA;AACA,QAAA;AACA,QAAA;AAGD,MAAA;AACD,IAAA;AACA,IAAA;AAEA,IAAA;AACC,MAAA;AACD,IAAA;AACI,IAAA;AACH,MAAA;AACD,IAAA;AACC,MAAA;AACC,QAAA;AACA,MAAA;AACF,IAAA;AAEI,IAAA;AACH,MAAA;AACD,IAAA;AAGC,MAAA;AACC,QAAA;AACD,MAAA;AAEA,MAAA;AACC,QAAA;AAAiB,UAAA;AACX,UAAA;AAEL,QAAA;AACD,QAAA;AACD,MAAA;AAEA,MAAA;AACC,QAAA;AACA,QAAA;AACA,MAAA;AACF,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AAEO,EAAA;AACN,IAAA;AACA,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACA,MAAA;AACC,QAAA;AACA,QAAA;AAGC,UAAA;AACC,YAAA;AAAiC,UAAA;AAEnC,QAAA;AACD,MAAA;AACD,IAAA;AACA,EAAA;AACF;ADghBS;AACA;AACA;AACA;AACA;AACA","file":"/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/workflow/mod.cjs","sourcesContent":[null,"import {\n\tCriticalError,\n\tEntryInProgressError,\n\tHistoryDivergedError,\n\tJoinError,\n\tRaceError,\n\tRollbackCheckpointError,\n\tRollbackError,\n\ttype RunWorkflowOptions,\n\treplayWorkflowFromStep,\n\trunWorkflow,\n\tStepExhaustedError,\n\ttype WorkflowErrorEvent,\n} from \"@rivetkit/workflow-engine\";\nimport invariant from \"invariant\";\nimport {\n\tACTOR_CONTEXT_INTERNAL_SYMBOL,\n\tdefineRunHandler,\n\ttype RunContext,\n\ttype RunControl,\n} from \"@/actor/config\";\nimport type { AnyStaticActorInstance } from \"@/actor/definition\";\nimport { isActorAbortedError, RivetError } from \"@/actor/errors\";\nimport type { EventSchemaConfig, QueueSchemaConfig } from \"@/actor/schema\";\nimport type { AnyDatabaseProvider } from \"@/common/database/config\";\nimport { stringifyError } from \"@/utils\";\nimport { WorkflowContext } from \"./context\";\nimport { ActorWorkflowControlDriver, ActorWorkflowDriver } from \"./driver\";\nimport { createWorkflowInspectorAdapter } from \"./inspector\";\n\nexport type {\n\tTryBlockCatchKind,\n\tTryBlockConfig,\n\tTryBlockFailure,\n\tTryBlockResult,\n\tTryStepCatchKind,\n\tTryStepConfig,\n\tTryStepFailure,\n\tTryStepResult,\n\tWorkflowError,\n\tWorkflowErrorEvent,\n} from \"@rivetkit/workflow-engine\";\nexport { Loop } from \"@rivetkit/workflow-engine\";\nexport {\n\ttype WorkflowBranchConfig,\n\ttype WorkflowBranchContextOf,\n\tWorkflowContext,\n\ttype WorkflowContextOf,\n\ttype WorkflowLoopConfig,\n\ttype WorkflowLoopContextOf,\n\ttype WorkflowStepConfig,\n\tWorkflowStepContext,\n\ttype WorkflowStepContextOf,\n\ttype WorkflowTryConfig,\n\ttype WorkflowTryStepConfig,\n} from \"./context\";\n\nfunction shouldRethrowWorkflowError(error: unknown): boolean {\n\tif (\n\t\terror instanceof CriticalError ||\n\t\terror instanceof JoinError ||\n\t\terror instanceof RaceError ||\n\t\terror instanceof RollbackError ||\n\t\terror instanceof StepExhaustedError\n\t) {\n\t\treturn false;\n\t}\n\n\tif (\n\t\terror instanceof EntryInProgressError ||\n\t\terror instanceof HistoryDivergedError ||\n\t\terror instanceof RollbackCheckpointError\n\t) {\n\t\treturn true;\n\t}\n\n\treturn true;\n}\n\nfunction workflowReplayInFlightError(): RivetError {\n\treturn new RivetError(\n\t\t\"actor\",\n\t\t\"workflow_in_flight\",\n\t\t\"Workflow replay is unavailable while the workflow is currently in flight.\",\n\t\t{\n\t\t\tpublic: true,\n\t\t\tstatusCode: 409,\n\t\t},\n\t);\n}\n\nfunction isWorkflowReplayBlockedByRunningEntry(error: unknown): boolean {\n\treturn (\n\t\terror instanceof Error &&\n\t\terror.message ===\n\t\t\t\"Cannot replay a workflow while a step is currently running\"\n\t);\n}\n\nfunction isRunHandlerUnavailable(error: unknown): boolean {\n\treturn (\n\t\terror instanceof RivetError &&\n\t\terror.group === \"actor\" &&\n\t\terror.code === \"run_handler_unavailable\"\n\t);\n}\n\nexport interface WorkflowOptions<\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig = Record<never, never>,\n\tTQueues extends QueueSchemaConfig = Record<never, never>,\n> {\n\tonError?: (\n\t\tctx: RunContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t\tevent: WorkflowErrorEvent,\n\t) => void | Promise<void>;\n}\n\nexport function workflow<\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig = Record<never, never>,\n\tTQueues extends QueueSchemaConfig = Record<never, never>,\n>(\n\tfn: (\n\t\tctx: WorkflowContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t) => Promise<unknown>,\n\toptions: WorkflowOptions<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t> = {},\n): (\n\tc: RunContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>,\n) => Promise<void> {\n\tconst onError = options.onError;\n\ttype WorkflowInspectorRegistration = ReturnType<\n\t\ttypeof createWorkflowInspectorAdapter\n\t> & { control?: RunControl };\n\tconst workflowInspectors = new Map<string, WorkflowInspectorRegistration>();\n\n\tfunction getWorkflowInspector(actorId: string) {\n\t\tlet workflowInspector = workflowInspectors.get(actorId);\n\t\tif (!workflowInspector) {\n\t\t\tworkflowInspector = createWorkflowInspectorAdapter();\n\t\t\tworkflowInspectors.set(actorId, workflowInspector);\n\t\t}\n\t\treturn workflowInspector;\n\t}\n\n\tasync function run(\n\t\trunCtx: RunContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<void> {\n\t\tconst actor = (\n\t\t\trunCtx as unknown as {\n\t\t\t\t[ACTOR_CONTEXT_INTERNAL_SYMBOL]?: AnyStaticActorInstance;\n\t\t\t}\n\t\t)[ACTOR_CONTEXT_INTERNAL_SYMBOL];\n\t\tinvariant(actor, \"workflow() requires an actor instance\");\n\t\tconst workflowInspector = getWorkflowInspector(actor.id);\n\n\t\tconst driver = new ActorWorkflowDriver(actor, runCtx);\n\t\tconst controlDriver = new ActorWorkflowControlDriver(actor, runCtx);\n\t\tworkflowInspector.setReplayFromStep(async (entryId) => {\n\t\t\tconst workflowState = await workflowInspector.adapter.getState();\n\t\t\tif (workflowState === \"pending\" || workflowState === \"running\") {\n\t\t\t\tthrow workflowReplayInFlightError();\n\t\t\t}\n\n\t\t\tconst control = workflowInspector.control;\n\t\t\tinvariant(control, \"workflow Inspector control is not initialized\");\n\t\t\ttry {\n\t\t\t\treturn await control.run.withInactive(\n\t\t\t\t\t{ restartOnSuccess: true },\n\t\t\t\t\tasync () => {\n\t\t\t\t\t\tconst snapshot = await replayWorkflowFromStep(\n\t\t\t\t\t\t\tactor.id,\n\t\t\t\t\t\t\tcontrolDriver,\n\t\t\t\t\t\t\tentryId,\n\t\t\t\t\t\t\t{ scheduleAlarm: false },\n\t\t\t\t\t\t);\n\t\t\t\t\t\tworkflowInspector.update(snapshot);\n\t\t\t\t\t\treturn workflowInspector.adapter.getHistory();\n\t\t\t\t\t},\n\t\t\t\t);\n\t\t\t} catch (error) {\n\t\t\t\tif (\n\t\t\t\t\tisWorkflowReplayBlockedByRunningEntry(error) ||\n\t\t\t\t\tisRunHandlerUnavailable(error)\n\t\t\t\t) {\n\t\t\t\t\tthrow workflowReplayInFlightError();\n\t\t\t\t}\n\t\t\t\tthrow error;\n\t\t\t}\n\t\t});\n\n\t\tconst handle = runWorkflow(\n\t\t\tactor.id,\n\t\t\tasync (ctx) => await fn(new WorkflowContext(ctx, runCtx)),\n\t\t\tundefined,\n\t\t\tdriver,\n\t\t\t{\n\t\t\t\tmode: \"live\",\n\t\t\t\t// The actor logger and the engine's pino logger are runtime\n\t\t\t\t// compatible but not structurally assignable.\n\t\t\t\tlogger: runCtx.log as RunWorkflowOptions[\"logger\"],\n\t\t\t\tonHistoryUpdated: workflowInspector.update,\n\t\t\t\tonError: onError\n\t\t\t\t\t? async (event) => await onError(runCtx, event)\n\t\t\t\t\t: undefined,\n\t\t\t},\n\t\t);\n\t\tworkflowInspector.setGetState(async () => await handle.getState());\n\n\t\tconst onAbort = () => {\n\t\t\thandle.evict();\n\t\t};\n\t\tif (runCtx.abortSignal.aborted) {\n\t\t\tonAbort();\n\t\t} else {\n\t\t\trunCtx.abortSignal.addEventListener(\"abort\", onAbort, {\n\t\t\t\tonce: true,\n\t\t\t});\n\t\t}\n\n\t\ttry {\n\t\t\tawait handle.result;\n\t\t} catch (error) {\n\t\t\t// `abortSignal.aborted` is delivered on a separate async hop and\n\t\t\t// races the rejection, so detect the sleep abort structurally too.\n\t\t\tif (runCtx.abortSignal.aborted || isActorAbortedError(error)) {\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tif (shouldRethrowWorkflowError(error)) {\n\t\t\t\trunCtx.log.error({\n\t\t\t\t\tmsg: \"workflow run failed\",\n\t\t\t\t\terror: stringifyError(error),\n\t\t\t\t});\n\t\t\t\tthrow error;\n\t\t\t}\n\n\t\t\trunCtx.log.warn({\n\t\t\t\tmsg: \"workflow failed and will sleep until woken\",\n\t\t\t\terror: stringifyError(error),\n\t\t\t});\n\t\t} finally {\n\t\t\trunCtx.abortSignal.removeEventListener(\"abort\", onAbort);\n\t\t}\n\t}\n\n\treturn defineRunHandler(run, {\n\t\ticon: \"diagram-project\",\n\t\tinspectorKind: \"workflow\",\n\t\tcreateInspector: ({ actorId, control }) => {\n\t\t\tconst workflowInspector = getWorkflowInspector(actorId);\n\t\t\tworkflowInspector.control = control;\n\t\t\treturn {\n\t\t\t\tinspector: { workflow: workflowInspector.adapter },\n\t\t\t\tdispose: () => {\n\t\t\t\t\t// Do not let a stale disposer remove a newly-created adapter for\n\t\t\t\t\t// the same actor id.\n\t\t\t\t\tif (workflowInspectors.get(actorId) === workflowInspector) {\n\t\t\t\t\t\tworkflowInspectors.delete(actorId);\n\t\t\t\t\t}\n\t\t\t\t},\n\t\t\t};\n\t\t},\n\t});\n}\n","export const WORKFLOW_GUARD_KV_KEY = \"__rivet_actor_workflow_guard_triggered\";\n","import type {\n\tBranchConfig,\n\tBranchOutput,\n\tEntryKindType,\n\tLoopConfig,\n\tLoopResult,\n\tStepConfig,\n\tTryBlockConfig,\n\tTryBlockResult,\n\tTryStepConfig,\n\tTryStepResult,\n\tWorkflowContextInterface,\n\tWorkflowQueueMessage,\n} from \"@rivetkit/workflow-engine\";\nimport type {\n\tQueueFilterName,\n\tQueueNextBatchOptions,\n\tQueueNextOptions,\n\tQueueResultMessageForName,\n\tRunContext,\n} from \"@/actor/config\";\nimport type {\n\tAnyActorDefinition,\n\tBaseActorDefinition,\n} from \"@/actor/definition\";\nimport type {\n\tEventSchemaConfig,\n\tInferEventArgs,\n\tInferSchemaMap,\n\tQueueSchemaConfig,\n} from \"@/actor/schema\";\nimport type { Client } from \"@/client/client\";\nimport type {\n\tAnyDatabaseProvider,\n\tInferDatabaseClient,\n} from \"@/common/database/config\";\nimport type { Registry } from \"@/registry\";\nimport { WORKFLOW_GUARD_KV_KEY } from \"./constants\";\n\ntype WorkflowActorQueueNextOptions<\n\tTName extends string,\n\tTCompletable extends boolean,\n> = Omit<QueueNextOptions<TName, TCompletable>, \"signal\">;\n\ntype WorkflowActorQueueNextOptionsFallback<TCompletable extends boolean> = Omit<\n\tQueueNextOptions<string, TCompletable>,\n\t\"signal\"\n>;\n\ntype WorkflowActorQueueNextBatchOptions<\n\tTName extends string,\n\tTCompletable extends boolean,\n> = Omit<QueueNextBatchOptions<TName, TCompletable>, \"signal\">;\n\ntype WorkflowActorQueueNextBatchOptionsFallback<TCompletable extends boolean> =\n\tOmit<QueueNextBatchOptions<string, TCompletable>, \"signal\">;\n\n// Step run callbacks receive a WorkflowStepContext, which is the only place\n// actor data (state/db/vars/client) may be touched.\ntype WorkflowStepRun<\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = (\n\tstep: WorkflowStepContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>,\n) => Promise<T>;\n\n// Step rollback callbacks compensate a committed step, so they also run with a\n// WorkflowStepContext to mutate actor data.\ntype WorkflowStepRollback<\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = (\n\tstep: WorkflowStepContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>,\n\toutput: T,\n) => Promise<void>;\n\n// Orchestration callbacks (try/loop/race/join) receive a WorkflowContext,\n// because inside them you sequence further steps rather than touch actor data.\ntype WorkflowContextRun<\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = (\n\tctx: WorkflowContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>,\n) => Promise<T>;\n\nexport type WorkflowStepConfig<\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = Omit<StepConfig<T>, \"run\" | \"rollback\"> & {\n\trun: WorkflowStepRun<\n\t\tT,\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n\trollback?: WorkflowStepRollback<\n\t\tT,\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n};\n\nexport type WorkflowTryStepConfig<\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = Omit<TryStepConfig<T>, \"run\" | \"rollback\"> & {\n\trun: WorkflowStepRun<\n\t\tT,\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n\trollback?: WorkflowStepRollback<\n\t\tT,\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n};\n\nexport type WorkflowTryConfig<\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = Omit<TryBlockConfig<T>, \"run\"> & {\n\trun: WorkflowContextRun<\n\t\tT,\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n};\n\nexport type WorkflowLoopConfig<\n\tS,\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = Omit<LoopConfig<S, T>, \"run\"> & {\n\trun: (\n\t\tctx: WorkflowContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t\tstate: S,\n\t) => Promise<\n\t\tLoopResult<S, T> | (S extends undefined ? undefined | void : never)\n\t>;\n};\n\nexport type WorkflowBranchConfig<\n\tTOutput,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = {\n\trun: WorkflowContextRun<\n\t\tTOutput,\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n};\n\n// Marks a step context inactive once its step has finished. Module-private so it\n// never appears on the public surface.\nconst DEACTIVATE_STEP = Symbol(\"workflow.step.deactivate\");\n\n/**\n * The context handed to a workflow step (`step` / `tryStep` callbacks). This is\n * the only scope where actor data (state, vars, db, client) and side effects\n * (broadcast, queue.send) are reachable. It is valid only while its step is\n * executing; using it after the step resolves throws.\n */\nexport class WorkflowStepContext<\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig = Record<never, never>,\n\tTQueues extends QueueSchemaConfig = Record<never, never>,\n> {\n\t#runCtx: RunContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n\t#active = true;\n\t#onGuardViolation: () => void;\n\n\tconstructor(\n\t\trunCtx: RunContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t\tonGuardViolation: () => void,\n\t) {\n\t\tthis.#runCtx = runCtx;\n\t\tthis.#onGuardViolation = onGuardViolation;\n\t}\n\n\t[DEACTIVATE_STEP](): void {\n\t\tthis.#active = false;\n\t}\n\n\t#ensureActive(feature: string): void {\n\t\tif (!this.#active) {\n\t\t\tthis.#onGuardViolation();\n\t\t\tthrow new Error(\n\t\t\t\t`${feature} is only available inside workflow steps`,\n\t\t\t);\n\t\t}\n\t}\n\n\tget actorId(): string {\n\t\treturn this.#runCtx.actorId;\n\t}\n\n\tget name(): string {\n\t\treturn this.#runCtx.name;\n\t}\n\n\tget key(): string[] {\n\t\treturn this.#runCtx.key;\n\t}\n\n\tget log() {\n\t\treturn this.#runCtx.log;\n\t}\n\n\tget abortSignal(): AbortSignal {\n\t\treturn this.#runCtx.abortSignal;\n\t}\n\n\tget state(): TState extends never ? never : TState {\n\t\tthis.#ensureActive(\"state\");\n\t\treturn this.#runCtx.state as TState extends never ? never : TState;\n\t}\n\n\tget vars(): TVars extends never ? never : TVars {\n\t\tthis.#ensureActive(\"vars\");\n\t\treturn this.#runCtx.vars as TVars extends never ? never : TVars;\n\t}\n\n\tget db(): TDatabase extends never ? never : InferDatabaseClient<TDatabase> {\n\t\tthis.#ensureActive(\"db\");\n\t\treturn this.#runCtx.db as TDatabase extends never\n\t\t\t? never\n\t\t\t: InferDatabaseClient<TDatabase>;\n\t}\n\n\tclient<R extends Registry<any> = Registry<any>>(): Client<R> {\n\t\tthis.#ensureActive(\"client\");\n\t\treturn this.#runCtx.client<R>();\n\t}\n\n\tbroadcast<K extends keyof TEvents & string>(\n\t\tname: K,\n\t\t...args: InferEventArgs<InferSchemaMap<TEvents>[K]>\n\t): void;\n\tbroadcast(\n\t\tname: keyof TEvents extends never ? string : never,\n\t\t...args: Array<unknown>\n\t): void;\n\tbroadcast(name: string, ...args: Array<unknown>): void {\n\t\tthis.#ensureActive(\"broadcast\");\n\t\tthis.#runCtx.broadcast(\n\t\t\tname as never,\n\t\t\t...(args as unknown[] as never[]),\n\t\t);\n\t}\n\n\tget queue() {\n\t\tconst self = this;\n\t\tfunction send<K extends keyof TQueues & string>(\n\t\t\tname: K,\n\t\t\tbody: InferSchemaMap<TQueues>[K],\n\t\t): Promise<void>;\n\t\tfunction send(\n\t\t\tname: keyof TQueues extends never ? string : never,\n\t\t\tbody: unknown,\n\t\t): Promise<void>;\n\t\tasync function send(name: string, body: unknown): Promise<void> {\n\t\t\tself.#ensureActive(\"queue.send\");\n\t\t\tawait self.#runCtx.queue.send(name as never, body as never);\n\t\t}\n\t\treturn { send };\n\t}\n\n\t/**\n\t * Holds the actor awake for the duration of the provided promise. The actor\n\t * cannot idle-sleep or finalize the sleep grace period until the promise\n\t * settles.\n\t */\n\tkeepAwake<T>(promise: Promise<T>): Promise<T> {\n\t\tthis.#ensureActive(\"keepAwake\");\n\t\treturn this.#runCtx.keepAwake(promise);\n\t}\n\n\t/**\n\t * Registers a promise that the sleep grace period will wait on. Use this for\n\t * best-effort flush/cleanup work that may complete inside the grace window.\n\t */\n\twaitUntil(promise: Promise<void>): void {\n\t\tthis.#ensureActive(\"waitUntil\");\n\t\tthis.#runCtx.waitUntil(promise);\n\t}\n\n\tdestroy(): void {\n\t\tthis.#ensureActive(\"destroy\");\n\t\tthis.#runCtx.destroy();\n\t}\n}\n\n/**\n * The context handed to the workflow function and to orchestration callbacks\n * (`try` / `loop` / `race` / `join`). It exposes the deterministic, replayable\n * workflow primitives (step, sleep, queue waits, control flow). It deliberately\n * does NOT expose actor data; reach `state`, `db`, `vars`, `client`, and\n * `broadcast` through the step context passed to `step` / `tryStep`.\n */\nexport class WorkflowContext<\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig = Record<never, never>,\n\tTQueues extends QueueSchemaConfig = Record<never, never>,\n> {\n\t#inner: WorkflowContextInterface;\n\t#runCtx: RunContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n\tconstructor(\n\t\tinner: WorkflowContextInterface,\n\t\trunCtx: RunContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t) {\n\t\tthis.#inner = inner;\n\t\tthis.#runCtx = runCtx;\n\t}\n\n\tget workflowId(): string {\n\t\treturn this.#inner.workflowId;\n\t}\n\n\tget abortSignal(): AbortSignal {\n\t\treturn this.#inner.abortSignal;\n\t}\n\n\tget actorId(): string {\n\t\treturn this.#runCtx.actorId;\n\t}\n\n\tget name(): string {\n\t\treturn this.#runCtx.name;\n\t}\n\n\tget key(): string[] {\n\t\treturn this.#runCtx.key;\n\t}\n\n\tget log() {\n\t\treturn this.#runCtx.log;\n\t}\n\n\tget queue() {\n\t\tconst self = this;\n\t\tfunction next<\n\t\t\tconst TName extends QueueFilterName<TQueues>,\n\t\t\tconst TCompletable extends boolean = false,\n\t\t>(\n\t\t\tname: string,\n\t\t\topts?: WorkflowActorQueueNextOptions<TName, TCompletable>,\n\t\t): Promise<QueueResultMessageForName<TQueues, TName, TCompletable>>;\n\t\tfunction next<const TCompletable extends boolean = false>(\n\t\t\tname: string,\n\t\t\topts?: WorkflowActorQueueNextOptionsFallback<TCompletable>,\n\t\t): Promise<\n\t\t\tQueueResultMessageForName<\n\t\t\t\tTQueues,\n\t\t\t\tQueueFilterName<TQueues>,\n\t\t\t\tTCompletable\n\t\t\t>\n\t\t>;\n\t\t// The implementation signature stays broad so the schema-typed public\n\t\t// overloads above remain compatible with it.\n\t\tasync function next(name: string, opts?: any): Promise<any> {\n\t\t\tconst message = await self.#inner.queue.next(name, opts);\n\t\t\treturn self.#toActorQueueMessage(message);\n\t\t}\n\n\t\tfunction nextBatch<\n\t\t\tconst TName extends QueueFilterName<TQueues>,\n\t\t\tconst TCompletable extends boolean = false,\n\t\t>(\n\t\t\tname: string,\n\t\t\topts?: WorkflowActorQueueNextBatchOptions<TName, TCompletable>,\n\t\t): Promise<\n\t\t\tArray<QueueResultMessageForName<TQueues, TName, TCompletable>>\n\t\t>;\n\t\tfunction nextBatch<const TCompletable extends boolean = false>(\n\t\t\tname: string,\n\t\t\topts?: WorkflowActorQueueNextBatchOptionsFallback<TCompletable>,\n\t\t): Promise<\n\t\t\tArray<\n\t\t\t\tQueueResultMessageForName<\n\t\t\t\t\tTQueues,\n\t\t\t\t\tQueueFilterName<TQueues>,\n\t\t\t\t\tTCompletable\n\t\t\t\t>\n\t\t\t>\n\t\t>;\n\t\tasync function nextBatch(name: string, opts?: any): Promise<any> {\n\t\t\tconst messages = await self.#inner.queue.nextBatch(name, opts);\n\t\t\treturn messages.map((message) =>\n\t\t\t\tself.#toActorQueueMessage(message),\n\t\t\t);\n\t\t}\n\n\t\treturn {\n\t\t\tnext,\n\t\t\tnextBatch,\n\t\t};\n\t}\n\n\tstep<T>(\n\t\tname: string,\n\t\trun: WorkflowStepRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<T>;\n\tstep<T>(\n\t\tconfig: WorkflowStepConfig<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<T>;\n\tasync step<T>(\n\t\tnameOrConfig:\n\t\t\t| string\n\t\t\t| WorkflowStepConfig<\n\t\t\t\t\tT,\n\t\t\t\t\tTState,\n\t\t\t\t\tTConnParams,\n\t\t\t\t\tTConnState,\n\t\t\t\t\tTVars,\n\t\t\t\t\tTInput,\n\t\t\t\t\tTDatabase,\n\t\t\t\t\tTEvents,\n\t\t\t\t\tTQueues\n\t\t\t >,\n\t\trun?: WorkflowStepRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<T> {\n\t\tif (typeof nameOrConfig === \"string\") {\n\t\t\tif (!run) {\n\t\t\t\tthrow new Error(\"Step run function missing\");\n\t\t\t}\n\t\t\tconst stepRun = run;\n\t\t\treturn await this.#wrapActive(() =>\n\t\t\t\tthis.#inner.step(nameOrConfig, () => this.#runStep(stepRun)),\n\t\t\t);\n\t\t}\n\t\tconst stepConfig = nameOrConfig;\n\t\tconst rollback = stepConfig.rollback;\n\t\tconst config: StepConfig<T> = {\n\t\t\t...stepConfig,\n\t\t\trun: () => this.#runStep(stepConfig.run),\n\t\t\trollback: rollback\n\t\t\t\t? (_ctx, output) => this.#runRollback(rollback, output)\n\t\t\t\t: undefined,\n\t\t};\n\t\treturn await this.#wrapActive(() => this.#inner.step(config));\n\t}\n\n\ttryStep<T>(\n\t\tname: string,\n\t\trun: WorkflowStepRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<TryStepResult<T>>;\n\ttryStep<T>(\n\t\tconfig: WorkflowTryStepConfig<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<TryStepResult<T>>;\n\tasync tryStep<T>(\n\t\tnameOrConfig:\n\t\t\t| string\n\t\t\t| WorkflowTryStepConfig<\n\t\t\t\t\tT,\n\t\t\t\t\tTState,\n\t\t\t\t\tTConnParams,\n\t\t\t\t\tTConnState,\n\t\t\t\t\tTVars,\n\t\t\t\t\tTInput,\n\t\t\t\t\tTDatabase,\n\t\t\t\t\tTEvents,\n\t\t\t\t\tTQueues\n\t\t\t >,\n\t\trun?: WorkflowStepRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<TryStepResult<T>> {\n\t\tif (typeof nameOrConfig === \"string\") {\n\t\t\tif (!run) {\n\t\t\t\tthrow new Error(\"Step run function missing\");\n\t\t\t}\n\t\t\tconst stepRun = run;\n\t\t\treturn await this.#wrapActive(() =>\n\t\t\t\tthis.#inner.tryStep(nameOrConfig, () => this.#runStep(stepRun)),\n\t\t\t);\n\t\t}\n\t\tconst stepConfig = nameOrConfig;\n\t\tconst rollback = stepConfig.rollback;\n\t\tconst config: TryStepConfig<T> = {\n\t\t\t...stepConfig,\n\t\t\trun: () => this.#runStep(stepConfig.run),\n\t\t\trollback: rollback\n\t\t\t\t? (_ctx, output) => this.#runRollback(rollback, output)\n\t\t\t\t: undefined,\n\t\t};\n\t\treturn await this.#wrapActive(() => this.#inner.tryStep(config));\n\t}\n\n\ttry<T>(\n\t\tname: string,\n\t\trun: WorkflowContextRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<TryBlockResult<T>>;\n\ttry<T>(\n\t\tconfig: WorkflowTryConfig<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<TryBlockResult<T>>;\n\tasync try<T>(\n\t\tnameOrConfig:\n\t\t\t| string\n\t\t\t| WorkflowTryConfig<\n\t\t\t\t\tT,\n\t\t\t\t\tTState,\n\t\t\t\t\tTConnParams,\n\t\t\t\t\tTConnState,\n\t\t\t\t\tTVars,\n\t\t\t\t\tTInput,\n\t\t\t\t\tTDatabase,\n\t\t\t\t\tTEvents,\n\t\t\t\t\tTQueues\n\t\t\t >,\n\t\trun?: WorkflowContextRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<TryBlockResult<T>> {\n\t\tif (typeof nameOrConfig === \"string\") {\n\t\t\tif (!run) {\n\t\t\t\tthrow new Error(\"Try run function missing\");\n\t\t\t}\n\t\t\tconst tryRun = run;\n\t\t\treturn await this.#wrapActive(() =>\n\t\t\t\tthis.#inner.try(nameOrConfig, async (ctx) =>\n\t\t\t\t\ttryRun(this.#createChildContext(ctx)),\n\t\t\t\t),\n\t\t\t);\n\t\t}\n\t\tconst tryConfig = nameOrConfig;\n\t\tconst config: TryBlockConfig<T> = {\n\t\t\t...tryConfig,\n\t\t\trun: async (ctx) => tryConfig.run(this.#createChildContext(ctx)),\n\t\t};\n\t\treturn await this.#wrapActive(() => this.#inner.try(config));\n\t}\n\n\tloop<T>(\n\t\tname: string,\n\t\trun: (\n\t\t\tctx: WorkflowContext<\n\t\t\t\tTState,\n\t\t\t\tTConnParams,\n\t\t\t\tTConnState,\n\t\t\t\tTVars,\n\t\t\t\tTInput,\n\t\t\t\tTDatabase,\n\t\t\t\tTEvents,\n\t\t\t\tTQueues\n\t\t\t>,\n\t\t) => Promise<LoopResult<undefined, T> | undefined | void>,\n\t): Promise<T>;\n\tloop<S, T>(\n\t\tconfig: WorkflowLoopConfig<\n\t\t\tS,\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<T>;\n\tasync loop(\n\t\tnameOrConfig:\n\t\t\t| string\n\t\t\t| WorkflowLoopConfig<\n\t\t\t\t\tany,\n\t\t\t\t\tany,\n\t\t\t\t\tTState,\n\t\t\t\t\tTConnParams,\n\t\t\t\t\tTConnState,\n\t\t\t\t\tTVars,\n\t\t\t\t\tTInput,\n\t\t\t\t\tTDatabase,\n\t\t\t\t\tTEvents,\n\t\t\t\t\tTQueues\n\t\t\t >,\n\t\trun?: (\n\t\t\tctx: WorkflowContext<\n\t\t\t\tTState,\n\t\t\t\tTConnParams,\n\t\t\t\tTConnState,\n\t\t\t\tTVars,\n\t\t\t\tTInput,\n\t\t\t\tTDatabase,\n\t\t\t\tTEvents,\n\t\t\t\tTQueues\n\t\t\t>,\n\t\t) => Promise<LoopResult<undefined, any> | undefined | void>,\n\t): Promise<any> {\n\t\tif (typeof nameOrConfig === \"string\") {\n\t\t\tif (!run) {\n\t\t\t\tthrow new Error(\"Loop run function missing\");\n\t\t\t}\n\t\t\tconst loopRun = run;\n\t\t\treturn await this.#wrapActive(() =>\n\t\t\t\tthis.#inner.loop(\n\t\t\t\t\tnameOrConfig,\n\t\t\t\t\t// A void return (no explicit Loop result) is undefined at\n\t\t\t\t\t// runtime, which the engine treats as continue.\n\t\t\t\t\tasync (\n\t\t\t\t\t\tctx,\n\t\t\t\t\t): Promise<LoopResult<undefined, any> | undefined> =>\n\t\t\t\t\t\t(await loopRun(this.#createChildContext(ctx))) ??\n\t\t\t\t\t\tundefined,\n\t\t\t\t),\n\t\t\t);\n\t\t}\n\t\tconst loopConfig = nameOrConfig;\n\t\tconst wrapped: LoopConfig<any, any> = {\n\t\t\t...loopConfig,\n\t\t\trun: (async (ctx, state) =>\n\t\t\t\t(await loopConfig.run(this.#createChildContext(ctx), state)) ??\n\t\t\t\tundefined) as LoopConfig<any, any>[\"run\"],\n\t\t};\n\t\treturn await this.#wrapActive(() => this.#inner.loop(wrapped));\n\t}\n\n\tsleep(name: string, durationMs: number): Promise<void> {\n\t\treturn this.#inner.sleep(name, durationMs);\n\t}\n\n\tsleepUntil(name: string, timestampMs: number): Promise<void> {\n\t\treturn this.#inner.sleepUntil(name, timestampMs);\n\t}\n\n\tasync rollbackCheckpoint(name: string): Promise<void> {\n\t\tawait this.#wrapActive(() => this.#inner.rollbackCheckpoint(name));\n\t}\n\n\tasync join<\n\t\tT extends Record<\n\t\t\tstring,\n\t\t\tWorkflowBranchConfig<\n\t\t\t\tunknown,\n\t\t\t\tTState,\n\t\t\t\tTConnParams,\n\t\t\t\tTConnState,\n\t\t\t\tTVars,\n\t\t\t\tTInput,\n\t\t\t\tTDatabase,\n\t\t\t\tTEvents,\n\t\t\t\tTQueues\n\t\t\t>\n\t\t>,\n\t>(\n\t\tname: string,\n\t\tbranches: T,\n\t): Promise<{ [K in keyof T]: Awaited<ReturnType<T[K][\"run\"]>> }>;\n\tasync join<T extends Record<string, BranchConfig<unknown>>>(\n\t\tname: string,\n\t\tbranches: T,\n\t): Promise<{ [K in keyof T]: BranchOutput<T[K]> }>;\n\tasync join(\n\t\tname: string,\n\t\tbranches: Record<string, { run: (ctx: any) => Promise<unknown> }>,\n\t) {\n\t\tconst wrappedBranches = Object.fromEntries(\n\t\t\tObject.entries(branches).map(([key, branch]) => [\n\t\t\t\tkey,\n\t\t\t\t{\n\t\t\t\t\trun: async (ctx: WorkflowContextInterface) =>\n\t\t\t\t\t\tbranch.run(this.#createChildContext(ctx)),\n\t\t\t\t},\n\t\t\t]),\n\t\t) as Record<string, BranchConfig<unknown>>;\n\t\treturn await this.#wrapActive(() =>\n\t\t\tthis.#inner.join(name, wrappedBranches),\n\t\t);\n\t}\n\n\tasync race<T>(\n\t\tname: string,\n\t\tbranches: Array<{\n\t\t\tname: string;\n\t\t\trun: WorkflowContextRun<\n\t\t\t\tT,\n\t\t\t\tTState,\n\t\t\t\tTConnParams,\n\t\t\t\tTConnState,\n\t\t\t\tTVars,\n\t\t\t\tTInput,\n\t\t\t\tTDatabase,\n\t\t\t\tTEvents,\n\t\t\t\tTQueues\n\t\t\t>;\n\t\t}>,\n\t): Promise<{ winner: string; value: T }> {\n\t\tconst wrappedBranches = branches.map((branch) => ({\n\t\t\tname: branch.name,\n\t\t\trun: (ctx: WorkflowContextInterface) =>\n\t\t\t\tbranch.run(this.#createChildContext(ctx)),\n\t\t}));\n\t\treturn (await this.#wrapActive(() =>\n\t\t\tthis.#inner.race(name, wrappedBranches),\n\t\t)) as { winner: string; value: T };\n\t}\n\n\tasync removed(name: string, originalType: EntryKindType): Promise<void> {\n\t\tawait this.#wrapActive(() => this.#inner.removed(name, originalType));\n\t}\n\n\tgetVersion(name: string, latest: number): Promise<number> {\n\t\treturn this.#wrapActive(() => this.#inner.getVersion(name, latest));\n\t}\n\n\tisEvicted(): boolean {\n\t\treturn this.#inner.isEvicted();\n\t}\n\n\t// Runs a user step body inside a fresh step context and deactivates the step\n\t// context once the body settles. Workflows are roll-forward only, so state and\n\t// vars mutations made before a failure always remain visible.\n\tasync #runStep<T>(\n\t\trun: WorkflowStepRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<T> {\n\t\tconst stepCtx = new WorkflowStepContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>(this.#runCtx, () => this.#markGuardTriggered());\n\n\t\ttry {\n\t\t\treturn await run(stepCtx);\n\t\t} finally {\n\t\t\tstepCtx[DEACTIVATE_STEP]();\n\t\t}\n\t}\n\n\t#toActorQueueMessage<T>(\n\t\tmessage: WorkflowQueueMessage<T>,\n\t): WorkflowQueueMessage<T> & { id: bigint } {\n\t\tlet id: bigint;\n\t\ttry {\n\t\t\tid = BigInt(message.id);\n\t\t} catch {\n\t\t\tthrow new Error(`Invalid queue message id \"${message.id}\"`);\n\t\t}\n\t\treturn {\n\t\t\tid,\n\t\t\tname: message.name,\n\t\t\tbody: message.body,\n\t\t\tcreatedAt: message.createdAt,\n\t\t\t...(message.complete ? { complete: message.complete } : {}),\n\t\t};\n\t}\n\n\t// Runs a step rollback compensation with an active step context.\n\tasync #runRollback<T>(\n\t\trollback: WorkflowStepRollback<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t\toutput: T,\n\t): Promise<void> {\n\t\tconst stepCtx = new WorkflowStepContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>(this.#runCtx, () => this.#markGuardTriggered());\n\t\ttry {\n\t\t\tawait rollback(stepCtx, output);\n\t\t} finally {\n\t\t\tstepCtx[DEACTIVATE_STEP]();\n\t\t}\n\t}\n\n\tasync #wrapActive<T>(run: () => Promise<T>): Promise<T> {\n\t\treturn await this.#runCtx.internalKeepAwake(run);\n\t}\n\n\t// Records that a step context was used outside its step. Mirrors the value\n\t// onto actor state and a KV flag so callers can observe the violation.\n\t#markGuardTriggered(): void {\n\t\ttry {\n\t\t\tconst state = this.#runCtx.state as Record<string, unknown>;\n\t\t\tif (\n\t\t\t\tstate &&\n\t\t\t\ttypeof state === \"object\" &&\n\t\t\t\t\"guardTriggered\" in state\n\t\t\t) {\n\t\t\t\t(state as Record<string, unknown>).guardTriggered = true;\n\t\t\t}\n\t\t} catch {\n\t\t\t// Ignore if state is unavailable\n\t\t}\n\n\t\tthis.#runCtx.waitUntil(\n\t\t\t(async () => {\n\t\t\t\ttry {\n\t\t\t\t\tawait this.#runCtx.kv.put(WORKFLOW_GUARD_KV_KEY, \"true\");\n\t\t\t\t} catch (error) {\n\t\t\t\t\tthis.#runCtx.log.error({\n\t\t\t\t\t\tmsg: \"failed to persist workflow guard flag\",\n\t\t\t\t\t\terror,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t})(),\n\t\t);\n\t}\n\n\t#createChildContext(\n\t\tctx: WorkflowContextInterface,\n\t): WorkflowContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t> {\n\t\treturn new WorkflowContext(ctx, this.#runCtx);\n\t}\n}\n\nexport type WorkflowContextOf<AD extends AnyActorDefinition> =\n\tAD extends BaseActorDefinition<\n\t\tinfer S,\n\t\tinfer CP,\n\t\tinfer CS,\n\t\tinfer V,\n\t\tinfer I,\n\t\tinfer DB extends AnyDatabaseProvider,\n\t\tinfer E extends EventSchemaConfig,\n\t\tinfer Q extends QueueSchemaConfig,\n\t\tany\n\t>\n\t\t? WorkflowContext<S, CP, CS, V, I, DB, E, Q>\n\t\t: never;\n\nexport type WorkflowLoopContextOf<AD extends AnyActorDefinition> =\n\tWorkflowContextOf<AD>;\n\nexport type WorkflowBranchContextOf<AD extends AnyActorDefinition> =\n\tWorkflowContextOf<AD>;\n\nexport type WorkflowStepContextOf<AD extends AnyActorDefinition> =\n\tAD extends BaseActorDefinition<\n\t\tinfer S,\n\t\tinfer CP,\n\t\tinfer CS,\n\t\tinfer V,\n\t\tinfer I,\n\t\tinfer DB extends AnyDatabaseProvider,\n\t\tinfer E extends EventSchemaConfig,\n\t\tinfer Q extends QueueSchemaConfig,\n\t\tany\n\t>\n\t\t? WorkflowStepContext<S, CP, CS, V, I, DB, E, Q>\n\t\t: never;\n","import type {\n\tEngineDriver,\n\tKVEntry,\n\tKVWrite,\n\tMessage,\n\tWorkflowMessageDriver,\n\tWorkflowTelemetryDriver,\n} from \"@rivetkit/workflow-engine\";\nimport type { RunContext } from \"@/actor/config\";\nimport type { AnyStaticActorInstance } from \"@/actor/definition\";\nimport { makeWorkflowKey, workflowStoragePrefix } from \"@/actor/keys\";\nimport type { RawAccess } from \"@/common/database/config\";\n\nconst WORKFLOW_STORAGE_PREFIX = workflowStoragePrefix();\n// Keep workflow flushes below depot's 320-dirty-page commit ceiling. The\n// former actor-KV transport allowed 976 KiB batches; SQLite also dirties table\n// and index pages, so it needs the same conservative budget as core imports.\nconst WORKFLOW_SQLITE_MAX_BATCH_ROWS = 128;\nconst WORKFLOW_SQLITE_MAX_BATCH_BYTES = 512 * 1024;\n\n// Mirrors the element shape returned by `queueManager.receive`. The actor\n// instance is reached through a loose type here, so the call's result is\n// untyped and the message must be annotated explicitly.\ninterface ReceivedQueueMessage {\n\tid: bigint;\n\tname: string;\n\tbody: unknown;\n\tcreatedAt: number;\n\tcomplete?: (response?: unknown) => Promise<void>;\n}\n\ntype KVEntryTuple = [Uint8Array, Uint8Array];\n\nfunction stripWorkflowKey(prefixed: Uint8Array): Uint8Array {\n\treturn prefixed.slice(WORKFLOW_STORAGE_PREFIX.length);\n}\n\nfunction computeUpperBound(prefix: Uint8Array): Uint8Array | null {\n\tconst upperBound = prefix.slice();\n\tfor (let i = upperBound.length - 1; i >= 0; i--) {\n\t\tif (upperBound[i] !== 0xff) {\n\t\t\tupperBound[i]++;\n\t\t\treturn upperBound.slice(0, i + 1);\n\t\t}\n\t}\n\treturn null;\n}\n\nfunction normalizeSqlBlob(value: unknown): Uint8Array {\n\tif (value instanceof Uint8Array) {\n\t\treturn value;\n\t}\n\tif (value instanceof ArrayBuffer) {\n\t\treturn new Uint8Array(value);\n\t}\n\tif (ArrayBuffer.isView(value)) {\n\t\treturn new Uint8Array(value.buffer, value.byteOffset, value.byteLength);\n\t}\n\tif (Array.isArray(value)) {\n\t\tconst bytes = new Uint8Array(value.length);\n\t\tfor (const [i, item] of value.entries()) {\n\t\t\tif (!Number.isInteger(item) || item < 0 || item > 255) {\n\t\t\t\tthrow new Error(\"workflow sqlite value was not a byte array\");\n\t\t\t}\n\t\t\tbytes[i] = item;\n\t\t}\n\t\treturn bytes;\n\t}\n\tthrow new Error(\"workflow sqlite value was not a blob\");\n}\n\nfunction runtimeDbFromContext(\n\trunCtx?: RunContext<any, any, any, any, any, any, any, any>,\n): RawAccess | undefined {\n\tconst db = (runCtx as unknown as { db?: unknown } | undefined)?.db;\n\tif (\n\t\tdb &&\n\t\ttypeof db === \"object\" &&\n\t\t\"execute\" in db &&\n\t\t\"transaction\" in db\n\t) {\n\t\treturn db as RawAccess;\n\t}\n\treturn undefined;\n}\n\nclass WorkflowStorage {\n\t#db: RawAccess;\n\n\tconstructor(db?: RawAccess) {\n\t\tif (!db) {\n\t\t\tthrow new Error(\n\t\t\t\t\"workflow storage requires embedded SQLite; actor KV workflow storage is no longer supported\",\n\t\t\t);\n\t\t}\n\t\tthis.#db = db;\n\t}\n\n\tasync get(key: Uint8Array): Promise<Uint8Array | null> {\n\t\tconst prefixed = makeWorkflowKey(key);\n\t\tconst rows = await this.#db.execute<{ value: unknown }>(\n\t\t\t\"SELECT value FROM _rivet_wf_kv WHERE key = ?\",\n\t\t\tprefixed,\n\t\t);\n\t\treturn rows[0]?.value == null ? null : normalizeSqlBlob(rows[0].value);\n\t}\n\n\tasync set(key: Uint8Array, value: Uint8Array): Promise<void> {\n\t\tawait this.batch([{ key, value }]);\n\t}\n\n\tasync delete(key: Uint8Array): Promise<void> {\n\t\tconst prefixed = makeWorkflowKey(key);\n\t\tawait this.#db.execute(\n\t\t\t\"DELETE FROM _rivet_wf_kv WHERE key = ?\",\n\t\t\tprefixed,\n\t\t);\n\t}\n\n\tasync batchDelete(keys: Uint8Array[]): Promise<void> {\n\t\tif (keys.length === 0) return;\n\t\tawait this.deleteRawKeys(keys.map((key) => makeWorkflowKey(key)));\n\t}\n\n\tasync deletePrefix(prefix: Uint8Array): Promise<void> {\n\t\tconst start = makeWorkflowKey(prefix);\n\t\tconst end = computeUpperBound(start);\n\t\tif (end) {\n\t\t\tawait this.#db.execute(\n\t\t\t\t\"DELETE FROM _rivet_wf_kv WHERE key >= ? AND key < ?\",\n\t\t\t\tstart,\n\t\t\t\tend,\n\t\t\t);\n\t\t\treturn;\n\t\t}\n\t\tconst entries = await this.listRaw(start);\n\t\tawait this.deleteRawKeys(\n\t\t\tentries\n\t\t\t\t.filter(([key]) => keyStartsWith(key, start))\n\t\t\t\t.map(([key]) => key),\n\t\t);\n\t}\n\n\tasync deleteRange(start: Uint8Array, end: Uint8Array): Promise<void> {\n\t\tconst prefixedStart = makeWorkflowKey(start);\n\t\tconst prefixedEnd = makeWorkflowKey(end);\n\t\tawait this.#db.execute(\n\t\t\t\"DELETE FROM _rivet_wf_kv WHERE key >= ? AND key < ?\",\n\t\t\tprefixedStart,\n\t\t\tprefixedEnd,\n\t\t);\n\t}\n\n\tasync list(prefix: Uint8Array): Promise<KVEntry[]> {\n\t\tconst prefixed = makeWorkflowKey(prefix);\n\t\tconst entries = await this.listRaw(prefixed);\n\t\treturn entries\n\t\t\t.filter(([key]) => keyStartsWith(key, prefixed))\n\t\t\t.map(([key, value]: KVEntryTuple) => ({\n\t\t\t\tkey: stripWorkflowKey(key),\n\t\t\t\tvalue,\n\t\t\t}));\n\t}\n\n\tasync batch(writes: KVWrite[]): Promise<void> {\n\t\tif (writes.length === 0) return;\n\t\tfor (const chunk of chunkWorkflowWrites(writes)) {\n\t\t\tawait this.#db.transaction(async (tx) => {\n\t\t\t\tfor (const { key, value } of chunk) {\n\t\t\t\t\tawait tx.execute(\n\t\t\t\t\t\t\"INSERT INTO _rivet_wf_kv (key, value) VALUES (?, ?) ON CONFLICT(key) DO UPDATE SET value = excluded.value\",\n\t\t\t\t\t\tmakeWorkflowKey(key),\n\t\t\t\t\t\tvalue,\n\t\t\t\t\t);\n\t\t\t\t}\n\t\t\t});\n\t\t}\n\t}\n\n\tasync listRaw(prefix: Uint8Array): Promise<KVEntryTuple[]> {\n\t\tconst end = computeUpperBound(prefix);\n\t\tconst rows = end\n\t\t\t? await this.#db.execute<{ key: unknown; value: unknown }>(\n\t\t\t\t\t\"SELECT key, value FROM _rivet_wf_kv WHERE key >= ? AND key < ? ORDER BY key ASC\",\n\t\t\t\t\tprefix,\n\t\t\t\t\tend,\n\t\t\t\t)\n\t\t\t: await this.#db.execute<{ key: unknown; value: unknown }>(\n\t\t\t\t\t\"SELECT key, value FROM _rivet_wf_kv WHERE key >= ? ORDER BY key ASC\",\n\t\t\t\t\tprefix,\n\t\t\t\t);\n\t\treturn rows.map((row) => [\n\t\t\tnormalizeSqlBlob(row.key),\n\t\t\tnormalizeSqlBlob(row.value),\n\t\t]);\n\t}\n\n\tasync deleteRawKeys(keys: Uint8Array[]): Promise<void> {\n\t\tfor (\n\t\t\tlet start = 0;\n\t\t\tstart < keys.length;\n\t\t\tstart += WORKFLOW_SQLITE_MAX_BATCH_ROWS\n\t\t) {\n\t\t\tawait this.#db.transaction(async (tx) => {\n\t\t\t\tfor (const key of keys.slice(\n\t\t\t\t\tstart,\n\t\t\t\t\tstart + WORKFLOW_SQLITE_MAX_BATCH_ROWS,\n\t\t\t\t)) {\n\t\t\t\t\tawait tx.execute(\n\t\t\t\t\t\t\"DELETE FROM _rivet_wf_kv WHERE key = ?\",\n\t\t\t\t\t\tkey,\n\t\t\t\t\t);\n\t\t\t\t}\n\t\t\t});\n\t\t}\n\t}\n}\n\nexport function chunkWorkflowWrites(writes: KVWrite[]): KVWrite[][] {\n\tconst chunks: KVWrite[][] = [];\n\tlet chunk: KVWrite[] = [];\n\tlet chunkBytes = 0;\n\tfor (const write of writes) {\n\t\tconst writeBytes =\n\t\t\tWORKFLOW_STORAGE_PREFIX.length +\n\t\t\twrite.key.length +\n\t\t\twrite.value.length;\n\t\tif (\n\t\t\tchunk.length > 0 &&\n\t\t\t(chunk.length >= WORKFLOW_SQLITE_MAX_BATCH_ROWS ||\n\t\t\t\tchunkBytes + writeBytes > WORKFLOW_SQLITE_MAX_BATCH_BYTES)\n\t\t) {\n\t\t\tchunks.push(chunk);\n\t\t\tchunk = [];\n\t\t\tchunkBytes = 0;\n\t\t}\n\t\tchunk.push(write);\n\t\tchunkBytes += writeBytes;\n\t}\n\tif (chunk.length > 0) chunks.push(chunk);\n\treturn chunks;\n}\n\nfunction keyStartsWith(key: Uint8Array, prefix: Uint8Array): boolean {\n\tif (key.length < prefix.length) return false;\n\tfor (let i = 0; i < prefix.length; i++) {\n\t\tif (key[i] !== prefix[i]) return false;\n\t}\n\treturn true;\n}\n\nclass ActorWorkflowMessageDriver implements WorkflowMessageDriver {\n\t#actor: AnyStaticActorInstance;\n\t#runCtx: RunContext<any, any, any, any, any, any, any, any>;\n\n\tconstructor(\n\t\tactor: AnyStaticActorInstance,\n\t\trunCtx: RunContext<any, any, any, any, any, any, any, any>,\n\t) {\n\t\tthis.#actor = actor;\n\t\tthis.#runCtx = runCtx;\n\t}\n\n\tasync addMessage(message: Message): Promise<void> {\n\t\tawait this.#runCtx.internalKeepAwake(\n\t\t\tthis.#actor.queueManager.enqueue(message.name, message.data),\n\t\t);\n\t}\n\n\tasync receiveMessages(opts: {\n\t\tnames?: readonly string[];\n\t\tcount: number;\n\t\tcompletable: boolean;\n\t}): Promise<Message[]> {\n\t\tconst messages = await this.#runCtx.internalKeepAwake(\n\t\t\tthis.#actor.queueManager.receive(\n\t\t\t\topts.names && opts.names.length > 0\n\t\t\t\t\t? [...opts.names]\n\t\t\t\t\t: undefined,\n\t\t\t\topts.count,\n\t\t\t\t0,\n\t\t\t\tundefined,\n\t\t\t\topts.completable,\n\t\t\t),\n\t\t);\n\t\treturn messages.map((message: ReceivedQueueMessage) => ({\n\t\t\tid: message.id.toString(),\n\t\t\tname: message.name,\n\t\t\tdata: message.body,\n\t\t\tsentAt: message.createdAt,\n\t\t\t...(opts.completable\n\t\t\t\t? {\n\t\t\t\t\t\tcomplete: async (response?: unknown) => {\n\t\t\t\t\t\t\tawait this.#runCtx.internalKeepAwake(\n\t\t\t\t\t\t\t\tthis.#actor.queueManager.completeMessage(\n\t\t\t\t\t\t\t\t\tmessage,\n\t\t\t\t\t\t\t\t\tresponse,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t},\n\t\t\t\t\t}\n\t\t\t\t: {}),\n\t\t}));\n\t}\n\n\tasync completeMessage(\n\t\tmessageId: string,\n\t\tresponse?: unknown,\n\t): Promise<void> {\n\t\tlet parsedId: bigint;\n\t\ttry {\n\t\t\tparsedId = BigInt(messageId);\n\t\t} catch {\n\t\t\treturn;\n\t\t}\n\n\t\tawait this.#runCtx.internalKeepAwake(\n\t\t\tthis.#actor.queueManager.completeMessageById(parsedId, response),\n\t\t);\n\t}\n}\n\nexport class ActorWorkflowDriver implements EngineDriver {\n\treadonly atomicBatch = true;\n\treadonly workerPollInterval = 100;\n\treadonly messageDriver: WorkflowMessageDriver;\n\treadonly telemetry: WorkflowTelemetryDriver;\n\t#actor: AnyStaticActorInstance;\n\t#runCtx: RunContext<any, any, any, any, any, any, any, any>;\n\t#storage: WorkflowStorage;\n\n\tconstructor(\n\t\tactor: AnyStaticActorInstance,\n\t\trunCtx: RunContext<any, any, any, any, any, any, any, any>,\n\t) {\n\t\tthis.#actor = actor;\n\t\tthis.#runCtx = runCtx;\n\t\tthis.messageDriver = new ActorWorkflowMessageDriver(actor, runCtx);\n\t\tthis.telemetry = {\n\t\t\tstartSpan: () => runCtx.run.startWorkflowSpan(),\n\t\t};\n\t\tthis.#storage = new WorkflowStorage(runtimeDbFromContext(runCtx));\n\t}\n\n\t/** Keeps the engine's history storage out of the workflow run's trace. */\n\t#untraced<T>(run: () => Promise<T>): Promise<T> {\n\t\treturn this.#runCtx.internalKeepAwake(\n\t\t\tthis.#runCtx.run.runOutsideWorkflowSpan(run),\n\t\t);\n\t}\n\n\tasync get(key: Uint8Array): Promise<Uint8Array | null> {\n\t\treturn await this.#untraced(() => this.#storage.get(key));\n\t}\n\n\tasync set(key: Uint8Array, value: Uint8Array): Promise<void> {\n\t\tawait this.#untraced(() => this.#storage.set(key, value));\n\t}\n\n\tasync delete(key: Uint8Array): Promise<void> {\n\t\tawait this.#untraced(() => this.#storage.delete(key));\n\t}\n\n\tasync batchDelete(keys: Uint8Array[]): Promise<void> {\n\t\tawait this.#untraced(() => this.#storage.batchDelete(keys));\n\t}\n\n\tasync deletePrefix(prefix: Uint8Array): Promise<void> {\n\t\tawait this.#untraced(() => this.#storage.deletePrefix(prefix));\n\t}\n\n\tasync deleteRange(start: Uint8Array, end: Uint8Array): Promise<void> {\n\t\tawait this.#untraced(() => this.#storage.deleteRange(start, end));\n\t}\n\n\tasync list(prefix: Uint8Array): Promise<KVEntry[]> {\n\t\treturn await this.#untraced(() => this.#storage.list(prefix));\n\t}\n\n\tasync batch(writes: KVWrite[]): Promise<void> {\n\t\tif (writes.length === 0) return;\n\n\t\tawait this.#untraced(() =>\n\t\t\tthis.#actor.stateManager.saveStateAndWorkflowBatch(writes),\n\t\t);\n\t}\n\n\tasync setAlarm(_workflowId: string, wakeAt: number): Promise<void> {\n\t\tawait this.#runCtx.internalKeepAwake(\n\t\t\tthis.#actor.driver.setAlarm(this.#actor, wakeAt),\n\t\t);\n\t}\n\n\tasync clearAlarm(_workflowId: string): Promise<void> {\n\t\t// No dedicated clear alarm support in actor drivers.\n\t\treturn;\n\t}\n\n\twaitForMessages(\n\t\tmessageNames: string[],\n\t\tabortSignal: AbortSignal,\n\t): Promise<void> {\n\t\treturn this.#actor.queueManager.waitForNames(\n\t\t\tmessageNames.length > 0 ? messageNames : undefined,\n\t\t\tabortSignal,\n\t\t);\n\t}\n}\n\nclass NoopWorkflowMessageDriver implements WorkflowMessageDriver {\n\tasync addMessage(_message: Message): Promise<void> {\n\t\tthrow new Error(\"Workflow control driver does not support messages\");\n\t}\n\n\tasync receiveMessages(_opts: {\n\t\tnames?: readonly string[];\n\t\tcount: number;\n\t\tcompletable: boolean;\n\t}): Promise<Message[]> {\n\t\tthrow new Error(\"Workflow control driver does not support messages\");\n\t}\n\n\tasync completeMessage(\n\t\t_messageId: string,\n\t\t_response?: unknown,\n\t): Promise<void> {\n\t\tthrow new Error(\"Workflow control driver does not support messages\");\n\t}\n}\n\nexport class ActorWorkflowControlDriver implements EngineDriver {\n\treadonly workerPollInterval = 100;\n\treadonly messageDriver: WorkflowMessageDriver =\n\t\tnew NoopWorkflowMessageDriver();\n\t#actor: AnyStaticActorInstance;\n\t#storage: WorkflowStorage;\n\n\tconstructor(\n\t\tactor: AnyStaticActorInstance,\n\t\trunCtx?: RunContext<any, any, any, any, any, any, any, any>,\n\t) {\n\t\tthis.#actor = actor;\n\t\tthis.#storage = new WorkflowStorage(runtimeDbFromContext(runCtx));\n\t}\n\n\tasync get(key: Uint8Array): Promise<Uint8Array | null> {\n\t\treturn await this.#storage.get(key);\n\t}\n\n\tasync set(key: Uint8Array, value: Uint8Array): Promise<void> {\n\t\tawait this.#storage.set(key, value);\n\t}\n\n\tasync delete(key: Uint8Array): Promise<void> {\n\t\tawait this.#storage.delete(key);\n\t}\n\n\tasync batchDelete(keys: Uint8Array[]): Promise<void> {\n\t\tawait this.#storage.batchDelete(keys);\n\t}\n\n\tasync deletePrefix(prefix: Uint8Array): Promise<void> {\n\t\tawait this.#storage.deletePrefix(prefix);\n\t}\n\n\tasync deleteRange(start: Uint8Array, end: Uint8Array): Promise<void> {\n\t\tawait this.#storage.deleteRange(start, end);\n\t}\n\n\tasync list(prefix: Uint8Array): Promise<KVEntry[]> {\n\t\treturn await this.#storage.list(prefix);\n\t}\n\n\tasync batch(writes: KVWrite[]): Promise<void> {\n\t\tif (writes.length === 0) {\n\t\t\treturn;\n\t\t}\n\n\t\tawait this.#storage.batch(writes);\n\t}\n\n\tasync setAlarm(_workflowId: string, wakeAt: number): Promise<void> {\n\t\tawait this.#actor.driver.setAlarm(this.#actor, wakeAt);\n\t}\n\n\tasync clearAlarm(_workflowId: string): Promise<void> {\n\t\treturn;\n\t}\n\n\twaitForMessages(\n\t\t_messageNames: string[],\n\t\t_abortSignal: AbortSignal,\n\t): Promise<void> {\n\t\tthrow new Error(\"Workflow control driver does not support messages\");\n\t}\n}\n"]}
|
|
1
|
+
{"version":3,"sources":["/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/workflow/mod.cjs","../../../src/workflow/mod.ts","../../../src/workflow/context.ts","../../../src/workflow/constants.ts","../../../src/workflow/driver.ts"],"names":[],"mappings":"AAAA;AACE;AACF,yDAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B;AACE;AACA;AACF,yDAA8B;AAC9B;AACE;AACA;AACF,yDAA8B;AAC9B;AACE;AACF,yDAA8B;AAC9B;AACE;AACA;AACF,yDAA8B;AAC9B;AACA;ACrBA;AACC;AACA;AACA;AACA;AACA;AACA;AACA;AAEA;AACA;AACA;AAAA,2DAEM;AACP,4FAAsB;ADqBtB;AACA;AEtBA;AFwBA;AACA;AGvCO,IAAM,sBAAA,EAAwB,wCAAA;AHyCrC;AACA;AEkPA,IAAM,gBAAA,kBAAkB,MAAA,CAAO,0BAA0B,CAAA;AAOlD,SAAS,qBAAA,CAAsB,MAAA,EAAuB;AAC5D,EAAA,MAAM,MAAA,EACL,MAAA,CACC,+CAA6B,CAAA;AAC/B,EAAA,GAAA,CAAA,CAAI,MAAA,GAAA,KAAA,EAAA,KAAA,EAAA,EAAA,KAAA,CAAO,MAAA,EAAA,IAAW,IAAA,EAAM;AAC3B,IAAA,MAAM,IAAI,iCAAA,CAAa,CAAA;AAAA,EACxB;AACD;AAQO,IAAM,oBAAA,EAAN,MASL;AAAA,EACD,CAAA,MAAA;AAAA,EAUA,CAAA,OAAA,EAAU,IAAA;AAAA,EACV,CAAA,gBAAA;AAAA,EAEA,WAAA,CACC,MAAA,EAUA,gBAAA,EACC;AACD,IAAA,IAAA,CAAK,CAAA,OAAA,EAAU,MAAA;AACf,IAAA,IAAA,CAAK,CAAA,iBAAA,EAAoB,gBAAA;AAAA,EAC1B;AAAA,EAEA,CAAC,eAAe,CAAA,CAAA,EAAU;AACzB,IAAA,IAAA,CAAK,CAAA,OAAA,EAAU,KAAA;AAAA,EAChB;AAAA,EAEA,CAAA,YAAA,CAAc,OAAA,EAAuB;AACpC,IAAA,GAAA,CAAI,CAAC,IAAA,CAAK,CAAA,MAAA,EAAS;AAClB,MAAA,IAAA,CAAK,CAAA,gBAAA,CAAkB,CAAA;AACvB,MAAA,MAAM,IAAI,KAAA;AAAA,QACT,CAAA,EAAA;AACD,MAAA;AACD,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACE,IAAA;AACL,IAAA;AACD,EAAA;AAEI,EAAA;AACE,IAAA;AACL,IAAA;AACD,EAAA;AAEI,EAAA;AACE,IAAA;AACL,IAAA;AAGD,EAAA;AAEA,EAAA;AACM,IAAA;AACL,IAAA;AACD,EAAA;AAUA,EAAA;AACM,IAAA;AACA,IAAA;AACJ,MAAA;AACI,MAAA;AACL,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AASA,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AACA,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAOA,EAAA;AACM,IAAA;AACL,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAAA;AAMA,EAAA;AACM,IAAA;AACA,IAAA;AACN,EAAA;AAEA,EAAA;AACM,IAAA;AACA,IAAA;AACN,EAAA;AACD;AASa;AAUZ,EAAA;AACA,EAAA;AAUA,EAAA;AAaM,IAAA;AACA,IAAA;AACN,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AACD,EAAA;AAEI,EAAA;AACH,IAAA;AAoBA,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AAuBA,IAAA;AACC,MAAA;AACA,MAAA;AAAqB,QAAA;AAErB,MAAA;AACD,IAAA;AAEA,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AACD,EAAA;AA6BM,EAAA;AA0BD,IAAA;AACH,MAAA;AACC,QAAA;AACD,MAAA;AACA,MAAA;AACA,MAAA;AAA8B,QAAA;AAE9B,MAAA;AACD,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACI,MAAA;AACH,MAAA;AACA,MAAA;AAGD,IAAA;AACA,IAAA;AACD,EAAA;AA6BM,EAAA;AA0BD,IAAA;AACH,MAAA;AACC,QAAA;AACD,MAAA;AACA,MAAA;AACA,MAAA;AAA8B,QAAA;AAE9B,MAAA;AACD,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACI,MAAA;AACH,MAAA;AACA,MAAA;AAGD,IAAA;AACA,IAAA;AACD,EAAA;AA6BM,EAAA;AA0BD,IAAA;AACH,MAAA;AACC,QAAA;AACD,MAAA;AACA,MAAA;AACA,MAAA;AAA8B,QAAA;AACjB,UAAA;AAAI,UAAA;AAEhB,QAAA;AACD,MAAA;AACD,IAAA;AACA,IAAA;AACA,IAAA;AACI,MAAA;AACH,MAAA;AACD,IAAA;AACA,IAAA;AACD,EAAA;AA+BM,EAAA;AA4BD,IAAA;AACH,MAAA;AACC,QAAA;AACD,MAAA;AACA,MAAA;AACA,MAAA;AAA8B,QAAA;AACjB,UAAA;AACX;AAAA;AAAA,UAAA;AAQD,QAAA;AACD,MAAA;AACD,IAAA;AACA,IAAA;AACA,IAAA;AACI,MAAA;AACH,MAAA;AAGD,IAAA;AACA,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEA,EAAA;AACC,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAyBM,EAAA;AAIL,IAAA;AACC,MAAA;AACC,QAAA;AACA,QAAA;AAAA,UAAA;AAGA,QAAA;AACA,MAAA;AACF,IAAA;AACA,IAAA;AAA8B,MAAA;AAE9B,IAAA;AACD,EAAA;AAEM,EAAA;AAiBL,IAAA;AACC,MAAA;AACA,MAAA;AAEC,IAAA;AACF,IAAA;AAA+B,MAAA;AAE/B,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEA,EAAA;AACC,IAAA;AACD,EAAA;AAEA,EAAA;AACC,IAAA;AACD,EAAA;AAAA;AAAA;AAAA;AAKM,EAAA;AAaL,IAAA;AAWA,IAAA;AACI,IAAA;AACH,MAAA;AACD,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AAEA,EAAA;AAGK,IAAA;AACA,IAAA;AACH,MAAA;AACD,IAAA;AACC,MAAA;AACD,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACI,MAAA;AACL,IAAA;AACD,EAAA;AAAA;AAGM,EAAA;AAcL,IAAA;AAUA,IAAA;AACI,IAAA;AACH,MAAA;AACD,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAAA;AAAA;AAIA,EAAA;AACK,IAAA;AACH,MAAA;AACA,MAAA;AAKE,QAAA;AACF,MAAA;AACD,IAAA;AAEA,IAAA;AAEK,IAAA;AACH,MAAA;AACA,QAAA;AACC,UAAA;AACD,QAAA;AACC,UAAA;AAAuB,YAAA;AACjB,YAAA;AACL,UAAA;AAEF,QAAA;AACE,MAAA;AACJ,IAAA;AACD,EAAA;AAEA,EAAA;AAaC,IAAA;AACA,IAAA;AACD,EAAA;AACD;AFlvBS;AACA;AIlWH;AAIA;AACA;AAeG;AACD,EAAA;AACR;AAES;AACF,EAAA;AACN,EAAA;AACK,IAAA;AACH,MAAA;AACA,MAAA;AACD,IAAA;AACD,EAAA;AACO,EAAA;AACR;AAES;AACJ,EAAA;AACH,IAAA;AACD,EAAA;AACI,EAAA;AACH,IAAA;AACD,EAAA;AACI,EAAA;AACH,IAAA;AACD,EAAA;AACI,EAAA;AACH,IAAA;AACA,IAAA;AACC,MAAA;AACC,QAAA;AACD,MAAA;AACA,MAAA;AACD,IAAA;AACA,IAAA;AACD,EAAA;AACM,EAAA;AACP;AAES;AAID,EAAA;AACH,IAAA;AACH,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACE,QAAA;AACD,QAAA;AACD,MAAA;AACD,IAAA;AACD,EAAA;AACD;AAES;AAGF,EAAA;AAEL,EAAA;AAQA,IAAA;AACD,EAAA;AACO,EAAA;AACR;AAEM;AACL,EAAA;AAEA,EAAA;AACM,IAAA;AACJ,MAAA;AACC,QAAA;AACD,MAAA;AACD,IAAA;AACK,IAAA;AACN,EAAA;AAEM,EAAA;AA1HP,IAAA;AA2HE,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AACA,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AACD,EAAA;AAEM,EAAA;AACD,IAAA;AACJ,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACA,IAAA;AACI,IAAA;AACH,MAAA;AACC,QAAA;AACA,QAAA;AACA,QAAA;AACD,MAAA;AACA,MAAA;AACD,IAAA;AACA,IAAA;AACA,IAAA;AACC,MAAA;AAGD,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACA,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACA,MAAA;AACD,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACA,IAAA;AACA,IAAA;AAGE,MAAA;AACA,MAAA;AACC,IAAA;AACJ,EAAA;AAEM,EAAA;AACD,IAAA;AACJ,IAAA;AACC,MAAA;AACC,QAAA;AACC,UAAA;AAAS,YAAA;AACR,YAAA;AACmB,YAAA;AACnB,UAAA;AAEF,QAAA;AACA,MAAA;AACF,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACA,IAAA;AAEG,MAAA;AACA,MAAA;AACA,MAAA;AAEA,IAAA;AACA,MAAA;AACA,MAAA;AACD,IAAA;AACF,IAAA;AACC,MAAA;AACA,MAAA;AACA,IAAA;AACF,EAAA;AAEM,EAAA;AACL,IAAA;AAKC,MAAA;AACC,QAAA;AAAuB,UAAA;AACtB,UAAA;AAED,QAAA;AACC,UAAA;AAAS,YAAA;AACR,YAAA;AACA,UAAA;AAEF,QAAA;AACA,MAAA;AACF,IAAA;AACD,EAAA;AACD;AAEgB;AACT,EAAA;AACF,EAAA;AACA,EAAA;AACJ,EAAA;AACC,IAAA;AAKC,IAAA;AAIA,MAAA;AACA,MAAA;AACA,MAAA;AACD,IAAA;AACA,IAAA;AACA,IAAA;AACD,EAAA;AACI,EAAA;AACG,EAAA;AACR;AAES;AACJ,EAAA;AACJ,EAAA;AACK,IAAA;AACL,EAAA;AACO,EAAA;AACR;AAEM;AACL,EAAA;AACA,EAAA;AAEA,EAAA;AAIM,IAAA;AACA,IAAA;AACN,EAAA;AAEM,EAAA;AACL,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AAEM,EAAA;AAKL,IAAA;AACC,MAAA;AACC,QAAA;AAGA,QAAA;AACA,QAAA;AACA,QAAA;AACA,QAAA;AACD,MAAA;AACD,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACI,MAAA;AAED,QAAA;AACC,UAAA;AAAmB,YAAA;AACO,cAAA;AACxB,cAAA;AACA,YAAA;AACD,UAAA;AAEF,QAAA;AACD,MAAA;AAED,IAAA;AACH,EAAA;AAEM,EAAA;AAID,IAAA;AACA,IAAA;AACH,MAAA;AACD,IAAA;AACC,MAAA;AACD,IAAA;AAEA,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AACD;AAEa;AACH,iBAAA;AACA,kBAAA;AACA,EAAA;AACA,EAAA;AACT,EAAA;AACA,EAAA;AACA,EAAA;AAEA,EAAA;AAIM,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACJ,MAAA;AACD,IAAA;AACK,IAAA;AACN,EAAA;AAAA;AAGA,EAAA;AACC,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACD,IAAA;AAEJ,IAAA;AAAqB,MAAA;AAErB,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AAEM,EAAA;AAEL,IAAA;AACD,EAAA;AAEA,EAAA;AAIC,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AACD,EAAA;AACD;AAEM;AACC,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AAKL,IAAA;AACD,EAAA;AAEM,EAAA;AAIL,IAAA;AACD,EAAA;AACD;AAEa;AACH,kBAAA;AACA,kBAAA;AAET,EAAA;AACA,EAAA;AAEA,EAAA;AAIM,IAAA;AACA,IAAA;AACN,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACD,IAAA;AACH,MAAA;AACD,IAAA;AAEA,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEM,EAAA;AACL,IAAA;AACD,EAAA;AAEA,EAAA;AAIC,IAAA;AACD,EAAA;AACD;AJ2NS;AACA;ACzrBA;AAeA;AAEP,EAAA;AAMA,IAAA;AACD,EAAA;AAGC,EAAA;AAIA,IAAA;AACD,EAAA;AAEO,EAAA;AACR;AAES;AACD,EAAA;AACN,IAAA;AACA,IAAA;AACA,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AACD,EAAA;AACD;AAES;AAEP,EAAA;AAIF;AAES;AAEP,EAAA;AAIF;AA2BgB;AA4CT,EAAA;AAMA,EAAA;AAEN,EAAA;AAIC,IAAA;AAGD,EAAA;AAEA,EAAA;AAIC,IAAA;AACI,IAAA;AACC,IAAA;AACJ,MAAA;AACA,MAAA;AACD,IAAA;AACA,IAAA;AACD,EAAA;AAEA,EAAA;AAYC,IAAA;AAKA,IAAA;AACA,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACD,IAAA;AAEA,IAAA;AACA,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACC,QAAA;AACD,MAAA;AAEA,MAAA;AACA,MAAA;AACA,MAAA;AACC,QAAA;AAAyB,UAAA;AACC,UAAA;AAExB,YAAA;AAAuB,cAAA;AAChB,cAAA;AACN,cAAA;AACA,cAAA;AACuB,YAAA;AAExB,YAAA;AACA,YAAA;AAA4C,UAAA;AAE9C,QAAA;AACD,MAAA;AACC,QAAA;AAIC,UAAA;AACD,QAAA;AACA,QAAA;AACD,MAAA;AACA,IAAA;AAED,IAAA;AACC,MAAA;AACA,MAAA;AACC,QAAA;AACA,QAAA;AACD,MAAA;AACA,MAAA;AACA,MAAA;AACA,MAAA;AACC,QAAA;AAAM;AAAA;AAGN,QAAA;AACA,QAAA;AACA,QAAA;AAGG,UAAA;AAGC,YAAA;AAAA,UAAA;AAED,UAAA;AACD,QAAA;AAEH,MAAA;AACD,IAAA;AACA,IAAA;AAEA,IAAA;AACC,MAAA;AACD,IAAA;AACI,IAAA;AACH,MAAA;AACD,IAAA;AACC,MAAA;AACC,QAAA;AACA,MAAA;AACF,IAAA;AAEI,IAAA;AACH,MAAA;AACD,IAAA;AAGC,MAAA;AACC,QAAA;AACD,MAAA;AAEA,MAAA;AACC,QAAA;AAAiB,UAAA;AACX,UAAA;AAEL,QAAA;AACD,QAAA;AACD,MAAA;AAEA,MAAA;AACC,QAAA;AACA,QAAA;AACA,MAAA;AACF,IAAA;AACC,MAAA;AACD,IAAA;AACD,EAAA;AAEO,EAAA;AACN,IAAA;AACA,IAAA;AACA,IAAA;AACC,MAAA;AACA,MAAA;AACC,QAAA;AACA,QAAA;AACD,MAAA;AACA,MAAA;AACA,MAAA;AACC,QAAA;AACA,QAAA;AAGC,UAAA;AACC,YAAA;AAA6B,UAAA;AAE/B,QAAA;AACD,MAAA;AACD,IAAA;AACA,EAAA;AACF;AD8hBS;AACA;AACA;AACA;AACA;AACA","file":"/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/workflow/mod.cjs","sourcesContent":[null,"import {\n\tCriticalError,\n\tEntryInProgressError,\n\tHistoryDivergedError,\n\tJoinError,\n\tRaceError,\n\tRollbackCheckpointError,\n\tRollbackError,\n\ttype RunWorkflowOptions,\n\treplayWorkflowFromStep,\n\trunWorkflow,\n\tStepExhaustedError,\n\ttype WorkflowErrorEvent,\n} from \"@rivetkit/workflow-engine\";\nimport invariant from \"invariant\";\nimport {\n\tACTOR_CONTEXT_INTERNAL_SYMBOL,\n\tdefineRunHandler,\n\ttype RunContext,\n\ttype RunControl,\n} from \"@/actor/config\";\nimport type { AnyStaticActorInstance } from \"@/actor/definition\";\nimport { isActorAbortedError, RivetError } from \"@/actor/errors\";\nimport type { EventSchemaConfig, QueueSchemaConfig } from \"@/actor/schema\";\nimport type { AnyDatabaseProvider } from \"@/common/database/config\";\nimport { stringifyError } from \"@/utils\";\nimport { throwIfGenerationLost, WorkflowContext } from \"./context\";\nimport { ActorWorkflowControlDriver, ActorWorkflowDriver } from \"./driver\";\nimport { createWorkflowInspectorAdapter } from \"./inspector\";\n\nexport type {\n\tTryBlockCatchKind,\n\tTryBlockConfig,\n\tTryBlockFailure,\n\tTryBlockResult,\n\tTryStepCatchKind,\n\tTryStepConfig,\n\tTryStepFailure,\n\tTryStepResult,\n\tWorkflowError,\n\tWorkflowErrorEvent,\n} from \"@rivetkit/workflow-engine\";\nexport { Loop } from \"@rivetkit/workflow-engine\";\nexport {\n\ttype WorkflowBranchConfig,\n\ttype WorkflowBranchContextOf,\n\tWorkflowContext,\n\ttype WorkflowContextOf,\n\ttype WorkflowLoopConfig,\n\ttype WorkflowLoopContextOf,\n\ttype WorkflowStepConfig,\n\tWorkflowStepContext,\n\ttype WorkflowStepContextOf,\n\ttype WorkflowTryConfig,\n\ttype WorkflowTryStepConfig,\n} from \"./context\";\n\nfunction shouldRethrowWorkflowError(error: unknown): boolean {\n\tif (\n\t\terror instanceof CriticalError ||\n\t\terror instanceof JoinError ||\n\t\terror instanceof RaceError ||\n\t\terror instanceof RollbackError ||\n\t\terror instanceof StepExhaustedError\n\t) {\n\t\treturn false;\n\t}\n\n\tif (\n\t\terror instanceof EntryInProgressError ||\n\t\terror instanceof HistoryDivergedError ||\n\t\terror instanceof RollbackCheckpointError\n\t) {\n\t\treturn true;\n\t}\n\n\treturn true;\n}\n\nfunction workflowReplayInFlightError(): RivetError {\n\treturn new RivetError(\n\t\t\"actor\",\n\t\t\"workflow_in_flight\",\n\t\t\"Workflow replay is unavailable while the workflow is currently in flight.\",\n\t\t{\n\t\t\tpublic: true,\n\t\t\tstatusCode: 409,\n\t\t},\n\t);\n}\n\nfunction isWorkflowReplayBlockedByRunningEntry(error: unknown): boolean {\n\treturn (\n\t\terror instanceof Error &&\n\t\terror.message ===\n\t\t\t\"Cannot replay a workflow while a step is currently running\"\n\t);\n}\n\nfunction isRunHandlerUnavailable(error: unknown): boolean {\n\treturn (\n\t\terror instanceof RivetError &&\n\t\terror.group === \"actor\" &&\n\t\terror.code === \"run_handler_unavailable\"\n\t);\n}\n\nexport interface WorkflowOptions<\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig = Record<never, never>,\n\tTQueues extends QueueSchemaConfig = Record<never, never>,\n> {\n\tonError?: (\n\t\tctx: RunContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t\tevent: WorkflowErrorEvent,\n\t) => void | Promise<void>;\n}\n\nexport function workflow<\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig = Record<never, never>,\n\tTQueues extends QueueSchemaConfig = Record<never, never>,\n>(\n\tfn: (\n\t\tctx: WorkflowContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t) => Promise<unknown>,\n\toptions: WorkflowOptions<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t> = {},\n): (\n\tc: RunContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>,\n) => Promise<void> {\n\tconst onError = options.onError;\n\ttype WorkflowInspectorRegistration = ReturnType<\n\t\ttypeof createWorkflowInspectorAdapter\n\t> & { control?: RunControl };\n\t// Keyed by actor id and generation so a lost generation that is still running cannot rebind\n\t// the control or state of the generation that replaced it.\n\tconst workflowInspectors = new Map<string, WorkflowInspectorRegistration>();\n\n\tfunction workflowInspectorKey(\n\t\tactorId: string,\n\t\tactorGeneration: number | undefined,\n\t): string {\n\t\treturn actorGeneration === undefined\n\t\t\t? actorId\n\t\t\t: `${actorId}#${actorGeneration}`;\n\t}\n\n\tfunction getWorkflowInspector(\n\t\tactorId: string,\n\t\tactorGeneration: number | undefined,\n\t) {\n\t\tconst key = workflowInspectorKey(actorId, actorGeneration);\n\t\tlet workflowInspector = workflowInspectors.get(key);\n\t\tif (!workflowInspector) {\n\t\t\tworkflowInspector = createWorkflowInspectorAdapter();\n\t\t\tworkflowInspectors.set(key, workflowInspector);\n\t\t}\n\t\treturn workflowInspector;\n\t}\n\n\tasync function run(\n\t\trunCtx: RunContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<void> {\n\t\tconst actor = (\n\t\t\trunCtx as unknown as {\n\t\t\t\t[ACTOR_CONTEXT_INTERNAL_SYMBOL]?: AnyStaticActorInstance;\n\t\t\t}\n\t\t)[ACTOR_CONTEXT_INTERNAL_SYMBOL];\n\t\tinvariant(actor, \"workflow() requires an actor instance\");\n\t\tconst actorGeneration = (actor as { generation?: number }).generation;\n\t\tconst workflowInspector = getWorkflowInspector(\n\t\t\tactor.id,\n\t\t\tactorGeneration,\n\t\t);\n\n\t\tconst driver = new ActorWorkflowDriver(actor, runCtx);\n\t\tconst controlDriver = new ActorWorkflowControlDriver(actor, runCtx);\n\t\tworkflowInspector.setReplayFromStep(async (entryId) => {\n\t\t\tconst workflowState = await workflowInspector.adapter.getState();\n\t\t\tif (workflowState === \"pending\" || workflowState === \"running\") {\n\t\t\t\tthrow workflowReplayInFlightError();\n\t\t\t}\n\n\t\t\tconst control = workflowInspector.control;\n\t\t\tinvariant(control, \"workflow Inspector control is not initialized\");\n\t\t\ttry {\n\t\t\t\treturn await control.run.withInactive(\n\t\t\t\t\t{ restartOnSuccess: true },\n\t\t\t\t\tasync () => {\n\t\t\t\t\t\tconst snapshot = await replayWorkflowFromStep(\n\t\t\t\t\t\t\tactor.id,\n\t\t\t\t\t\t\tcontrolDriver,\n\t\t\t\t\t\t\tentryId,\n\t\t\t\t\t\t\t{ scheduleAlarm: false },\n\t\t\t\t\t\t);\n\t\t\t\t\t\tworkflowInspector.update(snapshot);\n\t\t\t\t\t\treturn workflowInspector.adapter.getHistory();\n\t\t\t\t\t},\n\t\t\t\t);\n\t\t\t} catch (error) {\n\t\t\t\tif (\n\t\t\t\t\tisWorkflowReplayBlockedByRunningEntry(error) ||\n\t\t\t\t\tisRunHandlerUnavailable(error)\n\t\t\t\t) {\n\t\t\t\t\tthrow workflowReplayInFlightError();\n\t\t\t\t}\n\t\t\t\tthrow error;\n\t\t\t}\n\t\t});\n\n\t\tconst handle = runWorkflow(\n\t\t\tactor.id,\n\t\t\tasync (ctx) => {\n\t\t\t\tthrowIfGenerationLost(runCtx);\n\t\t\t\treturn await fn(new WorkflowContext(ctx, runCtx));\n\t\t\t},\n\t\t\tundefined,\n\t\t\tdriver,\n\t\t\t{\n\t\t\t\tmode: \"live\",\n\t\t\t\t// The actor logger and the engine's pino logger are runtime\n\t\t\t\t// compatible but not structurally assignable.\n\t\t\t\tlogger: runCtx.log as RunWorkflowOptions[\"logger\"],\n\t\t\t\tonHistoryUpdated: workflowInspector.update,\n\t\t\t\tonError: onError\n\t\t\t\t\t? async (event) => {\n\t\t\t\t\t\t\t// A lost generation starts no new user hook.\n\t\t\t\t\t\t\tif (\n\t\t\t\t\t\t\t\t(actor as { isLost?: boolean }).isLost === true\n\t\t\t\t\t\t\t) {\n\t\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\tawait onError(runCtx, event);\n\t\t\t\t\t\t}\n\t\t\t\t\t: undefined,\n\t\t\t},\n\t\t);\n\t\tworkflowInspector.setGetState(async () => await handle.getState());\n\n\t\tconst onAbort = () => {\n\t\t\thandle.evict();\n\t\t};\n\t\tif (runCtx.abortSignal.aborted) {\n\t\t\tonAbort();\n\t\t} else {\n\t\t\trunCtx.abortSignal.addEventListener(\"abort\", onAbort, {\n\t\t\t\tonce: true,\n\t\t\t});\n\t\t}\n\n\t\ttry {\n\t\t\tawait handle.result;\n\t\t} catch (error) {\n\t\t\t// `abortSignal.aborted` is delivered on a separate async hop and\n\t\t\t// races the rejection, so detect the sleep abort structurally too.\n\t\t\tif (runCtx.abortSignal.aborted || isActorAbortedError(error)) {\n\t\t\t\treturn;\n\t\t\t}\n\n\t\t\tif (shouldRethrowWorkflowError(error)) {\n\t\t\t\trunCtx.log.error({\n\t\t\t\t\tmsg: \"workflow run failed\",\n\t\t\t\t\terror: stringifyError(error),\n\t\t\t\t});\n\t\t\t\tthrow error;\n\t\t\t}\n\n\t\t\trunCtx.log.warn({\n\t\t\t\tmsg: \"workflow failed and will sleep until woken\",\n\t\t\t\terror: stringifyError(error),\n\t\t\t});\n\t\t} finally {\n\t\t\trunCtx.abortSignal.removeEventListener(\"abort\", onAbort);\n\t\t}\n\t}\n\n\treturn defineRunHandler(run, {\n\t\ticon: \"diagram-project\",\n\t\tinspectorKind: \"workflow\",\n\t\tcreateInspector: ({ actorId, actorGeneration, control }) => {\n\t\t\tconst key = workflowInspectorKey(actorId, actorGeneration);\n\t\t\tconst workflowInspector = getWorkflowInspector(\n\t\t\t\tactorId,\n\t\t\t\tactorGeneration,\n\t\t\t);\n\t\t\tworkflowInspector.control = control;\n\t\t\treturn {\n\t\t\t\tinspector: { workflow: workflowInspector.adapter },\n\t\t\t\tdispose: () => {\n\t\t\t\t\t// Do not let a stale disposer remove a newly-created adapter for\n\t\t\t\t\t// the same actor generation.\n\t\t\t\t\tif (workflowInspectors.get(key) === workflowInspector) {\n\t\t\t\t\t\tworkflowInspectors.delete(key);\n\t\t\t\t\t}\n\t\t\t\t},\n\t\t\t};\n\t\t},\n\t});\n}\n","import type {\n\tBranchConfig,\n\tBranchOutput,\n\tEntryKindType,\n\tLoopConfig,\n\tLoopResult,\n\tStepConfig,\n\tTryBlockConfig,\n\tTryBlockResult,\n\tTryStepConfig,\n\tTryStepResult,\n\tWorkflowContextInterface,\n\tWorkflowQueueMessage,\n} from \"@rivetkit/workflow-engine\";\nimport { EvictedError } from \"@rivetkit/workflow-engine\";\nimport type {\n\tQueueFilterName,\n\tQueueNextBatchOptions,\n\tQueueNextOptions,\n\tQueueResultMessageForName,\n\tRunContext,\n} from \"@/actor/config\";\nimport { ACTOR_CONTEXT_INTERNAL_SYMBOL } from \"@/actor/config\";\nimport type {\n\tAnyActorDefinition,\n\tBaseActorDefinition,\n} from \"@/actor/definition\";\nimport type {\n\tEventSchemaConfig,\n\tInferEventArgs,\n\tInferSchemaMap,\n\tQueueSchemaConfig,\n} from \"@/actor/schema\";\nimport type { Client } from \"@/client/client\";\nimport type {\n\tAnyDatabaseProvider,\n\tInferDatabaseClient,\n} from \"@/common/database/config\";\nimport type { Registry } from \"@/registry\";\nimport { WORKFLOW_GUARD_KV_KEY } from \"./constants\";\n\ntype WorkflowActorQueueNextOptions<\n\tTName extends string,\n\tTCompletable extends boolean,\n> = Omit<QueueNextOptions<TName, TCompletable>, \"signal\">;\n\ntype WorkflowActorQueueNextOptionsFallback<TCompletable extends boolean> = Omit<\n\tQueueNextOptions<string, TCompletable>,\n\t\"signal\"\n>;\n\ntype WorkflowActorQueueNextBatchOptions<\n\tTName extends string,\n\tTCompletable extends boolean,\n> = Omit<QueueNextBatchOptions<TName, TCompletable>, \"signal\">;\n\ntype WorkflowActorQueueNextBatchOptionsFallback<TCompletable extends boolean> =\n\tOmit<QueueNextBatchOptions<string, TCompletable>, \"signal\">;\n\n// Step run callbacks receive a WorkflowStepContext, which is the only place\n// actor data (state/db/vars/client) may be touched.\ntype WorkflowStepRun<\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = (\n\tstep: WorkflowStepContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>,\n) => Promise<T>;\n\n// Step rollback callbacks compensate a committed step, so they also run with a\n// WorkflowStepContext to mutate actor data.\ntype WorkflowStepRollback<\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = (\n\tstep: WorkflowStepContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>,\n\toutput: T,\n) => Promise<void>;\n\n// Orchestration callbacks (try/loop/race/join) receive a WorkflowContext,\n// because inside them you sequence further steps rather than touch actor data.\ntype WorkflowContextRun<\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = (\n\tctx: WorkflowContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>,\n) => Promise<T>;\n\nexport type WorkflowStepConfig<\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = Omit<StepConfig<T>, \"run\" | \"rollback\"> & {\n\trun: WorkflowStepRun<\n\t\tT,\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n\trollback?: WorkflowStepRollback<\n\t\tT,\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n};\n\nexport type WorkflowTryStepConfig<\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = Omit<TryStepConfig<T>, \"run\" | \"rollback\"> & {\n\trun: WorkflowStepRun<\n\t\tT,\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n\trollback?: WorkflowStepRollback<\n\t\tT,\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n};\n\nexport type WorkflowTryConfig<\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = Omit<TryBlockConfig<T>, \"run\"> & {\n\trun: WorkflowContextRun<\n\t\tT,\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n};\n\nexport type WorkflowLoopConfig<\n\tS,\n\tT,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = Omit<LoopConfig<S, T>, \"run\"> & {\n\trun: (\n\t\tctx: WorkflowContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t\tstate: S,\n\t) => Promise<\n\t\tLoopResult<S, T> | (S extends undefined ? undefined | void : never)\n\t>;\n};\n\nexport type WorkflowBranchConfig<\n\tTOutput,\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig,\n\tTQueues extends QueueSchemaConfig,\n> = {\n\trun: WorkflowContextRun<\n\t\tTOutput,\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n};\n\n// Marks a step context inactive once its step has finished. Module-private so it\n// never appears on the public surface.\nconst DEACTIVATE_STEP = Symbol(\"workflow.step.deactivate\");\n\n/**\n * Stops the workflow before it starts more user code once its actor generation was declared\n * lost. The engine treats the eviction like a sleep and replays from history in the next\n * generation.\n */\nexport function throwIfGenerationLost(runCtx: unknown): void {\n\tconst actor = (\n\t\trunCtx as { [ACTOR_CONTEXT_INTERNAL_SYMBOL]?: { isLost?: boolean } }\n\t)[ACTOR_CONTEXT_INTERNAL_SYMBOL];\n\tif (actor?.isLost === true) {\n\t\tthrow new EvictedError();\n\t}\n}\n\n/**\n * The context handed to a workflow step (`step` / `tryStep` callbacks). This is\n * the only scope where actor data (state, vars, db, client) and side effects\n * (broadcast, queue.send) are reachable. It is valid only while its step is\n * executing; using it after the step resolves throws.\n */\nexport class WorkflowStepContext<\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig = Record<never, never>,\n\tTQueues extends QueueSchemaConfig = Record<never, never>,\n> {\n\t#runCtx: RunContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n\t#active = true;\n\t#onGuardViolation: () => void;\n\n\tconstructor(\n\t\trunCtx: RunContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t\tonGuardViolation: () => void,\n\t) {\n\t\tthis.#runCtx = runCtx;\n\t\tthis.#onGuardViolation = onGuardViolation;\n\t}\n\n\t[DEACTIVATE_STEP](): void {\n\t\tthis.#active = false;\n\t}\n\n\t#ensureActive(feature: string): void {\n\t\tif (!this.#active) {\n\t\t\tthis.#onGuardViolation();\n\t\t\tthrow new Error(\n\t\t\t\t`${feature} is only available inside workflow steps`,\n\t\t\t);\n\t\t}\n\t}\n\n\tget actorId(): string {\n\t\treturn this.#runCtx.actorId;\n\t}\n\n\tget name(): string {\n\t\treturn this.#runCtx.name;\n\t}\n\n\tget key(): string[] {\n\t\treturn this.#runCtx.key;\n\t}\n\n\tget log() {\n\t\treturn this.#runCtx.log;\n\t}\n\n\tget abortSignal(): AbortSignal {\n\t\treturn this.#runCtx.abortSignal;\n\t}\n\n\tget state(): TState extends never ? never : TState {\n\t\tthis.#ensureActive(\"state\");\n\t\treturn this.#runCtx.state as TState extends never ? never : TState;\n\t}\n\n\tget vars(): TVars extends never ? never : TVars {\n\t\tthis.#ensureActive(\"vars\");\n\t\treturn this.#runCtx.vars as TVars extends never ? never : TVars;\n\t}\n\n\tget db(): TDatabase extends never ? never : InferDatabaseClient<TDatabase> {\n\t\tthis.#ensureActive(\"db\");\n\t\treturn this.#runCtx.db as TDatabase extends never\n\t\t\t? never\n\t\t\t: InferDatabaseClient<TDatabase>;\n\t}\n\n\tclient<R extends Registry<any> = Registry<any>>(): Client<R> {\n\t\tthis.#ensureActive(\"client\");\n\t\treturn this.#runCtx.client<R>();\n\t}\n\n\tbroadcast<K extends keyof TEvents & string>(\n\t\tname: K,\n\t\t...args: InferEventArgs<InferSchemaMap<TEvents>[K]>\n\t): void;\n\tbroadcast(\n\t\tname: keyof TEvents extends never ? string : never,\n\t\t...args: Array<unknown>\n\t): void;\n\tbroadcast(name: string, ...args: Array<unknown>): void {\n\t\tthis.#ensureActive(\"broadcast\");\n\t\tthis.#runCtx.broadcast(\n\t\t\tname as never,\n\t\t\t...(args as unknown[] as never[]),\n\t\t);\n\t}\n\n\tget queue() {\n\t\tconst self = this;\n\t\tfunction send<K extends keyof TQueues & string>(\n\t\t\tname: K,\n\t\t\tbody: InferSchemaMap<TQueues>[K],\n\t\t): Promise<void>;\n\t\tfunction send(\n\t\t\tname: keyof TQueues extends never ? string : never,\n\t\t\tbody: unknown,\n\t\t): Promise<void>;\n\t\tasync function send(name: string, body: unknown): Promise<void> {\n\t\t\tself.#ensureActive(\"queue.send\");\n\t\t\tawait self.#runCtx.queue.send(name as never, body as never);\n\t\t}\n\t\treturn { send };\n\t}\n\n\t/**\n\t * Holds the actor awake for the duration of the provided promise. The actor\n\t * cannot idle-sleep or finalize the sleep grace period until the promise\n\t * settles.\n\t */\n\tkeepAwake<T>(promise: Promise<T>): Promise<T> {\n\t\tthis.#ensureActive(\"keepAwake\");\n\t\treturn this.#runCtx.keepAwake(promise);\n\t}\n\n\t/**\n\t * Registers a promise that the sleep grace period will wait on. Use this for\n\t * best-effort flush/cleanup work that may complete inside the grace window.\n\t */\n\twaitUntil(promise: Promise<void>): void {\n\t\tthis.#ensureActive(\"waitUntil\");\n\t\tthis.#runCtx.waitUntil(promise);\n\t}\n\n\tdestroy(): void {\n\t\tthis.#ensureActive(\"destroy\");\n\t\tthis.#runCtx.destroy();\n\t}\n}\n\n/**\n * The context handed to the workflow function and to orchestration callbacks\n * (`try` / `loop` / `race` / `join`). It exposes the deterministic, replayable\n * workflow primitives (step, sleep, queue waits, control flow). It deliberately\n * does NOT expose actor data; reach `state`, `db`, `vars`, `client`, and\n * `broadcast` through the step context passed to `step` / `tryStep`.\n */\nexport class WorkflowContext<\n\tTState,\n\tTConnParams,\n\tTConnState,\n\tTVars,\n\tTInput,\n\tTDatabase extends AnyDatabaseProvider,\n\tTEvents extends EventSchemaConfig = Record<never, never>,\n\tTQueues extends QueueSchemaConfig = Record<never, never>,\n> {\n\t#inner: WorkflowContextInterface;\n\t#runCtx: RunContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t>;\n\tconstructor(\n\t\tinner: WorkflowContextInterface,\n\t\trunCtx: RunContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t) {\n\t\tthis.#inner = inner;\n\t\tthis.#runCtx = runCtx;\n\t}\n\n\tget workflowId(): string {\n\t\treturn this.#inner.workflowId;\n\t}\n\n\tget abortSignal(): AbortSignal {\n\t\treturn this.#inner.abortSignal;\n\t}\n\n\tget actorId(): string {\n\t\treturn this.#runCtx.actorId;\n\t}\n\n\tget name(): string {\n\t\treturn this.#runCtx.name;\n\t}\n\n\tget key(): string[] {\n\t\treturn this.#runCtx.key;\n\t}\n\n\tget log() {\n\t\treturn this.#runCtx.log;\n\t}\n\n\tget queue() {\n\t\tconst self = this;\n\t\tfunction next<\n\t\t\tconst TName extends QueueFilterName<TQueues>,\n\t\t\tconst TCompletable extends boolean = false,\n\t\t>(\n\t\t\tname: string,\n\t\t\topts?: WorkflowActorQueueNextOptions<TName, TCompletable>,\n\t\t): Promise<QueueResultMessageForName<TQueues, TName, TCompletable>>;\n\t\tfunction next<const TCompletable extends boolean = false>(\n\t\t\tname: string,\n\t\t\topts?: WorkflowActorQueueNextOptionsFallback<TCompletable>,\n\t\t): Promise<\n\t\t\tQueueResultMessageForName<\n\t\t\t\tTQueues,\n\t\t\t\tQueueFilterName<TQueues>,\n\t\t\t\tTCompletable\n\t\t\t>\n\t\t>;\n\t\t// The implementation signature stays broad so the schema-typed public\n\t\t// overloads above remain compatible with it.\n\t\tasync function next(name: string, opts?: any): Promise<any> {\n\t\t\tconst message = await self.#inner.queue.next(name, opts);\n\t\t\treturn self.#toActorQueueMessage(message);\n\t\t}\n\n\t\tfunction nextBatch<\n\t\t\tconst TName extends QueueFilterName<TQueues>,\n\t\t\tconst TCompletable extends boolean = false,\n\t\t>(\n\t\t\tname: string,\n\t\t\topts?: WorkflowActorQueueNextBatchOptions<TName, TCompletable>,\n\t\t): Promise<\n\t\t\tArray<QueueResultMessageForName<TQueues, TName, TCompletable>>\n\t\t>;\n\t\tfunction nextBatch<const TCompletable extends boolean = false>(\n\t\t\tname: string,\n\t\t\topts?: WorkflowActorQueueNextBatchOptionsFallback<TCompletable>,\n\t\t): Promise<\n\t\t\tArray<\n\t\t\t\tQueueResultMessageForName<\n\t\t\t\t\tTQueues,\n\t\t\t\t\tQueueFilterName<TQueues>,\n\t\t\t\t\tTCompletable\n\t\t\t\t>\n\t\t\t>\n\t\t>;\n\t\tasync function nextBatch(name: string, opts?: any): Promise<any> {\n\t\t\tconst messages = await self.#inner.queue.nextBatch(name, opts);\n\t\t\treturn messages.map((message) =>\n\t\t\t\tself.#toActorQueueMessage(message),\n\t\t\t);\n\t\t}\n\n\t\treturn {\n\t\t\tnext,\n\t\t\tnextBatch,\n\t\t};\n\t}\n\n\tstep<T>(\n\t\tname: string,\n\t\trun: WorkflowStepRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<T>;\n\tstep<T>(\n\t\tconfig: WorkflowStepConfig<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<T>;\n\tasync step<T>(\n\t\tnameOrConfig:\n\t\t\t| string\n\t\t\t| WorkflowStepConfig<\n\t\t\t\t\tT,\n\t\t\t\t\tTState,\n\t\t\t\t\tTConnParams,\n\t\t\t\t\tTConnState,\n\t\t\t\t\tTVars,\n\t\t\t\t\tTInput,\n\t\t\t\t\tTDatabase,\n\t\t\t\t\tTEvents,\n\t\t\t\t\tTQueues\n\t\t\t >,\n\t\trun?: WorkflowStepRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<T> {\n\t\tif (typeof nameOrConfig === \"string\") {\n\t\t\tif (!run) {\n\t\t\t\tthrow new Error(\"Step run function missing\");\n\t\t\t}\n\t\t\tconst stepRun = run;\n\t\t\treturn await this.#wrapActive(() =>\n\t\t\t\tthis.#inner.step(nameOrConfig, () => this.#runStep(stepRun)),\n\t\t\t);\n\t\t}\n\t\tconst stepConfig = nameOrConfig;\n\t\tconst rollback = stepConfig.rollback;\n\t\tconst config: StepConfig<T> = {\n\t\t\t...stepConfig,\n\t\t\trun: () => this.#runStep(stepConfig.run),\n\t\t\trollback: rollback\n\t\t\t\t? (_ctx, output) => this.#runRollback(rollback, output)\n\t\t\t\t: undefined,\n\t\t};\n\t\treturn await this.#wrapActive(() => this.#inner.step(config));\n\t}\n\n\ttryStep<T>(\n\t\tname: string,\n\t\trun: WorkflowStepRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<TryStepResult<T>>;\n\ttryStep<T>(\n\t\tconfig: WorkflowTryStepConfig<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<TryStepResult<T>>;\n\tasync tryStep<T>(\n\t\tnameOrConfig:\n\t\t\t| string\n\t\t\t| WorkflowTryStepConfig<\n\t\t\t\t\tT,\n\t\t\t\t\tTState,\n\t\t\t\t\tTConnParams,\n\t\t\t\t\tTConnState,\n\t\t\t\t\tTVars,\n\t\t\t\t\tTInput,\n\t\t\t\t\tTDatabase,\n\t\t\t\t\tTEvents,\n\t\t\t\t\tTQueues\n\t\t\t >,\n\t\trun?: WorkflowStepRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<TryStepResult<T>> {\n\t\tif (typeof nameOrConfig === \"string\") {\n\t\t\tif (!run) {\n\t\t\t\tthrow new Error(\"Step run function missing\");\n\t\t\t}\n\t\t\tconst stepRun = run;\n\t\t\treturn await this.#wrapActive(() =>\n\t\t\t\tthis.#inner.tryStep(nameOrConfig, () => this.#runStep(stepRun)),\n\t\t\t);\n\t\t}\n\t\tconst stepConfig = nameOrConfig;\n\t\tconst rollback = stepConfig.rollback;\n\t\tconst config: TryStepConfig<T> = {\n\t\t\t...stepConfig,\n\t\t\trun: () => this.#runStep(stepConfig.run),\n\t\t\trollback: rollback\n\t\t\t\t? (_ctx, output) => this.#runRollback(rollback, output)\n\t\t\t\t: undefined,\n\t\t};\n\t\treturn await this.#wrapActive(() => this.#inner.tryStep(config));\n\t}\n\n\ttry<T>(\n\t\tname: string,\n\t\trun: WorkflowContextRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<TryBlockResult<T>>;\n\ttry<T>(\n\t\tconfig: WorkflowTryConfig<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<TryBlockResult<T>>;\n\tasync try<T>(\n\t\tnameOrConfig:\n\t\t\t| string\n\t\t\t| WorkflowTryConfig<\n\t\t\t\t\tT,\n\t\t\t\t\tTState,\n\t\t\t\t\tTConnParams,\n\t\t\t\t\tTConnState,\n\t\t\t\t\tTVars,\n\t\t\t\t\tTInput,\n\t\t\t\t\tTDatabase,\n\t\t\t\t\tTEvents,\n\t\t\t\t\tTQueues\n\t\t\t >,\n\t\trun?: WorkflowContextRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<TryBlockResult<T>> {\n\t\tif (typeof nameOrConfig === \"string\") {\n\t\t\tif (!run) {\n\t\t\t\tthrow new Error(\"Try run function missing\");\n\t\t\t}\n\t\t\tconst tryRun = run;\n\t\t\treturn await this.#wrapActive(() =>\n\t\t\t\tthis.#inner.try(nameOrConfig, async (ctx) =>\n\t\t\t\t\ttryRun(this.#createChildContext(ctx)),\n\t\t\t\t),\n\t\t\t);\n\t\t}\n\t\tconst tryConfig = nameOrConfig;\n\t\tconst config: TryBlockConfig<T> = {\n\t\t\t...tryConfig,\n\t\t\trun: async (ctx) => tryConfig.run(this.#createChildContext(ctx)),\n\t\t};\n\t\treturn await this.#wrapActive(() => this.#inner.try(config));\n\t}\n\n\tloop<T>(\n\t\tname: string,\n\t\trun: (\n\t\t\tctx: WorkflowContext<\n\t\t\t\tTState,\n\t\t\t\tTConnParams,\n\t\t\t\tTConnState,\n\t\t\t\tTVars,\n\t\t\t\tTInput,\n\t\t\t\tTDatabase,\n\t\t\t\tTEvents,\n\t\t\t\tTQueues\n\t\t\t>,\n\t\t) => Promise<LoopResult<undefined, T> | undefined | void>,\n\t): Promise<T>;\n\tloop<S, T>(\n\t\tconfig: WorkflowLoopConfig<\n\t\t\tS,\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<T>;\n\tasync loop(\n\t\tnameOrConfig:\n\t\t\t| string\n\t\t\t| WorkflowLoopConfig<\n\t\t\t\t\tany,\n\t\t\t\t\tany,\n\t\t\t\t\tTState,\n\t\t\t\t\tTConnParams,\n\t\t\t\t\tTConnState,\n\t\t\t\t\tTVars,\n\t\t\t\t\tTInput,\n\t\t\t\t\tTDatabase,\n\t\t\t\t\tTEvents,\n\t\t\t\t\tTQueues\n\t\t\t >,\n\t\trun?: (\n\t\t\tctx: WorkflowContext<\n\t\t\t\tTState,\n\t\t\t\tTConnParams,\n\t\t\t\tTConnState,\n\t\t\t\tTVars,\n\t\t\t\tTInput,\n\t\t\t\tTDatabase,\n\t\t\t\tTEvents,\n\t\t\t\tTQueues\n\t\t\t>,\n\t\t) => Promise<LoopResult<undefined, any> | undefined | void>,\n\t): Promise<any> {\n\t\tif (typeof nameOrConfig === \"string\") {\n\t\t\tif (!run) {\n\t\t\t\tthrow new Error(\"Loop run function missing\");\n\t\t\t}\n\t\t\tconst loopRun = run;\n\t\t\treturn await this.#wrapActive(() =>\n\t\t\t\tthis.#inner.loop(\n\t\t\t\t\tnameOrConfig,\n\t\t\t\t\t// A void return (no explicit Loop result) is undefined at\n\t\t\t\t\t// runtime, which the engine treats as continue.\n\t\t\t\t\tasync (\n\t\t\t\t\t\tctx,\n\t\t\t\t\t): Promise<LoopResult<undefined, any> | undefined> =>\n\t\t\t\t\t\t(await loopRun(this.#createChildContext(ctx))) ??\n\t\t\t\t\t\tundefined,\n\t\t\t\t),\n\t\t\t);\n\t\t}\n\t\tconst loopConfig = nameOrConfig;\n\t\tconst wrapped: LoopConfig<any, any> = {\n\t\t\t...loopConfig,\n\t\t\trun: (async (ctx, state) =>\n\t\t\t\t(await loopConfig.run(this.#createChildContext(ctx), state)) ??\n\t\t\t\tundefined) as LoopConfig<any, any>[\"run\"],\n\t\t};\n\t\treturn await this.#wrapActive(() => this.#inner.loop(wrapped));\n\t}\n\n\tsleep(name: string, durationMs: number): Promise<void> {\n\t\treturn this.#inner.sleep(name, durationMs);\n\t}\n\n\tsleepUntil(name: string, timestampMs: number): Promise<void> {\n\t\treturn this.#inner.sleepUntil(name, timestampMs);\n\t}\n\n\tasync rollbackCheckpoint(name: string): Promise<void> {\n\t\tawait this.#wrapActive(() => this.#inner.rollbackCheckpoint(name));\n\t}\n\n\tasync join<\n\t\tT extends Record<\n\t\t\tstring,\n\t\t\tWorkflowBranchConfig<\n\t\t\t\tunknown,\n\t\t\t\tTState,\n\t\t\t\tTConnParams,\n\t\t\t\tTConnState,\n\t\t\t\tTVars,\n\t\t\t\tTInput,\n\t\t\t\tTDatabase,\n\t\t\t\tTEvents,\n\t\t\t\tTQueues\n\t\t\t>\n\t\t>,\n\t>(\n\t\tname: string,\n\t\tbranches: T,\n\t): Promise<{ [K in keyof T]: Awaited<ReturnType<T[K][\"run\"]>> }>;\n\tasync join<T extends Record<string, BranchConfig<unknown>>>(\n\t\tname: string,\n\t\tbranches: T,\n\t): Promise<{ [K in keyof T]: BranchOutput<T[K]> }>;\n\tasync join(\n\t\tname: string,\n\t\tbranches: Record<string, { run: (ctx: any) => Promise<unknown> }>,\n\t) {\n\t\tconst wrappedBranches = Object.fromEntries(\n\t\t\tObject.entries(branches).map(([key, branch]) => [\n\t\t\t\tkey,\n\t\t\t\t{\n\t\t\t\t\trun: async (ctx: WorkflowContextInterface) =>\n\t\t\t\t\t\tbranch.run(this.#createChildContext(ctx)),\n\t\t\t\t},\n\t\t\t]),\n\t\t) as Record<string, BranchConfig<unknown>>;\n\t\treturn await this.#wrapActive(() =>\n\t\t\tthis.#inner.join(name, wrappedBranches),\n\t\t);\n\t}\n\n\tasync race<T>(\n\t\tname: string,\n\t\tbranches: Array<{\n\t\t\tname: string;\n\t\t\trun: WorkflowContextRun<\n\t\t\t\tT,\n\t\t\t\tTState,\n\t\t\t\tTConnParams,\n\t\t\t\tTConnState,\n\t\t\t\tTVars,\n\t\t\t\tTInput,\n\t\t\t\tTDatabase,\n\t\t\t\tTEvents,\n\t\t\t\tTQueues\n\t\t\t>;\n\t\t}>,\n\t): Promise<{ winner: string; value: T }> {\n\t\tconst wrappedBranches = branches.map((branch) => ({\n\t\t\tname: branch.name,\n\t\t\trun: (ctx: WorkflowContextInterface) =>\n\t\t\t\tbranch.run(this.#createChildContext(ctx)),\n\t\t}));\n\t\treturn (await this.#wrapActive(() =>\n\t\t\tthis.#inner.race(name, wrappedBranches),\n\t\t)) as { winner: string; value: T };\n\t}\n\n\tasync removed(name: string, originalType: EntryKindType): Promise<void> {\n\t\tawait this.#wrapActive(() => this.#inner.removed(name, originalType));\n\t}\n\n\tgetVersion(name: string, latest: number): Promise<number> {\n\t\treturn this.#wrapActive(() => this.#inner.getVersion(name, latest));\n\t}\n\n\tisEvicted(): boolean {\n\t\treturn this.#inner.isEvicted();\n\t}\n\n\t// Runs a user step body inside a fresh step context and deactivates the step\n\t// context once the body settles. Workflows are roll-forward only, so state and\n\t// vars mutations made before a failure always remain visible.\n\tasync #runStep<T>(\n\t\trun: WorkflowStepRun<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t): Promise<T> {\n\t\tconst stepCtx = new WorkflowStepContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>(this.#runCtx, () => this.#markGuardTriggered());\n\n\t\tthrowIfGenerationLost(this.#runCtx);\n\t\ttry {\n\t\t\treturn await run(stepCtx);\n\t\t} finally {\n\t\t\tstepCtx[DEACTIVATE_STEP]();\n\t\t}\n\t}\n\n\t#toActorQueueMessage<T>(\n\t\tmessage: WorkflowQueueMessage<T>,\n\t): WorkflowQueueMessage<T> & { id: bigint } {\n\t\tlet id: bigint;\n\t\ttry {\n\t\t\tid = BigInt(message.id);\n\t\t} catch {\n\t\t\tthrow new Error(`Invalid queue message id \"${message.id}\"`);\n\t\t}\n\t\treturn {\n\t\t\tid,\n\t\t\tname: message.name,\n\t\t\tbody: message.body,\n\t\t\tcreatedAt: message.createdAt,\n\t\t\t...(message.complete ? { complete: message.complete } : {}),\n\t\t};\n\t}\n\n\t// Runs a step rollback compensation with an active step context.\n\tasync #runRollback<T>(\n\t\trollback: WorkflowStepRollback<\n\t\t\tT,\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>,\n\t\toutput: T,\n\t): Promise<void> {\n\t\tconst stepCtx = new WorkflowStepContext<\n\t\t\tTState,\n\t\t\tTConnParams,\n\t\t\tTConnState,\n\t\t\tTVars,\n\t\t\tTInput,\n\t\t\tTDatabase,\n\t\t\tTEvents,\n\t\t\tTQueues\n\t\t>(this.#runCtx, () => this.#markGuardTriggered());\n\t\tthrowIfGenerationLost(this.#runCtx);\n\t\ttry {\n\t\t\tawait rollback(stepCtx, output);\n\t\t} finally {\n\t\t\tstepCtx[DEACTIVATE_STEP]();\n\t\t}\n\t}\n\n\tasync #wrapActive<T>(run: () => Promise<T>): Promise<T> {\n\t\treturn await this.#runCtx.internalKeepAwake(run);\n\t}\n\n\t// Records that a step context was used outside its step. Mirrors the value\n\t// onto actor state and a KV flag so callers can observe the violation.\n\t#markGuardTriggered(): void {\n\t\ttry {\n\t\t\tconst state = this.#runCtx.state as Record<string, unknown>;\n\t\t\tif (\n\t\t\t\tstate &&\n\t\t\t\ttypeof state === \"object\" &&\n\t\t\t\t\"guardTriggered\" in state\n\t\t\t) {\n\t\t\t\t(state as Record<string, unknown>).guardTriggered = true;\n\t\t\t}\n\t\t} catch {\n\t\t\t// Ignore if state is unavailable\n\t\t}\n\n\t\tthis.#runCtx.waitUntil(\n\t\t\t(async () => {\n\t\t\t\ttry {\n\t\t\t\t\tawait this.#runCtx.kv.put(WORKFLOW_GUARD_KV_KEY, \"true\");\n\t\t\t\t} catch (error) {\n\t\t\t\t\tthis.#runCtx.log.error({\n\t\t\t\t\t\tmsg: \"failed to persist workflow guard flag\",\n\t\t\t\t\t\terror,\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t})(),\n\t\t);\n\t}\n\n\t#createChildContext(\n\t\tctx: WorkflowContextInterface,\n\t): WorkflowContext<\n\t\tTState,\n\t\tTConnParams,\n\t\tTConnState,\n\t\tTVars,\n\t\tTInput,\n\t\tTDatabase,\n\t\tTEvents,\n\t\tTQueues\n\t> {\n\t\t// Every try, loop, join, and race body runs through a child context.\n\t\tthrowIfGenerationLost(this.#runCtx);\n\t\treturn new WorkflowContext(ctx, this.#runCtx);\n\t}\n}\n\nexport type WorkflowContextOf<AD extends AnyActorDefinition> =\n\tAD extends BaseActorDefinition<\n\t\tinfer S,\n\t\tinfer CP,\n\t\tinfer CS,\n\t\tinfer V,\n\t\tinfer I,\n\t\tinfer DB extends AnyDatabaseProvider,\n\t\tinfer E extends EventSchemaConfig,\n\t\tinfer Q extends QueueSchemaConfig,\n\t\tany\n\t>\n\t\t? WorkflowContext<S, CP, CS, V, I, DB, E, Q>\n\t\t: never;\n\nexport type WorkflowLoopContextOf<AD extends AnyActorDefinition> =\n\tWorkflowContextOf<AD>;\n\nexport type WorkflowBranchContextOf<AD extends AnyActorDefinition> =\n\tWorkflowContextOf<AD>;\n\nexport type WorkflowStepContextOf<AD extends AnyActorDefinition> =\n\tAD extends BaseActorDefinition<\n\t\tinfer S,\n\t\tinfer CP,\n\t\tinfer CS,\n\t\tinfer V,\n\t\tinfer I,\n\t\tinfer DB extends AnyDatabaseProvider,\n\t\tinfer E extends EventSchemaConfig,\n\t\tinfer Q extends QueueSchemaConfig,\n\t\tany\n\t>\n\t\t? WorkflowStepContext<S, CP, CS, V, I, DB, E, Q>\n\t\t: never;\n","export const WORKFLOW_GUARD_KV_KEY = \"__rivet_actor_workflow_guard_triggered\";\n","import type {\n\tEngineDriver,\n\tKVEntry,\n\tKVWrite,\n\tMessage,\n\tWorkflowMessageDriver,\n\tWorkflowTelemetryDriver,\n} from \"@rivetkit/workflow-engine\";\nimport type { RunContext } from \"@/actor/config\";\nimport type { AnyStaticActorInstance } from \"@/actor/definition\";\nimport { makeWorkflowKey, workflowStoragePrefix } from \"@/actor/keys\";\nimport type { RawAccess } from \"@/common/database/config\";\nimport { throwIfGenerationLost } from \"./context\";\n\nconst WORKFLOW_STORAGE_PREFIX = workflowStoragePrefix();\n// Keep workflow flushes below depot's 320-dirty-page commit ceiling. The\n// former actor-KV transport allowed 976 KiB batches; SQLite also dirties table\n// and index pages, so it needs the same conservative budget as core imports.\nconst WORKFLOW_SQLITE_MAX_BATCH_ROWS = 128;\nconst WORKFLOW_SQLITE_MAX_BATCH_BYTES = 512 * 1024;\n\n// Mirrors the element shape returned by `queueManager.receive`. The actor\n// instance is reached through a loose type here, so the call's result is\n// untyped and the message must be annotated explicitly.\ninterface ReceivedQueueMessage {\n\tid: bigint;\n\tname: string;\n\tbody: unknown;\n\tcreatedAt: number;\n\tcomplete?: (response?: unknown) => Promise<void>;\n}\n\ntype KVEntryTuple = [Uint8Array, Uint8Array];\n\nfunction stripWorkflowKey(prefixed: Uint8Array): Uint8Array {\n\treturn prefixed.slice(WORKFLOW_STORAGE_PREFIX.length);\n}\n\nfunction computeUpperBound(prefix: Uint8Array): Uint8Array | null {\n\tconst upperBound = prefix.slice();\n\tfor (let i = upperBound.length - 1; i >= 0; i--) {\n\t\tif (upperBound[i] !== 0xff) {\n\t\t\tupperBound[i]++;\n\t\t\treturn upperBound.slice(0, i + 1);\n\t\t}\n\t}\n\treturn null;\n}\n\nfunction normalizeSqlBlob(value: unknown): Uint8Array {\n\tif (value instanceof Uint8Array) {\n\t\treturn value;\n\t}\n\tif (value instanceof ArrayBuffer) {\n\t\treturn new Uint8Array(value);\n\t}\n\tif (ArrayBuffer.isView(value)) {\n\t\treturn new Uint8Array(value.buffer, value.byteOffset, value.byteLength);\n\t}\n\tif (Array.isArray(value)) {\n\t\tconst bytes = new Uint8Array(value.length);\n\t\tfor (const [i, item] of value.entries()) {\n\t\t\tif (!Number.isInteger(item) || item < 0 || item > 255) {\n\t\t\t\tthrow new Error(\"workflow sqlite value was not a byte array\");\n\t\t\t}\n\t\t\tbytes[i] = item;\n\t\t}\n\t\treturn bytes;\n\t}\n\tthrow new Error(\"workflow sqlite value was not a blob\");\n}\n\nfunction guardRawAccessAgainstLost(\n\taccess: RawAccess,\n\trunCtx: RunContext<any, any, any, any, any, any, any, any> | undefined,\n): RawAccess {\n\treturn {\n\t\t...access,\n\t\texecute: ((...args: Parameters<RawAccess[\"execute\"]>) => {\n\t\t\tthrowIfGenerationLost(runCtx);\n\t\t\treturn access.execute(...args);\n\t\t}) as RawAccess[\"execute\"],\n\t\ttransaction: (callback, options) => {\n\t\t\tthrowIfGenerationLost(runCtx);\n\t\t\treturn access.transaction(\n\t\t\t\t(tx) => callback(guardRawAccessAgainstLost(tx, runCtx)),\n\t\t\t\toptions,\n\t\t\t);\n\t\t},\n\t};\n}\n\nfunction runtimeDbFromContext(\n\trunCtx?: RunContext<any, any, any, any, any, any, any, any>,\n): RawAccess | undefined {\n\tconst db = (runCtx as unknown as { db?: unknown } | undefined)?.db;\n\tif (\n\t\tdb &&\n\t\ttypeof db === \"object\" &&\n\t\t\"execute\" in db &&\n\t\t\"transaction\" in db\n\t) {\n\t\t// The client can come from a user database provider. Workflow storage calls it after\n\t\t// awaits, including inside its own transactions, so every call checks that the\n\t\t// generation was not lost first.\n\t\treturn guardRawAccessAgainstLost(db as RawAccess, runCtx);\n\t}\n\treturn undefined;\n}\n\nclass WorkflowStorage {\n\t#db: RawAccess;\n\n\tconstructor(db?: RawAccess) {\n\t\tif (!db) {\n\t\t\tthrow new Error(\n\t\t\t\t\"workflow storage requires embedded SQLite; actor KV workflow storage is no longer supported\",\n\t\t\t);\n\t\t}\n\t\tthis.#db = db;\n\t}\n\n\tasync get(key: Uint8Array): Promise<Uint8Array | null> {\n\t\tconst prefixed = makeWorkflowKey(key);\n\t\tconst rows = await this.#db.execute<{ value: unknown }>(\n\t\t\t\"SELECT value FROM _rivet_wf_kv WHERE key = ?\",\n\t\t\tprefixed,\n\t\t);\n\t\treturn rows[0]?.value == null ? null : normalizeSqlBlob(rows[0].value);\n\t}\n\n\tasync set(key: Uint8Array, value: Uint8Array): Promise<void> {\n\t\tawait this.batch([{ key, value }]);\n\t}\n\n\tasync delete(key: Uint8Array): Promise<void> {\n\t\tconst prefixed = makeWorkflowKey(key);\n\t\tawait this.#db.execute(\n\t\t\t\"DELETE FROM _rivet_wf_kv WHERE key = ?\",\n\t\t\tprefixed,\n\t\t);\n\t}\n\n\tasync batchDelete(keys: Uint8Array[]): Promise<void> {\n\t\tif (keys.length === 0) return;\n\t\tawait this.deleteRawKeys(keys.map((key) => makeWorkflowKey(key)));\n\t}\n\n\tasync deletePrefix(prefix: Uint8Array): Promise<void> {\n\t\tconst start = makeWorkflowKey(prefix);\n\t\tconst end = computeUpperBound(start);\n\t\tif (end) {\n\t\t\tawait this.#db.execute(\n\t\t\t\t\"DELETE FROM _rivet_wf_kv WHERE key >= ? AND key < ?\",\n\t\t\t\tstart,\n\t\t\t\tend,\n\t\t\t);\n\t\t\treturn;\n\t\t}\n\t\tconst entries = await this.listRaw(start);\n\t\tawait this.deleteRawKeys(\n\t\t\tentries\n\t\t\t\t.filter(([key]) => keyStartsWith(key, start))\n\t\t\t\t.map(([key]) => key),\n\t\t);\n\t}\n\n\tasync deleteRange(start: Uint8Array, end: Uint8Array): Promise<void> {\n\t\tconst prefixedStart = makeWorkflowKey(start);\n\t\tconst prefixedEnd = makeWorkflowKey(end);\n\t\tawait this.#db.execute(\n\t\t\t\"DELETE FROM _rivet_wf_kv WHERE key >= ? AND key < ?\",\n\t\t\tprefixedStart,\n\t\t\tprefixedEnd,\n\t\t);\n\t}\n\n\tasync list(prefix: Uint8Array): Promise<KVEntry[]> {\n\t\tconst prefixed = makeWorkflowKey(prefix);\n\t\tconst entries = await this.listRaw(prefixed);\n\t\treturn entries\n\t\t\t.filter(([key]) => keyStartsWith(key, prefixed))\n\t\t\t.map(([key, value]: KVEntryTuple) => ({\n\t\t\t\tkey: stripWorkflowKey(key),\n\t\t\t\tvalue,\n\t\t\t}));\n\t}\n\n\tasync batch(writes: KVWrite[]): Promise<void> {\n\t\tif (writes.length === 0) return;\n\t\tfor (const chunk of chunkWorkflowWrites(writes)) {\n\t\t\tawait this.#db.transaction(async (tx) => {\n\t\t\t\tfor (const { key, value } of chunk) {\n\t\t\t\t\tawait tx.execute(\n\t\t\t\t\t\t\"INSERT INTO _rivet_wf_kv (key, value) VALUES (?, ?) ON CONFLICT(key) DO UPDATE SET value = excluded.value\",\n\t\t\t\t\t\tmakeWorkflowKey(key),\n\t\t\t\t\t\tvalue,\n\t\t\t\t\t);\n\t\t\t\t}\n\t\t\t});\n\t\t}\n\t}\n\n\tasync listRaw(prefix: Uint8Array): Promise<KVEntryTuple[]> {\n\t\tconst end = computeUpperBound(prefix);\n\t\tconst rows = end\n\t\t\t? await this.#db.execute<{ key: unknown; value: unknown }>(\n\t\t\t\t\t\"SELECT key, value FROM _rivet_wf_kv WHERE key >= ? AND key < ? ORDER BY key ASC\",\n\t\t\t\t\tprefix,\n\t\t\t\t\tend,\n\t\t\t\t)\n\t\t\t: await this.#db.execute<{ key: unknown; value: unknown }>(\n\t\t\t\t\t\"SELECT key, value FROM _rivet_wf_kv WHERE key >= ? ORDER BY key ASC\",\n\t\t\t\t\tprefix,\n\t\t\t\t);\n\t\treturn rows.map((row) => [\n\t\t\tnormalizeSqlBlob(row.key),\n\t\t\tnormalizeSqlBlob(row.value),\n\t\t]);\n\t}\n\n\tasync deleteRawKeys(keys: Uint8Array[]): Promise<void> {\n\t\tfor (\n\t\t\tlet start = 0;\n\t\t\tstart < keys.length;\n\t\t\tstart += WORKFLOW_SQLITE_MAX_BATCH_ROWS\n\t\t) {\n\t\t\tawait this.#db.transaction(async (tx) => {\n\t\t\t\tfor (const key of keys.slice(\n\t\t\t\t\tstart,\n\t\t\t\t\tstart + WORKFLOW_SQLITE_MAX_BATCH_ROWS,\n\t\t\t\t)) {\n\t\t\t\t\tawait tx.execute(\n\t\t\t\t\t\t\"DELETE FROM _rivet_wf_kv WHERE key = ?\",\n\t\t\t\t\t\tkey,\n\t\t\t\t\t);\n\t\t\t\t}\n\t\t\t});\n\t\t}\n\t}\n}\n\nexport function chunkWorkflowWrites(writes: KVWrite[]): KVWrite[][] {\n\tconst chunks: KVWrite[][] = [];\n\tlet chunk: KVWrite[] = [];\n\tlet chunkBytes = 0;\n\tfor (const write of writes) {\n\t\tconst writeBytes =\n\t\t\tWORKFLOW_STORAGE_PREFIX.length +\n\t\t\twrite.key.length +\n\t\t\twrite.value.length;\n\t\tif (\n\t\t\tchunk.length > 0 &&\n\t\t\t(chunk.length >= WORKFLOW_SQLITE_MAX_BATCH_ROWS ||\n\t\t\t\tchunkBytes + writeBytes > WORKFLOW_SQLITE_MAX_BATCH_BYTES)\n\t\t) {\n\t\t\tchunks.push(chunk);\n\t\t\tchunk = [];\n\t\t\tchunkBytes = 0;\n\t\t}\n\t\tchunk.push(write);\n\t\tchunkBytes += writeBytes;\n\t}\n\tif (chunk.length > 0) chunks.push(chunk);\n\treturn chunks;\n}\n\nfunction keyStartsWith(key: Uint8Array, prefix: Uint8Array): boolean {\n\tif (key.length < prefix.length) return false;\n\tfor (let i = 0; i < prefix.length; i++) {\n\t\tif (key[i] !== prefix[i]) return false;\n\t}\n\treturn true;\n}\n\nclass ActorWorkflowMessageDriver implements WorkflowMessageDriver {\n\t#actor: AnyStaticActorInstance;\n\t#runCtx: RunContext<any, any, any, any, any, any, any, any>;\n\n\tconstructor(\n\t\tactor: AnyStaticActorInstance,\n\t\trunCtx: RunContext<any, any, any, any, any, any, any, any>,\n\t) {\n\t\tthis.#actor = actor;\n\t\tthis.#runCtx = runCtx;\n\t}\n\n\tasync addMessage(message: Message): Promise<void> {\n\t\tawait this.#runCtx.internalKeepAwake(\n\t\t\tthis.#actor.queueManager.enqueue(message.name, message.data),\n\t\t);\n\t}\n\n\tasync receiveMessages(opts: {\n\t\tnames?: readonly string[];\n\t\tcount: number;\n\t\tcompletable: boolean;\n\t}): Promise<Message[]> {\n\t\tconst messages = await this.#runCtx.internalKeepAwake(\n\t\t\tthis.#actor.queueManager.receive(\n\t\t\t\topts.names && opts.names.length > 0\n\t\t\t\t\t? [...opts.names]\n\t\t\t\t\t: undefined,\n\t\t\t\topts.count,\n\t\t\t\t0,\n\t\t\t\tundefined,\n\t\t\t\topts.completable,\n\t\t\t),\n\t\t);\n\t\treturn messages.map((message: ReceivedQueueMessage) => ({\n\t\t\tid: message.id.toString(),\n\t\t\tname: message.name,\n\t\t\tdata: message.body,\n\t\t\tsentAt: message.createdAt,\n\t\t\t...(opts.completable\n\t\t\t\t? {\n\t\t\t\t\t\tcomplete: async (response?: unknown) => {\n\t\t\t\t\t\t\tawait this.#runCtx.internalKeepAwake(\n\t\t\t\t\t\t\t\tthis.#actor.queueManager.completeMessage(\n\t\t\t\t\t\t\t\t\tmessage,\n\t\t\t\t\t\t\t\t\tresponse,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t},\n\t\t\t\t\t}\n\t\t\t\t: {}),\n\t\t}));\n\t}\n\n\tasync completeMessage(\n\t\tmessageId: string,\n\t\tresponse?: unknown,\n\t): Promise<void> {\n\t\tlet parsedId: bigint;\n\t\ttry {\n\t\t\tparsedId = BigInt(messageId);\n\t\t} catch {\n\t\t\treturn;\n\t\t}\n\n\t\tawait this.#runCtx.internalKeepAwake(\n\t\t\tthis.#actor.queueManager.completeMessageById(parsedId, response),\n\t\t);\n\t}\n}\n\nexport class ActorWorkflowDriver implements EngineDriver {\n\treadonly atomicBatch = true;\n\treadonly workerPollInterval = 100;\n\treadonly messageDriver: WorkflowMessageDriver;\n\treadonly telemetry: WorkflowTelemetryDriver;\n\t#actor: AnyStaticActorInstance;\n\t#runCtx: RunContext<any, any, any, any, any, any, any, any>;\n\t#storage: WorkflowStorage;\n\n\tconstructor(\n\t\tactor: AnyStaticActorInstance,\n\t\trunCtx: RunContext<any, any, any, any, any, any, any, any>,\n\t) {\n\t\tthis.#actor = actor;\n\t\tthis.#runCtx = runCtx;\n\t\tthis.messageDriver = new ActorWorkflowMessageDriver(actor, runCtx);\n\t\tthis.telemetry = {\n\t\t\tstartSpan: () => runCtx.run.startWorkflowSpan(),\n\t\t};\n\t\tthis.#storage = new WorkflowStorage(runtimeDbFromContext(runCtx));\n\t}\n\n\t/** Keeps the engine's history storage out of the workflow run's trace. */\n\t#untraced<T>(run: () => Promise<T>): Promise<T> {\n\t\treturn this.#runCtx.internalKeepAwake(\n\t\t\tthis.#runCtx.run.runOutsideWorkflowSpan(run),\n\t\t);\n\t}\n\n\tasync get(key: Uint8Array): Promise<Uint8Array | null> {\n\t\treturn await this.#untraced(() => this.#storage.get(key));\n\t}\n\n\tasync set(key: Uint8Array, value: Uint8Array): Promise<void> {\n\t\tawait this.#untraced(() => this.#storage.set(key, value));\n\t}\n\n\tasync delete(key: Uint8Array): Promise<void> {\n\t\tawait this.#untraced(() => this.#storage.delete(key));\n\t}\n\n\tasync batchDelete(keys: Uint8Array[]): Promise<void> {\n\t\tawait this.#untraced(() => this.#storage.batchDelete(keys));\n\t}\n\n\tasync deletePrefix(prefix: Uint8Array): Promise<void> {\n\t\tawait this.#untraced(() => this.#storage.deletePrefix(prefix));\n\t}\n\n\tasync deleteRange(start: Uint8Array, end: Uint8Array): Promise<void> {\n\t\tawait this.#untraced(() => this.#storage.deleteRange(start, end));\n\t}\n\n\tasync list(prefix: Uint8Array): Promise<KVEntry[]> {\n\t\treturn await this.#untraced(() => this.#storage.list(prefix));\n\t}\n\n\tasync batch(writes: KVWrite[]): Promise<void> {\n\t\tif (writes.length === 0) return;\n\n\t\tawait this.#untraced(() =>\n\t\t\tthis.#actor.stateManager.saveStateAndWorkflowBatch(writes),\n\t\t);\n\t}\n\n\tasync setAlarm(_workflowId: string, wakeAt: number): Promise<void> {\n\t\tawait this.#runCtx.internalKeepAwake(\n\t\t\tthis.#actor.driver.setAlarm(this.#actor, wakeAt),\n\t\t);\n\t}\n\n\tasync clearAlarm(_workflowId: string): Promise<void> {\n\t\t// No dedicated clear alarm support in actor drivers.\n\t\treturn;\n\t}\n\n\twaitForMessages(\n\t\tmessageNames: string[],\n\t\tabortSignal: AbortSignal,\n\t): Promise<void> {\n\t\treturn this.#actor.queueManager.waitForNames(\n\t\t\tmessageNames.length > 0 ? messageNames : undefined,\n\t\t\tabortSignal,\n\t\t);\n\t}\n}\n\nclass NoopWorkflowMessageDriver implements WorkflowMessageDriver {\n\tasync addMessage(_message: Message): Promise<void> {\n\t\tthrow new Error(\"Workflow control driver does not support messages\");\n\t}\n\n\tasync receiveMessages(_opts: {\n\t\tnames?: readonly string[];\n\t\tcount: number;\n\t\tcompletable: boolean;\n\t}): Promise<Message[]> {\n\t\tthrow new Error(\"Workflow control driver does not support messages\");\n\t}\n\n\tasync completeMessage(\n\t\t_messageId: string,\n\t\t_response?: unknown,\n\t): Promise<void> {\n\t\tthrow new Error(\"Workflow control driver does not support messages\");\n\t}\n}\n\nexport class ActorWorkflowControlDriver implements EngineDriver {\n\treadonly workerPollInterval = 100;\n\treadonly messageDriver: WorkflowMessageDriver =\n\t\tnew NoopWorkflowMessageDriver();\n\t#actor: AnyStaticActorInstance;\n\t#storage: WorkflowStorage;\n\n\tconstructor(\n\t\tactor: AnyStaticActorInstance,\n\t\trunCtx?: RunContext<any, any, any, any, any, any, any, any>,\n\t) {\n\t\tthis.#actor = actor;\n\t\tthis.#storage = new WorkflowStorage(runtimeDbFromContext(runCtx));\n\t}\n\n\tasync get(key: Uint8Array): Promise<Uint8Array | null> {\n\t\treturn await this.#storage.get(key);\n\t}\n\n\tasync set(key: Uint8Array, value: Uint8Array): Promise<void> {\n\t\tawait this.#storage.set(key, value);\n\t}\n\n\tasync delete(key: Uint8Array): Promise<void> {\n\t\tawait this.#storage.delete(key);\n\t}\n\n\tasync batchDelete(keys: Uint8Array[]): Promise<void> {\n\t\tawait this.#storage.batchDelete(keys);\n\t}\n\n\tasync deletePrefix(prefix: Uint8Array): Promise<void> {\n\t\tawait this.#storage.deletePrefix(prefix);\n\t}\n\n\tasync deleteRange(start: Uint8Array, end: Uint8Array): Promise<void> {\n\t\tawait this.#storage.deleteRange(start, end);\n\t}\n\n\tasync list(prefix: Uint8Array): Promise<KVEntry[]> {\n\t\treturn await this.#storage.list(prefix);\n\t}\n\n\tasync batch(writes: KVWrite[]): Promise<void> {\n\t\tif (writes.length === 0) {\n\t\t\treturn;\n\t\t}\n\n\t\tawait this.#storage.batch(writes);\n\t}\n\n\tasync setAlarm(_workflowId: string, wakeAt: number): Promise<void> {\n\t\tawait this.#actor.driver.setAlarm(this.#actor, wakeAt);\n\t}\n\n\tasync clearAlarm(_workflowId: string): Promise<void> {\n\t\treturn;\n\t}\n\n\twaitForMessages(\n\t\t_messageNames: string[],\n\t\t_abortSignal: AbortSignal,\n\t): Promise<void> {\n\t\tthrow new Error(\"Workflow control driver does not support messages\");\n\t}\n}\n"]}
|
|
@@ -1,9 +1,9 @@
|
|
|
1
1
|
import { WorkflowErrorEvent } from '@rivetkit/workflow-engine';
|
|
2
2
|
export { Loop, TryBlockCatchKind, TryBlockConfig, TryBlockFailure, TryBlockResult, TryStepCatchKind, TryStepConfig, TryStepFailure, TryStepResult, WorkflowError, WorkflowErrorEvent } from '@rivetkit/workflow-engine';
|
|
3
|
-
import { E as EventSchemaConfig, Q as QueueSchemaConfig, h as RunContext } from '../config-
|
|
3
|
+
import { E as EventSchemaConfig, Q as QueueSchemaConfig, h as RunContext } from '../config-CuKzfH58.cjs';
|
|
4
4
|
import { A as AnyDatabaseProvider } from '../config-ey2I2zz5.cjs';
|
|
5
|
-
import { d as WorkflowContext } from '../context-
|
|
6
|
-
export { e as WorkflowBranchConfig, W as WorkflowBranchContextOf, a as WorkflowContextOf, f as WorkflowLoopConfig, b as WorkflowLoopContextOf, g as WorkflowStepConfig, h as WorkflowStepContext, c as WorkflowStepContextOf, i as WorkflowTryConfig, j as WorkflowTryStepConfig } from '../context-
|
|
5
|
+
import { d as WorkflowContext } from '../context-DPCxyjax.cjs';
|
|
6
|
+
export { e as WorkflowBranchConfig, W as WorkflowBranchContextOf, a as WorkflowContextOf, f as WorkflowLoopConfig, b as WorkflowLoopContextOf, g as WorkflowStepConfig, h as WorkflowStepContext, c as WorkflowStepContextOf, i as WorkflowTryConfig, j as WorkflowTryStepConfig } from '../context-DPCxyjax.cjs';
|
|
7
7
|
import 'zod/v4';
|
|
8
8
|
import '@rivetkit/rivetkit-wasm';
|
|
9
9
|
import 'zod';
|
|
@@ -1,9 +1,9 @@
|
|
|
1
1
|
import { WorkflowErrorEvent } from '@rivetkit/workflow-engine';
|
|
2
2
|
export { Loop, TryBlockCatchKind, TryBlockConfig, TryBlockFailure, TryBlockResult, TryStepCatchKind, TryStepConfig, TryStepFailure, TryStepResult, WorkflowError, WorkflowErrorEvent } from '@rivetkit/workflow-engine';
|
|
3
|
-
import { E as EventSchemaConfig, Q as QueueSchemaConfig, h as RunContext } from '../config-
|
|
3
|
+
import { E as EventSchemaConfig, Q as QueueSchemaConfig, h as RunContext } from '../config-RV6gIa2L.js';
|
|
4
4
|
import { A as AnyDatabaseProvider } from '../config-ey2I2zz5.js';
|
|
5
|
-
import { d as WorkflowContext } from '../context-
|
|
6
|
-
export { e as WorkflowBranchConfig, W as WorkflowBranchContextOf, a as WorkflowContextOf, f as WorkflowLoopConfig, b as WorkflowLoopContextOf, g as WorkflowStepConfig, h as WorkflowStepContext, c as WorkflowStepContextOf, i as WorkflowTryConfig, j as WorkflowTryStepConfig } from '../context-
|
|
5
|
+
import { d as WorkflowContext } from '../context-DoMKZBQA.js';
|
|
6
|
+
export { e as WorkflowBranchConfig, W as WorkflowBranchContextOf, a as WorkflowContextOf, f as WorkflowLoopConfig, b as WorkflowLoopContextOf, g as WorkflowStepConfig, h as WorkflowStepContext, c as WorkflowStepContextOf, i as WorkflowTryConfig, j as WorkflowTryStepConfig } from '../context-DoMKZBQA.js';
|
|
7
7
|
import 'zod/v4';
|
|
8
8
|
import '@rivetkit/rivetkit-wasm';
|
|
9
9
|
import 'zod';
|