rake-db 2.3.8 → 2.3.9

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # rake-db
 
+## 2.3.9
+
+### Patch Changes
+
+- Add migrate and rollback callbacks
+
 ## 2.3.8
 
 ### Patch Changes

package/dist/index.d.ts

@@ -389,6 +389,7 @@ declare namespace RakeDbAst {
     };
 }
 
+declare type Db = DbResult<typeof columnTypes>;
 declare type RakeDbConfig = {
     migrationsPath: string;
     migrationsTable: string;
@@ -396,6 +397,10 @@ declare type RakeDbConfig = {
     noPrimaryKey?: NoPrimaryKeyOption;
     appCodeUpdater?: AppCodeUpdater;
     useCodeUpdater?: boolean;
+    beforeMigrate?(db: Db): Promise<void>;
+    afterMigrate?(db: Db): Promise<void>;
+    beforeRollback?(db: Db): Promise<void>;
+    afterRollback?(db: Db): Promise<void>;
 } & QueryLogOptions;
 declare type AppCodeUpdater = (params: {
     ast: RakeDbAst;

package/dist/index.js

@@ -1137,8 +1137,9 @@ var __spreadValues$1 = (a, b) => {
     }
   return a;
 };
+const getDb = (adapter) => pqb.createDb({ adapter, columnTypes: pqb.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;
@@ -1156,8 +1157,14 @@ const migrateOrRollback = async (options, config, args, up) => {
     delete config.appCodeUpdater;
   const appCodeUpdaterCache = {};
   for (const opts of pqb.toArray(options)) {
-    const db = new pqb.Adapter(opts);
-    const migratedVersions = await getMigratedVersionsMap(db, config);
+    const adapter = new pqb.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]) {
@@ -1165,11 +1172,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;
   }