front: convert times from NGE to OSRD

Closes: #8389

Author: emersion

front/src/applications/operationalStudies/components/MacroEditor/ngeToOsrd.ts

@@ -1,7 +1,16 @@
 import { osrdEditoastApi, type TrainScheduleBase } from 'common/api/osrdEditoastApi';
 import type { AppDispatch } from 'store';
+import { formatToIsoDate } from 'utils/date';
+import { calculateTimeDifferenceInSeconds, formatDurationAsISO8601 } from 'utils/timeManipulation';
 
-import type { NetzgrafikDto, NGETrainrunEvent, TrainrunSection, Node, Trainrun } from './types';
+import type {
+  NetzgrafikDto,
+  NGETrainrunEvent,
+  TrainrunSection,
+  Node,
+  TimeLock,
+  Trainrun,
+} from './types';
 
 const createdTrainrun = new Map<number, number>();
 
@@ -11,14 +20,9 @@ const getTrainrunSectionsByTrainrunId = (trainrunSections: TrainrunSection[], tr
 const getNodeById = (nodes: Node[], nodeId: number | string) =>
   nodes.find((node) => node.id === nodeId);
 
-// TODO: add a dynamic values when available
-const DEFAULT_PAYLOAD: Pick<
-  TrainScheduleBase,
-  'constraint_distribution' | 'rolling_stock_name' | 'start_time'
-> = {
+const DEFAULT_PAYLOAD: Pick<TrainScheduleBase, 'constraint_distribution' | 'rolling_stock_name'> = {
   constraint_distribution: 'STANDARD',
   rolling_stock_name: '',
-  start_time: '2024-07-15T08:00:00+02:00',
 };
 
 const createPathItemFromNode = (node: Node, index: number) => {
@@ -30,10 +34,19 @@ const createPathItemFromNode = (node: Node, index: number) => {
   };
 };
 
+const getTimeLockDate = (timeLock: TimeLock, baseDate: Date): Date | null => {
+  if (timeLock.time === null) return null;
+  return new Date(baseDate.getTime() + timeLock.consecutiveTime! * 60 * 1000);
+};
+
+const formatDateDifference = (start: Date, stop: Date) =>
+  formatDurationAsISO8601(calculateTimeDifferenceInSeconds(start, stop));
+
 const createTrainSchedulePayload = (
   trainrunSections: TrainrunSection[],
   nodes: Node[],
-  trainrun: Trainrun
+  trainrun: Trainrun,
+  startDate: Date
 ) => {
   // TODO: check that the trainrunSections format is still compatible
   const path = trainrunSections.flatMap((section, index) => {
@@ -42,15 +55,43 @@ const createTrainSchedulePayload = (
     if (!sourceNode || !targetNode) return [];
     const originPathItem = createPathItemFromNode(sourceNode, index);
     if (index === trainrunSections.length - 1) {
-      const destinationPathItem = createPathItemFromNode(targetNode, index);
+      const destinationPathItem = createPathItemFromNode(targetNode, index + 1);
       return [originPathItem, destinationPathItem];
     }
     return [originPathItem];
   });
 
+  const baseDate = new Date(startDate);
+  baseDate.setMinutes(0, 0, 0);
+
+  // The departure time of the first section is guaranteed to be non-null
+  const startTime = getTimeLockDate(trainrunSections[0].sourceDeparture, baseDate)!;
+
+  const schedule = trainrunSections.flatMap((section, index) => {
+    const nextSection = trainrunSections[index + 1];
+
+    // TODO: extract isNonStopTransit from transitions
+    let arrival = getTimeLockDate(section.targetArrival, baseDate);
+    const departure = nextSection ? getTimeLockDate(nextSection.sourceDeparture, baseDate) : null;
+    if (!arrival && !departure) {
+      return [];
+    }
+
+    // If missing arrival time, default to a zero stop duration
+    arrival = arrival || departure!;
+
+    return {
+      at: `${section.targetNodeId}-${index + 1}`,
+      arrival: formatDateDifference(startTime, arrival),
+      stop_for: departure ? formatDateDifference(arrival, departure) : null,
+    };
+  });
+
   return {
-    path,
     train_name: trainrun.name,
+    start_time: formatToIsoDate(startTime),
+    path,
+    schedule,
   };
 };
 
@@ -74,13 +115,19 @@ const handleOperation = async ({
   );
   switch (type) {
     case 'create': {
+      const startDate = new Date();
       const newTrainSchedules = await dispatch(
         osrdEditoastApi.endpoints.postV2TimetableByIdTrainSchedule.initiate({
           id: timeTableId,
           body: [
             {
               ...DEFAULT_PAYLOAD,
-              ...createTrainSchedulePayload(trainrunSectionsByTrainrunId, nodes, trainrun),
+              ...createTrainSchedulePayload(
+                trainrunSectionsByTrainrunId,
+                nodes,
+                trainrun,
+                startDate
+              ),
             },
           ],
         })
@@ -105,14 +152,13 @@ const handleOperation = async ({
           id: trainrunIdToUpdate,
         })
       ).unwrap();
+      const startDate = new Date(trainSchedule.start_time);
       await dispatch(
         osrdEditoastApi.endpoints.putV2TrainScheduleById.initiate({
           id: trainrunIdToUpdate,
           trainScheduleForm: {
             ...trainSchedule,
-            ...createTrainSchedulePayload(trainrunSectionsByTrainrunId, nodes, trainrun),
-            // TODO: convert NGE times to OSRD schedule
-            schedule: [],
+            ...createTrainSchedulePayload(trainrunSectionsByTrainrunId, nodes, trainrun, startDate),
           },
         })
       ).unwrap();