SimulationResultsExport.tsx

import { useCallback, useMemo } from 'react';

import { Button } from '@osrd-project/ui-core';
import { Download, File } from '@osrd-project/ui-icons';
import { pdf } from '@react-pdf/renderer';
import { useTranslation } from 'react-i18next';

import type {
  OperationalPointWithTimeAndSpeed,
  PathPropertiesFormatted,
  SimulationResponseSuccess,
} from 'applications/operationalStudies/types';
import type {
  PathfindingResultSuccess,
  RollingStockWithLiveries,
  TrainScheduleBase,
} from 'common/api/osrdEditoastApi';

import exportTrainCSV from './exportTrainCSV';
import SimulationReportSheetScenario from './SimulationReportSheetScenario';
import type { SimulationSheetData } from './types';

type SimulationResultExportProps = {
  path: PathfindingResultSuccess;
  scenarioData: { name: string; infraName: string };
  train: TrainScheduleBase;
  simulatedTrain: SimulationResponseSuccess;
  pathElectrifications: PathPropertiesFormatted['electrifications'];
  operationalPoints: OperationalPointWithTimeAndSpeed[];
  rollingStock: RollingStockWithLiveries;
  mapCanvas?: string;
};

const SimulationResultExport = ({
  path,
  scenarioData,
  train,
  simulatedTrain,
  pathElectrifications,
  operationalPoints,
  rollingStock,
  mapCanvas,
}: SimulationResultExportProps) => {
  const { t } = useTranslation('simulation');

  const simulationSheetData: SimulationSheetData = useMemo(
    () => ({
      trainName: train.train_name,
      departure_time: '',
      simulation: simulatedTrain,
      creationDate: new Date(),
      rollingStock,
      speedLimitByTag: train.speed_limit_tag,
    }),
    [simulatedTrain]
  );

  const exportTrainPDF = useCallback(async () => {
    const doc = (
      <SimulationReportSheetScenario
        path={path}
        scenarioData={scenarioData}
        trainData={simulationSheetData}
        operationalPointsList={operationalPoints}
        mapCanvas={mapCanvas}
      />
    );
    const blob = await pdf(doc).toBlob();
    const url = URL.createObjectURL(blob);
    window.open(url, '_blank');
    URL.revokeObjectURL(url);
  }, [path, scenarioData, simulationSheetData, operationalPoints, mapCanvas]);

  return (
    <div className="simulation-sheet-container">
      {/* Export simulation PDF */}
      <Button
        onClick={exportTrainPDF}
        variant="Quiet"
        label={t('simulationSheet')}
        size="medium"
        leadingIcon={<File />}
      />

      {/* Export simulation CSV */}
      <Button
        onClick={() =>
          exportTrainCSV(simulatedTrain, operationalPoints, pathElectrifications, train)
        }
        variant="Quiet"
        label=".csv"
        size="medium"
        leadingIcon={<Download />}
      />
    </div>
  );
};

export default SimulationResultExport;