AK-21/Graphite-Industrial-Intelligence

1"use strict";2var __defProp = Object.defineProperty;3var __getOwnPropDesc = Object.getOwnPropertyDescriptor;4var __getOwnPropNames = Object.getOwnPropertyNames;5var __hasOwnProp = Object.prototype.hasOwnProperty;6var __export = (target, all) => {7  for (var name in all)8    __defProp(target, name, { get: all[name], enumerable: true });9};10var __copyProps = (to, from, except, desc) => {11  if (from && typeof from === "object" || typeof from === "function") {12    for (let key of __getOwnPropNames(from))13      if (!__hasOwnProp.call(to, key) && key !== except)14        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });15  }16  return to;17};18var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);19var driver_exports = {};20__export(driver_exports, {21  VercelPgDatabase: () => VercelPgDatabase,22  VercelPgDriver: () => VercelPgDriver,23  drizzle: () => drizzle24});25module.exports = __toCommonJS(driver_exports);26var import_postgres = require("@vercel/postgres");27var import_entity = require("../entity.cjs");28var import_logger = require("../logger.cjs");29var import_db = require("../pg-core/db.cjs");30var import_pg_core = require("../pg-core/index.cjs");31var import_relations = require("../relations.cjs");32var import_utils = require("../utils.cjs");33var import_session = require("./session.cjs");34class VercelPgDriver {35  constructor(client, dialect, options = {}) {36    this.client = client;37    this.dialect = dialect;38    this.options = options;39  }40  static [import_entity.entityKind] = "VercelPgDriver";41  createSession(schema) {42    return new import_session.VercelPgSession(this.client, this.dialect, schema, {43      logger: this.options.logger,44      cache: this.options.cache45    });46  }47}48class VercelPgDatabase extends import_db.PgDatabase {49  static [import_entity.entityKind] = "VercelPgDatabase";50}51function construct(client, config = {}) {52  const dialect = new import_pg_core.PgDialect({ casing: config.casing });53  let logger;54  if (config.logger === true) {55    logger = new import_logger.DefaultLogger();56  } else if (config.logger !== false) {57    logger = config.logger;58  }59  let schema;60  if (config.schema) {61    const tablesConfig = (0, import_relations.extractTablesRelationalConfig)(62      config.schema,63      import_relations.createTableRelationsHelpers64    );65    schema = {66      fullSchema: config.schema,67      schema: tablesConfig.tables,68      tableNamesMap: tablesConfig.tableNamesMap69    };70  }71  const driver = new VercelPgDriver(client, dialect, { logger, cache: config.cache });72  const session = driver.createSession(schema);73  const db = new VercelPgDatabase(dialect, session, schema);74  db.$client = client;75  db.$cache = config.cache;76  if (db.$cache) {77    db.$cache["invalidate"] = config.cache?.onMutate;78  }79  return db;80}81function drizzle(...params) {82  if ((0, import_utils.isConfig)(params[0])) {83    const { client, ...drizzleConfig } = params[0];84    return construct(client ?? import_postgres.sql, drizzleConfig);85  }86  return construct(params[0] ?? import_postgres.sql, params[1]);87}88((drizzle2) => {89  function mock(config) {90    return construct({}, config);91  }92  drizzle2.mock = mock;93})(drizzle || (drizzle = {}));94// Annotate the CommonJS export names for ESM import in node:950 && (module.exports = {96  VercelPgDatabase,97  VercelPgDriver,98  drizzle99});100//# sourceMappingURL=driver.cjs.map