booking operations and trains scheduling also allocations

apps/edr-freight-web/backoffice/src/utils/wagonAvailability.ts

@@ -0,0 +1,33 @@
+import { Freight } from "@edr/types";
+
+import type { Wagon } from "@/services/wagon.service";
+
+export function wagonMatchesScheduleDirection(
+  wagon: Pick<Wagon, "status" | "readiness">,
+  scheduleDirection?: string | null,
+  options?: { allowPinned?: boolean },
+): boolean {
+  if (options?.allowPinned) return true;
+  if (wagon.status !== Freight.WagonStatus.Available) return false;
+  if (!scheduleDirection || scheduleDirection === "DOMESTIC") return true;
+  if (scheduleDirection === "IMPORT") {
+    return wagon.readiness === Freight.WagonReadiness.ImportReady;
+  }
+  if (scheduleDirection === "EXPORT") {
+    return wagon.readiness === Freight.WagonReadiness.ExportReady;
+  }
+  return true;
+}
+
+export function filterWagonsForSchedule(
+  wagons: Wagon[],
+  scheduleDirection?: string | null,
+  pinnedWagonIds?: Set<string>,
+): Wagon[] {
+  return wagons.filter((wagon) => {
+    const isPinned = pinnedWagonIds?.has(wagon.id) ?? false;
+    return wagonMatchesScheduleDirection(wagon, scheduleDirection, {
+      allowPinned: isPinned,
+    });
+  });
+}