driver.js 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. import Client from "better-sqlite3";
  2. import { entityKind } from "../entity.js";
  3. import { DefaultLogger } from "../logger.js";
  4. import {
  5. createTableRelationsHelpers,
  6. extractTablesRelationalConfig
  7. } from "../relations.js";
  8. import { BaseSQLiteDatabase } from "../sqlite-core/db.js";
  9. import { SQLiteSyncDialect } from "../sqlite-core/dialect.js";
  10. import { isConfig } from "../utils.js";
  11. import { BetterSQLiteSession } from "./session.js";
  12. class BetterSQLite3Database extends BaseSQLiteDatabase {
  13. static [entityKind] = "BetterSQLite3Database";
  14. }
  15. function construct(client, config = {}) {
  16. const dialect = new SQLiteSyncDialect({ casing: config.casing });
  17. let logger;
  18. if (config.logger === true) {
  19. logger = new DefaultLogger();
  20. } else if (config.logger !== false) {
  21. logger = config.logger;
  22. }
  23. let schema;
  24. if (config.schema) {
  25. const tablesConfig = extractTablesRelationalConfig(
  26. config.schema,
  27. createTableRelationsHelpers
  28. );
  29. schema = {
  30. fullSchema: config.schema,
  31. schema: tablesConfig.tables,
  32. tableNamesMap: tablesConfig.tableNamesMap
  33. };
  34. }
  35. const session = new BetterSQLiteSession(client, dialect, schema, { logger });
  36. const db = new BetterSQLite3Database("sync", dialect, session, schema);
  37. db.$client = client;
  38. return db;
  39. }
  40. function drizzle(...params) {
  41. if (params[0] === void 0 || typeof params[0] === "string") {
  42. const instance = params[0] === void 0 ? new Client() : new Client(params[0]);
  43. return construct(instance, params[1]);
  44. }
  45. if (isConfig(params[0])) {
  46. const { connection, client, ...drizzleConfig } = params[0];
  47. if (client) return construct(client, drizzleConfig);
  48. if (typeof connection === "object") {
  49. const { source, ...options } = connection;
  50. const instance2 = new Client(source, options);
  51. return construct(instance2, drizzleConfig);
  52. }
  53. const instance = new Client(connection);
  54. return construct(instance, drizzleConfig);
  55. }
  56. return construct(params[0], params[1]);
  57. }
  58. ((drizzle2) => {
  59. function mock(config) {
  60. return construct({}, config);
  61. }
  62. drizzle2.mock = mock;
  63. })(drizzle || (drizzle = {}));
  64. export {
  65. BetterSQLite3Database,
  66. drizzle
  67. };
  68. //# sourceMappingURL=driver.js.map