import { useMemo } from 'react';
import { usePagination, UsePaginationInstanceProps, UsePaginationOptions, UsePaginationState } from 'react-table';
import { pick, values } from 'lodash-es';

import { IRelatableStateInstance, PageSetter, PageSizeSetter, TableAddOnReturn } from '../relatable.types';

import { DEFAULT_PAGE_SIZE_OPTIONS } from '../constants';

export interface IWithPaginationOptions<Data extends object = any> extends UsePaginationOptions<Data> {
  onPageChange?: PageSetter;
  onPageSizeChange?: PageSizeSetter;
  pageSizeOptions?: number[];

  // react-table state overrides https://react-table.js.org/api/usePagination
  pageSize?: number;
  pageIndex?: number;
}

export interface IWithPaginationState<Data extends object = any> extends UsePaginationState<Data> {
}

export interface IWithPaginationInstance<Data extends object = any> extends UsePaginationInstanceProps<Data>, IRelatableStateInstance<Data, IWithPaginationState<Data>> {
  customPageSizeOptions: number[];
  onCustomPageSizeChange?: PageSizeSetter;
  onCustomPageChange?: PageSetter;
}

export default function withPagination<Data extends object = any>(options: IWithPaginationOptions<Data> = {}): TableAddOnReturn {
  const {
    pageSize,
    pageIndex,
    onPageSizeChange,
    onPageChange,
    pageSizeOptions = DEFAULT_PAGE_SIZE_OPTIONS,
    ...tableParams
  } = options;
  const stateParams = pick(options, ['pageSize', 'pageIndex']);

  return [
    withPagination.name,
    null,
    () => false,
    () => useMemo((): Partial<IWithPaginationInstance> => ({
      ...tableParams,
      customPageSizeOptions: pageSizeOptions,
      // @todo: figure out strategy for these
      onCustomPageSizeChange: onPageSizeChange,
      onCustomPageChange: onPageChange,
    }), [onPageSizeChange, onPageChange, ...values(tableParams)]),
    () => useMemo(() => stateParams, values(stateParams)),
    usePagination,
  ];
}