import { useMemo } from "react";
import { ArrowDown, ArrowUp, ArrowUpDown } from "lucide-react";
import { Input } from "@/shared/ui/input";
import { Button } from "@/shared/ui/button";
import { Checkbox } from "@/shared/ui/checkbox";
import { cn } from "@/shared/ui/cn";
import { Table, TableBody, TableCaption, TableCell, TableHead, TableHeader, TableRow } from "@/shared/ui/table";
import { AnswerReveal } from "./answer-motion";
import { AskAboutContent, SelectForAnalysis } from "./answer-ui-interactions";

import { useAnswerNodeState } from "./answer-view-state-provider";
import type { AnswerTableState } from "./answer-view-state";

export function InteractiveDataTable({ title, columns, rows, source, state: controlledState, onStateChange }: {
  title: string; columns: string[]; rows: string[][]; source?: string;
  state?: Omit<AnswerTableState, "type">;
  onStateChange?: (patch: Partial<Omit<AnswerTableState, "type">>) => void;
}) {
  const [localState, patchLocal] = useAnswerNodeState<AnswerTableState>("Table", { title, columns, rows });
  const state = controlledState ?? localState;
  const patch = onStateChange ?? patchLocal;
  // The owning schema node migrates data changes. A complete restored snapshot
  // can change chart columns/range and restore a valid selection atomically.
  const { query, selection, sort } = state;
  const setSelection = (selection: number[]) => patch({ selection });
  const filtered = useMemo(() => {
    const items = rows.map((cells, id) => ({ cells, id }))
      .filter(({ cells }) => cells.join(" ").toLocaleLowerCase().includes(query.trim().toLocaleLowerCase()));
    if (sort) items.sort((a, b) => {
      const aValue = a.cells[sort.column], bValue = b.cells[sort.column];
      const compared = /^[-+]?\d+(\.\d+)?%?$/.test(aValue) && /^[-+]?\d+(\.\d+)?%?$/.test(bValue)
        ? Number.parseFloat(aValue) - Number.parseFloat(bValue)
        : aValue.localeCompare(bValue, "zh-CN", { numeric: true });
      return sort.descending ? -compared : compared;
    });
    return items;
  }, [rows, sort, query]);
  const selected = filtered.filter(({ id }) => selection.includes(id));
  const quote = selected.map(({ cells }) => cells.map((cell, index) => `${columns[index]}：${cell}`).join("；")).join("\n");
  return <div className="min-w-0 space-y-2" data-interactive-table>
    <div className="flex flex-wrap items-center gap-2">
      <Input aria-label={`搜索${title}`} placeholder="搜索当前数据" value={query} className="h-9 min-w-0 flex-1 text-xs"
        onChange={(event) => { patch({ query: event.currentTarget.value, selection: [] }); }} />
      <span className="text-[11px] tabular-nums text-muted-foreground">{filtered.length} / {rows.length} 行</span>
    </div>
    <div role="region" aria-label={title} tabIndex={0}
      className="answer-ui-table overflow-x-auto rounded-lg border border-border focus-visible:outline-2 focus-visible:outline-offset-2 focus-visible:outline-ring">
      <Table>
        <TableCaption className="caption-top font-medium">{title}</TableCaption>
        <TableHeader className="bg-muted/50"><TableRow>
          <TableHead scope="col" className="w-10"><Checkbox aria-label={`选择${title}的全部筛选结果`}
            checked={filtered.length > 0 && selected.length === Math.min(filtered.length, 100)} disabled={!filtered.length}
            onChange={(event) => setSelection(event.target.checked ? filtered.slice(0, 100).map(({ id }) => id) : [])} /></TableHead>
          {columns.map((column, index) => <TableHead key={index} scope="col" className="whitespace-nowrap"
            aria-sort={sort?.column === index ? sort.descending ? "descending" : "ascending" : "none"}>
            <Button size="sm" variant="ghost" className="h-9 gap-1 px-1 text-xs" aria-label={`按${column}排序`}
              onClick={() => patch({ sort: sort?.column === index
                ? sort.descending ? null : { column: index, descending: true }
                : { column: index, descending: false } })}>
              {column}{sort?.column === index ? sort.descending ? <ArrowDown size={12} /> : <ArrowUp size={12} /> : <ArrowUpDown size={12} />}
            </Button>
          </TableHead>)}
        </TableRow></TableHeader>
        <TableBody>
          {filtered.map(({ cells, id }) => <TableRow key={id} data-state={selection.includes(id) ? "selected" : undefined}>
            <TableCell className="w-10"><Checkbox aria-label={`选择第 ${id + 1} 行`} checked={selection.includes(id)} disabled={!selection.includes(id) && selection.length >= 100}
              onChange={(event) => setSelection(event.target.checked ? [...selection, id] : selection.filter((value) => value !== id))} /></TableCell>
            {cells.map((cell, index) => <TableCell key={index}
              className={cn("min-w-24", /^[-+]?\d+(\.\d+)?%?$/.test(cell) && "text-right tabular-nums")}>{cell}</TableCell>)}
          </TableRow>)}
          {!filtered.length ? <TableRow><TableCell colSpan={columns.length + 1} className="py-6 text-center text-xs text-muted-foreground">
            {rows.length ? "当前筛选没有匹配的数据。" : "没有已加载的数据。"}
          </TableCell></TableRow> : null}
        </TableBody>
      </Table>
    </div>
    <AnswerReveal open={selected.length > 0} className="flex flex-wrap items-center justify-between gap-1">
      <span className="text-xs text-muted-foreground">已选 {selected.length} 行{filtered.length > 100 ? " · 每次最多选择前 100 行" : ""}</span>
      <div className="flex flex-wrap gap-1">
        <Button type="button" variant="ghost" size="sm" className="h-9 text-xs" onClick={() => setSelection([])}>取消选行</Button>
        <SelectForAnalysis label="将所选行加入分析" title={title} text={`${quote}${source ? `\n来源：${source}` : ""}`} />
        <AskAboutContent title={title} text={`${quote}${source ? `\n来源：${source}` : ""}`} />
      </div>
    </AnswerReveal>
    {!selected.length && filtered.length ? <p className="text-[11px] text-muted-foreground">勾选数据行可加入分析或继续提问。</p> : null}
  </div>;
}