app.module.ts

import { MiddlewareConsumer, Module, NestModule } from '@nestjs/common';
import { ConfigModule, ConfigService } from '@nestjs/config';
import { APP_GUARD } from '@nestjs/core';
import { ThrottlerModule, ThrottlerGuard } from '@nestjs/throttler';
import { WinstonModule } from 'nest-winston';
import { DatabaseModule } from './database/database.module';
import { GeoModule } from './geo/geo.module';
import { IpfsModule } from './ipfs/ipfs.module';
import { SorobanModule } from './soroban/soroban.module';
import { GistsModule } from './gists/gists.module';
import { HealthModule } from './health/health.module';
import { MetricsModule } from './metrics/metrics.module';
import { ShutdownModule } from './common/shutdown/shutdown.module';
import { envValidationSchema } from './config/env.validation';
import { CorrelationIdMiddleware } from './common/middleware/correlation-id.middleware';
import { InFlightRequestMiddleware } from './common/shutdown/in-flight.middleware';
import { buildWinstonOptions } from './common/logger/winston.config';

@Module({
  imports: [
    ConfigModule.forRoot({
      isGlobal: true,
      validationSchema: envValidationSchema,
      validationOptions: { abortEarly: false },
    }),
    WinstonModule.forRootAsync({
      imports: [ConfigModule],
      inject: [ConfigService],
      useFactory: (config: ConfigService) =>
        buildWinstonOptions(
          config.get<string>('NODE_ENV', 'development'),
          config.get<string>('LOG_DIR', 'logs'),
        ),
    }),
    ThrottlerModule.forRootAsync({
      imports: [ConfigModule],
      inject: [ConfigService],
      useFactory: (config: ConfigService) => [
        {
          ttl: config.get<number>('THROTTLE_TTL_MS', 60000),
          limit: config.get<number>('THROTTLE_LIMIT', 10),
        },
      ],
    }),
    DatabaseModule,
    GeoModule,
    IpfsModule,
    SorobanModule,
    GistsModule,
    HealthModule,
    MetricsModule,
    ShutdownModule,
  ],
  providers: [
    {
      provide: APP_GUARD,
      useClass: ThrottlerGuard,
    },
  ],
})
export class AppModule implements NestModule {
  configure(consumer: MiddlewareConsumer) {
    // In-flight-request middleware must run before correlation-id so that
    // every response (including 404s) is counted toward the drain window.
    consumer
      .apply(InFlightRequestMiddleware, CorrelationIdMiddleware)
      .forRoutes('*');
  }
}