rake-db 2.10.25 → 2.10.27

package/dist/index.mjs

@@ -3,7 +3,7 @@ import { getCallerFilePath, singleQuote, toSnakeCase, isRawSQL, toArray, snakeCa
 import path, { join } from 'path';
 import { readdir, mkdir, writeFile, stat, readFile } from 'fs/promises';
 import prompts from 'prompts';
-import * as url from 'url';
+import { pathToFileURL } from 'url';
 
 var __defProp$6 = Object.defineProperty;
 var __defProps$5 = Object.defineProperties;
@@ -45,8 +45,9 @@ const migrationConfigDefaults = {
 const processRakeDbConfig = (config) => {
   var _a;
   const result = __spreadValues$6(__spreadValues$6({}, migrationConfigDefaults), config);
-  if (!result.recurrentPath)
+  if (!result.recurrentPath) {
     result.recurrentPath = path.join(result.migrationsPath, "recurrent");
+  }
   if (config.appCodeUpdater && (!("baseTable" in config) || !config.baseTable)) {
     throw new Error(
       "`baseTable` option is required in `rakeDb` for `appCodeUpdater`"
@@ -62,16 +63,22 @@ const processRakeDbConfig = (config) => {
         "Failed to determine path to db script. Please set basePath option of rakeDb"
       );
     }
+    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`
+      );
+    }
     result.basePath = path.dirname(filePath);
     result.dbScript = path.basename(filePath);
   }
-  if (!path.isAbsolute(result.migrationsPath)) {
+  if ("migrationsPath" in result && !path.isAbsolute(result.migrationsPath)) {
     result.migrationsPath = path.resolve(
       result.basePath,
       result.migrationsPath
     );
   }
-  if (!path.isAbsolute(result.recurrentPath)) {
+  if ("recurrentPath" in result && !path.isAbsolute(result.recurrentPath)) {
     result.recurrentPath = path.resolve(result.basePath, result.recurrentPath);
   }
   if ("baseTable" in config) {
@@ -202,35 +209,63 @@ const getTextAfterTo = (input) => {
 const getTextAfterFrom = (input) => {
   return getTextAfterRegExp(input, /(From|-from|_from)[A-Z-_]/, 4);
 };
-const getMigrationFiles = async (config, up) => {
-  const { migrationsPath } = config;
+const getMigrations = async (config, up) => {
+  if ("migrations" in config) {
+    const result = [];
+    const { migrations, basePath } = config;
+    for (const key in migrations) {
+      result.push({
+        path: path.resolve(basePath, key),
+        version: getVersion(path.basename(key)),
+        change: migrations[key]
+      });
+    }
+    return result;
+  }
+  const { migrationsPath, import: imp } = config;
   let files;
   try {
     files = await readdir(migrationsPath);
   } catch (_) {
     return [];
   }
-  const sort = up ? sortAsc : sortDesc;
-  return sort(files.filter((file) => path.basename(file).includes("."))).map(
-    (file) => {
-      if (!file.endsWith(".ts")) {
-        throw new Error(
-          `Only .ts files are supported for migration, received: ${file}`
-        );
-      }
-      const timestampMatch = file.match(/^(\d{14})\D/);
-      if (!timestampMatch) {
-        throw new Error(
-          `Migration file name should start with 14 digit version, received ${file}`
-        );
+  files = files.filter((file) => path.basename(file).includes("."));
+  files = (up ? sortAsc : sortDesc)(files);
+  return files.map((file) => {
+    checkExt(file);
+    const filePath = path.resolve(migrationsPath, file);
+    return {
+      path: filePath,
+      version: getVersion(file),
+      async change() {
+        try {
+          await imp(filePath);
+        } catch (err) {
+          if (err.code !== "ERR_UNSUPPORTED_ESM_URL_SCHEME")
+            throw err;
+          await imp(pathToFileURL(filePath).pathname);
+        }
       }
-      return {
-        path: path.resolve(migrationsPath, file),
-        version: timestampMatch[1]
-      };
-    }
-  );
+    };
+  });
 };
+function checkExt(filePath) {
+  const ext = path.extname(filePath);
+  if (ext !== ".ts" && ext !== ".js" && ext !== ".mjs") {
+    throw new Error(
+      `Only .ts and .js files are supported for migration, received: ${path}`
+    );
+  }
+}
+function getVersion(path2) {
+  const timestampMatch = path2.match(/^(\d{14})\D/);
+  if (!timestampMatch) {
+    throw new Error(
+      `Migration file name should start with 14 digit version, received ${path2}`
+    );
+  }
+  return timestampMatch[1];
+}
 const sortAsc = (arr) => arr.sort();
 const sortDesc = (arr) => arr.sort((a, b) => a > b ? -1 : 1);
 const joinColumns = (columns) => {
@@ -2168,7 +2203,7 @@ const getDb = (adapter) => createDb$1({ adapter });
 const migrateOrRollback = async (options, config, args, up) => {
   var _a, _b, _c, _d, _e, _f;
   config = __spreadValues$1({}, config);
-  const files = await getMigrationFiles(config, up);
+  const files = await getMigrations(config, up);
   let count = up ? Infinity : 1;
   let argI = 0;
   const num = args[0] === "all" ? Infinity : parseInt(args[0]);
@@ -2242,13 +2277,7 @@ const processMigration = async (db, up, file, config, options, appCodeUpdaterCac
     clearChanges();
     let changes = changeCache[file.path];
     if (!changes) {
-      try {
-        await config.import(file.path);
-      } catch (err) {
-        if (err.code !== "ERR_UNSUPPORTED_ESM_URL_SCHEME")
-          throw err;
-        await config.import(url.pathToFileURL(file.path).pathname);
-      }
+      await file.change();
       changes = getCurrentChanges();
       changeCache[file.path] = changes;
     }