@rivetkit/db 0.9.2 → 0.9.5

package/dist/config-D99afWH0.d.cts

@@ -0,0 +1,25 @@
+type DatabaseProvider<DB extends RawAccess> = {
+    /**
+     * Creates a new database client for the actor.
+     * The result is passed to the actor context as `c.db`.
+     * @experimental
+     */
+    createClient: (ctx: {
+        getDatabase: () => Promise<string | unknown>;
+    }) => Promise<DB>;
+    /**
+     * Runs before the actor has started.
+     * Use this to run migrations or other setup tasks.
+     * @experimental
+     */
+    onMigrate: (client: DB) => void | Promise<void>;
+};
+type ExecuteFunction = (query: string, ...args: unknown[]) => Promise<unknown[]>;
+type RawAccess = {
+    /**
+     * Executes a raw SQL query.
+     */
+    execute: ExecuteFunction;
+};
+
+export type { DatabaseProvider as D, RawAccess as R };

package/dist/config-D99afWH0.d.ts

@@ -0,0 +1,25 @@
+type DatabaseProvider<DB extends RawAccess> = {
+    /**
+     * Creates a new database client for the actor.
+     * The result is passed to the actor context as `c.db`.
+     * @experimental
+     */
+    createClient: (ctx: {
+        getDatabase: () => Promise<string | unknown>;
+    }) => Promise<DB>;
+    /**
+     * Runs before the actor has started.
+     * Use this to run migrations or other setup tasks.
+     * @experimental
+     */
+    onMigrate: (client: DB) => void | Promise<void>;
+};
+type ExecuteFunction = (query: string, ...args: unknown[]) => Promise<unknown[]>;
+type RawAccess = {
+    /**
+     * Executes a raw SQL query.
+     */
+    execute: ExecuteFunction;
+};
+
+export type { DatabaseProvider as D, RawAccess as R };

package/dist/drizzle/mod.cjs

@@ -1,12 +1,9 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } } function _createStarExport(obj) { Object.keys(obj) .filter((key) => key !== "default" && key !== "__esModule") .forEach((key) => { if (exports.hasOwnProperty(key)) { return; } Object.defineProperty(exports, key, {enumerable: true, configurable: true, get: () => obj[key]}); }); }// src/drizzle/mod.ts
-var _bettersqlite3 = require('better-sqlite3'); var Database = _interopRequireWildcard(_bettersqlite3);
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function _createStarExport(obj) { Object.keys(obj) .filter((key) => key !== "default" && key !== "__esModule") .forEach((key) => { if (exports.hasOwnProperty(key)) { return; } Object.defineProperty(exports, key, {enumerable: true, configurable: true, get: () => obj[key]}); }); }// src/drizzle/mod.ts
+var _bettersqlite3 = require('better-sqlite3'); var _bettersqlite32 = _interopRequireDefault(_bettersqlite3);
 
 
-var _bettersqlite32 = require('drizzle-orm/better-sqlite3');
+var _bettersqlite33 = require('drizzle-orm/better-sqlite3');
 var _durablesqlite = require('drizzle-orm/durable-sqlite');
-
-
-
 var _migrator = require('drizzle-orm/durable-sqlite/migrator');
 var _sqlitecore = require('drizzle-orm/sqlite-core'); _createStarExport(_sqlitecore);
 var _drizzlekit = require('drizzle-kit');
