@rvoh/dream 1.4.2 → 1.5.0

package/dist/cjs/src/dream/QueryDriver/Postgres.js

@@ -1,6 +1,168 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+// after building for esm, importing pg using the following:
+//
+//  import * as pg from 'pg'
+//
+// will crash. This is difficult to discover, since it only happens
+// when being imported from our esm build.
+// eslint-disable-next-line @typescript-eslint/ban-ts-comment
+// @ts-ignore
+const pg_1 = require("pg");
+const kysely_1 = require("kysely");
+const index_js_1 = require("../../cli/index.js");
+const dataTypes_js_1 = require("../../db/dataTypes.js");
+const index_js_2 = require("../../dream-app/index.js");
+const camelize_js_1 = require("../../helpers/camelize.js");
+const customPgParsers_js_1 = require("../../helpers/customPgParsers.js");
+const EnvInternal_js_1 = require("../../helpers/EnvInternal.js");
+const createDb_js_1 = require("./helpers/pg/createDb.js");
+const dropDb_js_1 = require("./helpers/pg/dropDb.js");
+const loadPgClient_js_1 = require("./helpers/pg/loadPgClient.js");
 const Kysely_js_1 = require("./Kysely.js");
+const pgTypes = pg_1.default.types;
 class PostgresQueryDriver extends Kysely_js_1.default {
+    /**
+     * create the database. Must respond to the NODE_ENV value.
+     */
+    static async dbCreate(connectionName) {
+        const dreamApp = index_js_2.default.getOrFail();
+        const primaryDbConf = dreamApp.dbConnectionConfig(connectionName, 'primary');
+        index_js_1.default.logger.logStartProgress(`creating ${primaryDbConf.name}...`);
+        await (0, createDb_js_1.default)(connectionName, 'primary');
+        index_js_1.default.logger.logEndProgress();
+        // TODO: add support for creating replicas. Began doing it below, but it is very tricky,
+        // and we don't need it at the moment, so kicking off for future development when we have more time
+        // to flesh this out.
+        // if (connectionRetriever.hasReplicaConfig(connectionName)) {
+        //   const replicaDbConf = connectionRetriever.getConnectionConf('replica')
+        //   console.log(`creating ${process.env[replicaDbConf.name]}`)
+        //   await createDb('replica')
+        // }
+    }
+    /**
+     * delete the database. Must respond to the NODE_ENV value.
+     */
+    static async dbDrop(connectionName) {
+        const dreamApp = index_js_2.default.getOrFail();
+        const primaryDbConf = dreamApp.dbConnectionConfig(connectionName, 'primary');
+        index_js_1.default.logger.logStartProgress(`dropping ${primaryDbConf.name}...`);
+        await (0, dropDb_js_1.default)(connectionName, 'primary');
+        index_js_1.default.logger.logEndProgress();
+        // TODO: add support for dropping replicas. Began doing it below, but it is very tricky,
+        // and we don't need it at the moment, so kicking off for future development when we have more time
+        // to flesh this out.
+        // if (connectionRetriever.hasReplicaConfig(connectionName)) {
+        //   const replicaDbConf = connectionRetriever.getConnectionConf('replica')
+        //   console.log(`dropping ${process.env[replicaDbConf.name]}`)
+        //   await _dropDb('replica')
+        // }
+    }
+    /**
+     * @internal
+     *
+     * this method is called when dream is initializing, and is used
+     * to configure the database to utilize custom type parsers for
+     * a variety of data types.
+     *
+     * @param connectionName - the name of the connection you are doing this for
+     * @returns void
+     */
+    static async setDatabaseTypeParsers(connectionName) {
+        const kyselyDb = this.dbFor(connectionName, 'primary');
+        pgTypes.setTypeParser(pgTypes.builtins.DATE, customPgParsers_js_1.parsePostgresDate);
+        pgTypes.setTypeParser(pgTypes.builtins.TIMESTAMP, customPgParsers_js_1.parsePostgresDatetime);
+        pgTypes.setTypeParser(pgTypes.builtins.TIMESTAMPTZ, customPgParsers_js_1.parsePostgresDatetime);
+        pgTypes.setTypeParser(pgTypes.builtins.NUMERIC, customPgParsers_js_1.parsePostgresDecimal);
+        pgTypes.setTypeParser(pgTypes.builtins.INT8, customPgParsers_js_1.parsePostgresBigint);
+        const textArrayOid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.TEXT);
+        if (textArrayOid) {
+            let oid;
+            const textArrayParser = pgTypes.getTypeParser(textArrayOid);
+            function transformPostgresArray(transformer) {
+                return (value) => textArrayParser(value).map(str => transformer(str));
+            }
+            const enumArrayOids = await (0, customPgParsers_js_1.findEnumArrayOids)(kyselyDb);
+            enumArrayOids.forEach((enumArrayOid) => pgTypes.setTypeParser(enumArrayOid, textArrayParser));
+            oid = await (0, customPgParsers_js_1.findCitextArrayOid)(kyselyDb);
+            if (oid)
+                pgTypes.setTypeParser(oid, textArrayParser);
+            oid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.UUID);
+            if (oid)
+                pgTypes.setTypeParser(oid, textArrayParser);
+            oid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.DATE);
+            if (oid)
+                pgTypes.setTypeParser(oid, transformPostgresArray(customPgParsers_js_1.parsePostgresDate));
+            oid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.TIMESTAMP);
+            if (oid)
+                pgTypes.setTypeParser(oid, transformPostgresArray(customPgParsers_js_1.parsePostgresDatetime));
+            oid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.TIMESTAMPTZ);
+            if (oid)
+                pgTypes.setTypeParser(oid, transformPostgresArray(customPgParsers_js_1.parsePostgresDatetime));
+            oid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.NUMERIC);
+            if (oid)
+                pgTypes.setTypeParser(oid, transformPostgresArray(customPgParsers_js_1.parsePostgresDecimal));
+            oid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.INT8);
+            if (oid)
+                pgTypes.setTypeParser(oid, transformPostgresArray(customPgParsers_js_1.parsePostgresBigint));
+        }
+    }
+    /**
+     * @internal
+     *
+     * this is used by the SchemaBuilder to store column data permanently
+     * within the types/dream.ts file.
+     */
+    static async getColumnData(connectionName, tableName, associationData) {
+        const db = this.dbFor(connectionName, 'primary');
+        const sqlQuery = (0, kysely_1.sql) `SELECT column_name, udt_name::regtype, is_nullable, data_type FROM information_schema.columns WHERE table_name = ${tableName}`;
+        const columnToDBTypeMap = await sqlQuery.execute(db);
+        const rows = columnToDBTypeMap.rows;
+        const columnData = {};
+        rows.forEach(row => {
+            const isEnum = ['USER-DEFINED', 'ARRAY'].includes(row.dataType) && !(0, dataTypes_js_1.isPrimitiveDataType)(row.udtName);
+            const isArray = ['ARRAY'].includes(row.dataType);
+            const associationMetadata = associationData[row.columnName];
+            columnData[(0, camelize_js_1.default)(row.columnName)] = {
+                dbType: row.udtName,
+                allowNull: row.isNullable === 'YES',
+                enumType: isEnum ? this.enumType(row) : null,
+                enumValues: isEnum ? `${this.enumType(row)}Values` : null,
+                isArray,
+                foreignKey: associationMetadata?.foreignKey || null,
+            };
+        });
+        return Object.keys(columnData)
+            .sort()
+            .reduce((acc, key) => {
+            if (columnData[key] === undefined)
+                return acc;
+            acc[key] = columnData[key];
+            return acc;
+        }, {});
+    }
+    static async duplicateDatabase(connectionName) {
+        const dreamApp = index_js_2.default.getOrFail();
+        const parallelTests = dreamApp.parallelTests;
+        if (!parallelTests)
+            return;
+        index_js_1.default.logger.logStartProgress(`duplicating db for parallel tests...`);
+        const dbConf = dreamApp.dbConnectionConfig(connectionName, 'primary');
+        const client = await (0, loadPgClient_js_1.default)({ useSystemDb: true, connectionName });
+        if (EnvInternal_js_1.default.boolean('DREAM_CORE_DEVELOPMENT')) {
+            const replicaTestWorkerDatabaseName = `replica_test_${dbConf.name}`;
+            index_js_1.default.logger.logContinueProgress(`creating fake replica test database ${replicaTestWorkerDatabaseName}...`, { logPrefix: '  ├ [db]', logPrefixColor: 'cyan' });
+            await client.query(`DROP DATABASE IF EXISTS ${replicaTestWorkerDatabaseName};`);
+            await client.query(`CREATE DATABASE ${replicaTestWorkerDatabaseName} TEMPLATE ${dbConf.name};`);
+        }
+        for (let i = 2; i <= parallelTests; i++) {
+            const workerDatabaseName = `${dbConf.name}_${i}`;
+            index_js_1.default.logger.logContinueProgress(`creating duplicate test database ${workerDatabaseName} for concurrent tests...`, { logPrefix: '  ├ [db]', logPrefixColor: 'cyan' });
+            await client.query(`DROP DATABASE IF EXISTS ${workerDatabaseName};`);
+            await client.query(`CREATE DATABASE ${workerDatabaseName} TEMPLATE ${dbConf.name};`);
+        }
+        await client.end();
+        index_js_1.default.logger.logEndProgress();
+    }
 }
 exports.default = PostgresQueryDriver;

