rivetkit 2.0.27 → 2.0.29-rc.1

package/dist/tsup/{chunk-G7YZSSWV.cjs → chunk-QT7PPLSB.cjs}

@@ -1,20 +1,20 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
 
-var _chunkXNKWXMITcjs = require('./chunk-XNKWXMIT.cjs');
+var _chunkMG2TWYPFcjs = require('./chunk-MG2TWYPF.cjs');
 
 
-var _chunk2TZH6VO6cjs = require('./chunk-2TZH6VO6.cjs');
+var _chunkSE37OCA2cjs = require('./chunk-SE37OCA2.cjs');
 
 // src/client/mod.ts
 function createClient(endpointOrConfig) {
   const configInput = endpointOrConfig === void 0 ? {} : typeof endpointOrConfig === "string" ? { endpoint: endpointOrConfig } : endpointOrConfig;
-  const config = _chunk2TZH6VO6cjs.ClientConfigSchema.parse(configInput);
-  const driver = new (0, _chunkXNKWXMITcjs.RemoteManagerDriver)(config);
-  return _chunkXNKWXMITcjs.createClientWithDriver.call(void 0, driver, config);
+  const config = _chunkSE37OCA2cjs.ClientConfigSchema.parse(configInput);
+  const driver = new (0, _chunkMG2TWYPFcjs.RemoteManagerDriver)(config);
+  return _chunkMG2TWYPFcjs.createClientWithDriver.call(void 0, driver, config);
 }
 
 
 
 exports.createClient = createClient;
-//# sourceMappingURL=chunk-G7YZSSWV.cjs.map
+//# sourceMappingURL=chunk-QT7PPLSB.cjs.map

package/dist/tsup/{chunk-G7YZSSWV.cjs.map → chunk-QT7PPLSB.cjs.map}

