UNPKG

drizzle-orm

Version:

Drizzle ORM package for SQL databases

70 lines 2.31 kB
import { Database } from "bun:sqlite"; import { entityKind } from "../entity.js"; import { DefaultLogger } from "../logger.js"; import { createTableRelationsHelpers, extractTablesRelationalConfig } from "../relations.js"; import { BaseSQLiteDatabase } from "../sqlite-core/db.js"; import { SQLiteSyncDialect } from "../sqlite-core/dialect.js"; import { isConfig } from "../utils.js"; import { SQLiteBunSession } from "./session.js"; class BunSQLiteDatabase extends BaseSQLiteDatabase { static [entityKind] = "BunSQLiteDatabase"; } function construct(client, config = {}) { const dialect = new SQLiteSyncDialect({ 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 SQLiteBunSession(client, dialect, schema, { logger }); const db = new BunSQLiteDatabase("sync", dialect, session, schema); db.$client = client; return db; } function drizzle(...params) { if (params[0] === void 0 || typeof params[0] === "string") { const instance = params[0] === void 0 ? new Database() : new Database(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") { const { source, ...opts } = connection; const options = Object.values(opts).filter((v) => v !== void 0).length ? opts : void 0; const instance2 = new Database(source, options); return construct(instance2, drizzleConfig); } const instance = new Database(connection); return construct(instance, drizzleConfig); } return construct(params[0], params[1]); } ((drizzle2) => { function mock(config) { return construct({}, config); } drizzle2.mock = mock; })(drizzle || (drizzle = {})); export { BunSQLiteDatabase, drizzle }; //# sourceMappingURL=driver.js.map