import { computed, reactive } from 'vue-demi'

export type PaginationReq<T extends object> = T & {
  currenPage: number,
  pageSize: number
}

export interface PaginationRes<T> {
  total: number,
  data: Array<T>
}

export function usePagination (size = 10) {
  const pagination = reactive({
    currenPage: 1,
    pageSize: size,
    total: 0
  })

  const pages = computed(() => Math.ceil(pagination.total / pagination.pageSize) || 1)

  function wrapperPagePromise<T> (p: Promise<PaginationRes<T>>) {
    return p.then(({ total, data }) => {
      pagination.total = total ?? 0
      return data
    })
  }

  /**
   * 自动分页
   * @param queryFunc
   */
  function wrapperPagePromiseFn <T extends object, U> (queryFunc: (params: PaginationReq<T>) => Promise<PaginationRes<U>>) {
    return (params: T) => {
      const queryParam = {
        ...params,
        ...pagination
      }
      return wrapperPagePromise(queryFunc(queryParam as PaginationReq<T>))
    }
  }

  return { pagination, pages, wrapperPagePromise, wrapperPagePromiseFn }
}