rake-db 2.3.8 → 2.3.10

package/dist/index.mjs

@@ -23,8 +23,9 @@ var __spreadValues$6 = (a, b) => {
 };
 var __spreadProps$5 = (a, b) => __defProps$5(a, __getOwnPropDescs$5(b));
 const migrationConfigDefaults = {
-  migrationsPath: path.resolve("src", "migrations"),
+  migrationsPath: path.resolve("src", "db", "migrations"),
   migrationsTable: "schemaMigrations",
+  commands: {},
   requireTs: (path2) => import(path2),
   log: true,
   logger: console,
@@ -1110,8 +1111,9 @@ var __spreadValues$1 = (a, b) => {
     }
   return a;
 };
+const getDb = (adapter) => createDb$1({ adapter, columnTypes });
 const migrateOrRollback = async (options, config, args, up) => {
-  var _a;
+  var _a, _b, _c, _d, _e;
   config = __spreadValues$1({}, config);
   const files = await getMigrationFiles(config, up);
   let count = up ? Infinity : 1;
@@ -1129,8 +1131,14 @@ const migrateOrRollback = async (options, config, args, up) => {
     delete config.appCodeUpdater;
   const appCodeUpdaterCache = {};
   for (const opts of toArray(options)) {
-    const db = new Adapter(opts);
-    const migratedVersions = await getMigratedVersionsMap(db, config);
+    const adapter = new Adapter(opts);
+    let db;
+    if (up) {
+      await ((_a = config.beforeMigrate) == null ? void 0 : _a.call(config, db != null ? db : db = getDb(adapter)));
+    } else {
+      await ((_b = config.beforeRollback) == null ? void 0 : _b.call(config, db != null ? db : db = getDb(adapter)));
+    }
+    const migratedVersions = await getMigratedVersionsMap(adapter, config);
     try {
       for (const file of files) {
         if (up && migratedVersions[file.version] || !up && !migratedVersions[file.version]) {
@@ -1138,11 +1146,23 @@ const migrateOrRollback = async (options, config, args, up) => {
         }
         if (count-- <= 0)
           break;
-        await processMigration(db, up, file, config, opts, appCodeUpdaterCache);
-        (_a = config.logger) == null ? void 0 : _a.log(`${file.path} ${up ? "migrated" : "rolled back"}`);
+        await processMigration(
+          adapter,
+          up,
+          file,
+          config,
+          opts,
+          appCodeUpdaterCache
+        );
+        (_c = config.logger) == null ? void 0 : _c.log(`${file.path} ${up ? "migrated" : "rolled back"}`);
+      }
+      if (up) {
+        await ((_d = config.afterMigrate) == null ? void 0 : _d.call(config, db != null ? db : db = getDb(adapter)));
+      } else {
+        await ((_e = config.afterRollback) == null ? void 0 : _e.call(config, db != null ? db : db = getDb(adapter)));
       }
     } finally {
-      await db.close();
+      await adapter.close();
     }
     delete config.appCodeUpdater;
   }
@@ -1915,6 +1935,8 @@ const rakeDb = async (options, partialConfig = {}, args = process.argv.slice(2))
     await generate(config, args.slice(1));
   } else if (command === "pull") {
     await pullDbStructure(toArray(options)[0], config);
+  } else if (config.commands[command]) {
+    await config.commands[command](toArray(options), config, args.slice(1));
   } else {
     printHelp();
   }