BackTop.js 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109
  1. "use strict";
  2. "use client";
  3. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
  4. var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard").default;
  5. Object.defineProperty(exports, "__esModule", {
  6. value: true
  7. });
  8. exports.default = void 0;
  9. var _react = _interopRequireWildcard(require("react"));
  10. var _VerticalAlignTopOutlined = _interopRequireDefault(require("@ant-design/icons/VerticalAlignTopOutlined"));
  11. var _classnames = _interopRequireDefault(require("classnames"));
  12. var _rcMotion = _interopRequireDefault(require("rc-motion"));
  13. var _ref = require("rc-util/lib/ref");
  14. var _getScroll = _interopRequireDefault(require("../_util/getScroll"));
  15. var _scrollTo = _interopRequireDefault(require("../_util/scrollTo"));
  16. var _throttleByAnimationFrame = _interopRequireDefault(require("../_util/throttleByAnimationFrame"));
  17. var _configProvider = require("../config-provider");
  18. var _context = require("../config-provider/context");
  19. var _context2 = _interopRequireDefault(require("./context"));
  20. var _FloatButton = _interopRequireWildcard(require("./FloatButton"));
  21. var __rest = void 0 && (void 0).__rest || function (s, e) {
  22. var t = {};
  23. for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0) t[p] = s[p];
  24. if (s != null && typeof Object.getOwnPropertySymbols === "function") for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
  25. if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i])) t[p[i]] = s[p[i]];
  26. }
  27. return t;
  28. };
  29. const defaultIcon = /*#__PURE__*/_react.default.createElement(_VerticalAlignTopOutlined.default, null);
  30. const BackTop = /*#__PURE__*/_react.default.forwardRef((props, ref) => {
  31. var _a;
  32. const {
  33. backTopIcon: contextIcon
  34. } = (0, _context.useComponentConfig)('floatButton');
  35. const {
  36. prefixCls: customizePrefixCls,
  37. className,
  38. type = 'default',
  39. shape = 'circle',
  40. visibilityHeight = 400,
  41. icon,
  42. target,
  43. onClick,
  44. duration = 450
  45. } = props,
  46. restProps = __rest(props, ["prefixCls", "className", "type", "shape", "visibilityHeight", "icon", "target", "onClick", "duration"]);
  47. const mergedIcon = (_a = icon !== null && icon !== void 0 ? icon : contextIcon) !== null && _a !== void 0 ? _a : defaultIcon;
  48. const [visible, setVisible] = (0, _react.useState)(visibilityHeight === 0);
  49. const internalRef = _react.default.useRef(null);
  50. _react.default.useImperativeHandle(ref, () => ({
  51. nativeElement: internalRef.current
  52. }));
  53. const getDefaultTarget = () => {
  54. var _a;
  55. return ((_a = internalRef.current) === null || _a === void 0 ? void 0 : _a.ownerDocument) || window;
  56. };
  57. const handleScroll = (0, _throttleByAnimationFrame.default)(e => {
  58. const scrollTop = (0, _getScroll.default)(e.target);
  59. setVisible(scrollTop >= visibilityHeight);
  60. });
  61. (0, _react.useEffect)(() => {
  62. const getTarget = target || getDefaultTarget;
  63. const container = getTarget();
  64. handleScroll({
  65. target: container
  66. });
  67. container === null || container === void 0 ? void 0 : container.addEventListener('scroll', handleScroll);
  68. return () => {
  69. handleScroll.cancel();
  70. container === null || container === void 0 ? void 0 : container.removeEventListener('scroll', handleScroll);
  71. };
  72. }, [target]);
  73. const scrollToTop = e => {
  74. (0, _scrollTo.default)(0, {
  75. getContainer: target || getDefaultTarget,
  76. duration
  77. });
  78. onClick === null || onClick === void 0 ? void 0 : onClick(e);
  79. };
  80. const {
  81. getPrefixCls
  82. } = (0, _react.useContext)(_configProvider.ConfigContext);
  83. const prefixCls = getPrefixCls(_FloatButton.floatButtonPrefixCls, customizePrefixCls);
  84. const rootPrefixCls = getPrefixCls();
  85. const groupShape = (0, _react.useContext)(_context2.default);
  86. const mergedShape = groupShape || shape;
  87. const contentProps = Object.assign({
  88. prefixCls,
  89. icon: mergedIcon,
  90. type,
  91. shape: mergedShape
  92. }, restProps);
  93. return /*#__PURE__*/_react.default.createElement(_rcMotion.default, {
  94. visible: visible,
  95. motionName: `${rootPrefixCls}-fade`
  96. }, ({
  97. className: motionClassName
  98. }, setRef) => (/*#__PURE__*/_react.default.createElement(_FloatButton.default, Object.assign({
  99. ref: (0, _ref.composeRef)(internalRef, setRef)
  100. }, contentProps, {
  101. onClick: scrollToTop,
  102. className: (0, _classnames.default)(className, motionClassName)
  103. }))));
  104. });
  105. if (process.env.NODE_ENV !== 'production') {
  106. BackTop.displayName = 'BackTop';
  107. }
  108. var _default = exports.default = BackTop;