"use client";

import { flexRender, type Header, type Table as TanStackTable } from "@tanstack/react-table";

import {
  Table,
  TableBody,
  TableCell,
  TableHead,
  TableHeader,
  TableRow,
} from "@/components/ui/table";

type Props<T> = {
  table: TanStackTable<T>;
  columnCount: number;
  emptyMessage?: string;
  /**
   * Render a filter cell per column. Return a `<TableCell>` (or `null` for an empty cell).
   * The shell wraps non-cell returns in a padded `<TableCell>` automatically.
   */
  renderFilterCell?: (header: Header<T, unknown>) => React.ReactNode;
};

export function DataTableShell<T>({
  table,
  columnCount,
  emptyMessage = "No results.",
  renderFilterCell,
}: Props<T>) {
  const headerGroups = table.getHeaderGroups();

  return (
    <div className="overflow-x-auto rounded-lg border bg-card">
      <Table>
        <TableHeader className="bg-muted/15">
          {headerGroups.map((headerGroup) => (
            <TableRow key={headerGroup.id}>
              {headerGroup.headers.map((header) => (
                <TableHead
                  key={header.id}
                  className="min-w-[120px] px-3 py-2.5 font-medium whitespace-nowrap"
                >
                  {header.isPlaceholder
                    ? null
                    : flexRender(header.column.columnDef.header, header.getContext())}
                </TableHead>
              ))}
            </TableRow>
          ))}
          {renderFilterCell ? (
            <TableRow className="bg-muted/5 hover:bg-muted/5">
              {headerGroups[0]?.headers.map((header) => {
                const content = renderFilterCell(header);
                return content ?? <TableCell key={`filter-${header.id}`} className="p-2" />;
              })}
            </TableRow>
          ) : null}
        </TableHeader>
        <TableBody>
          {table.getRowModel().rows.length ? (
            table.getRowModel().rows.map((row) => (
              <TableRow key={row.id} className="align-top hover:bg-muted/30">
                {row.getVisibleCells().map((cell) => (
                  <TableCell key={cell.id} className="px-3 py-3 align-top text-sm">
                    {flexRender(cell.column.columnDef.cell, cell.getContext())}
                  </TableCell>
                ))}
              </TableRow>
            ))
          ) : (
            <TableRow>
              <TableCell colSpan={columnCount} className="h-24 text-center text-muted-foreground">
                {emptyMessage}
              </TableCell>
            </TableRow>
          )}
        </TableBody>
      </Table>
    </div>
  );
}