Vizzuality/landgriffon

import { CallHandler, ExecutionContext, NestInterceptor } from '@nestjs/common';
import { Observable } from 'rxjs';

export class SetScenarioIdsInterceptor implements NestInterceptor {
  intercept(context: ExecutionContext, next: CallHandler): Observable<any> {
    const request: any = context.switchToHttp().getRequest();

    // For dtos of impact and actual-vs-scenario
    if (request.query.scenarioId) {
      request.query.scenarioIds = [request.query.scenarioId];
    }

    if (request.query.comparedScenarioId) {
      request.query.scenarioIds = [request.query.comparedScenarioId];
    }
    // For dto of scenario vs scenario
    if (request.query.baseScenarioId && request.query.comparedScenarioId) {
      request.query.scenarioIds = [
        request.query.baseScenarioId,
        request.query.comparedScenarioId,
      ];
    }

    return next.handle();
  }
}