editoast: add train_schedule object to search endpoint

Signed-off-by: Ethan Perruzza <[email protected]>

Author: EthanPERRUZZA

editoast/openapi.yaml

@@ -8750,6 +8750,7 @@ components:
       - $ref: '#/components/schemas/SearchResultItemProject'
       - $ref: '#/components/schemas/SearchResultItemStudy'
       - $ref: '#/components/schemas/SearchResultItemScenario'
+      - $ref: '#/components/schemas/SearchResultItemTrainSchedule'
       description: A search result item that depends on the query's `object`
     SearchResultItemOperationalPoint:
       type: object
@@ -8983,6 +8984,77 @@ components:
           format: int64
         line_name:
           type: string
+    SearchResultItemTrainSchedule:
+      type: object
+      description: A search result item for a query with `object = "trainschedule"`
+      required:
+      - id
+      - train_name
+      - labels
+      - rolling_stock_name
+      - timetable_id
+      - start_time
+      - schedule
+      - margins
+      - initial_speed
+      - comfort
+      - path
+      - constraint_distribution
+      - speed_limit_tag
+      - power_restrictions
+      - options
+      properties:
+        comfort:
+          type: integer
+          format: int32
+        constraint_distribution:
+          type: integer
+          format: int32
+        id:
+          type: integer
+          format: int64
+          minimum: 0
+        initial_speed:
+          type: number
+          format: double
+        labels:
+          type: array
+          items:
+            type: string
+        margins:
+          type: array
+          items:
+            type: string
+        options:
+          type: array
+          items:
+            type: string
+        path:
+          type: array
+          items:
+            type: string
+        power_restrictions:
+          type: array
+          items:
+            type: string
+        rolling_stock_name:
+          type: string
+        schedule:
+          type: array
+          items:
+            type: string
+        speed_limit_tag:
+          type: array
+          items:
+            type: string
+        start_time:
+          type: string
+        timetable_id:
+          type: integer
+          format: int64
+          minimum: 0
+        train_name:
+          type: string
     Side:
       type: string
       enum:

editoast/src/views/search.rs

@@ -344,7 +344,9 @@ async fn search(
     Json(SearchPayload { object, query, dry }): Json<SearchPayload>,
 ) -> Result<Json<serde_json::Value>> {
     let roles: HashSet<BuiltinRole> = match object.as_str() {
-        "track" | "operationalpoint" | "signal" => HashSet::from([BuiltinRole::InfraRead]),
+        "track" | "operationalpoint" | "signal" | "trainschedule" => {
+            HashSet::from([BuiltinRole::InfraRead])
+        }
         "project" | "study" | "scenario" => HashSet::from([BuiltinRole::OpsRead]),
         _ => {
             return Err(SearchApiError::ObjectType {
@@ -683,6 +685,47 @@ pub(super) struct SearchResultItemScenario {
     tags: Vec<String>,
 }
 
+#[derive(Search, Serialize, ToSchema)]
+#[search(
+    table = "train_schedule",
+    column(name = "timetable_id", data_type = "integer"),
+    column(name = "train_name", data_type = "string")
+)]
+#[allow(unused)]
+/// A search result item for a query with `object = "trainschedule"`
+pub(super) struct SearchResultItemTrainSchedule {
+    #[search(sql = "train_schedule.id")]
+    id: u64,
+    #[search(sql = "train_schedule.train_name")]
+    train_name: String, // useless puisqu'on en a besoin pour la request?
+    #[search(sql = "train_schedule.labels")]
+    labels: Vec<String>,
+    #[search(sql = "train_schedule.rolling_stock_name")]
+    rolling_stock_name: String,
+    #[search(sql = "train_schedule.timetable_id")]
+    timetable_id: u64, // useless puisqu'on en a besoin pour la request?
+    #[search(sql = "train_schedule.start_time")]
+    start_time: String,
+    #[search(sql = "train_schedule.schedule")]
+    schedule: Vec<String>,
+    #[search(sql = "train_schedule.margins")]
+    margins: Vec<String>,
+    #[search(sql = "train_schedule.initial_speed")]
+    initial_speed: f64,
+    #[search(sql = "train_schedule.comfort")]
+    comfort: i16,
+    #[search(sql = "train_schedule.path")]
+    path: Vec<String>,
+    #[search(sql = "train_schedule.constraint_distribution")]
+    constraint_distribution: i16,
+    #[search(sql = "train_schedule.speed_limit_tag")]
+    speed_limit_tag: Vec<String>,
+    #[search(sql = "train_schedule.power_restrictions")]
+    power_restrictions: Vec<String>,
+    #[search(sql = "train_schedule.options")]
+    options: Vec<String>,
+}
+
 /// See [editoast_search::SearchConfigStore::find]
 #[derive(SearchConfigStore)]
 #[search_config_store(
@@ -692,5 +735,6 @@ pub(super) struct SearchResultItemScenario {
     object(name = "project", config = SearchResultItemProject),
     object(name = "study", config = SearchResultItemStudy),
     object(name = "scenario", config = SearchResultItemScenario),
+    object(name = "trainschedule", config = SearchResultItemTrainSchedule),
 )]
 pub struct SearchConfigFinder;