@erfanium/drizzle-orm 0.1.0 → 1.0.1

package/netlify-db/driver.cjs

@@ -1,160 +1,106 @@
-"use strict";
-var __create = Object.create;
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __getProtoOf = Object.getPrototypeOf;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+const require_runtime = require('../_virtual/_rolldown/runtime.cjs');
+const require_netlify_db_session = require('./session.cjs');
+const require_netlify_db_codecs = require('./codecs.cjs');
+let __entity_ts = require("../entity.cjs");
+let __utils_ts = require("../utils.cjs");
+let __logger_ts = require("../logger.cjs");
+let __pg_core_async_db_ts = require("../pg-core/async/db.cjs");
+let __pg_core_dialect_ts = require("../pg-core/dialect.cjs");
+let _neondatabase_serverless = require("@neondatabase/serverless");
+let __node_postgres_driver_ts = require("../node-postgres/driver.cjs");
+
+//#region src/netlify-db/driver.ts
+var NetlifyDbDatabase = class NetlifyDbDatabase extends __pg_core_async_db_ts.PgAsyncDatabase {
+	static [__entity_ts.entityKind] = "NetlifyDbDatabase";
+	$withAuth(token) {
+		const session = new require_netlify_db_session.NetlifyDbSession(this.session.httpClient, this.session.pool, this.dialect, this._.relations, {
+			...this.session.options,
+			authToken: token
+		});
+		return new NetlifyDbDatabase(this.dialect, session, this._.relations);
+	}
+	async batch(batch) {
+		return this.session.batch(batch);
+	}
 };
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
-  // If the importer is in node compatibility mode or this is not an ESM
-  // file that has been converted to a CommonJS file using a Babel-
-  // compatible transform (i.e. "__esModule" has not been set), then set
-  // "default" to the CommonJS "module.exports" for node compatibility.
-  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
-  mod
-));
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-var driver_exports = {};
-__export(driver_exports, {
-  NetlifyDbDatabase: () => NetlifyDbDatabase,
-  NetlifyDbDriver: () => NetlifyDbDriver,
-  drizzle: () => drizzle
-});
-module.exports = __toCommonJS(driver_exports);
-var import_serverless = require("@neondatabase/serverless");
-var import_db = require("@netlify/db");
-var import_entity = require("../entity.cjs");
-var import_logger = require("../logger.cjs");
-var import_driver = require("../node-postgres/driver.cjs");
-var import_db2 = require("../pg-core/db.cjs");
-var import_dialect = require("../pg-core/dialect.cjs");
-var V1 = __toESM(require("../relations.cjs"), 1);
-var import_utils = require("../utils.cjs");
-var import_session = require("./session.cjs");
-class NetlifyDbDriver {
-  constructor(httpClient, pool, dialect, options = {}) {
-    this.httpClient = httpClient;
-    this.pool = pool;
-    this.dialect = dialect;
-    this.options = options;
-    this.initMappers();
-  }
-  static [import_entity.entityKind] = "NetlifyDbDriver";
-  createSession(schema) {
-    return new import_session.NetlifyDbSession(this.httpClient, this.pool, this.dialect, schema, {
-      logger: this.options.logger,
-      cache: this.options.cache
-    });
-  }
-  initMappers() {
-    import_serverless.types.setTypeParser(import_serverless.types.builtins.TIMESTAMPTZ, (val) => val);
-    import_serverless.types.setTypeParser(import_serverless.types.builtins.TIMESTAMP, (val) => val);
-    import_serverless.types.setTypeParser(import_serverless.types.builtins.DATE, (val) => val);
-    import_serverless.types.setTypeParser(import_serverless.types.builtins.INTERVAL, (val) => val);
-    import_serverless.types.setTypeParser(1231, (val) => val);
-    import_serverless.types.setTypeParser(1115, (val) => val);
-    import_serverless.types.setTypeParser(1185, (val) => val);
-    import_serverless.types.setTypeParser(1187, (val) => val);
-    import_serverless.types.setTypeParser(1182, (val) => val);
-  }
-}
-class NetlifyDbDatabase extends import_db2.PgDatabase {
-  static [import_entity.entityKind] = "NetlifyDbDatabase";
-  async batch(batch) {
-    return this.session.batch(
-      batch
-    );
-  }
-}
 function construct(httpClient, pool, config = {}) {
-  const dialect = new import_dialect.PgDialect({ casing: config.casing });
-  let logger;
-  if (config.logger === true) {
-    logger = new import_logger.DefaultLogger();
-  } else if (config.logger !== false) {
-    logger = config.logger;
-  }
-  let schema;
-  if (config.schema) {
-    const tablesConfig = V1.extractTablesRelationalConfig(
-      config.schema,
-      V1.createTableRelationsHelpers
-    );
-    schema = {
-      fullSchema: config.schema,
-      schema: tablesConfig.tables,
-      tableNamesMap: tablesConfig.tableNamesMap
-    };
-  }
-  const driver = new NetlifyDbDriver(httpClient, pool, dialect, { logger, cache: config.cache });
-  const session = driver.createSession(schema);
-  const db = new NetlifyDbDatabase(
-    dialect,
-    session,
-    schema
-  );
-  db.$client = { http: httpClient, pool };
-  db.$cache = config.cache;
-  if (db.$cache) {
-    db.$cache["invalidate"] = config.cache?.onMutate;
-  }
-  return db;
+	const dialect = new __pg_core_dialect_ts.PgDialect({
+		codecs: config.codecs ?? require_netlify_db_codecs.netlifyDbCodecs,
+		useJitMappers: (0, __utils_ts.jitCompatCheck)(config.jit)
+	});
+	let logger;
+	if (config.logger === true) logger = new __logger_ts.DefaultLogger();
+	else if (config.logger !== false) logger = config.logger;
+	const relations = config.relations ?? {};
+	const session = new require_netlify_db_session.NetlifyDbSession(httpClient, pool, dialect, relations ?? {}, {
+		logger,
+		cache: config.cache,
+		transactionCodecs: config.transactionCodecs ?? require_netlify_db_codecs.netlifyDbTransactionCodecs
+	});
+	_neondatabase_serverless.types.setTypeParser(_neondatabase_serverless.types.builtins.TIMESTAMPTZ, (val) => val);
+	_neondatabase_serverless.types.setTypeParser(_neondatabase_serverless.types.builtins.TIMESTAMP, (val) => val);
+	_neondatabase_serverless.types.setTypeParser(_neondatabase_serverless.types.builtins.DATE, (val) => val);
+	_neondatabase_serverless.types.setTypeParser(_neondatabase_serverless.types.builtins.INTERVAL, (val) => val);
+	_neondatabase_serverless.types.setTypeParser(1231, (val) => val);
+	_neondatabase_serverless.types.setTypeParser(1115, (val) => val);
+	_neondatabase_serverless.types.setTypeParser(1185, (val) => val);
+	_neondatabase_serverless.types.setTypeParser(1187, (val) => val);
+	_neondatabase_serverless.types.setTypeParser(1182, (val) => val);
+	const db = new NetlifyDbDatabase(dialect, session, relations);
+	db.$client = {
+		http: httpClient,
+		pool
+	};
+	db.$cache = config.cache;
+	if (db.$cache) db.$cache["invalidate"] = config.cache?.onMutate;
+	return db;
 }
 function drizzle(...params) {
-  if (params.length === 0 || params.length === 1 && (0, import_utils.isConfig)(params[0]) && !("connection" in params[0]) && !("client" in params[0])) {
-    const drizzleConfig = params[0] ?? {};
-    const connection = (0, import_db.getDatabase)();
-    if (connection.driver === "server") {
-      return (0, import_driver.drizzle)({ client: connection.pool, ...drizzleConfig });
-    }
-    return construct(connection.httpClient, connection.pool, drizzleConfig);
-  }
-  if (typeof params[0] === "string") {
-    const connectionString = params[0];
-    const httpClient = (0, import_serverless.neon)(connectionString);
-    const pool = new import_serverless.Pool({ connectionString });
-    return construct(httpClient, pool, params[1]);
-  }
-  if ((0, import_utils.isConfig)(params[0])) {
-    const { connection, client, ...drizzleConfig } = params[0];
-    if (client) {
-      if ("driver" in client) {
-        if (client.driver === "serverless") {
-          return construct(client.httpClient, client.pool, drizzleConfig);
-        }
-        return (0, import_driver.drizzle)({ client: client.pool, ...drizzleConfig });
-      }
-      return construct(client.http, client.pool, drizzleConfig);
-    }
-    const connectionString = typeof connection === "string" ? connection : connection.connectionString;
-    const httpClient = (0, import_serverless.neon)(connectionString);
-    const pool = new import_serverless.Pool({ connectionString });
-    return construct(httpClient, pool, drizzleConfig);
-  }
-  throw new Error("Invalid arguments. Expected a connection string or a config object.");
+	if (params.length === 0 || params.length === 1 && (0, __utils_ts.isConfig)(params[0]) && !("connection" in params[0]) && !("client" in params[0])) {
+		const drizzleConfig = params[0] ?? {};
+		const connectionString = process.env["NETLIFY_DB_URL"];
+		if (!connectionString) throw new Error("NETLIFY_DB_URL environment variable is not set. Provide a connection string or client to drizzle().");
+		if (process.env["NETLIFY_DB_DRIVER"] === "server") return (0, __node_postgres_driver_ts.drizzle)({
+			connection: connectionString,
+			...drizzleConfig
+		});
+		return construct((0, _neondatabase_serverless.neon)(connectionString), new _neondatabase_serverless.Pool({ connectionString }), drizzleConfig);
+	}
+	if (typeof params[0] === "string") {
+		const connectionString = params[0];
+		return construct((0, _neondatabase_serverless.neon)(connectionString), new _neondatabase_serverless.Pool({ connectionString }), params[1]);
+	}
+	if ((0, __utils_ts.isConfig)(params[0])) {
+		const { connection, client, ...drizzleConfig } = params[0];
+		if (client) {
+			if ("driver" in client) {
+				if (client.driver === "serverless") return construct(client.httpClient, client.pool, drizzleConfig);
+				return (0, __node_postgres_driver_ts.drizzle)({
+					client: client.pool,
+					...drizzleConfig
+				});
+			}
+			return construct(client.http, client.pool, drizzleConfig);
+		}
+		const connectionString = typeof connection === "string" ? connection : connection.connectionString;
+		return construct((0, _neondatabase_serverless.neon)(connectionString), new _neondatabase_serverless.Pool({ connectionString }), drizzleConfig);
+	}
+	throw new Error("Invalid arguments. Expected a connection string or a config object.");
 }
