import {
  type ColumnDef,
  getCoreRowModel,
  getFacetedRowModel,
  getFacetedUniqueValues,
  getFilteredRowModel,
  getPaginationRowModel,
  getSortedRowModel,
  type TableOptions,
  type TableState,
  useReactTable
} from "@tanstack/react-table";
import { useEffect } from "react";

import { useI18n } from "../../../hooks/useI18n.js";
import { type FormOptions, Operation } from "../../../interfaces";
import type { JSONRecord } from "../../../interfaces/JSONRecord.js";
import { getComponent } from "../../../registries/components";
import type { DefaultCellOperations } from "../components/DefaultCellOperations";

export interface UseTableProps<Data extends object = JSONRecord> extends Omit<TableOptions<Data>, "onClick" | "getCoreRowModel"> {
  operations: Operation<Data>[];
  metadata?: Record<string, unknown>;
  i18n?: FormOptions["i18n"];
  onClick?: (data: Data, operation: Operation<Data>) => void;
  manualFaceted?: boolean;
  onChange?: (query: TableState) => void;
  pageSizes?: number[];
}

export function useTable<Data extends object = JSONRecord>(props: UseTableProps<Data>) {
  const Operations = getComponent<typeof DefaultCellOperations<Data>>("CellOperations");
  const { t } = useI18n(props.i18n);

  const operations = props.operations.length
    ? ([
        {
          id: "operations",
          header: t("Operations"),
          cell: (info) => (
            <Operations info={info} operations={props.operations} metadata={props.metadata} onClick={props.onClick} i18n={t} />
          )
        }
      ] satisfies ColumnDef<Data>[])
    : [];

  const tableInstance = useReactTable({
    ...props,
    columns: [...props.columns, ...operations],
    getCoreRowModel: getCoreRowModel(),
    getPaginationRowModel: getPaginationRowModel(),
    getFilteredRowModel: !props.manualFiltering ? props.getFilteredRowModel || getFilteredRowModel() : undefined,
    getSortedRowModel: !props.manualSorting ? props.getSortedRowModel || getSortedRowModel() : undefined,
    getFacetedRowModel: !props.manualFaceted ? props.getFacetedRowModel || getFacetedRowModel() : undefined, // client-side faceting
    getFacetedUniqueValues: !props.manualFaceted ? props.getFacetedUniqueValues || getFacetedUniqueValues() : undefined // generate unique values for select filter/autocomplete
  });

  const { columnFilters, sorting, pagination, columnOrder, columnPinning, columnSizing, columnSizingInfo, columnVisibility, globalFilter } =
    tableInstance.getState();

  useEffect(() => {
    if (props.onChange) {
      props.onChange(tableInstance.getState());
    }
  }, [columnFilters, sorting, pagination, columnOrder, columnPinning, columnSizing, columnSizingInfo, columnVisibility, globalFilter]);

  return {
    i18n: t,
    tableInstance
  };
}