rake-db 2.25.17 → 2.26.0

package/dist/index.mjs

@@ -1,8 +1,10 @@
-import { DomainColumn, escapeForMigration, ArrayColumn, EnumColumn, defaultSchemaConfig, getColumnTypes, parseTableData, escapeString, tableDataMethods, ColumnType, parseTableDataInput, UnknownColumn, raw, TransactionAdapter, logParamToLogObject, createDb as createDb$1, Adapter, makeColumnTypes, makeColumnsByType, RawSQL, CustomTypeColumn, assignDbDataToColumn, PostgisGeographyPointColumn, pushTableDataCode, primaryKeyInnerToCode, indexInnerToCode, excludeInnerToCode, constraintInnerToCode, referencesArgsToCode, TimestampTZColumn, TimestampColumn } from 'pqb';
-import { singleQuote, toSnakeCase, isRawSQL, toCamelCase, toArray, snakeCaseKey, emptyObject, setCurrentColumnName, consumeColumnName, ColumnTypeBase, setDefaultLanguage, deepCompare, getImportPath, pathToLog, emptyArray, getStackTrace, codeToString, addCode, quoteObjectKey, backtickQuote } from 'orchid-core';
+import { singleQuote, isRawSQL, toSnakeCase, toCamelCase, toArray, snakeCaseKey, emptyObject, setCurrentColumnName, consumeColumnName, ColumnTypeBase, setDefaultLanguage, deepCompare, getImportPath, pathToLog, emptyArray, getStackTrace, codeToString, addCode, quoteObjectKey, backtickQuote } from 'orchid-core';
+import { escapeForMigration, ArrayColumn, DomainColumn, EnumColumn, defaultSchemaConfig, getColumnTypes, parseTableData, escapeString, tableDataMethods, ColumnType, parseTableDataInput, UnknownColumn, raw, TransactionAdapter, logParamToLogObject, createDb as createDb$1, Adapter, makeColumnTypes, makeColumnsByType, RawSQL, CustomTypeColumn, assignDbDataToColumn, PostgisGeographyPointColumn, pushTableDataCode, primaryKeyInnerToCode, indexInnerToCode, excludeInnerToCode, constraintInnerToCode, referencesArgsToCode, TimestampTZColumn, TimestampColumn } from 'pqb';
 import path, { join } from 'path';
 import { pathToFileURL, fileURLToPath } from 'node:url';
 import fs, { mkdir, writeFile, readdir, stat, readFile } from 'fs/promises';
