changes

apps/edr-freight-web/backoffice/src/services/trainScheduling.service.ts

@@ -8,6 +8,8 @@ import type {
   BookableSchedule,
   BookingWindow,
   AssignBookingsPayload,
+  BookingLoadResult,
+  BookingUnloadResult,
   CompositionRemovalEntry,
   UnassignedBookingsResponse,
   CreateTrainSchedulePayload,
@@ -35,6 +37,7 @@ import type {
   UploadImportDjiboutiDocumentPayload,
   WagonAllocationAttemptResult,
   YardOption,
+  YardWorkResult,
 } from "@/types/trainScheduling";
 
 interface BookingReferenceDataResponse {
@@ -330,6 +333,35 @@ export const trainSchedulingService = {
     return unwrap(response.data);
   },
 
+  getYardWork: async (scheduleId: string): Promise<YardWorkResult> => {
+    const response = await client.get<YardWorkResult>(
+      URL_CONSTANTS.TRAIN_SCHEDULING.YARD_WORK(scheduleId),
+    );
+    return unwrap(response.data);
+  },
+
+  loadScheduleBooking: async (
+    scheduleId: string,
+    bookingId: string,
+  ): Promise<BookingLoadResult> => {
+    const response = await client.post<BookingLoadResult>(
+      URL_CONSTANTS.TRAIN_SCHEDULING.BOOKING_LOAD(scheduleId, bookingId),
+      {},
+    );
+    return unwrap(response.data);
+  },
+
+  unloadScheduleBooking: async (
+    scheduleId: string,
+    bookingId: string,
+  ): Promise<BookingUnloadResult> => {
+    const response = await client.post<BookingUnloadResult>(
+      URL_CONSTANTS.TRAIN_SCHEDULING.BOOKING_UNLOAD(scheduleId, bookingId),
+      {},
+    );
+    return unwrap(response.data);
+  },
+
   getIntercityCandidates: async (
     scheduleId: string,
   ): Promise<IntercityCandidatesResult> => {