UNPKG

drizzle-orm

Version:

Drizzle ORM package for SQL databases

74 lines 2.17 kB
import { SQL } from "bun"; import { entityKind } from "../entity.js"; import { DefaultLogger } from "../logger.js"; import { PgDatabase } from "../pg-core/db.js"; import { PgDialect } from "../pg-core/dialect.js"; import { createTableRelationsHelpers, extractTablesRelationalConfig } from "../relations.js"; import { isConfig } from "../utils.js"; import { BunSQLSession } from "./session.js"; class BunSQLDatabase extends PgDatabase { static [entityKind] = "BunSQLDatabase"; } function construct(client, config = {}) { const dialect = new PgDialect({ casing: config.casing }); let logger; if (config.logger === true) { logger = new DefaultLogger(); } else if (config.logger !== false) { logger = config.logger; } let schema; if (config.schema) { const tablesConfig = extractTablesRelationalConfig( config.schema, createTableRelationsHelpers ); schema = { fullSchema: config.schema, schema: tablesConfig.tables, tableNamesMap: tablesConfig.tableNamesMap }; } const session = new BunSQLSession(client, dialect, schema, { logger }); const db = new BunSQLDatabase(dialect, session, schema); db.$client = client; return db; } function drizzle(...params) { if (typeof params[0] === "string") { const instance = new SQL(params[0]); return construct(instance, params[1]); } if (isConfig(params[0])) { const { connection, client, ...drizzleConfig } = params[0]; if (client) return construct(client, drizzleConfig); if (typeof connection === "object" && connection.url !== void 0) { const { url, ...config } = connection; const instance2 = new SQL({ url, ...config }); return construct(instance2, drizzleConfig); } const instance = new SQL(connection); return construct(instance, drizzleConfig); } return construct(params[0], params[1]); } ((drizzle2) => { function mock(config) { return construct({ options: { parsers: {}, serializers: {} } }, config); } drizzle2.mock = mock; })(drizzle || (drizzle = {})); export { BunSQLDatabase, drizzle }; //# sourceMappingURL=driver.js.map