@rvoh/dream 0.35.1 → 0.35.3

package/dist/cjs/src/Dream.js

@@ -37,8 +37,8 @@ const CannotCallUndestroyOnANonSoftDeleteModel_js_1 = require("./errors/CannotCa
 const ConstructorOnlyForInternalUse_js_1 = require("./errors/ConstructorOnlyForInternalUse.js");
 const CreateOrFindByFailedToCreateAndFind_js_1 = require("./errors/CreateOrFindByFailedToCreateAndFind.js");
 const GlobalNameNotSet_js_1 = require("./errors/dream-application/GlobalNameNotSet.js");
+const DreamMissingRequiredOverride_js_1 = require("./errors/DreamMissingRequiredOverride.js");
 const MissingSerializersDefinition_js_1 = require("./errors/MissingSerializersDefinition.js");
-const MissingTable_js_1 = require("./errors/MissingTable.js");
 const NonExistentScopeProvidedToResort_js_1 = require("./errors/NonExistentScopeProvidedToResort.js");
 const CalendarDate_js_1 = require("./helpers/CalendarDate.js");
 const cloneDeepSafe_js_1 = require("./helpers/cloneDeepSafe.js");
@@ -54,14 +54,14 @@ class Dream {
      * @internal
      *
      * This getter will throw an error when developers use .toEqual instead of
-     * useToMatchDreamModels or useToMatchDreamModel in a jest spec. This
+     * useToMatchDreamModels or useToMatchDreamModel in a vite spec. This
      * must be the first getter in the class in order for this to work, so don't move it.
      *
      */
     get _useToMatchDreamModels() {
         throw new Error(`
       Hi there! It looks like you're trying to compare a Dream model in
-      a Jest expectation using \`toEqual\`. That won't work.
+      a Vite expectation using \`toEqual\`. That won't work.
       Instead, use \`toMatchDreamModel\` or \`toMatchDreamModels\`.
 
       For example, instead of:
@@ -99,10 +99,10 @@ class Dream {
      */
     static globallyInitializingDecorators = false;
     get schema() {
-        throw new Error('Must define schema getter in ApplicationModel');
+        throw new DreamMissingRequiredOverride_js_1.default(this.constructor, 'schema');
     }
     get globalSchema() {
-        throw new Error('Must define schema getter in ApplicationModel');
+        throw new DreamMissingRequiredOverride_js_1.default(this.constructor, 'globalSchema');
     }
     /**
      * Shadows #primaryKey, a getter which can be overwritten to customize the id field
@@ -1234,18 +1234,7 @@ class Dream {
      * @returns A Kysely query. Depending on the type passed, it will return either a SelectQueryBuilder, DeleteQueryBuilder, UpdateQueryBuilder, or an InsertQueryBuilder
      */
     static toKysely(type) {
-        switch (type) {
-            case 'select':
-                return this.query().dbFor('select').selectFrom(this.table);
-            case 'delete':
-                return this.query().dbFor('delete').deleteFrom(this.table);
-            case 'update':
-                return this.query().dbFor('update').updateTable(this.table);
-            case 'insert':
-                return this.query().dbFor('insert').insertInto(this.table);
-            default:
-                throw new Error('never');
-        }
+        return this.query().toKysely(type);
     }
     /**
      * Applies transaction to a new Query scoped
@@ -1748,7 +1737,7 @@ class Dream {
      * @returns The table name for this model
      */
     get table() {
-        throw new MissingTable_js_1.default(this.constructor);
+        throw new DreamMissingRequiredOverride_js_1.default(this.constructor, 'table');
     }
     /**
      * @internal

package/dist/cjs/src/bin/helpers/sync.js

@@ -5,17 +5,16 @@ const fs = require("fs/promises");
 const path = require("path");
 const index_js_1 = require("../../cli/index.js");
 const colorize_js_1 = require("../../cli/logger/loggable/colorize.js");
-const ConnectionConfRetriever_js_1 = require("../../db/ConnectionConfRetriever.js");
 const index_js_2 = require("../../dream-application/index.js");
+const autoGeneratedFileDisclaimer_js_1 = require("../../helpers/cli/autoGeneratedFileDisclaimer.js");
 const compact_js_1 = require("../../helpers/compact.js");
 const EnvInternal_js_1 = require("../../helpers/EnvInternal.js");
 const dreamPath_js_1 = require("../../helpers/path/dreamPath.js");
 const snakeify_js_1 = require("../../helpers/snakeify.js");
 const sspawn_js_1 = require("../../helpers/sspawn.js");
-const autoGeneratedFileDisclaimer_js_1 = require("../../helpers/cli/autoGeneratedFileDisclaimer.js");
 async function writeSyncFile() {
-    const dbConf = new ConnectionConfRetriever_js_1.default().getConnectionConf('primary');
     const dreamApp = index_js_2.default.getOrFail();
+    const dbConf = dreamApp.dbConnectionConfig('primary');
     const dbSyncFilePath = path.join((0, dreamPath_js_1.default)('types'), 'db.ts');
     const absoluteDbSyncPath = path.join(dreamApp.projectRoot, dbSyncFilePath);
     await (0, sspawn_js_1.default)(`kysely-codegen --dialect=postgres --url=postgres://${dbConf.user}:${dbConf.password}@${dbConf.host}:${dbConf.port}/${dbConf.name} --out-file=${absoluteDbSyncPath}`, {

package/dist/cjs/src/bin/index.js

@@ -1,7 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 const index_js_1 = require("../cli/index.js");
-const ConnectionConfRetriever_js_1 = require("../db/ConnectionConfRetriever.js");
 const index_js_2 = require("../dream-application/index.js");
 const EnvInternal_js_1 = require("../helpers/EnvInternal.js");
 const SchemaBuilder_js_1 = require("../helpers/cli/SchemaBuilder.js");
@@ -29,8 +28,8 @@ class DreamBin {
         await new SchemaBuilder_js_1.default().build();
     }
     static async dbCreate() {
-        const connectionRetriever = new ConnectionConfRetriever_js_1.default();
-        const primaryDbConf = connectionRetriever.getConnectionConf('primary');
+        const dreamApp = index_js_2.default.getOrFail();
+        const primaryDbConf = dreamApp.dbConnectionConfig('primary');
         index_js_1.default.logger.logStartProgress(`creating ${primaryDbConf.name}...`);
         await (0, createDb_js_1.default)('primary');
         index_js_1.default.logger.logEndProgress();
@@ -44,8 +43,8 @@ class DreamBin {
         // }
     }
     static async dbDrop() {
-        const connectionRetriever = new ConnectionConfRetriever_js_1.default();
-        const primaryDbConf = connectionRetriever.getConnectionConf('primary');
+        const dreamApp = index_js_2.default.getOrFail();
+        const primaryDbConf = dreamApp.dbConnectionConfig('primary');
         index_js_1.default.logger.logStartProgress(`dropping ${primaryDbConf.name}...`);
         await (0, dropDb_js_1.default)('primary');
         index_js_1.default.logger.logEndProgress();
@@ -59,24 +58,24 @@ class DreamBin {
         // }
     }
     static async dbMigrate() {
-        const connectionRetriever = new ConnectionConfRetriever_js_1.default();
-        const primaryDbConf = connectionRetriever.getConnectionConf('primary');
+        const dreamApp = index_js_2.default.getOrFail();
+        const primaryDbConf = dreamApp.dbConnectionConfig('primary');
         index_js_1.default.logger.logStartProgress(`migrating ${primaryDbConf.name}...`);
         await (0, runMigration_js_1.default)({ mode: 'migrate' });
-        await this.duplicateDatabase();
         index_js_1.default.logger.logEndProgress();
+        await this.duplicateDatabase();
     }
     static async dbRollback(opts) {
-        const connectionRetriever = new ConnectionConfRetriever_js_1.default();
-        const primaryDbConf = connectionRetriever.getConnectionConf('primary');
+        const dreamApp = index_js_2.default.getOrFail();
+        const primaryDbConf = dreamApp.dbConnectionConfig('primary');
         index_js_1.default.logger.logStartProgress(`rolling back ${primaryDbConf.name}...`);
         let step = opts.steps;
         while (step > 0) {
             await (0, runMigration_js_1.default)({ mode: 'rollback' });
             step -= 1;
         }
-        await this.duplicateDatabase();
         index_js_1.default.logger.logEndProgress();
+        await this.duplicateDatabase();
     }
     static async generateDream(fullyQualifiedModelName, columnsWithTypes, options) {
         await (0, generateDream_js_1.default)({ fullyQualifiedModelName, columnsWithTypes, options });
@@ -96,26 +95,27 @@ class DreamBin {
         index_js_1.default.logger.logEndProgress();
     }
     static async duplicateDatabase() {
-        const parallelTests = index_js_2.default.getOrFail().parallelTests;
+        const dreamApp = index_js_2.default.getOrFail();
+        const parallelTests = dreamApp.parallelTests;
         if (!parallelTests)
             return;
-        const connectionRetriever = new ConnectionConfRetriever_js_1.default();
-        const dbConf = connectionRetriever.getConnectionConf('primary');
+        index_js_1.default.logger.logStartProgress(`duplicating db for parallel tests...`);
+        const dbConf = dreamApp.dbConnectionConfig('primary');
         const client = await (0, loadPgClient_js_1.default)({ useSystemDb: true });
         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}...`);
+            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}`;
-            console.log(`creating duplicate test database ${workerDatabaseName} for concurrent tests`);
-            index_js_1.default.logger.logContinueProgress(`creating duplicate test database ${workerDatabaseName} for concurrent tests...`);
+            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 = DreamBin;

package/dist/cjs/src/db/DreamDbConnection.js

@@ -13,16 +13,16 @@ exports.closeAllDbConnections = closeAllDbConnections;
 const pg_1 = require("pg");
 const kysely_1 = require("kysely");
 const index_js_1 = require("../dream-application/index.js");
-const ConnectionConfRetriever_js_1 = require("./ConnectionConfRetriever.js");
 let connections = {};
 class DreamDbConnection {
     static getConnection(connectionType) {
-        const connectionName = getConnectionTypeName(connectionType);
+        const dreamApp = index_js_1.default.getOrFail();
+        const connectionName = this.getConnectionTypeName(connectionType);
         const connection = connections[connectionName];
         if (connection) {
             return connection;
         }
-        const connectionConf = new ConnectionConfRetriever_js_1.default().getConnectionConf(connectionType);
+        const connectionConf = dreamApp.dbConnectionConfig(connectionType);
         const dbConn = new kysely_1.Kysely({
             log(event) {
                 const dreamApp = index_js_1.default.getOrFail();
@@ -34,7 +34,7 @@ class DreamDbConnection {
                 pool: new pg_1.default.Pool({
                     user: connectionConf.user || '',
                     password: connectionConf.password || '',
-                    database: getDatabaseName(connectionConf.name),
+                    database: dreamApp.dbName(connectionType),
                     host: connectionConf.host || 'localhost',
                     port: connectionConf.port || 5432,
                     ssl: connectionConf.useSsl ? sslConfig(connectionConf) : false,
@@ -42,7 +42,7 @@ class DreamDbConnection {
             }),
             plugins: [new kysely_1.CamelCasePlugin({ underscoreBetweenUppercaseLetters: true })],
         });
-        connections[getConnectionTypeName(connectionType)] = dbConn;
+        connections[this.getConnectionTypeName(connectionType)] = dbConn;
         return dbConn;
     }
     static async dropAllConnections() {
@@ -51,19 +51,13 @@ class DreamDbConnection {
             delete connections[key];
         }
     }
+    static getConnectionTypeName(connectionType) {
+        return index_js_1.default.getOrFail().parallelDatabasesEnabled
+            ? `${connectionType}_${process.env.VITEST_POOL_ID}`
+            : connectionType;
+    }
 }
 exports.default = DreamDbConnection;
-function getConnectionTypeName(connectionType) {
-    return parallelDatabasesEnabled() ? `${connectionType}_${process.env.JEST_WORKER_ID}` : connectionType;
-}
-function getDatabaseName(dbName) {
-    return parallelDatabasesEnabled() ? `${dbName}_${process.env.JEST_WORKER_ID}` : dbName;
-}
-function parallelDatabasesEnabled() {
-    return (!!index_js_1.default.getOrFail().parallelTests &&
-        !Number.isNaN(Number(process.env.JEST_WORKER_ID)) &&
-        Number(process.env.JEST_WORKER_ID) > 1);
-}
 // eslint-disable-next-line @typescript-eslint/no-unused-vars
 function sslConfig(connectionConf) {
     // TODO: properly configure (https://rvohealth.atlassian.net/browse/PDTC-2914)

package/dist/cjs/src/dream-application/helpers/importers/importModels.js

@@ -5,7 +5,7 @@ exports.setCachedModels = setCachedModels;
 exports.getModelsOrFail = getModelsOrFail;
 exports.getModelsOrBlank = getModelsOrBlank;
 const Dream_js_1 = require("../../../Dream.js");
-const MissingTable_js_1 = require("../../../errors/MissingTable.js");
+const DreamMissingRequiredOverride_js_1 = require("../../../errors/DreamMissingRequiredOverride.js");
 const DreamImporter_js_1 = require("../DreamImporter.js");
 const globalModelKeyFromPath_js_1 = require("../globalModelKeyFromPath.js");
 let _models;
@@ -35,7 +35,7 @@ async function importModels(modelsPath, modelImportCb) {
             catch (error) {
                 // ApplicationModel will automatically raise an exception here,
                 // since it does not have a table.
-                if (!(error instanceof MissingTable_js_1.default))
+                if (!(error instanceof DreamMissingRequiredOverride_js_1.default))
                     throw error;
             }
         }
@@ -54,7 +54,7 @@ async function importModels(modelsPath, modelImportCb) {
             catch (error) {
                 // ApplicationModel will automatically raise an exception here,
                 // since it does not have a table.
-                if (!(error instanceof MissingTable_js_1.default))
+                if (!(error instanceof DreamMissingRequiredOverride_js_1.default))
                     throw error;
             }
         }

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

@@ -228,6 +228,28 @@ Try setting it to something valid, like:
     get serializers() {
         return (0, importSerializers_js_1.getSerializersOrFail)();
     }
+    dbName(connection) {
+        const conf = this.dbConnectionConfig(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)
+            throw new Error(`
+      Cannot find a connection config given the following connection and node environment:
+        connection: ${connection}
+        NODE_ENV: ${EnvInternal_js_1.default.nodeEnv}
+    `);
+        return conf;
+    }
+    get hasReplicaConfig() {
+        return !!this.dbCredentials.replica;
+    }
+    get parallelDatabasesEnabled() {
+        return (!!this.parallelTests &&
+            !Number.isNaN(Number(process.env.VITEST_POOL_ID)) &&
+            Number(process.env.VITEST_POOL_ID) > 1);
+    }
     async load(resourceType, resourcePath, importCb) {
         switch (resourceType) {
             case 'models':

package/dist/cjs/src/errors/DreamMissingRequiredOverride.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+class DreamMissingRequiredOverride extends Error {
+    dreamClass;
+    methodName;
+    constructor(dreamClass, methodName) {
+        super();
+        this.dreamClass = dreamClass;
+        this.methodName = methodName;
+    }
+    get message() {
+        return `Dream class ${this.dreamClass.sanitizedName} missing required ${this.methodName} override.`;
+    }
+}
+exports.default = DreamMissingRequiredOverride;

package/dist/cjs/src/helpers/db/createDb.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = createDb;
-const ConnectionConfRetriever_js_1 = require("../../db/ConnectionConfRetriever.js");
+const index_js_1 = require("../../dream-application/index.js");
 const EnvInternal_js_1 = require("../EnvInternal.js");
 const loadPgClient_js_1 = require("./loadPgClient.js");
 async function createDb(connection, dbName) {
@@ -9,8 +9,8 @@ async function createDb(connection, dbName) {
     // so there is no way to drop in production
     if (EnvInternal_js_1.default.isProduction)
         return false;
-    const connectionRetriever = new ConnectionConfRetriever_js_1.default();
-    const dbConf = connectionRetriever.getConnectionConf(connection);
+    const dreamApp = index_js_1.default.getOrFail();
+    const dbConf = dreamApp.dbConnectionConfig(connection);
     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');

package/dist/cjs/src/helpers/db/dropDb.js

@@ -1,17 +1,17 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.default = dropDb;
-const ConnectionConfRetriever_js_1 = require("../../db/ConnectionConfRetriever.js");
 const index_js_1 = require("../../dream-application/index.js");
 const EnvInternal_js_1 = require("../EnvInternal.js");
 const loadPgClient_js_1 = require("./loadPgClient.js");
+const index_js_2 = require("../../cli/index.js");
 async function dropDb(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 connectionRetriever = new ConnectionConfRetriever_js_1.default();
-    const dbConf = connectionRetriever.getConnectionConf(connection);
+    const dreamApp = index_js_1.default.getOrFail();
+    const dbConf = dreamApp.dbConnectionConfig(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');
@@ -25,12 +25,15 @@ async function maybeDropDuplicateDatabases(client, dbName) {
         return;
     if (EnvInternal_js_1.default.boolean('DREAM_CORE_DEVELOPMENT')) {
         const replicaTestWorkerDatabaseName = `replica_test_${dbName}`;
-        console.log(`dropping fake replica test database ${replicaTestWorkerDatabaseName}`);
+        index_js_2.default.logger.logContinueProgress(`dropping fake replica test database ${replicaTestWorkerDatabaseName}`, { logPrefix: '  ├ [db]', logPrefixColor: 'cyan' });
         await client.query(`DROP DATABASE IF EXISTS ${replicaTestWorkerDatabaseName};`);
     }
     for (let i = 2; i <= parallelTests; i++) {
         const workerDatabaseName = `${dbName}_${i}`;
-        console.log(`dropping duplicate test database ${workerDatabaseName}`);
+        index_js_2.default.logger.logContinueProgress(`dropping duplicate test database ${workerDatabaseName}`, {
+            logPrefix: '  ├ [db]',
+            logPrefixColor: 'cyan',
+        });
         await client.query(`DROP DATABASE IF EXISTS ${workerDatabaseName};`);
     }
 }

package/dist/esm/src/Dream.js

@@ -35,8 +35,8 @@ import CannotCallUndestroyOnANonSoftDeleteModel from './errors/CannotCallUndestr
 import ConstructorOnlyForInternalUse from './errors/ConstructorOnlyForInternalUse.js';
 import CreateOrFindByFailedToCreateAndFind from './errors/CreateOrFindByFailedToCreateAndFind.js';
 import GlobalNameNotSet from './errors/dream-application/GlobalNameNotSet.js';
+import DreamMissingRequiredOverride from './errors/DreamMissingRequiredOverride.js';
 import MissingSerializer from './errors/MissingSerializersDefinition.js';
-import MissingTable from './errors/MissingTable.js';
 import NonExistentScopeProvidedToResort from './errors/NonExistentScopeProvidedToResort.js';
 import CalendarDate from './helpers/CalendarDate.js';
 import cloneDeepSafe from './helpers/cloneDeepSafe.js';
@@ -52,14 +52,14 @@ export default class Dream {
      * @internal
      *
      * This getter will throw an error when developers use .toEqual instead of
-     * useToMatchDreamModels or useToMatchDreamModel in a jest spec. This
+     * useToMatchDreamModels or useToMatchDreamModel in a vite spec. This
      * must be the first getter in the class in order for this to work, so don't move it.
      *
      */
     get _useToMatchDreamModels() {
         throw new Error(`
       Hi there! It looks like you're trying to compare a Dream model in
-      a Jest expectation using \`toEqual\`. That won't work.
+      a Vite expectation using \`toEqual\`. That won't work.
       Instead, use \`toMatchDreamModel\` or \`toMatchDreamModels\`.
 
       For example, instead of:
@@ -97,10 +97,10 @@ export default class Dream {
      */
     static globallyInitializingDecorators = false;
     get schema() {
-        throw new Error('Must define schema getter in ApplicationModel');
+        throw new DreamMissingRequiredOverride(this.constructor, 'schema');
     }
     get globalSchema() {
-        throw new Error('Must define schema getter in ApplicationModel');
+        throw new DreamMissingRequiredOverride(this.constructor, 'globalSchema');
     }
     /**
      * Shadows #primaryKey, a getter which can be overwritten to customize the id field
@@ -1232,18 +1232,7 @@ export default class Dream {
      * @returns A Kysely query. Depending on the type passed, it will return either a SelectQueryBuilder, DeleteQueryBuilder, UpdateQueryBuilder, or an InsertQueryBuilder
      */
     static toKysely(type) {
-        switch (type) {
-            case 'select':
-                return this.query().dbFor('select').selectFrom(this.table);
-            case 'delete':
-                return this.query().dbFor('delete').deleteFrom(this.table);
-            case 'update':
-                return this.query().dbFor('update').updateTable(this.table);
-            case 'insert':
-                return this.query().dbFor('insert').insertInto(this.table);
-            default:
-                throw new Error('never');
-        }
+        return this.query().toKysely(type);
     }
     /**
      * Applies transaction to a new Query scoped
@@ -1746,7 +1735,7 @@ export default class Dream {
      * @returns The table name for this model
      */
     get table() {
-        throw new MissingTable(this.constructor);
+        throw new DreamMissingRequiredOverride(this.constructor, 'table');
     }
     /**
      * @internal

package/dist/esm/src/bin/helpers/sync.js

@@ -2,17 +2,16 @@ import * as fs from 'fs/promises';
 import * as path from 'path';
 import DreamCLI from '../../cli/index.js';
 import colorize from '../../cli/logger/loggable/colorize.js';
-import ConnectionConfRetriever from '../../db/ConnectionConfRetriever.js';
 import DreamApplication from '../../dream-application/index.js';
+import autogeneratedFileDisclaimer from '../../helpers/cli/autoGeneratedFileDisclaimer.js';
 import compact from '../../helpers/compact.js';
 import EnvInternal from '../../helpers/EnvInternal.js';
 import dreamPath from '../../helpers/path/dreamPath.js';
 import snakeify from '../../helpers/snakeify.js';
 import sspawn from '../../helpers/sspawn.js';
-import autogeneratedFileDisclaimer from '../../helpers/cli/autoGeneratedFileDisclaimer.js';
 export default async function writeSyncFile() {
-    const dbConf = new ConnectionConfRetriever().getConnectionConf('primary');
     const dreamApp = DreamApplication.getOrFail();
+    const dbConf = dreamApp.dbConnectionConfig('primary');
     const dbSyncFilePath = path.join(dreamPath('types'), 'db.ts');
     const absoluteDbSyncPath = path.join(dreamApp.projectRoot, dbSyncFilePath);
     await sspawn(`kysely-codegen --dialect=postgres --url=postgres://${dbConf.user}:${dbConf.password}@${dbConf.host}:${dbConf.port}/${dbConf.name} --out-file=${absoluteDbSyncPath}`, {

package/dist/esm/src/bin/index.js

@@ -1,5 +1,4 @@
 import DreamCLI from '../cli/index.js';
-import ConnectionConfRetriever from '../db/ConnectionConfRetriever.js';
 import DreamApplication from '../dream-application/index.js';
 import EnvInternal from '../helpers/EnvInternal.js';
 import SchemaBuilder from '../helpers/cli/SchemaBuilder.js';
@@ -27,8 +26,8 @@ export default class DreamBin {
         await new SchemaBuilder().build();
     }
     static async dbCreate() {
-        const connectionRetriever = new ConnectionConfRetriever();
-        const primaryDbConf = connectionRetriever.getConnectionConf('primary');
+        const dreamApp = DreamApplication.getOrFail();
+        const primaryDbConf = dreamApp.dbConnectionConfig('primary');
         DreamCLI.logger.logStartProgress(`creating ${primaryDbConf.name}...`);
         await createDb('primary');
         DreamCLI.logger.logEndProgress();
@@ -42,8 +41,8 @@ export default class DreamBin {
         // }
     }
     static async dbDrop() {
-        const connectionRetriever = new ConnectionConfRetriever();
-        const primaryDbConf = connectionRetriever.getConnectionConf('primary');
+        const dreamApp = DreamApplication.getOrFail();
+        const primaryDbConf = dreamApp.dbConnectionConfig('primary');
         DreamCLI.logger.logStartProgress(`dropping ${primaryDbConf.name}...`);
         await _dropDb('primary');
         DreamCLI.logger.logEndProgress();
@@ -57,24 +56,24 @@ export default class DreamBin {
         // }
     }
     static async dbMigrate() {
-        const connectionRetriever = new ConnectionConfRetriever();
-        const primaryDbConf = connectionRetriever.getConnectionConf('primary');
+        const dreamApp = DreamApplication.getOrFail();
+        const primaryDbConf = dreamApp.dbConnectionConfig('primary');
         DreamCLI.logger.logStartProgress(`migrating ${primaryDbConf.name}...`);
         await runMigration({ mode: 'migrate' });
-        await this.duplicateDatabase();
         DreamCLI.logger.logEndProgress();
+        await this.duplicateDatabase();
     }
     static async dbRollback(opts) {
-        const connectionRetriever = new ConnectionConfRetriever();
-        const primaryDbConf = connectionRetriever.getConnectionConf('primary');
+        const dreamApp = DreamApplication.getOrFail();
+        const primaryDbConf = dreamApp.dbConnectionConfig('primary');
         DreamCLI.logger.logStartProgress(`rolling back ${primaryDbConf.name}...`);
         let step = opts.steps;
         while (step > 0) {
             await runMigration({ mode: 'rollback' });
             step -= 1;
         }
-        await this.duplicateDatabase();
         DreamCLI.logger.logEndProgress();
+        await this.duplicateDatabase();
     }
     static async generateDream(fullyQualifiedModelName, columnsWithTypes, options) {
         await generateDream({ fullyQualifiedModelName, columnsWithTypes, options });
@@ -94,25 +93,26 @@ export default class DreamBin {
         DreamCLI.logger.logEndProgress();
     }
     static async duplicateDatabase() {
-        const parallelTests = DreamApplication.getOrFail().parallelTests;
+        const dreamApp = DreamApplication.getOrFail();
+        const parallelTests = dreamApp.parallelTests;
         if (!parallelTests)
             return;
-        const connectionRetriever = new ConnectionConfRetriever();
-        const dbConf = connectionRetriever.getConnectionConf('primary');
+        DreamCLI.logger.logStartProgress(`duplicating db for parallel tests...`);
+        const dbConf = dreamApp.dbConnectionConfig('primary');
         const client = await loadPgClient({ useSystemDb: true });
         if (EnvInternal.boolean('DREAM_CORE_DEVELOPMENT')) {
             const replicaTestWorkerDatabaseName = `replica_test_${dbConf.name}`;
-            DreamCLI.logger.logContinueProgress(`creating fake replica test database ${replicaTestWorkerDatabaseName}...`);
+            DreamCLI.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}`;
-            console.log(`creating duplicate test database ${workerDatabaseName} for concurrent tests`);
-            DreamCLI.logger.logContinueProgress(`creating duplicate test database ${workerDatabaseName} for concurrent tests...`);
+            DreamCLI.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();
+        DreamCLI.logger.logEndProgress();
     }
 }

package/dist/esm/src/db/DreamDbConnection.js

@@ -9,16 +9,16 @@
 import pg from 'pg';
 import { CamelCasePlugin, Kysely, PostgresDialect } from 'kysely';
 import DreamApplication from '../dream-application/index.js';
-import ConnectionConfRetriever from './ConnectionConfRetriever.js';
 let connections = {};
 export default class DreamDbConnection {
     static getConnection(connectionType) {
-        const connectionName = getConnectionTypeName(connectionType);
+        const dreamApp = DreamApplication.getOrFail();
+        const connectionName = this.getConnectionTypeName(connectionType);
         const connection = connections[connectionName];
         if (connection) {
             return connection;
         }
-        const connectionConf = new ConnectionConfRetriever().getConnectionConf(connectionType);
+        const connectionConf = dreamApp.dbConnectionConfig(connectionType);
         const dbConn = new Kysely({
             log(event) {
                 const dreamApp = DreamApplication.getOrFail();
@@ -30,7 +30,7 @@ export default class DreamDbConnection {
                 pool: new pg.Pool({
                     user: connectionConf.user || '',
                     password: connectionConf.password || '',
-                    database: getDatabaseName(connectionConf.name),
+                    database: dreamApp.dbName(connectionType),
                     host: connectionConf.host || 'localhost',
                     port: connectionConf.port || 5432,
                     ssl: connectionConf.useSsl ? sslConfig(connectionConf) : false,
@@ -38,7 +38,7 @@ export default class DreamDbConnection {
             }),
             plugins: [new CamelCasePlugin({ underscoreBetweenUppercaseLetters: true })],
         });
-        connections[getConnectionTypeName(connectionType)] = dbConn;
+        connections[this.getConnectionTypeName(connectionType)] = dbConn;
         return dbConn;
     }
     static async dropAllConnections() {
@@ -47,17 +47,11 @@ export default class DreamDbConnection {
             delete connections[key];
         }
     }
-}
-function getConnectionTypeName(connectionType) {
-    return parallelDatabasesEnabled() ? `${connectionType}_${process.env.JEST_WORKER_ID}` : connectionType;
-}
-function getDatabaseName(dbName) {
-    return parallelDatabasesEnabled() ? `${dbName}_${process.env.JEST_WORKER_ID}` : dbName;
-}
-function parallelDatabasesEnabled() {
-    return (!!DreamApplication.getOrFail().parallelTests &&
-        !Number.isNaN(Number(process.env.JEST_WORKER_ID)) &&
-        Number(process.env.JEST_WORKER_ID) > 1);
+    static getConnectionTypeName(connectionType) {
+        return DreamApplication.getOrFail().parallelDatabasesEnabled
+            ? `${connectionType}_${process.env.VITEST_POOL_ID}`
+            : connectionType;
+    }
 }
 // eslint-disable-next-line @typescript-eslint/no-unused-vars
 function sslConfig(connectionConf) {

package/dist/esm/src/dream-application/helpers/importers/importModels.js

@@ -1,5 +1,5 @@
 import Dream from '../../../Dream.js';
-import MissingTable from '../../../errors/MissingTable.js';
+import DreamMissingRequiredOverride from '../../../errors/DreamMissingRequiredOverride.js';
 import DreamImporter from '../DreamImporter.js';
 import globalModelKeyFromPath from '../globalModelKeyFromPath.js';
 let _models;
@@ -29,7 +29,7 @@ export default async function importModels(modelsPath, modelImportCb) {
             catch (error) {
                 // ApplicationModel will automatically raise an exception here,
                 // since it does not have a table.
-                if (!(error instanceof MissingTable))
+                if (!(error instanceof DreamMissingRequiredOverride))
                     throw error;
             }
         }
@@ -48,7 +48,7 @@ export default async function importModels(modelsPath, modelImportCb) {
             catch (error) {
                 // ApplicationModel will automatically raise an exception here,
                 // since it does not have a table.
-                if (!(error instanceof MissingTable))
+                if (!(error instanceof DreamMissingRequiredOverride))
                     throw error;
             }
         }