rake-db 2.27.18 → 2.27.20

package/dist/index.mjs

@@ -1,6 +1,6 @@
-import { defaultSchemaConfig, makeColumnTypes, getStackTrace, singleQuote, isRawSQL, escapeForMigration, ArrayColumn, toSnakeCase, toCamelCase, DomainColumn, toArray, EnumColumn, snakeCaseKey, getColumnTypes, parseTableData, emptyObject, escapeString, tableDataMethods, setCurrentColumnName, consumeColumnName, ColumnType, ColumnTypeBase, parseTableDataInput, UnknownColumn, setDefaultLanguage, deepCompare, raw, logParamToLogObject, createDbWithAdapter, getImportPath, pathToLog, emptyArray, colors, makeColumnsByType, RawSQL, CustomTypeColumn, assignDbDataToColumn, PostgisGeographyPointColumn, exhaustive, codeToString, addCode, quoteObjectKey, pushTableDataCode, primaryKeyInnerToCode, indexInnerToCode, excludeInnerToCode, constraintInnerToCode, referencesArgsToCode, backtickQuote, TimestampTZColumn, TimestampColumn } from 'pqb';
+import { isRawSQL, escapeForMigration, ArrayColumn, toSnakeCase, toCamelCase, DomainColumn, toArray, EnumColumn, defaultSchemaConfig, snakeCaseKey, getColumnTypes, parseTableData, emptyObject, escapeString, tableDataMethods, setCurrentColumnName, consumeColumnName, ColumnType, ColumnTypeBase, parseTableDataInput, UnknownColumn, setDefaultLanguage, deepCompare, raw, singleQuote, logParamToLogObject, createDbWithAdapter, getImportPath, pathToLog, emptyArray, makeColumnTypes, getStackTrace, colors, makeColumnsByType, RawSQL, CustomTypeColumn, assignDbDataToColumn, PostgisGeographyPointColumn, exhaustive, codeToString, addCode, quoteObjectKey, pushTableDataCode, primaryKeyInnerToCode, indexInnerToCode, excludeInnerToCode, constraintInnerToCode, referencesArgsToCode, backtickQuote, TimestampTZColumn, TimestampColumn } from 'pqb';
 import path, { join } from 'path';
