import * as React from 'react'; import Animate from 'rc-animate'; import Icon from '../icon'; import addEventListener from 'rc-util/lib/Dom/addEventListener'; import classNames from 'classnames'; import omit from 'object.omit'; function getScroll(w, top) { let ret = w[`page${top ? 'Y' : 'X'}Offset`]; const method = `scroll${top ? 'Top' : 'Left'}`; if (typeof ret !== 'number') { const d = w.document; // ie6,7,8 standard mode ret = d.documentElement[method]; if (typeof ret !== 'number') { // quirks mode ret = d.body[method]; } } return ret; } interface BackTopProps { visibilityHeight?: number; onClick?: (event) => void; prefixCls?: string; className?: string; } export default class BackTop extends React.Component { static defaultProps = { onClick() {}, visibilityHeight: 400, prefixCls: 'ant-back-top', }; scrollEvent: any; constructor(props) { super(props); const scrollTop = getScroll(window, true); this.state = { visible: scrollTop > this.props.visibilityHeight, }; } scrollToTop = (e) => { if (e) { e.preventDefault(); } this.setScrollTop(0); this.props.onClick(e); } setScrollTop(value) { document.body.scrollTop = value; document.documentElement.scrollTop = value; } handleScroll = () => { const scrollTop = getScroll(window, true); this.setState({ visible: scrollTop > this.props.visibilityHeight, }); } componentDidMount() { this.scrollEvent = addEventListener(window, 'scroll', this.handleScroll); } componentWillUnmount() { if (this.scrollEvent) { this.scrollEvent.remove(); } } render() { const { prefixCls, className, children } = this.props; const classString = classNames({ [prefixCls]: true, [className]: !!className, }); const defaultElement = (
); // fix https://fb.me/react-unknown-prop const divProps = omit(this.props, [ 'prefixCls', 'className', 'children', 'visibilityHeight', ]); return ( { this.state.visible ?
{children || defaultElement}
: null }
); } }