rivetkit 2.0.42 → 2.1.0-rc.1

package/src/workflow/inspector.ts

@@ -0,0 +1,268 @@
+import * as cbor from "cbor-x";
+import { createNanoEvents } from "nanoevents";
+import type {
+	BranchStatus,
+	BranchStatusType,
+	EntryKind,
+	EntryStatus,
+	Location,
+	SleepState,
+	WorkflowHistoryEntry,
+	WorkflowHistorySnapshot,
+	WorkflowEntryMetadataSnapshot,
+} from "@rivetkit/workflow-engine";
+import { encodeWorkflowHistoryTransport } from "@/inspector/transport";
+import type * as inspectorSchema from "@/schemas/actor-inspector/mod";
+import * as transport from "@/schemas/transport/mod";
+import { assertUnreachable, bufferToArrayBuffer } from "@/utils";
+
+export interface WorkflowInspectorAdapter {
+	getHistory: () => inspectorSchema.WorkflowHistory | null;
+	onHistoryUpdated: (
+		listener: (history: inspectorSchema.WorkflowHistory) => void,
+	) => () => void;
+}
+
+export function createWorkflowInspectorAdapter(): {
+	adapter: WorkflowInspectorAdapter;
+	update: (snapshot: WorkflowHistorySnapshot) => void;
+} {
+	const emitter = createNanoEvents<{
+		updated: (history: inspectorSchema.WorkflowHistory) => void;
+	}>();
+	let history: inspectorSchema.WorkflowHistory | null = null;
+
+	const adapter: WorkflowInspectorAdapter = {
+		getHistory: () => history,
+		onHistoryUpdated: (listener) => emitter.on("updated", listener),
+	};
+
+	const update = (snapshot: WorkflowHistorySnapshot) => {
+		const transportHistory = toWorkflowHistory(snapshot);
+		const next = encodeWorkflowHistoryTransport(transportHistory);
+		history = next;
+		emitter.emit("updated", next);
+	};
+
+	return { adapter, update };
+}
+
+function encodeCbor(value: unknown): ArrayBuffer {
+	return bufferToArrayBuffer(cbor.encode(value));
+}
+
+function encodeOptionalCbor(value: unknown): ArrayBuffer | null {
+	if (value === undefined) {
+		return null;
+	}
+	return encodeCbor(value);
+}
+
+function toU64(value: number): bigint {
+	return BigInt(Math.max(0, Math.floor(value)));
+}
+
+function toWorkflowLocation(
+	location: Location,
+): transport.WorkflowLocation {
+	return location.map((segment) => {
+		if (typeof segment === "number") {
+			return { tag: "WorkflowNameIndex", val: segment };
+		}
+		return {
+			tag: "WorkflowLoopIterationMarker",
+			val: {
+				loop: segment.loop,
+				iteration: segment.iteration,
+			},
+		};
+	});
+}
+
+function toWorkflowEntryKind(
+	kind: EntryKind,
+): transport.WorkflowEntryKind {
+	switch (kind.type) {
+		case "step":
+			return {
+				tag: "WorkflowStepEntry",
+				val: {
+					output: encodeOptionalCbor(kind.data.output),
+					error: kind.data.error ?? null,
+				},
+			};
+		case "loop":
+			return {
+				tag: "WorkflowLoopEntry",
+				val: {
+					state: encodeCbor(kind.data.state),
+					iteration: kind.data.iteration,
+					output: encodeOptionalCbor(kind.data.output),
+				},
+			};
+		case "sleep":
+			return {
+				tag: "WorkflowSleepEntry",
+				val: {
+					deadline: toU64(kind.data.deadline),
+					state: toWorkflowSleepState(kind.data.state),
+				},
+			};
+		case "message":
+			return {
+				tag: "WorkflowMessageEntry",
+				val: {
+					name: kind.data.name,
+					messageData: encodeCbor(kind.data.data),
+				},
+			};
+		case "rollback_checkpoint":
+			return {
+				tag: "WorkflowRollbackCheckpointEntry",
+				val: { name: kind.data.name },
+			};
+		case "join":
+			return {
+				tag: "WorkflowJoinEntry",
+				val: { branches: toWorkflowBranchStatusMap(kind.data.branches) },
+			};
+		case "race":
+			return {
+				tag: "WorkflowRaceEntry",
+				val: {
+					winner: kind.data.winner ?? null,
+					branches: toWorkflowBranchStatusMap(kind.data.branches),
+				},
+			};
+		case "removed":
+			return {
+				tag: "WorkflowRemovedEntry",
+				val: {
+					originalType: kind.data.originalType,
+					originalName: kind.data.originalName ?? null,
+				},
+			};
+		default:
+			assertUnreachable(kind as never);
+	}
+}
+
+function toWorkflowEntry(
+	entry: WorkflowHistoryEntry,
+): transport.WorkflowEntry {
+	return {
+		id: entry.id,
+		location: toWorkflowLocation(entry.location),
+		kind: toWorkflowEntryKind(entry.kind),
+	};
+}
+
+function toWorkflowEntryStatus(
+	status: EntryStatus,
+): transport.WorkflowEntryStatus {
+	switch (status) {
+		case "pending":
+			return transport.WorkflowEntryStatus.PENDING;
+		case "running":
+			return transport.WorkflowEntryStatus.RUNNING;
+		case "completed":
+			return transport.WorkflowEntryStatus.COMPLETED;
+		case "failed":
+			return transport.WorkflowEntryStatus.FAILED;
+		case "exhausted":
+			return transport.WorkflowEntryStatus.EXHAUSTED;
+		default:
+			assertUnreachable(status as never);
+	}
+}
+
+function toWorkflowSleepState(
+	state: SleepState,
+): transport.WorkflowSleepState {
+	switch (state) {
+		case "pending":
+			return transport.WorkflowSleepState.PENDING;
+		case "completed":
+			return transport.WorkflowSleepState.COMPLETED;
+		case "interrupted":
+			return transport.WorkflowSleepState.INTERRUPTED;
+		default:
+			assertUnreachable(state as never);
+	}
+}
+
+function toWorkflowBranchStatusType(
+	status: BranchStatusType,
+): transport.WorkflowBranchStatusType {
+	switch (status) {
+		case "pending":
+			return transport.WorkflowBranchStatusType.PENDING;
+		case "running":
+			return transport.WorkflowBranchStatusType.RUNNING;
+		case "completed":
+			return transport.WorkflowBranchStatusType.COMPLETED;
+		case "failed":
+			return transport.WorkflowBranchStatusType.FAILED;
+		case "cancelled":
+			return transport.WorkflowBranchStatusType.CANCELLED;
+		default:
+			assertUnreachable(status as never);
+	}
+}
+
+function toWorkflowBranchStatus(
+	status: BranchStatus,
+): transport.WorkflowBranchStatus {
+	return {
+		status: toWorkflowBranchStatusType(status.status),
+		output: encodeOptionalCbor(status.output),
+		error: status.error ?? null,
+	};
+}
+
+function toWorkflowBranchStatusMap(
+	branches: Record<string, BranchStatus>,
+): ReadonlyMap<string, transport.WorkflowBranchStatus> {
+	return new Map(
+		Object.entries(branches).map(([name, status]) => [
+			name,
+			toWorkflowBranchStatus(status),
+		]),
+	);
+}
+
+function toWorkflowEntryMetadata(
+	metadata: WorkflowEntryMetadataSnapshot,
+): transport.WorkflowEntryMetadata {
+	return {
+		status: toWorkflowEntryStatus(metadata.status),
+		error: metadata.error ?? null,
+		attempts: metadata.attempts,
+		lastAttemptAt: toU64(metadata.lastAttemptAt),
+		createdAt: toU64(metadata.createdAt),
+		completedAt:
+			metadata.completedAt === undefined
+				? null
+				: toU64(metadata.completedAt),
+		rollbackCompletedAt:
+			metadata.rollbackCompletedAt === undefined
+				? null
+				: toU64(metadata.rollbackCompletedAt),
+		rollbackError: metadata.rollbackError ?? null,
+	};
+}
+
+function toWorkflowHistory(
+	snapshot: WorkflowHistorySnapshot,
+): transport.WorkflowHistory {
+	const entryMetadata = new Map<string, transport.WorkflowEntryMetadata>();
+	for (const [id, metadata] of snapshot.entryMetadata) {
+		entryMetadata.set(id, toWorkflowEntryMetadata(metadata));
+	}
+
+	return {
+		nameRegistry: snapshot.nameRegistry,
+		entries: snapshot.entries.map((entry) => toWorkflowEntry(entry)),
+		entryMetadata,
+	};
+}