+import 'url';
+import 'node:path';
 
 const getFirstWordAndRest = (input) => {
   const i = input.search(/(?=[A-Z])|[-_ ]/);
@@ -29,12 +31,6 @@ const getTextAfterTo = (input) => {
 const getTextAfterFrom = (input) => {
   return getTextAfterRegExp(input, /(From|-from|_from| from)[A-Z-_ ]/, 4);
 };
-const joinWords = (...words) => {
-  return words.slice(1).reduce(
-    (acc, word) => acc + word[0].toUpperCase() + word.slice(1),
-    words[0]
-  );
-};
 const joinColumns = (columns) => {
   return columns.map((column) => `"${column}"`).join(", ");
 };
@@ -79,19 +75,13 @@ const transaction = (adapter, fn) => {
   return adapter.transaction(begin, fn);
 };
 const queryLock = (trx) => trx.query(`SELECT pg_advisory_xact_lock('${RAKE_DB_LOCK_KEY}')`);
-const exhaustive = (_) => {
-  throw new Error("Condition was not exhaustive");
-};
-const pluralize = (w, count, append = "s") => {
-  return count === 1 ? w : w + append;
-};
 
 let currentChanges = [];
 const clearChanges = () => {
   currentChanges = [];
 };
 const getCurrentChanges = () => currentChanges;
-const pushChange = (fn) => currentChanges.push(fn);
+const pushChange = (change) => currentChanges.push(change);
 
 const versionToString = (config, version) => config.migrationId === "timestamp" ? `${version}` : `${version}`.padStart(config.migrationId.serial, "0");
 const columnTypeToSql = (item) => {
@@ -3062,17 +3052,17 @@ function makeMigrateFn(defaultCount, up, fn) {
     return adapters;
   };
 }
-const migrate = makeMigrateFn(
+const fullMigrate = makeMigrateFn(
   Infinity,
   true,
   (trx, config, set, versions, count, force) => migrateOrRollback(trx, config, set, versions, count, true, false, force)
 );
-const rollback = makeMigrateFn(
+const fullRollback = makeMigrateFn(
   1,
   false,
   (trx, config, set, versions, count, force) => migrateOrRollback(trx, config, set, versions, count, false, false, force)
 );
-const redo = makeMigrateFn(
+const fullRedo = makeMigrateFn(
   1,
   true,
   async (trx, config, set, versions, count, force) => {
@@ -3093,7 +3083,7 @@ const redo = makeMigrateFn(
   }
 );
 const getDb = (adapter) => createDb$1({ adapter });
-const migrateOrRollback = async (trx, config, set, versions, count, up, redo2, force, skipLock) => {
+const migrateOrRollback = async (trx, config, set, versions, count, up, redo, force, skipLock) => {
   const { sequence, map: versionsMap } = versions;
   if (up) {
     const rollbackTo = checkMigrationOrder(config, set, versions, force);
@@ -3114,7 +3104,7 @@ const migrateOrRollback = async (trx, config, set, versions, count, up, redo2, f
         versions,
         sequence.length - i,
         false,
-        redo2
+        redo
       );
       set.migrations.reverse();
     }
@@ -3126,7 +3116,7 @@ const migrateOrRollback = async (trx, config, set, versions, count, up, redo2, f
     db ?? (db = getDb(trx));
     const { migrations: migrations2 } = set;
     await beforeMigrate?.({ db, migrations: migrations2 });
-    await config.beforeChange?.({ db, migrations: migrations2, up, redo: redo2 });
+    await config.beforeChange?.({ db, migrations: migrations2, up, redo });
   }
   let loggedAboutStarting = false;
   let migrations;
@@ -3135,14 +3125,16 @@ const migrateOrRollback = async (trx, config, set, versions, count, up, redo2, f
       continue;
     }
     if (count-- <= 0) break;
-    if (!loggedAboutStarting && (!redo2 || !up)) {
+    if (!loggedAboutStarting && (!redo || !up)) {
       loggedAboutStarting = true;
       config.logger?.log(
-        `${redo2 ? "Reapplying migrations for" : up ? "Migrating" : "Rolling back"} database ${trx.config.connectionString ? new URL(trx.config.connectionString).pathname.slice(1) : trx.config.database}
+        `${redo ? "Reapplying migrations for" : up ? "Migrating" : "Rolling back"} database ${trx.config.connectionString ? new URL(trx.config.connectionString).pathname.slice(1) : trx.config.database}
 `
       );
     }
-    await runMigration(trx, up, file, config);
+    const changes = await getChanges(file, config);
+    const adapter = await runMigration(trx, up, changes, config);
+    await changeMigratedVersion(adapter, up, file, config);
     (migrations ?? (migrations = [])).push(file);
     if (up) {
       const name = path.basename(file.path);
@@ -3161,7 +3153,7 @@ const migrateOrRollback = async (trx, config, set, versions, count, up, redo2, f
   const afterMigrate = config[up ? "afterMigrate" : "afterRollback"];
   if (config.afterChange || afterMigrate) {
     db ?? (db = getDb(trx));
-    await config.afterChange?.({ db, up, redo: redo2, migrations });
+    await config.afterChange?.({ db, up, redo, migrations });
     await afterMigrate?.({ db, migrations });
   }
   return migrations;
@@ -3204,31 +3196,37 @@ Run \`**db command** up force\` to rollback the above migrations and migrate all
   return;
 };
 const changeCache = {};
-const runMigration = async (trx, up, file, config) => {
+const getChanges = async (file, config) => {
   clearChanges();
-  let changes = changeCache[file.path];
+  let changes = file.path ? changeCache[file.path] : void 0;
   if (!changes) {
     const module = await file.load();
     const exported = module?.default && toArray(module.default);
-    if (config.forceDefaultExports && !exported) {
+    if (config?.forceDefaultExports && !exported) {
       throw new RakeDbError(
         `Missing a default export in ${file.path} migration`
       );
     }
     changes = exported || getCurrentChanges();
-    changeCache[file.path] = changes;
+    if (file.path) changeCache[file.path] = changes;
   }
+  return changes;
+};
+const runMigration = async (trx, up, changes, config) => {
   const db = createMigrationInterface(trx, up, config);
   if (changes.length) {
     const from = up ? 0 : changes.length - 1;
     const to = up ? changes.length : -1;
     const step = up ? 1 : -1;
     for (let i = from; i !== to; i += step) {
-      await changes[i](db, up);
+      await changes[i].fn(db, up);
     }
   }
+  return db.adapter;
+};
+const changeMigratedVersion = async (adapter, up, file, config) => {
   await (up ? saveMigratedVersion : deleteMigratedVersion)(
-    db.adapter,
+    adapter,
     file.version,
     path.basename(file.path).slice(file.version.length + 1),
     config
@@ -3638,7 +3636,7 @@ const dropDb = async (options, config) => {
 const resetDb = async (options, config) => {
   await dropDb(options, config);
   await createDb(options, config);
-  await migrate({}, options, config);
+  await fullMigrate({}, options, config);
 };
 
 const filterSchema = (table) => `${table} !~ '^pg_' AND ${table} != 'information_schema'`;
@@ -4602,6 +4600,10 @@ const checkIfIsOuterRecursiveFkey = (data, table, references) => {
   return false;
 };
 
+const exhaustive = (_) => {
+  throw new Error("Condition was not exhaustive");
+};
+
 const astToGenerateItems = (config, asts, currentSchema) => {
   return asts.map((ast) => astToGenerateItem(config, ast, currentSchema));
 };
@@ -5642,7 +5644,7 @@ const rebase = async (options, config) => {
     return files2;
   }, []).sort((a, b) => +b.version - +a.version);
   set.migrations = migrationsDown;
-  await redo(
+  await fullRedo(
     ctx,
     options,
     {
@@ -5682,25 +5684,26 @@ const rakeDb = (options, partialConfig, args = process.argv.slice(2)) => {
     }
     throw err;
   });
-  return Object.assign(change, {
+  return Object.assign(makeChange(config), {
     promise
   });
 };
 rakeDb.lazy = (options, partialConfig) => {
   const config = processRakeDbConfig(partialConfig);
   return {
-    change,
+    change: makeChange(config),
     run(args, conf) {
       return runCommand(options, conf ? { ...config, ...conf } : config, args);
     }
   };
 };
-function change(fn) {
-  pushChange(fn);
-  return fn;
-}
+const makeChange = (config) => (fn) => {
+  const change = { fn, config };
+  pushChange(change);
+  return change;
+};
 const rakeDbAliases = {
-  migrate: "up",
+  fullMigrate: "up",
   rollback: "down",
   s: "status",
   rec: "recurrent"
@@ -5776,7 +5779,7 @@ ${Object.entries(helpArguments).map(
   };
 };
 const upCommand = {
-  run: (options, config, args) => migrate({}, options, config, args).then(
+  run: (options, config, args) => fullMigrate({}, options, config, args).then(
     () => runRecurrentMigrations(options, config)
   ),
   help: "migrate pending migrations",
@@ -5787,7 +5790,7 @@ const upCommand = {
   }
 };
 const downCommand = {
-  run: (options, config, args) => rollback({}, options, config, args),
+  run: (options, config, args) => fullRollback({}, options, config, args),
   help: "rollback migrated migrations",
   helpArguments: {
     "no arguments": "rollback one last migration",
@@ -5827,7 +5830,7 @@ const rakeDbCommands = {
   down: downCommand,
   rollback: downCommand,
   redo: {
-    run: (options, config, args) => redo({}, options, config, args).then(
+    run: (options, config, args) => fullRedo({}, options, config, args).then(
       () => runRecurrentMigrations(options, config)
     ),
     help: "rollback and migrate, run recurrent"
@@ -5859,5 +5862,18 @@ const rakeDbCommands = {
   }
 };
 
-export { Migration, NoMigrationsTableError, RAKE_DB_LOCK_KEY, addColumnComment, addColumnExclude, addColumnIndex, addOrDropEnumValues, astToMigration, changeCache, changeEnumValues, clearChanges, cmpRawSql, colors, columnToSql, columnTypeToSql, commentsToQuery, concatSchemaAndName, constraintToSql, createDb, createMigrationInterface, dbColumnToAst, deleteMigratedVersion, dropDb, encodeColumnDefault, excludesToQuery, exhaustive, generateTimeStamp, getColumnName, getConstraintName, getCurrentChanges, getDatabaseAndUserFromOptions, getDbStructureTableData, getDbTableColumnsChecks, getExcludeName, getFirstWordAndRest, getForeignKeyTable, getIndexName, getMigratedVersionsMap, getSchemaAndTableFromName, getTextAfterFrom, getTextAfterTo, identityToSql, indexesToQuery, instantiateDbColumn, interpolateSqlValues, introspectDbSchema, joinColumns, joinWords, makeDbStructureColumnsShape, makeDomainsMap, makeFileVersion, makePopulateEnumQuery, makeStructureToAstCtx, migrate, migrateOrRollback, migrationConfigDefaults, nameColumnChecks, newMigration, pluralize, primaryKeyToSql, processRakeDbConfig, promptConfirm, promptSelect, promptText, pushChange, queryLock, quoteCustomType, quoteNameFromString, quoteSchemaTable, quoteTable, quoteWithSchema, rakeDb, rakeDbAliases, rakeDbCommands, redo, referencesToSql, renameType, resetDb, rollback, saveMigratedVersion, structureToAst, tableToAst, transaction, versionToString, writeMigrationFile };
+const migrateFiles = async (db, files) => {
+  const qb = db.$qb;
+  await qb.ensureTransaction(async () => {
+    const adapter = qb.internal.transactionStorage.getStore()?.adapter;
+    for (const load of files) {
+      clearChanges();
+      const changes = await getChanges({ load });
+      const config = changes[0]?.config;
+      await runMigration(adapter, true, changes, config);
+    }
+  });
+};
+
+export { astToMigration, concatSchemaAndName, createMigrationInterface, dbColumnToAst, encodeColumnDefault, fullMigrate, getConstraintName, getDbStructureTableData, getDbTableColumnsChecks, getExcludeName, getIndexName, getSchemaAndTableFromName, instantiateDbColumn, introspectDbSchema, makeDomainsMap, makeFileVersion, makeStructureToAstCtx, migrateFiles, migrationConfigDefaults, promptSelect, rakeDb, rakeDbCommands, saveMigratedVersion, structureToAst, tableToAst, writeMigrationFile };
 //# sourceMappingURL=index.mjs.map