import { cn } from "@/lib/utils";
import { Skeleton } from "@/components/ui/skeleton";
import {
  Table,
  TableBody,
  TableCell,
  TableHead,
  TableHeader,
  TableRow,
} from "@/components/ui/table";

interface TableSkeletonProps extends Omit<
  React.ComponentProps<"div">,
  "children"
> {
  rows?: number;
  columns?: number;
  /** Use grid layout: search 10/12, page + controls 2/12 (e.g. fix-packages table) */
  toolbarVariant?: "default" | "search10Page2";
}

export function TableSkeleton({
  rows = 5,
  columns = 6,
  toolbarVariant = "default",
  className,
  ...rest
}: TableSkeletonProps) {
  const searchSkeleton = (
    <div className="relative flex justify-start w-full">
      <Skeleton className="h-10 w-full max-w-sm" />
    </div>
  );
  const pageSkeleton = (
    <div className="flex items-center justify-end gap-2">
      <Skeleton className="h-5 w-12 shrink-0" />
      <Skeleton className="h-10 w-20 shrink-0" />
    </div>
  );

  return (
    <div className={cn("space-y-4", className)} {...rest}>
      {/* Search and toolbar skeleton */}
      <div
        className={
          toolbarVariant === "search10Page2"
            ? "grid grid-cols-12 gap-4 items-center"
            : "flex flex-col gap-4 sm:flex-row sm:items-center sm:justify-between"
        }
      >
        {toolbarVariant === "search10Page2" ? (
          <>
            <div className="col-span-12 sm:col-span-10">{searchSkeleton}</div>
            <div className="col-span-12 sm:col-span-2">{pageSkeleton}</div>
          </>
        ) : (
          <>
            {searchSkeleton}
            {pageSkeleton}
          </>
        )}
      </div>

      {/* Results count skeleton */}
      <Skeleton className="h-5 w-48" />

      {/* Table skeleton */}
      <div className="rounded-md border">
        <Table>
          <TableHeader>
            <TableRow>
              {Array.from({ length: columns }).map((_, i) => (
                <TableHead key={i}>
                  <Skeleton className="h-4 w-24" />
                </TableHead>
              ))}
            </TableRow>
          </TableHeader>
          <TableBody>
            {Array.from({ length: rows }).map((_, rowIndex) => (
              <TableRow key={rowIndex}>
                {Array.from({ length: columns }).map((_, colIndex) => (
                  <TableCell key={colIndex}>
                    <Skeleton className="h-4 w-full" />
                  </TableCell>
                ))}
              </TableRow>
            ))}
          </TableBody>
        </Table>
      </div>

      {/* Pagination skeleton */}
      <div className="flex items-center justify-between">
        <Skeleton className="h-5 w-32" />
        <div className="flex items-center gap-1">
          <Skeleton className="h-10 w-10" />
          <Skeleton className="h-10 w-10" />
          <Skeleton className="h-10 w-10" />
          <Skeleton className="h-10 w-10" />
          <Skeleton className="h-10 w-10" />
        </div>
      </div>
    </div>
  );
}