import { Trace } from "@/types";

// Consolidated status color utility for traces
export const getTraceStatusColor = (status: Trace["status"]) => {
  switch (status) {
    case "completed":
      return "bg-green-500/10 text-green-700 border-green-200";
    case "processing":
      return "bg-blue-500/10 text-blue-700 border-blue-200";
    case "error":
      return "bg-red-500/10 text-red-700 border-red-200";
    default:
      return "bg-gray-500/10 text-gray-700 border-gray-200";
  }
};

// Consolidated status variant utility for knowledge graphs
export const getKnowledgeGraphStatusVariant = (
  status: string
): "default" | "secondary" | "destructive" | "outline" => {
  switch (status) {
    case "created":
      return "secondary";
    case "enriched":
    case "perturbed":
    case "analyzed":
    case "causal":
      return "default";
    case "error":
      return "destructive";
    default:
      return "outline";
  }
};

// Consolidated status message utility
export const getStatusMessage = (status: string) => {
  switch (status) {
    case "created":
      return "Created";
    case "enriched":
      return "Prompt Reconstructed";
    case "perturbed":
      return "Perturbation Tested";
    case "analyzed":
    case "causal":
      return "Processed"; // Removed "Causal Analyzed" since checkmarks already indicate completion
    case "error":
      return "Error";
    case "processing":
      return "Processing";
    case "completed":
      return "Completed";
    default:
      return "Unknown";
  }
};

// Pipeline status colors
export const getPipelineStatusColor = (status: string) => {
  switch (status) {
    case "completed":
      return "bg-green-500/10 text-green-700 border-green-200";
    case "running":
    case "processing":
      return "bg-blue-500/10 text-blue-700 border-blue-200";
    case "failed":
    case "error":
      return "bg-red-500/10 text-red-700 border-red-200";
    case "pending":
      return "bg-yellow-500/10 text-yellow-700 border-yellow-200";
    default:
      return "bg-gray-500/10 text-gray-700 border-gray-200";
  }
};