package/src/workflow/mod.ts

@@ -0,0 +1,122 @@
+import { ACTOR_CONTEXT_INTERNAL_SYMBOL } from "@/actor/contexts/base/actor";
+import type { RunContext } from "@/actor/contexts/run";
+import type { AnyDatabaseProvider } from "@/actor/database";
+import type { AnyActorInstance } from "@/actor/instance/mod";
+import type { EventSchemaConfig, QueueSchemaConfig } from "@/actor/schema";
+import { RUN_FUNCTION_CONFIG_SYMBOL } from "@/actor/config";
+import { stringifyError } from "@/utils";
+import { runWorkflow } from "@rivetkit/workflow-engine";
+import invariant from "invariant";
+import { ActorWorkflowContext } from "./context";
+import { ActorWorkflowDriver } from "./driver";
+import { createWorkflowInspectorAdapter } from "./inspector";
+
+export { Loop } from "@rivetkit/workflow-engine";
+export { ActorWorkflowContext } from "./context";
+
+export function workflow<
+	TState,
+	TConnParams,
+	TConnState,
+	TVars,
+	TInput,
+	TDatabase extends AnyDatabaseProvider,
+	TEvents extends EventSchemaConfig = Record<never, never>,
+	TQueues extends QueueSchemaConfig = Record<never, never>,
+>(
+	fn: (
+		ctx: ActorWorkflowContext<
+			TState,
+			TConnParams,
+			TConnState,
+			TVars,
+			TInput,
+			TDatabase,
+			TEvents,
+			TQueues
+		>,
+	) => Promise<unknown>,
+): (
+	c: RunContext<
+		TState,
+		TConnParams,
+		TConnState,
+		TVars,
+		TInput,
+		TDatabase,
+		TEvents,
+		TQueues
+	>,
+) => Promise<void> {
+	const workflowInspector = createWorkflowInspectorAdapter();
+
+	async function run(
+		runCtx: RunContext<
+			TState,
+			TConnParams,
+			TConnState,
+			TVars,
+			TInput,
+			TDatabase,
+			TEvents,
+			TQueues
+		>,
+	): Promise<void> {
+		const actor = (
+			runCtx as unknown as {
+				[ACTOR_CONTEXT_INTERNAL_SYMBOL]?: AnyActorInstance;
+			}
+		)[ACTOR_CONTEXT_INTERNAL_SYMBOL];
+		invariant(actor, "workflow() requires an actor instance");
+
+		const driver = new ActorWorkflowDriver(actor, runCtx);
+
+		const handle = runWorkflow(
+			actor.id,
+			async (ctx) => await fn(new ActorWorkflowContext(ctx, runCtx)),
+			undefined,
+			driver,
+			{
+				mode: "live",
+				logger: runCtx.log,
+				onHistoryUpdated: workflowInspector.update,
+			},
+		);
+
+		const onAbort = () => {
+			handle.evict();
+		};
+		if (runCtx.abortSignal.aborted) {
+			onAbort();
+		} else {
+			runCtx.abortSignal.addEventListener("abort", onAbort, {
+				once: true,
+			});
+		}
+
+		try {
+			await handle.result;
+		} catch (error) {
+			runCtx.log.error({
+				msg: "workflow run failed",
+				error: stringifyError(error),
+			});
+			throw error;
+		} finally {
+			runCtx.abortSignal.removeEventListener("abort", onAbort);
+		}
+	}
+
+	const runWithConfig = run as typeof run & {
+		[RUN_FUNCTION_CONFIG_SYMBOL]?: {
+			icon?: string;
+			inspector?: { workflow: typeof workflowInspector.adapter };
+		};
+	};
+	runWithConfig[RUN_FUNCTION_CONFIG_SYMBOL] = {
+		icon: "diagram-project",
+		inspector: { workflow: workflowInspector.adapter },
+	};
+
+	return runWithConfig;
+}

package/dist/tsup/chunk-2IJTYN6K.cjs

