'use client';

import type { CSSProperties } from 'react';
import { Cell, Pie, PieChart, ResponsiveContainer, Tooltip } from 'recharts';
import type { TodayChartBucket } from '@/types/today';
import { TodayChartFrame } from '@/components/today/TodayChartFrame';
import {
  TODAY_CHART_COLORS,
  TODAY_CHART_TOOLTIP_STYLE,
} from '@/components/today/chart-theme';

interface TodayDonutChartProps {
  data: TodayChartBucket[];
  labelForCode: (code: string) => string;
  colorForCode?: (code: string, index: number) => string;
  swatchStyleForCode?: (code: string, index: number) => CSSProperties;
  variant?: 'donut' | 'pie';
  sideLegend?: boolean;
}

export function TodayDonutChart({
  data,
  labelForCode,
  colorForCode,
  swatchStyleForCode,
  variant = 'donut',
  sideLegend = false,
}: TodayDonutChartProps) {
  const chartData = data.map((item) => ({
    ...item,
    displayLabel: labelForCode(item.code),
  }));

  const resolveColor = (code: string, index: number) =>
    colorForCode?.(code, index) ??
    TODAY_CHART_COLORS[index % TODAY_CHART_COLORS.length];

  const resolveSwatchStyle = (code: string, index: number): CSSProperties =>
    swatchStyleForCode?.(code, index) ?? {
      backgroundColor: resolveColor(code, index),
      borderColor: 'rgba(0, 0, 0, 0.18)',
    };

  const innerRadius = variant === 'pie' ? 0 : 62;
  const outerRadius = sideLegend ? 100 : 92;

  const chart = (
    <TodayChartFrame>
      <ResponsiveContainer width="100%" height="100%">
        <PieChart margin={{ top: 0, right: 0, bottom: 0, left: 0 }}>
        <Pie
          data={chartData}
          dataKey="count"
          nameKey="displayLabel"
          cx="50%"
          cy="50%"
          innerRadius={innerRadius}
          outerRadius={outerRadius}
          paddingAngle={variant === 'pie' ? 1 : 2}
          stroke="transparent"
        >
          {chartData.map((entry, index) => (
            <Cell key={entry.code} fill={resolveColor(entry.code, index)} />
          ))}
        </Pie>
        <Tooltip
          contentStyle={TODAY_CHART_TOOLTIP_STYLE}
          formatter={(value, _name, item) => {
            const row = item?.payload as TodayChartBucket | undefined;
            return [value, row ? labelForCode(row.code) : ''];
          }}
        />
      </PieChart>
      </ResponsiveContainer>
    </TodayChartFrame>
  );

  if (!sideLegend) {
    return chart;
  }

  const rowClass = 'flex h-4 items-center text-xs leading-none';
  const legendInset = 'px-12';

  return (
    <div className={`flex h-full min-h-0 items-center overflow-hidden ${legendInset}`}>
      <div className="flex min-h-0 min-w-0 flex-1 items-center overflow-hidden py-0.5">
        <div className="flex max-h-full min-w-0 flex-col items-start gap-1.5 overflow-y-auto">
          {chartData.map((entry, index) => (
            <span key={entry.code} className={rowClass}>
              <span
                className="inline-block h-3 w-3 rounded-sm border"
                style={resolveSwatchStyle(entry.code, index)}
                aria-hidden
              />
            </span>
          ))}
        </div>

        <div className="ml-2 flex min-w-0 flex-col items-start gap-1.5 overflow-hidden">
          {chartData.map((entry) => (
            <span
              key={entry.code}
              className={`${rowClass} max-w-full truncate text-left text-text-primary`}
            >
              {entry.displayLabel}
            </span>
          ))}
        </div>

        <div className="ml-3 flex shrink-0 flex-col items-end gap-1.5">
          {chartData.map((entry) => (
            <span
              key={entry.code}
              className={`${rowClass} tabular-nums text-right text-text-muted`}
            >
              {entry.count}
            </span>
          ))}
        </div>
      </div>

      <div className="ml-4 flex h-full min-h-0 w-[min(100%,220px)] max-w-[48%] shrink-0 items-center justify-center">
        <ResponsiveContainer width="100%" height="100%">
          <PieChart margin={{ top: 0, right: 0, bottom: 0, left: 0 }}>
            <Pie
              data={chartData}
              dataKey="count"
              nameKey="displayLabel"
              cx="50%"
              cy="50%"
              innerRadius={innerRadius}
              outerRadius={outerRadius}
              paddingAngle={variant === 'pie' ? 1 : 2}
              stroke="transparent"
            >
              {chartData.map((entry, index) => (
                <Cell key={entry.code} fill={resolveColor(entry.code, index)} />
              ))}
            </Pie>
            <Tooltip
              contentStyle={TODAY_CHART_TOOLTIP_STYLE}
              formatter={(value, _name, item) => {
                const row = item?.payload as TodayChartBucket | undefined;
                return [value, row ? labelForCode(row.code) : ''];
              }}
            />
          </PieChart>
        </ResponsiveContainer>
      </div>
    </div>
  );
}