rivetkit 2.1.5 → 2.1.6-rc.1

package/dist/tsup/{chunk-ZFY5J2EP.cjs → chunk-ZZLJ5TSM.cjs}

@@ -2,7 +2,7 @@
 
 
 
-var _chunkSR3KQE7Qcjs = require('./chunk-SR3KQE7Q.cjs');
+var _chunkSQFCIDCGcjs = require('./chunk-SQFCIDCG.cjs');
 
 
 
@@ -96,15 +96,15 @@ function defineConfig(config) {
 }
 function createProxyCallback(waDb, mutex, isClosed) {
   return async (sql, params, method) => {
-    return mutex.run(async () => {
+    return await mutex.run(async () => {
       if (isClosed()) {
         throw new Error("database is closed");
       }
       if (method === "run") {
-        await waDb.run(sql, _chunkSR3KQE7Qcjs.toSqliteBindings.call(void 0, params));
+        await waDb.run(sql, _chunkSQFCIDCGcjs.toSqliteBindings.call(void 0, params));
         return { rows: [] };
       }
-      const result = await waDb.query(sql, _chunkSR3KQE7Qcjs.toSqliteBindings.call(void 0, params));
+      const result = await waDb.query(sql, _chunkSQFCIDCGcjs.toSqliteBindings.call(void 0, params));
       if (method === "get") {
         return { rows: result.rows[0] };
       }
@@ -112,24 +112,20 @@ function createProxyCallback(waDb, mutex, isClosed) {
     });
   };
 }
-async function runInlineMigrations(waDb, mutex, migrations) {
-  await mutex.run(
-    () => waDb.exec(`
+async function runInlineMigrations(waDb, migrations) {
+  await waDb.exec(`
 		CREATE TABLE IF NOT EXISTS __drizzle_migrations (
 			id INTEGER PRIMARY KEY AUTOINCREMENT,
 			hash TEXT NOT NULL,
 			created_at INTEGER
 		)
-	`)
-  );
+	`);
   let lastCreatedAt = 0;
-  await mutex.run(
-    () => waDb.exec(
-      "SELECT id, hash, created_at FROM __drizzle_migrations ORDER BY created_at DESC LIMIT 1",
-      (row) => {
-        lastCreatedAt = Number(row[2]) || 0;
-      }
-    )
+  await waDb.exec(
+    "SELECT id, hash, created_at FROM __drizzle_migrations ORDER BY created_at DESC LIMIT 1",
+    (row) => {
+      lastCreatedAt = Number(row[2]) || 0;
+    }
   );
   const journal = migrations.journal;
   if (!(journal == null ? void 0 : journal.entries)) return;
@@ -138,18 +134,15 @@ async function runInlineMigrations(waDb, mutex, migrations) {
     const migrationKey = `m${String(entry.idx).padStart(4, "0")}`;
     const sql = migrations.migrations[migrationKey];
     if (!sql) continue;
-    await mutex.run(() => waDb.exec(sql));
-    await mutex.run(
-      () => waDb.run(
-        "INSERT INTO __drizzle_migrations (hash, created_at) VALUES (?, ?)",
-        [entry.tag, entry.when]
-      )
+    await waDb.exec(sql);
+    await waDb.run(
+      "INSERT INTO __drizzle_migrations (hash, created_at) VALUES (?, ?)",
+      [entry.tag, entry.when]
     );
   }
 }
 function db(config) {
-  let waDbInstance = null;
-  const mutex = new (0, _chunkSR3KQE7Qcjs.AsyncMutex)();
+  const clientToRawDb = /* @__PURE__ */ new WeakMap();
   return {
     createClient: async (ctx) => {
       if (!ctx.sqliteVfs) {
@@ -157,9 +150,9 @@ function db(config) {
           "SqliteVfs instance not provided in context. The driver must provide a sqliteVfs instance."
         );
       }
-      const kvStore = _chunkSR3KQE7Qcjs.createActorKvStore.call(void 0, ctx.kv);
+      const kvStore = _chunkSQFCIDCGcjs.createActorKvStore.call(void 0, ctx.kv, ctx.metrics);
       const waDb = await ctx.sqliteVfs.open(ctx.actorId, kvStore);
-      waDbInstance = waDb;
+      const mutex = new (0, _chunkSQFCIDCGcjs.AsyncMutex)();
       let closed = false;
       const ensureOpen = () => {
         if (closed) {
@@ -168,19 +161,19 @@ function db(config) {
       };
       const callback = createProxyCallback(waDb, mutex, () => closed);
       const client = _sqliteproxy.drizzle.call(void 0, callback, config);
-      return Object.assign(client, {
+      const result = Object.assign(client, {
         execute: async (query, ...args) => {
-          return mutex.run(async () => {
+          return await mutex.run(async () => {
             ensureOpen();
             if (args.length > 0) {
-              const result = await waDb.query(
+              const result2 = await waDb.query(
                 query,
-                _chunkSR3KQE7Qcjs.toSqliteBindings.call(void 0, args)
+                _chunkSQFCIDCGcjs.toSqliteBindings.call(void 0, args)
               );
-              return result.rows.map((row) => {
+              return result2.rows.map((row) => {
                 const obj = {};
-                for (let i = 0; i < result.columns.length; i++) {
-                  obj[result.columns[i]] = row[i];
+                for (let i = 0; i < result2.columns.length; i++) {
+                  obj[result2.columns[i]] = row[i];
                 }
                 return obj;
               });
@@ -204,22 +197,24 @@ function db(config) {
           });
         },
         close: async () => {
-          await mutex.run(async () => {
-            if (closed) {
-              return;
-            }
+          const shouldClose = await mutex.run(async () => {
+            if (closed) return false;
             closed = true;
-            await waDb.close();
-            waDbInstance = null;
+            return true;
           });
+          if (shouldClose) {
+            await waDb.close();
+          }
         }
       });
+      clientToRawDb.set(result, waDb);
+      return result;
     },
-    onMigrate: async (_client) => {
-      if ((config == null ? void 0 : config.migrations) && waDbInstance) {
+    onMigrate: async (client) => {
+      const waDb = clientToRawDb.get(client);
+      if ((config == null ? void 0 : config.migrations) && waDb) {
         await runInlineMigrations(
-          waDbInstance,
-          mutex,
+          waDb,
           config.migrations
         );
       }
@@ -254,4 +249,4 @@ function db(config) {
 
 
 exports.blob = _sqlitecore.blob; exports.check = _sqlitecore.check; exports.extractUsedTable = _sqlitecore.extractUsedTable; exports.foreignKey = _sqlitecore.foreignKey; exports.getTableConfig = _sqlitecore.getTableConfig; exports.getViewConfig = _sqlitecore.getViewConfig; exports.index = _sqlitecore.index; exports.integer = _sqlitecore.integer; exports.numeric = _sqlitecore.numeric; exports.primaryKey = _sqlitecore.primaryKey; exports.real = _sqlitecore.real; exports.sqliteTable = _sqlitecore.sqliteTable; exports.sqliteTableCreator = _sqlitecore.sqliteTableCreator; exports.sqliteView = _sqlitecore.sqliteView; exports.text = _sqlitecore.text; exports.unique = _sqlitecore.unique; exports.uniqueIndex = _sqlitecore.uniqueIndex; exports.uniqueKeyName = _sqlitecore.uniqueKeyName; exports.view = _sqlitecore.view; exports.defineConfig = defineConfig; exports.db = db; exports.mod_exports = mod_exports;
-//# sourceMappingURL=chunk-ZFY5J2EP.cjs.map
+//# sourceMappingURL=chunk-ZZLJ5TSM.cjs.map

package/dist/tsup/chunk-ZZLJ5TSM.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/chunk-ZZLJ5TSM.cjs","../../src/db/drizzle/mod.ts","../../src/db/drizzle/sqlite-core.ts"],"names":[],"mappings":"AAAA;AACE;AACA;AACA;AACF,wDAA6B;AAC7B;AACE;AACA;AACF,wDAA6B;AAC7B;AACA;ACVA,IAAA,YAAA,EAAA,CAAA,CAAA;AAAA,wCAAA,WAAA,EAAA;AAAA,EAAA,IAAA,EAAA,CAAA,EAAA,GAAA,gBAAA;AAAA,EAAA,KAAA,EAAA,CAAA,EAAA,GAAA,iBAAA;AAAA,EAAA,EAAA,EAAA,CAAA,EAAA,GAAA,EAAA;AAAA,EAAA,YAAA,EAAA,CAAA,EAAA,GAAA,YAAA;AAAA,EAAA,gBAAA,EAAA,CAAA,EAAA,GAAA,4BAAA;AAAA,EAAA,UAAA,EAAA,CAAA,EAAA,GAAA,sBAAA;AAAA,EAAA,cAAA,EAAA,CAAA,EAAA,GAAA,0BAAA;AAAA,EAAA,aAAA,EAAA,CAAA,EAAA,GAAA,yBAAA;AAAA,EAAA,KAAA,EAAA,CAAA,EAAA,GAAA,iBAAA;AAAA,EAAA,OAAA,EAAA,CAAA,EAAA,GAAA,mBAAA;AAAA,EAAA,OAAA,EAAA,CAAA,EAAA,GAAA,mBAAA;AAAA,EAAA,UAAA,EAAA,CAAA,EAAA,GAAA,sBAAA;AAAA,EAAA,IAAA,EAAA,CAAA,EAAA,GAAA,gBAAA;AAAA,EAAA,WAAA,EAAA,CAAA,EAAA,GAAA,uBAAA;AAAA,EAAA,kBAAA,EAAA,CAAA,EAAA,GAAA,8BAAA;AAAA,EAAA,UAAA,EAAA,CAAA,EAAA,GAAA,sBAAA;AAAA,EAAA,IAAA,EAAA,CAAA,EAAA,GAAA,gBAAA;AAAA,EAAA,MAAA,EAAA,CAAA,EAAA,GAAA,kBAAA;AAAA,EAAA,WAAA,EAAA,CAAA,EAAA,GAAA,uBAAA;AAAA,EAAA,aAAA,EAAA,CAAA,EAAA,GAAA,yBAAA;AAAA,EAAA,IAAA,EAAA,CAAA,EAAA,GAAA;AAAA,CAAA,CAAA;AACA;AACC;AAAW,uDAEL;ADkCP;AACA;AEvCA,IAAA,oBAAA,EAAA,CAAA,CAAA;AAAA,wCAAA,mBAAA,EAAA;AAAA,EAAA,IAAA,EAAA,CAAA,EAAA,GAAA,gBAAA;AAAA,EAAA,KAAA,EAAA,CAAA,EAAA,GAAA,iBAAA;AAAA,EAAA,gBAAA,EAAA,CAAA,EAAA,GAAA,4BAAA;AAAA,EAAA,UAAA,EAAA,CAAA,EAAA,GAAA,sBAAA;AAAA,EAAA,cAAA,EAAA,CAAA,EAAA,GAAA,0BAAA;AAAA,EAAA,aAAA,EAAA,CAAA,EAAA,GAAA,yBAAA;AAAA,EAAA,KAAA,EAAA,CAAA,EAAA,GAAA,iBAAA;AAAA,EAAA,OAAA,EAAA,CAAA,EAAA,GAAA,mBAAA;AAAA,EAAA,OAAA,EAAA,CAAA,EAAA,GAAA,mBAAA;AAAA,EAAA,UAAA,EAAA,CAAA,EAAA,GAAA,sBAAA;AAAA,EAAA,IAAA,EAAA,CAAA,EAAA,GAAA,gBAAA;AAAA,EAAA,WAAA,EAAA,CAAA,EAAA,GAAA,uBAAA;AAAA,EAAA,kBAAA,EAAA,CAAA,EAAA,GAAA,8BAAA;AAAA,EAAA,UAAA,EAAA,CAAA,EAAA,GAAA,sBAAA;AAAA,EAAA,IAAA,EAAA,CAAA,EAAA,GAAA,gBAAA;AAAA,EAAA,MAAA,EAAA,CAAA,EAAA,GAAA,kBAAA;AAAA,EAAA,WAAA,EAAA,CAAA,EAAA,GAAA,uBAAA;AAAA,EAAA,aAAA,EAAA,CAAA,EAAA,GAAA,yBAAA;AAAA,EAAA,IAAA,EAAA,CAAA,EAAA,GAAA;AAAA,CAAA,CAAA;AAAA,0CAAA,mBAAA,EAAA,gBAAA,CAAA;AAAA,kHAAc;AACd;AACC;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AAAA;AFiED;AACA;AC9EA,0CAAA,WAAA,EAAc,mBAAA,CAAA;AAEd,yCAAkE;AAE3D,SAAS,YAAA,CACf,MAAA,EACS;AACT,EAAA,OAAO,sCAAA;AAAqB,IAC3B,OAAA,EAAS,QAAA;AAAA,IACT,MAAA,EAAQ,gBAAA;AAAA,IACR,GAAG;AAAA,EACJ,CAAC,CAAA;AACF;AAYA,SAAS,mBAAA,CACR,IAAA,EACA,KAAA,EACA,QAAA,EACC;AACD,EAAA,OAAO,MAAA,CACN,GAAA,EACA,MAAA,EACA,MAAA,EAAA,GAC4B;AAC5B,IAAA,OAAO,MAAM,KAAA,CAAM,GAAA,CAAI,MAAA,CAAA,EAAA,GAAY;AAClC,MAAA,GAAA,CAAI,QAAA,CAAS,CAAA,EAAG;AACf,QAAA,MAAM,IAAI,KAAA,CAAM,oBAAoB,CAAA;AAAA,MACrC;AAEA,MAAA,GAAA,CAAI,OAAA,IAAW,KAAA,EAAO;AACrB,QAAA,MAAM,IAAA,CAAK,GAAA,CAAI,GAAA,EAAK,gDAAA,MAAuB,CAAC,CAAA;AAC5C,QAAA,OAAO,EAAE,IAAA,EAAM,CAAC,EAAE,CAAA;AAAA,MACnB;AAGA,MAAA,MAAM,OAAA,EAAS,MAAM,IAAA,CAAK,KAAA,CAAM,GAAA,EAAK,gDAAA,MAAuB,CAAC,CAAA;AAI7D,MAAA,GAAA,CAAI,OAAA,IAAW,KAAA,EAAO;AACrB,QAAA,OAAO,EAAE,IAAA,EAAM,MAAA,CAAO,IAAA,CAAK,CAAC,EAAE,CAAA;AAAA,MAC/B;AAEA,MAAA,OAAO,EAAE,IAAA,EAAM,MAAA,CAAO,KAAK,CAAA;AAAA,IAC5B,CAAC,CAAA;AAAA,EACF,CAAA;AACD;AAMA,MAAA,SAAe,mBAAA,CACd,IAAA,EACA,UAAA,EACgB;AAEhB,EAAA,MAAM,IAAA,CAAK,IAAA,CAAK,CAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,CAAA,CAMf,CAAA;AAGD,EAAA,IAAI,cAAA,EAAgB,CAAA;AACpB,EAAA,MAAM,IAAA,CAAK,IAAA;AAAA,IACV,wFAAA;AAAA,IACA,CAAC,GAAA,EAAA,GAAQ;AACR,MAAA,cAAA,EAAgB,MAAA,CAAO,GAAA,CAAI,CAAC,CAAC,EAAA,GAAK,CAAA;AAAA,IACnC;AAAA,EACD,CAAA;AAGA,EAAA,MAAM,QAAA,EAAU,UAAA,CAAW,OAAA;AAC3B,EAAA,GAAA,CAAI,CAAA,CAAC,QAAA,GAAA,KAAA,EAAA,KAAA,EAAA,EAAA,OAAA,CAAS,OAAA,CAAA,EAAS,MAAA;AAEvB,EAAA,IAAA,CAAA,MAAW,MAAA,GAAS,OAAA,CAAQ,OAAA,EAAS;AACpC,IAAA,GAAA,CAAI,KAAA,CAAM,KAAA,GAAQ,aAAA,EAAe,QAAA;AAIjC,IAAA,MAAM,aAAA,EAAe,CAAA,CAAA,EAAI,MAAA,CAAO,KAAA,CAAM,GAAG,CAAA,CAAE,QAAA,CAAS,CAAA,EAAG,GAAG,CAAC,CAAA,CAAA;AACb,IAAA;AACpC,IAAA;AAGS,IAAA;AAGR,IAAA;AACV,MAAA;AACsB,MAAA;AACvB,IAAA;AACD,EAAA;AACD;AAM+D;AAOT,EAAA;AAE9C,EAAA;AACuB,IAAA;AAER,MAAA;AACT,QAAA;AACT,UAAA;AACD,QAAA;AACD,MAAA;AAEsD,MAAA;AACI,MAAA;AAI7B,MAAA;AAChB,MAAA;AACY,MAAA;AACZ,QAAA;AACyB,UAAA;AACrC,QAAA;AACD,MAAA;AAGwD,MAAA;AAGH,MAAA;AAEhB,MAAA;AASd,QAAA;AACc,UAAA;AACvB,YAAA;AAEU,YAAA;AACM,cAAA;AACzB,gBAAA;AACqB,gBAAA;AACtB,cAAA;AAC2C,cAAA;AACJ,gBAAA;AAIrC,gBAAA;AAE8B,kBAAA;AAC/B,gBAAA;AACO,gBAAA;AACP,cAAA;AACF,YAAA;AAG4C,YAAA;AACT,YAAA;AACxB,YAAA;AACV,cAAA;AACuC,cAAA;AACpB,gBAAA;AACH,kBAAA;AACf,gBAAA;AACsC,gBAAA;AACD,gBAAA;AACT,kBAAA;AAC5B,gBAAA;AACgB,gBAAA;AACjB,cAAA;AACD,YAAA;AACO,YAAA;AACP,UAAA;AACF,QAAA;AACmB,QAAA;AAC8B,UAAA;AAC5B,YAAA;AACV,YAAA;AACF,YAAA;AACP,UAAA;AACgB,UAAA;AACC,YAAA;AAClB,UAAA;AACD,QAAA;AACoB,MAAA;AAES,MAAA;AACvB,MAAA;AACR,IAAA;AAC6B,IAAA;AACmB,MAAA;AACrB,MAAA;AACnB,QAAA;AACL,UAAA;AACO,UAAA;AACR,QAAA;AACD,MAAA;AACD,IAAA;AAC6B,IAAA;AACT,MAAA;AACpB,IAAA;AACD,EAAA;AACD;ADTgE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA","file":"/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/chunk-ZZLJ5TSM.cjs","sourcesContent":[null,"import type { IDatabase } from \"@rivetkit/sqlite-vfs\";\nimport {\n\tdrizzle as proxyDrizzle,\n\ttype SqliteRemoteDatabase,\n} from \"drizzle-orm/sqlite-proxy\";\nimport type { DatabaseProvider, RawAccess } from \"../config\";\nimport { AsyncMutex, createActorKvStore, toSqliteBindings } from \"../shared\";\n\nexport * from \"./sqlite-core\";\n\nimport { type Config, defineConfig as originalDefineConfig } from \"drizzle-kit\";\n\nexport function defineConfig(\n\tconfig: Partial<Config & { driver: \"durable-sqlite\" }>,\n): Config {\n\treturn originalDefineConfig({\n\t\tdialect: \"sqlite\",\n\t\tdriver: \"durable-sqlite\",\n\t\t...config,\n\t});\n}\n\ninterface DatabaseFactoryConfig<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n> {\n\tschema?: TSchema;\n\tmigrations?: any;\n}\n\n/**\n * Create a sqlite-proxy async callback from a @rivetkit/sqlite Database\n */\nfunction createProxyCallback(\n\twaDb: IDatabase,\n\tmutex: AsyncMutex,\n\tisClosed: () => boolean,\n) {\n\treturn async (\n\t\tsql: string,\n\t\tparams: any[],\n\t\tmethod: \"run\" | \"all\" | \"values\" | \"get\",\n\t): Promise<{ rows: any }> => {\n\t\treturn await mutex.run(async () => {\n\t\t\tif (isClosed()) {\n\t\t\t\tthrow new Error(\"database is closed\");\n\t\t\t}\n\n\t\t\tif (method === \"run\") {\n\t\t\t\tawait waDb.run(sql, toSqliteBindings(params));\n\t\t\t\treturn { rows: [] };\n\t\t\t}\n\n\t\t\t// For all/get/values, use parameterized query\n\t\t\tconst result = await waDb.query(sql, toSqliteBindings(params));\n\n\t\t\t// drizzle's mapResultRow accesses rows by column index (positional arrays)\n\t\t\t// so we return raw arrays for all methods\n\t\t\tif (method === \"get\") {\n\t\t\t\treturn { rows: result.rows[0] };\n\t\t\t}\n\n\t\t\treturn { rows: result.rows };\n\t\t});\n\t};\n}\n\n/**\n * Run inline migrations via the @rivetkit/sqlite Database.\n * Migrations use the same embedded format as drizzle-orm's durable-sqlite.\n */\nasync function runInlineMigrations(\n\twaDb: IDatabase,\n\tmigrations: any,\n): Promise<void> {\n\t// Create migrations table\n\tawait waDb.exec(`\n\t\tCREATE TABLE IF NOT EXISTS __drizzle_migrations (\n\t\t\tid INTEGER PRIMARY KEY AUTOINCREMENT,\n\t\t\thash TEXT NOT NULL,\n\t\t\tcreated_at INTEGER\n\t\t)\n\t`);\n\n\t// Get the last applied migration\n\tlet lastCreatedAt = 0;\n\tawait waDb.exec(\n\t\t\"SELECT id, hash, created_at FROM __drizzle_migrations ORDER BY created_at DESC LIMIT 1\",\n\t\t(row) => {\n\t\t\tlastCreatedAt = Number(row[2]) || 0;\n\t\t},\n\t);\n\n\t// Apply pending migrations from journal entries\n\tconst journal = migrations.journal;\n\tif (!journal?.entries) return;\n\n\tfor (const entry of journal.entries) {\n\t\tif (entry.when <= lastCreatedAt) continue;\n\n\t\t// Find the migration SQL from the migrations map\n\t\t// The key format is \"m\" + zero-padded index (e.g. \"m0000\")\n\t\tconst migrationKey = `m${String(entry.idx).padStart(4, \"0\")}`;\n\t\tconst sql = migrations.migrations[migrationKey];\n\t\tif (!sql) continue;\n\n\t\t// Execute migration SQL\n\t\tawait waDb.exec(sql);\n\n\t\t// Record migration\n\t\tawait waDb.run(\n\t\t\t\"INSERT INTO __drizzle_migrations (hash, created_at) VALUES (?, ?)\",\n\t\t\t[entry.tag, entry.when],\n\t\t);\n\t}\n}\n\nexport function db<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n>(\n\tconfig?: DatabaseFactoryConfig<TSchema>,\n): DatabaseProvider<SqliteRemoteDatabase<TSchema> & RawAccess> {\n\t// Map from drizzle client to the underlying @rivetkit/sqlite Database.\n\t// Uses WeakMap so entries are garbage-collected when the client is.\n\t// This replaces the previous shared `waDbInstance` variable which caused\n\t// a race when multiple actors of the same type created databases\n\t// concurrently: the last writer won, and earlier actors' migrations\n\t// ran on the wrong database.\n\tconst clientToRawDb = new WeakMap<object, IDatabase>();\n\n\treturn {\n\t\tcreateClient: async (ctx) => {\n\t\t\t// Construct KV-backed client using actor driver's KV operations\n\t\t\tif (!ctx.sqliteVfs) {\n\t\t\t\tthrow new Error(\n\t\t\t\t\t\"SqliteVfs instance not provided in context. The driver must provide a sqliteVfs instance.\",\n\t\t\t\t);\n\t\t\t}\n\n\t\t\tconst kvStore = createActorKvStore(ctx.kv, ctx.metrics);\n\t\t\tconst waDb = await ctx.sqliteVfs.open(ctx.actorId, kvStore);\n\t\t\t// Per-client mutex so actors of the same type do not serialize\n\t\t\t// against each other. Each actor has its own database handle and\n\t\t\t// its own closed flag, so there is no shared state to guard.\n\t\t\tconst mutex = new AsyncMutex();\n\t\t\tlet closed = false;\n\t\t\tconst ensureOpen = () => {\n\t\t\t\tif (closed) {\n\t\t\t\t\tthrow new Error(\"database is closed\");\n\t\t\t\t}\n\t\t\t};\n\n\t\t\t// Create the async proxy callback\n\t\t\tconst callback = createProxyCallback(waDb, mutex, () => closed);\n\n\t\t\t// Create the drizzle instance using sqlite-proxy\n\t\t\tconst client = proxyDrizzle<TSchema>(callback, config);\n\n\t\t\tconst result = Object.assign(client, {\n\t\t\t\texecute: async <\n\t\t\t\t\tTRow extends Record<string, unknown> = Record<\n\t\t\t\t\t\tstring,\n\t\t\t\t\t\tunknown\n\t\t\t\t\t>,\n\t\t\t\t>(\n\t\t\t\t\tquery: string,\n\t\t\t\t\t...args: unknown[]\n\t\t\t\t): Promise<TRow[]> => {\n\t\t\t\t\treturn await mutex.run(async () => {\n\t\t\t\t\t\tensureOpen();\n\n\t\t\t\t\t\tif (args.length > 0) {\n\t\t\t\t\t\t\tconst result = await waDb.query(\n\t\t\t\t\t\t\t\tquery,\n\t\t\t\t\t\t\t\ttoSqliteBindings(args),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn result.rows.map((row: unknown[]) => {\n\t\t\t\t\t\t\t\tconst obj: Record<string, unknown> = {};\n\t\t\t\t\t\t\t\tfor (\n\t\t\t\t\t\t\t\t\tlet i = 0;\n\t\t\t\t\t\t\t\t\ti < result.columns.length;\n\t\t\t\t\t\t\t\t\ti++\n\t\t\t\t\t\t\t\t) {\n\t\t\t\t\t\t\t\t\tobj[result.columns[i]] = row[i];\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\treturn obj;\n\t\t\t\t\t\t\t}) as TRow[];\n\t\t\t\t\t\t}\n\t\t\t\t\t\t// Use exec for non-parameterized queries since\n\t\t\t\t\t\t// @rivetkit/sqlite's query() can crash on some statements.\n\t\t\t\t\t\tconst results: Record<string, unknown>[] = [];\n\t\t\t\t\t\tlet columnNames: string[] | null = null;\n\t\t\t\t\t\tawait waDb.exec(\n\t\t\t\t\t\t\tquery,\n\t\t\t\t\t\t\t(row: unknown[], columns: string[]) => {\n\t\t\t\t\t\t\t\tif (!columnNames) {\n\t\t\t\t\t\t\t\t\tcolumnNames = columns;\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\tconst obj: Record<string, unknown> = {};\n\t\t\t\t\t\t\t\tfor (let i = 0; i < row.length; i++) {\n\t\t\t\t\t\t\t\t\tobj[columnNames[i]] = row[i];\n\t\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\t\tresults.push(obj);\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t);\n\t\t\t\t\t\treturn results as TRow[];\n\t\t\t\t\t});\n\t\t\t\t},\n\t\t\t\tclose: async () => {\n\t\t\t\t\tconst shouldClose = await mutex.run(async () => {\n\t\t\t\t\t\tif (closed) return false;\n\t\t\t\t\t\tclosed = true;\n\t\t\t\t\t\treturn true;\n\t\t\t\t\t});\n\t\t\t\t\tif (shouldClose) {\n\t\t\t\t\t\tawait waDb.close();\n\t\t\t\t\t}\n\t\t\t\t},\n\t\t\t} satisfies RawAccess);\n\n\t\t\tclientToRawDb.set(result, waDb);\n\t\t\treturn result;\n\t\t},\n\t\tonMigrate: async (client) => {\n\t\t\tconst waDb = clientToRawDb.get(client as object);\n\t\t\tif (config?.migrations && waDb) {\n\t\t\t\tawait runInlineMigrations(\n\t\t\t\t\twaDb,\n\t\t\t\t\tconfig.migrations,\n\t\t\t\t);\n\t\t\t}\n\t\t},\n\t\tonDestroy: async (client) => {\n\t\t\tawait client.close();\n\t\t},\n\t};\n}\n","export * from \"drizzle-orm/sqlite-core\";\nexport {\n\tblob,\n\tcheck,\n\textractUsedTable,\n\tforeignKey,\n\tgetTableConfig,\n\tgetViewConfig,\n\tindex,\n\tinteger,\n\tnumeric,\n\tprimaryKey,\n\treal,\n\tsqliteTable,\n\tsqliteTableCreator,\n\tsqliteView,\n\ttext,\n\tunique,\n\tuniqueIndex,\n\tuniqueKeyName,\n\tview,\n} from \"drizzle-orm/sqlite-core\";\n"]}

package/dist/tsup/client/mod.cjs

@@ -1,5 +1,6 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
+var _chunkYAE3MEJMcjs = require('../chunk-YAE3MEJM.cjs');
 
 
 
@@ -8,15 +9,17 @@
 
 
 
-var _chunkIIJNPVPQcjs = require('../chunk-IIJNPVPQ.cjs');
-require('../chunk-JC6BEPE7.cjs');
-require('../chunk-TI5PXQGG.cjs');
+
+var _chunkGUHXWPGBcjs = require('../chunk-GUHXWPGB.cjs');
+require('../chunk-V3JSZR5P.cjs');
+require('../chunk-CGGGBIDP.cjs');
+require('../chunk-HNE2AK6C.cjs');
 require('../chunk-NIYZDWMW.cjs');
 
 
-var _chunkPB5AEMKQcjs = require('../chunk-PB5AEMKQ.cjs');
-require('../chunk-HYPIHCDT.cjs');
-require('../chunk-L47L3ZWJ.cjs');
+var _chunkA4KEUCB6cjs = require('../chunk-A4KEUCB6.cjs');
+require('../chunk-MPLMTJY5.cjs');
+require('../chunk-6G7ZNM27.cjs');
 require('../chunk-HDQ2JUQT.cjs');
 
 
@@ -29,5 +32,5 @@ require('../chunk-HDQ2JUQT.cjs');
 
 
 
-exports.ActorClientError = _chunkIIJNPVPQcjs.ActorClientError; exports.ActorConnDisposed = _chunkIIJNPVPQcjs.ActorConnDisposed; exports.ActorConnRaw = _chunkIIJNPVPQcjs.ActorConnRaw; exports.ActorError = _chunkIIJNPVPQcjs.ActorError; exports.ActorHandleRaw = _chunkIIJNPVPQcjs.ActorHandleRaw; exports.InternalError = _chunkIIJNPVPQcjs.InternalError; exports.KV_KEYS = _chunkPB5AEMKQcjs.KEYS; exports.MalformedResponseMessage = _chunkIIJNPVPQcjs.MalformedResponseMessage; exports.ManagerError = _chunkIIJNPVPQcjs.ManagerError; exports.createClient = _chunkIIJNPVPQcjs.createClient;
+exports.ActorClientError = _chunkGUHXWPGBcjs.ActorClientError; exports.ActorConnDisposed = _chunkGUHXWPGBcjs.ActorConnDisposed; exports.ActorConnRaw = _chunkGUHXWPGBcjs.ActorConnRaw; exports.ActorError = _chunkGUHXWPGBcjs.ActorError; exports.ActorHandleRaw = _chunkGUHXWPGBcjs.ActorHandleRaw; exports.InternalError = _chunkGUHXWPGBcjs.InternalError; exports.KV_KEYS = _chunkA4KEUCB6cjs.KEYS; exports.MalformedResponseMessage = _chunkGUHXWPGBcjs.MalformedResponseMessage; exports.ManagerError = _chunkGUHXWPGBcjs.ManagerError; exports.createClient = _chunkYAE3MEJMcjs.createClient;
 //# sourceMappingURL=mod.cjs.map

package/dist/tsup/client/mod.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/client/mod.cjs"],"names":[],"mappings":"AAAA;AACE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACF,yDAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B;AACE;AACF,yDAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B;AACE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACF,okBAAC","file":"/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/client/mod.cjs"}
+{"version":3,"sources":["/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/client/mod.cjs"],"names":[],"mappings":"AAAA;AACE;AACF,yDAA8B;AAC9B;AACE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACF,yDAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B;AACE;AACF,yDAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B,iCAA8B;AAC9B;AACE;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACA;AACF,okBAAC","file":"/home/runner/work/rivet/rivet/rivetkit-typescript/packages/rivetkit/dist/tsup/client/mod.cjs"}

package/dist/tsup/client/mod.d.cts

@@ -1,18 +1,18 @@
-import { a5 as Registry, C as Client } from '../config-iPj5l1bL.cjs';
-export { aB as ActorAccessor, aC as ActorActionFunction, aD as ActorClientError, aE as ActorConn, aF as ActorConnDisposed, aG as ActorConnRaw, aH as ActorConnStatus, f as ActorDefinition, aI as ActorError, aJ as ActorHandle, aK as ActorHandleRaw, t as AnyActorDefinition, aL as ClientRaw, aM as ConnectionStateCallback, aN as CreateOptions, aO as CreateRequest, E as Encoding, aP as EventUnsubscribe, aQ as ExtractActorsFromRegistry, aR as ExtractRegistryFromClient, aS as GetOptions, aT as GetWithIdOptions, aU as InternalError, aV as MalformedResponseMessage, aW as ManagerError, aX as QueryOptions, aY as Region, aZ as StatusChangeCallback } from '../config-iPj5l1bL.cjs';
+import { R as Registry, C as Client } from '../config-ehT-_3BB.cjs';
+export { k as ActorAccessor, l as ActorActionFunction, m as ActorClientError, n as ActorConn, o as ActorConnDisposed, p as ActorConnRaw, q as ActorConnStatus, d as ActorDefinition, r as ActorError, s as ActorHandle, t as ActorHandleRaw, c as AnyActorDefinition, u as ClientRaw, v as ConnectionStateCallback, w as CreateOptions, x as CreateRequest, E as Encoding, y as EventUnsubscribe, z as ExtractActorsFromRegistry, B as ExtractRegistryFromClient, G as GetOptions, D as GetWithIdOptions, F as InternalError, H as MalformedResponseMessage, J as ManagerError, K as QueryOptions, L as Region, S as StatusChangeCallback } from '../config-ehT-_3BB.cjs';
 export { K as KV_KEYS } from '../keys-CydblqMh.cjs';
 import z from 'zod/v4';
 import { GetUpgradeWebSocket } from '../utils.cjs';
 import '@rivetkit/virtual-websocket';
-import '../config-BiNoIHRs.cjs';
+import '../config-DROwzBLT.cjs';
 import '@rivetkit/sqlite-vfs';
 import '@rivetkit/traces';
 import 'hono';
 import 'vbare';
-import 'pino';
+import '@standard-schema/spec';
 import 'drizzle-orm/sqlite-core';
 import 'zod';
-import '@standard-schema/spec';
+import 'pino';
 import 'nanoevents';
 import '../v3-DnYObHH3.cjs';
 import '@rivetkit/bare-ts';
@@ -61,4 +61,4 @@ type ClientConfigInput = z.input<typeof ClientConfigSchema>;
  */
 declare function createClient<A extends Registry<any>>(endpointOrConfig?: string | ClientConfigInput): Client<A>;
 
-export { Client, createClient };
+export { Client, type ClientConfigInput, createClient };

package/dist/tsup/client/mod.d.ts

@@ -1,18 +1,18 @@
-import { a5 as Registry, C as Client } from '../config-Qj-zLJPc.js';
-export { aB as ActorAccessor, aC as ActorActionFunction, aD as ActorClientError, aE as ActorConn, aF as ActorConnDisposed, aG as ActorConnRaw, aH as ActorConnStatus, f as ActorDefinition, aI as ActorError, aJ as ActorHandle, aK as ActorHandleRaw, t as AnyActorDefinition, aL as ClientRaw, aM as ConnectionStateCallback, aN as CreateOptions, aO as CreateRequest, E as Encoding, aP as EventUnsubscribe, aQ as ExtractActorsFromRegistry, aR as ExtractRegistryFromClient, aS as GetOptions, aT as GetWithIdOptions, aU as InternalError, aV as MalformedResponseMessage, aW as ManagerError, aX as QueryOptions, aY as Region, aZ as StatusChangeCallback } from '../config-Qj-zLJPc.js';
+import { R as Registry, C as Client } from '../config-C2Wwnc69.js';
+export { k as ActorAccessor, l as ActorActionFunction, m as ActorClientError, n as ActorConn, o as ActorConnDisposed, p as ActorConnRaw, q as ActorConnStatus, d as ActorDefinition, r as ActorError, s as ActorHandle, t as ActorHandleRaw, c as AnyActorDefinition, u as ClientRaw, v as ConnectionStateCallback, w as CreateOptions, x as CreateRequest, E as Encoding, y as EventUnsubscribe, z as ExtractActorsFromRegistry, B as ExtractRegistryFromClient, G as GetOptions, D as GetWithIdOptions, F as InternalError, H as MalformedResponseMessage, J as ManagerError, K as QueryOptions, L as Region, S as StatusChangeCallback } from '../config-C2Wwnc69.js';
 export { K as KV_KEYS } from '../keys-CydblqMh.js';
 import z from 'zod/v4';
 import { GetUpgradeWebSocket } from '../utils.js';
 import '@rivetkit/virtual-websocket';
-import '../config-BiNoIHRs.js';
+import '../config-DROwzBLT.js';
 import '@rivetkit/sqlite-vfs';
 import '@rivetkit/traces';
 import 'hono';
 import 'vbare';
-import 'pino';
+import '@standard-schema/spec';
 import 'drizzle-orm/sqlite-core';
 import 'zod';
-import '@standard-schema/spec';
+import 'pino';
 import 'nanoevents';
 import '../v3-DnYObHH3.js';
 import '@rivetkit/bare-ts';
@@ -61,4 +61,4 @@ type ClientConfigInput = z.input<typeof ClientConfigSchema>;
  */
 declare function createClient<A extends Registry<any>>(endpointOrConfig?: string | ClientConfigInput): Client<A>;
 
-export { Client, createClient };
+export { Client, type ClientConfigInput, createClient };

package/dist/tsup/client/mod.js

@@ -1,3 +1,6 @@
+import {
+  createClient
+} from "../chunk-T5KYKM6R.js";
 import {
   ActorClientError,
   ActorConnDisposed,
@@ -6,17 +9,17 @@ import {
   ActorHandleRaw,
   InternalError,
   MalformedResponseMessage,
-  ManagerError,
-  createClient
-} from "../chunk-JPXO2H55.js";
-import "../chunk-2OK7S6QF.js";
-import "../chunk-MIX2KB6U.js";
+  ManagerError
+} from "../chunk-WW27B6DM.js";
+import "../chunk-I5I6OALK.js";
+import "../chunk-C4EB42ET.js";
+import "../chunk-PQWI44WD.js";
 import "../chunk-LXUQ667X.js";
 import {
   KEYS
-} from "../chunk-TADUYCHF.js";
-import "../chunk-EIATSBYZ.js";
-import "../chunk-7HTNH26M.js";
+} from "../chunk-C22JYHVT.js";
+import "../chunk-VWYO36X4.js";
+import "../chunk-AKUJ5OTO.js";
 import "../chunk-424PT5DM.js";
 export {
   ActorClientError,

package/dist/tsup/common/log.cjs

@@ -6,8 +6,8 @@
 
 
 
-var _chunkHYPIHCDTcjs = require('../chunk-HYPIHCDT.cjs');
-require('../chunk-L47L3ZWJ.cjs');
+var _chunkMPLMTJY5cjs = require('../chunk-MPLMTJY5.cjs');
+require('../chunk-6G7ZNM27.cjs');
 require('../chunk-HDQ2JUQT.cjs');
 
 
@@ -17,5 +17,5 @@ require('../chunk-HDQ2JUQT.cjs');
 
 
 
-exports.LogLevelSchema = _chunkHYPIHCDTcjs.LogLevelSchema; exports.configureBaseLogger = _chunkHYPIHCDTcjs.configureBaseLogger; exports.configureDefaultLogger = _chunkHYPIHCDTcjs.configureDefaultLogger; exports.getBaseLogger = _chunkHYPIHCDTcjs.getBaseLogger; exports.getIncludeTarget = _chunkHYPIHCDTcjs.getIncludeTarget; exports.getLogger = _chunkHYPIHCDTcjs.getLogger; exports.getPinoLevel = _chunkHYPIHCDTcjs.getPinoLevel;
+exports.LogLevelSchema = _chunkMPLMTJY5cjs.LogLevelSchema; exports.configureBaseLogger = _chunkMPLMTJY5cjs.configureBaseLogger; exports.configureDefaultLogger = _chunkMPLMTJY5cjs.configureDefaultLogger; exports.getBaseLogger = _chunkMPLMTJY5cjs.getBaseLogger; exports.getIncludeTarget = _chunkMPLMTJY5cjs.getIncludeTarget; exports.getLogger = _chunkMPLMTJY5cjs.getLogger; exports.getPinoLevel = _chunkMPLMTJY5cjs.getPinoLevel;
 //# sourceMappingURL=log.cjs.map

package/dist/tsup/common/log.js

@@ -6,8 +6,8 @@ import {
   getIncludeTarget,
   getLogger,
   getPinoLevel
-} from "../chunk-EIATSBYZ.js";
-import "../chunk-7HTNH26M.js";
+} from "../chunk-VWYO36X4.js";
+import "../chunk-AKUJ5OTO.js";
 import "../chunk-424PT5DM.js";
 export {
   LogLevelSchema,

package/dist/tsup/common/websocket.cjs

@@ -1,10 +1,10 @@
 "use strict";Object.defineProperty(exports, "__esModule", {value: true});
 
-var _chunkJC6BEPE7cjs = require('../chunk-JC6BEPE7.cjs');
-require('../chunk-HYPIHCDT.cjs');
-require('../chunk-L47L3ZWJ.cjs');
+var _chunkV3JSZR5Pcjs = require('../chunk-V3JSZR5P.cjs');
+require('../chunk-MPLMTJY5.cjs');
+require('../chunk-6G7ZNM27.cjs');
 require('../chunk-HDQ2JUQT.cjs');
 
 
-exports.importWebSocket = _chunkJC6BEPE7cjs.importWebSocket;
+exports.importWebSocket = _chunkV3JSZR5Pcjs.importWebSocket;
 //# sourceMappingURL=websocket.cjs.map

package/dist/tsup/common/websocket.js

@@ -1,8 +1,8 @@
 import {
   importWebSocket
-} from "../chunk-2OK7S6QF.js";
-import "../chunk-EIATSBYZ.js";
-import "../chunk-7HTNH26M.js";
+} from "../chunk-I5I6OALK.js";
+import "../chunk-VWYO36X4.js";
+import "../chunk-AKUJ5OTO.js";
 import "../chunk-424PT5DM.js";
 export {
   importWebSocket