train

apps/edr-freight-api/src/modules/train-scheduling/dto/create-container-train-schedule.dto.ts

@@ -20,15 +20,26 @@ export class CreateContainerTrainScheduleDto {
   @IsDateString()
   scheduleDate!: string;
 
-  @ApiProperty({
+  @ApiPropertyOptional({
+    format: 'uuid',
+    description:
+      'Built train (Train Builder) to run this departure — its locomotive set is used. Provide either trainId or locomotiveIds.',
+  })
+  @IsOptional()
+  @IsUUID()
+  trainId?: string;
+
+  @ApiPropertyOptional({
     type: [String],
     format: 'uuid',
-    description: 'Locomotives pulling the train (minimum 2 — front and back)',
+    description:
+      'Hand-picked locomotives pulling the train (minimum 2 — front and back). Ignored when trainId is provided.',
   })
+  @IsOptional()
   @IsArray()
   @ArrayMinSize(2, { message: 'A train must be pulled by at least two locomotives' })
   @IsUUID('all', { each: true })
-  locomotiveIds!: string[];
+  locomotiveIds?: string[];
 
   @ApiPropertyOptional({ description: 'Maximum total booking weight allowed on this train' })
   @IsOptional()