icynet-auth-server/src/modules/features/two-factor/two-factor.module.ts

import { MiddlewareConsumer, Module, NestModule } from '@nestjs/common';
import { AuthMiddleware } from 'src/middleware/auth.middleware';
import { FlashMiddleware } from 'src/middleware/flash.middleware';
import { UserModule } from 'src/modules/objects/user/user.module';
import { TwoFactorController } from './two-factor.controller';

@Module({
  imports: [UserModule],
  controllers: [TwoFactorController],
})
export class TwoFactorModule implements NestModule {
  configure(consumer: MiddlewareConsumer) {
    consumer.apply(AuthMiddleware).forRoutes('two-factor');
    consumer.apply(FlashMiddleware).forRoutes('*');
  }
}