usePagination.js 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475
  1. "use strict";
  2. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
  3. Object.defineProperty(exports, "__esModule", {
  4. value: true
  5. });
  6. exports.default = exports.DEFAULT_PAGE_SIZE = void 0;
  7. exports.getPaginationParam = getPaginationParam;
  8. var _react = require("react");
  9. var _extendsObject = _interopRequireDefault(require("../../_util/extendsObject"));
  10. var __rest = void 0 && (void 0).__rest || function (s, e) {
  11. var t = {};
  12. for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0) t[p] = s[p];
  13. if (s != null && typeof Object.getOwnPropertySymbols === "function") for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
  14. if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i])) t[p[i]] = s[p[i]];
  15. }
  16. return t;
  17. };
  18. const DEFAULT_PAGE_SIZE = exports.DEFAULT_PAGE_SIZE = 10;
  19. function getPaginationParam(mergedPagination, pagination) {
  20. const param = {
  21. current: mergedPagination.current,
  22. pageSize: mergedPagination.pageSize
  23. };
  24. const paginationObj = pagination && typeof pagination === 'object' ? pagination : {};
  25. Object.keys(paginationObj).forEach(pageProp => {
  26. const value = mergedPagination[pageProp];
  27. if (typeof value !== 'function') {
  28. param[pageProp] = value;
  29. }
  30. });
  31. return param;
  32. }
  33. function usePagination(total, onChange, pagination) {
  34. const _a = pagination && typeof pagination === 'object' ? pagination : {},
  35. {
  36. total: paginationTotal = 0
  37. } = _a,
  38. paginationObj = __rest(_a, ["total"]);
  39. const [innerPagination, setInnerPagination] = (0, _react.useState)(() => ({
  40. current: 'defaultCurrent' in paginationObj ? paginationObj.defaultCurrent : 1,
  41. pageSize: 'defaultPageSize' in paginationObj ? paginationObj.defaultPageSize : DEFAULT_PAGE_SIZE
  42. }));
  43. // ============ Basic Pagination Config ============
  44. const mergedPagination = (0, _extendsObject.default)(innerPagination, paginationObj, {
  45. total: paginationTotal > 0 ? paginationTotal : total
  46. });
  47. // Reset `current` if data length or pageSize changed
  48. const maxPage = Math.ceil((paginationTotal || total) / mergedPagination.pageSize);
  49. if (mergedPagination.current > maxPage) {
  50. // Prevent a maximum page count of 0
  51. mergedPagination.current = maxPage || 1;
  52. }
  53. const refreshPagination = (current, pageSize) => {
  54. setInnerPagination({
  55. current: current !== null && current !== void 0 ? current : 1,
  56. pageSize: pageSize || mergedPagination.pageSize
  57. });
  58. };
  59. const onInternalChange = (current, pageSize) => {
  60. var _a;
  61. if (pagination) {
  62. (_a = pagination.onChange) === null || _a === void 0 ? void 0 : _a.call(pagination, current, pageSize);
  63. }
  64. refreshPagination(current, pageSize);
  65. onChange(current, pageSize || (mergedPagination === null || mergedPagination === void 0 ? void 0 : mergedPagination.pageSize));
  66. };
  67. if (pagination === false) {
  68. return [{}, () => {}];
  69. }
  70. return [Object.assign(Object.assign({}, mergedPagination), {
  71. onChange: onInternalChange
  72. }), refreshPagination];
  73. }
  74. var _default = exports.default = usePagination;