import { cn } from "../lib/cn";
import { SESSION_STATUS_PRESENTATION, type SessionDisplayStatus } from "../session-status-model";
import { useOpenGeniText, type OpenGeniText } from "../messages/runtime";

/** A server status, or the display-only `blocked` (see `sessionDisplayStatus`). */
type SessionStatusValue = SessionDisplayStatus;

export type SessionStatusMeta = {
  label: string;
  /** Token-backed color classes for the dot and tinted badge. */
  dotClassName: string;
  badgeClassName: string;
  /** Live states breathe; terminal states hold still. */
  pulse: boolean;
};

export const SESSION_STATUS_META = (
  t: OpenGeniText,
): Record<SessionStatusValue, SessionStatusMeta> => ({
  queued: {
    label: SESSION_STATUS_PRESENTATION(t).queued.label,
    dotClassName: "bg-og-status-queued",
    badgeClassName: "text-og-fg-muted border-og-border bg-og-status-queued/10",
    pulse: SESSION_STATUS_PRESENTATION(t).queued.pulse,
  },
  running: {
    label: SESSION_STATUS_PRESENTATION(t).running.label,
    dotClassName: "bg-og-status-running",
    badgeClassName: "text-og-status-running border-og-status-running/30 bg-og-status-running/10",
    pulse: SESSION_STATUS_PRESENTATION(t).running.pulse,
  },
  recovering: {
    label: SESSION_STATUS_PRESENTATION(t).recovering.label,
    dotClassName: "bg-og-status-running",
    badgeClassName: "text-og-status-running border-og-status-running/30 bg-og-status-running/10",
    pulse: SESSION_STATUS_PRESENTATION(t).recovering.pulse,
  },
  waiting_capacity: {
    label: SESSION_STATUS_PRESENTATION(t).waiting_capacity.label,
    dotClassName: "bg-og-status-waiting",
    badgeClassName: "text-og-status-waiting border-og-status-waiting/35 bg-og-status-waiting/10",
    pulse: SESSION_STATUS_PRESENTATION(t).waiting_capacity.pulse,
  },
  idle: {
    label: SESSION_STATUS_PRESENTATION(t).idle.label,
    dotClassName: "bg-og-status-idle",
    badgeClassName: "text-og-status-idle border-og-status-idle/30 bg-og-status-idle/10",
    pulse: SESSION_STATUS_PRESENTATION(t).idle.pulse,
  },
  requires_action: {
    label: SESSION_STATUS_PRESENTATION(t).requires_action.label,
    dotClassName: "bg-og-status-waiting",
    badgeClassName: "text-og-status-waiting border-og-status-waiting/35 bg-og-status-waiting/10",
    pulse: SESSION_STATUS_PRESENTATION(t).requires_action.pulse,
  },
  blocked: {
    label: SESSION_STATUS_PRESENTATION(t).blocked.label,
    dotClassName: "bg-og-status-failed",
    badgeClassName: "text-og-status-failed border-og-status-failed/35 bg-og-status-failed/10",
    pulse: SESSION_STATUS_PRESENTATION(t).blocked.pulse,
  },
  failed: {
    label: SESSION_STATUS_PRESENTATION(t).failed.label,
    dotClassName: "bg-og-status-failed",
    badgeClassName: "text-og-status-failed border-og-status-failed/35 bg-og-status-failed/10",
    pulse: SESSION_STATUS_PRESENTATION(t).failed.pulse,
  },
  cancelled: {
    label: SESSION_STATUS_PRESENTATION(t).cancelled.label,
    dotClassName: "bg-og-status-cancelled",
    badgeClassName: "text-og-fg-subtle border-og-border bg-og-status-cancelled/10",
    pulse: SESSION_STATUS_PRESENTATION(t).cancelled.pulse,
  },
});

export type SessionStatusProps = {
  status: SessionStatusValue;
  /** Override the label ("Running" -> "Deploying", ...). */
  label?: string | undefined;
  size?: "sm" | "md" | undefined;
  className?: string | undefined;
};

/** Status badge with a breathing dot for live states. */
export function SessionStatus({ status, label, size = "md", className }: SessionStatusProps) {
  const t = useOpenGeniText();
  const meta = SESSION_STATUS_META(t)[status];
  return (
    <span
      data-status={status}
      className={cn(
        "og-root inline-flex shrink-0 items-center rounded-full border font-medium",
        size === "sm" ? "gap-1 px-1.5 py-px text-og-xs" : "gap-1.5 px-2 py-0.5 text-og-control",
        meta.badgeClassName,
        className,
      )}
    >
      <StatusDot status={status} className={size === "sm" ? "size-1" : "size-1.5"} />
      {label ?? meta.label}
    </span>
  );
}

export type StatusDotProps = {
  status: SessionStatusValue;
  className?: string | undefined;
};

/** Just the dot — for dense rows and tiles. */
export function StatusDot({ status, className }: StatusDotProps) {
  const t = useOpenGeniText();
  const meta = SESSION_STATUS_META(t)[status];
  return (
    <span
      className={cn(
        "relative inline-flex size-1.5 shrink-0 rounded-full",
        meta.dotClassName,
        className,
      )}
    >
      {meta.pulse ? (
        <span className={cn("absolute inset-0 animate-og-pulse rounded-full", meta.dotClassName)} />
      ) : null}
    </span>
  );
}