import {
  AccessorFn,
  ColumnDef,
  DeepKeys,
  DeepValue,
  DisplayColumnDef,
  GroupColumnDef,
  IdentifiedColumnDef,
  RowData,
} from '@tanstack/react-table'

type CustomFieldColumnDef = {
  isHidden?: boolean
}

export type CustomColumnDef<T, Y = unknown> = ColumnDef<T, Y> & CustomFieldColumnDef

export type ColumnHelper<TData extends RowData> = {
  accessor: <
    TAccessor extends AccessorFn<TData> | DeepKeys<TData>,
    TValue extends TAccessor extends AccessorFn<TData, infer TReturn>
      ? TReturn
      : TAccessor extends DeepKeys<TData>
      ? DeepValue<TData, TAccessor>
      : never,
  >(
    accessor: TAccessor,
    column: TAccessor extends AccessorFn<TData>
      ? DisplayColumnDef<TData, TValue> & CustomFieldColumnDef
      : IdentifiedColumnDef<TData, TValue> & CustomFieldColumnDef,
  ) => CustomColumnDef<TData, TValue>
  display: (column: DisplayColumnDef<TData>) => CustomColumnDef<TData, unknown>
  group: (column: GroupColumnDef<TData>) => CustomColumnDef<TData, unknown>
}