| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768 |
- import Client from "better-sqlite3";
- 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 { BetterSQLiteSession } from "./session.js";
- class BetterSQLite3Database extends BaseSQLiteDatabase {
- static [entityKind] = "BetterSQLite3Database";
- }
- 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 BetterSQLiteSession(client, dialect, schema, { logger });
- const db = new BetterSQLite3Database("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 Client() : new Client(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, ...options } = connection;
- const instance2 = new Client(source, options);
- return construct(instance2, drizzleConfig);
- }
- const instance = new Client(connection);
- return construct(instance, drizzleConfig);
- }
- return construct(params[0], params[1]);
- }
- ((drizzle2) => {
- function mock(config) {
- return construct({}, config);
- }
- drizzle2.mock = mock;
- })(drizzle || (drizzle = {}));
- export {
- BetterSQLite3Database,
- drizzle
- };
- //# sourceMappingURL=driver.js.map
|