import type { FreightType } from "@/types/trainScheduling";

const isContainerFreight = (freightType?: string | null) => freightType === "CONTAINER";

/** Show container number placement step when train includes container cargo. */
export function shouldShowContainerPlacementStep(params: {
  containerUnitCount: number;
  scheduleFreightType?: FreightType | string | null;
  bookingFreightTypes: Array<FreightType | string | null | undefined>;
}): boolean {
  if (params.containerUnitCount > 0) return true;
  if (isContainerFreight(params.scheduleFreightType)) return true;
  return params.bookingFreightTypes.some(isContainerFreight);
}
