import type { BuiltTrainStatus } from "@/services/trainBuilder.service";

/** Badge color per built-train lifecycle status (Mantine palette keys). */
export const trainStatusColor = (status: BuiltTrainStatus | string): string => {
  switch (status) {
    case "AVAILABLE":
      return "edr-green";
    case "SCHEDULED":
      return "blue";
    case "IN_SERVICE":
      return "teal";
    case "UNDER_MAINTENANCE":
      return "yellow";
    case "OUT_OF_SERVICE":
      return "red";
    default:
      return "gray";
  }
};

export const trainStatusLabel = (status: BuiltTrainStatus | string): string =>
  String(status)
    .toLowerCase()
    .replace(/_/g, " ")
    .replace(/^\w/, (c) => c.toUpperCase());