feature(SeparatedConnection): added option hooks to change default connection

Author: Stradivario

src/injection.tokens.ts

@@ -37,6 +37,12 @@ export interface Config {
       useNewUrlParser: boolean;
     };
   };
+  hooks?: {
+    connection: <T = any>() => Promise<T>;
+    down: (fileName: string) => Promise<any>;
+    up: (fileName: string) => Promise<any>;
+    status: (fileNames: string[]) => Promise<ReturnType[]>;
+  };
   dateTimeFormat?: () => string;
   outDir: string;
   migrationsDir?: string;

src/services/migration/migration.service.ts

@@ -24,6 +24,9 @@ export class MigrationService {
   ) {}
 
   async connect() {
+    if (this.configService.config.hooks) {
+      return this.configService.config.hooks.connection() as never;
+    }
     await this.database.mongooseConnect();
     return this.database.connect();
   }
@@ -62,14 +65,22 @@ export class MigrationService {
         });
         throw error;
       }
-      const collection = client
-        .db()
-        .collection(this.configService.config.changelogCollectionName);
+
       const { fileName } = item;
       const appliedAt = new Date();
 
       try {
-        await collection.insertOne({ fileName, appliedAt });
+        if (
+          this.configService.config.hooks &&
+          typeof this.configService.config.hooks.up === 'function'
+        ) {
+          await this.configService.config.hooks.up(fileName);
+        } else {
+          const collection = client
+            .db()
+            .collection(this.configService.config.changelogCollectionName);
+          await collection.insertOne({ fileName, appliedAt });
+        }
       } catch (err) {
         await logger.error({
           migrated,
@@ -135,11 +146,20 @@ export class MigrationService {
         });
         throw error;
       }
-      const collection = client
-        .db()
-        .collection(this.configService.config.changelogCollectionName);
+
       try {
-        await collection.deleteOne({ fileName: lastAppliedItem.fileName });
+        if (
+          this.configService.config.hooks &&
+          typeof this.configService.config.hooks.down === 'function'
+        ) {
+          await this.configService.config.hooks.down(lastAppliedItem.fileName);
+        } else {
+          const collection = client
+            .db()
+            .collection(this.configService.config.changelogCollectionName);
+          await collection.deleteOne({ fileName: lastAppliedItem.fileName });
+        }
+
         const res: ReturnType = {
           fileName: lastAppliedItem.fileName,
           appliedAt: new Date(),
@@ -230,6 +250,12 @@ export class MigrationService {
 
   async statusInternal() {
     const fileNames = await this.migrationsResolver.getFileNames();
+    if (
+      this.configService.config.hooks &&
+      typeof this.configService.config.hooks.status === 'function'
+    ) {
+      return this.configService.config.hooks.status(fileNames);
+    }
     const client = await this.connect();
     const collection = client
       .db()

xmigrate.ts

@@ -3,6 +3,38 @@ export default async () => {
     defaultTemplate: 'typescript',
     outDir: './.xmigrate',
     typescript: true,
+    hooks: {
+      connection: () => {
+        return {
+          db: () => ({ collection: () => ({ updateOne: () => null }) }),
+        };
+      },
+      up: (fileName: string) => {
+        /* Save up migration */
+        // Migrations.save({ fileName, createdAt: new Date() });
+        console.log('up');
+        return {};
+      },
+      down: (fileName: string) => {
+        /* Remove migration from changelog */
+        // Migrations.remove(fileName);
+        console.log('down');
+        return {};
+      },
+      status: (fileNames: string[]) => {
+        /* Get all migrations from database */
+        const prevMigrations = [
+          { fileName: '2020102412521-pesho.ts', appliedAt: new Date() },
+        ];
+        return fileNames.map((fileName) => {
+          const itemInLog = prevMigrations.find((m) => m.fileName === fileName);
+          const appliedAt = itemInLog
+            ? (itemInLog.appliedAt as Date).toJSON()
+            : 'PENDING';
+          return { fileName, appliedAt, result: null };
+        });
+      },
+    },
     // dateTimeFormat: () => new Date().toISOString(),
     mongodb: {
       url: 'mongodb://localhost:27017',