driver.js 2.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. import { Database } from "bun:sqlite";
  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 { SQLiteBunSession } from "./session.js";
  12. class BunSQLiteDatabase extends BaseSQLiteDatabase {
  13. static [entityKind] = "BunSQLiteDatabase";
  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 SQLiteBunSession(client, dialect, schema, { logger });
  36. const db = new BunSQLiteDatabase("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 Database() : new Database(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, ...opts } = connection;
  50. const options = Object.values(opts).filter((v) => v !== void 0).length ? opts : void 0;
  51. const instance2 = new Database(source, options);
  52. return construct(instance2, drizzleConfig);
  53. }
  54. const instance = new Database(connection);
  55. return construct(instance, drizzleConfig);
  56. }
  57. return construct(params[0], params[1]);
  58. }
  59. ((drizzle2) => {
  60. function mock(config) {
  61. return construct({}, config);
  62. }
  63. drizzle2.mock = mock;
  64. })(drizzle || (drizzle = {}));
  65. export {
  66. BunSQLiteDatabase,
  67. drizzle
  68. };
  69. //# sourceMappingURL=driver.js.map