-import { fileURLToPath, pathToFileURL } from 'node:url';
+import { pathToFileURL, fileURLToPath } from 'node:url';
 import fs, { mkdir, writeFile, readdir, stat, readFile } from 'fs/promises';
 
 class RakeDbError extends Error {
@@ -15,154 +15,6 @@ const clearChanges = () => {
 const getCurrentChanges = () => currentChanges;
 const pushChange = (change) => currentChanges.push(change);
 
-const migrationConfigDefaults = {
-  schemaConfig: defaultSchemaConfig,
-  migrationsPath: path.join("src", "db", "migrations"),
-  migrationId: { serial: 4 },
-  migrationsTable: "schemaMigrations",
-  snakeCase: false,
-  commands: {},
-  log: true,
-  logger: console,
-  import() {
-    throw new Error(
-      "Add `import: (path) => import(path),` setting to `rakeDb` config"
-    );
-  }
-};
-const ensureMigrationsPath = (config) => {
-  if (!config.migrationsPath) {
-    config.migrationsPath = migrationConfigDefaults.migrationsPath;
-  }
-  if (!path.isAbsolute(config.migrationsPath)) {
-    config.migrationsPath = path.resolve(
-      config.basePath,
-      config.migrationsPath
-    );
-  }
-  return config;
-};
-const ensureBasePathAndDbScript = (config, intermediateCallers = 0) => {
-  if (config.basePath && config.dbScript) return config;
-  let filePath = getStackTrace()?.[3 + intermediateCallers].getFileName();
-  if (!filePath) {
-    throw new Error(
-      "Failed to determine path to db script. Please set basePath option of rakeDb"
-    );
-  }
-  if (filePath.startsWith("file://")) {
-    filePath = fileURLToPath(filePath);
-  }
-  const ext = path.extname(filePath);
-  if (ext !== ".ts" && ext !== ".js" && ext !== ".mjs") {
-    throw new Error(
-      `Add a .ts suffix to the "${path.basename(filePath)}" when calling it`
-    );
-  }
-  config.basePath = path.dirname(filePath);
-  config.dbScript = path.basename(filePath);
-  return config;
-};
-const processRakeDbConfig = (config) => {
-  const result = { ...migrationConfigDefaults, ...config };
-  if (!result.log) {
-    delete result.logger;
-  }
-  ensureBasePathAndDbScript(result, 1);
-  ensureMigrationsPath(result);
-  if (!result.recurrentPath) {
-    result.recurrentPath = path.join(
-      result.migrationsPath,
-      "recurrent"
-    );
-  }
-  if ("recurrentPath" in result && !path.isAbsolute(result.recurrentPath)) {
-    result.recurrentPath = path.resolve(result.basePath, result.recurrentPath);
-  }
-  if ("baseTable" in config && config.baseTable) {
-    const { types, snakeCase, language } = config.baseTable.prototype;
-    result.columnTypes = types || makeColumnTypes(defaultSchemaConfig);
-    if (snakeCase) result.snakeCase = true;
-    if (language) result.language = language;
-  } else {
-    const ct = "columnTypes" in config && config.columnTypes;
-    result.columnTypes = (typeof ct === "function" ? ct(
-      makeColumnTypes(defaultSchemaConfig)
-    ) : ct) || makeColumnTypes(defaultSchemaConfig);
-  }
-  if (config.migrationId === "serial") {
-    result.migrationId = { serial: 4 };
-  }
-  return result;
-};
-
-const getFirstWordAndRest = (input) => {
-  const i = input.search(/(?=[A-Z])|[-_ ]/);
-  if (i !== -1) {
-    const restStart = input[i] === "-" || input[i] === "_" || input[i] === " " ? i + 1 : i;
-    const rest = input.slice(restStart);
-    return [input.slice(0, i), rest[0].toLowerCase() + rest.slice(1)];
-  } else {
-    return [input];
-  }
-};
-const getTextAfterRegExp = (input, regex, length) => {
-  let i = input.search(regex);
-  if (i === -1) return;
-  if (input[i] === "-" || input[i] === "_" || input[i] === " ") i++;
-  i += length;
-  const start = input[i] == "-" || input[i] === "_" || input[i] === " " ? i + 1 : i;
-  const text = input.slice(start);
-  return text[0].toLowerCase() + text.slice(1);
-};
-const getTextAfterTo = (input) => {
-  return getTextAfterRegExp(input, /(To|-to|_to| to)[A-Z-_ ]/, 2);
-};
-const getTextAfterFrom = (input) => {
-  return getTextAfterRegExp(input, /(From|-from|_from| from)[A-Z-_ ]/, 4);
-};
-const joinColumns = (columns) => {
-  return columns.map((column) => `"${column}"`).join(", ");
-};
-const quoteWithSchema = ({
-  schema,
-  name
-}) => quoteTable(schema, name);
-const quoteTable = (schema, table) => schema ? `"${schema}"."${table}"` : `"${table}"`;
-const getSchemaAndTableFromName = (name) => {
-  const i = name.indexOf(".");
-  return i !== -1 ? [name.slice(0, i), name.slice(i + 1)] : [void 0, name];
-};
-const quoteNameFromString = (string) => {
-  return quoteTable(...getSchemaAndTableFromName(string));
-};
-const quoteCustomType = (s) => {
-  const [schema, type] = getSchemaAndTableFromName(s);
-  return schema ? '"' + schema + '".' + type : type;
-};
-const quoteSchemaTable = (arg) => {
-  return singleQuote(concatSchemaAndName(arg));
-};
-const concatSchemaAndName = ({
-  schema,
-  name
-}) => {
-  return schema ? `${schema}.${name}` : name;
-};
-const makePopulateEnumQuery = (item) => {
-  const [schema, name] = getSchemaAndTableFromName(item.enumName);
-  return {
-    text: `SELECT unnest(enum_range(NULL::${quoteTable(schema, name)}))::text`,
-    then(result) {
-      item.options.push(...result.rows.map(([value]) => value));
-    }
-  };
-};
-const transaction = (adapter, fn) => {
-  return adapter.transaction(void 0, fn);
-};
-const queryLock = (trx) => trx.query(`SELECT pg_advisory_xact_lock('${RAKE_DB_LOCK_KEY}')`);
-
 const versionToString = (config, version) => config.migrationId === "timestamp" ? `${version}` : `${version}`.padStart(config.migrationId.serial, "0");
 const columnTypeToSql = (item) => {
   return item.data.isOfCustomType ? item instanceof DomainColumn ? quoteNameFromString(item.dataType) : quoteCustomType(item.toSQL()) : item.toSQL();
@@ -3066,6 +2918,9 @@ const createMigrationsTable = async (db, config) => {
 };
 
 const RAKE_DB_LOCK_KEY = "8582141715823621641";
+const transactionIfSingle = (params, fn) => {
+  return params.config.transaction === "single" ? transaction(params.adapter, fn) : fn(params.adapter);
+};
 function makeMigrateFn(up, fn) {
   return async (params) => {
     const ctx = params.ctx || {};
@@ -3074,7 +2929,7 @@ function makeMigrateFn(up, fn) {
     const force = params.force ?? false;
     let migrations;
     try {
-      await transaction(params.adapter, async (trx) => {
+      await transactionIfSingle(params, async (trx) => {
         const versions = await getMigratedVersionsMap(
           ctx,
           trx,
@@ -3085,7 +2940,7 @@ function makeMigrateFn(up, fn) {
       });
     } catch (err) {
       if (err instanceof NoMigrationsTableError) {
-        await transaction(params.adapter, async (trx) => {
+        await transactionIfSingle(params, async (trx) => {
           await createMigrationsTable(trx, params.config);
           const versions = await getMigratedVersionsMap(
             ctx,
@@ -3204,6 +3059,7 @@ const migrateOrRollback = async (trx, config, set, versions, count, up, redo2, f
   }
   let loggedAboutStarting = false;
   let migrations;
+  const migrationRunner = config.transaction === "single" ? runMigration : runMigrationInOwnTransaction;
   for (const file of set.migrations) {
     if (up && versionsMap[file.version] || !up && !versionsMap[file.version]) {
       continue;
@@ -3217,7 +3073,7 @@ const migrateOrRollback = async (trx, config, set, versions, count, up, redo2, f
       );
     }
     const changes = await getChanges(file, config);
-    const adapter = await runMigration(trx, up, changes, config);
+    const adapter = await migrationRunner(trx, up, changes, config);
     await changeMigratedVersion(adapter, up, file, config);
     (migrations ?? (migrations = [])).push(file);
     if (up) {
@@ -3296,6 +3152,9 @@ const getChanges = async (file, config) => {
   }
   return changes;
 };
+const runMigrationInOwnTransaction = (adapter, ...rest) => {
+  return transaction(adapter, (trx) => runMigration(trx, ...rest));
+};
 const runMigration = async (trx, up, changes, config) => {
   const db = createMigrationInterface(trx, up, config);
   if (changes.length) {
@@ -3317,6 +3176,176 @@ const changeMigratedVersion = async (adapter, up, file, config) => {
   );
 };
 
+const getFirstWordAndRest = (input) => {
+  const i = input.search(/(?=[A-Z])|[-_ ]/);
+  if (i !== -1) {
+    const restStart = input[i] === "-" || input[i] === "_" || input[i] === " " ? i + 1 : i;
+    const rest = input.slice(restStart);
+    return [input.slice(0, i), rest[0].toLowerCase() + rest.slice(1)];
+  } else {
+    return [input];
+  }
+};
+const getTextAfterRegExp = (input, regex, length) => {
+  let i = input.search(regex);
+  if (i === -1) return;
+  if (input[i] === "-" || input[i] === "_" || input[i] === " ") i++;
+  i += length;
+  const start = input[i] == "-" || input[i] === "_" || input[i] === " " ? i + 1 : i;
+  const text = input.slice(start);
+  return text[0].toLowerCase() + text.slice(1);
+};
+const getTextAfterTo = (input) => {
+  return getTextAfterRegExp(input, /(To|-to|_to| to)[A-Z-_ ]/, 2);
+};
+const getTextAfterFrom = (input) => {
+  return getTextAfterRegExp(input, /(From|-from|_from| from)[A-Z-_ ]/, 4);
+};
+const joinColumns = (columns) => {
+  return columns.map((column) => `"${column}"`).join(", ");
+};
+const quoteWithSchema = ({
+  schema,
+  name
+}) => quoteTable(schema, name);
+const quoteTable = (schema, table) => schema ? `"${schema}"."${table}"` : `"${table}"`;
+const getSchemaAndTableFromName = (name) => {
+  const i = name.indexOf(".");
+  return i !== -1 ? [name.slice(0, i), name.slice(i + 1)] : [void 0, name];
+};
+const quoteNameFromString = (string) => {
+  return quoteTable(...getSchemaAndTableFromName(string));
+};
+const quoteCustomType = (s) => {
+  const [schema, type] = getSchemaAndTableFromName(s);
+  return schema ? '"' + schema + '".' + type : type;
+};
+const quoteSchemaTable = (arg) => {
+  return singleQuote(concatSchemaAndName(arg));
+};
+const concatSchemaAndName = ({
+  schema,
+  name
+}) => {
+  return schema ? `${schema}.${name}` : name;
+};
+const makePopulateEnumQuery = (item) => {
+  const [schema, name] = getSchemaAndTableFromName(item.enumName);
+  return {
+    text: `SELECT unnest(enum_range(NULL::${quoteTable(schema, name)}))::text`,
+    then(result) {
+      item.options.push(...result.rows.map(([value]) => value));
+    }
+  };
+};
+const transaction = (adapter, fn) => {
+  return adapter.transaction(void 0, fn);
+};
+const queryLock = (trx) => trx.query(`SELECT pg_advisory_xact_lock('${RAKE_DB_LOCK_KEY}')`);
+const getCliParam = (args, name) => {
+  if (args) {
+    const key = "--" + name;
+    for (let i = 0; i < args.length; i += 1) {
+      const arg = args[i];
+      if (arg === key) return args[i + 1];
+      else if (arg.startsWith(key)) return arg.slice(key.length + 1);
+    }
+  }
+  return;
+};
+
+const migrationConfigDefaults = {
+  schemaConfig: defaultSchemaConfig,
+  migrationsPath: path.join("src", "db", "migrations"),
+  migrationId: { serial: 4 },
+  migrationsTable: "schemaMigrations",
+  snakeCase: false,
+  commands: {},
+  log: true,
+  logger: console,
+  import() {
+    throw new Error(
+      "Add `import: (path) => import(path),` setting to `rakeDb` config"
+    );
+  }
+};
+const ensureMigrationsPath = (config) => {
+  if (!config.migrationsPath) {
+    config.migrationsPath = migrationConfigDefaults.migrationsPath;
+  }
+  if (!path.isAbsolute(config.migrationsPath)) {
+    config.migrationsPath = path.resolve(
+      config.basePath,
+      config.migrationsPath
+    );
+  }
+  return config;
+};
+const ensureBasePathAndDbScript = (config, intermediateCallers = 0) => {
+  if (config.basePath && config.dbScript) return config;
+  let filePath = getStackTrace()?.[3 + intermediateCallers].getFileName();
+  if (!filePath) {
+    throw new Error(
+      "Failed to determine path to db script. Please set basePath option of rakeDb"
+    );
+  }
+  if (filePath.startsWith("file://")) {
+    filePath = fileURLToPath(filePath);
+  }
+  const ext = path.extname(filePath);
+  if (ext !== ".ts" && ext !== ".js" && ext !== ".mjs") {
+    throw new Error(
+      `Add a .ts suffix to the "${path.basename(filePath)}" when calling it`
+    );
+  }
+  config.basePath = path.dirname(filePath);
+  config.dbScript = path.basename(filePath);
+  return config;
+};
+const processRakeDbConfig = (config, args) => {
+  const result = { ...migrationConfigDefaults, ...config };
+  if (!result.log) {
+    delete result.logger;
+  }
+  ensureBasePathAndDbScript(result, 1);
+  ensureMigrationsPath(result);
+  if (!result.recurrentPath) {
+    result.recurrentPath = path.join(
+      result.migrationsPath,
+      "recurrent"
+    );
+  }
+  if ("recurrentPath" in result && !path.isAbsolute(result.recurrentPath)) {
+    result.recurrentPath = path.resolve(result.basePath, result.recurrentPath);
+  }
+  if ("baseTable" in config && config.baseTable) {
+    const { types, snakeCase, language } = config.baseTable.prototype;
+    result.columnTypes = types || makeColumnTypes(defaultSchemaConfig);
+    if (snakeCase) result.snakeCase = true;
+    if (language) result.language = language;
+  } else {
+    const ct = "columnTypes" in config && config.columnTypes;
+    result.columnTypes = (typeof ct === "function" ? ct(
+      makeColumnTypes(defaultSchemaConfig)
+    ) : ct) || makeColumnTypes(defaultSchemaConfig);
+  }
+  if (config.migrationId === "serial") {
+    result.migrationId = { serial: 4 };
+  }
+  const transaction = getCliParam(args, "transaction");
+  if (transaction) {
+    if (transaction !== "single" && transaction !== "per-migration") {
+      throw new Error(
+        `Unsupported transaction param ${transaction}, expected single or per-migration`
+      );
+    }
+    result.transaction = transaction;
+  } else if (!result.transaction) {
+    result.transaction = "single";
+  }
+  return result;
+};
+
 const ESC = "\x1B";
 const CSI = `${ESC}[`;
 const cursorShow = `${CSI}?25h`;
@@ -5784,7 +5813,7 @@ const rakeDbCommands = {
 };
 
 const rakeDbWithAdapters = (adapters, partialConfig, args = process.argv.slice(2)) => {
-  const config = processRakeDbConfig(partialConfig);
+  const config = processRakeDbConfig(partialConfig, args);
   const promise = runCommand(
     adapters,
     config,
@@ -5838,7 +5867,8 @@ const makeMigrateAdapter = (config) => {
         columnTypes: conf.columnTypes || makeColumnTypes(defaultSchemaConfig),
         migrationId: conf.migrationId || migrationConfigDefaults.migrationId,
         migrationsTable: conf.migrationsTable || migrationConfigDefaults.migrationsTable,
-        import: conf.import || migrationConfigDefaults.import
+        import: conf.import || migrationConfigDefaults.import,
+        transaction: conf.transaction || "single"
       }
     });
   };