import * as React from 'react';

import { classNames } from '../../lib/utils';

const Table = React.forwardRef<HTMLTableElement, React.ComponentProps<'table'>>(
  function Table({ className, ...props }, ref) {
    return (
      <div
        data-slot="table-container"
        className="relative w-full overflow-x-auto"
      >
        <table
          data-slot="table"
          className={classNames('w-full caption-bottom text-sm', className)}
          {...props}
          ref={ref}
        />
      </div>
    );
  }
);

const TableHeader = React.forwardRef<
  HTMLTableSectionElement,
  React.ComponentProps<'thead'>
>(function TableHeader({ className, ...props }, ref) {
  return (
    <thead
      data-slot="table-header"
      className={classNames('[&_tr]:border-b', className)}
      {...props}
      ref={ref}
    />
  );
});

const TableBody = React.forwardRef<
  HTMLTableSectionElement,
  React.ComponentProps<'tbody'>
>(function TableBody({ className, ...props }, ref) {
  return (
    <tbody
      data-slot="table-body"
      className={classNames('[&_tr:last-child]:border-0', className)}
      {...props}
      ref={ref}
    />
  );
});

const TableFooter = React.forwardRef<
  HTMLTableSectionElement,
  React.ComponentProps<'tfoot'>
>(function TableFooter({ className, ...props }, ref) {
  return (
    <tfoot
      data-slot="table-footer"
      className={classNames(
        'border-t bg-muted/50 font-medium [&>tr]:last:border-b-0',
        className
      )}
      {...props}
      ref={ref}
    />
  );
});

const TableRow = React.forwardRef<
  HTMLTableRowElement,
  React.ComponentProps<'tr'>
>(function TableRow({ className, ...props }, ref) {
  return (
    <tr
      data-slot="table-row"
      className={classNames(
        'border-b transition-colors hover:bg-muted/50 data-[state=selected]:bg-muted',
        className
      )}
      {...props}
      ref={ref}
    />
  );
});

const TableHead = React.forwardRef<
  HTMLTableHeaderCellElement,
  React.ComponentProps<'th'>
>(function TableHead({ className, ...props }, ref) {
  return (
    <th
      data-slot="table-head"
      className={classNames(
        'h-9 px-2 text-left align-middle text-[11px] font-semibold uppercase tracking-wider whitespace-nowrap text-muted-foreground [&:has([role=checkbox])]:pr-0',
        className
      )}
      {...props}
      ref={ref}
    />
  );
});

const TableCell = React.forwardRef<
  HTMLTableDataCellElement,
  React.ComponentProps<'td'>
>(function TableCell({ className, ...props }, ref) {
  return (
    <td
      data-slot="table-cell"
      className={classNames(
        'px-2 py-2.5 align-middle whitespace-nowrap [&:has([role=checkbox])]:pr-0',
        className
      )}
      {...props}
      ref={ref}
    />
  );
});

const TableCaption = React.forwardRef<
  HTMLElement,
  React.ComponentProps<'caption'>
>(function TableCaption({ className, ...props }, ref) {
  return (
    <caption
      data-slot="table-caption"
      className={classNames('mt-4 text-sm text-muted-foreground', className)}
      {...props}
      ref={ref}
    />
  );
});

export {
  Table,
  TableHeader,
  TableBody,
  TableFooter,
  TableHead,
  TableRow,
  TableCell,
  TableCaption,
};