import { TaskStatus } from '@/lib/types';

const classMap: Record<TaskStatus, string> = {
  Completed: 'completed',
  'In Progress': 'progress',
  'Not Started': 'pending',
  'Needs Correction': 'correction',
  Locked: 'locked',
  Submitted: 'progress'
};

export function StatusBadge({ status }: { status: TaskStatus | string }) {
  const safeClass = classMap[status as TaskStatus] ?? 'pending';
  return <span className={`status ${safeClass}`}>{status}</span>;
}