import { Skeleton } from "@/components/shadcn/skeleton/skeleton";
import {
  DEFAULT_SCAN_JOBS_TAB,
  SCAN_JOBS_TAB,
  type ScanJobsTab,
} from "@/types";

const AccountCellSkeleton = () => (
  <td className="px-3 py-4">
    <div className="flex items-center gap-4">
      <Skeleton className="size-9 rounded-xl" />
      <div className="flex flex-col gap-1">
        <Skeleton className="h-3.5 w-20 rounded" />
        <div className="bg-bg-neutral-tertiary border-border-neutral-tertiary flex h-6 w-32 items-center gap-1 rounded-full border-2 px-2">
          <Skeleton className="h-3 w-20 rounded" />
          <Skeleton className="h-3.5 w-3.5 rounded" />
        </div>
      </div>
    </div>
  </td>
);

const ScanInfoCellSkeleton = () => (
  <td className="px-3 py-4">
    <div className="flex flex-col gap-1">
      <Skeleton className="h-3.5 w-32 rounded" />
      <div className="bg-bg-neutral-tertiary border-border-neutral-tertiary flex h-6 w-32 items-center gap-1 rounded-full border-2 px-2">
        <Skeleton className="h-3 w-20 rounded" />
        <Skeleton className="h-3.5 w-3.5 rounded" />
      </div>
    </div>
  </td>
);

const ProgressCellSkeleton = () => (
  <td className="px-3 py-4">
    <div className="flex min-w-[220px] items-center gap-3">
      <Skeleton className="h-2 w-[140px] rounded-full" />
      <Skeleton className="h-3.5 w-9 rounded" />
    </div>
  </td>
);

const ScheduleCellSkeleton = () => (
  <td className="px-3 py-4">
    <div className="flex flex-col gap-1">
      <Skeleton className="h-4 w-20 rounded" />
      <Skeleton className="h-3 w-32 rounded" />
    </div>
  </td>
);

const DateCellSkeleton = () => (
  <td className="px-3 py-4">
    <div className="flex flex-col gap-1">
      <Skeleton className="h-4 w-24 rounded" />
      <Skeleton className="h-3 w-20 rounded" />
    </div>
  </td>
);

const ResourcesCellSkeleton = () => (
  <td className="px-3 py-4">
    <Skeleton className="h-6 w-12 rounded" />
  </td>
);

const DurationCellSkeleton = () => (
  <td className="px-3 py-4">
    <Skeleton className="h-4 w-16 rounded" />
  </td>
);

const StatusCellSkeleton = () => (
  <td className="px-3 py-4">
    <Skeleton className="h-6 w-20 rounded-md" />
  </td>
);

const ActionsCellSkeleton = () => (
  <td className="px-2 py-4">
    <Skeleton className="size-7 rounded" />
  </td>
);

const HeaderLabel = ({
  width,
  sortable = false,
}: {
  width: string;
  sortable?: boolean;
}) => (
  <div className="flex h-8 items-center gap-1">
    <Skeleton className={`h-4 ${width} rounded`} />
    {sortable && <Skeleton className="size-3.5 rounded" />}
  </div>
);

interface ColumnDescriptor {
  headerWidth: string;
  sortable?: boolean;
  Cell: () => React.JSX.Element;
}

const ACCOUNT_COLUMN: ColumnDescriptor = {
  headerWidth: "w-14",
  Cell: AccountCellSkeleton,
};
const SCAN_INFO_COLUMN: ColumnDescriptor = {
  headerWidth: "w-8",
  sortable: true,
  Cell: ScanInfoCellSkeleton,
};
const PROGRESS_COLUMN: ColumnDescriptor = {
  headerWidth: "w-14",
  Cell: ProgressCellSkeleton,
};
const SCHEDULE_COLUMN: ColumnDescriptor = {
  headerWidth: "w-14",
  sortable: true,
  Cell: ScheduleCellSkeleton,
};
const LAUNCHED_COLUMN: ColumnDescriptor = {
  headerWidth: "w-14",
  Cell: DateCellSkeleton,
};
const RESOURCES_COLUMN: ColumnDescriptor = {
  headerWidth: "w-16",
  Cell: ResourcesCellSkeleton,
};
const DURATION_COLUMN: ColumnDescriptor = {
  headerWidth: "w-14",
  Cell: DurationCellSkeleton,
};
const STATUS_COLUMN: ColumnDescriptor = {
  headerWidth: "w-10",
  Cell: StatusCellSkeleton,
};
const COMPLETED_COLUMN: ColumnDescriptor = {
  headerWidth: "w-16",
  sortable: true,
  Cell: DateCellSkeleton,
};
const NEXT_RUN_COLUMN: ColumnDescriptor = {
  headerWidth: "w-14",
  sortable: true,
  Cell: DateCellSkeleton,
};