@@ -18,34 +15,42 @@ function defineConfig(config) {
   });
 }
 function db(config) {
-  return async (ctx) => {
-    const conn = await ctx.createDatabase();
-    if (!conn) {
-      throw new Error(
-        "Cannot create database connection, or database feature is not enabled."
-      );
-    }
-    if (typeof conn === "object" && conn && "exec" in conn) {
-      const client2 = _durablesqlite.drizzle.call(void 0, conn, config);
-      return {
-        client: client2,
-        onMigrate: async () => {
-          await _migrator.migrate.call(void 0, client2, config == null ? void 0 : config.migrations);
-        }
-      };
-    }
-    const client = _bettersqlite32.drizzle.call(void 0, {
-      client: new Database(conn),
-      ...config
-    });
-    return {
-      client,
-      onMigrate: async () => {
+  return {
+    createClient: async (ctx) => {
+      if (!ctx.getDatabase) {
+        throw new Error("createDatabase method is not available in context.");
+      }
+      const conn = await ctx.getDatabase();
+      if (!conn) {
+        throw new Error(
+          "Cannot create database connection, or database feature is not enabled."
+        );
+      }
+      if (isSQLiteShim(conn)) {
+        const client2 = _durablesqlite.drizzle.call(void 0, conn, config);
+        return Object.assign(client2, {
+          // client.$client.exec is the underlying SQLite client
+          execute: async (query, ...args) => client2.$client.exec(query, ...args)
+        });
+      }
+      const client = _bettersqlite33.drizzle.call(void 0, {
+        client: new (0, _bettersqlite32.default)(conn),
+        ...config
+      });
+      return Object.assign(client, {
+        execute: async (query, ...args) => client.$client.prepare(query).all(...args)
+      });
+    },
+    onMigrate: async (client) => {
+      if (config == null ? void 0 : config.migrations) {
         await _migrator.migrate.call(void 0, client, config == null ? void 0 : config.migrations);
       }
-    };
+    }
   };
 }
+function isSQLiteShim(conn) {
+  return typeof conn === "object" && conn !== null && "exec" in conn && typeof conn.exec === "function";
+}
 
 
 

package/dist/drizzle/mod.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/home/nathan/rivetkit/packages/db/dist/drizzle/mod.cjs","../../src/drizzle/mod.ts"],"names":["client"],"mappings":"AAAA;ACAA,uGAA0B;AAC1B;AAEC;AAAW,4DACL;AACP,2DAA0C;AAC1C;AACC;AACA;AAAW,+DACL;AAGP,qFAAc;AAEd,yCAAkE;AAE3D,SAAS,YAAA,CACf,MAAA,EACS;AAGT,EAAA,OAAO,sCAAA;AAAqB,IAC3B,OAAA,EAAS,QAAA;AAAA,IACT,MAAA,EAAQ,gBAAA;AAAA,IACR,GAAG;AAAA,EACJ,CAAC,CAAA;AACF;AAYO,SAAS,EAAA,CAGf,MAAA,EACkD;AAClD,EAAA,OAAO,MAAA,CAAO,GAAA,EAAA,GAAQ;AACrB,IAAA,MAAM,KAAA,EAAO,MAAM,GAAA,CAAI,cAAA,CAAe,CAAA;AAEtC,IAAA,GAAA,CAAI,CAAC,IAAA,EAAM;AACV,MAAA,MAAM,IAAI,KAAA;AAAA,QACT;AAAA,MACD,CAAA;AAAA,IACD;AAEA,IAAA,GAAA,CAAI,OAAO,KAAA,IAAS,SAAA,GAAY,KAAA,GAAQ,OAAA,GAAU,IAAA,EAAM;AAGvD,MAAA,MAAMA,QAAAA,EAAS,oCAAA,IAAe,EAAM,MAAM,CAAA;AAC1C,MAAA,OAAO;AAAA,QACN,MAAA,EAAAA,OAAAA;AAAA,QACA,SAAA,EAAW,MAAA,CAAA,EAAA,GAAY;AACtB,UAAA,MAAM,+BAAA,OAAeA,EAAQ,OAAA,GAAA,KAAA,EAAA,KAAA,EAAA,EAAA,MAAA,CAAQ,UAAU,CAAA;AAAA,QAChD;AAAA,MACD,CAAA;AAAA,IACD;AAEA,IAAA,MAAM,OAAA,EAAS,qCAAA;AAAc,MAC5B,MAAA,EAAQ,IAAI,QAAA,CAAS,IAAc,CAAA;AAAA,MACnC,GAAG;AAAA,IACJ,CAAC,CAAA;AAED,IAAA,OAAO;AAAA,MACN,MAAA;AAAA,MACA,SAAA,EAAW,MAAA,CAAA,EAAA,GAAY;AACtB,QAAA,MAAM,+BAAA,MAAc,EAAQ,OAAA,GAAA,KAAA,EAAA,KAAA,EAAA,EAAA,MAAA,CAAQ,UAAU,CAAA;AAAA,MAC/C;AAAA,IACD,CAAA;AAAA,EACD,CAAA;AACD;AD5BA;AACE;AACA;AACF,qDAAC","file":"/home/nathan/rivetkit/packages/db/dist/drizzle/mod.cjs","sourcesContent":[null,"import * as Database from \"better-sqlite3\";\nimport {\n\ttype BetterSQLite3Database,\n\tdrizzle as sqliteDrizzle,\n} from \"drizzle-orm/better-sqlite3\";\nimport { drizzle as durableDrizzle } from \"drizzle-orm/durable-sqlite\";\nimport {\n\tmigrate as durableMigrate,\n\tmigrate as sqliteMigrate,\n} from \"drizzle-orm/durable-sqlite/migrator\";\nimport type { DatabaseFactory } from \"@/config\";\n\nexport * from \"drizzle-orm/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\t// This is a workaround to avoid the \"drizzle-kit\" import issue in the examples.\n\t// It allows us to use the same defineConfig function in both the main package and the examples.\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\t/**\n\t * The database schema.\n\t */\n\tschema?: TSchema;\n\tmigrations?: any;\n}\n\nexport function db<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n>(\n\tconfig?: DatabaseFactoryConfig<TSchema>,\n): DatabaseFactory<BetterSQLite3Database<TSchema>> {\n\treturn async (ctx) => {\n\t\tconst conn = await ctx.createDatabase();\n\n\t\tif (!conn) {\n\t\t\tthrow new Error(\n\t\t\t\t\"Cannot create database connection, or database feature is not enabled.\",\n\t\t\t);\n\t\t}\n\n\t\tif (typeof conn === \"object\" && conn && \"exec\" in conn) {\n\t\t\t// If the connection is already an object with exec method, return it\n\t\t\t// i.e. in serverless environments (Cloudflare Workers)\n\t\t\tconst client = durableDrizzle(conn, config);\n\t\t\treturn {\n\t\t\t\tclient,\n\t\t\t\tonMigrate: async () => {\n\t\t\t\t\tawait durableMigrate(client, config?.migrations);\n\t\t\t\t},\n\t\t\t};\n\t\t}\n\n\t\tconst client = sqliteDrizzle({\n\t\t\tclient: new Database(conn as string),\n\t\t\t...config,\n\t\t});\n\n\t\treturn {\n\t\t\tclient,\n\t\t\tonMigrate: async () => {\n\t\t\t\tawait sqliteMigrate(client, config?.migrations);\n\t\t\t},\n\t\t};\n\t};\n}\n"]}
+{"version":3,"sources":["/home/nathan/rivetkit/packages/db/dist/drizzle/mod.cjs","../../src/drizzle/mod.ts"],"names":["client"],"mappings":"AAAA;ACAA,6GAAqB;AACrB;AAEC;AAAW,4DACL;AACP,2DAA0C;AAC1C,+DAA0C;AAG1C,qFAAc;AAEd,yCAAkE;AAG3D,SAAS,YAAA,CACf,MAAA,EACS;AAGT,EAAA,OAAO,sCAAA;AAAqB,IAC3B,OAAA,EAAS,QAAA;AAAA,IACT,MAAA,EAAQ,gBAAA;AAAA,IACR,GAAG;AAAA,EACJ,CAAC,CAAA;AACF;AAYO,SAAS,EAAA,CAGf,MAAA,EAC+D;AAC/D,EAAA,OAAO;AAAA,IACN,YAAA,EAAc,MAAA,CAAO,GAAA,EAAA,GAAQ;AAE5B,MAAA,GAAA,CAAI,CAAC,GAAA,CAAI,WAAA,EAAa;AACrB,QAAA,MAAM,IAAI,KAAA,CAAM,oDAAoD,CAAA;AAAA,MACrE;AAEA,MAAA,MAAM,KAAA,EAAO,MAAM,GAAA,CAAI,WAAA,CAAY,CAAA;AAEnC,MAAA,GAAA,CAAI,CAAC,IAAA,EAAM;AACV,QAAA,MAAM,IAAI,KAAA;AAAA,UACT;AAAA,QACD,CAAA;AAAA,MACD;AAEA,MAAA,GAAA,CAAI,YAAA,CAAa,IAAI,CAAA,EAAG;AAGvB,QAAA,MAAMA,QAAAA,EAAS,oCAAA,IAAoC,EAAM,MAAM,CAAA;AAC/D,QAAA,OAAO,MAAA,CAAO,MAAA,CAAOA,OAAAA,EAAQ;AAAA;AAAA,UAE5B,OAAA,EAAS,MAAA,CAAO,KAAA,EAAA,GAAU,IAAA,EAAA,GACzBA,OAAAA,CAAO,OAAA,CAAQ,IAAA,CAAK,KAAA,EAAO,GAAG,IAAI;AAAA,QACpC,CAAqB,CAAA;AAAA,MACtB;AAGA,MAAA,MAAM,OAAA,EAAS,qCAAA;AAAc,QAC5B,MAAA,EAAQ,IAAI,4BAAA,CAAS,IAAc,CAAA;AAAA,QACnC,GAAG;AAAA,MACJ,CAAC,CAAA;AAED,MAAA,OAAO,MAAA,CAAO,MAAA,CAAO,MAAA,EAAQ;AAAA,QAC5B,OAAA,EAAS,MAAA,CAAO,KAAA,EAAA,GAAU,IAAA,EAAA,GACzB,MAAA,CAAO,OAAA,CAAQ,OAAA,CAAQ,KAAK,CAAA,CAAE,GAAA,CAAI,GAAG,IAAI;AAAA,MAC3C,CAAqB,CAAA;AAAA,IACtB,CAAA;AAAA,IACA,SAAA,EAAW,MAAA,CAAO,MAAA,EAAA,GAAW;AAE5B,MAAA,GAAA,CAAI,OAAA,GAAA,KAAA,EAAA,KAAA,EAAA,EAAA,MAAA,CAAQ,UAAA,EAAY;AACvB,QAAA,MAAM,+BAAA,MAAe,EAAQ,OAAA,GAAA,KAAA,EAAA,KAAA,EAAA,EAAA,MAAA,CAAQ,UAAU,CAAA;AAAA,MAChD;AAAA,IACD;AAAA,EACD,CAAA;AACD;AAEA,SAAS,YAAA,CAAa,IAAA,EAAmC;AACxD,EAAA,OACC,OAAO,KAAA,IAAS,SAAA,GAChB,KAAA,IAAS,KAAA,GACT,OAAA,GAAU,KAAA,GACV,OAAQ,IAAA,CAAa,KAAA,IAAS,UAAA;AAEhC;ADzCA;AACE;AACA;AACF,qDAAC","file":"/home/nathan/rivetkit/packages/db/dist/drizzle/mod.cjs","sourcesContent":[null,"import Database from \"better-sqlite3\";\nimport {\n\ttype BetterSQLite3Database,\n\tdrizzle as sqliteDrizzle,\n} from \"drizzle-orm/better-sqlite3\";\nimport { drizzle as durableDrizzle } from \"drizzle-orm/durable-sqlite\";\nimport { migrate as durableMigrate } from \"drizzle-orm/durable-sqlite/migrator\";\nimport type { DatabaseProvider, RawAccess } from \"@/config\";\n\nexport * from \"drizzle-orm/sqlite-core\";\n\nimport { type Config, defineConfig as originalDefineConfig } from \"drizzle-kit\";\nimport type { SQLiteShim } from \"@/utils\";\n\nexport function defineConfig(\n\tconfig: Partial<Config & { driver: \"durable-sqlite\" }>,\n): Config {\n\t// This is a workaround to avoid the \"drizzle-kit\" import issue in the examples.\n\t// It allows us to use the same defineConfig function in both the main package and the examples.\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\t/**\n\t * The database schema.\n\t */\n\tschema?: TSchema;\n\tmigrations?: any;\n}\n\nexport function db<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n>(\n\tconfig?: DatabaseFactoryConfig<TSchema>,\n): DatabaseProvider<BetterSQLite3Database<TSchema> & RawAccess> {\n\treturn {\n\t\tcreateClient: async (ctx) => {\n\t\t\t// Create a database connection using the provided context\n\t\t\tif (!ctx.getDatabase) {\n\t\t\t\tthrow new Error(\"createDatabase method is not available in context.\");\n\t\t\t}\n\n\t\t\tconst conn = await ctx.getDatabase();\n\n\t\t\tif (!conn) {\n\t\t\t\tthrow new Error(\n\t\t\t\t\t\"Cannot create database connection, or database feature is not enabled.\",\n\t\t\t\t);\n\t\t\t}\n\n\t\t\tif (isSQLiteShim(conn)) {\n\t\t\t\t// If the connection is already an object with exec method, return it\n\t\t\t\t// i.e. in serverless environments (Cloudflare Workers)\n\t\t\t\tconst client = durableDrizzle<TSchema, SQLiteShim>(conn, config);\n\t\t\t\treturn Object.assign(client, {\n\t\t\t\t\t// client.$client.exec is the underlying SQLite client\n\t\t\t\t\texecute: async (query, ...args) =>\n\t\t\t\t\t\tclient.$client.exec(query, ...args),\n\t\t\t\t} satisfies RawAccess);\n\t\t\t}\n\n\t\t\t// Create a database client using the connection\n\t\t\tconst client = sqliteDrizzle({\n\t\t\t\tclient: new Database(conn as string),\n\t\t\t\t...config,\n\t\t\t});\n\n\t\t\treturn Object.assign(client, {\n\t\t\t\texecute: async (query, ...args) =>\n\t\t\t\t\tclient.$client.prepare(query).all(...args),\n\t\t\t} satisfies RawAccess);\n\t\t},\n\t\tonMigrate: async (client) => {\n\t\t\t// Run migrations if provided in the config\n\t\t\tif (config?.migrations) {\n\t\t\t\tawait durableMigrate(client, config?.migrations);\n\t\t\t}\n\t\t},\n\t};\n}\n\nfunction isSQLiteShim(conn: unknown): conn is SQLiteShim {\n\treturn (\n\t\ttypeof conn === \"object\" &&\n\t\tconn !== null &&\n\t\t\"exec\" in conn &&\n\t\ttypeof (conn as any).exec === \"function\"\n\t);\n}\n"]}

