rivetkit 2.0.3 → 2.0.4

package/README.md

@@ -0,0 +1,11 @@
+# RivetKit
+
+_Library to build and scale stateful workloads_
+
+[Learn More →](https://github.com/rivet-dev/rivetkit)
+
+[Discord](https://rivet.dev/discord) — [Documentation](https://rivetkit.org) — [Issues](https://github.com/rivet-dev/rivetkit/issues)
+
+## License
+
+Apache 2.0

package/dist/schemas/actor-persist/v1.ts

@@ -18,20 +18,7 @@ export function writePersistedSubscription(bc: bare.ByteCursor, x: PersistedSubs
     bare.writeString(bc, x.eventName)
 }
 
-function read0(bc: bare.ByteCursor): ArrayBuffer | null {
-    return bare.readBool(bc)
-        ? bare.readData(bc)
-        : null
-}
-
-function write0(bc: bare.ByteCursor, x: ArrayBuffer | null): void {
-    bare.writeBool(bc, x !== null)
-    if (x !== null) {
-        bare.writeData(bc, x)
-    }
-}
-
-function read1(bc: bare.ByteCursor): readonly PersistedSubscription[] {
+function read0(bc: bare.ByteCursor): readonly PersistedSubscription[] {
     const len = bare.readUintSafe(bc)
     if (len === 0) { return [] }
     const result = [readPersistedSubscription(bc)]
@@ -41,7 +28,7 @@ function read1(bc: bare.ByteCursor): readonly PersistedSubscription[] {
     return result
 }
 
-function write1(bc: bare.ByteCursor, x: readonly PersistedSubscription[]): void {
+function write0(bc: bare.ByteCursor, x: readonly PersistedSubscription[]): void {
     bare.writeUintSafe(bc, x.length)
     for (let i = 0; i < x.length; i++) {
         writePersistedSubscription(bc, x[i])
@@ -55,7 +42,6 @@ export type PersistedConnection = {
     readonly driverState: ArrayBuffer,
     readonly parameters: ArrayBuffer,
     readonly state: ArrayBuffer,
-    readonly auth: ArrayBuffer | null,
     readonly subscriptions: readonly PersistedSubscription[],
     readonly lastSeen: u64,
 }
@@ -68,8 +54,7 @@ export function readPersistedConnection(bc: bare.ByteCursor): PersistedConnectio
         driverState: bare.readData(bc),
         parameters: bare.readData(bc),
         state: bare.readData(bc),
-        auth: read0(bc),
-        subscriptions: read1(bc),
+        subscriptions: read0(bc),
         lastSeen: bare.readU64(bc),
     }
 }
@@ -81,11 +66,23 @@ export function writePersistedConnection(bc: bare.ByteCursor, x: PersistedConnec
     bare.writeData(bc, x.driverState)
     bare.writeData(bc, x.parameters)
     bare.writeData(bc, x.state)
-    write0(bc, x.auth)
-    write1(bc, x.subscriptions)
+    write0(bc, x.subscriptions)
     bare.writeU64(bc, x.lastSeen)
 }
 
+function read1(bc: bare.ByteCursor): ArrayBuffer | null {
+    return bare.readBool(bc)
+        ? bare.readData(bc)
+        : null
+}
+
+function write1(bc: bare.ByteCursor, x: ArrayBuffer | null): void {
+    bare.writeBool(bc, x !== null)
+    if (x !== null) {
+        bare.writeData(bc, x)
+    }
+}
+
 export type GenericPersistedScheduleEvent = {
     readonly action: string,
     readonly args: ArrayBuffer | null,
@@ -94,13 +91,13 @@ export type GenericPersistedScheduleEvent = {
 export function readGenericPersistedScheduleEvent(bc: bare.ByteCursor): GenericPersistedScheduleEvent {
     return {
         action: bare.readString(bc),
-        args: read0(bc),
+        args: read1(bc),
     }
 }
 
 export function writeGenericPersistedScheduleEvent(bc: bare.ByteCursor, x: GenericPersistedScheduleEvent): void {
     bare.writeString(bc, x.action)
-    write0(bc, x.args)
+    write1(bc, x.args)
 }
 
 export type PersistedScheduleEventKind =
@@ -193,7 +190,7 @@ export type PersistedActor = {
 
 export function readPersistedActor(bc: bare.ByteCursor): PersistedActor {
     return {
-        input: read0(bc),
+        input: read1(bc),
         hasInitialized: bare.readBool(bc),
         state: bare.readData(bc),
         connections: read2(bc),
@@ -202,7 +199,7 @@ export function readPersistedActor(bc: bare.ByteCursor): PersistedActor {
 }
 
 export function writePersistedActor(bc: bare.ByteCursor, x: PersistedActor): void {
-    write0(bc, x.input)
+    write1(bc, x.input)
     bare.writeBool(bc, x.hasInitialized)
     bare.writeData(bc, x.state)
     write2(bc, x.connections)

package/dist/schemas/client-protocol/v1.ts

@@ -51,6 +51,7 @@ function write1(bc: bare.ByteCursor, x: uint | null): void {
 }
 
 export type Error = {
+    readonly group: string,
     readonly code: string,
     readonly message: string,
     readonly metadata: ArrayBuffer | null,
@@ -59,6 +60,7 @@ export type Error = {
 
 export function readError(bc: bare.ByteCursor): Error {
     return {
+        group: bare.readString(bc),
         code: bare.readString(bc),
         message: bare.readString(bc),
         metadata: read0(bc),
@@ -67,6 +69,7 @@ export function readError(bc: bare.ByteCursor): Error {
 }
 
 export function writeError(bc: bare.ByteCursor, x: Error): void {
+    bare.writeString(bc, x.group)
     bare.writeString(bc, x.code)
     bare.writeString(bc, x.message)
     write0(bc, x.metadata)
@@ -357,6 +360,7 @@ export function decodeHttpActionResponse(bytes: Uint8Array): HttpActionResponse
 }
 
 export type HttpResponseError = {
+    readonly group: string,
     readonly code: string,
     readonly message: string,
     readonly metadata: ArrayBuffer | null,
@@ -364,6 +368,7 @@ export type HttpResponseError = {
 
 export function readHttpResponseError(bc: bare.ByteCursor): HttpResponseError {
     return {
+        group: bare.readString(bc),
         code: bare.readString(bc),
         message: bare.readString(bc),
         metadata: read0(bc),
@@ -371,6 +376,7 @@ export function readHttpResponseError(bc: bare.ByteCursor): HttpResponseError {
 }
 
 export function writeHttpResponseError(bc: bare.ByteCursor, x: HttpResponseError): void {
+    bare.writeString(bc, x.group)
     bare.writeString(bc, x.code)
     bare.writeString(bc, x.message)
     write0(bc, x.metadata)

package/dist/tsup/actor/errors.cjs

@@ -31,11 +31,11 @@
 
 
 
-var _chunkGIR3AFFIcjs = require('../chunk-GIR3AFFI.cjs');
 
 
 
 
+var _chunkSBHHJ6QScjs = require('../chunk-SBHHJ6QS.cjs');
 
 
 
@@ -65,5 +65,13 @@ var _chunkGIR3AFFIcjs = require('../chunk-GIR3AFFI.cjs');
 
 
 
-exports.ActionNotFound = _chunkGIR3AFFIcjs.ActionNotFound; exports.ActionTimedOut = _chunkGIR3AFFIcjs.ActionTimedOut; exports.ActorAlreadyExists = _chunkGIR3AFFIcjs.ActorAlreadyExists; exports.ActorError = _chunkGIR3AFFIcjs.ActorError; exports.ActorNotFound = _chunkGIR3AFFIcjs.ActorNotFound; exports.ConnNotFound = _chunkGIR3AFFIcjs.ConnNotFound; exports.ConnStateNotEnabled = _chunkGIR3AFFIcjs.ConnStateNotEnabled; exports.DatabaseNotEnabled = _chunkGIR3AFFIcjs.DatabaseNotEnabled; exports.FetchHandlerNotDefined = _chunkGIR3AFFIcjs.FetchHandlerNotDefined; exports.Forbidden = _chunkGIR3AFFIcjs.Forbidden; exports.INTERNAL_ERROR_CODE = _chunkGIR3AFFIcjs.INTERNAL_ERROR_CODE; exports.INTERNAL_ERROR_DESCRIPTION = _chunkGIR3AFFIcjs.INTERNAL_ERROR_DESCRIPTION; exports.IncorrectConnToken = _chunkGIR3AFFIcjs.IncorrectConnToken; exports.InternalError = _chunkGIR3AFFIcjs.InternalError; exports.InvalidActionRequest = _chunkGIR3AFFIcjs.InvalidActionRequest; exports.InvalidEncoding = _chunkGIR3AFFIcjs.InvalidEncoding; exports.InvalidFetchResponse = _chunkGIR3AFFIcjs.InvalidFetchResponse; exports.InvalidParams = _chunkGIR3AFFIcjs.InvalidParams; exports.InvalidQueryJSON = _chunkGIR3AFFIcjs.InvalidQueryJSON; exports.InvalidRequest = _chunkGIR3AFFIcjs.InvalidRequest; exports.InvalidStateType = _chunkGIR3AFFIcjs.InvalidStateType; exports.MalformedMessage = _chunkGIR3AFFIcjs.MalformedMessage; exports.MessageTooLong = _chunkGIR3AFFIcjs.MessageTooLong; exports.ProxyError = _chunkGIR3AFFIcjs.ProxyError; exports.StateNotEnabled = _chunkGIR3AFFIcjs.StateNotEnabled; exports.USER_ERROR_CODE = _chunkGIR3AFFIcjs.USER_ERROR_CODE; exports.Unauthorized = _chunkGIR3AFFIcjs.Unauthorized; exports.Unreachable = _chunkGIR3AFFIcjs.Unreachable; exports.Unsupported = _chunkGIR3AFFIcjs.Unsupported; exports.UserError = _chunkGIR3AFFIcjs.UserError; exports.VarsNotEnabled = _chunkGIR3AFFIcjs.VarsNotEnabled; exports.WebSocketHandlerNotDefined = _chunkGIR3AFFIcjs.WebSocketHandlerNotDefined;
+
+
+
+
+
+
+
+
+exports.ActionNotFound = _chunkSBHHJ6QScjs.ActionNotFound; exports.ActionTimedOut = _chunkSBHHJ6QScjs.ActionTimedOut; exports.ActorAlreadyExists = _chunkSBHHJ6QScjs.ActorAlreadyExists; exports.ActorError = _chunkSBHHJ6QScjs.ActorError; exports.ActorNotFound = _chunkSBHHJ6QScjs.ActorNotFound; exports.ConnNotFound = _chunkSBHHJ6QScjs.ConnNotFound; exports.ConnStateNotEnabled = _chunkSBHHJ6QScjs.ConnStateNotEnabled; exports.DatabaseNotEnabled = _chunkSBHHJ6QScjs.DatabaseNotEnabled; exports.FeatureNotImplemented = _chunkSBHHJ6QScjs.FeatureNotImplemented; exports.FetchHandlerNotDefined = _chunkSBHHJ6QScjs.FetchHandlerNotDefined; exports.Forbidden = _chunkSBHHJ6QScjs.Forbidden; exports.INTERNAL_ERROR_CODE = _chunkSBHHJ6QScjs.INTERNAL_ERROR_CODE; exports.INTERNAL_ERROR_DESCRIPTION = _chunkSBHHJ6QScjs.INTERNAL_ERROR_DESCRIPTION; exports.IncorrectConnToken = _chunkSBHHJ6QScjs.IncorrectConnToken; exports.InternalError = _chunkSBHHJ6QScjs.InternalError; exports.InvalidActionRequest = _chunkSBHHJ6QScjs.InvalidActionRequest; exports.InvalidEncoding = _chunkSBHHJ6QScjs.InvalidEncoding; exports.InvalidFetchResponse = _chunkSBHHJ6QScjs.InvalidFetchResponse; exports.InvalidParams = _chunkSBHHJ6QScjs.InvalidParams; exports.InvalidQueryJSON = _chunkSBHHJ6QScjs.InvalidQueryJSON; exports.InvalidRequest = _chunkSBHHJ6QScjs.InvalidRequest; exports.InvalidStateType = _chunkSBHHJ6QScjs.InvalidStateType; exports.MalformedMessage = _chunkSBHHJ6QScjs.MalformedMessage; exports.MessageTooLong = _chunkSBHHJ6QScjs.MessageTooLong; exports.MissingActorHeader = _chunkSBHHJ6QScjs.MissingActorHeader; exports.ProxyError = _chunkSBHHJ6QScjs.ProxyError; exports.RouteNotFound = _chunkSBHHJ6QScjs.RouteNotFound; exports.StateNotEnabled = _chunkSBHHJ6QScjs.StateNotEnabled; exports.USER_ERROR_CODE = _chunkSBHHJ6QScjs.USER_ERROR_CODE; exports.Unauthorized = _chunkSBHHJ6QScjs.Unauthorized; exports.Unreachable = _chunkSBHHJ6QScjs.Unreachable; exports.Unsupported = _chunkSBHHJ6QScjs.Unsupported; exports.UserError = _chunkSBHHJ6QScjs.UserError; exports.VarsNotEnabled = _chunkSBHHJ6QScjs.VarsNotEnabled; exports.WebSocketHandlerNotDefined = _chunkSBHHJ6QScjs.WebSocketHandlerNotDefined; exports.WebSocketsNotEnabled = _chunkSBHHJ6QScjs.WebSocketsNotEnabled;
 //# sourceMappingURL=errors.cjs.map

package/dist/tsup/actor/errors.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/Users/nathan/rivetkit/packages/rivetkit/dist/tsup/actor/errors.cjs"],"names":[],"mappings":"AAAA;AACE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACF,yDAA8B;AAC9B;AACE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACF,u7DAAC","file":"/Users/nathan/rivetkit/packages/rivetkit/dist/tsup/actor/errors.cjs"}
+{"version":3,"sources":["/Users/nathan/rivetkit/packages/rivetkit/dist/tsup/actor/errors.cjs"],"names":[],"mappings":"AAAA;AACE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACF,yDAA8B;AAC9B;AACE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACF,msEAAC","file":"/Users/nathan/rivetkit/packages/rivetkit/dist/tsup/actor/errors.cjs"}

package/dist/tsup/actor/errors.d.cts

@@ -1,4 +1,4 @@
-import { D as DeconstructedError } from '../utils-CT0cv4jd.cjs';
+import { D as DeconstructedError } from '../utils-fwx3o3K9.cjs';
 import 'hono';
 import 'hono/utils/http-status';
 
@@ -13,13 +13,14 @@ interface ActorErrorOptions extends ErrorOptions {
     metadata?: unknown;
 }
 declare class ActorError extends Error {
-    readonly code: string;
     __type: string;
     public: boolean;
     metadata?: unknown;
     statusCode: number;
+    readonly group: string;
+    readonly code: string;
     static isActorError(error: unknown): error is ActorError | DeconstructedError;
-    constructor(code: string, message: string, opts?: ActorErrorOptions);
+    constructor(group: string, code: string, message: string, opts?: ActorErrorOptions);
     toString(): string;
     /**
      * Serialize error for HTTP response
@@ -139,5 +140,17 @@ declare class WebSocketHandlerNotDefined extends ActorError {
 declare class InvalidFetchResponse extends ActorError {
     constructor();
 }
+declare class MissingActorHeader extends ActorError {
+    constructor();
+}
+declare class WebSocketsNotEnabled extends ActorError {
+    constructor();
+}
+declare class FeatureNotImplemented extends ActorError {
+    constructor(feature: string);
+}
+declare class RouteNotFound extends ActorError {
+    constructor();
+}
 
-export { ActionNotFound, ActionTimedOut, ActorAlreadyExists, ActorError, ActorNotFound, ConnNotFound, ConnStateNotEnabled, DatabaseNotEnabled, FetchHandlerNotDefined, Forbidden, INTERNAL_ERROR_CODE, INTERNAL_ERROR_DESCRIPTION, IncorrectConnToken, InternalError, type InternalErrorMetadata, InvalidActionRequest, InvalidEncoding, InvalidFetchResponse, InvalidParams, InvalidQueryJSON, InvalidRequest, InvalidStateType, type InvalidStateTypeOptions, MalformedMessage, MessageTooLong, ProxyError, StateNotEnabled, USER_ERROR_CODE, Unauthorized, Unreachable, Unsupported, UserError, type UserErrorOptions, VarsNotEnabled, WebSocketHandlerNotDefined };
+export { ActionNotFound, ActionTimedOut, ActorAlreadyExists, ActorError, ActorNotFound, ConnNotFound, ConnStateNotEnabled, DatabaseNotEnabled, FeatureNotImplemented, FetchHandlerNotDefined, Forbidden, INTERNAL_ERROR_CODE, INTERNAL_ERROR_DESCRIPTION, IncorrectConnToken, InternalError, type InternalErrorMetadata, InvalidActionRequest, InvalidEncoding, InvalidFetchResponse, InvalidParams, InvalidQueryJSON, InvalidRequest, InvalidStateType, type InvalidStateTypeOptions, MalformedMessage, MessageTooLong, MissingActorHeader, ProxyError, RouteNotFound, StateNotEnabled, USER_ERROR_CODE, Unauthorized, Unreachable, Unsupported, UserError, type UserErrorOptions, VarsNotEnabled, WebSocketHandlerNotDefined, WebSocketsNotEnabled };

package/dist/tsup/actor/errors.d.ts

@@ -1,4 +1,4 @@
-import { D as DeconstructedError } from '../utils-CT0cv4jd.js';
+import { D as DeconstructedError } from '../utils-fwx3o3K9.js';
 import 'hono';
 import 'hono/utils/http-status';
 
@@ -13,13 +13,14 @@ interface ActorErrorOptions extends ErrorOptions {
     metadata?: unknown;
 }
 declare class ActorError extends Error {
-    readonly code: string;
     __type: string;
     public: boolean;
     metadata?: unknown;
     statusCode: number;
+    readonly group: string;
+    readonly code: string;
     static isActorError(error: unknown): error is ActorError | DeconstructedError;
-    constructor(code: string, message: string, opts?: ActorErrorOptions);
+    constructor(group: string, code: string, message: string, opts?: ActorErrorOptions);
     toString(): string;
     /**
      * Serialize error for HTTP response
@@ -139,5 +140,17 @@ declare class WebSocketHandlerNotDefined extends ActorError {
 declare class InvalidFetchResponse extends ActorError {
     constructor();
 }
+declare class MissingActorHeader extends ActorError {
+    constructor();
+}
+declare class WebSocketsNotEnabled extends ActorError {
+    constructor();
+}
+declare class FeatureNotImplemented extends ActorError {
+    constructor(feature: string);
+}
+declare class RouteNotFound extends ActorError {
+    constructor();
+}
 
-export { ActionNotFound, ActionTimedOut, ActorAlreadyExists, ActorError, ActorNotFound, ConnNotFound, ConnStateNotEnabled, DatabaseNotEnabled, FetchHandlerNotDefined, Forbidden, INTERNAL_ERROR_CODE, INTERNAL_ERROR_DESCRIPTION, IncorrectConnToken, InternalError, type InternalErrorMetadata, InvalidActionRequest, InvalidEncoding, InvalidFetchResponse, InvalidParams, InvalidQueryJSON, InvalidRequest, InvalidStateType, type InvalidStateTypeOptions, MalformedMessage, MessageTooLong, ProxyError, StateNotEnabled, USER_ERROR_CODE, Unauthorized, Unreachable, Unsupported, UserError, type UserErrorOptions, VarsNotEnabled, WebSocketHandlerNotDefined };
+export { ActionNotFound, ActionTimedOut, ActorAlreadyExists, ActorError, ActorNotFound, ConnNotFound, ConnStateNotEnabled, DatabaseNotEnabled, FeatureNotImplemented, FetchHandlerNotDefined, Forbidden, INTERNAL_ERROR_CODE, INTERNAL_ERROR_DESCRIPTION, IncorrectConnToken, InternalError, type InternalErrorMetadata, InvalidActionRequest, InvalidEncoding, InvalidFetchResponse, InvalidParams, InvalidQueryJSON, InvalidRequest, InvalidStateType, type InvalidStateTypeOptions, MalformedMessage, MessageTooLong, MissingActorHeader, ProxyError, RouteNotFound, StateNotEnabled, USER_ERROR_CODE, Unauthorized, Unreachable, Unsupported, UserError, type UserErrorOptions, VarsNotEnabled, WebSocketHandlerNotDefined, WebSocketsNotEnabled };

package/dist/tsup/actor/errors.js

@@ -7,6 +7,7 @@ import {
   ConnNotFound,
   ConnStateNotEnabled,
   DatabaseNotEnabled,
+  FeatureNotImplemented,
   FetchHandlerNotDefined,
   Forbidden,
   INTERNAL_ERROR_CODE,
@@ -22,7 +23,9 @@ import {
   InvalidStateType,
   MalformedMessage,
   MessageTooLong,
+  MissingActorHeader,
   ProxyError,
+  RouteNotFound,
   StateNotEnabled,
   USER_ERROR_CODE,
   Unauthorized,
@@ -30,8 +33,9 @@ import {
   Unsupported,
   UserError,
   VarsNotEnabled,
-  WebSocketHandlerNotDefined
-} from "../chunk-INGJP237.js";
+  WebSocketHandlerNotDefined,
+  WebSocketsNotEnabled
+} from "../chunk-NFU2BBT5.js";
 export {
   ActionNotFound,
   ActionTimedOut,
@@ -41,6 +45,7 @@ export {
   ConnNotFound,
   ConnStateNotEnabled,
   DatabaseNotEnabled,
+  FeatureNotImplemented,
   FetchHandlerNotDefined,
   Forbidden,
   INTERNAL_ERROR_CODE,
@@ -56,7 +61,9 @@ export {
   InvalidStateType,
   MalformedMessage,
   MessageTooLong,
+  MissingActorHeader,
   ProxyError,
+  RouteNotFound,
   StateNotEnabled,
   USER_ERROR_CODE,
   Unauthorized,
@@ -64,6 +71,7 @@ export {
   Unsupported,
   UserError,
   VarsNotEnabled,
-  WebSocketHandlerNotDefined
+  WebSocketHandlerNotDefined,
+  WebSocketsNotEnabled
 };
 //# sourceMappingURL=errors.js.map

package/dist/tsup/{chunk-6PDXBYI5.js → chunk-3F2YSRJL.js}

@@ -1,17 +1,13 @@
 import {
   createFileSystemOrMemoryDriver,
-  createInlineClientDriver,
   createManagerRouter
-} from "./chunk-OV6AYD4S.js";
-import {
-  createClient
-} from "./chunk-UVUPOS46.js";
+} from "./chunk-XJQHKJ4P.js";
 import {
   RunConfigSchema
-} from "./chunk-KJCJLKRM.js";
+} from "./chunk-TQ62L3X7.js";
 import {
   getLogger
-} from "./chunk-XFSS33EQ.js";
+} from "./chunk-LV2S3OU3.js";
 
 // src/test/mod.ts
 import { createServer } from "net";
@@ -27,9 +23,8 @@ var ConfigSchema = RunConfigSchema.removeDefault().extend({
 }).default({});
 
 // src/test/log.ts
-var LOGGER_NAME = "nodejs";
 function logger() {
-  return getLogger(LOGGER_NAME);
+  return getLogger("test");
 }
 
 // src/test/mod.ts
@@ -43,11 +38,9 @@ function serve(registry, inputConfig) {
   const runConfig = RunConfigSchema.parse(inputConfig);
   const driver = inputConfig.driver ?? createFileSystemOrMemoryDriver(false);
   const managerDriver = driver.manager(registry.config, config);
-  const inlineClientDriver = createInlineClientDriver(managerDriver);
   const { router } = createManagerRouter(
     registry.config,
     runConfig,
-    inlineClientDriver,
     managerDriver,
     false
   );
@@ -59,7 +52,8 @@ function serve(registry, inputConfig) {
     port: config.port
   });
   nodeWebSocket.injectWebSocket(server);
-  logger().info("rivetkit started", {
+  logger().info({
+    msg: "rivetkit started",
     hostname: config.hostname,
     port: config.port,
     definitions: Object.keys(registry.config.use).length
@@ -77,16 +71,7 @@ async function setupTest(c, registry) {
   c.onTestFinished(
     async () => await new Promise((resolve) => server.close(() => resolve()))
   );
-  const client = createClient(`http://127.0.0.1:${port}`);
-  c.onTestFinished(async () => await client.dispose());
-  return {
-    client,
-    mockDriver: {
-      actorDriver: {
-        setCreateVarsContext: setDriverContextFn
-      }
-    }
-  };
+  throw "TODO: Fix engine port";
 }
 async function getPort() {
   const MIN_PORT = 1e4;
@@ -129,4 +114,4 @@ export {
   setupTest,
   getPort
 };
-//# sourceMappingURL=chunk-6PDXBYI5.js.map
+//# sourceMappingURL=chunk-3F2YSRJL.js.map

package/dist/tsup/chunk-3F2YSRJL.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/test/mod.ts","../../src/test/config.ts","../../src/test/log.ts"],"sourcesContent":["import { createServer } from \"node:net\";\nimport { serve as honoServe, type ServerType } from \"@hono/node-server\";\nimport { createNodeWebSocket } from \"@hono/node-ws\";\nimport { type TestContext, vi } from \"vitest\";\nimport { type Client, createClient } from \"@/client/mod\";\nimport { chooseDefaultDriver } from \"@/drivers/default\";\nimport { createFileSystemOrMemoryDriver } from \"@/drivers/file-system/mod\";\nimport { getInspectorUrl } from \"@/inspector/utils\";\nimport { createManagerRouter } from \"@/manager/router\";\nimport type { Registry } from \"@/registry/mod\";\nimport { RunConfigSchema } from \"@/registry/run-config\";\nimport { ConfigSchema, type InputConfig } from \"./config\";\nimport { logger } from \"./log\";\n\nfunction serve(registry: Registry<any>, inputConfig?: InputConfig): ServerType {\n\t// Configure default configuration\n\tinputConfig ??= {};\n\n\tconst config = ConfigSchema.parse(inputConfig);\n\n\tlet upgradeWebSocket: any;\n\tif (!config.getUpgradeWebSocket) {\n\t\tconfig.getUpgradeWebSocket = () => upgradeWebSocket!;\n\t}\n\n\t// Create router\n\tconst runConfig = RunConfigSchema.parse(inputConfig);\n\tconst driver = inputConfig.driver ?? createFileSystemOrMemoryDriver(false);\n\tconst managerDriver = driver.manager(registry.config, config);\n\tconst { router } = createManagerRouter(\n\t\tregistry.config,\n\t\trunConfig,\n\t\tmanagerDriver,\n\t\tfalse,\n\t);\n\n\t// Inject WebSocket\n\tconst nodeWebSocket = createNodeWebSocket({ app: router });\n\tupgradeWebSocket = nodeWebSocket.upgradeWebSocket;\n\n\tconst server = honoServe({\n\t\tfetch: router.fetch,\n\t\thostname: config.hostname,\n\t\tport: config.port,\n\t});\n\tnodeWebSocket.injectWebSocket(server);\n\n\tlogger().info({\n\t\tmsg: \"rivetkit started\",\n\t\thostname: config.hostname,\n\t\tport: config.port,\n\t\tdefinitions: Object.keys(registry.config.use).length,\n\t});\n\n\treturn server;\n}\n\nexport interface SetupTestResult<A extends Registry<any>> {\n\tclient: Client<A>;\n\tmockDriver: {\n\t\tactorDriver: {\n\t\t\tsetCreateVarsContext: (ctx: any) => void;\n\t\t};\n\t};\n}\n\n// Must use `TestContext` since global hooks do not work when running concurrently\nexport async function setupTest<A extends Registry<any>>(\n\tc: TestContext,\n\tregistry: A,\n): Promise<SetupTestResult<A>> {\n\tvi.useFakeTimers();\n\n\t// Set up mock driver for testing createVars context\n\tconst mockDriverContext: any = {};\n\tconst setDriverContextFn = (ctx: any) => {\n\t\tmockDriverContext.current = ctx;\n\t};\n\n\t// We don't need to modify the driver context anymore since we're testing with the actual context\n\n\t// Start server with a random port\n\tconst port = await getPort();\n\tconst server = serve(registry, { port });\n\tc.onTestFinished(\n\t\tasync () => await new Promise((resolve) => server.close(() => resolve())),\n\t);\n\n\tthrow \"TODO: Fix engine port\";\n\n\t// // TODO: Figure out how to make this the correct endpoint\n\t// // Create client\n\t// const client = createClient<A>(`http://127.0.0.1:${port}`);\n\t// c.onTestFinished(async () => await client.dispose());\n\t//\n\t// return {\n\t// \tclient,\n\t// \tmockDriver: {\n\t// \t\tactorDriver: {\n\t// \t\t\tsetCreateVarsContext: setDriverContextFn,\n\t// \t\t},\n\t// \t},\n\t// };\n}\n\nexport async function getPort(): Promise<number> {\n\t// Pick random port between 10000 and 65535 (avoiding well-known and registered ports)\n\tconst MIN_PORT = 10000;\n\tconst MAX_PORT = 65535;\n\tconst getRandomPort = () =>\n\t\tMath.floor(Math.random() * (MAX_PORT - MIN_PORT + 1)) + MIN_PORT;\n\n\tlet port = getRandomPort();\n\tlet maxAttempts = 10;\n\n\twhile (maxAttempts > 0) {\n\t\ttry {\n\t\t\t// Try to create a server on the port to check if it's available\n\t\t\tconst server = await new Promise<any>((resolve, reject) => {\n\t\t\t\tconst server = createServer();\n\n\t\t\t\tserver.once(\"error\", (err: Error & { code?: string }) => {\n\t\t\t\t\tif (err.code === \"EADDRINUSE\") {\n\t\t\t\t\t\treject(new Error(`Port ${port} is in use`));\n\t\t\t\t\t} else {\n\t\t\t\t\t\treject(err);\n\t\t\t\t\t}\n\t\t\t\t});\n\n\t\t\t\tserver.once(\"listening\", () => {\n\t\t\t\t\tresolve(server);\n\t\t\t\t});\n\n\t\t\t\tserver.listen(port);\n\t\t\t});\n\n\t\t\t// Close the server since we're just checking availability\n\t\t\tawait new Promise<void>((resolve) => {\n\t\t\t\tserver.close(() => resolve());\n\t\t\t});\n\n\t\t\treturn port;\n\t\t} catch (err) {\n\t\t\t// If port is in use, try a different one\n\t\t\tmaxAttempts--;\n\t\t\tif (maxAttempts <= 0) {\n\t\t\t\tbreak;\n\t\t\t}\n\t\t\tport = getRandomPort();\n\t\t}\n\t}\n\n\tthrow new Error(\"Could not find an available port after multiple attempts\");\n}\n","import { z } from \"zod\";\nimport { RunConfigSchema } from \"@/registry/run-config\";\n\nexport const ConfigSchema = RunConfigSchema.removeDefault()\n\t.extend({\n\t\thostname: z\n\t\t\t.string()\n\t\t\t.optional()\n\t\t\t.default(process.env.HOSTNAME ?? \"127.0.0.1\"),\n\t\tport: z\n\t\t\t.number()\n\t\t\t.optional()\n\t\t\t.default(Number.parseInt(process.env.PORT ?? \"8080\")),\n\t})\n\t.default({});\nexport type InputConfig = z.input<typeof ConfigSchema>;\n","import { getLogger } from \"@/common/log\";\n\nexport function logger() {\n\treturn getLogger(\"test\");\n}\n"],"mappings":";;;;;;;;;;;;AAAA,SAAS,oBAAoB;AAC7B,SAAS,SAAS,iBAAkC;AACpD,SAAS,2BAA2B;AACpC,SAA2B,UAAU;;;ACHrC,SAAS,SAAS;AAGX,IAAM,eAAe,gBAAgB,cAAc,EACxD,OAAO;AAAA,EACP,UAAU,EACR,OAAO,EACP,SAAS,EACT,QAAQ,QAAQ,IAAI,YAAY,WAAW;AAAA,EAC7C,MAAM,EACJ,OAAO,EACP,SAAS,EACT,QAAQ,OAAO,SAAS,QAAQ,IAAI,QAAQ,MAAM,CAAC;AACtD,CAAC,EACA,QAAQ,CAAC,CAAC;;;ACZL,SAAS,SAAS;AACxB,SAAO,UAAU,MAAM;AACxB;;;AFUA,SAAS,MAAM,UAAyB,aAAuC;AAE9E,kBAAgB,CAAC;AAEjB,QAAM,SAAS,aAAa,MAAM,WAAW;AAE7C,MAAI;AACJ,MAAI,CAAC,OAAO,qBAAqB;AAChC,WAAO,sBAAsB,MAAM;AAAA,EACpC;AAGA,QAAM,YAAY,gBAAgB,MAAM,WAAW;AACnD,QAAM,SAAS,YAAY,UAAU,+BAA+B,KAAK;AACzE,QAAM,gBAAgB,OAAO,QAAQ,SAAS,QAAQ,MAAM;AAC5D,QAAM,EAAE,OAAO,IAAI;AAAA,IAClB,SAAS;AAAA,IACT;AAAA,IACA;AAAA,IACA;AAAA,EACD;AAGA,QAAM,gBAAgB,oBAAoB,EAAE,KAAK,OAAO,CAAC;AACzD,qBAAmB,cAAc;AAEjC,QAAM,SAAS,UAAU;AAAA,IACxB,OAAO,OAAO;AAAA,IACd,UAAU,OAAO;AAAA,IACjB,MAAM,OAAO;AAAA,EACd,CAAC;AACD,gBAAc,gBAAgB,MAAM;AAEpC,SAAO,EAAE,KAAK;AAAA,IACb,KAAK;AAAA,IACL,UAAU,OAAO;AAAA,IACjB,MAAM,OAAO;AAAA,IACb,aAAa,OAAO,KAAK,SAAS,OAAO,GAAG,EAAE;AAAA,EAC/C,CAAC;AAED,SAAO;AACR;AAYA,eAAsB,UACrB,GACA,UAC8B;AAC9B,KAAG,cAAc;AAGjB,QAAM,oBAAyB,CAAC;AAChC,QAAM,qBAAqB,CAAC,QAAa;AACxC,sBAAkB,UAAU;AAAA,EAC7B;AAKA,QAAM,OAAO,MAAM,QAAQ;AAC3B,QAAM,SAAS,MAAM,UAAU,EAAE,KAAK,CAAC;AACvC,IAAE;AAAA,IACD,YAAY,MAAM,IAAI,QAAQ,CAAC,YAAY,OAAO,MAAM,MAAM,QAAQ,CAAC,CAAC;AAAA,EACzE;AAEA,QAAM;AAeP;AAEA,eAAsB,UAA2B;AAEhD,QAAM,WAAW;AACjB,QAAM,WAAW;AACjB,QAAM,gBAAgB,MACrB,KAAK,MAAM,KAAK,OAAO,KAAK,WAAW,WAAW,EAAE,IAAI;AAEzD,MAAI,OAAO,cAAc;AACzB,MAAI,cAAc;AAElB,SAAO,cAAc,GAAG;AACvB,QAAI;AAEH,YAAM,SAAS,MAAM,IAAI,QAAa,CAAC,SAAS,WAAW;AAC1D,cAAMA,UAAS,aAAa;AAE5B,QAAAA,QAAO,KAAK,SAAS,CAAC,QAAmC;AACxD,cAAI,IAAI,SAAS,cAAc;AAC9B,mBAAO,IAAI,MAAM,QAAQ,IAAI,YAAY,CAAC;AAAA,UAC3C,OAAO;AACN,mBAAO,GAAG;AAAA,UACX;AAAA,QACD,CAAC;AAED,QAAAA,QAAO,KAAK,aAAa,MAAM;AAC9B,kBAAQA,OAAM;AAAA,QACf,CAAC;AAED,QAAAA,QAAO,OAAO,IAAI;AAAA,MACnB,CAAC;AAGD,YAAM,IAAI,QAAc,CAAC,YAAY;AACpC,eAAO,MAAM,MAAM,QAAQ,CAAC;AAAA,MAC7B,CAAC;AAED,aAAO;AAAA,IACR,SAAS,KAAK;AAEb;AACA,UAAI,eAAe,GAAG;AACrB;AAAA,MACD;AACA,aAAO,cAAc;AAAA,IACtB;AAAA,EACD;AAEA,QAAM,IAAI,MAAM,0DAA0D;AAC3E;","names":["server"]}