"use client";

import * as React from "react";

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);
                // Key on the array element itself; a key set inside the returned component
                // (e.g. MasterTextFilterCell's inner TableCell) does not count for this list.
                return (
                  <React.Fragment key={`filter-${header.id}`}>
                    {content ?? <TableCell className="p-2" />}
                  </React.Fragment>
                );
              })}
            </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="max-w-[22rem] px-3 py-3 align-top text-sm whitespace-normal break-words [&_span]:whitespace-normal"
                  >
                    {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>
  );
}