"use client";
import { useCallback, useEffect, useRef, useState } from "react";
import { DndContext, PointerSensor, useSensor, useSensors, closestCenter, type DragEndEvent } from "@dnd-kit/core";
import { SortableContext, useSortable, verticalListSortingStrategy, arrayMove } from "@dnd-kit/sortable";
import { CSS } from "@dnd-kit/utilities";
import { Plus, GripVertical, Trash2 } from "lucide-react";
import { Popover, PopoverTrigger, PopoverContent } from "@/components/ui/primitives";
import { api } from "@/lib/client/api";
import { cn } from "@/lib/utils";
import type { Property, RowOut } from "@/lib/databases/types";
import { useDb } from "../hooks";
import { Cell } from "../cells/Cell";
import { TypeIcon } from "../meta";
import { PropertyEditor } from "../PropertyEditor";
import { groupRows } from "../group";
import { OPTION_CLASS } from "../colors";

const MIN_W = 80;
export function TableView() {
  const ctx = useDb(); const cols = ctx.visibleProps; const widths = ctx.view.config.widths ?? {};
  const [editing, setEditing] = useState<string | null>(null); const [sel, setSel] = useState<Set<string>>(new Set()); const [editProp, setEditProp] = useState<string | null>(null); const [adding, setAdding] = useState(false);
  const [localW, setLocalW] = useState<Record<string, number>>({}); const w = (p: Property) => localW[p.id] ?? widths[p.id] ?? p.width ?? (p.type === "title" ? 260 : 180);
  const gb = ctx.view.config.groupBy ? ctx.props.find((p) => p.id === ctx.view.config.groupBy) : undefined; const groups = groupRows(ctx.rows, gb);
  const canReorder = !gb && !(ctx.view.config.sorts?.length) && !ctx.search; const sensors = useSensors(useSensor(PointerSensor, { activationConstraint: { distance: 6 } }));
  const flat = groups.flatMap((g) => g.rows);
  const nav = useCallback((e: React.KeyboardEvent) => {
    const el = (e.target as HTMLElement).closest<HTMLElement>("[data-cell]"); if (!el || editing) return; const [r, c] = el.dataset.cell!.split(":").map(Number);
    const go = (nr: number, nc: number) => { const t = el.closest("table")?.querySelector<HTMLElement>(`[data-cell="${nr}:${nc}"]`); if (t) { e.preventDefault(); t.focus(); } };
    if (e.key === "ArrowDown") go(r + 1, c); else if (e.key === "ArrowUp") go(r - 1, c); else if (e.key === "ArrowRight" || (e.key === "Tab" && !e.shiftKey)) go(r, c + 1); else if (e.key === "ArrowLeft" || (e.key === "Tab" && e.shiftKey)) go(r, c - 1);
    else if (e.key === "Enter" || e.key === " ") { e.preventDefault(); const row = flat[r], p = cols[c]; if (row && p) { if (p.type === "checkbox") ctx.setCell(row.id, { [p.id]: !row.values[p.id] }); else setEditing(`${row.id}:${p.id}`); } }
    else if (e.key === "Escape") setEditing(null);
  }, [editing, flat, cols, ctx]);
  const startResize = (p: Property, e: React.PointerEvent) => { e.preventDefault(); const x0 = e.clientX, w0 = w(p); let cur = w0; const move = (ev: PointerEvent) => { cur = Math.max(MIN_W, w0 + ev.clientX - x0); setLocalW((s) => ({ ...s, [p.id]: cur })); }; const up = () => { window.removeEventListener("pointermove", move); window.removeEventListener("pointerup", up); ctx.patchView({ widths: { ...widths, [p.id]: cur } }); }; window.addEventListener("pointermove", move); window.addEventListener("pointerup", up); };
  const onDragEnd = (e: DragEndEvent) => { const { active, over } = e; if (!over || active.id === over.id) return; const ids = flat.map((r) => r.id); const next = arrayMove(ids, ids.indexOf(String(active.id)), ids.indexOf(String(over.id))); api(`/api/databases/${ctx.db.id}/rows`, { method: "PATCH", json: { order: next } }).then(() => ctx.refresh()); };
  useEffect(() => { setSel((s) => new Set([...s].filter((id) => flat.some((r) => r.id === id)))); }, [ctx.rows.length]); // eslint-disable-line react-hooks/exhaustive-deps
  let ri = 0;
  return (
    <div className="relative">
      {sel.size > 0 && <div className="mb-2 flex items-center gap-2 rounded-md border border-line bg-bg-elev px-2 py-1 text-[12px] shadow-card"><span className="font-medium">{sel.size} selected</span><button type="button" onClick={async () => { await ctx.removeRows([...sel]); setSel(new Set()); }} className="flex h-7 items-center gap-1 rounded-md px-2 text-danger hover:bg-danger/10"><Trash2 className="size-3.5" /> Delete</button><button type="button" onClick={() => setSel(new Set())} className="ml-auto text-fg-muted hover:text-fg">Clear</button></div>}
      <div className="overflow-x-auto rounded-md border border-line">
        <DndContext sensors={sensors} collisionDetection={closestCenter} onDragEnd={onDragEnd}><SortableContext items={flat.map((r) => r.id)} strategy={verticalListSortingStrategy}>
          <table className="w-max min-w-full border-collapse text-[13px]" onKeyDown={nav}>
            <thead><tr className="border-b border-line bg-bg-sunk/40">
              <th className="sticky left-0 z-[3] w-8 bg-bg p-0"><div className="h-9" /></th>
              {cols.map((p, ci) => (
                <th key={p.id} style={{ width: w(p), minWidth: w(p), maxWidth: w(p) }} className={cn("group/h relative p-0 text-left font-normal", ci === 0 && "sticky left-8 z-[3] bg-bg")}>
                  <Popover open={editProp === p.id} onOpenChange={(o) => setEditProp(o ? p.id : null)}><PopoverTrigger asChild><button type="button" className="flex h-9 w-full items-center gap-1.5 truncate px-2 text-[12px] font-medium text-fg-muted hover:bg-bg-sunk"><TypeIcon type={p.type} /><span className="truncate">{p.name}</span></button></PopoverTrigger><PopoverContent className="w-auto"><PropertyEditor prop={p} onClose={() => setEditProp(null)} /></PopoverContent></Popover>
                  <div onPointerDown={(e) => startResize(p, e)} className="absolute right-0 top-0 z-[1] h-full w-2 cursor-col-resize touch-none opacity-0 hover:opacity-100 group-hover/h:opacity-100"><div className="mx-auto h-full w-px bg-line-strong" /></div>
                </th>))}
              <th className="w-10 p-0"><Popover open={adding} onOpenChange={setAdding}><PopoverTrigger asChild><button type="button" className="grid h-9 w-10 place-items-center text-fg-faint hover:bg-bg-sunk hover:text-fg"><Plus className="size-4" /></button></PopoverTrigger><PopoverContent className="w-auto"><PropertyEditor onClose={() => setAdding(false)} /></PopoverContent></Popover></th>
            </tr></thead>
            <tbody>
              {groups.map((g) => (g.rows.length || !gb ? [
                gb && <tr key={`g-${g.key}`}><td colSpan={cols.length + 2} className="sticky left-0 bg-bg px-1 pb-1 pt-3"><span className={cn("inline-flex items-center gap-1.5 rounded-md px-1.5 py-0.5 text-[12px] font-medium", OPTION_CLASS[g.color])}>{g.label}<span className="opacity-60">{g.rows.length}</span></span></td></tr>,
                ...g.rows.map((row) => <Row key={row.id} row={row} ri={ri++} cols={cols} w={w} editing={editing} setEditing={setEditing} selected={sel.has(row.id)} onSelect={(v) => setSel((s) => { const n = new Set(s); v ? n.add(row.id) : n.delete(row.id); return n; })} canReorder={canReorder} />),
                gb && <tr key={`a-${g.key}`}><td colSpan={cols.length + 2} className="sticky left-0 bg-bg p-0"><button type="button" onClick={() => ctx.addRow(g.value === undefined ? {} : { [gb.id]: g.value })} className="flex h-8 items-center gap-1 px-2 text-[12px] text-fg-faint hover:text-fg"><Plus className="size-3.5" /> New</button></td></tr>,
              ] : null))}
              {!gb && <tr><td colSpan={cols.length + 2} className="sticky left-0 bg-bg p-0"><button type="button" onClick={() => ctx.addRow()} className="flex h-9 w-full items-center gap-1 px-2 text-[13px] text-fg-faint hover:bg-bg-sunk hover:text-fg"><Plus className="size-4" /> New row</button></td></tr>}
            </tbody>
            <tfoot><tr><td colSpan={cols.length + 2} className="sticky left-0 px-2 py-1.5 text-[11px] text-fg-faint">{ctx.total} row{ctx.total === 1 ? "" : "s"}{ctx.rows.length < ctx.total ? ` · showing ${ctx.rows.length}` : ""}</td></tr></tfoot>
          </table>
        </SortableContext></DndContext>
      </div>
    </div>
  );
}
function Row({ row, ri, cols, w, editing, setEditing, selected, onSelect, canReorder }: { row: RowOut; ri: number; cols: Property[]; w: (p: Property) => number; editing: string | null; setEditing: (k: string | null) => void; selected: boolean; onSelect: (v: boolean) => void; canReorder: boolean }) {
  const { attributes, listeners, setNodeRef, transform, transition, isDragging } = useSortable({ id: row.id, disabled: !canReorder });
  return (
    <tr ref={setNodeRef} style={{ transform: CSS.Transform.toString(transform), transition }} className={cn("group/row border-b border-line hover:bg-bg-sunk/60", isDragging && "relative z-10 bg-bg-elev shadow-pop", selected && "bg-accent-soft/40")}>
      <td className="sticky left-0 z-[2] w-8 bg-bg p-0 align-top"><div className="flex h-9 items-center justify-center gap-0.5"><input type="checkbox" checked={selected} onChange={(e) => onSelect(e.target.checked)} className={cn("size-3.5 accent-[var(--accent)]", !selected && "opacity-0 group-hover/row:opacity-100")} aria-label="Select row" />{canReorder && <button type="button" {...attributes} {...listeners} className="hidden cursor-grab text-fg-faint group-hover/row:block md:block md:opacity-0 md:group-hover/row:opacity-100"><GripVertical className="size-3.5" /></button>}</div></td>
      {cols.map((p, ci) => { const key = `${row.id}:${p.id}`; return (
        <td key={p.id} style={{ width: w(p), minWidth: w(p), maxWidth: w(p) }} className={cn("p-0 align-top", ci === 0 && "sticky left-8 z-[2] bg-bg group-hover/row:bg-bg-sunk", ci > 0 && "border-l border-line")}>
          <div data-cell={`${ri}:${ci}`} tabIndex={0} className="h-full min-h-9 outline-none focus-visible:ring-2 focus-visible:ring-inset focus-visible:ring-accent/40"><Cell p={p} row={row} editing={editing === key} onEdit={(o) => setEditing(o ? key : null)} /></div>
        </td>); })}
      <td className="p-0" />
    </tr>
  );
}