@@ -1,278 +0,0 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true});
-
-var _chunkX35U3YNXcjs = require('./chunk-X35U3YNX.cjs');
-
-
-
-var _chunkH4TB4X25cjs = require('./chunk-H4TB4X25.cjs');
-
-// src/driver-test-suite/log.ts
-function logger() {
-  return _chunkH4TB4X25cjs.getLogger.call(void 0, "test-suite");
-}
-
-// fixtures/driver-test-suite/hibernation.ts
-var HIBERNATION_SLEEP_TIMEOUT = 500;
-var hibernationActor = _chunkX35U3YNXcjs.actor.call(void 0, {
-  state: {
-    sleepCount: 0,
-    wakeCount: 0
-  },
-  createConnState: (c) => {
-    return {
-      count: 0,
-      connectCount: 0,
-      disconnectCount: 0
-    };
-  },
-  onWake: (c) => {
-    c.state.wakeCount += 1;
-  },
-  onSleep: (c) => {
-    c.state.sleepCount += 1;
-  },
-  onConnect: (c, conn) => {
-    conn.state.connectCount += 1;
-  },
-  onDisconnect: (c, conn) => {
-    conn.state.disconnectCount += 1;
-  },
-  actions: {
-    // Basic RPC that returns a simple value
-    ping: (c) => {
-      return "pong";
-    },
-    // Increment the connection's count
-    connIncrement: (c) => {
-      c.conn.state.count += 1;
-      return c.conn.state.count;
-    },
-    // Get the connection's count
-    getConnCount: (c) => {
-      return c.conn.state.count;
-    },
-    // Get the connection's lifecycle counts
-    getConnLifecycleCounts: (c) => {
-      return {
-        connectCount: c.conn.state.connectCount,
-        disconnectCount: c.conn.state.disconnectCount
-      };
-    },
-    // Get all connection IDs
-    getConnectionIds: (c) => {
-      return c.conns.entries().map((x) => x[0]).toArray();
-    },
-    // Get actor sleep/wake counts
-    getActorCounts: (c) => {
-      return {
-        sleepCount: c.state.sleepCount,
-        wakeCount: c.state.wakeCount
-      };
-    },
-    // Trigger sleep
-    triggerSleep: (c) => {
-      c.sleep();
-    }
-  },
-  options: {
-    sleepTimeout: HIBERNATION_SLEEP_TIMEOUT
-  }
-});
-
-// fixtures/driver-test-suite/sleep.ts
-var SLEEP_TIMEOUT = 1e3;
-var sleep = _chunkX35U3YNXcjs.actor.call(void 0, {
-  state: { startCount: 0, sleepCount: 0 },
-  onWake: (c) => {
-    c.state.startCount += 1;
-  },
-  onSleep: (c) => {
-    c.state.sleepCount += 1;
-  },
-  actions: {
-    triggerSleep: (c) => {
-      c.sleep();
-    },
-    getCounts: (c) => {
-      return {
-        startCount: c.state.startCount,
-        sleepCount: c.state.sleepCount
-      };
-    },
-    setAlarm: async (c, duration) => {
-      await c.schedule.after(duration, "onAlarm");
-    },
-    onAlarm: (c) => {
-      c.log.info("alarm called");
-    }
-  },
-  options: {
-    sleepTimeout: SLEEP_TIMEOUT
-  }
-});
-var sleepWithLongRpc = _chunkX35U3YNXcjs.actor.call(void 0, {
-  state: { startCount: 0, sleepCount: 0 },
-  createVars: () => ({}),
-  onWake: (c) => {
-    c.state.startCount += 1;
-  },
-  onSleep: (c) => {
-    c.state.sleepCount += 1;
-  },
-  actions: {
-    getCounts: (c) => {
-      return {
-        startCount: c.state.startCount,
-        sleepCount: c.state.sleepCount
-      };
-    },
-    longRunningRpc: async (c) => {
-      c.log.info("starting long running rpc");
-      c.vars.longRunningResolve = _chunkH4TB4X25cjs.promiseWithResolvers.call(void 0, );
-      c.broadcast("waiting");
-      await c.vars.longRunningResolve.promise;
-      c.log.info("finished long running rpc");
-    },
-    finishLongRunningRpc: (c) => {
-      var _a;
-      return (_a = c.vars.longRunningResolve) == null ? void 0 : _a.resolve();
-    }
-  },
-  options: {
-    sleepTimeout: SLEEP_TIMEOUT
-  }
-});
-var sleepWithRawHttp = _chunkX35U3YNXcjs.actor.call(void 0, {
-  state: { startCount: 0, sleepCount: 0, requestCount: 0 },
-  onWake: (c) => {
-    c.state.startCount += 1;
-  },
-  onSleep: (c) => {
-    c.state.sleepCount += 1;
-  },
-  onRequest: async (c, request) => {
-    c.state.requestCount += 1;
-    const url = new URL(request.url);
-    if (url.pathname === "/long-request") {
-      const duration = parseInt(
-        url.searchParams.get("duration") || "1000"
-      );
-      c.log.info({ msg: "starting long fetch request", duration });
-      await new Promise((resolve) => setTimeout(resolve, duration));
-      c.log.info("finished long fetch request");
-      return new Response(JSON.stringify({ completed: true }), {
-        headers: { "Content-Type": "application/json" }
-      });
-    }
-    return new Response("Not Found", { status: 404 });
-  },
-  actions: {
-    getCounts: (c) => {
-      return {
-        startCount: c.state.startCount,
-        sleepCount: c.state.sleepCount,
-        requestCount: c.state.requestCount
-      };
-    }
-  },
-  options: {
-    sleepTimeout: SLEEP_TIMEOUT
-  }
-});
-var sleepWithRawWebSocket = _chunkX35U3YNXcjs.actor.call(void 0, {
-  state: { startCount: 0, sleepCount: 0, connectionCount: 0 },
-  onWake: (c) => {
-    c.state.startCount += 1;
-  },
-  onSleep: (c) => {
-    c.state.sleepCount += 1;
-  },
-  onWebSocket: (c, websocket) => {
-    c.state.connectionCount += 1;
-    c.log.info({
-      msg: "websocket connected",
-      connectionCount: c.state.connectionCount
-    });
-    websocket.send(
-      JSON.stringify({
-        type: "connected",
-        connectionCount: c.state.connectionCount
-      })
-    );
-    websocket.addEventListener("message", (event) => {
-      const data = event.data;
-      if (typeof data === "string") {
-        try {
-          const parsed = JSON.parse(data);
-          if (parsed.type === "getCounts") {
-            websocket.send(
-              JSON.stringify({
-                type: "counts",
-                startCount: c.state.startCount,
-                sleepCount: c.state.sleepCount,
-                connectionCount: c.state.connectionCount
-              })
-            );
-          } else if (parsed.type === "keepAlive") {
-            websocket.send(JSON.stringify({ type: "ack" }));
-          }
-        } catch (e) {
-          websocket.send(data);
-        }
-      }
-    });
-    websocket.addEventListener("close", () => {
-      c.state.connectionCount -= 1;
-      c.log.info({
-        msg: "websocket disconnected",
-        connectionCount: c.state.connectionCount
-      });
-    });
-  },
-  actions: {
-    getCounts: (c) => {
-      return {
-        startCount: c.state.startCount,
-        sleepCount: c.state.sleepCount,
-        connectionCount: c.state.connectionCount
-      };
-    }
-  },
-  options: {
-    sleepTimeout: SLEEP_TIMEOUT
-  }
-});
-var sleepWithNoSleepOption = _chunkX35U3YNXcjs.actor.call(void 0, {
-  state: { startCount: 0, sleepCount: 0 },
-  onWake: (c) => {
-    c.state.startCount += 1;
-  },
-  onSleep: (c) => {
-    c.state.sleepCount += 1;
-  },
-  actions: {
-    getCounts: (c) => {
-      return {
-        startCount: c.state.startCount,
-        sleepCount: c.state.sleepCount
-      };
-    }
-  },
-  options: {
-    sleepTimeout: SLEEP_TIMEOUT,
-    noSleep: true
-  }
-});
-
-
-
-
-
-
-
-
-
-
-
-exports.logger = logger; exports.HIBERNATION_SLEEP_TIMEOUT = HIBERNATION_SLEEP_TIMEOUT; exports.hibernationActor = hibernationActor; exports.SLEEP_TIMEOUT = SLEEP_TIMEOUT; exports.sleep = sleep; exports.sleepWithLongRpc = sleepWithLongRpc; exports.sleepWithRawHttp = sleepWithRawHttp; exports.sleepWithRawWebSocket = sleepWithRawWebSocket; exports.sleepWithNoSleepOption = sleepWithNoSleepOption;
-//# sourceMappingURL=chunk-2IJTYN6K.cjs.map

