Refactor app.module.ts to update MigrationsService import and add MigrationsModule

Author: tacxou

src/app.module.ts

@@ -20,7 +20,8 @@ import { MailerModule } from '@nestjs-modules/mailer';
 import { HandlebarsAdapter } from '@nestjs-modules/mailer/dist/adapters/handlebars.adapter';
 import { MailadmService } from '~/settings/mailadm.service';
 import { FactorydriveModule } from '@the-software-compagny/nestjs_module_factorydrive';
-import { MigrationsService } from './migrations.service';
+import { MigrationsService } from './migrations/migrations.service';
+import { MigrationsModule } from './migrations/migrations.module';
 
 @Module({
   imports: [
@@ -93,11 +94,11 @@ import { MigrationsService } from './migrations.service';
     CoreModule.register(),
     ManagementModule.register(),
     SettingsModule.register(),
+    MigrationsModule.register(),
   ],
   controllers: [AppController],
   providers: [
     AppService,
-    MigrationsService,
     {
       provide: APP_GUARD,
       useClass: AuthGuard('jwt'),

src/management/identities/identities.service.ts

@@ -51,7 +51,6 @@ export class IdentitiesService extends AbstractServiceSchema {
     data?: IdentitiesUpsertDto,
     options?: QueryOptions<T>,
   ): Promise<[HttpStatus.OK | HttpStatus.CREATED, ModifyResult<Query<T, T, any, T>>]> {
-
     data = this.transformNullsToString(data);
     const identity = await this.model.findOne<Identities>(filters).exec();
     this.logger.log(`Upserting identity with filters ${JSON.stringify(filters)}`);
@@ -76,7 +75,7 @@ export class IdentitiesService extends AbstractServiceSchema {
         'inetOrgPerson.employeeNumber and inetOrgPerson.employeeType are required for create identity.',
       );
     }
-    if ( data.inetOrgPerson?.employeeNumber.indexOf('174981') >= 0 || data.inetOrgPerson?.employeeNumber.indexOf('162982') >= 0 ){
+    if (data.inetOrgPerson?.employeeNumber.indexOf('174981') >= 0 || data.inetOrgPerson?.employeeNumber.indexOf('162982') >= 0) {
       console.log('test');
     }
     //controle si l identité est fusionnée si c est la bonne à mettre à jour puisqu elle a 2 employeeNumber

src/migrations.service.ts

@@ -0,0 +1,32 @@
+import { Logger } from "@nestjs/common"
+import { InjectConnection } from "@nestjs/mongoose"
+import { Connection } from "mongoose"
+
+export default class EmployeeNumber1616027432511 {
+  public constructor(@InjectConnection() private connection: Connection) {
+
+  }
+
+  public async up(): Promise<void> {
+    Logger.log('EmployeeNumber1616027432511 up')
+
+    Logger.log('this.connection.readyState', this.connection.readyState)
+
+    return new Promise((resolve) => {
+      const interval = setInterval(() => {
+        Logger.log('EmployeeNumber1616027432511 up in progress')
+      }, 1_000)
+
+      setTimeout(() => {
+        clearInterval(interval)
+        Logger.log('EmployeeNumber1616027432511 up done')
+
+        resolve()
+      }, 5_000)
+    })
+  }
+
+  public async down(): Promise<void> {
+    Logger.log('EmployeeNumber1616027432511 down')
+  }
+}

src/migrations/1316027432511-employee-number.ts

@@ -0,0 +1,39 @@
+import { DynamicModule, Logger, Module } from '@nestjs/common';
+import { glob } from 'glob';
+import chalk from 'chalk';
+import { MigrationsService } from './migrations.service';
+
+@Module({
+  imports: [],
+  providers: [
+    MigrationsService,
+  ],
+})
+export class MigrationsModule {
+  public static async register(): Promise<DynamicModule> {
+    const jobs = [];
+    const files = await glob(`./jobs/*.js`, {
+      cwd: __dirname,
+      root: __dirname,
+    });
+
+    for (const file of files) {
+      const migration = await import(`${__dirname}/${file}`);
+
+      if (!migration.default) {
+        Logger.log(chalk.yellow(`Migration ${chalk.bold('<' + file + '>')} does not have a default export !`));
+        return;
+      }
+
+      jobs.push(migration.default);
+    }
+
+    return {
+      module: this,
+      providers: [
+        ...Reflect.getMetadata('providers', this),
+        ...jobs,
+      ],
+    }
+  }
+}

src/migrations/jobs/1316027432511-employee-number.ts

@@ -0,0 +1,82 @@
+import { Injectable, Logger, OnModuleInit } from '@nestjs/common'
+import { InjectModel } from '@nestjs/mongoose'
+import { Model } from 'mongoose'
+import { glob } from 'glob'
+import readline from 'readline'
+import chalk from 'chalk'
+
+function startLoader(message) {
+  let currentFrame = 0;
+  const spinnerFrames = ['-', '\\', '|', '/'];
+
+  const loaderInterval = setInterval(() => {
+    readline.cursorTo(process.stdout, 0);
+    process.stdout.write(`${spinnerFrames[currentFrame]} ${message} `);
+    currentFrame = (currentFrame + 1) % spinnerFrames.length;
+  }, 100);
+
+  return loaderInterval;
+}
+
+function stopLoader(loaderInterval) {
+  clearInterval(loaderInterval);
+  readline.cursorTo(process.stdout, 0);
+  process.stdout.clearLine(0);
+}
+
+@Injectable()
+export class MigrationsService implements OnModuleInit {
+  private readonly logger = new Logger(`${chalk.bold.red(MigrationsService.name)}\x1b[33m`)
+
+  protected migrations = new Map<string, any>()
+
+  public constructor() { }
+
+  public async onModuleInit() {
+    await this.runMigrations()
+  }
+
+  public async runMigrations() {
+    this.logger.log(chalk.yellow('Checking migrations files...'));
+    await this._loadMigrationsFiles()
+
+
+    const loader = startLoader('Migration en cours...');
+    await this._executeMigrations();
+    stopLoader(loader);
+  }
+
+  private async _loadMigrationsFiles() {
+    const files = await glob(`./migrations/*.js`, {
+      cwd: __dirname,
+      root: __dirname,
+    });
+
+    for (const file of files) {
+      const migration = await import(`${__dirname}/${file}`);
+
+      if (!migration.default) {
+        this.logger.log(chalk.yellow(`Migration ${chalk.bold('<' + file + '>')} does not have a default export !`));
+        return;
+      }
+
+      this.migrations.set(file, migration)
+    }
+  }
+
+  private async _executeMigrations() {
+    console.log('this.migrations.keys()', this.migrations.keys())
+    for (const key of this.migrations.keys()) {
+      const migration = this.migrations.get(key);
+      const instance = new migration.default();
+
+      if (typeof instance.up !== 'function') {
+        this.logger.log(chalk.yellow(`Migration ${chalk.bold('<' + key + '>')} does not have an up method !`));
+        break;
+      }
+
+      this.logger.log(chalk.yellow(`Running migration ${chalk.bold('<' + key + '>')}...`));
+      await instance.up();
+    }
+  }
+}