backend/src/unit/unit.module.ts

import {
  MiddlewareConsumer,
  Module,
  NestModule,
  RequestMethod,
} from '@nestjs/common';
import { UnitService } from './services/unit.service';
import { UnitController } from './controllers/unit.controller';
import { ConfigModule } from '@nestjs/config';
import { SpaceRepositoryModule } from '@app/common/modules/space/space.repository.module';
import { SpaceRepository } from '@app/common/modules/space/repositories';
import { SpaceTypeRepositoryModule } from '@app/common/modules/space-type/space.type.repository.module';
import { SpaceTypeRepository } from '@app/common/modules/space-type/repositories';
import { CheckFloorTypeMiddleware } from 'src/middleware/CheckFloorTypeMiddleware';

@Module({
  imports: [ConfigModule, SpaceRepositoryModule, SpaceTypeRepositoryModule],
  controllers: [UnitController],
  providers: [UnitService, SpaceRepository, SpaceTypeRepository],
  exports: [UnitService],
})
export class UnitModule implements NestModule {
  configure(consumer: MiddlewareConsumer) {
    consumer.apply(CheckFloorTypeMiddleware).forRoutes({
      path: '/unit',
      method: RequestMethod.POST,
    });
  }
}