rake-db 2.3.8 → 2.3.10

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # rake-db
 
+## 2.3.10
+
+### Patch Changes
+
+- Add custom commands to rake-db
+
+## 2.3.9
+
+### Patch Changes
+
+- Add migrate and rollback callbacks
+
 ## 2.3.8
 
 ### Patch Changes

package/dist/index.d.ts

@@ -389,13 +389,19 @@ declare namespace RakeDbAst {
     };
 }
 
+declare type Db = DbResult<typeof columnTypes>;
 declare type RakeDbConfig = {
     migrationsPath: string;
     migrationsTable: string;
+    commands: Record<string, (options: AdapterOptions[], config: RakeDbConfig, args: string[]) => Promise<void>>;
     requireTs(path: string): Promise<void>;
     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

@@ -50,8 +50,9 @@ var __spreadValues$6 = (a, b) => {
 };
 var __spreadProps$5 = (a, b) => __defProps$5(a, __getOwnPropDescs$5(b));
 const migrationConfigDefaults = {
-  migrationsPath: path__default["default"].resolve("src", "migrations"),
+  migrationsPath: path__default["default"].resolve("src", "db", "migrations"),
   migrationsTable: "schemaMigrations",
+  commands: {},
   requireTs: (path2) => (function (t) { return Promise.resolve().then(function () { return /*#__PURE__*/_interopNamespace(require(t)); }); })(path2),
   log: true,
   logger: console,
@@ -1137,8 +1138,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 +1158,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 +1173,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;
   }
@@ -1942,6 +1962,8 @@ const rakeDb = async (options, partialConfig = {}, args = process.argv.slice(2))
     await generate(config, args.slice(1));
   } else if (command === "pull") {
     await pullDbStructure(pqb.toArray(options)[0], config);
+  } else if (config.commands[command]) {
+    await config.commands[command](pqb.toArray(options), config, args.slice(1));
   } else {
     printHelp();
   }