shadcn-ui/apps/www/components/chart-toolbar.tsx

"use client"

import { cn } from "@/lib/utils"
import { ChartCodeViewer } from "@/components/chart-code-viewer"
import { Separator } from "@/registry/new-york/ui/separator"

import "@/styles/mdx.css"
import {
  AreaChart,
  BarChartBig,
  Hexagon,
  LineChart,
  MousePointer2,
  PieChart,
  Radar,
} from "lucide-react"

import { ChartCopyButton } from "@/components/chart-copy-button"
import { Chart } from "@/components/chart-display"

export function ChartToolbar({
  chart,
  className,
  children,
}: {
  chart: Chart
} & React.ComponentProps<"div">) {
  return (
    <div className={cn("flex items-center gap-2", className)}>
      <div className="flex items-center gap-1.5 pl-1 text-[13px] text-muted-foreground [&>svg]:h-[0.9rem] [&>svg]:w-[0.9rem]">
        <ChartTitle chart={chart} />
      </div>
      <div className="ml-auto flex items-center gap-2 [&>form]:flex">
        <ChartCopyButton
          event="copy_chart_code"
          name={chart.name}
          code={chart.files?.[0]?.content ?? ""}
          className="[&_svg]-h-3 h-6 w-6 rounded-[6px] bg-transparent text-foreground shadow-none hover:bg-muted dark:text-foreground [&_svg]:w-3"
        />
        <Separator orientation="vertical" className="mx-0 hidden h-4 md:flex" />
        <ChartCodeViewer chart={chart}>{children}</ChartCodeViewer>
      </div>
    </div>
  )
}

function ChartTitle({ chart }: { chart: Chart }) {
  const { categories } = chart

  if (!categories?.length) {
    return null
  }

  if (categories.includes("charts-line")) {
    return (
      <>
        <LineChart /> Chart
      </>
    )
  }

  if (categories.includes("charts-bar")) {
    return (
      <>
        <BarChartBig /> Chart
      </>
    )
  }

  if (categories.includes("charts-pie")) {
    return (
      <>
        <PieChart /> Chart
      </>
    )
  }

  if (categories.includes("charts-area")) {
    return (
      <>
        <AreaChart /> Chart
      </>
    )
  }

  if (categories.includes("charts-radar")) {
    return (
      <>
        <Hexagon /> Chart
      </>
    )
  }

  if (categories.includes("charts-radial")) {
    return (
      <>
        <Radar /> Chart
      </>
    )
  }

  if (categories.includes("charts-tooltip")) {
    return (
      <>
        <MousePointer2 />
        Tooltip
      </>
    )
  }

  return categories[0]
}