Vizzuality/landgriffon

import {
  Controller,
  Get,
  Query,
  Res,
  UseInterceptors,
  ValidationPipe,
} from '@nestjs/common';
import { ImpactService } from 'modules/impact/impact.service';
import { ActualVsScenarioImpactService } from 'modules/impact/comparison/actual-vs-scenario.service';
import { ScenarioVsScenarioImpactService } from 'modules/impact/comparison/scenario-vs-scenario.service';
import { ApiOperation, ApiTags } from '@nestjs/swagger';
import { SetScenarioIdsInterceptor } from 'modules/impact/set-scenario-ids.interceptor';
import { ImpactReportService } from 'modules/impact/reports/impact.report';
import { Response } from 'express';
import {
  GetActualVsScenarioImpactTableDto,
  GetImpactTableDto,
  GetScenarioVsScenarioImpactTableDto,
} from 'modules/impact/dto/impact-table.dto';
import { CheckUserOwnsScenario } from 'modules/authorization/formodule/scenario-ownership.interceptor';

@Controller('/api/v1/impact')
@ApiTags('Impact')
export class ImpactReportController {
  constructor(
    private readonly impactService: ImpactService,
    private readonly impactReports: ImpactReportService,
    private readonly actualVsScenarioImpactService: ActualVsScenarioImpactService,
    private readonly scenarioVsScenarioService: ScenarioVsScenarioImpactService,
  ) {}

  @ApiOperation({
    description: 'Get a Impact Table CSV Report',
  })
  @UseInterceptors(SetScenarioIdsInterceptor)
  @Get('table/report')
  async getImpactReport(
    @Query(ValidationPipe) impactTableDto: GetImpactTableDto,
    @Res() res: Response,
  ): Promise<void> {
    const { data } = await this.impactService.getImpactTable(impactTableDto, {
      disablePagination: true,
    });
    const report: string = await this.impactReports.generateImpactReport(
      data.impactTable,
    );
    res.setHeader('Content-Type', 'text/csv');
    res.setHeader(
      'Content-Disposition',
      'attachment; filename=impact_report.csv',
    );
    res.send(report);
  }

  @ApiOperation({
    description:
      'Get a Actual Vs Scenario Impact Table CSV Report for a given scenario',
  })
  @CheckUserOwnsScenario({
    bypassIfScenarioIsPublic: true,
    isComparisonMode: true,
  })
  @UseInterceptors(SetScenarioIdsInterceptor)
  @Get('compare/scenario/vs/actual/report')
  async getActualVsScenarioImpactReport(
    @Query(ValidationPipe)
    actualVsScenarioImpactTableDto: GetActualVsScenarioImpactTableDto,
    @Res() res: Response,
  ): Promise<void> {
    const { data } =
      await this.actualVsScenarioImpactService.getActualVsScenarioImpactTable(
        actualVsScenarioImpactTableDto,
        { disablePagination: true },
      );
    const report: string = await this.impactReports.generateImpactReport(
      data.impactTable,
    );
    res.setHeader('Content-Type', 'text/csv');
    res.setHeader(
      'Content-Disposition',
      'attachment; filename=impact_report.csv',
    );
    res.send(report);
  }

  @ApiOperation({
    description:
      'Get a Scenario Vs Scenario Impact Table CSV Report for 2 Scenarios',
  })
  @CheckUserOwnsScenario({
    bypassIfScenarioIsPublic: true,
    isComparisonMode: true,
  })
  @UseInterceptors(SetScenarioIdsInterceptor)
  @Get('compare/scenario/vs/scenario/report')
  async getTwoScenariosImpacReport(
    @Query(ValidationPipe)
    scenarioVsScenarioImpactTableDto: GetScenarioVsScenarioImpactTableDto,
    @Res() res: Response,
  ): Promise<void> {
    const { data } =
      await this.scenarioVsScenarioService.getScenarioVsScenarioImpactTable(
        scenarioVsScenarioImpactTableDto,
        { disablePagination: true },
      );
    const report: string = await this.impactReports.generateImpactReport(
      data.impactTable,
    );
    res.setHeader('Content-Type', 'text/csv');
    res.setHeader(
      'Content-Disposition',
      'attachment; filename=impact_report.csv',
    );
    res.send(report);
  }
}