-((drizzle2) => {
-  function mock(config) {
-    return construct({}, {}, config);
-  }
-  drizzle2.mock = mock;
+(function(_drizzle) {
+	function mock(config) {
+		return construct({}, {}, config);
+	}
+	_drizzle.mock = mock;
 })(drizzle || (drizzle = {}));
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  NetlifyDbDatabase,
-  NetlifyDbDriver,
-  drizzle
+
+//#endregion
+exports.NetlifyDbDatabase = NetlifyDbDatabase;
+Object.defineProperty(exports, 'drizzle', {
+  enumerable: true,
+  get: function () {
+    return drizzle;
+  }
 });
 //# sourceMappingURL=driver.cjs.map

package/netlify-db/driver.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/netlify-db/driver.ts"],"sourcesContent":["import { neon, Pool, types } from '@neondatabase/serverless';\nimport { getDatabase } from '@netlify/db';\nimport type { BatchItem, BatchResponse } from '~/batch.ts';\nimport type { Cache } from '~/cache/core/cache.ts';\nimport { entityKind } from '~/entity.ts';\nimport type { Logger } from '~/logger.ts';\nimport { DefaultLogger } from '~/logger.ts';\nimport type { NeonHttpClient, NeonHttpQueryResultHKT } from '~/neon-http/session.ts';\nimport { drizzle as drizzleNodePg, type NodePgDatabase } from '~/node-postgres/driver.ts';\nimport type { NodePgClient } from '~/node-postgres/session.ts';\nimport { PgDatabase } from '~/pg-core/db.ts';\nimport { PgDialect } from '~/pg-core/dialect.ts';\nimport * as V1 from '~/relations.ts';\nimport { type DrizzleConfig, isConfig } from '~/utils.ts';\nimport { type NetlifyDbClient, NetlifyDbSession } from './session.ts';\n\nexport interface ServerlessDrizzleClient {\n\tdriver: 'serverless';\n\thttpClient: NeonHttpClient;\n\tpool: Pool;\n\tconnectionString: string;\n}\n\nexport interface ServerDrizzleClient {\n\tdriver: 'server';\n\tpool: NodePgClient;\n\tconnectionString: string;\n}\n\nexport type DrizzleClient = ServerlessDrizzleClient | ServerDrizzleClient;\n\nexport interface NetlifyDbDriverOptions {\n\tlogger?: Logger;\n\tcache?: Cache;\n}\n\nexport class NetlifyDbDriver {\n\tstatic readonly [entityKind]: string = 'NetlifyDbDriver';\n\n\tconstructor(\n\t\tprivate httpClient: NeonHttpClient,\n\t\tprivate pool: Pool,\n\t\tprivate dialect: PgDialect,\n\t\tprivate options: NetlifyDbDriverOptions = {},\n\t) {\n\t\tthis.initMappers();\n\t}\n\n\tcreateSession(\n\t\tschema: V1.RelationalSchemaConfig<V1.TablesRelationalConfig> | undefined,\n\t): NetlifyDbSession<Record<string, unknown>, V1.TablesRelationalConfig> {\n\t\treturn new NetlifyDbSession(this.httpClient, this.pool, this.dialect, schema, {\n\t\t\tlogger: this.options.logger,\n\t\t\tcache: this.options.cache,\n\t\t});\n\t}\n\n\tinitMappers() {\n\t\ttypes.setTypeParser(types.builtins.TIMESTAMPTZ, (val: any) => val);\n\t\ttypes.setTypeParser(types.builtins.TIMESTAMP, (val: any) => val);\n\t\ttypes.setTypeParser(types.builtins.DATE, (val: any) => val);\n\t\ttypes.setTypeParser(types.builtins.INTERVAL, (val: any) => val);\n\t\ttypes.setTypeParser(1231, (val: any) => val);\n\t\ttypes.setTypeParser(1115, (val: any) => val);\n\t\ttypes.setTypeParser(1185, (val: any) => val);\n\t\ttypes.setTypeParser(1187, (val: any) => val);\n\t\ttypes.setTypeParser(1182, (val: any) => val);\n\t}\n}\n\nexport class NetlifyDbDatabase<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n> extends PgDatabase<NeonHttpQueryResultHKT, TSchema> {\n\tstatic override readonly [entityKind]: string = 'NetlifyDbDatabase';\n\n\tasync batch<U extends BatchItem<'pg'>, T extends Readonly<[U, ...U[]]>>(\n\t\tbatch: T,\n\t): Promise<BatchResponse<T>> {\n\t\treturn (<NetlifyDbSession<TSchema, V1.ExtractTablesWithRelations<TSchema>>> this.session).batch(\n\t\t\tbatch,\n\t\t) as Promise<BatchResponse<T>>;\n\t}\n}\n\nfunction construct<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n>(\n\thttpClient: NeonHttpClient,\n\tpool: Pool,\n\tconfig: DrizzleConfig<TSchema> = {},\n): NetlifyDbDatabase<TSchema> & {\n\t$client: NetlifyDbClient;\n} {\n\tconst dialect = new PgDialect({ casing: config.casing });\n\tlet logger;\n\tif (config.logger === true) {\n\t\tlogger = new DefaultLogger();\n\t} else if (config.logger !== false) {\n\t\tlogger = config.logger;\n\t}\n\n\tlet schema: V1.RelationalSchemaConfig<V1.TablesRelationalConfig> | undefined;\n\tif (config.schema) {\n\t\tconst tablesConfig = V1.extractTablesRelationalConfig(\n\t\t\tconfig.schema,\n\t\t\tV1.createTableRelationsHelpers,\n\t\t);\n\t\tschema = {\n\t\t\tfullSchema: config.schema,\n\t\t\tschema: tablesConfig.tables,\n\t\t\ttableNamesMap: tablesConfig.tableNamesMap,\n\t\t};\n\t}\n\n\tconst driver = new NetlifyDbDriver(httpClient, pool, dialect, { logger, cache: config.cache });\n\tconst session = driver.createSession(schema);\n\n\tconst db = new NetlifyDbDatabase(\n\t\tdialect,\n\t\tsession,\n\t\tschema as V1.RelationalSchemaConfig<V1.ExtractTablesWithRelations<TSchema>> | undefined,\n\t);\n\t(<any> db).$client = { http: httpClient, pool } satisfies NetlifyDbClient;\n\t(<any> db).$cache = config.cache;\n\tif ((<any> db).$cache) {\n\t\t(<any> db).$cache['invalidate'] = config.cache?.onMutate;\n\t}\n\n\treturn db as any;\n}\n\nexport function drizzle<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n>():\n\t| (NetlifyDbDatabase<TSchema> & { $client: NetlifyDbClient })\n\t| (NodePgDatabase<TSchema> & { $client: NodePgClient });\n\nexport function drizzle<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n>(\n\tconfig: DrizzleConfig<TSchema>,\n):\n\t| (NetlifyDbDatabase<TSchema> & { $client: NetlifyDbClient })\n\t| (NodePgDatabase<TSchema> & { $client: NodePgClient });\n\nexport function drizzle<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n>(\n\tconfig: DrizzleConfig<TSchema> & { client: ServerlessDrizzleClient },\n): NetlifyDbDatabase<TSchema> & { $client: NetlifyDbClient };\n\nexport function drizzle<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n>(\n\tconfig: DrizzleConfig<TSchema> & { client: ServerDrizzleClient },\n): NodePgDatabase<TSchema> & { $client: NodePgClient };\n\nexport function drizzle<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n>(\n\tconfig: DrizzleConfig<TSchema> & { client: DrizzleClient },\n):\n\t| (NetlifyDbDatabase<TSchema> & { $client: NetlifyDbClient })\n\t| (NodePgDatabase<TSchema> & { $client: NodePgClient });\n\nexport function drizzle<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n>(\n\t...params: [\n\t\tstring,\n\t] | [\n\t\tstring,\n\t\tDrizzleConfig<TSchema>,\n\t] | [\n\t\t(\n\t\t\t& DrizzleConfig<TSchema>\n\t\t\t& ({\n\t\t\t\tconnection: string | { connectionString: string };\n\t\t\t} | {\n\t\t\t\tclient: NetlifyDbClient | DrizzleClient;\n\t\t\t})\n\t\t),\n\t]\n): NetlifyDbDatabase<TSchema> & { $client: NetlifyDbClient };\n\nexport function drizzle<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n>(\n\t...params: [] | [\n\t\tDrizzleConfig<TSchema>,\n\t] | [\n\t\tstring,\n\t] | [\n\t\tstring,\n\t\tDrizzleConfig<TSchema>,\n\t] | [\n\t\t(\n\t\t\t& DrizzleConfig<TSchema>\n\t\t\t& ({\n\t\t\t\tconnection: string | { connectionString: string };\n\t\t\t} | {\n\t\t\t\tclient: NetlifyDbClient | DrizzleClient;\n\t\t\t})\n\t\t),\n\t]\n):\n\t| (NetlifyDbDatabase<TSchema> & { $client: NetlifyDbClient })\n\t| (NodePgDatabase<TSchema> & { $client: NodePgClient })\n{\n\t// Zero-config: read env vars\n\tif (\n\t\tparams.length === 0\n\t\t|| (params.length === 1 && isConfig(params[0]) && !('connection' in (params[0] as any))\n\t\t\t&& !('client' in (params[0] as any)))\n\t) {\n\t\tconst drizzleConfig = (params[0] ?? {}) as DrizzleConfig<TSchema>;\n\t\tconst connection = getDatabase();\n\n\t\tif (connection.driver === 'server') {\n\t\t\treturn drizzleNodePg({ client: connection.pool, ...drizzleConfig }) as any;\n\t\t}\n\n\t\treturn construct(connection.httpClient, connection.pool, drizzleConfig) as any;\n\t}\n\n\tif (typeof params[0] === 'string') {\n\t\tconst connectionString = params[0];\n\t\tconst httpClient = neon(connectionString);\n\t\tconst pool = new Pool({ connectionString });\n\t\treturn construct(httpClient, pool, params[1]) as any;\n\t}\n\n\tif (isConfig(params[0])) {\n\t\tconst { connection, client, ...drizzleConfig } = params[0] as {\n\t\t\tconnection?: { connectionString: string } | string;\n\t\t\tclient?: NetlifyDbClient | DrizzleClient;\n\t\t} & DrizzleConfig<TSchema>;\n\n\t\tif (client) {\n\t\t\tif ('driver' in client) {\n\t\t\t\tif (client.driver === 'serverless') {\n\t\t\t\t\treturn construct(client.httpClient, client.pool, drizzleConfig);\n\t\t\t\t}\n\t\t\t\treturn drizzleNodePg({ client: client.pool, ...drizzleConfig }) as any;\n\t\t\t}\n\t\t\treturn construct(client.http, client.pool, drizzleConfig);\n\t\t}\n\n\t\tconst connectionString = typeof connection === 'string' ? connection : connection!.connectionString;\n\t\tconst httpClient = neon(connectionString);\n\t\tconst pool = new Pool({ connectionString });\n\n\t\treturn construct(httpClient, pool, drizzleConfig) as any;\n\t}\n\n\tthrow new Error('Invalid arguments. Expected a connection string or a config object.');\n}\n\nexport namespace drizzle {\n\texport function mock<\n\t\tTSchema extends Record<string, unknown> = Record<string, never>,\n\t>(\n\t\tconfig?: DrizzleConfig<TSchema>,\n\t): NetlifyDbDatabase<TSchema> & {\n\t\t$client: '$client is not available on drizzle.mock()';\n\t} {\n\t\treturn construct({} as any, {} as any, config) as any;\n\t}\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,wBAAkC;AAClC,gBAA4B;AAG5B,oBAA2B;AAE3B,oBAA8B;AAE9B,oBAA8D;AAE9D,IAAAA,aAA2B;AAC3B,qBAA0B;AAC1B,SAAoB;AACpB,mBAA6C;AAC7C,qBAAuD;AAsBhD,MAAM,gBAAgB;AAAA,EAG5B,YACS,YACA,MACA,SACA,UAAkC,CAAC,GAC1C;AAJO;AACA;AACA;AACA;AAER,SAAK,YAAY;AAAA,EAClB;AAAA,EATA,QAAiB,wBAAU,IAAY;AAAA,EAWvC,cACC,QACuE;AACvE,WAAO,IAAI,gCAAiB,KAAK,YAAY,KAAK,MAAM,KAAK,SAAS,QAAQ;AAAA,MAC7E,QAAQ,KAAK,QAAQ;AAAA,MACrB,OAAO,KAAK,QAAQ;AAAA,IACrB,CAAC;AAAA,EACF;AAAA,EAEA,cAAc;AACb,4BAAM,cAAc,wBAAM,SAAS,aAAa,CAAC,QAAa,GAAG;AACjE,4BAAM,cAAc,wBAAM,SAAS,WAAW,CAAC,QAAa,GAAG;AAC/D,4BAAM,cAAc,wBAAM,SAAS,MAAM,CAAC,QAAa,GAAG;AAC1D,4BAAM,cAAc,wBAAM,SAAS,UAAU,CAAC,QAAa,GAAG;AAC9D,4BAAM,cAAc,MAAM,CAAC,QAAa,GAAG;AAC3C,4BAAM,cAAc,MAAM,CAAC,QAAa,GAAG;AAC3C,4BAAM,cAAc,MAAM,CAAC,QAAa,GAAG;AAC3C,4BAAM,cAAc,MAAM,CAAC,QAAa,GAAG;AAC3C,4BAAM,cAAc,MAAM,CAAC,QAAa,GAAG;AAAA,EAC5C;AACD;AAEO,MAAM,0BAEH,sBAA4C;AAAA,EACrD,QAA0B,wBAAU,IAAY;AAAA,EAEhD,MAAM,MACL,OAC4B;AAC5B,WAA4E,KAAK,QAAS;AAAA,MACzF;AAAA,IACD;AAAA,EACD;AACD;AAEA,SAAS,UAGR,YACA,MACA,SAAiC,CAAC,GAGjC;AACD,QAAM,UAAU,IAAI,yBAAU,EAAE,QAAQ,OAAO,OAAO,CAAC;AACvD,MAAI;AACJ,MAAI,OAAO,WAAW,MAAM;AAC3B,aAAS,IAAI,4BAAc;AAAA,EAC5B,WAAW,OAAO,WAAW,OAAO;AACnC,aAAS,OAAO;AAAA,EACjB;AAEA,MAAI;AACJ,MAAI,OAAO,QAAQ;AAClB,UAAM,eAAe,GAAG;AAAA,MACvB,OAAO;AAAA,MACP,GAAG;AAAA,IACJ;AACA,aAAS;AAAA,MACR,YAAY,OAAO;AAAA,MACnB,QAAQ,aAAa;AAAA,MACrB,eAAe,aAAa;AAAA,IAC7B;AAAA,EACD;AAEA,QAAM,SAAS,IAAI,gBAAgB,YAAY,MAAM,SAAS,EAAE,QAAQ,OAAO,OAAO,MAAM,CAAC;AAC7F,QAAM,UAAU,OAAO,cAAc,MAAM;AAE3C,QAAM,KAAK,IAAI;AAAA,IACd;AAAA,IACA;AAAA,IACA;AAAA,EACD;AACA,EAAO,GAAI,UAAU,EAAE,MAAM,YAAY,KAAK;AAC9C,EAAO,GAAI,SAAS,OAAO;AAC3B,MAAW,GAAI,QAAQ;AACtB,IAAO,GAAI,OAAO,YAAY,IAAI,OAAO,OAAO;AAAA,EACjD;AAEA,SAAO;AACR;AAwDO,SAAS,WAGZ,QAoBJ;AAEC,MACC,OAAO,WAAW,KACd,OAAO,WAAW,SAAK,uBAAS,OAAO,CAAC,CAAC,KAAK,EAAE,gBAAiB,OAAO,CAAC,MACzE,EAAE,YAAa,OAAO,CAAC,IAC1B;AACD,UAAM,gBAAiB,OAAO,CAAC,KAAK,CAAC;AACrC,UAAM,iBAAa,uBAAY;AAE/B,QAAI,WAAW,WAAW,UAAU;AACnC,iBAAO,cAAAC,SAAc,EAAE,QAAQ,WAAW,MAAM,GAAG,cAAc,CAAC;AAAA,IACnE;AAEA,WAAO,UAAU,WAAW,YAAY,WAAW,MAAM,aAAa;AAAA,EACvE;AAEA,MAAI,OAAO,OAAO,CAAC,MAAM,UAAU;AAClC,UAAM,mBAAmB,OAAO,CAAC;AACjC,UAAM,iBAAa,wBAAK,gBAAgB;AACxC,UAAM,OAAO,IAAI,uBAAK,EAAE,iBAAiB,CAAC;AAC1C,WAAO,UAAU,YAAY,MAAM,OAAO,CAAC,CAAC;AAAA,EAC7C;AAEA,UAAI,uBAAS,OAAO,CAAC,CAAC,GAAG;AACxB,UAAM,EAAE,YAAY,QAAQ,GAAG,cAAc,IAAI,OAAO,CAAC;AAKzD,QAAI,QAAQ;AACX,UAAI,YAAY,QAAQ;AACvB,YAAI,OAAO,WAAW,cAAc;AACnC,iBAAO,UAAU,OAAO,YAAY,OAAO,MAAM,aAAa;AAAA,QAC/D;AACA,mBAAO,cAAAA,SAAc,EAAE,QAAQ,OAAO,MAAM,GAAG,cAAc,CAAC;AAAA,MAC/D;AACA,aAAO,UAAU,OAAO,MAAM,OAAO,MAAM,aAAa;AAAA,IACzD;AAEA,UAAM,mBAAmB,OAAO,eAAe,WAAW,aAAa,WAAY;AACnF,UAAM,iBAAa,wBAAK,gBAAgB;AACxC,UAAM,OAAO,IAAI,uBAAK,EAAE,iBAAiB,CAAC;AAE1C,WAAO,UAAU,YAAY,MAAM,aAAa;AAAA,EACjD;AAEA,QAAM,IAAI,MAAM,qEAAqE;AACtF;AAAA,CAEO,CAAUC,aAAV;AACC,WAAS,KAGf,QAGC;AACD,WAAO,UAAU,CAAC,GAAU,CAAC,GAAU,MAAM;AAAA,EAC9C;AARO,EAAAA,SAAS;AAAA,GADA;","names":["import_db","drizzleNodePg","drizzle"]}
+{"version":3,"file":"driver.cjs","names":["PgAsyncDatabase","entityKind","NetlifyDbSession","PgDialect","netlifyDbCodecs","DefaultLogger","netlifyDbTransactionCodecs","types","Pool"],"sources":["../../src/netlify-db/driver.ts"],"sourcesContent":["import { type HTTPQueryOptions, neon, Pool, types } from '@neondatabase/serverless';\nimport type { BatchItem, BatchResponse } from '~/batch.ts';\nimport { entityKind } from '~/entity.ts';\nimport { DefaultLogger } from '~/logger.ts';\nimport type { NeonHttpClient, NeonHttpQueryResultHKT } from '~/neon-http/session.ts';\nimport { drizzle as drizzleNodePg, type NodePgDatabase } from '~/node-postgres/driver.ts';\nimport type { NodePgClient } from '~/node-postgres/session.ts';\nimport { PgAsyncDatabase } from '~/pg-core/async/db.ts';\nimport type { PgCodecs } from '~/pg-core/codecs.ts';\nimport { PgDialect } from '~/pg-core/dialect.ts';\nimport type { DrizzlePgConfig } from '~/pg-core/utils.ts';\nimport type { AnyRelations, EmptyRelations } from '~/relations.ts';\nimport { isConfig, jitCompatCheck } from '~/utils.ts';\nimport { netlifyDbCodecs, netlifyDbTransactionCodecs } from './codecs.ts';\nimport { type NetlifyDbClient, NetlifyDbSession } from './session.ts';\n\nexport interface ServerlessDrizzleClient {\n\tdriver: 'serverless';\n\thttpClient: NeonHttpClient;\n\tpool: Pool;\n\tconnectionString: string;\n}\n\nexport interface ServerDrizzleClient {\n\tdriver: 'server';\n\tpool: NodePgClient;\n\tconnectionString: string;\n}\n\nexport type DrizzleClient = ServerlessDrizzleClient | ServerDrizzleClient;\n\nexport interface DrizzleNetlifyConfig<TRelations extends AnyRelations = EmptyRelations>\n\textends DrizzlePgConfig<TRelations>\n{\n\t/** Netlify utilizes different driver for transactions, thus requiring separate set of codecs */\n\ttransactionCodecs?: PgCodecs | undefined;\n}\n\nexport class NetlifyDbDatabase<\n\tTRelations extends AnyRelations = EmptyRelations,\n> extends PgAsyncDatabase<NeonHttpQueryResultHKT, TRelations> {\n\tstatic override readonly [entityKind]: string = 'NetlifyDbDatabase';\n\n\tdeclare session: NetlifyDbSession<TRelations>;\n\n\t$withAuth(\n\t\ttoken: Exclude<HTTPQueryOptions<true, true>['authToken'], undefined>,\n\t): Omit<this, '$withAuth'> {\n\t\tconst session = new NetlifyDbSession(this.session.httpClient, this.session.pool, this.dialect, this._.relations, {\n\t\t\t...this.session.options,\n\t\t\tauthToken: token,\n\t\t});\n\n\t\treturn new NetlifyDbDatabase(this.dialect, session, this._.relations) as any;\n\t}\n\n\tasync batch<U extends BatchItem<'pg'>, T extends Readonly<[U, ...U[]]>>(\n\t\tbatch: T,\n\t): Promise<BatchResponse<T>> {\n\t\treturn this.session.batch(\n\t\t\tbatch,\n\t\t) as Promise<BatchResponse<T>>;\n\t}\n}\n\nfunction construct<\n\tTRelations extends AnyRelations = EmptyRelations,\n>(\n\thttpClient: NeonHttpClient,\n\tpool: Pool,\n\tconfig: DrizzleNetlifyConfig<TRelations> = {},\n): NetlifyDbDatabase<TRelations> & {\n\t$client: NetlifyDbClient;\n} {\n\tconst dialect = new PgDialect({\n\t\tcodecs: config.codecs ?? netlifyDbCodecs,\n\t\tuseJitMappers: jitCompatCheck(config.jit),\n\t});\n\n\tlet logger;\n\tif (config.logger === true) {\n\t\tlogger = new DefaultLogger();\n\t} else if (config.logger !== false) {\n\t\tlogger = config.logger;\n\t}\n\tconst relations = config.relations ?? {} as TRelations;\n\n\tconst session = new NetlifyDbSession(httpClient, pool, dialect, relations ?? {}, {\n\t\tlogger: logger,\n\t\tcache: config.cache,\n\t\ttransactionCodecs: config.transactionCodecs ?? netlifyDbTransactionCodecs,\n\t});\n\n\ttypes.setTypeParser(types.builtins.TIMESTAMPTZ, (val) => val);\n\ttypes.setTypeParser(types.builtins.TIMESTAMP, (val) => val);\n\ttypes.setTypeParser(types.builtins.DATE, (val) => val);\n\ttypes.setTypeParser(types.builtins.INTERVAL, (val) => val);\n\ttypes.setTypeParser(1231, (val) => val);\n\ttypes.setTypeParser(1115, (val) => val);\n\ttypes.setTypeParser(1185, (val) => val);\n\ttypes.setTypeParser(1187, (val) => val);\n\ttypes.setTypeParser(1182, (val) => val);\n\n\tconst db = new NetlifyDbDatabase(\n\t\tdialect,\n\t\tsession,\n\t\trelations,\n\t);\n\t(<any> db).$client = { http: httpClient, pool } satisfies NetlifyDbClient;\n\t(<any> db).$cache = config.cache;\n\tif ((<any> db).$cache) {\n\t\t(<any> db).$cache['invalidate'] = config.cache?.onMutate;\n\t}\n\n\treturn db as any;\n}\n\nexport function drizzle<\n\tTRelations extends AnyRelations = EmptyRelations,\n>():\n\t| (NetlifyDbDatabase<TRelations> & { $client: NetlifyDbClient })\n\t| (NodePgDatabase<TRelations> & { $client: NodePgClient });\n\nexport function drizzle<\n\tTRelations extends AnyRelations = EmptyRelations,\n>(\n\tconfig: DrizzleNetlifyConfig<TRelations>,\n):\n\t| (NetlifyDbDatabase<TRelations> & { $client: NetlifyDbClient })\n\t| (NodePgDatabase<TRelations> & { $client: NodePgClient });\n\nexport function drizzle<\n\tTRelations extends AnyRelations = EmptyRelations,\n>(\n\tconfig: DrizzleNetlifyConfig<TRelations> & { client: ServerlessDrizzleClient },\n): NetlifyDbDatabase<TRelations> & { $client: NetlifyDbClient };\n\nexport function drizzle<\n\tTRelations extends AnyRelations = EmptyRelations,\n>(\n\tconfig: DrizzleNetlifyConfig<TRelations> & { client: ServerDrizzleClient },\n): NodePgDatabase<TRelations> & { $client: NodePgClient };\n\nexport function drizzle<\n\tTRelations extends AnyRelations = EmptyRelations,\n>(\n\tconfig: DrizzleNetlifyConfig<TRelations> & { client: DrizzleClient },\n):\n\t| (NetlifyDbDatabase<TRelations> & { $client: NetlifyDbClient })\n\t| (NodePgDatabase<TRelations> & { $client: NodePgClient });\n\nexport function drizzle<\n\tTRelations extends AnyRelations = EmptyRelations,\n>(\n\t...params: [\n\t\tstring,\n\t] | [\n\t\tstring,\n\t\tDrizzleNetlifyConfig<TRelations>,\n\t] | [\n\t\t(\n\t\t\t& DrizzleNetlifyConfig<TRelations>\n\t\t\t& ({\n\t\t\t\tconnection: string | { connectionString: string };\n\t\t\t} | {\n\t\t\t\tclient: NetlifyDbClient | DrizzleClient;\n\t\t\t})\n\t\t),\n\t]\n): NetlifyDbDatabase<TRelations> & { $client: NetlifyDbClient };\n\nexport function drizzle<TRelations extends AnyRelations = EmptyRelations>(\n\t...params: [] | [\n\t\tDrizzleNetlifyConfig<TRelations>,\n\t] | [\n\t\tstring,\n\t] | [\n\t\tstring,\n\t\tDrizzleNetlifyConfig<TRelations>,\n\t] | [\n\t\t(\n\t\t\t& DrizzleNetlifyConfig<TRelations>\n\t\t\t& ({\n\t\t\t\tconnection: string | { connectionString: string };\n\t\t\t} | {\n\t\t\t\tclient: NetlifyDbClient | DrizzleClient;\n\t\t\t})\n\t\t),\n\t]\n):\n\t| (NetlifyDbDatabase<TRelations> & { $client: NetlifyDbClient })\n\t| (NodePgDatabase<TRelations> & { $client: NodePgClient })\n{\n\t// Zero-config: read env vars\n\tif (\n\t\tparams.length === 0\n\t\t|| (params.length === 1 && isConfig(params[0]) && !('connection' in (params[0] as any))\n\t\t\t&& !('client' in (params[0] as any)))\n\t) {\n\t\tconst drizzleConfig = (params[0] ?? {}) as DrizzleNetlifyConfig<TRelations>;\n\t\tconst connectionString = process.env['NETLIFY_DB_URL'];\n\n\t\tif (!connectionString) {\n\t\t\tthrow new Error(\n\t\t\t\t'NETLIFY_DB_URL environment variable is not set. '\n\t\t\t\t\t+ 'Provide a connection string or client to drizzle().',\n\t\t\t);\n\t\t}\n\n\t\tconst driver = process.env['NETLIFY_DB_DRIVER'];\n\n\t\tif (driver === 'server') {\n\t\t\treturn drizzleNodePg({ connection: connectionString, ...drizzleConfig }) as any;\n\t\t}\n\n\t\tconst httpClient = neon(connectionString);\n\t\tconst pool = new Pool({ connectionString });\n\t\treturn construct(httpClient, pool, drizzleConfig) as any;\n\t}\n\n\tif (typeof params[0] === 'string') {\n\t\tconst connectionString = params[0];\n\t\tconst httpClient = neon(connectionString);\n\t\tconst pool = new Pool({ connectionString });\n\t\treturn construct(httpClient, pool, params[1]) as any;\n\t}\n\n\tif (isConfig(params[0])) {\n\t\tconst { connection, client, ...drizzleConfig } = params[0] as {\n\t\t\tconnection?: { connectionString: string } | string;\n\t\t\tclient?: NetlifyDbClient | DrizzleClient;\n\t\t} & DrizzleNetlifyConfig<TRelations>;\n\n\t\tif (client) {\n\t\t\tif ('driver' in client) {\n\t\t\t\tif (client.driver === 'serverless') {\n\t\t\t\t\treturn construct(client.httpClient, client.pool, drizzleConfig);\n\t\t\t\t}\n\t\t\t\treturn drizzleNodePg({ client: client.pool, ...drizzleConfig }) as any;\n\t\t\t}\n\t\t\treturn construct(client.http, client.pool, drizzleConfig);\n\t\t}\n\n\t\tconst connectionString = typeof connection === 'string' ? connection : connection!.connectionString;\n\t\tconst httpClient = neon(connectionString);\n\t\tconst pool = new Pool({ connectionString });\n\n\t\treturn construct(httpClient, pool, drizzleConfig) as any;\n\t}\n\n\tthrow new Error('Invalid arguments. Expected a connection string or a config object.');\n}\n\nexport namespace drizzle {\n\texport function mock<\n\t\tTRelations extends AnyRelations = EmptyRelations,\n\t>(\n\t\tconfig?: DrizzleNetlifyConfig<TRelations>,\n\t): NetlifyDbDatabase<TRelations> & {\n\t\t$client: '$client is not available on drizzle.mock()';\n\t} {\n\t\treturn construct({} as any, {} as any, config) as any;\n\t}\n}\n"],"mappings":";;;;;;;;;;;;;AAsCA,IAAa,oBAAb,MAAa,0BAEHA,sCAAoD;CAC7D,QAA0BC,0BAAsB;CAIhD,UACC,OAC0B;EAC1B,MAAM,UAAU,IAAIC,4CAAiB,KAAK,QAAQ,YAAY,KAAK,QAAQ,MAAM,KAAK,SAAS,KAAK,EAAE,WAAW;GAChH,GAAG,KAAK,QAAQ;GAChB,WAAW;GACX,CAAC;AAEF,SAAO,IAAI,kBAAkB,KAAK,SAAS,SAAS,KAAK,EAAE,UAAU;;CAGtE,MAAM,MACL,OAC4B;AAC5B,SAAO,KAAK,QAAQ,MACnB,MACA;;;AAIH,SAAS,UAGR,YACA,MACA,SAA2C,EAAE,EAG5C;CACD,MAAM,UAAU,IAAIC,+BAAU;EAC7B,QAAQ,OAAO,UAAUC;EACzB,8CAA8B,OAAO,IAAI;EACzC,CAAC;CAEF,IAAI;AACJ,KAAI,OAAO,WAAW,KACrB,UAAS,IAAIC,2BAAe;UAClB,OAAO,WAAW,MAC5B,UAAS,OAAO;CAEjB,MAAM,YAAY,OAAO,aAAa,EAAE;CAExC,MAAM,UAAU,IAAIH,4CAAiB,YAAY,MAAM,SAAS,aAAa,EAAE,EAAE;EACxE;EACR,OAAO,OAAO;EACd,mBAAmB,OAAO,qBAAqBI;EAC/C,CAAC;AAEF,gCAAM,cAAcC,+BAAM,SAAS,cAAc,QAAQ,IAAI;AAC7D,gCAAM,cAAcA,+BAAM,SAAS,YAAY,QAAQ,IAAI;AAC3D,gCAAM,cAAcA,+BAAM,SAAS,OAAO,QAAQ,IAAI;AACtD,gCAAM,cAAcA,+BAAM,SAAS,WAAW,QAAQ,IAAI;AAC1D,gCAAM,cAAc,OAAO,QAAQ,IAAI;AACvC,gCAAM,cAAc,OAAO,QAAQ,IAAI;AACvC,gCAAM,cAAc,OAAO,QAAQ,IAAI;AACvC,gCAAM,cAAc,OAAO,QAAQ,IAAI;AACvC,gCAAM,cAAc,OAAO,QAAQ,IAAI;CAEvC,MAAM,KAAK,IAAI,kBACd,SACA,SACA,UACA;AACD,CAAO,GAAI,UAAU;EAAE,MAAM;EAAY;EAAM;AAC/C,CAAO,GAAI,SAAS,OAAO;AAC3B,KAAW,GAAI,OACd,CAAO,GAAI,OAAO,gBAAgB,OAAO,OAAO;AAGjD,QAAO;;AAyDR,SAAgB,QACf,GAAG,QAoBJ;AAEC,KACC,OAAO,WAAW,KACd,OAAO,WAAW,8BAAc,OAAO,GAAG,IAAI,EAAE,gBAAiB,OAAO,OACxE,EAAE,YAAa,OAAO,KACzB;EACD,MAAM,gBAAiB,OAAO,MAAM,EAAE;EACtC,MAAM,mBAAmB,QAAQ,IAAI;AAErC,MAAI,CAAC,iBACJ,OAAM,IAAI,MACT,sGAEA;AAKF,MAFe,QAAQ,IAAI,yBAEZ,SACd,+CAAqB;GAAE,YAAY;GAAkB,GAAG;GAAe,CAAC;AAKzE,SAAO,6CAFiB,iBAAiB,EAC5B,IAAIC,8BAAK,EAAE,kBAAkB,CAAC,EACR,cAAc;;AAGlD,KAAI,OAAO,OAAO,OAAO,UAAU;EAClC,MAAM,mBAAmB,OAAO;AAGhC,SAAO,6CAFiB,iBAAiB,EAC5B,IAAIA,8BAAK,EAAE,kBAAkB,CAAC,EACR,OAAO,GAAG;;AAG9C,8BAAa,OAAO,GAAG,EAAE;EACxB,MAAM,EAAE,YAAY,QAAQ,GAAG,kBAAkB,OAAO;AAKxD,MAAI,QAAQ;AACX,OAAI,YAAY,QAAQ;AACvB,QAAI,OAAO,WAAW,aACrB,QAAO,UAAU,OAAO,YAAY,OAAO,MAAM,cAAc;AAEhE,kDAAqB;KAAE,QAAQ,OAAO;KAAM,GAAG;KAAe,CAAC;;AAEhE,UAAO,UAAU,OAAO,MAAM,OAAO,MAAM,cAAc;;EAG1D,MAAM,mBAAmB,OAAO,eAAe,WAAW,aAAa,WAAY;AAInF,SAAO,6CAHiB,iBAAiB,EAC5B,IAAIA,8BAAK,EAAE,kBAAkB,CAAC,EAER,cAAc;;AAGlD,OAAM,IAAI,MAAM,sEAAsE;;;CAI/E,SAAS,KAGf,QAGC;AACD,SAAO,UAAU,EAAE,EAAS,EAAE,EAAS,OAAO"}

package/netlify-db/driver.d.cts

@@ -1,16 +1,14 @@
-import { Pool } from '@neondatabase/serverless';
-import type { BatchItem, BatchResponse } from "../batch.cjs";
-import type { Cache } from "../cache/core/cache.cjs";
-import { entityKind } from "../entity.cjs";
-import type { Logger } from "../logger.cjs";
-import type { NeonHttpClient, NeonHttpQueryResultHKT } from "../neon-http/session.cjs";
-import { type NodePgDatabase } from "../node-postgres/driver.cjs";
-import type { NodePgClient } from "../node-postgres/session.cjs";
-import { PgDatabase } from "../pg-core/db.cjs";
-import { PgDialect } from "../pg-core/dialect.cjs";
-import * as V1 from "../relations.cjs";
-import { type DrizzleConfig } from "../utils.cjs";
-import { type NetlifyDbClient, NetlifyDbSession } from "./session.cjs";
+import { type HTTPQueryOptions, Pool } from '@neondatabase/serverless';
+import type { BatchItem, BatchResponse } from '../batch.cjs';
+import { entityKind } from '../entity.cjs';
+import type { NeonHttpClient, NeonHttpQueryResultHKT } from '../neon-http/session.cjs';
+import { type NodePgDatabase } from '../node-postgres/driver.cjs';
+import type { NodePgClient } from '../node-postgres/session.cjs';
+import { PgAsyncDatabase } from '../pg-core/async/db.cjs';
+import type { PgCodecs } from '../pg-core/codecs.cjs';
+import type { DrizzlePgConfig } from '../pg-core/utils.cjs';
+import type { AnyRelations, EmptyRelations } from '../relations.cjs';
+import { type NetlifyDbClient, NetlifyDbSession } from './session.cjs';
 export interface ServerlessDrizzleClient {
     driver: 'serverless';
     httpClient: NeonHttpClient;
@@ -23,69 +21,61 @@ export interface ServerDrizzleClient {
     connectionString: string;
 }
 export type DrizzleClient = ServerlessDrizzleClient | ServerDrizzleClient;
-export interface NetlifyDbDriverOptions {
-    logger?: Logger;
-    cache?: Cache;
+export interface DrizzleNetlifyConfig<TRelations extends AnyRelations = EmptyRelations> extends DrizzlePgConfig<TRelations> {
+    /** Netlify utilizes different driver for transactions, thus requiring separate set of codecs */
+    transactionCodecs?: PgCodecs | undefined;
 }
-export declare class NetlifyDbDriver {
-    private httpClient;
-    private pool;
-    private dialect;
-    private options;
-    static readonly [entityKind]: string;
-    constructor(httpClient: NeonHttpClient, pool: Pool, dialect: PgDialect, options?: NetlifyDbDriverOptions);
-    createSession(schema: V1.RelationalSchemaConfig<V1.TablesRelationalConfig> | undefined): NetlifyDbSession<Record<string, unknown>, V1.TablesRelationalConfig>;
-    initMappers(): void;
-}
-export declare class NetlifyDbDatabase<TSchema extends Record<string, unknown> = Record<string, never>> extends PgDatabase<NeonHttpQueryResultHKT, TSchema> {
+export declare class NetlifyDbDatabase<TRelations extends AnyRelations = EmptyRelations> extends PgAsyncDatabase<NeonHttpQueryResultHKT, TRelations> {
     static readonly [entityKind]: string;
+    session: NetlifyDbSession<TRelations>;
+    $withAuth(token: Exclude<HTTPQueryOptions<true, true>['authToken'], undefined>): Omit<this, '$withAuth'>;
     batch<U extends BatchItem<'pg'>, T extends Readonly<[U, ...U[]]>>(batch: T): Promise<BatchResponse<T>>;
 }
-export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>>(): (NetlifyDbDatabase<TSchema> & {
+export declare function drizzle<TRelations extends AnyRelations = EmptyRelations>(): (NetlifyDbDatabase<TRelations> & {
     $client: NetlifyDbClient;
-}) | (NodePgDatabase<TSchema> & {
+}) | (NodePgDatabase<TRelations> & {
     $client: NodePgClient;
 });
-export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>>(config: DrizzleConfig<TSchema>): (NetlifyDbDatabase<TSchema> & {
+export declare function drizzle<TRelations extends AnyRelations = EmptyRelations>(config: DrizzleNetlifyConfig<TRelations>): (NetlifyDbDatabase<TRelations> & {
     $client: NetlifyDbClient;
-}) | (NodePgDatabase<TSchema> & {
+}) | (NodePgDatabase<TRelations> & {
     $client: NodePgClient;
 });
-export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>>(config: DrizzleConfig<TSchema> & {
+export declare function drizzle<TRelations extends AnyRelations = EmptyRelations>(config: DrizzleNetlifyConfig<TRelations> & {
     client: ServerlessDrizzleClient;
-}): NetlifyDbDatabase<TSchema> & {
+}): NetlifyDbDatabase<TRelations> & {
     $client: NetlifyDbClient;
 };
-export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>>(config: DrizzleConfig<TSchema> & {
+export declare function drizzle<TRelations extends AnyRelations = EmptyRelations>(config: DrizzleNetlifyConfig<TRelations> & {
     client: ServerDrizzleClient;
-}): NodePgDatabase<TSchema> & {
+}): NodePgDatabase<TRelations> & {
     $client: NodePgClient;
 };
-export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>>(config: DrizzleConfig<TSchema> & {
+export declare function drizzle<TRelations extends AnyRelations = EmptyRelations>(config: DrizzleNetlifyConfig<TRelations> & {
     client: DrizzleClient;
-}): (NetlifyDbDatabase<TSchema> & {
+}): (NetlifyDbDatabase<TRelations> & {
     $client: NetlifyDbClient;
-}) | (NodePgDatabase<TSchema> & {
+}) | (NodePgDatabase<TRelations> & {
     $client: NodePgClient;
 });
-export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>>(...params: [
+export declare function drizzle<TRelations extends AnyRelations = EmptyRelations>(...params: [
     string
 ] | [
     string,
-    DrizzleConfig<TSchema>
+    DrizzleNetlifyConfig<TRelations>
 ] | [
-    (DrizzleConfig<TSchema> & ({
+    (DrizzleNetlifyConfig<TRelations> & ({
         connection: string | {
             connectionString: string;
         };
     } | {
         client: NetlifyDbClient | DrizzleClient;
     }))
-]): NetlifyDbDatabase<TSchema> & {
+]): NetlifyDbDatabase<TRelations> & {
     $client: NetlifyDbClient;
 };
 export declare namespace drizzle {
-    function mock<TSchema extends Record<string, unknown> = Record<string, never>>(config?: DrizzleConfig<TSchema>): NetlifyDbDatabase<TSchema> & {
+    function mock<TRelations extends AnyRelations = EmptyRelations>(config?: DrizzleNetlifyConfig<TRelations>): NetlifyDbDatabase<TRelations> & {
         $client: '$client is not available on drizzle.mock()';
     };
 }

package/netlify-db/driver.d.ts

@@ -1,16 +1,14 @@
-import { Pool } from '@neondatabase/serverless';
-import type { BatchItem, BatchResponse } from "../batch.js";
-import type { Cache } from "../cache/core/cache.js";
-import { entityKind } from "../entity.js";
-import type { Logger } from "../logger.js";
-import type { NeonHttpClient, NeonHttpQueryResultHKT } from "../neon-http/session.js";
-import { type NodePgDatabase } from "../node-postgres/driver.js";
-import type { NodePgClient } from "../node-postgres/session.js";
-import { PgDatabase } from "../pg-core/db.js";
-import { PgDialect } from "../pg-core/dialect.js";
-import * as V1 from "../relations.js";
-import { type DrizzleConfig } from "../utils.js";
-import { type NetlifyDbClient, NetlifyDbSession } from "./session.js";
+import { type HTTPQueryOptions, Pool } from '@neondatabase/serverless';
+import type { BatchItem, BatchResponse } from '../batch.js';
+import { entityKind } from '../entity.js';
+import type { NeonHttpClient, NeonHttpQueryResultHKT } from '../neon-http/session.js';
+import { type NodePgDatabase } from '../node-postgres/driver.js';
+import type { NodePgClient } from '../node-postgres/session.js';
+import { PgAsyncDatabase } from '../pg-core/async/db.js';
+import type { PgCodecs } from '../pg-core/codecs.js';
+import type { DrizzlePgConfig } from '../pg-core/utils.js';
+import type { AnyRelations, EmptyRelations } from '../relations.js';
+import { type NetlifyDbClient, NetlifyDbSession } from './session.js';
 export interface ServerlessDrizzleClient {
     driver: 'serverless';
     httpClient: NeonHttpClient;
@@ -23,69 +21,61 @@ export interface ServerDrizzleClient {
     connectionString: string;
 }
 export type DrizzleClient = ServerlessDrizzleClient | ServerDrizzleClient;
-export interface NetlifyDbDriverOptions {
-    logger?: Logger;
-    cache?: Cache;
+export interface DrizzleNetlifyConfig<TRelations extends AnyRelations = EmptyRelations> extends DrizzlePgConfig<TRelations> {
+    /** Netlify utilizes different driver for transactions, thus requiring separate set of codecs */
+    transactionCodecs?: PgCodecs | undefined;
 }
-export declare class NetlifyDbDriver {
-    private httpClient;
-    private pool;
-    private dialect;
-    private options;
-    static readonly [entityKind]: string;
-    constructor(httpClient: NeonHttpClient, pool: Pool, dialect: PgDialect, options?: NetlifyDbDriverOptions);
-    createSession(schema: V1.RelationalSchemaConfig<V1.TablesRelationalConfig> | undefined): NetlifyDbSession<Record<string, unknown>, V1.TablesRelationalConfig>;
-    initMappers(): void;
-}
-export declare class NetlifyDbDatabase<TSchema extends Record<string, unknown> = Record<string, never>> extends PgDatabase<NeonHttpQueryResultHKT, TSchema> {
+export declare class NetlifyDbDatabase<TRelations extends AnyRelations = EmptyRelations> extends PgAsyncDatabase<NeonHttpQueryResultHKT, TRelations> {
     static readonly [entityKind]: string;
+    session: NetlifyDbSession<TRelations>;
+    $withAuth(token: Exclude<HTTPQueryOptions<true, true>['authToken'], undefined>): Omit<this, '$withAuth'>;
     batch<U extends BatchItem<'pg'>, T extends Readonly<[U, ...U[]]>>(batch: T): Promise<BatchResponse<T>>;
 }
-export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>>(): (NetlifyDbDatabase<TSchema> & {
+export declare function drizzle<TRelations extends AnyRelations = EmptyRelations>(): (NetlifyDbDatabase<TRelations> & {
     $client: NetlifyDbClient;
-}) | (NodePgDatabase<TSchema> & {
+}) | (NodePgDatabase<TRelations> & {
     $client: NodePgClient;
 });
-export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>>(config: DrizzleConfig<TSchema>): (NetlifyDbDatabase<TSchema> & {
+export declare function drizzle<TRelations extends AnyRelations = EmptyRelations>(config: DrizzleNetlifyConfig<TRelations>): (NetlifyDbDatabase<TRelations> & {
     $client: NetlifyDbClient;
-}) | (NodePgDatabase<TSchema> & {
+}) | (NodePgDatabase<TRelations> & {
     $client: NodePgClient;
 });
-export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>>(config: DrizzleConfig<TSchema> & {
+export declare function drizzle<TRelations extends AnyRelations = EmptyRelations>(config: DrizzleNetlifyConfig<TRelations> & {
     client: ServerlessDrizzleClient;
-}): NetlifyDbDatabase<TSchema> & {
+}): NetlifyDbDatabase<TRelations> & {
     $client: NetlifyDbClient;
 };
-export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>>(config: DrizzleConfig<TSchema> & {
+export declare function drizzle<TRelations extends AnyRelations = EmptyRelations>(config: DrizzleNetlifyConfig<TRelations> & {
     client: ServerDrizzleClient;
-}): NodePgDatabase<TSchema> & {
+}): NodePgDatabase<TRelations> & {
     $client: NodePgClient;
 };
-export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>>(config: DrizzleConfig<TSchema> & {
+export declare function drizzle<TRelations extends AnyRelations = EmptyRelations>(config: DrizzleNetlifyConfig<TRelations> & {
     client: DrizzleClient;
-}): (NetlifyDbDatabase<TSchema> & {
+}): (NetlifyDbDatabase<TRelations> & {
     $client: NetlifyDbClient;
-}) | (NodePgDatabase<TSchema> & {
+}) | (NodePgDatabase<TRelations> & {
     $client: NodePgClient;
 });
-export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>>(...params: [
+export declare function drizzle<TRelations extends AnyRelations = EmptyRelations>(...params: [
     string
 ] | [
     string,
-    DrizzleConfig<TSchema>
+    DrizzleNetlifyConfig<TRelations>
 ] | [
-    (DrizzleConfig<TSchema> & ({
+    (DrizzleNetlifyConfig<TRelations> & ({
         connection: string | {
             connectionString: string;
         };
     } | {
         client: NetlifyDbClient | DrizzleClient;
     }))
-]): NetlifyDbDatabase<TSchema> & {
+]): NetlifyDbDatabase<TRelations> & {
     $client: NetlifyDbClient;
 };
 export declare namespace drizzle {
-    function mock<TSchema extends Record<string, unknown> = Record<string, never>>(config?: DrizzleConfig<TSchema>): NetlifyDbDatabase<TSchema> & {
+    function mock<TRelations extends AnyRelations = EmptyRelations>(config?: DrizzleNetlifyConfig<TRelations>): NetlifyDbDatabase<TRelations> & {
         $client: '$client is not available on drizzle.mock()';
     };
 }