@@ -1 +1 @@
-{"version":3,"sources":["/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/chunk-G7YZSSWV.cjs","../../src/client/mod.ts"],"names":[],"mappings":"AAAA;AACE;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACmCO,SAAS,YAAA,CACf,gBAAA,EACY;AAEZ,EAAA,MAAM,YAAA,EACL,iBAAA,IAAqB,KAAA,EAAA,EAClB,CAAC,EAAA,EACD,OAAO,iBAAA,IAAqB,SAAA,EAC3B,EAAE,QAAA,EAAU,iBAAiB,EAAA,EAC7B,gBAAA;AACL,EAAA,MAAM,OAAA,EAAS,oCAAA,CAAmB,KAAA,CAAM,WAAW,CAAA;AAGnD,EAAA,MAAM,OAAA,EAAS,IAAI,0CAAA,CAAoB,MAAM,CAAA;AAC7C,EAAA,OAAO,sDAAA,MAA0B,EAAQ,MAAM,CAAA;AAChD;AD3CA;AACA;AACE;AACF,oCAAC","file":"/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/chunk-G7YZSSWV.cjs","sourcesContent":[null,"import type { Registry } from \"@/registry/mod\";\nimport { RemoteManagerDriver } from \"@/remote-manager-driver/mod\";\nimport {\n\ttype Client,\n\ttype ClientConfigInput,\n\tcreateClientWithDriver,\n} from \"./client\";\nimport { ClientConfigSchema } from \"./config\";\n\nexport {\n\tActorDefinition,\n\tAnyActorDefinition,\n} from \"@/actor/definition\";\nexport type { Encoding } from \"@/actor/protocol/serde\";\nexport {\n\tActorClientError,\n\tActorError,\n\tInternalError,\n\tMalformedResponseMessage,\n\tManagerError,\n} from \"@/client/errors\";\nexport type { CreateRequest } from \"@/manager/protocol/query\";\nexport type { ActorActionFunction } from \"./actor-common\";\nexport type { ActorConn, EventUnsubscribe } from \"./actor-conn\";\nexport { ActorConnRaw } from \"./actor-conn\";\nexport type { ActorHandle } from \"./actor-handle\";\nexport { ActorHandleRaw } from \"./actor-handle\";\nexport type {\n\tActorAccessor,\n\tClient,\n\tClientRaw,\n\tCreateOptions,\n\tExtractActorsFromRegistry,\n\tExtractRegistryFromClient,\n\tGetOptions,\n\tGetWithIdOptions,\n\tQueryOptions,\n\tRegion,\n} from \"./client\";\n\n/**\n * Creates a client with the actor accessor proxy.\n */\nexport function createClient<A extends Registry<any>>(\n\tendpointOrConfig?: string | ClientConfigInput,\n): Client<A> {\n\t// Parse config\n\tconst configInput =\n\t\tendpointOrConfig === undefined\n\t\t\t? {}\n\t\t\t: typeof endpointOrConfig === \"string\"\n\t\t\t\t? { endpoint: endpointOrConfig }\n\t\t\t\t: endpointOrConfig;\n\tconst config = ClientConfigSchema.parse(configInput);\n\n\t// Create client\n\tconst driver = new RemoteManagerDriver(config);\n\treturn createClientWithDriver<A>(driver, config);\n}\n"]}
+{"version":3,"sources":["/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/chunk-QT7PPLSB.cjs","../../src/client/mod.ts"],"names":[],"mappings":"AAAA;AACE;AACA;AACF,wDAA6B;AAC7B;AACE;AACF,wDAA6B;AAC7B;AACA;ACmCO,SAAS,YAAA,CACf,gBAAA,EACY;AAEZ,EAAA,MAAM,YAAA,EACL,iBAAA,IAAqB,KAAA,EAAA,EAClB,CAAC,EAAA,EACD,OAAO,iBAAA,IAAqB,SAAA,EAC3B,EAAE,QAAA,EAAU,iBAAiB,EAAA,EAC7B,gBAAA;AACL,EAAA,MAAM,OAAA,EAAS,oCAAA,CAAmB,KAAA,CAAM,WAAW,CAAA;AAGnD,EAAA,MAAM,OAAA,EAAS,IAAI,0CAAA,CAAoB,MAAM,CAAA;AAC7C,EAAA,OAAO,sDAAA,MAA0B,EAAQ,MAAM,CAAA;AAChD;AD3CA;AACA;AACE;AACF,oCAAC","file":"/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/chunk-QT7PPLSB.cjs","sourcesContent":[null,"import type { Registry } from \"@/registry/mod\";\nimport { RemoteManagerDriver } from \"@/remote-manager-driver/mod\";\nimport {\n\ttype Client,\n\ttype ClientConfigInput,\n\tcreateClientWithDriver,\n} from \"./client\";\nimport { ClientConfigSchema } from \"./config\";\n\nexport {\n\tActorDefinition,\n\tAnyActorDefinition,\n} from \"@/actor/definition\";\nexport type { Encoding } from \"@/actor/protocol/serde\";\nexport {\n\tActorClientError,\n\tActorError,\n\tInternalError,\n\tMalformedResponseMessage,\n\tManagerError,\n} from \"@/client/errors\";\nexport type { CreateRequest } from \"@/manager/protocol/query\";\nexport type { ActorActionFunction } from \"./actor-common\";\nexport type { ActorConn, EventUnsubscribe } from \"./actor-conn\";\nexport { ActorConnRaw } from \"./actor-conn\";\nexport type { ActorHandle } from \"./actor-handle\";\nexport { ActorHandleRaw } from \"./actor-handle\";\nexport type {\n\tActorAccessor,\n\tClient,\n\tClientRaw,\n\tCreateOptions,\n\tExtractActorsFromRegistry,\n\tExtractRegistryFromClient,\n\tGetOptions,\n\tGetWithIdOptions,\n\tQueryOptions,\n\tRegion,\n} from \"./client\";\n\n/**\n * Creates a client with the actor accessor proxy.\n */\nexport function createClient<A extends Registry<any>>(\n\tendpointOrConfig?: string | ClientConfigInput,\n): Client<A> {\n\t// Parse config\n\tconst configInput =\n\t\tendpointOrConfig === undefined\n\t\t\t? {}\n\t\t\t: typeof endpointOrConfig === \"string\"\n\t\t\t\t? { endpoint: endpointOrConfig }\n\t\t\t\t: endpointOrConfig;\n\tconst config = ClientConfigSchema.parse(configInput);\n\n\t// Create client\n\tconst driver = new RemoteManagerDriver(config);\n\treturn createClientWithDriver<A>(driver, config);\n}\n"]}