package/dist/cjs/src/{helpers/db → dream/QueryDriver/helpers/kysely}/runMigration.js

@@ -4,15 +4,19 @@ exports.default = runMigration;
 const kysely_1 = require("kysely");
 const fs = require("node:fs/promises");
 const path = require("node:path");
-const index_js_1 = require("../../cli/index.js");
-const colorize_js_1 = require("../../cli/logger/loggable/colorize.js");
-const DreamDbConnection_js_1 = require("../../db/DreamDbConnection.js");
-const index_js_2 = require("../../db/index.js");
-const index_js_3 = require("../../dream-app/index.js");
-async function runMigration({ mode = 'migrate' } = {}) {
-    const dreamApp = index_js_3.default.getOrFail();
-    const migrationFolder = path.join(dreamApp.projectRoot, dreamApp.paths.db, 'migrations');
-    const kyselyDb = (0, index_js_2.default)('primary');
+const colorize_js_1 = require("../../../../cli/logger/loggable/colorize.js");
+const DreamDbConnection_js_1 = require("../../../../db/DreamDbConnection.js");
+const index_js_1 = require("../../../../db/index.js");
+const index_js_2 = require("../../../../dream-app/index.js");
+const index_js_3 = require("../../../../cli/index.js");
+async function runMigration({ connectionName, mode = 'migrate', dialectProvider, }) {
+    const dreamApp = index_js_2.default.getOrFail();
+    const migrationFolder = connectionName === 'default'
+        ? path.join(dreamApp.projectRoot, dreamApp.paths.db, 'migrations')
+        : path.join(dreamApp.projectRoot, dreamApp.paths.db, 'migrations', connectionName);
+    // Ensure the migration folder exists
+    await fs.mkdir(migrationFolder, { recursive: true });
+    const kyselyDb = (0, index_js_1.default)(connectionName, 'primary', dialectProvider);
     const migrator = new kysely_1.Migrator({
         db: kyselyDb,
         allowUnorderedMigrations: true,
@@ -28,7 +32,7 @@ async function runMigration({ mode = 'migrate' } = {}) {
     else if (mode === 'rollback') {
         await rollback(migrator);
     }
-    await DreamDbConnection_js_1.default.dropAllConnections();
+    await (0, DreamDbConnection_js_1.closeAllConnectionsForConnectionName)(connectionName);
 }
 async function migrate(migrator) {
     let nextMigrationRequiringNewTransaction = await findNextMigrationRequiringNewTransaction(migrator);
@@ -74,9 +78,9 @@ async function rollback(migrator) {
         await handleError(error, 'rollback');
 }
 async function handleError(error, mode) {
-    await DreamDbConnection_js_1.default.dropAllConnections();
-    index_js_3.default.logWithLevel('error', `failed to ${migratedActionCurrentTense(mode)}`);
-    index_js_3.default.logWithLevel('error', error);
+    await (0, DreamDbConnection_js_1.closeAllDbConnections)();
+    index_js_2.default.logWithLevel('error', `failed to ${migratedActionCurrentTense(mode)}`);
+    index_js_2.default.logWithLevel('error', error);
     process.exit(1);
 }
 function migratedActionCurrentTense(mode) {
@@ -88,12 +92,12 @@ function migratedActionPastTense(mode) {
 function logResults(results, mode) {
     results?.forEach(it => {
         if (it.status === 'Success') {
-            index_js_1.default.logger.logContinueProgress((0, colorize_js_1.default)(`[db]`, { color: 'cyan' }) +
+            index_js_3.default.logger.logContinueProgress((0, colorize_js_1.default)(`[db]`, { color: 'cyan' }) +
                 ` migration "${it.migrationName}" was ${migratedActionPastTense(mode)} successfully`, { logPrefixColor: 'cyan' });
         }
         else if (it.status === 'Error') {
-            index_js_1.default.logger.logContinueProgress(JSON.stringify(it, null, 2));
-            index_js_1.default.logger.logContinueProgress((0, colorize_js_1.default)(`failed to ${migratedActionCurrentTense(mode)} migration "${it.migrationName}"`, {
+            index_js_3.default.logger.logContinueProgress(JSON.stringify(it, null, 2));
+            index_js_3.default.logger.logContinueProgress((0, colorize_js_1.default)(`failed to ${migratedActionCurrentTense(mode)} migration "${it.migrationName}"`, {
                 color: 'redBright',
             }));
         }

package/dist/cjs/src/{helpers/db → dream/QueryDriver/helpers/pg}/createDb.js

@@ -1,20 +1,20 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = createDb;
-const index_js_1 = require("../../dream-app/index.js");
-const EnvInternal_js_1 = require("../EnvInternal.js");
+const index_js_1 = require("../../../../dream-app/index.js");
+const EnvInternal_js_1 = require("../../../../helpers/EnvInternal.js");
 const loadPgClient_js_1 = require("./loadPgClient.js");
-async function createDb(connection, dbName) {
+async function createDb(connectionName, connectionType, dbName) {
     // this was only ever written to clear the db between tests or in development,
     // so there is no way to drop in production
     if (EnvInternal_js_1.default.isProduction)
         return false;
     const dreamApp = index_js_1.default.getOrFail();
-    const dbConf = dreamApp.dbConnectionConfig(connection);
+    const dbConf = dreamApp.dbConnectionConfig(connectionName, connectionType);
     dbName ||= dbConf.name || null;
     if (!dbName)
         throw new Error('Must either pass a dbName to the create function, or else ensure that DB_NAME is set in the env');
-    const client = await (0, loadPgClient_js_1.default)({ useSystemDb: true });
+    const client = await (0, loadPgClient_js_1.default)({ useSystemDb: true, connectionName });
     await client.query(`CREATE DATABASE ${dbName};`);
     await client.end();
 }

package/dist/cjs/src/{helpers/db → dream/QueryDriver/helpers/pg}/dropDb.js

@@ -1,21 +1,21 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = dropDb;
-const index_js_1 = require("../../cli/index.js");
-const index_js_2 = require("../../dream-app/index.js");
-const EnvInternal_js_1 = require("../EnvInternal.js");
+const index_js_1 = require("../../../../cli/index.js");
+const index_js_2 = require("../../../../dream-app/index.js");
+const EnvInternal_js_1 = require("../../../../helpers/EnvInternal.js");
 const loadPgClient_js_1 = require("./loadPgClient.js");
-async function dropDb(connection, dbName) {
+async function dropDb(connectionName, connection, dbName) {
     // this was only ever written to clear the db between tests or in development,
     // so there is no way to drop in production
     if (EnvInternal_js_1.default.isProduction)
         return false;
     const dreamApp = index_js_2.default.getOrFail();
-    const dbConf = dreamApp.dbConnectionConfig(connection);
+    const dbConf = dreamApp.dbConnectionConfig(connectionName, connection);
     dbName ||= dbConf.name || null;
     if (!dbName)
         throw new Error('Must either pass a dbName to the drop function, or else ensure that DB_NAME is set in the env');
-    const client = await (0, loadPgClient_js_1.default)({ useSystemDb: true });
+    const client = await (0, loadPgClient_js_1.default)({ useSystemDb: true, connectionName });
     await maybeDropDuplicateDatabases(client, dbName);
     await client.query(`DROP DATABASE IF EXISTS ${dbName};`);
 }

package/dist/cjs/src/{helpers/db → dream/QueryDriver/helpers/pg}/loadPgClient.js

@@ -10,10 +10,12 @@ exports.default = loadPgClient;
 // eslint-disable-next-line @typescript-eslint/ban-ts-comment
 // @ts-ignore
 const pg_1 = require("pg");
-const index_js_1 = require("../../dream-app/index.js");
-async function loadPgClient({ useSystemDb } = {}) {
+const index_js_1 = require("../../../../dream-app/index.js");
+async function loadPgClient({ connectionName, useSystemDb, }) {
     const dreamconf = index_js_1.default.getOrFail();
-    const creds = dreamconf.dbCredentials.primary;
+    const creds = dreamconf.dbCredentialsFor(connectionName)?.primary;
+    if (!creds)
+        throw new Error(`failed to load db credentials for connection: ${connectionName}`);
     const client = new pg_1.default.Client({
         host: creds.host || 'localhost',
         port: creds.port,

package/dist/cjs/src/dream/internal/destroyDream.js

@@ -1,9 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = destroyDream;
+const index_js_1 = require("../../dream-app/index.js");
 const destroyAssociatedRecords_js_1 = require("./destroyAssociatedRecords.js");
 const runHooksFor_js_1 = require("./runHooksFor.js");
-const softDeleteDream_js_1 = require("./softDeleteDream.js");
 /**
  * @internal
  *
@@ -59,13 +59,6 @@ function shouldSoftDelete(dream, reallyDestroy) {
  * deleting by one of the beforeDestroy model hooks
  */
 async function maybeDestroyDream(dream, txn, reallyDestroy) {
-    if (shouldSoftDelete(dream, reallyDestroy)) {
-        await (0, softDeleteDream_js_1.default)(dream, txn);
-    }
-    else if (!dream['_preventDeletion']) {
-        await txn.kyselyTransaction
-            .deleteFrom(dream.table)
-            .where(dream['_primaryKey'], '=', dream.primaryKeyValue())
-            .execute();
-    }
+    const dbDriverClass = index_js_1.default.getOrFail().dbConnectionQueryDriverClass(dream.connectionName);
+    await dbDriverClass.destroyDream(dream, txn, reallyDestroy);
 }

package/dist/cjs/src/dream/internal/saveDream.js

@@ -31,7 +31,7 @@ async function saveDream(dream, txn = null, { skipHooks = false } = {}) {
     // BeforeSave/Update actions may clear all the data that we intended to save, leaving us with
     // an invalid update command. The Sortable decorator is an example of this.
     if (!alreadyPersisted || hasUnsavedData) {
-        const data = await Query_js_1.default.dbDriverClass().saveDream(dream, txn);
+        const data = await Query_js_1.default.dbDriverClass(dream.connectionName || 'default').saveDream(dream, txn);
         dream['isPersisted'] = true;
         dream.setAttributes(data);
     }

package/dist/cjs/src/dream/internal/similarity/SimilarityBuilder.js

@@ -11,6 +11,7 @@ const namespaceColumn_js_1 = require("../../../helpers/namespaceColumn.js");
 const constants_js_1 = require("../../constants.js");
 const similaritySelectSql_js_1 = require("./similaritySelectSql.js");
 const similarityWhereSql_js_1 = require("./similarityWhereSql.js");
+const setPosition_js_1 = require("../../../decorators/field/sortable/helpers/setPosition.js");
 class SimilarityBuilder extends ConnectedToDB_js_1.default {
     whereStatement;
     whereNotStatement;
@@ -217,8 +218,9 @@ class SimilarityBuilder extends ConnectedToDB_js_1.default {
         const schema = this.dreamClass.prototype.schema;
         const primaryKeyName = this.dreamClass.primaryKey;
         const { tableName, tableAlias, columnName } = similarityStatement;
+        const queryDriverClass = (0, setPosition_js_1.getPostgresQueryDriver)(this.dreamInstance.connectionName);
         // eslint-disable-next-line @typescript-eslint/unbound-method
-        const { ref } = this.dbFor('select').dynamic;
+        const { ref } = queryDriverClass.dbFor(this.dreamInstance.connectionName, this.dbConnectionType('select')).dynamic;
         const validatedTableAlias = (0, validateTableAlias_js_1.default)(tableAlias);
         const validatedPrimaryKey = (0, validateColumn_js_1.default)(schema, tableName, primaryKeyName);
         const nestedQuery = this.buildNestedSelectQuery({
@@ -248,7 +250,9 @@ class SimilarityBuilder extends ConnectedToDB_js_1.default {
             statementType,
         });
         const trigramSearchAlias = this.similaritySearchId(tableAlias, columnName);
-        const selectQuery = this.dbFor('select')
+        const queryDriverClass = (0, setPosition_js_1.getPostgresQueryDriver)(this.dreamInstance.connectionName);
+        const selectQuery = queryDriverClass
+            .dbFor(this.dreamInstance.connectionName, this.dbConnectionType('select'))
             .selectFrom(validatedTableAlias)
             .select(`${trigramSearchAlias}.trigram_search_id`)
             .innerJoin(nestedQuery.as(trigramSearchAlias), join => join.onRef((0, namespaceColumn_js_1.default)(validatedPrimaryKey, validatedTableAlias), '=', (0, namespaceColumn_js_1.default)('trigram_search_id', trigramSearchAlias)));
@@ -260,7 +264,9 @@ class SimilarityBuilder extends ConnectedToDB_js_1.default {
         const { columnName, opsStatement, tableName } = similarityStatement;
         const validatedTable = (0, validateTable_js_1.default)(schema, tableName);
         const validatedPrimaryKey = (0, validateColumn_js_1.default)(schema, tableName, primaryKeyName);
-        let nestedQuery = this.dbFor('select')
+        const queryDriverClass = (0, setPosition_js_1.getPostgresQueryDriver)(this.dreamInstance.connectionName);
+        let nestedQuery = queryDriverClass
+            .dbFor(this.dreamInstance.connectionName, this.dbConnectionType('select'))
             .selectFrom(tableName)
             .select(eb => {
             const tableNameRef = eb.ref(validatedTable);

package/dist/cjs/src/dream-app/index.js

@@ -1,29 +1,19 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.GeneratorImportStyles = exports.DreamAppAllowedPackageManagersEnumValues = void 0;
-// after building for esm, importing pg using the following:
-//
-//  import * as pg from 'pg'
-//
-// will crash. This is difficult to discover, since it only happens
-// when being imported from our esm build.
-// eslint-disable-next-line @typescript-eslint/ban-ts-comment
-// @ts-ignore
-const pg_1 = require("pg");
 const util = require("node:util");
-const index_js_1 = require("../db/index.js");
 const validateTable_js_1 = require("../db/validators/validateTable.js");
-const index_js_2 = require("../encrypt/index.js");
+const Query_js_1 = require("../dream/Query.js");
+const index_js_1 = require("../encrypt/index.js");
 const DreamAppInitMissingCallToLoadModels_js_1 = require("../errors/dream-app/DreamAppInitMissingCallToLoadModels.js");
 const DreamAppInitMissingMissingProjectRoot_js_1 = require("../errors/dream-app/DreamAppInitMissingMissingProjectRoot.js");
 const CalendarDate_js_1 = require("../helpers/CalendarDate.js");
-const customPgParsers_js_1 = require("../helpers/customPgParsers.js");
 const DateTime_js_1 = require("../helpers/DateTime.js");
 const EnvInternal_js_1 = require("../helpers/EnvInternal.js");
 const cache_js_1 = require("./cache.js");
 const importModels_js_1 = require("./helpers/importers/importModels.js");
 const importSerializers_js_1 = require("./helpers/importers/importSerializers.js");
-const pgTypes = pg_1.default.types;
+const Postgres_js_1 = require("../dream/QueryDriver/Postgres.js");
 // this needs to be done top-level to ensure proper configuration
 DateTime_js_1.Settings.defaultZone = 'UTC';
 class DreamApp {
@@ -47,7 +37,7 @@ class DreamApp {
             (0, importSerializers_js_1.setCachedSerializers)({});
         (0, cache_js_1.cacheDreamApp)(dreamApp);
         if (!EnvInternal_js_1.default.boolean('BYPASS_DB_CONNECTIONS_DURING_INIT'))
-            await this.setDatabaseTypeParsers();
+            await this.setDatabaseTypeParsers(dreamApp);
         await deferCb?.(dreamApp);
         for (const plugin of dreamApp.plugins) {
             await plugin(dreamApp);
@@ -90,51 +80,18 @@ class DreamApp {
      *
      *
      */
-    static async setDatabaseTypeParsers() {
-        const kyselyDb = (0, index_js_1.default)('primary');
-        pgTypes.setTypeParser(pgTypes.builtins.DATE, customPgParsers_js_1.parsePostgresDate);
-        pgTypes.setTypeParser(pgTypes.builtins.TIMESTAMP, customPgParsers_js_1.parsePostgresDatetime);
-        pgTypes.setTypeParser(pgTypes.builtins.TIMESTAMPTZ, customPgParsers_js_1.parsePostgresDatetime);
-        pgTypes.setTypeParser(pgTypes.builtins.NUMERIC, customPgParsers_js_1.parsePostgresDecimal);
-        pgTypes.setTypeParser(pgTypes.builtins.INT8, customPgParsers_js_1.parsePostgresBigint);
-        const textArrayOid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.TEXT);
-        if (textArrayOid) {
-            let oid;
-            const textArrayParser = pgTypes.getTypeParser(textArrayOid);
-            function transformPostgresArray(transformer) {
-                return (value) => textArrayParser(value).map(str => transformer(str));
-            }
-            const enumArrayOids = await (0, customPgParsers_js_1.findEnumArrayOids)(kyselyDb);
-            enumArrayOids.forEach((enumArrayOid) => pgTypes.setTypeParser(enumArrayOid, textArrayParser));
-            oid = await (0, customPgParsers_js_1.findCitextArrayOid)(kyselyDb);
-            if (oid)
-                pgTypes.setTypeParser(oid, textArrayParser);
-            oid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.UUID);
-            if (oid)
-                pgTypes.setTypeParser(oid, textArrayParser);
-            oid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.DATE);
-            if (oid)
-                pgTypes.setTypeParser(oid, transformPostgresArray(customPgParsers_js_1.parsePostgresDate));
-            oid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.TIMESTAMP);
-            if (oid)
-                pgTypes.setTypeParser(oid, transformPostgresArray(customPgParsers_js_1.parsePostgresDatetime));
-            oid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.TIMESTAMPTZ);
-            if (oid)
-                pgTypes.setTypeParser(oid, transformPostgresArray(customPgParsers_js_1.parsePostgresDatetime));
-            oid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.NUMERIC);
-            if (oid)
-                pgTypes.setTypeParser(oid, transformPostgresArray(customPgParsers_js_1.parsePostgresDecimal));
-            oid = await (0, customPgParsers_js_1.findCorrespondingArrayOid)(kyselyDb, pgTypes.builtins.INT8);
-            if (oid)
-                pgTypes.setTypeParser(oid, transformPostgresArray(customPgParsers_js_1.parsePostgresBigint));
+    static async setDatabaseTypeParsers(dreamApp) {
+        for (const connectionName of Object.keys(dreamApp._dbCredentials)) {
+            const dbDriverClass = Query_js_1.default.dbDriverClass(connectionName);
+            await dbDriverClass.setDatabaseTypeParsers(connectionName);
         }
     }
     static checkKey(encryptionIdentifier, key, algorithm) {
-        if (!index_js_2.default.validateKey(key, algorithm))
+        if (!index_js_1.default.validateKey(key, algorithm))
             console.warn(`
 Your current key value for ${encryptionIdentifier} encryption is invalid.
 Try setting it to something valid, like:
-  ${index_js_2.default.generateKey(algorithm)}
+  ${index_js_1.default.generateKey(algorithm)}
 
 (This was done by calling:
   Encrypt.generateKey('${algorithm}')
@@ -168,10 +125,15 @@ Try setting it to something valid, like:
     get specialHooks() {
         return this._specialHooks;
     }
-    _dbCredentials;
+    _dbCredentials = {};
     get dbCredentials() {
         return this._dbCredentials;
     }
+    dbCredentialsFor(connectionName) {
+        if (this._dbCredentials[connectionName])
+            return this._dbCredentials[connectionName];
+        return null;
+    }
     _encryption;
     get encryption() {
         return this._encryption;
@@ -227,7 +189,7 @@ Try setting it to something valid, like:
     loadedModels = false;
     constructor(opts) {
         if (opts?.db)
-            this._dbCredentials = opts.db;
+            this._dbCredentials['default'] = opts.db;
         if (opts?.primaryKeyType)
             this._primaryKeyType = opts.primaryKeyType;
         if (opts?.projectRoot)
@@ -254,22 +216,38 @@ Try setting it to something valid, like:
     get serializers() {
         return (0, importSerializers_js_1.getSerializersOrFail)();
     }
-    dbName(connection) {
-        const conf = this.dbConnectionConfig(connection);
+    dbName(connectionName, connection) {
+        const conf = this.dbConnectionConfig(connectionName, connection);
         return this.parallelDatabasesEnabled ? `${conf.name}_${process.env.VITEST_POOL_ID}` : conf.name;
     }
-    dbConnectionConfig(connection) {
-        const conf = this.dbCredentials?.[connection] || this.dbCredentials?.primary;
-        if (!conf)
+    dbConnectionConfig(connectionName, connection) {
+        const conf = this.dbCredentialsFor(connectionName)?.[connection] || this.dbCredentialsFor(connectionName)?.primary;
+        if (!conf) {
             throw new Error(`
       Cannot find a connection config given the following connection and node environment:
+        connectionName: ${connectionName}
         connection: ${connection}
         NODE_ENV: ${EnvInternal_js_1.default.nodeEnv}
     `);
+        }
         return conf;
     }
-    get hasReplicaConfig() {
-        return !!this.dbCredentials.replica;
+    dbConnectionQueryDriverClass(connectionName) {
+        const conf = this.dbCredentialsFor(connectionName);
+        if (!conf) {
+            throw new Error(`
+      Cannot find a db credentials for the given connectionName:
+        connectionName: ${connectionName}
+        NODE_ENV: ${EnvInternal_js_1.default.nodeEnv}
+    `);
+        }
+        return (conf.queryDriverClass) || (Postgres_js_1.default);
+    }
+    dbConnectionKeys() {
+        return Object.keys(this.dbCredentials);
+    }
+    hasReplicaConfig(connectionName) {
+        return !!this.dbCredentials[connectionName]?.replica;
     }
     get parallelDatabasesEnabled() {
         return (!!this.parallelTests &&
@@ -290,10 +268,15 @@ Try setting it to something valid, like:
     plugin(cb) {
         this._plugins.push(cb);
     }
-    set(applyOption, options) {
+    set(applyOption, options, secondaryOptions) {
         switch (applyOption) {
             case 'db':
-                this._dbCredentials = options;
+                if (typeof options === 'string') {
+                    this._dbCredentials[options] = secondaryOptions;
+                }
+                else {
+                    this._dbCredentials['default'] = options;
+                }
                 break;
             case 'encryption':
                 this._encryption = options;