import {
  ComparisonChart,
  LineageChart,
  LossCurveChart,
  RetentionMatrix,
} from "../charts/index.ts";
import type { KnowledgeChartSpec } from "./registry.ts";

export function KnowledgeChart({ spec }: { spec: KnowledgeChartSpec }) {
  switch (spec.kind) {
    case "comparison":
      return (
        <ComparisonChart
          meta={spec.meta}
          data={spec.data}
          baselineLabel={spec.baselineLabel}
          trainedLabel={spec.trainedLabel}
          domain={spec.domain}
          format={spec.format}
          digits={spec.digits}
        />
      );
    case "loss":
      return (
        <LossCurveChart
          meta={spec.meta}
          series={spec.series}
          xDomain={spec.xDomain}
          yDomain={spec.yDomain}
          yScale={spec.yScale}
          xFormat={spec.xFormat}
          yFormat={spec.yFormat}
          digits={spec.digits}
        />
      );
    case "lineage":
      return <LineageChart meta={spec.meta} nodes={spec.nodes} edges={spec.edges} />;
    case "matrix":
      return (
        <RetentionMatrix
          meta={spec.meta}
          rows={spec.rows}
          cols={spec.cols}
          cells={spec.cells}
          domain={spec.domain}
          format={spec.format}
          digits={spec.digits}
        />
      );
    default: {
      const exhaustive: never = spec;
      void exhaustive;
      return null;
    }
  }
}