'use client';
import { Table2 } from 'lucide-react';
import { useId, useState, type ReactNode } from 'react';
import { t } from '@/i18n';
import { cn } from '@/lib/cn';
import type { Provenance } from '@/lib/types';
import type { ProvenancePayload } from '@/components/data/provenance-context';
import { SourceLine } from './source-line';

export interface TableColumn {
  key: string;
  label: string;
  numeric?: boolean;
}
export interface TableData {
  columns: TableColumn[];
  rows: Array<Record<string, string>>;
}

/**
 * Wraps any chart: optional heading, the chart, a source attribution line (clickable → provenance),
 * and the accessible data-table toggle. `summary` is the auto-generated text summary announced to AT.
 */
export function ChartFrame({
  title,
  subtitle,
  summary,
  provenance,
  payload,
  table,
  actions,
  legend,
  note,
  children,
  className,
  minHeight,
}: {
  title?: ReactNode;
  subtitle?: ReactNode;
  summary: string;
  provenance?: Provenance | null;
  payload?: ProvenancePayload | null;
  table?: TableData;
  actions?: ReactNode;
  legend?: ReactNode;
  note?: ReactNode;
  children: ReactNode;
  className?: string;
  /** Reserve height to avoid layout shift while the chart mounts. */
  minHeight?: number;
}) {
  const [showTable, setShowTable] = useState(false);
  const id = useId();
  return (
    <figure className={cn('min-w-0', className)} aria-describedby={`${id}-sum`}>
      {title || actions ? (
        <figcaption className="mb-2 flex items-start justify-between gap-3">
          <div className="min-w-0">
            {title ? <div className="text-sm font-semibold text-ink">{title}</div> : null}
            {subtitle ? <div className="text-xs text-ink-2">{subtitle}</div> : null}
          </div>
          {actions ? <div className="flex shrink-0 items-center gap-1">{actions}</div> : null}
        </figcaption>
      ) : null}
      {legend}
      <div style={minHeight ? { minHeight } : undefined}>{children}</div>
      <p id={`${id}-sum`} className="sr-only">
        {summary}
      </p>
      <div className="mt-1.5 flex flex-wrap items-center justify-between gap-x-3 gap-y-1">
        <div className="min-w-0 flex-1">
          <SourceLine provenance={provenance} payload={payload} />
          {note ? <p className="text-2xs text-ink-3">{note}</p> : null}
        </div>
        {table ? (
          <button
            type="button"
            onClick={() => setShowTable((s) => !s)}
            aria-expanded={showTable}
            aria-controls={`${id}-table`}
            className="inline-flex min-h-[32px] items-center gap-1 rounded-sm px-1.5 text-2xs text-ink-3 hover:bg-surface-2 hover:text-ink"
          >
            <Table2 size={12} aria-hidden />
            {showTable ? t('common.hideTable') : t('common.viewTable')}
          </button>
        ) : null}
      </div>
      {table && showTable ? (
        <div id={`${id}-table`} className="mt-2 max-h-72 overflow-auto border-y border-rule">
          <table className="w-full text-xs tnum">
            <caption className="sr-only">{t('chart.table.caption')}</caption>
            <thead className="sticky top-0 bg-paper">
              <tr>
                {table.columns.map((c) => (
                  <th key={c.key} scope="col" className={cn('px-2 py-1.5 text-left font-medium text-ink-2', c.numeric && 'text-right')}>
                    {c.label}
                  </th>
                ))}
              </tr>
            </thead>
            <tbody className="divide-y divide-rule">
              {table.rows.map((r, i) => (
                <tr key={i}>
                  {table.columns.map((c) => (
                    <td key={c.key} className={cn('px-2 py-1', c.numeric && 'text-right')}>
                      {r[c.key] ?? t('common.na')}
                    </td>
                  ))}
                </tr>
              ))}
            </tbody>
          </table>
        </div>
      ) : null}
    </figure>
  );
}

/** Legend for ≥ 2 series (a single series needs none — the title names it). */
export function Legend({ items }: { items: Array<{ label: string; color: string; dashed?: boolean; shape?: 'line' | 'rect' }> }) {
  if (items.length < 2) return null;
  return (
    <ul className="mb-2 flex flex-wrap gap-x-4 gap-y-1 text-xs text-ink-2" aria-label={t('common.legend')}>
      {items.map((it) => (
        <li key={it.label} className="inline-flex items-center gap-1.5">
          {it.shape === 'rect' ? (
            <span aria-hidden className="inline-block h-2.5 w-2.5 rounded-xs" style={{ background: it.color }} />
          ) : (
            <span aria-hidden className="inline-block h-0.5 w-4 rounded-full" style={{ background: it.color, ...(it.dashed ? { backgroundImage: `repeating-linear-gradient(90deg, ${it.color} 0 3px, transparent 3px 6px)`, backgroundColor: 'transparent' } : {}) }} />
          )}
          <span>{it.label}</span>
        </li>
      ))}
    </ul>
  );
}