package/dist/tsup/chunk-2IJTYN6K.cjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/chunk-2IJTYN6K.cjs","../../src/driver-test-suite/log.ts","../../fixtures/driver-test-suite/hibernation.ts","../../fixtures/driver-test-suite/sleep.ts"],"names":[],"mappings":"AAAA;AACE;AACF,wDAA6B;AAC7B;AACE;AACA;AACF,wDAA6B;AAC7B;AACA;ACNO,SAAS,MAAA,CAAA,EAAS;AACxB,EAAA,OAAO,yCAAA,YAAsB,CAAA;AAC9B;ADQA;AACA;AEXO,IAAM,0BAAA,EAA4B,GAAA;AAQlC,IAAM,iBAAA,EAAmB,qCAAA;AAAM,EACrC,KAAA,EAAO;AAAA,IACN,UAAA,EAAY,CAAA;AAAA,IACZ,SAAA,EAAW;AAAA,EACZ,CAAA;AAAA,EACA,eAAA,EAAiB,CAAC,CAAA,EAAA,GAA4B;AAC7C,IAAA,OAAO;AAAA,MACN,KAAA,EAAO,CAAA;AAAA,MACP,YAAA,EAAc,CAAA;AAAA,MACd,eAAA,EAAiB;AAAA,IAClB,CAAA;AAAA,EACD,CAAA;AAAA,EACA,MAAA,EAAQ,CAAC,CAAA,EAAA,GAAM;AACd,IAAA,CAAA,CAAE,KAAA,CAAM,UAAA,GAAa,CAAA;AAAA,EACtB,CAAA;AAAA,EACA,OAAA,EAAS,CAAC,CAAA,EAAA,GAAM;AACf,IAAA,CAAA,CAAE,KAAA,CAAM,WAAA,GAAc,CAAA;AAAA,EACvB,CAAA;AAAA,EACA,SAAA,EAAW,CAAC,CAAA,EAAG,IAAA,EAAA,GAAS;AACvB,IAAA,IAAA,CAAK,KAAA,CAAM,aAAA,GAAgB,CAAA;AAAA,EAC5B,CAAA;AAAA,EACA,YAAA,EAAc,CAAC,CAAA,EAAG,IAAA,EAAA,GAAS;AAC1B,IAAA,IAAA,CAAK,KAAA,CAAM,gBAAA,GAAmB,CAAA;AAAA,EAC/B,CAAA;AAAA,EACA,OAAA,EAAS;AAAA;AAAA,IAER,IAAA,EAAM,CAAC,CAAA,EAAA,GAAM;AACZ,MAAA,OAAO,MAAA;AAAA,IACR,CAAA;AAAA;AAAA,IAEA,aAAA,EAAe,CAAC,CAAA,EAAA,GAAM;AACrB,MAAA,CAAA,CAAE,IAAA,CAAK,KAAA,CAAM,MAAA,GAAS,CAAA;AACtB,MAAA,OAAO,CAAA,CAAE,IAAA,CAAK,KAAA,CAAM,KAAA;AAAA,IACrB,CAAA;AAAA;AAAA,IAEA,YAAA,EAAc,CAAC,CAAA,EAAA,GAAM;AACpB,MAAA,OAAO,CAAA,CAAE,IAAA,CAAK,KAAA,CAAM,KAAA;AAAA,IACrB,CAAA;AAAA;AAAA,IAEA,sBAAA,EAAwB,CAAC,CAAA,EAAA,GAAM;AAC9B,MAAA,OAAO;AAAA,QACN,YAAA,EAAc,CAAA,CAAE,IAAA,CAAK,KAAA,CAAM,YAAA;AAAA,QAC3B,eAAA,EAAiB,CAAA,CAAE,IAAA,CAAK,KAAA,CAAM;AAAA,MAC/B,CAAA;AAAA,IACD,CAAA;AAAA;AAAA,IAEA,gBAAA,EAAkB,CAAC,CAAA,EAAA,GAAM;AACxB,MAAA,OAAO,CAAA,CAAE,KAAA,CACP,OAAA,CAAQ,CAAA,CACR,GAAA,CAAI,CAAC,CAAA,EAAA,GAAM,CAAA,CAAE,CAAC,CAAC,CAAA,CACf,OAAA,CAAQ,CAAA;AAAA,IACX,CAAA;AAAA;AAAA,IAEA,cAAA,EAAgB,CAAC,CAAA,EAAA,GAAM;AACtB,MAAA,OAAO;AAAA,QACN,UAAA,EAAY,CAAA,CAAE,KAAA,CAAM,UAAA;AAAA,QACpB,SAAA,EAAW,CAAA,CAAE,KAAA,CAAM;AAAA,MACpB,CAAA;AAAA,IACD,CAAA;AAAA;AAAA,IAEA,YAAA,EAAc,CAAC,CAAA,EAAA,GAAM;AACpB,MAAA,CAAA,CAAE,KAAA,CAAM,CAAA;AAAA,IACT;AAAA,EACD,CAAA;AAAA,EACA,OAAA,EAAS;AAAA,IACR,YAAA,EAAc;AAAA,EACf;AACD,CAAC,CAAA;AFGD;AACA;AG9EO,IAAM,cAAA,EAAgB,GAAA;AAEtB,IAAM,MAAA,EAAQ,qCAAA;AAAM,EAC1B,KAAA,EAAO,EAAE,UAAA,EAAY,CAAA,EAAG,UAAA,EAAY,EAAE,CAAA;AAAA,EACtC,MAAA,EAAQ,CAAC,CAAA,EAAA,GAAM;AACd,IAAA,CAAA,CAAE,KAAA,CAAM,WAAA,GAAc,CAAA;AAAA,EACvB,CAAA;AAAA,EACA,OAAA,EAAS,CAAC,CAAA,EAAA,GAAM;AACf,IAAA,CAAA,CAAE,KAAA,CAAM,WAAA,GAAc,CAAA;AAAA,EACvB,CAAA;AAAA,EACA,OAAA,EAAS;AAAA,IACR,YAAA,EAAc,CAAC,CAAA,EAAA,GAAM;AACpB,MAAA,CAAA,CAAE,KAAA,CAAM,CAAA;AAAA,IACT,CAAA;AAAA,IACA,SAAA,EAAW,CAAC,CAAA,EAAA,GAAM;AACjB,MAAA,OAAO;AAAA,QACN,UAAA,EAAY,CAAA,CAAE,KAAA,CAAM,UAAA;AAAA,QACpB,UAAA,EAAY,CAAA,CAAE,KAAA,CAAM;AAAA,MACrB,CAAA;AAAA,IACD,CAAA;AAAA,IACA,QAAA,EAAU,MAAA,CAAO,CAAA,EAAG,QAAA,EAAA,GAAqB;AACxC,MAAA,MAAM,CAAA,CAAE,QAAA,CAAS,KAAA,CAAM,QAAA,EAAU,SAAS,CAAA;AAAA,IAC3C,CAAA;AAAA,IACA,OAAA,EAAS,CAAC,CAAA,EAAA,GAAM;AACf,MAAA,CAAA,CAAE,GAAA,CAAI,IAAA,CAAK,cAAc,CAAA;AAAA,IAC1B;AAAA,EACD,CAAA;AAAA,EACA,OAAA,EAAS;AAAA,IACR,YAAA,EAAc;AAAA,EACf;AACD,CAAC,CAAA;AAEM,IAAM,iBAAA,EAAmB,qCAAA;AAAM,EACrC,KAAA,EAAO,EAAE,UAAA,EAAY,CAAA,EAAG,UAAA,EAAY,EAAE,CAAA;AAAA,EACtC,UAAA,EAAY,CAAA,EAAA,GAAA,CACV,CAAC,CAAA,CAAA;AAAA,EACH,MAAA,EAAQ,CAAC,CAAA,EAAA,GAAM;AACd,IAAA,CAAA,CAAE,KAAA,CAAM,WAAA,GAAc,CAAA;AAAA,EACvB,CAAA;AAAA,EACA,OAAA,EAAS,CAAC,CAAA,EAAA,GAAM;AACf,IAAA,CAAA,CAAE,KAAA,CAAM,WAAA,GAAc,CAAA;AAAA,EACvB,CAAA;AAAA,EACA,OAAA,EAAS;AAAA,IACR,SAAA,EAAW,CAAC,CAAA,EAAA,GAAM;AACjB,MAAA,OAAO;AAAA,QACN,UAAA,EAAY,CAAA,CAAE,KAAA,CAAM,UAAA;AAAA,QACpB,UAAA,EAAY,CAAA,CAAE,KAAA,CAAM;AAAA,MACrB,CAAA;AAAA,IACD,CAAA;AAAA,IACA,cAAA,EAAgB,MAAA,CAAO,CAAA,EAAA,GAAM;AAC5B,MAAA,CAAA,CAAE,GAAA,CAAI,IAAA,CAAK,2BAA2B,CAAA;AACtC,MAAA,CAAA,CAAE,IAAA,CAAK,mBAAA,EAAqB,oDAAA,CAAqB;AACjD,MAAA,CAAA,CAAE,SAAA,CAAU,SAAS,CAAA;AACrB,MAAA,MAAM,CAAA,CAAE,IAAA,CAAK,kBAAA,CAAmB,OAAA;AAChC,MAAA,CAAA,CAAE,GAAA,CAAI,IAAA,CAAK,2BAA2B,CAAA;AAAA,IACvC,CAAA;AAAA,IACA,oBAAA,EAAsB,CAAC,CAAA,EAAA,GAAG;AA3D5B,MAAA,IAAA,EAAA;AA2D+B,MAAA,OAAA,CAAA,GAAA,EAAA,CAAA,CAAE,IAAA,CAAK,kBAAA,EAAA,GAAP,KAAA,EAAA,KAAA,EAAA,EAAA,EAAA,CAA2B,OAAA,CAAA,CAAA;AAAA,IAAA;AAAA,EACzD,CAAA;AAAA,EACA,OAAA,EAAS;AAAA,IACR,YAAA,EAAc;AAAA,EACf;AACD,CAAC,CAAA;AAEM,IAAM,iBAAA,EAAmB,qCAAA;AAAM,EACrC,KAAA,EAAO,EAAE,UAAA,EAAY,CAAA,EAAG,UAAA,EAAY,CAAA,EAAG,YAAA,EAAc,EAAE,CAAA;AAAA,EACvD,MAAA,EAAQ,CAAC,CAAA,EAAA,GAAM;AACd,IAAA,CAAA,CAAE,KAAA,CAAM,WAAA,GAAc,CAAA;AAAA,EACvB,CAAA;AAAA,EACA,OAAA,EAAS,CAAC,CAAA,EAAA,GAAM;AACf,IAAA,CAAA,CAAE,KAAA,CAAM,WAAA,GAAc,CAAA;AAAA,EACvB,CAAA;AAAA,EACA,SAAA,EAAW,MAAA,CAAO,CAAA,EAAG,OAAA,EAAA,GAAY;AAChC,IAAA,CAAA,CAAE,KAAA,CAAM,aAAA,GAAgB,CAAA;AACxB,IAAA,MAAM,IAAA,EAAM,IAAI,GAAA,CAAI,OAAA,CAAQ,GAAG,CAAA;AAE/B,IAAA,GAAA,CAAI,GAAA,CAAI,SAAA,IAAa,eAAA,EAAiB;AACrC,MAAA,MAAM,SAAA,EAAW,QAAA;AAAA,QAChB,GAAA,CAAI,YAAA,CAAa,GAAA,CAAI,UAAU,EAAA,GAAK;AAAA,MACrC,CAAA;AACA,MAAA,CAAA,CAAE,GAAA,CAAI,IAAA,CAAK,EAAE,GAAA,EAAK,6BAAA,EAA+B,SAAS,CAAC,CAAA;AAC3D,MAAA,MAAM,IAAI,OAAA,CAAQ,CAAC,OAAA,EAAA,GAAY,UAAA,CAAW,OAAA,EAAS,QAAQ,CAAC,CAAA;AAC5D,MAAA,CAAA,CAAE,GAAA,CAAI,IAAA,CAAK,6BAA6B,CAAA;AACxC,MAAA,OAAO,IAAI,QAAA,CAAS,IAAA,CAAK,SAAA,CAAU,EAAE,SAAA,EAAW,KAAK,CAAC,CAAA,EAAG;AAAA,QACxD,OAAA,EAAS,EAAE,cAAA,EAAgB,mBAAmB;AAAA,MAC/C,CAAC,CAAA;AAAA,IACF;AAEA,IAAA,OAAO,IAAI,QAAA,CAAS,WAAA,EAAa,EAAE,MAAA,EAAQ,IAAI,CAAC,CAAA;AAAA,EACjD,CAAA;AAAA,EACA,OAAA,EAAS;AAAA,IACR,SAAA,EAAW,CAAC,CAAA,EAAA,GAAM;AACjB,MAAA,OAAO;AAAA,QACN,UAAA,EAAY,CAAA,CAAE,KAAA,CAAM,UAAA;AAAA,QACpB,UAAA,EAAY,CAAA,CAAE,KAAA,CAAM,UAAA;AAAA,QACpB,YAAA,EAAc,CAAA,CAAE,KAAA,CAAM;AAAA,MACvB,CAAA;AAAA,IACD;AAAA,EACD,CAAA;AAAA,EACA,OAAA,EAAS;AAAA,IACR,YAAA,EAAc;AAAA,EACf;AACD,CAAC,CAAA;AAEM,IAAM,sBAAA,EAAwB,qCAAA;AAAM,EAC1C,KAAA,EAAO,EAAE,UAAA,EAAY,CAAA,EAAG,UAAA,EAAY,CAAA,EAAG,eAAA,EAAiB,EAAE,CAAA;AAAA,EAC1D,MAAA,EAAQ,CAAC,CAAA,EAAA,GAAM;AACd,IAAA,CAAA,CAAE,KAAA,CAAM,WAAA,GAAc,CAAA;AAAA,EACvB,CAAA;AAAA,EACA,OAAA,EAAS,CAAC,CAAA,EAAA,GAAM;AACf,IAAA,CAAA,CAAE,KAAA,CAAM,WAAA,GAAc,CAAA;AAAA,EACvB,CAAA;AAAA,EACA,WAAA,EAAa,CAAC,CAAA,EAAG,SAAA,EAAA,GAAkC;AAClD,IAAA,CAAA,CAAE,KAAA,CAAM,gBAAA,GAAmB,CAAA;AAC3B,IAAA,CAAA,CAAE,GAAA,CAAI,IAAA,CAAK;AAAA,MACV,GAAA,EAAK,qBAAA;AAAA,MACL,eAAA,EAAiB,CAAA,CAAE,KAAA,CAAM;AAAA,IAC1B,CAAC,CAAA;AAED,IAAA,SAAA,CAAU,IAAA;AAAA,MACT,IAAA,CAAK,SAAA,CAAU;AAAA,QACd,IAAA,EAAM,WAAA;AAAA,QACN,eAAA,EAAiB,CAAA,CAAE,KAAA,CAAM;AAAA,MAC1B,CAAC;AAAA,IACF,CAAA;AAEA,IAAA,SAAA,CAAU,gBAAA,CAAiB,SAAA,EAAW,CAAC,KAAA,EAAA,GAAe;AACrD,MAAA,MAAM,KAAA,EAAO,KAAA,CAAM,IAAA;AACnB,MAAA,GAAA,CAAI,OAAO,KAAA,IAAS,QAAA,EAAU;AAC7B,QAAA,IAAI;AACH,UAAA,MAAM,OAAA,EAAS,IAAA,CAAK,KAAA,CAAM,IAAI,CAAA;AAC9B,UAAA,GAAA,CAAI,MAAA,CAAO,KAAA,IAAS,WAAA,EAAa;AAChC,YAAA,SAAA,CAAU,IAAA;AAAA,cACT,IAAA,CAAK,SAAA,CAAU;AAAA,gBACd,IAAA,EAAM,QAAA;AAAA,gBACN,UAAA,EAAY,CAAA,CAAE,KAAA,CAAM,UAAA;AAAA,gBACpB,UAAA,EAAY,CAAA,CAAE,KAAA,CAAM,UAAA;AAAA,gBACpB,eAAA,EAAiB,CAAA,CAAE,KAAA,CAAM;AAAA,cAC1B,CAAC;AAAA,YACF,CAAA;AAAA,UACD,EAAA,KAAA,GAAA,CAAW,MAAA,CAAO,KAAA,IAAS,WAAA,EAAa;AAEvC,YAAA,SAAA,CAAU,IAAA,CAAK,IAAA,CAAK,SAAA,CAAU,EAAE,IAAA,EAAM,MAAM,CAAC,CAAC,CAAA;AAAA,UAC/C;AAAA,QACD,EAAA,UAAQ;AAEP,UAAA,SAAA,CAAU,IAAA,CAAK,IAAI,CAAA;AAAA,QACpB;AAAA,MACD;AAAA,IACD,CAAC,CAAA;AAED,IAAA,SAAA,CAAU,gBAAA,CAAiB,OAAA,EAAS,CAAA,EAAA,GAAM;AACzC,MAAA,CAAA,CAAE,KAAA,CAAM,gBAAA,GAAmB,CAAA;AAC3B,MAAA,CAAA,CAAE,GAAA,CAAI,IAAA,CAAK;AAAA,QACV,GAAA,EAAK,wBAAA;AAAA,QACL,eAAA,EAAiB,CAAA,CAAE,KAAA,CAAM;AAAA,MAC1B,CAAC,CAAA;AAAA,IACF,CAAC,CAAA;AAAA,EACF,CAAA;AAAA,EACA,OAAA,EAAS;AAAA,IACR,SAAA,EAAW,CAAC,CAAA,EAAA,GAAM;AACjB,MAAA,OAAO;AAAA,QACN,UAAA,EAAY,CAAA,CAAE,KAAA,CAAM,UAAA;AAAA,QACpB,UAAA,EAAY,CAAA,CAAE,KAAA,CAAM,UAAA;AAAA,QACpB,eAAA,EAAiB,CAAA,CAAE,KAAA,CAAM;AAAA,MAC1B,CAAA;AAAA,IACD;AAAA,EACD,CAAA;AAAA,EACA,OAAA,EAAS;AAAA,IACR,YAAA,EAAc;AAAA,EACf;AACD,CAAC,CAAA;AAEM,IAAM,uBAAA,EAAyB,qCAAA;AAAM,EAC3C,KAAA,EAAO,EAAE,UAAA,EAAY,CAAA,EAAG,UAAA,EAAY,EAAE,CAAA;AAAA,EACtC,MAAA,EAAQ,CAAC,CAAA,EAAA,GAAM;AACd,IAAA,CAAA,CAAE,KAAA,CAAM,WAAA,GAAc,CAAA;AAAA,EACvB,CAAA;AAAA,EACA,OAAA,EAAS,CAAC,CAAA,EAAA,GAAM;AACf,IAAA,CAAA,CAAE,KAAA,CAAM,WAAA,GAAc,CAAA;AAAA,EACvB,CAAA;AAAA,EACA,OAAA,EAAS;AAAA,IACR,SAAA,EAAW,CAAC,CAAA,EAAA,GAAM;AACjB,MAAA,OAAO;AAAA,QACN,UAAA,EAAY,CAAA,CAAE,KAAA,CAAM,UAAA;AAAA,QACpB,UAAA,EAAY,CAAA,CAAE,KAAA,CAAM;AAAA,MACrB,CAAA;AAAA,IACD;AAAA,EACD,CAAA;AAAA,EACA,OAAA,EAAS;AAAA,IACR,YAAA,EAAc,aAAA;AAAA,IACd,OAAA,EAAS;AAAA,EACV;AACD,CAAC,CAAA;AHsED;AACA;AACE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACF,4YAAC","file":"/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/chunk-2IJTYN6K.cjs","sourcesContent":[null,"import { getLogger } from \"@/common/log\";\n\nexport function logger() {\n\treturn getLogger(\"test-suite\");\n}\n","import { actor } from \"rivetkit\";\n\nexport const HIBERNATION_SLEEP_TIMEOUT = 500;\n\nexport type HibernationConnState = {\n\tcount: number;\n\tconnectCount: number;\n\tdisconnectCount: number;\n};\n\nexport const hibernationActor = actor({\n\tstate: {\n\t\tsleepCount: 0,\n\t\twakeCount: 0,\n\t},\n\tcreateConnState: (c): HibernationConnState => {\n\t\treturn {\n\t\t\tcount: 0,\n\t\t\tconnectCount: 0,\n\t\t\tdisconnectCount: 0,\n\t\t};\n\t},\n\tonWake: (c) => {\n\t\tc.state.wakeCount += 1;\n\t},\n\tonSleep: (c) => {\n\t\tc.state.sleepCount += 1;\n\t},\n\tonConnect: (c, conn) => {\n\t\tconn.state.connectCount += 1;\n\t},\n\tonDisconnect: (c, conn) => {\n\t\tconn.state.disconnectCount += 1;\n\t},\n\tactions: {\n\t\t// Basic RPC that returns a simple value\n\t\tping: (c) => {\n\t\t\treturn \"pong\";\n\t\t},\n\t\t// Increment the connection's count\n\t\tconnIncrement: (c) => {\n\t\t\tc.conn.state.count += 1;\n\t\t\treturn c.conn.state.count;\n\t\t},\n\t\t// Get the connection's count\n\t\tgetConnCount: (c) => {\n\t\t\treturn c.conn.state.count;\n\t\t},\n\t\t// Get the connection's lifecycle counts\n\t\tgetConnLifecycleCounts: (c) => {\n\t\t\treturn {\n\t\t\t\tconnectCount: c.conn.state.connectCount,\n\t\t\t\tdisconnectCount: c.conn.state.disconnectCount,\n\t\t\t};\n\t\t},\n\t\t// Get all connection IDs\n\t\tgetConnectionIds: (c) => {\n\t\t\treturn c.conns\n\t\t\t\t.entries()\n\t\t\t\t.map((x) => x[0])\n\t\t\t\t.toArray();\n\t\t},\n\t\t// Get actor sleep/wake counts\n\t\tgetActorCounts: (c) => {\n\t\t\treturn {\n\t\t\t\tsleepCount: c.state.sleepCount,\n\t\t\t\twakeCount: c.state.wakeCount,\n\t\t\t};\n\t\t},\n\t\t// Trigger sleep\n\t\ttriggerSleep: (c) => {\n\t\t\tc.sleep();\n\t\t},\n\t},\n\toptions: {\n\t\tsleepTimeout: HIBERNATION_SLEEP_TIMEOUT,\n\t},\n});\n","import { actor, type UniversalWebSocket } from \"rivetkit\";\nimport { promiseWithResolvers } from \"rivetkit/utils\";\n\nexport const SLEEP_TIMEOUT = 1000;\n\nexport const sleep = actor({\n\tstate: { startCount: 0, sleepCount: 0 },\n\tonWake: (c) => {\n\t\tc.state.startCount += 1;\n\t},\n\tonSleep: (c) => {\n\t\tc.state.sleepCount += 1;\n\t},\n\tactions: {\n\t\ttriggerSleep: (c) => {\n\t\t\tc.sleep();\n\t\t},\n\t\tgetCounts: (c) => {\n\t\t\treturn {\n\t\t\t\tstartCount: c.state.startCount,\n\t\t\t\tsleepCount: c.state.sleepCount,\n\t\t\t};\n\t\t},\n\t\tsetAlarm: async (c, duration: number) => {\n\t\t\tawait c.schedule.after(duration, \"onAlarm\");\n\t\t},\n\t\tonAlarm: (c) => {\n\t\t\tc.log.info(\"alarm called\");\n\t\t},\n\t},\n\toptions: {\n\t\tsleepTimeout: SLEEP_TIMEOUT,\n\t},\n});\n\nexport const sleepWithLongRpc = actor({\n\tstate: { startCount: 0, sleepCount: 0 },\n\tcreateVars: () =>\n\t\t({}) as { longRunningResolve: PromiseWithResolvers<void> },\n\tonWake: (c) => {\n\t\tc.state.startCount += 1;\n\t},\n\tonSleep: (c) => {\n\t\tc.state.sleepCount += 1;\n\t},\n\tactions: {\n\t\tgetCounts: (c) => {\n\t\t\treturn {\n\t\t\t\tstartCount: c.state.startCount,\n\t\t\t\tsleepCount: c.state.sleepCount,\n\t\t\t};\n\t\t},\n\t\tlongRunningRpc: async (c) => {\n\t\t\tc.log.info(\"starting long running rpc\");\n\t\t\tc.vars.longRunningResolve = promiseWithResolvers();\n\t\t\tc.broadcast(\"waiting\");\n\t\t\tawait c.vars.longRunningResolve.promise;\n\t\t\tc.log.info(\"finished long running rpc\");\n\t\t},\n\t\tfinishLongRunningRpc: (c) => c.vars.longRunningResolve?.resolve(),\n\t},\n\toptions: {\n\t\tsleepTimeout: SLEEP_TIMEOUT,\n\t},\n});\n\nexport const sleepWithRawHttp = actor({\n\tstate: { startCount: 0, sleepCount: 0, requestCount: 0 },\n\tonWake: (c) => {\n\t\tc.state.startCount += 1;\n\t},\n\tonSleep: (c) => {\n\t\tc.state.sleepCount += 1;\n\t},\n\tonRequest: async (c, request) => {\n\t\tc.state.requestCount += 1;\n\t\tconst url = new URL(request.url);\n\n\t\tif (url.pathname === \"/long-request\") {\n\t\t\tconst duration = parseInt(\n\t\t\t\turl.searchParams.get(\"duration\") || \"1000\",\n\t\t\t);\n\t\t\tc.log.info({ msg: \"starting long fetch request\", duration });\n\t\t\tawait new Promise((resolve) => setTimeout(resolve, duration));\n\t\t\tc.log.info(\"finished long fetch request\");\n\t\t\treturn new Response(JSON.stringify({ completed: true }), {\n\t\t\t\theaders: { \"Content-Type\": \"application/json\" },\n\t\t\t});\n\t\t}\n\n\t\treturn new Response(\"Not Found\", { status: 404 });\n\t},\n\tactions: {\n\t\tgetCounts: (c) => {\n\t\t\treturn {\n\t\t\t\tstartCount: c.state.startCount,\n\t\t\t\tsleepCount: c.state.sleepCount,\n\t\t\t\trequestCount: c.state.requestCount,\n\t\t\t};\n\t\t},\n\t},\n\toptions: {\n\t\tsleepTimeout: SLEEP_TIMEOUT,\n\t},\n});\n\nexport const sleepWithRawWebSocket = actor({\n\tstate: { startCount: 0, sleepCount: 0, connectionCount: 0 },\n\tonWake: (c) => {\n\t\tc.state.startCount += 1;\n\t},\n\tonSleep: (c) => {\n\t\tc.state.sleepCount += 1;\n\t},\n\tonWebSocket: (c, websocket: UniversalWebSocket) => {\n\t\tc.state.connectionCount += 1;\n\t\tc.log.info({\n\t\t\tmsg: \"websocket connected\",\n\t\t\tconnectionCount: c.state.connectionCount,\n\t\t});\n\n\t\twebsocket.send(\n\t\t\tJSON.stringify({\n\t\t\t\ttype: \"connected\",\n\t\t\t\tconnectionCount: c.state.connectionCount,\n\t\t\t}),\n\t\t);\n\n\t\twebsocket.addEventListener(\"message\", (event: any) => {\n\t\t\tconst data = event.data;\n\t\t\tif (typeof data === \"string\") {\n\t\t\t\ttry {\n\t\t\t\t\tconst parsed = JSON.parse(data);\n\t\t\t\t\tif (parsed.type === \"getCounts\") {\n\t\t\t\t\t\twebsocket.send(\n\t\t\t\t\t\t\tJSON.stringify({\n\t\t\t\t\t\t\t\ttype: \"counts\",\n\t\t\t\t\t\t\t\tstartCount: c.state.startCount,\n\t\t\t\t\t\t\t\tsleepCount: c.state.sleepCount,\n\t\t\t\t\t\t\t\tconnectionCount: c.state.connectionCount,\n\t\t\t\t\t\t\t}),\n\t\t\t\t\t\t);\n\t\t\t\t\t} else if (parsed.type === \"keepAlive\") {\n\t\t\t\t\t\t// Just acknowledge to keep connection alive\n\t\t\t\t\t\twebsocket.send(JSON.stringify({ type: \"ack\" }));\n\t\t\t\t\t}\n\t\t\t\t} catch {\n\t\t\t\t\t// Echo non-JSON messages\n\t\t\t\t\twebsocket.send(data);\n\t\t\t\t}\n\t\t\t}\n\t\t});\n\n\t\twebsocket.addEventListener(\"close\", () => {\n\t\t\tc.state.connectionCount -= 1;\n\t\t\tc.log.info({\n\t\t\t\tmsg: \"websocket disconnected\",\n\t\t\t\tconnectionCount: c.state.connectionCount,\n\t\t\t});\n\t\t});\n\t},\n\tactions: {\n\t\tgetCounts: (c) => {\n\t\t\treturn {\n\t\t\t\tstartCount: c.state.startCount,\n\t\t\t\tsleepCount: c.state.sleepCount,\n\t\t\t\tconnectionCount: c.state.connectionCount,\n\t\t\t};\n\t\t},\n\t},\n\toptions: {\n\t\tsleepTimeout: SLEEP_TIMEOUT,\n\t},\n});\n\nexport const sleepWithNoSleepOption = actor({\n\tstate: { startCount: 0, sleepCount: 0 },\n\tonWake: (c) => {\n\t\tc.state.startCount += 1;\n\t},\n\tonSleep: (c) => {\n\t\tc.state.sleepCount += 1;\n\t},\n\tactions: {\n\t\tgetCounts: (c) => {\n\t\t\treturn {\n\t\t\t\tstartCount: c.state.startCount,\n\t\t\t\tsleepCount: c.state.sleepCount,\n\t\t\t};\n\t\t},\n\t},\n\toptions: {\n\t\tsleepTimeout: SLEEP_TIMEOUT,\n\t\tnoSleep: true,\n\t},\n});\n"]}