98 lines
3.5 KiB
JavaScript
98 lines
3.5 KiB
JavaScript
|
"use client";
|
||
|
|
|
||
|
|
var __rest = this && this.__rest || function (s, e) {
|
||
|
|
var t = {};
|
||
|
|
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0) t[p] = s[p];
|
||
|
|
if (s != null && typeof Object.getOwnPropertySymbols === "function") for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
|
||
|
|
if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i])) t[p[i]] = s[p[i]];
|
||
|
|
}
|
||
|
|
return t;
|
||
|
|
};
|
||
|
|
import React, { useContext, useEffect, useState } from 'react';
|
||
|
|
import VerticalAlignTopOutlined from "@ant-design/icons/es/icons/VerticalAlignTopOutlined";
|
||
|
|
import classNames from 'classnames';
|
||
|
|
import CSSMotion from 'rc-motion';
|
||
|
|
import { composeRef } from "rc-util/es/ref";
|
||
|
|
import getScroll from '../_util/getScroll';
|
||
|
|
import scrollTo from '../_util/scrollTo';
|
||
|
|
import throttleByAnimationFrame from '../_util/throttleByAnimationFrame';
|
||
|
|
import { ConfigContext } from '../config-provider';
|
||
|
|
import FloatButtonGroupContext from './context';
|
||
|
|
import FloatButton, { floatButtonPrefixCls } from './FloatButton';
|
||
|
|
const BackTop = /*#__PURE__*/React.forwardRef((props, ref) => {
|
||
|
|
const {
|
||
|
|
prefixCls: customizePrefixCls,
|
||
|
|
className,
|
||
|
|
type = 'default',
|
||
|
|
shape = 'circle',
|
||
|
|
visibilityHeight = 400,
|
||
|
|
icon = /*#__PURE__*/React.createElement(VerticalAlignTopOutlined, null),
|
||
|
|
target,
|
||
|
|
onClick,
|
||
|
|
duration = 450
|
||
|
|
} = props,
|
||
|
|
restProps = __rest(props, ["prefixCls", "className", "type", "shape", "visibilityHeight", "icon", "target", "onClick", "duration"]);
|
||
|
|
const [visible, setVisible] = useState(visibilityHeight === 0);
|
||
|
|
const internalRef = React.useRef(null);
|
||
|
|
React.useImperativeHandle(ref, () => ({
|
||
|
|
nativeElement: internalRef.current
|
||
|
|
}));
|
||
|
|
const getDefaultTarget = () => {
|
||
|
|
var _a;
|
||
|
|
return ((_a = internalRef.current) === null || _a === void 0 ? void 0 : _a.ownerDocument) || window;
|
||
|
|
};
|
||
|
|
const handleScroll = throttleByAnimationFrame(e => {
|
||
|
|
const scrollTop = getScroll(e.target);
|
||
|
|
setVisible(scrollTop >= visibilityHeight);
|
||
|
|
});
|
||
|
|
useEffect(() => {
|
||
|
|
const getTarget = target || getDefaultTarget;
|
||
|
|
const container = getTarget();
|
||
|
|
handleScroll({
|
||
|
|
target: container
|
||
|
|
});
|
||
|
|
container === null || container === void 0 ? void 0 : container.addEventListener('scroll', handleScroll);
|
||
|
|
return () => {
|
||
|
|
handleScroll.cancel();
|
||
|
|
container === null || container === void 0 ? void 0 : container.removeEventListener('scroll', handleScroll);
|
||
|
|
};
|
||
|
|
}, [target]);
|
||
|
|
const scrollToTop = e => {
|
||
|
|
scrollTo(0, {
|
||
|
|
getContainer: target || getDefaultTarget,
|
||
|
|
duration
|
||
|
|
});
|
||
|
|
onClick === null || onClick === void 0 ? void 0 : onClick(e);
|
||
|
|
};
|
||
|
|
const {
|
||
|
|
getPrefixCls
|
||
|
|
} = useContext(ConfigContext);
|
||
|
|
const prefixCls = getPrefixCls(floatButtonPrefixCls, customizePrefixCls);
|
||
|
|
const rootPrefixCls = getPrefixCls();
|
||
|
|
const groupShape = useContext(FloatButtonGroupContext);
|
||
|
|
const mergedShape = groupShape || shape;
|
||
|
|
const contentProps = Object.assign({
|
||
|
|
prefixCls,
|
||
|
|
icon,
|
||
|
|
type,
|
||
|
|
shape: mergedShape
|
||
|
|
}, restProps);
|
||
|
|
return /*#__PURE__*/React.createElement(CSSMotion, {
|
||
|
|
visible: visible,
|
||
|
|
motionName: `${rootPrefixCls}-fade`
|
||
|
|
}, (_ref, setRef) => {
|
||
|
|
let {
|
||
|
|
className: motionClassName
|
||
|
|
} = _ref;
|
||
|
|
return /*#__PURE__*/React.createElement(FloatButton, Object.assign({
|
||
|
|
ref: composeRef(internalRef, setRef)
|
||
|
|
}, contentProps, {
|
||
|
|
onClick: scrollToTop,
|
||
|
|
className: classNames(className, motionClassName)
|
||
|
|
}));
|
||
|
|
});
|
||
|
|
});
|
||
|
|
if (process.env.NODE_ENV !== 'production') {
|
||
|
|
BackTop.displayName = 'BackTop';
|
||
|
|
}
|
||
|
|
export default BackTop;
|