const COLUMNS_BY_TAB: Record<ScanJobsTab, ColumnDescriptor[]> = {
  [SCAN_JOBS_TAB.ACTIVE]: [
    ACCOUNT_COLUMN,
    SCAN_INFO_COLUMN,
    PROGRESS_COLUMN,
    SCHEDULE_COLUMN,
    LAUNCHED_COLUMN,
  ],
  [SCAN_JOBS_TAB.COMPLETED]: [
    ACCOUNT_COLUMN,
    SCAN_INFO_COLUMN,
    RESOURCES_COLUMN,
    DURATION_COLUMN,
    STATUS_COLUMN,
    SCHEDULE_COLUMN,
    COMPLETED_COLUMN,
  ],
  [SCAN_JOBS_TAB.SCHEDULED]: [
    ACCOUNT_COLUMN,
    SCAN_INFO_COLUMN,
    SCHEDULE_COLUMN,
    NEXT_RUN_COLUMN,
  ],
};

interface SkeletonTableScansProps {
  tab?: ScanJobsTab;
  rows?: number;
}

export const SkeletonTableScans = ({
  tab = DEFAULT_SCAN_JOBS_TAB,
  rows = 6,
}: SkeletonTableScansProps = {}) => {
  const columns = COLUMNS_BY_TAB[tab];

  return (
    <div className="border-border-neutral-secondary bg-bg-neutral-secondary flex w-full flex-col justify-between gap-4 overflow-hidden rounded-[14px] border p-4 shadow-sm">
      {/* Toolbar — mirrors DataTable's flex-col → md:flex-row layout (no search, only total entries) */}
      <div className="flex flex-col items-start gap-3 md:flex-row md:items-center md:justify-between">
        <div className="w-full md:w-auto" />
        <div className="flex w-full flex-col items-start gap-2 md:ml-auto md:w-auto md:flex-row md:items-center md:gap-4">
          <Skeleton className="h-4 w-28 rounded" />
        </div>
      </div>

      {/* Table */}
      <table className="w-full">
        <thead>
          <tr className="border-border-neutral-secondary border-b">
            {columns.map((column, i) => (
              <th key={i} className="px-3 py-3 text-left">
                <HeaderLabel
                  width={column.headerWidth}
                  sortable={column.sortable}
                />
              </th>
            ))}
            {/* Actions - empty header */}
            <th className="w-10 py-3" />
          </tr>
        </thead>
        <tbody>
          {Array.from({ length: rows }).map((_, rowIdx) => (
            <tr
              key={rowIdx}
              className="border-border-neutral-secondary border-b last:border-b-0"
            >
              {columns.map(({ Cell }, colIdx) => (
                <Cell key={colIdx} />
              ))}
              <ActionsCellSkeleton />
            </tr>
          ))}
        </tbody>
      </table>

      {/* Pagination — mirrors DataTablePagination's "justify-end gap-6 py-1.5" */}
      <div className="flex w-full items-center justify-end gap-6 py-1.5">
        {/* Rows per page group */}
        <div className="flex items-center gap-3">
          <Skeleton className="h-3 w-20 rounded" />
          <Skeleton className="h-8 w-12 rounded-full" />
        </div>
        {/* Page info + 4 chevron nav buttons */}
        <div className="flex items-center gap-3">
          <Skeleton className="hidden h-3 w-20 rounded sm:block" />
          <div className="flex items-center gap-3">
            <Skeleton className="size-6 rounded" />
            <Skeleton className="size-6 rounded" />
            <Skeleton className="size-6 rounded" />
            <Skeleton className="size-6 rounded" />
          </div>
        </div>
      </div>
    </div>
  );
};