FloatButtonGroup.tsx

import React, { useRef, memo, useContext, useEffect } from 'react';
import CloseOutlined from '@ant-design/icons/CloseOutlined';
import FileTextOutlined from '@ant-design/icons/FileTextOutlined';
import classNames from 'classnames';
import CSSMotion from 'rc-motion';
import useMergedState from 'rc-util/lib/hooks/useMergedState';
import FloatButton, { floatButtonPrefixCls } from './FloatButton';
import type { ConfigConsumerProps } from '../config-provider';
import { ConfigContext } from '../config-provider';
import { FloatButtonGroupProvider } from './context';
import type { FloatButtonGroupProps } from './interface';
import useStyle from './style';

const FloatButtonGroup: React.FC<FloatButtonGroupProps> = (props) => {
  const {
    prefixCls: customizePrefixCls,
    className,
    style,
    shape = 'circle',
    type = 'default',
    icon = <FileTextOutlined />,
    closeIcon = <CloseOutlined />,
    description,
    trigger,
    children,
    clickOutAutoClose,
    onOpenChange,
  } = props;

  const { direction, getPrefixCls } = useContext<ConfigConsumerProps>(ConfigContext);
  const prefixCls = getPrefixCls(floatButtonPrefixCls, customizePrefixCls);
  const [wrapSSR, hashId] = useStyle(prefixCls);
  const groupPrefixCls = `${prefixCls}-group`;

  const groupCls = classNames(groupPrefixCls, hashId, className, {
    [`${groupPrefixCls}-rtl`]: direction === 'rtl',
    [`${groupPrefixCls}-${shape}`]: shape,
    [`${groupPrefixCls}-${shape}-shadow`]: !trigger,
  });

  const wrapperCls = classNames(hashId, `${groupPrefixCls}-wrap`);

  const [open, setOpen] = useMergedState(false, { value: props.open });

  const FloatButtonEL = useRef(null);

  const clickAction = useRef<React.HTMLAttributes<HTMLAnchorElement | HTMLButtonElement>>({});

  const hoverAction = useRef<React.HTMLAttributes<HTMLDivElement>>({});

  if (trigger === 'click') {
    if (clickOutAutoClose) {
      useEffect(() => {
        const BigestEl = document;
        const clickFn = (e: MouseEvent) => {
          let clickTarget = e.target as Node;
          while (clickTarget) {
            if (clickTarget == FloatButtonEL.current) {
              setOpen((prevState) => {
                onOpenChange?.(!prevState);
                return !prevState;
              });
              return;
            }
            clickTarget = clickTarget.parentNode!;
          }
          setOpen(false);
        };
        BigestEl?.addEventListener('click', clickFn);
        //  非严格模式下 会在组件卸载时自动 remove
        // return (
        //   BigestEl?.removeEventListener('click', clickFn)
        // )
      }, []);
    } else {
      clickAction.current = {
        onClick(e) {
          setOpen((prevState) => {
            onOpenChange?.(!prevState);
            return !prevState;
          });
        },
      };
    }
  }

  if (trigger === 'hover') {
    hoverAction.current = {
      onMouseEnter() {
        setOpen(true);
        onOpenChange?.(true);
      },
      onMouseLeave() {
        setOpen(false);
        onOpenChange?.(false);
      },
    };
  }

  return wrapSSR(
    <FloatButtonGroupProvider value={shape}>
      <div className={groupCls} style={style} {...hoverAction.current}>
        {trigger && ['click', 'hover'].includes(trigger) ? (
          <>
            <CSSMotion visible={open} motionName={`${groupPrefixCls}-wrap`}>
              {({ className: motionClassName }) => (
                <div className={classNames(motionClassName, wrapperCls)}>{children}</div>
              )}
            </CSSMotion>
            <FloatButton
              ref={FloatButtonEL}
              type={type}
              shape={shape}
              icon={open ? closeIcon : icon}
              description={description}
              {...clickAction.current}
            />
          </>
        ) : (
          children
        )}
      </div>
    </FloatButtonGroupProvider>,
  );
};

export default memo(FloatButtonGroup);