yaak-mountain-loop/apps/yaak-proxy/components/FlowTimeline.tsx

import type { Flow } from "@yaakapp-internal/proxy-lib";
import {
  EventViewer,
  EventViewerRow,
  formatMillis,
  Icon,
  type IconProps,
} from "@yaakapp-internal/ui";
import { useMemo } from "react";
import { type FlowPhase, flowPhases, type PhaseKind } from "../lib/phases";

const ICONS: Record<PhaseKind, { icon: IconProps["icon"]; color: IconProps["color"] }> = {
  start: { icon: "send_horizontal", color: "secondary" },
  step: { icon: "arrow_right", color: "secondary" },
  end: { icon: "check", color: "success" },
  error: { icon: "alert_triangle", color: "danger" },
  cancelled: { icon: "circle_off", color: "secondary" },
};

/** Yaak's event list with the flow's phases as rows, each saying how long it took */
export function FlowTimeline({ flow }: { flow: Flow }) {
  const phases = useMemo(() => flowPhases(flow), [flow]);
  return (
    <EventViewer<FlowPhase>
      events={phases}
      getEventKey={(p) => p.key}
      splitLayoutStorageKey="proxy_flow_timeline"
      enableKeyboardNav={false}
      emptyMessage="No timings recorded"
      renderRow={({ event, isActive, onClick }) => {
        const { icon, color } = ICONS[event.kind];
        return (
          <EventViewerRow
            isActive={isActive}
            onClick={onClick}
            icon={<Icon icon={icon} color={color} size="sm" />}
            content={
              <span title={event.detail}>
                {event.label}
                {event.detail != null && (
                  <span className="text-text-subtlest">{` ${event.detail}`}</span>
                )}
              </span>
            }
            columns="auto minmax(0,1fr) 6rem 6rem"
            cells={{
              trailing: [
                <span key="took" className="tabular-nums">
                  {event.kind === "start" ? "" : `+${formatMillis(event.tookMs)}`}
                </span>,
                <span key="at" className="tabular-nums opacity-50">
                  {formatMillis(event.atMs)}
                </span>,
              ],
            }}
          />
        );
      }}
    />
  );
}