import { BenchmarkResult } from "../../../types";
import { formatNumber, formatSize } from "../utils/formatters";
import { columns } from "../table/columns";

export const exportToCsv = (
  data: BenchmarkResult[],
  selectedDataset: string,
) => {
  // Convert data to CSV
  const headers = columns.map((col) => col.header).join(",");
  const rows = data
    .map((row) =>
      columns
        .map((col) => {
          const value = row[col.accessorKey as keyof BenchmarkResult];
          // Format numbers according to their display format
          if (col.accessorKey === "compression_ratio") {
            return `${formatNumber(value as number)}x`;
          } else if (
            col.accessorKey === "encode_time" ||
            col.accessorKey === "decode_time"
          ) {
            return formatNumber(value as number, 4);
          } else if (
            col.accessorKey === "original_size" ||
            col.accessorKey === "compressed_size"
          ) {
            return formatSize(value as number);
          } else if (typeof value === "number") {
            return formatNumber(value);
          }
          return value;
        })
        .join(","),
    )
    .join("\n");
  const csv = `${headers}\n${rows}`;

  // Create and trigger download
  const blob = new Blob([csv], { type: "text/csv" });
  const url = window.URL.createObjectURL(blob);
  const a = document.createElement("a");
  a.href = url;
  a.download = `benchmark-results${selectedDataset ? `-${selectedDataset}` : ""}.csv`;
  document.body.appendChild(a);
  a.click();
  document.body.removeChild(a);
  window.URL.revokeObjectURL(url);
};