editoast: update temporary speed limit import to use signals

Previously, the temporary speed limits creation endpoint expected to
retrieve directional track ranges in the request body. Since it might
not be possible to get that information directly, update the
endpoint so that it retrieves the list of execution and resume signals
associated with the temporary speed limits and then do a graph traversal
on the infrastructure track sections to determine from the signals on
which the track ranges the temporary speed limits apply.

Add an infrastructure id parameter to the request form.

Signed-off-by: Loup Federico <[email protected]>

editoast/openapi.yaml

@@ -2336,8 +2336,12 @@ paths:
               type: object
               required:
               - speed_limit_group_name
+              - infra_id
               - speed_limits
               properties:
+                infra_id:
+                  type: integer
+                  format: int64
                 speed_limit_group_name:
                   type: string
                 speed_limits:
@@ -2347,7 +2351,7 @@ paths:
                     required:
                     - start_date_time
                     - end_date_time
-                    - track_ranges
+                    - signals
                     - speed_limit
                     - obj_id
                     properties:
@@ -2356,16 +2360,16 @@ paths:
                         format: date-time
                       obj_id:
                         type: string
+                      signals:
+                        type: array
+                        items:
+                          $ref: '#/components/schemas/Sign'
                       speed_limit:
                         type: number
                         format: double
                       start_date_time:
                         type: string
                         format: date-time
-                      track_ranges:
-                        type: array
-                        items:
-                          $ref: '#/components/schemas/DirectionalTrackRange'
         required: true
       responses:
         '201':

editoast/src/infra_cache/graph.rs

@@ -82,6 +82,17 @@ impl<'a> Graph<'a> {
             .map(|groups| groups.keys().cloned().collect())
             .unwrap_or_default()
     }
+
+    /// Given an endpoint return all its neighbours indiscriminately
+    /// of their group.
+    pub fn get_all_neighbours(&'a self, track_endpoint: &TrackEndpoint) -> Vec<&'a TrackEndpoint> {
+        let groups = self.links.get(track_endpoint);
+        if let Some(groups) = groups {
+            groups.values().copied().collect::<Vec<_>>()
+        } else {
+            Vec::new()
+        }
+    }
 }
 
 #[cfg(test)]