package/dist/tsup/{chunk-VRZNWBDK.js → chunk-SDJFBUFE.js}

@@ -1,33 +1,29 @@
 import {
   createFileSystemOrMemoryDriver,
   createManagerRouter
-} from "./chunk-XU4GGB6J.js";
+} from "./chunk-QKDCNAPW.js";
 import {
   configureInspectorAccessToken
-} from "./chunk-2NNICHGY.js";
+} from "./chunk-EVQJYFFP.js";
+import {
+  createClient
+} from "./chunk-5RN6O2RW.js";
 import {
   createClientWithDriver
-} from "./chunk-BBTOBXEO.js";
+} from "./chunk-CHPLLKNL.js";
 import {
   ClientConfigSchema,
   RunnerConfigSchema
-} from "./chunk-5YDKTW6Y.js";
+} from "./chunk-CGIRSKRW.js";
 import {
   getLogger
-} from "./chunk-KSQQU7NC.js";
+} from "./chunk-YJXB7BPR.js";
 
 // src/test/mod.ts
 import { createServer } from "net";
 import { serve as honoServe } from "@hono/node-server";
 import { createNodeWebSocket } from "@hono/node-ws";
-import { vi } from "vitest";
-
-// src/test/config.ts
-import { z } from "zod";
-var ConfigSchema = RunnerConfigSchema.removeDefault().extend({
-  hostname: z.string().optional().default(process.env.HOSTNAME ?? "127.0.0.1"),
-  port: z.number().optional().default(Number.parseInt(process.env.PORT ?? "8080"))
-}).default({});
+import invariant from "invariant";
 
 // src/test/log.ts
 function logger() {
@@ -35,56 +31,59 @@ function logger() {
 }
 
 // src/test/mod.ts
-async function serve(registry, inputConfig) {
-  inputConfig ??= {};
-  const config = ConfigSchema.parse(inputConfig);
+async function setupTest(c, registry) {
+  registry.config.test.enabled = true;
+  const driver = await createFileSystemOrMemoryDriver(
+    true,
+    `/tmp/rivetkit-test-${crypto.randomUUID()}`
+  );
   let upgradeWebSocket;
-  if (!config.getUpgradeWebSocket) {
-    config.getUpgradeWebSocket = () => upgradeWebSocket;
-  }
-  const runConfig = RunnerConfigSchema.parse(inputConfig);
-  const driver = inputConfig.driver ?? await createFileSystemOrMemoryDriver(false);
+  const config = RunnerConfigSchema.parse({
+    driver,
+    getUpgradeWebSocket: () => upgradeWebSocket,
+    inspector: {
+      enabled: true,
+      token: () => "token"
+    }
+  });
   const managerDriver = driver.manager(registry.config, config);
-  const client = createClientWithDriver(
+  const internalClient = createClientWithDriver(
     managerDriver,
     ClientConfigSchema.parse({})
   );
   configureInspectorAccessToken(config, managerDriver);
   const { router } = createManagerRouter(
     registry.config,
-    runConfig,
+    config,
     managerDriver,
     driver,
-    client
+    internalClient
   );
   const nodeWebSocket = createNodeWebSocket({ app: router });
   upgradeWebSocket = nodeWebSocket.upgradeWebSocket;
+  const port = await getPort();
   const server = honoServe({
     fetch: router.fetch,
-    hostname: config.hostname,
-    port: config.port
+    hostname: "127.0.0.1",
+    port
   });
+  invariant(
+    nodeWebSocket.injectWebSocket !== void 0,
+    "should have injectWebSocket"
+  );
   nodeWebSocket.injectWebSocket(server);
-  logger().info({
-    msg: "rivetkit started",
-    hostname: config.hostname,
-    port: config.port,
-    definitions: Object.keys(registry.config.use).length
+  const endpoint = `http://127.0.0.1:${port}`;
+  logger().info({ msg: "test server listening", port });
+  c.onTestFinished(async () => {
+    await new Promise((resolve) => server.close(() => resolve(void 0)));
   });
-  return server;
-}
-async function setupTest(c, registry) {
-  vi.useFakeTimers();
-  const mockDriverContext = {};
-  const setDriverContextFn = (ctx) => {
-    mockDriverContext.current = ctx;
-  };
-  const port = await getPort();
-  const server = await serve(registry, { port });
-  c.onTestFinished(
-    async () => await new Promise((resolve) => server.close(() => resolve()))
-  );
-  throw "TODO: Fix engine port";
+  const client = createClient({
+    endpoint,
+    namespace: "default",
+    runnerName: "default"
+  });
+  c.onTestFinished(async () => await client.dispose());
+  return { client };
 }
 async function getPort() {
   const MIN_PORT = 1e4;
@@ -127,4 +126,4 @@ export {
   setupTest,
   getPort
 };
-//# sourceMappingURL=chunk-VRZNWBDK.js.map
+//# sourceMappingURL=chunk-SDJFBUFE.js.map

package/dist/tsup/chunk-SDJFBUFE.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/test/mod.ts","../../src/test/log.ts"],"sourcesContent":["import { createServer } from \"node:net\";\nimport { serve as honoServe } from \"@hono/node-server\";\nimport { createNodeWebSocket } from \"@hono/node-ws\";\nimport invariant from \"invariant\";\nimport { type TestContext, vi } from \"vitest\";\nimport { type Client, createClient } from \"@/client/mod\";\nimport { createFileSystemOrMemoryDriver } from \"@/drivers/file-system/mod\";\nimport { configureInspectorAccessToken } from \"@/inspector/utils\";\nimport { createManagerRouter } from \"@/manager/router\";\nimport { createClientWithDriver, type Registry, type RunConfig } from \"@/mod\";\nimport { ClientConfigSchema } from \"@/client/config\";\nimport { RunnerConfigSchema } from \"@/registry/run-config\";\nimport { logger } from \"./log\";\n\nexport interface SetupTestResult<A extends Registry<any>> {\n\tclient: Client<A>;\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\t// Force enable test mode\n\tregistry.config.test.enabled = true;\n\n\t// Create driver\n\tconst driver = await createFileSystemOrMemoryDriver(\n\t\ttrue,\n\t\t`/tmp/rivetkit-test-${crypto.randomUUID()}`,\n\t);\n\n\t// Build driver config\n\t// biome-ignore lint/style/useConst: Assigned later\n\tlet upgradeWebSocket: any;\n\tconst config: RunConfig = RunnerConfigSchema.parse({\n\t\tdriver,\n\t\tgetUpgradeWebSocket: () => upgradeWebSocket!,\n\t\tinspector: {\n\t\t\tenabled: true,\n\t\t\ttoken: () => \"token\",\n\t\t},\n\t});\n\n\t// Create router\n\tconst managerDriver = driver.manager(registry.config, config);\n\tconst internalClient = createClientWithDriver(\n\t\tmanagerDriver,\n\t\tClientConfigSchema.parse({}),\n\t);\n\tconfigureInspectorAccessToken(config, managerDriver);\n\tconst { router } = createManagerRouter(\n\t\tregistry.config,\n\t\tconfig,\n\t\tmanagerDriver,\n\t\tdriver,\n\t\tinternalClient,\n\t);\n\n\t// Inject WebSocket\n\tconst nodeWebSocket = createNodeWebSocket({ app: router });\n\tupgradeWebSocket = nodeWebSocket.upgradeWebSocket;\n\n\t// Start server\n\tconst port = await getPort();\n\tconst server = honoServe({\n\t\tfetch: router.fetch,\n\t\thostname: \"127.0.0.1\",\n\t\tport,\n\t});\n\tinvariant(\n\t\tnodeWebSocket.injectWebSocket !== undefined,\n\t\t\"should have injectWebSocket\",\n\t);\n\tnodeWebSocket.injectWebSocket(server);\n\tconst endpoint = `http://127.0.0.1:${port}`;\n\n\tlogger().info({ msg: \"test server listening\", port });\n\n\t// Cleanup on test finish\n\tc.onTestFinished(async () => {\n\t\tawait new Promise((resolve) => server.close(() => resolve(undefined)));\n\t});\n\n\t// Create client\n\tconst client = createClient<A>({\n\t\tendpoint,\n\t\tnamespace: \"default\",\n\t\trunnerName: \"default\",\n\t});\n\tc.onTestFinished(async () => await client.dispose());\n\n\treturn { client };\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 { getLogger } from \"@/common/log\";\n\nexport function logger() {\n\treturn getLogger(\"test\");\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAAA,SAAS,oBAAoB;AAC7B,SAAS,SAAS,iBAAiB;AACnC,SAAS,2BAA2B;AACpC,OAAO,eAAe;;;ACDf,SAAS,SAAS;AACxB,SAAO,UAAU,MAAM;AACxB;;;ADeA,eAAsB,UACrB,GACA,UAC8B;AAE9B,WAAS,OAAO,KAAK,UAAU;AAG/B,QAAM,SAAS,MAAM;AAAA,IACpB;AAAA,IACA,sBAAsB,OAAO,WAAW,CAAC;AAAA,EAC1C;AAIA,MAAI;AACJ,QAAM,SAAoB,mBAAmB,MAAM;AAAA,IAClD;AAAA,IACA,qBAAqB,MAAM;AAAA,IAC3B,WAAW;AAAA,MACV,SAAS;AAAA,MACT,OAAO,MAAM;AAAA,IACd;AAAA,EACD,CAAC;AAGD,QAAM,gBAAgB,OAAO,QAAQ,SAAS,QAAQ,MAAM;AAC5D,QAAM,iBAAiB;AAAA,IACtB;AAAA,IACA,mBAAmB,MAAM,CAAC,CAAC;AAAA,EAC5B;AACA,gCAA8B,QAAQ,aAAa;AACnD,QAAM,EAAE,OAAO,IAAI;AAAA,IAClB,SAAS;AAAA,IACT;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACD;AAGA,QAAM,gBAAgB,oBAAoB,EAAE,KAAK,OAAO,CAAC;AACzD,qBAAmB,cAAc;AAGjC,QAAM,OAAO,MAAM,QAAQ;AAC3B,QAAM,SAAS,UAAU;AAAA,IACxB,OAAO,OAAO;AAAA,IACd,UAAU;AAAA,IACV;AAAA,EACD,CAAC;AACD;AAAA,IACC,cAAc,oBAAoB;AAAA,IAClC;AAAA,EACD;AACA,gBAAc,gBAAgB,MAAM;AACpC,QAAM,WAAW,oBAAoB,IAAI;AAEzC,SAAO,EAAE,KAAK,EAAE,KAAK,yBAAyB,KAAK,CAAC;AAGpD,IAAE,eAAe,YAAY;AAC5B,UAAM,IAAI,QAAQ,CAAC,YAAY,OAAO,MAAM,MAAM,QAAQ,MAAS,CAAC,CAAC;AAAA,EACtE,CAAC;AAGD,QAAM,SAAS,aAAgB;AAAA,IAC9B;AAAA,IACA,WAAW;AAAA,IACX,YAAY;AAAA,EACb,CAAC;AACD,IAAE,eAAe,YAAY,MAAM,OAAO,QAAQ,CAAC;AAEnD,SAAO,EAAE,OAAO;AACjB;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"]}