package/dist/drizzle/mod.d.cts

@@ -1,5 +1,5 @@
 import { BetterSQLite3Database } from 'drizzle-orm/better-sqlite3';
-import { D as DatabaseFactory } from '../config-BIvph9_B.cjs';
+import { D as DatabaseProvider, R as RawAccess } from '../config-D99afWH0.cjs';
 export * from 'drizzle-orm/sqlite-core';
 import { Config } from 'drizzle-kit';
 
@@ -13,6 +13,6 @@ interface DatabaseFactoryConfig<TSchema extends Record<string, unknown> = Record
     schema?: TSchema;
     migrations?: any;
 }
-declare function db<TSchema extends Record<string, unknown> = Record<string, never>>(config?: DatabaseFactoryConfig<TSchema>): DatabaseFactory<BetterSQLite3Database<TSchema>>;
+declare function db<TSchema extends Record<string, unknown> = Record<string, never>>(config?: DatabaseFactoryConfig<TSchema>): DatabaseProvider<BetterSQLite3Database<TSchema> & RawAccess>;
 
 export { db, defineConfig };

package/dist/drizzle/mod.d.ts

@@ -1,5 +1,5 @@
 import { BetterSQLite3Database } from 'drizzle-orm/better-sqlite3';
-import { D as DatabaseFactory } from '../config-BIvph9_B.js';
+import { D as DatabaseProvider, R as RawAccess } from '../config-D99afWH0.js';
 export * from 'drizzle-orm/sqlite-core';
 import { Config } from 'drizzle-kit';
 
@@ -13,6 +13,6 @@ interface DatabaseFactoryConfig<TSchema extends Record<string, unknown> = Record
     schema?: TSchema;
     migrations?: any;
 }
-declare function db<TSchema extends Record<string, unknown> = Record<string, never>>(config?: DatabaseFactoryConfig<TSchema>): DatabaseFactory<BetterSQLite3Database<TSchema>>;
+declare function db<TSchema extends Record<string, unknown> = Record<string, never>>(config?: DatabaseFactoryConfig<TSchema>): DatabaseProvider<BetterSQLite3Database<TSchema> & RawAccess>;
 
 export { db, defineConfig };

package/dist/drizzle/mod.js

@@ -1,13 +1,10 @@
 // src/drizzle/mod.ts
-import * as Database from "better-sqlite3";
+import Database from "better-sqlite3";
 import {
   drizzle as sqliteDrizzle
 } from "drizzle-orm/better-sqlite3";
 import { drizzle as durableDrizzle } from "drizzle-orm/durable-sqlite";
