train

apps/edr-freight-api/src/modules/trains/train-builder.controller.ts

@@ -7,6 +7,7 @@ import {
   HttpStatus,
   Param,
   ParseUUIDPipe,
+  Patch,
   Post,
   Put,
   Query,
@@ -19,6 +20,7 @@ import { BuildTrainDto } from './dto/build-train.dto';
 import { ListBuiltTrainsQueryDto } from './dto/list-built-trains-query.dto';
 import { ReorderTrainWagonsDto } from './dto/reorder-train-wagons.dto';
 import { UpdateTrainLocomotivesDto } from './dto/update-train-locomotives.dto';
+import { UpdateTrainYardDto } from './dto/update-train-yard.dto';
 import { TrainBuilderService } from './train-builder.service';
 
 @ApiTags('train-builder')
@@ -57,6 +59,15 @@ export class TrainBuilderController {
     return this.trainBuilderService.setLocomotives(id, dto);
   }
 
+  @Patch(':id/yard')
+  @FleetManage()
+  @ApiOperation({
+    summary: 'Relocate the train — its locomotives and wagons move to the new yard with it',
+  })
+  setYard(@Param('id', ParseUUIDPipe) id: string, @Body() dto: UpdateTrainYardDto) {
+    return this.trainBuilderService.setYard(id, dto.currentYardId);
+  }
+
   @Post(':id/wagons')
   @FleetManage()
   @ApiOperation({ summary: "Append AVAILABLE wagons from the train's yard to the consist" })