"use client";

import { ResponsiveContainer, Treemap } from "recharts";

import {
  Card,
  CardContent,
  CardDescription,
  CardHeader,
  CardTitle,
} from "@/components/ui/card";
import { cn } from "@/lib/utils";

interface ChartCard22Props {
  title?: string;
  description?: string;
  className?: string;
}

const chartData = [
  {
    name: "الکترونیک",
    children: [
      { name: "گوشی", size: 4500, color: "var(--chart-1)" },
      { name: "لپ‌تاپ", size: 3200, color: "var(--chart-1)" },
      { name: "تبلت", size: 1800, color: "var(--chart-1)" },
    ],
  },
  {
    name: "پوشاک",
    children: [
      { name: "پیراهن", size: 2800, color: "var(--chart-2)" },
      { name: "شلوار", size: 2100, color: "var(--chart-2)" },
      { name: "کفش", size: 1600, color: "var(--chart-2)" },
    ],
  },
  {
    name: "خانه",
    children: [
      { name: "مبلمان", size: 2400, color: "var(--chart-3)" },
      { name: "دکور", size: 1200, color: "var(--chart-3)" },
    ],
  },
  {
    name: "ورزش",
    children: [
      { name: "تجهیزات", size: 1800, color: "var(--chart-4)" },
      { name: "پوشاک ورزشی", size: 900, color: "var(--chart-4)" },
    ],
  },
];

interface TreemapContentProps {
  x: number;
  y: number;
  width: number;
  height: number;
  name: string;
  color?: string;
  depth: number;
}

const CustomContent = ({
  x,
  y,
  width,
  height,
  name,
  color,
  depth,
}: TreemapContentProps) => {
  if (depth !== 2) return null;

  const showLabel = width > 50 && height > 30;

  return (
    <g>
      <rect
        x={x}
        y={y}
        width={width}
        height={height}
        fill={color || "var(--chart-1)"}
        stroke="var(--background)"
        strokeWidth={2}
        rx={4}
      />
      {showLabel && (
        <text
          x={x + width / 2}
          y={y + height / 2}
          textAnchor="middle"
          dominantBaseline="middle"
          className="fill-white text-xs font-medium"
          style={{ pointerEvents: "none" }}
        >
          {name}
        </text>
      )}
    </g>
  );
};

const ChartCard22 = ({
  title = "فروش به تفکیک دسته",
  description = "توزیع درآمد میان دسته‌های محصول",
  className,
}: ChartCard22Props) => {
  return (
    <Card
      dir="rtl"
      lang="fa"
      className={cn("w-full max-w-2xl text-start", className)}
    >
      <CardHeader>
        <CardTitle>{title}</CardTitle>
        <CardDescription>{description}</CardDescription>
      </CardHeader>
      <CardContent>
        <div className="h-64 w-full">
          <ResponsiveContainer width="100%" height="100%">
            <Treemap
              data={chartData}
              dataKey="size"
              aspectRatio={4 / 3}
              stroke="var(--background)"
              content={
                <CustomContent
                  x={0}
                  y={0}
                  width={0}
                  height={0}
                  name=""
                  depth={0}
                />
              }
            />
          </ResponsiveContainer>
        </div>
        <div className="mt-4 flex flex-wrap gap-4">
          {chartData.map((category, index) => (
            <div key={category.name} className="flex items-center gap-2">
              <div
                className="size-3 rounded-sm"
                style={{
                  backgroundColor: `var(--chart-${index + 1})`,
                }}
              />
              <span className="text-sm">{category.name}</span>
            </div>
          ))}
        </div>
      </CardContent>
    </Card>
  );
};

export { ChartCard22 };