-import {
-  migrate as durableMigrate,
-  migrate as sqliteMigrate
-} from "drizzle-orm/durable-sqlite/migrator";
+import { migrate as durableMigrate } from "drizzle-orm/durable-sqlite/migrator";
 export * from "drizzle-orm/sqlite-core";
 import { defineConfig as originalDefineConfig } from "drizzle-kit";
 function defineConfig(config) {
@@ -18,34 +15,42 @@ function defineConfig(config) {
   });
 }
 function db(config) {
-  return async (ctx) => {
-    const conn = await ctx.createDatabase();
-    if (!conn) {
-      throw new Error(
-        "Cannot create database connection, or database feature is not enabled."
-      );
-    }
-    if (typeof conn === "object" && conn && "exec" in conn) {
-      const client2 = durableDrizzle(conn, config);
-      return {
-        client: client2,
-        onMigrate: async () => {
-          await durableMigrate(client2, config == null ? void 0 : config.migrations);
-        }
-      };
-    }
-    const client = sqliteDrizzle({
-      client: new Database(conn),
-      ...config
-    });
-    return {
-      client,
-      onMigrate: async () => {
-        await sqliteMigrate(client, config == null ? void 0 : config.migrations);
+  return {
+    createClient: async (ctx) => {
+      if (!ctx.getDatabase) {
+        throw new Error("createDatabase method is not available in context.");
+      }
+      const conn = await ctx.getDatabase();
+      if (!conn) {
+        throw new Error(
+          "Cannot create database connection, or database feature is not enabled."
+        );
       }
-    };
+      if (isSQLiteShim(conn)) {
+        const client2 = durableDrizzle(conn, config);
+        return Object.assign(client2, {
+          // client.$client.exec is the underlying SQLite client
+          execute: async (query, ...args) => client2.$client.exec(query, ...args)
+        });
+      }
+      const client = sqliteDrizzle({
+        client: new Database(conn),
+        ...config
+      });
+      return Object.assign(client, {
+        execute: async (query, ...args) => client.$client.prepare(query).all(...args)
+      });
+    },
+    onMigrate: async (client) => {
+      if (config == null ? void 0 : config.migrations) {
+        await durableMigrate(client, config == null ? void 0 : config.migrations);
+      }
+    }
   };
 }
+function isSQLiteShim(conn) {
+  return typeof conn === "object" && conn !== null && "exec" in conn && typeof conn.exec === "function";
+}
 export {
   db,
   defineConfig

package/dist/drizzle/mod.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/drizzle/mod.ts"],"sourcesContent":["import * as Database from \"better-sqlite3\";\nimport {\n\ttype BetterSQLite3Database,\n\tdrizzle as sqliteDrizzle,\n} from \"drizzle-orm/better-sqlite3\";\nimport { drizzle as durableDrizzle } from \"drizzle-orm/durable-sqlite\";\nimport {\n\tmigrate as durableMigrate,\n\tmigrate as sqliteMigrate,\n} from \"drizzle-orm/durable-sqlite/migrator\";\nimport type { DatabaseFactory } from \"@/config\";\n\nexport * from \"drizzle-orm/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\t// This is a workaround to avoid the \"drizzle-kit\" import issue in the examples.\n\t// It allows us to use the same defineConfig function in both the main package and the examples.\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\t/**\n\t * The database schema.\n\t */\n\tschema?: TSchema;\n\tmigrations?: any;\n}\n\nexport function db<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n>(\n\tconfig?: DatabaseFactoryConfig<TSchema>,\n): DatabaseFactory<BetterSQLite3Database<TSchema>> {\n\treturn async (ctx) => {\n\t\tconst conn = await ctx.createDatabase();\n\n\t\tif (!conn) {\n\t\t\tthrow new Error(\n\t\t\t\t\"Cannot create database connection, or database feature is not enabled.\",\n\t\t\t);\n\t\t}\n\n\t\tif (typeof conn === \"object\" && conn && \"exec\" in conn) {\n\t\t\t// If the connection is already an object with exec method, return it\n\t\t\t// i.e. in serverless environments (Cloudflare Workers)\n\t\t\tconst client = durableDrizzle(conn, config);\n\t\t\treturn {\n\t\t\t\tclient,\n\t\t\t\tonMigrate: async () => {\n\t\t\t\t\tawait durableMigrate(client, config?.migrations);\n\t\t\t\t},\n\t\t\t};\n\t\t}\n\n\t\tconst client = sqliteDrizzle({\n\t\t\tclient: new Database(conn as string),\n\t\t\t...config,\n\t\t});\n\n\t\treturn {\n\t\t\tclient,\n\t\t\tonMigrate: async () => {\n\t\t\t\tawait sqliteMigrate(client, config?.migrations);\n\t\t\t},\n\t\t};\n\t};\n}\n"],"mappings":";AAAA,YAAY,cAAc;AAC1B;AAAA,EAEC,WAAW;AAAA,OACL;AACP,SAAS,WAAW,sBAAsB;AAC1C;AAAA,EACC,WAAW;AAAA,EACX,WAAW;AAAA,OACL;AAGP,cAAc;AAEd,SAAsB,gBAAgB,4BAA4B;AAE3D,SAAS,aACf,QACS;AAGT,SAAO,qBAAqB;AAAA,IAC3B,SAAS;AAAA,IACT,QAAQ;AAAA,IACR,GAAG;AAAA,EACJ,CAAC;AACF;AAYO,SAAS,GAGf,QACkD;AAClD,SAAO,OAAO,QAAQ;AACrB,UAAM,OAAO,MAAM,IAAI,eAAe;AAEtC,QAAI,CAAC,MAAM;AACV,YAAM,IAAI;AAAA,QACT;AAAA,MACD;AAAA,IACD;AAEA,QAAI,OAAO,SAAS,YAAY,QAAQ,UAAU,MAAM;AAGvD,YAAMA,UAAS,eAAe,MAAM,MAAM;AAC1C,aAAO;AAAA,QACN,QAAAA;AAAA,QACA,WAAW,YAAY;AACtB,gBAAM,eAAeA,SAAQ,iCAAQ,UAAU;AAAA,QAChD;AAAA,MACD;AAAA,IACD;AAEA,UAAM,SAAS,cAAc;AAAA,MAC5B,QAAQ,IAAI,SAAS,IAAc;AAAA,MACnC,GAAG;AAAA,IACJ,CAAC;AAED,WAAO;AAAA,MACN;AAAA,MACA,WAAW,YAAY;AACtB,cAAM,cAAc,QAAQ,iCAAQ,UAAU;AAAA,MAC/C;AAAA,IACD;AAAA,EACD;AACD;","names":["client"]}
+{"version":3,"sources":["../../src/drizzle/mod.ts"],"sourcesContent":["import Database from \"better-sqlite3\";\nimport {\n\ttype BetterSQLite3Database,\n\tdrizzle as sqliteDrizzle,\n} from \"drizzle-orm/better-sqlite3\";\nimport { drizzle as durableDrizzle } from \"drizzle-orm/durable-sqlite\";\nimport { migrate as durableMigrate } from \"drizzle-orm/durable-sqlite/migrator\";\nimport type { DatabaseProvider, RawAccess } from \"@/config\";\n\nexport * from \"drizzle-orm/sqlite-core\";\n\nimport { type Config, defineConfig as originalDefineConfig } from \"drizzle-kit\";\nimport type { SQLiteShim } from \"@/utils\";\n\nexport function defineConfig(\n\tconfig: Partial<Config & { driver: \"durable-sqlite\" }>,\n): Config {\n\t// This is a workaround to avoid the \"drizzle-kit\" import issue in the examples.\n\t// It allows us to use the same defineConfig function in both the main package and the examples.\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\t/**\n\t * The database schema.\n\t */\n\tschema?: TSchema;\n\tmigrations?: any;\n}\n\nexport function db<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n>(\n\tconfig?: DatabaseFactoryConfig<TSchema>,\n): DatabaseProvider<BetterSQLite3Database<TSchema> & RawAccess> {\n\treturn {\n\t\tcreateClient: async (ctx) => {\n\t\t\t// Create a database connection using the provided context\n\t\t\tif (!ctx.getDatabase) {\n\t\t\t\tthrow new Error(\"createDatabase method is not available in context.\");\n\t\t\t}\n\n\t\t\tconst conn = await ctx.getDatabase();\n\n\t\t\tif (!conn) {\n\t\t\t\tthrow new Error(\n\t\t\t\t\t\"Cannot create database connection, or database feature is not enabled.\",\n\t\t\t\t);\n\t\t\t}\n\n\t\t\tif (isSQLiteShim(conn)) {\n\t\t\t\t// If the connection is already an object with exec method, return it\n\t\t\t\t// i.e. in serverless environments (Cloudflare Workers)\n\t\t\t\tconst client = durableDrizzle<TSchema, SQLiteShim>(conn, config);\n\t\t\t\treturn Object.assign(client, {\n\t\t\t\t\t// client.$client.exec is the underlying SQLite client\n\t\t\t\t\texecute: async (query, ...args) =>\n\t\t\t\t\t\tclient.$client.exec(query, ...args),\n\t\t\t\t} satisfies RawAccess);\n\t\t\t}\n\n\t\t\t// Create a database client using the connection\n\t\t\tconst client = sqliteDrizzle({\n\t\t\t\tclient: new Database(conn as string),\n\t\t\t\t...config,\n\t\t\t});\n\n\t\t\treturn Object.assign(client, {\n\t\t\t\texecute: async (query, ...args) =>\n\t\t\t\t\tclient.$client.prepare(query).all(...args),\n\t\t\t} satisfies RawAccess);\n\t\t},\n\t\tonMigrate: async (client) => {\n\t\t\t// Run migrations if provided in the config\n\t\t\tif (config?.migrations) {\n\t\t\t\tawait durableMigrate(client, config?.migrations);\n\t\t\t}\n\t\t},\n\t};\n}\n\nfunction isSQLiteShim(conn: unknown): conn is SQLiteShim {\n\treturn (\n\t\ttypeof conn === \"object\" &&\n\t\tconn !== null &&\n\t\t\"exec\" in conn &&\n\t\ttypeof (conn as any).exec === \"function\"\n\t);\n}\n"],"mappings":";AAAA,OAAO,cAAc;AACrB;AAAA,EAEC,WAAW;AAAA,OACL;AACP,SAAS,WAAW,sBAAsB;AAC1C,SAAS,WAAW,sBAAsB;AAG1C,cAAc;AAEd,SAAsB,gBAAgB,4BAA4B;AAG3D,SAAS,aACf,QACS;AAGT,SAAO,qBAAqB;AAAA,IAC3B,SAAS;AAAA,IACT,QAAQ;AAAA,IACR,GAAG;AAAA,EACJ,CAAC;AACF;AAYO,SAAS,GAGf,QAC+D;AAC/D,SAAO;AAAA,IACN,cAAc,OAAO,QAAQ;AAE5B,UAAI,CAAC,IAAI,aAAa;AACrB,cAAM,IAAI,MAAM,oDAAoD;AAAA,MACrE;AAEA,YAAM,OAAO,MAAM,IAAI,YAAY;AAEnC,UAAI,CAAC,MAAM;AACV,cAAM,IAAI;AAAA,UACT;AAAA,QACD;AAAA,MACD;AAEA,UAAI,aAAa,IAAI,GAAG;AAGvB,cAAMA,UAAS,eAAoC,MAAM,MAAM;AAC/D,eAAO,OAAO,OAAOA,SAAQ;AAAA;AAAA,UAE5B,SAAS,OAAO,UAAU,SACzBA,QAAO,QAAQ,KAAK,OAAO,GAAG,IAAI;AAAA,QACpC,CAAqB;AAAA,MACtB;AAGA,YAAM,SAAS,cAAc;AAAA,QAC5B,QAAQ,IAAI,SAAS,IAAc;AAAA,QACnC,GAAG;AAAA,MACJ,CAAC;AAED,aAAO,OAAO,OAAO,QAAQ;AAAA,QAC5B,SAAS,OAAO,UAAU,SACzB,OAAO,QAAQ,QAAQ,KAAK,EAAE,IAAI,GAAG,IAAI;AAAA,MAC3C,CAAqB;AAAA,IACtB;AAAA,IACA,WAAW,OAAO,WAAW;AAE5B,UAAI,iCAAQ,YAAY;AACvB,cAAM,eAAe,QAAQ,iCAAQ,UAAU;AAAA,MAChD;AAAA,IACD;AAAA,EACD;AACD;AAEA,SAAS,aAAa,MAAmC;AACxD,SACC,OAAO,SAAS,YAChB,SAAS,QACT,UAAU,QACV,OAAQ,KAAa,SAAS;AAEhC;","names":["client"]}

package/dist/mod.cjs

@@ -1,30 +1,45 @@
-"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { newObj[key] = obj[key]; } } } newObj.default = obj; return newObj; } }// src/mod.ts
-var _bettersqlite3 = require('better-sqlite3'); var SQLite = _interopRequireWildcard(_bettersqlite3);
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }// src/mod.ts
+var _bettersqlite3 = require('better-sqlite3'); var _bettersqlite32 = _interopRequireDefault(_bettersqlite3);
+
+// src/utils.ts
+function isSQLiteShim(conn) {
+  return typeof conn === "object" && conn !== null && "exec" in conn && typeof conn.exec === "function";
+}
+
+// src/mod.ts
 function db({
   onMigrate
 } = {}) {
-  return async (ctx) => {
-    const conn = await ctx.createDatabase();
-    if (!conn) {
-      throw new Error(
-        "Cannot create database connection, or database feature is not enabled."
-      );
-    }
-    if (typeof conn === "object" && conn && "exec" in conn) {
-      return {
-        client: conn,
-        onMigrate: () => {
-          onMigrate == null ? void 0 : onMigrate(client);
+  return {
+    createClient: async (ctx) => {
+      if (!ctx.getDatabase) {
+        throw new Error("createDatabase method is not available in context.");
+      }
+      const conn = await ctx.getDatabase();
+      if (!conn) {
+        throw new Error(
+          "Cannot create database connection, or database feature is not enabled."
+        );
+      }
+      if (isSQLiteShim(conn)) {
+        return Object.assign({}, conn, {
+          execute: async (query, ...args) => {
+            return conn.exec(query, ...args);
+          }
+        });
+      }
+      const client = new (0, _bettersqlite32.default)(conn);
+      return Object.assign({}, client, {
+        execute: async (query, ...args) => {
+          return client.prepare(query).all(...args);
         }
-      };
-    }
-    const client = new SQLite(conn);
-    return {
-      client,
-      onMigrate: () => {
-        onMigrate == null ? void 0 : onMigrate(client);
+      });
+    },
+    onMigrate: async (client) => {
+      if (onMigrate) {
+        await onMigrate(client);
       }
-    };
+    }
   };
 }
 

package/dist/mod.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["/home/nathan/rivetkit/packages/db/dist/mod.cjs","../src/mod.ts"],"names":[],"mappings":"AAAA;ACAA,qGAAwB;AAcjB,SAAS,EAAA,CAAG;AAAA,EAClB;AACD,EAAA,EAA2B,CAAC,CAAA,EAAgC;AAC3D,EAAA,OAAO,MAAA,CAAO,GAAA,EAAA,GAAQ;AACrB,IAAA,MAAM,KAAA,EAAO,MAAM,GAAA,CAAI,cAAA,CAAe,CAAA;AAEtC,IAAA,GAAA,CAAI,CAAC,IAAA,EAAM;AACV,MAAA,MAAM,IAAI,KAAA;AAAA,QACT;AAAA,MACD,CAAA;AAAA,IACD;AAEA,IAAA,GAAA,CAAI,OAAO,KAAA,IAAS,SAAA,GAAY,KAAA,GAAQ,OAAA,GAAU,IAAA,EAAM;AAGvD,MAAA,OAAO;AAAA,QACN,MAAA,EAAQ,IAAA;AAAA,QACR,SAAA,EAAW,CAAA,EAAA,GAAM;AAChB,UAAA,UAAA,GAAA,KAAA,EAAA,KAAA,EAAA,EAAA,SAAA,CAAY,MAAA,CAAA;AAAA,QACb;AAAA,MACD,CAAA;AAAA,IACD;AAEA,IAAA,MAAM,OAAA,EAAS,IAAI,MAAA,CAAO,IAAc,CAAA;AACxC,IAAA,OAAO;AAAA,MACN,MAAA;AAAA,MACA,SAAA,EAAW,CAAA,EAAA,GAAM;AAChB,QAAA,UAAA,GAAA,KAAA,EAAA,KAAA,EAAA,EAAA,SAAA,CAAY,MAAA,CAAA;AAAA,MACb;AAAA,IACD,CAAA;AAAA,EACD,CAAA;AACD;ADhBA;AACE;AACF,gBAAC","file":"/home/nathan/rivetkit/packages/db/dist/mod.cjs","sourcesContent":[null,"import * as SQLite from \"better-sqlite3\";\nimport type { DatabaseFactory } from \"./config\";\n\n/**\n * On serverless environments, we use a shim, as not all methods are available.\n * This is a minimal shim that only includes the `exec` method, which is used for\n * running raw SQL commands.\n */\ntype SQLiteShim = Pick<SQLite.Database, \"exec\">;\n\ninterface DatabaseFactoryConfig {\n\tonMigrate?: (db: SQLiteShim) => void;\n}\n\nexport function db({\n\tonMigrate,\n}: DatabaseFactoryConfig = {}): DatabaseFactory<SQLiteShim> {\n\treturn async (ctx) => {\n\t\tconst conn = await ctx.createDatabase();\n\n\t\tif (!conn) {\n\t\t\tthrow new Error(\n\t\t\t\t\"Cannot create database connection, or database feature is not enabled.\",\n\t\t\t);\n\t\t}\n\n\t\tif (typeof conn === \"object\" && conn && \"exec\" in conn) {\n\t\t\t// if the connection is already an object with exec method, return it\n\t\t\t// i.e. in serverless environments (cloudflare)\n\t\t\treturn {\n\t\t\t\tclient: conn as SQLiteShim,\n\t\t\t\tonMigrate: () => {\n\t\t\t\t\tonMigrate?.(client);\n\t\t\t\t},\n\t\t\t};\n\t\t}\n\n\t\tconst client = new SQLite(conn as string);\n\t\treturn {\n\t\t\tclient,\n\t\t\tonMigrate: () => {\n\t\t\t\tonMigrate?.(client);\n\t\t\t},\n\t\t};\n\t};\n}\n"]}
+{"version":3,"sources":["/home/nathan/rivetkit/packages/db/dist/mod.cjs","../src/mod.ts","../src/utils.ts"],"names":[],"mappings":"AAAA;ACAA,6GAAmB;ADEnB;AACA;AEMO,SAAS,YAAA,CAAgB,IAAA,EAAuC;AACtE,EAAA,OACC,OAAO,KAAA,IAAS,SAAA,GAChB,KAAA,IAAS,KAAA,GACT,OAAA,GAAU,KAAA,GACV,OAAQ,IAAA,CAAoB,KAAA,IAAS,UAAA;AAEvC;AFTA;AACA;ACAO,SAAS,EAAA,CAAG;AAAA,EAClB;AACD,EAAA,EAA2B,CAAC,CAAA,EAAgC;AAC3D,EAAA,OAAO;AAAA,IACN,YAAA,EAAc,MAAA,CAAO,GAAA,EAAA,GAAQ;AAE5B,MAAA,GAAA,CAAI,CAAC,GAAA,CAAI,WAAA,EAAa;AACrB,QAAA,MAAM,IAAI,KAAA,CAAM,oDAAoD,CAAA;AAAA,MACrE;AAEA,MAAA,MAAM,KAAA,EAAO,MAAM,GAAA,CAAI,WAAA,CAAY,CAAA;AAEnC,MAAA,GAAA,CAAI,CAAC,IAAA,EAAM;AACV,QAAA,MAAM,IAAI,KAAA;AAAA,UACT;AAAA,QACD,CAAA;AAAA,MACD;AAEA,MAAA,GAAA,CAAI,YAAA,CAAa,IAAI,CAAA,EAAG;AAGvB,QAAA,OAAO,MAAA,CAAO,MAAA,CAAO,CAAC,CAAA,EAAG,IAAA,EAAM;AAAA,UAC9B,OAAA,EAAS,MAAA,CAAO,KAAA,EAAA,GAAU,IAAA,EAAA,GAAS;AAClC,YAAA,OAAO,IAAA,CAAK,IAAA,CAAK,KAAA,EAAO,GAAG,IAAI,CAAA;AAAA,UAChC;AAAA,QACD,CAAqB,CAAA;AAAA,MACtB;AAEA,MAAA,MAAM,OAAA,EAAS,IAAI,4BAAA,CAAO,IAAc,CAAA;AACxC,MAAA,OAAO,MAAA,CAAO,MAAA,CAAO,CAAC,CAAA,EAAG,MAAA,EAAQ;AAAA,QAChC,OAAA,EAAS,MAAA,CAAO,KAAA,EAAA,GAAU,IAAA,EAAA,GAAS;AAClC,UAAA,OAAO,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA,CAAE,GAAA,CAAI,GAAG,IAAI,CAAA;AAAA,QACzC;AAAA,MACD,CAAqB,CAAA;AAAA,IACtB,CAAA;AAAA,IACA,SAAA,EAAW,MAAA,CAAO,MAAA,EAAA,GAAW;AAE5B,MAAA,GAAA,CAAI,SAAA,EAAW;AACd,QAAA,MAAM,SAAA,CAAU,MAAM,CAAA;AAAA,MACvB;AAAA,IACD;AAAA,EACD,CAAA;AACD;ADNA;AACE;AACF,gBAAC","file":"/home/nathan/rivetkit/packages/db/dist/mod.cjs","sourcesContent":[null,"import SQLite from \"better-sqlite3\";\nimport type { DatabaseProvider, RawAccess } from \"./config\";\nimport { isSQLiteShim, type SQLiteShim } from \"./utils\";\n\ninterface DatabaseFactoryConfig {\n\tonMigrate?: (db: RawAccess) => Promise<void> | void;\n}\n\nexport function db({\n\tonMigrate,\n}: DatabaseFactoryConfig = {}): DatabaseProvider<RawAccess> {\n\treturn {\n\t\tcreateClient: async (ctx) => {\n\t\t\t// Create a database connection using the provided context\n\t\t\tif (!ctx.getDatabase) {\n\t\t\t\tthrow new Error(\"createDatabase method is not available in context.\");\n\t\t\t}\n\n\t\t\tconst conn = await ctx.getDatabase();\n\n\t\t\tif (!conn) {\n\t\t\t\tthrow new Error(\n\t\t\t\t\t\"Cannot create database connection, or database feature is not enabled.\",\n\t\t\t\t);\n\t\t\t}\n\n\t\t\tif (isSQLiteShim(conn)) {\n\t\t\t\t// If the connection is already an object with exec method, return it\n\t\t\t\t// i.e. in serverless environments (Cloudflare Workers)\n\t\t\t\treturn Object.assign({}, conn, {\n\t\t\t\t\texecute: async (query, ...args) => {\n\t\t\t\t\t\treturn conn.exec(query, ...args);\n\t\t\t\t\t},\n\t\t\t\t} satisfies RawAccess) as SQLiteShim & RawAccess;\n\t\t\t}\n\n\t\t\tconst client = new SQLite(conn as string);\n\t\t\treturn Object.assign({}, client, {\n\t\t\t\texecute: async (query, ...args) => {\n\t\t\t\t\treturn client.prepare(query).all(...args);\n\t\t\t\t},\n\t\t\t} satisfies RawAccess) as RawAccess;\n\t\t},\n\t\tonMigrate: async (client) => {\n\t\t\t// Run migrations if provided in the config\n\t\t\tif (onMigrate) {\n\t\t\t\tawait onMigrate(client);\n\t\t\t}\n\t\t},\n\t};\n}\n","/**\n * On serverless environments, we use a shim, as not all methods are available.\n * This is a minimal shim that only includes the `exec` method, which is used for\n * running raw SQL commands.\n */\nexport type SQLiteShim = {\n\texec: (query: string, ...args: unknown[]) => unknown[];\n};\n\nexport function isSQLiteShim<T>(conn: unknown): conn is SQLiteShim & T {\n\treturn (\n\t\ttypeof conn === \"object\" &&\n\t\tconn !== null &&\n\t\t\"exec\" in conn &&\n\t\ttypeof (conn as SQLiteShim).exec === \"function\"\n\t);\n}\n"]}

package/dist/mod.d.cts

@@ -1,15 +1,8 @@
-import * as SQLite from 'better-sqlite3';
-import { D as DatabaseFactory } from './config-BIvph9_B.cjs';
+import { R as RawAccess, D as DatabaseProvider } from './config-D99afWH0.cjs';
 
-/**
- * On serverless environments, we use a shim, as not all methods are available.
- * This is a minimal shim that only includes the `exec` method, which is used for
- * running raw SQL commands.
- */
-type SQLiteShim = Pick<SQLite.Database, "exec">;
 interface DatabaseFactoryConfig {
-    onMigrate?: (db: SQLiteShim) => void;
+    onMigrate?: (db: RawAccess) => Promise<void> | void;
 }
-declare function db({ onMigrate, }?: DatabaseFactoryConfig): DatabaseFactory<SQLiteShim>;
+declare function db({ onMigrate, }?: DatabaseFactoryConfig): DatabaseProvider<RawAccess>;
 
 export { db };

package/dist/mod.d.ts

@@ -1,15 +1,8 @@
-import * as SQLite from 'better-sqlite3';
-import { D as DatabaseFactory } from './config-BIvph9_B.js';
+import { R as RawAccess, D as DatabaseProvider } from './config-D99afWH0.js';
 
-/**
- * On serverless environments, we use a shim, as not all methods are available.
- * This is a minimal shim that only includes the `exec` method, which is used for
- * running raw SQL commands.
- */
-type SQLiteShim = Pick<SQLite.Database, "exec">;
 interface DatabaseFactoryConfig {
-    onMigrate?: (db: SQLiteShim) => void;
+    onMigrate?: (db: RawAccess) => Promise<void> | void;
 }
-declare function db({ onMigrate, }?: DatabaseFactoryConfig): DatabaseFactory<SQLiteShim>;
+declare function db({ onMigrate, }?: DatabaseFactoryConfig): DatabaseProvider<RawAccess>;
 
 export { db };

package/dist/mod.js

@@ -1,30 +1,45 @@
 // src/mod.ts
-import * as SQLite from "better-sqlite3";
+import SQLite from "better-sqlite3";
+
+// src/utils.ts
+function isSQLiteShim(conn) {
+  return typeof conn === "object" && conn !== null && "exec" in conn && typeof conn.exec === "function";
+}
+
+// src/mod.ts
 function db({
   onMigrate
 } = {}) {
-  return async (ctx) => {
-    const conn = await ctx.createDatabase();
-    if (!conn) {
-      throw new Error(
-        "Cannot create database connection, or database feature is not enabled."
-      );
-    }
-    if (typeof conn === "object" && conn && "exec" in conn) {
-      return {
-        client: conn,
-        onMigrate: () => {
-          onMigrate == null ? void 0 : onMigrate(client);
+  return {
+    createClient: async (ctx) => {
+      if (!ctx.getDatabase) {
+        throw new Error("createDatabase method is not available in context.");
+      }
+      const conn = await ctx.getDatabase();
+      if (!conn) {
+        throw new Error(
+          "Cannot create database connection, or database feature is not enabled."
+        );
+      }
+      if (isSQLiteShim(conn)) {
+        return Object.assign({}, conn, {
+          execute: async (query, ...args) => {
+            return conn.exec(query, ...args);
+          }
+        });
+      }
+      const client = new SQLite(conn);
+      return Object.assign({}, client, {
+        execute: async (query, ...args) => {
+          return client.prepare(query).all(...args);
         }
-      };
-    }
-    const client = new SQLite(conn);
-    return {
-      client,
-      onMigrate: () => {
-        onMigrate == null ? void 0 : onMigrate(client);
+      });
+    },
+    onMigrate: async (client) => {
+      if (onMigrate) {
+        await onMigrate(client);
       }
-    };
+    }
   };
 }
 export {

package/dist/mod.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/mod.ts"],"sourcesContent":["import * as SQLite from \"better-sqlite3\";\nimport type { DatabaseFactory } from \"./config\";\n\n/**\n * On serverless environments, we use a shim, as not all methods are available.\n * This is a minimal shim that only includes the `exec` method, which is used for\n * running raw SQL commands.\n */\ntype SQLiteShim = Pick<SQLite.Database, \"exec\">;\n\ninterface DatabaseFactoryConfig {\n\tonMigrate?: (db: SQLiteShim) => void;\n}\n\nexport function db({\n\tonMigrate,\n}: DatabaseFactoryConfig = {}): DatabaseFactory<SQLiteShim> {\n\treturn async (ctx) => {\n\t\tconst conn = await ctx.createDatabase();\n\n\t\tif (!conn) {\n\t\t\tthrow new Error(\n\t\t\t\t\"Cannot create database connection, or database feature is not enabled.\",\n\t\t\t);\n\t\t}\n\n\t\tif (typeof conn === \"object\" && conn && \"exec\" in conn) {\n\t\t\t// if the connection is already an object with exec method, return it\n\t\t\t// i.e. in serverless environments (cloudflare)\n\t\t\treturn {\n\t\t\t\tclient: conn as SQLiteShim,\n\t\t\t\tonMigrate: () => {\n\t\t\t\t\tonMigrate?.(client);\n\t\t\t\t},\n\t\t\t};\n\t\t}\n\n\t\tconst client = new SQLite(conn as string);\n\t\treturn {\n\t\t\tclient,\n\t\t\tonMigrate: () => {\n\t\t\t\tonMigrate?.(client);\n\t\t\t},\n\t\t};\n\t};\n}\n"],"mappings":";AAAA,YAAY,YAAY;AAcjB,SAAS,GAAG;AAAA,EAClB;AACD,IAA2B,CAAC,GAAgC;AAC3D,SAAO,OAAO,QAAQ;AACrB,UAAM,OAAO,MAAM,IAAI,eAAe;AAEtC,QAAI,CAAC,MAAM;AACV,YAAM,IAAI;AAAA,QACT;AAAA,MACD;AAAA,IACD;AAEA,QAAI,OAAO,SAAS,YAAY,QAAQ,UAAU,MAAM;AAGvD,aAAO;AAAA,QACN,QAAQ;AAAA,QACR,WAAW,MAAM;AAChB,iDAAY;AAAA,QACb;AAAA,MACD;AAAA,IACD;AAEA,UAAM,SAAS,IAAI,OAAO,IAAc;AACxC,WAAO;AAAA,MACN;AAAA,MACA,WAAW,MAAM;AAChB,+CAAY;AAAA,MACb;AAAA,IACD;AAAA,EACD;AACD;","names":[]}
+{"version":3,"sources":["../src/mod.ts","../src/utils.ts"],"sourcesContent":["import SQLite from \"better-sqlite3\";\nimport type { DatabaseProvider, RawAccess } from \"./config\";\nimport { isSQLiteShim, type SQLiteShim } from \"./utils\";\n\ninterface DatabaseFactoryConfig {\n\tonMigrate?: (db: RawAccess) => Promise<void> | void;\n}\n\nexport function db({\n\tonMigrate,\n}: DatabaseFactoryConfig = {}): DatabaseProvider<RawAccess> {\n\treturn {\n\t\tcreateClient: async (ctx) => {\n\t\t\t// Create a database connection using the provided context\n\t\t\tif (!ctx.getDatabase) {\n\t\t\t\tthrow new Error(\"createDatabase method is not available in context.\");\n\t\t\t}\n\n\t\t\tconst conn = await ctx.getDatabase();\n\n\t\t\tif (!conn) {\n\t\t\t\tthrow new Error(\n\t\t\t\t\t\"Cannot create database connection, or database feature is not enabled.\",\n\t\t\t\t);\n\t\t\t}\n\n\t\t\tif (isSQLiteShim(conn)) {\n\t\t\t\t// If the connection is already an object with exec method, return it\n\t\t\t\t// i.e. in serverless environments (Cloudflare Workers)\n\t\t\t\treturn Object.assign({}, conn, {\n\t\t\t\t\texecute: async (query, ...args) => {\n\t\t\t\t\t\treturn conn.exec(query, ...args);\n\t\t\t\t\t},\n\t\t\t\t} satisfies RawAccess) as SQLiteShim & RawAccess;\n\t\t\t}\n\n\t\t\tconst client = new SQLite(conn as string);\n\t\t\treturn Object.assign({}, client, {\n\t\t\t\texecute: async (query, ...args) => {\n\t\t\t\t\treturn client.prepare(query).all(...args);\n\t\t\t\t},\n\t\t\t} satisfies RawAccess) as RawAccess;\n\t\t},\n\t\tonMigrate: async (client) => {\n\t\t\t// Run migrations if provided in the config\n\t\t\tif (onMigrate) {\n\t\t\t\tawait onMigrate(client);\n\t\t\t}\n\t\t},\n\t};\n}\n","/**\n * On serverless environments, we use a shim, as not all methods are available.\n * This is a minimal shim that only includes the `exec` method, which is used for\n * running raw SQL commands.\n */\nexport type SQLiteShim = {\n\texec: (query: string, ...args: unknown[]) => unknown[];\n};\n\nexport function isSQLiteShim<T>(conn: unknown): conn is SQLiteShim & T {\n\treturn (\n\t\ttypeof conn === \"object\" &&\n\t\tconn !== null &&\n\t\t\"exec\" in conn &&\n\t\ttypeof (conn as SQLiteShim).exec === \"function\"\n\t);\n}\n"],"mappings":";AAAA,OAAO,YAAY;;;ACSZ,SAAS,aAAgB,MAAuC;AACtE,SACC,OAAO,SAAS,YAChB,SAAS,QACT,UAAU,QACV,OAAQ,KAAoB,SAAS;AAEvC;;;ADRO,SAAS,GAAG;AAAA,EAClB;AACD,IAA2B,CAAC,GAAgC;AAC3D,SAAO;AAAA,IACN,cAAc,OAAO,QAAQ;AAE5B,UAAI,CAAC,IAAI,aAAa;AACrB,cAAM,IAAI,MAAM,oDAAoD;AAAA,MACrE;AAEA,YAAM,OAAO,MAAM,IAAI,YAAY;AAEnC,UAAI,CAAC,MAAM;AACV,cAAM,IAAI;AAAA,UACT;AAAA,QACD;AAAA,MACD;AAEA,UAAI,aAAa,IAAI,GAAG;AAGvB,eAAO,OAAO,OAAO,CAAC,GAAG,MAAM;AAAA,UAC9B,SAAS,OAAO,UAAU,SAAS;AAClC,mBAAO,KAAK,KAAK,OAAO,GAAG,IAAI;AAAA,UAChC;AAAA,QACD,CAAqB;AAAA,MACtB;AAEA,YAAM,SAAS,IAAI,OAAO,IAAc;AACxC,aAAO,OAAO,OAAO,CAAC,GAAG,QAAQ;AAAA,QAChC,SAAS,OAAO,UAAU,SAAS;AAClC,iBAAO,OAAO,QAAQ,KAAK,EAAE,IAAI,GAAG,IAAI;AAAA,QACzC;AAAA,MACD,CAAqB;AAAA,IACtB;AAAA,IACA,WAAW,OAAO,WAAW;AAE5B,UAAI,WAAW;AACd,cAAM,UAAU,MAAM;AAAA,MACvB;AAAA,IACD;AAAA,EACD;AACD;","names":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rivetkit/db",
-  "version": "0.9.2",
+  "version": "0.9.5",
   "license": "Apache-2.0",
   "sideEffects": false,
   "type": "module",
@@ -53,7 +53,7 @@
   "stableVersion": "0.8.0",
   "dependencies": {
     "better-sqlite3": "^11.10.0",
-    "@rivetkit/core": "0.9.2"
+    "@rivetkit/core": "0.9.5"
   },
   "scripts": {
     "build": "tsup src/mod.ts src/drizzle/mod.ts",

package/dist/config-BIvph9_B.d.cts

@@ -1,10 +0,0 @@
-interface DatabaseConfig<DB> {
-    client: DB;
-    onMigrate: () => void;
-}
-interface DatabaseFactoryContext<DB> {
-    createDatabase: () => Promise<unknown>;
-}
-type DatabaseFactory<DB> = (ctx: DatabaseFactoryContext<DB>) => Promise<DatabaseConfig<DB>>;
-
-export type { DatabaseFactory as D };

package/dist/config-BIvph9_B.d.ts

@@ -1,10 +0,0 @@
-interface DatabaseConfig<DB> {
-    client: DB;
-    onMigrate: () => void;
-}
-interface DatabaseFactoryContext<DB> {
-    createDatabase: () => Promise<unknown>;
-}
-type DatabaseFactory<DB> = (ctx: DatabaseFactoryContext<DB>) => Promise<DatabaseConfig<DB>>;
-
-export type { DatabaseFactory as D };