比ant更豐富Modal組件功能實(shí)現(xiàn)示例詳解

更新時(shí)間：2022年11月29日 09:37:35 作者：孟祥_(kāi)成都

這篇文章主要為大家介紹了比ant更豐富Modal組件功能實(shí)現(xiàn)示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

有哪些比ant更豐富的功能

普通的modal組件如下：

我們寫(xiě)的modal額外支持，后面沒(méi)有蒙版，并且Modal框能夠拖拽

還支持渲染在文檔流里，上面的都是fixed布局，我們這個(gè)正常渲染到文檔下面：

render部分

    <RenderDialog
      {...restState}
      visible={visible}
      prefixCls={prefixCls}
      header={renderHeader}
      attach={attach}
      closeBtn={renderCloseIcon()}
      classPrefix={classPrefix}
      onClose={onClose}
      onConfirm={onConfirm}
      footer={footer === true ? defaultFooter() : footer}
      ref={dialogDom}
    />

大家記住這個(gè)RenderDialog，接下來(lái)都是上面?zhèn)鲄⒌慕忉專(zhuān)?/p>

resetState: 是對(duì)象，一堆屬性的集合，哪些屬性呢，我們往下看

// 其實(shí)默認(rèn)參數(shù)寫(xiě)到這里并不科學(xué)，因?yàn)閞eact有個(gè)靜態(tài)屬性defaultProps屬性支持合并props
  const [state, setState] = useSetState<DialogProps>({
    width: 520, // 默認(rèn)寬度是520
    visible: false, // 默認(rèn)visible是false
    zIndex: 2500, // 默認(rèn)zIndex 2500
    placement: 'center', // 默認(rèn)渲染到屏幕中間
    mode: 'modal', // 默認(rèn)的模式是modal是ant那種渲染結(jié)果，其他模式我們下面談
    showOverlay: true, // 是否展示透明黑色蒙版
    destroyOnClose: false, // 關(guān)閉彈窗的時(shí)候是否銷(xiāo)毀里面的內(nèi)容
    draggable: false, // 是否能拖拽modal
    preventScrollThrough: true, // 防止?jié)L動(dòng)穿透
    ...props,
  });

restState在下面，除了state上某些屬性。

 const {
   visible, // 控制對(duì)話框是否顯示
   attach, // 對(duì)話框掛載的節(jié)點(diǎn)，默認(rèn)掛在組件本身的位置。數(shù)據(jù)類(lèi)型為 String 時(shí)，會(huì)被當(dāng)作選擇器處理，進(jìn)行節(jié)點(diǎn)查詢(xún)。示例：'body' 或 () => document.body
   closeBtn, // 關(guān)閉按鈕，可以自定義。值為 true 顯示默認(rèn)關(guān)閉按鈕，值為 false 不顯示關(guān)閉按鈕。值類(lèi)型為 string 則直接顯示值，// 底部操作欄，默認(rèn)會(huì)有“確認(rèn)”和“取消”兩個(gè)按鈕。值為 true 顯示默認(rèn)操作按鈕，值為 false 不顯示任何內(nèi)容，值類(lèi)型為 Function 表示自定義底部?jī)?nèi)容
   footer = true, // 如果“取消”按鈕存在，則點(diǎn)擊“取消”按鈕時(shí)觸發(fā)，同時(shí)觸發(fā)關(guān)閉事件
   onCancel = noop, // 如果“確認(rèn)”按鈕存在，則點(diǎn)擊“確認(rèn)”按鈕時(shí)觸發(fā)，或者鍵盤(pán)按下回車(chē)鍵時(shí)觸發(fā)
   onConfirm = noop, // 如果“確認(rèn)”按鈕存在，則點(diǎn)擊“確認(rèn)”按鈕時(shí)觸發(fā)，或者鍵盤(pán)按下回車(chē)鍵時(shí)觸發(fā)
   cancelBtn = cancelText, // 取消按鈕，可自定義。值為 null 則不顯示取消按鈕。值類(lèi)型為字符串，則表示自定義按鈕文本，值類(lèi)型為 Object 則表示透?jìng)?Button 組件屬性。
   confirmBtn = confirmText, // 確認(rèn)按鈕。值為 null 則不顯示確認(rèn)按鈕。值類(lèi)型為字符串，則表示自定義按鈕文本，值類(lèi)型為 Object 則表示透?jìng)?Button 組件屬性。
   onClose = noop, // 關(guān)閉事件，點(diǎn)擊取消按鈕、點(diǎn)擊關(guān)閉按鈕、點(diǎn)擊蒙層、按下 ESC 等場(chǎng)景下觸發(fā)
   ...restState
 } = state;

說(shuō)了這么多，我們接著看RenderDialog組件上傳入的屬性。

prefixCls不講了，是css屬性前綴，一個(gè)字符串，接著看header屬性被包裝為renderHeader

const renderHeader = useMemo(() => {
    if (!state.header) return null;
    const iconMap = {
      info: <InfoCircleFilledIcon className={`${classPrefix}-is-info`} />,
      warning: <InfoCircleFilledIcon className={`${classPrefix}-is-warning`} />,
      error: <InfoCircleFilledIcon className={`${classPrefix}-is-error`} />,
      success: <CheckCircleFilledIcon className={`${classPrefix}-is-success`} />,
    };
    return (
      <div className={`${prefixCls}__header-content`}>
        {iconMap[state.theme]}
        {state.header}
      </div>
    );
    // eslint-disable-next-line react-hooks/exhaustive-deps
  }, [state.header, state.theme, prefixCls, classPrefix]);

其實(shí)就是在header的文字前面多了一個(gè)icon，比如成功的彈窗如下：

接著看closeBtn屬性

  const renderCloseIcon = () => {
    if (closeBtn === false) return null;
    if (closeBtn === true) return <CloseIcon style={{ verticalAlign: 'unset' }} />;
    return closeBtn || <CloseIcon style={{ verticalAlign: 'unset' }} />;
  };

這個(gè)是右上角關(guān)閉按鈕的Icon，很簡(jiǎn)單，如果是false，什么都不許安然，如果是undefined或者true渲染這個(gè)icon。

好了，我們把整個(gè)代碼放到下面，有代碼注釋?zhuān)瑳](méi)寫(xiě)注釋的是上面咋們已經(jīng)講過(guò)的內(nèi)容，接著就要進(jìn)入RenderDialog這個(gè)組件內(nèi)部了。

import 的部分省略了
// 渲染 footer的button方法
const renderDialogButton = (btn: TdDialogProps['cancelBtn'], defaultProps: ButtonProps) => {
  let result = null;
  if (isString(btn)) {
    result = <Button {...defaultProps}>{btn}</Button>;
  }  else if (isFunction(btn)) {
    result = btn();
  }
  return result;
};
const Dialog = forwardRef((props: DialogProps, ref: React.Ref<DialogInstance>) => {
  // 這部分忽略就好，用來(lái)獲取全局配置的css前綴字符串
  const { classPrefix } = useConfig();
  // 這個(gè)也忽略，獲取icon組件的
  const { CloseIcon, InfoCircleFilledIcon, CheckCircleFilledIcon } = useGlobalIcon({
    CloseIcon: TdCloseIcon,
    InfoCircleFilledIcon: TdInfoCircleFilledIcon,
    CheckCircleFilledIcon: TdCheckCircleFilledIcon,
  });
  // 用來(lái)引用dialog彈框的dom
  const dialogDom = useRef<HTMLDivElement>();
  const [state, setState] = useSetState<DialogProps>({
    width: 520,
    visible: false,
    zIndex: 2500,
    placement: 'center',
    mode: 'modal',
    showOverlay: true,
    destroyOnClose: false,
    draggable: false,
    preventScrollThrough: true,
    ...props,
  });
 // 國(guó)際化有關(guān)的
  const [local, t] = useLocaleReceiver('dialog');
  const confirmText = t(local.confirm);
  const cancelText = t(local.cancel);
  const {
    visible,
    attach,
    closeBtn,
    footer = true,
    onCancel = noop,
    onConfirm = noop,
    cancelBtn = cancelText,
    confirmBtn = confirmText,
    onClose = noop,
    ...restState
  } = state;
  useEffect(() => {
     setState((prevState) => ({
        ...prevState,
        ...props,
      }));
  }, [props, setState, isPlugin]);
  const prefixCls = `${classPrefix}-dialog`;
  const renderCloseIcon = () => {
    if (closeBtn === false) return null;
    if (closeBtn === true) return <CloseIcon style={{ verticalAlign: 'unset' }} />;
    return closeBtn || <CloseIcon style={{ verticalAlign: 'unset' }} />;
  };
 // 這里把一些外部方法暴露給調(diào)用者，只需要傳入ref就可以獲取
  React.useImperativeHandle(ref, () => ({
    show() {
      setState({ visible: true });
    },
    hide() {
      setState({ visible: false });
    },
    destroy() {
      setState({ visible: false, destroyOnClose: true });
    },
    update(newOptions) {
      setState((prevState) => ({
        ...prevState,
        ...(newOptions as DialogProps),
      }));
    },
  }));
  const renderHeader = useMemo(() => {
    if (!state.header) return null;
    const iconMap = {
      info: <InfoCircleFilledIcon className={`${classPrefix}-is-info`} />,
      warning: <InfoCircleFilledIcon className={`${classPrefix}-is-warning`} />,
      error: <InfoCircleFilledIcon className={`${classPrefix}-is-error`} />,
      success: <CheckCircleFilledIcon className={`${classPrefix}-is-success`} />,
    };
    return (
      <div className={`${prefixCls}__header-content`}>
        {iconMap[state.theme]}
        {state.header}
      </div>
    );
    // eslint-disable-next-line react-hooks/exhaustive-deps
  }, [state.header, state.theme, prefixCls, classPrefix]);
// 渲染footer的時(shí)候，點(diǎn)擊取消按鈕會(huì)用到
  const handleCancel = (e: React.MouseEvent<HTMLButtonElement>) => {
    onCancel({ e });
    onClose({ e, trigger: 'cancel' });
  };
// 渲染footer的時(shí)候，點(diǎn)擊確認(rèn)按鈕會(huì)用到
  const handleConfirm = (e: React.MouseEvent<HTMLButtonElement>) => {
    onConfirm({ e });
  };
  const defaultFooter = () => {
    const renderCancelBtn = renderDialogButton(cancelBtn, { variant: 'outline' });
    const renderConfirmBtn = renderDialogButton(confirmBtn, { theme: 'primary' });
    return (
      <>
        {renderCancelBtn &&
          React.cloneElement(renderCancelBtn, {
            onClick: handleCancel,
            ...renderCancelBtn.props,
          })}
        {renderConfirmBtn &&
          React.cloneElement(renderConfirmBtn, {
            onClick: handleConfirm,
            ...renderConfirmBtn.props,
          })}
      </>
    );
  };
  return (
    <RenderDialog
      {...restState}
      visible={visible}
      prefixCls={prefixCls}
      header={renderHeader}
      attach={attach}
      closeBtn={renderCloseIcon()}
      classPrefix={classPrefix}
      onClose={onClose}
      onConfirm={onConfirm}
      footer={footer === true ? defaultFooter() : footer}
      ref={dialogDom}
    />
  );
});
Dialog.displayName = 'Dialog';
Dialog.defaultProps = dialogDefaultProps;
export default Dialog;

接著，我們要渲染的部分其實(shí)很簡(jiǎn)單，包括

背后的黑色蒙層
彈框
- 彈框的標(biāo)題
- 彈框的內(nèi)容區(qū)域
- 彈框的footer
還需要彈框動(dòng)畫(huà)，比如zoom或者fade

渲染黑色蒙層

代碼如下，很簡(jiǎn)單

  const renderMask = () => {
    let maskElement;
    if (showOverlay) {
      maskElement = (
        <CSSTransition
          in={visible}
          appear
          timeout={transitionTime}
          classNames={`${prefixCls}-fade`}
          mountOnEnter
          unmountOnExit
          nodeRef={maskRef}
        >
          <div ref={maskRef} className={`${prefixCls}__mask`} />
        </CSSTransition>
      );
    }
    return maskElement;
  };

首先介紹一下CSSTransition，這是react-transition-group動(dòng)畫(huà)庫(kù)的一個(gè)組件，用來(lái)幫助我們實(shí)現(xiàn)css動(dòng)畫(huà)的。其中一些屬性說(shuō)明如下：

in： ture就是開(kāi)始動(dòng)畫(huà)，false就是停止動(dòng)畫(huà)
appear：boolean，為 false 時(shí)當(dāng) CSSTransition 控件加載完畢后不執(zhí)行動(dòng)畫(huà)，為 true 時(shí)控件加載完畢則立即執(zhí)行動(dòng)畫(huà)。如果要組件初次渲染就有動(dòng)畫(huà)，則需要設(shè)成 true。
timeout 動(dòng)畫(huà)時(shí)間
classNames：動(dòng)畫(huà)的類(lèi)名，比如classNames:'demo'，會(huì)自動(dòng)在進(jìn)入動(dòng)畫(huà)的時(shí)候幫你把類(lèi)名改為 demo-enter-active, demo-enter-done, 在退出動(dòng)畫(huà)同樣會(huì)有類(lèi)名的改變。
mountOnEnter：一進(jìn)來(lái)的時(shí)候不顯示dom元素
unmountOnExit：boolean，為 true 時(shí)組件將移除處于隱藏狀態(tài)的元素，為 false 時(shí)組件保持動(dòng)畫(huà)結(jié)束時(shí)的狀態(tài)而不移除元素。一般要設(shè)成 true。
nodeRef，獲取蒙層的ref

蒙層主要靠css實(shí)現(xiàn)，我們看下css

  position: fixed;
  top: 0;
  left: 0;
  width: 100%;
  height: 100%;
  z-index: 1;
  background: var(--td-mask-active);
  pointer-events: auto;

渲染彈框主體

也非常簡(jiǎn)單啊，我們把注釋寫(xiě)在下面的代碼里了，其中有一個(gè)需要小小注意的功能就是拖拽功能

// 渲染Dialog主體
  const renderDialog = () => {
    const dest: any = {};
    // 把width變?yōu)橛衟x結(jié)尾的字符串
    if (props.width !== undefined) {
      dest.width = GetCSSValue(props.width);
    }
    // normal 場(chǎng)景下，需要設(shè)置 zindex 為auto 避免出現(xiàn)多個(gè) dialog，normal 出現(xiàn)在最上層
    if (props.mode === 'normal') {
      dest.zIndex = 'auto';
    }
    // 獲取footer
    const footer = props.footer ? <div className={`${prefixCls}__footer`}>{props.footer}</div> : null;
    // 獲取header
    const { header } = props;
    // 獲取Dialog body
    const body = <div className={`${prefixCls}__body`}>{props.body || props.children}</div>;
    // 關(guān)閉按鈕，可以自定義。值為 true 顯示默認(rèn)關(guān)閉按鈕，值為 false 不顯示關(guān)閉按鈕。值類(lèi)型為 string 則直接顯示值，如：“關(guān)閉”。
    const closer = closeBtn && (
      <span onClick={handleCloseBtnClick} className={`${prefixCls}__close`}>
        {closeBtn}
      </span>
    );
    const validWindow = typeof window === 'object';
    // 獲取屏幕高度
    const screenHeight = validWindow ? window.innerHeight || document.documentElement.clientHeight : undefined;
    // 獲取屏幕寬度
    const screenWidth = validWindow ? window.innerWidth || document.documentElement.clientWidth : undefined;
    // 設(shè)置style
    const style = { ...dest, ...props.style };
    let dialogOffset = { x: 0, y: 0 };
    // 拖拽代碼實(shí)現(xiàn)部分
    const onDialogMove = (e: MouseEvent) => {
      // offsetWidth是指元素的寬 + padding + border的總和
      const { style, offsetWidth, offsetHeight } = dialog.current;
      // diffX是指彈框部分距離body左邊部分
      let diffX = e.clientX - dialogOffset.x;
      let diffY = e.clientY - dialogOffset.y;
      // 拖拽上左邊界限制
      if (diffX < 0) diffX = 0;
      if (diffY < 0) diffY = 0;
      // 右邊的限制
      if (screenWidth - offsetWidth - diffX < 0) diffX = screenWidth - offsetWidth;
      // 下邊的限制
      if (screenHeight - offsetHeight - diffY < 0) diffY = screenHeight - offsetHeight;
      style.position = 'absolute';
      style.left = `${diffX}px`;
      style.top = `${diffY}px`;
    };
    const onDialogMoveEnd = () => {
      // 恢復(fù)指針樣式為默認(rèn)，并且注銷(xiāo)mousemove， mouseup事件
      dialog.current.style.cursor = 'default';
      document.removeEventListener('mousemove', onDialogMove);
      document.removeEventListener('mouseup', onDialogMoveEnd);
    };
    // 拖拽開(kāi)始，對(duì)應(yīng)mouseDown事件
    const onDialogMoveStart = (e: React.MouseEvent<HTMLDivElement>) => {
      contentClickRef.current = true;
      // 阻止事件冒泡， mode === 'modeless才能拖拽
      if (canDraggable && e.currentTarget === e.target) {
        const { offsetLeft, offsetTop, offsetHeight, offsetWidth } = dialog.current;
        // 如果彈出框超出屏幕范圍 不能進(jìn)行拖拽
        if (offsetWidth > screenWidth || offsetHeight > screenHeight) return;
        // 拖拽樣式設(shè)置為move
        dialog.current.style.cursor = 'move';
        // 計(jì)算鼠標(biāo) e.clientX是鼠標(biāo)在屏幕的坐標(biāo)，offsetLeft是Dialog主體跟body的距離
        // 所以e.clientX - offsetLeft就是鼠標(biāo)在是Dialog主體上的橫坐標(biāo)
        const diffX = e.clientX - offsetLeft;
        const diffY = e.clientY - offsetTop;
        dialogOffset = {
          x: diffX,
          y: diffY,
        };
        // 此時(shí)把mousemove和mouseup事件也綁定一下，其實(shí)不建議綁定在這里直接操作dom
        document.addEventListener('mousemove', onDialogMove);
        document.addEventListener('mouseup', onDialogMoveEnd);
      }
    };
    // 頂部定位實(shí)現(xiàn)
    const positionStyle: any = {};
    if (props.top) {
      const topValue = GetCSSValue(props.top);
      positionStyle.paddingTop = topValue;
    }
    // 此處獲取定位方式 top 優(yōu)先級(jí)較高 存在時(shí) 默認(rèn)使用 top 定位
    const positionClass = classnames(
      `${prefixCls}__position`,
      { [`${prefixCls}--top`]: !!props.top },
      `${props.placement && !props.top ? `${prefixCls}--${props.placement}` : ''}`,
    );
    // 然后就是用css去渲染header body和footer
    const dialogElement = (
      <div className={isNormal ? '' : `${prefixCls}__wrap`}>
        <div className={isNormal ? '' : positionClass} style={positionStyle} onClick={onMaskClick} ref={dialogPosition}>
          <div
            ref={dialog}
            style={style}
            className={classnames(`${prefixCls}`, `${prefixCls}--default`)}
            onMouseDown={onDialogMoveStart}
          >
            <div className={classnames(`${prefixCls}__header`)}>
              {header}
              {closer}
            </div>
            {body}
            {footer}
          </div>
        </div>
      </div>
    );
    return (
      <CSSTransition
        in={props.visible}
        appear
        mountOnEnter
        unmountOnExit={destroyOnClose}
        timeout={transitionTime}
        classNames={`${prefixCls}-zoom`}
        onEntered={props.onOpened}
        onExited={onAnimateLeave}
        nodeRef={dialog}
      >
        {dialogElement}
      </CSSTransition>
    );
  };

我們這里貼一下css部分：

header：

.t-dialog__header {
    color: var(--td-text-color-primary);
    font: var(--td-font-title-medium);
    font-weight: 600;
    display: flex;
    align-items: flex-start;
    word-break: break-word;
}

這里注意下：word-wrap:break-word

它會(huì)把整個(gè)單詞看成一個(gè)整體，如果該行末端寬度不夠顯示整個(gè)單詞，它會(huì)自動(dòng)把整個(gè)單詞放到下一行，而不會(huì)把單詞截?cái)嗟舻摹?/p>

body

.t-dialog__body {
  padding: 16px 0;
  color: var(--td-text-color-secondary);
  font: var(--td-font-body-medium);
  overflow: auto;
  word-break: break-word;
}

footer

  width: 100%;
  text-align: right;
  padding: 16px 0 0 0;

好了，我們結(jié)合一下彈框和蒙層，看下render函數(shù)

const render = () => {
    // 。。。省略css部分
    // 如果不是 modal 模式 默認(rèn)沒(méi)有 mask 也就沒(méi)有相關(guān)點(diǎn)擊 mask 事件
    const dialog = (
      <div ref={wrap} className={wrapClass} style={wrapStyle} onKeyDown={handleKeyDown} tabIndex={0}>
        {mode === 'modal' && renderMask()}
        {dialogBody} // 這里就是我們上面講的renderDialog
      </div>
    );
    return dialog;
  };

設(shè)置body overflow：hiiden

為啥要設(shè)置body overflow：hiiden這個(gè)屬性呢，你打開(kāi)modal彈窗的時(shí)候，如果此時(shí)body還有滾動(dòng)條，那么你滾動(dòng)鼠標(biāo)滾輪還可以向下滑動(dòng)，但是一般情況下，我們打開(kāi)彈框，是希望用戶(hù)目標(biāo)鎖定在當(dāng)前交互，此時(shí)最好不要允許用戶(hù)滾動(dòng)界面。

當(dāng)然你也可以允許用戶(hù)滾動(dòng)，我們用一個(gè)preventScrollThrough參數(shù)控制。

先記住當(dāng)前body的css樣式，以及body的overflow的值，代碼如下

  useLayoutEffect(() => {
    bodyOverflow.current = document.body.style.overflow;
    bodyCssTextRef.current = document.body.style.cssText;
  }, []);

 const isModal = mode === 'modal';
useLayoutEffect(() => {
      // 只有modal數(shù)量小于1的時(shí)候才重置樣式，因?yàn)榭赡艹霈F(xiàn)多個(gè)彈框，那么關(guān)閉一個(gè)彈框就出現(xiàn)滾動(dòng)條明顯不對(duì)
      if (isModal) {
      const openDialogDom = document.querySelectorAll(`${prefixCls}__mode`);
      if (openDialogDom.length < 1) {
        document.body.style.cssText = bodyCssTextRef.current;
      }
    // 組件銷(xiāo)毀后重置 body 樣式
    return () => {
      if (isModal) {
        // 此處只能查詢(xún) mode 模式的 dialog 個(gè)數(shù) 因?yàn)?modeless 會(huì)點(diǎn)擊透?jìng)?normal 是正常文檔流
        const openDialogDom = document.querySelectorAll(`${prefixCls}__mode`);
        if (openDialogDom.length < 1) {
          document.body.style.cssText = bodyCssTextRef.current;
          document.body.style.overflow = bodyOverflow.current;
        }
      } 
    };
  }, [preventScrollThrough, attach, visible, mode, isModal, showInAttachedElement, prefixCls]);

上面的代碼還有一個(gè)問(wèn)題，就是我們需要preventScrollThrough這個(gè)參數(shù)去控制是否可以body滾動(dòng)頁(yè)面，這個(gè)也是算比ant更豐富的功能。

const isModal = mode === 'modal';
 useLayoutEffect(() => {
    // 處于顯示態(tài)
    if (visible) {
      // isModal表示是否是普通彈框，就是帶黑色蒙層的
      // bodyOverflow.current 引用的是body的overflow屬性
      // preventScrollThrough是代表是否可以滾動(dòng)body
      // ！showInAttachedElement表示不掛載到其他dom上
      if (isModal && bodyOverflow.current !== 'hidden' && preventScrollThrough && !showInAttachedElement) {
        // 求出滾動(dòng)條的寬度
        const scrollWidth = window.innerWidth - document.body.offsetWidth;
        // 減少回流
        if (bodyCssTextRef.current === '') {
          let bodyCssText = 'overflow: hidden;';
          if (scrollWidth > 0) {
            bodyCssText += `position: relative;width: calc(100% - ${scrollWidth}px);`;
          }
          document.body.style.cssText = bodyCssText;
        } else {
          if (scrollWidth > 0) {
            document.body.style.width = `calc(100% - ${scrollWidth}px)`;
            document.body.style.position = 'relative';
          }
          document.body.style.overflow = 'hidden';
        }
      }
      // 剛進(jìn)頁(yè)面就focus到彈框組件上
      if (wrap.current) {
        wrap.current.focus();
      }
    } else if (isModal) {
      const openDialogDom = document.querySelectorAll(`${prefixCls}__mode`);
      if (openDialogDom.length < 1) {
        document.body.style.cssText = bodyCssTextRef.current;
      }
    }
    // 組件銷(xiāo)毀后重置 body 樣式
    return () => {
      if (isModal) {
        // 此處只能查詢(xún) mode 模式的 dialog 個(gè)數(shù) 因?yàn)?modeless 會(huì)點(diǎn)擊透?jìng)?normal 是正常文檔流
        const openDialogDom = document.querySelectorAll(`${prefixCls}__mode`);
        if (openDialogDom.length < 1) {
          document.body.style.cssText = bodyCssTextRef.current;
          document.body.style.overflow = bodyOverflow.current;
        }
      } else {
        document.body.style.cssText = bodyCssTextRef.current;
        document.body.style.overflow = bodyOverflow.current;
      }
    };
  }, [preventScrollThrough, attach, visible, mode, isModal, showInAttachedElement, prefixCls]);

其實(shí)還有一個(gè)邏輯，是把彈窗渲染到任意dom里，需要一個(gè)Portal組件，我們這里就不說(shuō)了，后續(xù)將Popup或者叫trigger組件的時(shí)候我們講吧。一篇文檔內(nèi)容太多不好消化。

好了，主邏輯已經(jīng)寫(xiě)完了，很簡(jiǎn)單吧！

接下來(lái)看下完整代碼，沒(méi)有注釋的部分是上面已經(jīng)講過(guò)的

//省去了import
// 把css的數(shù)字轉(zhuǎn)為有px結(jié)尾的字符串，，這里其實(shí)應(yīng)該寫(xiě)到一個(gè)utils文件夾里，不應(yīng)該跟主代碼混在一起
function GetCSSValue(v: string | number) {
  return Number.isNaN(Number(v)) ? v : `${Number(v)}px`;
}
// 動(dòng)畫(huà)執(zhí)行時(shí)間，這里其實(shí)應(yīng)該寫(xiě)到一個(gè)constants文件里，不應(yīng)該跟主代碼混在一起
const transitionTime = 300;
const RenderDialog = forwardRef((props: RenderDialogProps, ref: React.Ref<HTMLDivElement>) => {
  // 這里不用看，跟國(guó)際化有關(guān)
  const [local] = useLocaleReceiver('dialog');
  const {
    prefixCls, 
    attach, // 對(duì)話框掛載的節(jié)點(diǎn)，默認(rèn)掛在組件本身的位置。數(shù)據(jù)類(lèi)型為 String 時(shí)，會(huì)被當(dāng)作選擇器處理，進(jìn)行節(jié)點(diǎn)查詢(xún)。示例：'body' 或 () => document.body
    visible, // 控制對(duì)話框是否顯示
    mode, // 對(duì)話框類(lèi)型，有三種：模態(tài)對(duì)話框、非模態(tài)對(duì)話框和普通對(duì)話框。彈出「模態(tài)對(duì)話框」時(shí)，只能操作對(duì)話框里面的內(nèi)容，不能操作其他內(nèi)容。彈出「非模態(tài)對(duì)話框」時(shí)，則可以操作頁(yè)面內(nèi)所有內(nèi)容?！钙胀▽?duì)話框」是指沒(méi)有脫離文檔流的對(duì)話框，可以在這個(gè)基礎(chǔ)上開(kāi)發(fā)更多的插件
    zIndex, // 對(duì)話框?qū)蛹?jí)，Web 側(cè)樣式默認(rèn)為 2500，移動(dòng)端和小程序樣式默認(rèn)為 1500
    showOverlay, // 是否顯示遮罩層
    onEscKeydown = noop,// 按下 ESC 時(shí)觸發(fā)事件
    onClosed = noop, // 對(duì)話框消失動(dòng)畫(huà)效果結(jié)束后觸發(fā)
    onClose = noop, // 關(guān)閉事件，點(diǎn)擊取消按鈕、點(diǎn)擊關(guān)閉按鈕、點(diǎn)擊蒙層、按下 ESC 等場(chǎng)景下觸發(fā)
    onCloseBtnClick = noop, // 點(diǎn)擊右上角關(guān)閉按鈕時(shí)觸發(fā)
    onOverlayClick = noop, // 如果蒙層存在，點(diǎn)擊蒙層時(shí)觸發(fā)
    onConfirm = noop, // 如果“確認(rèn)”按鈕存在，則點(diǎn)擊“確認(rèn)”按鈕時(shí)觸發(fā)，或者鍵盤(pán)按下回車(chē)鍵時(shí)觸發(fā)
    preventScrollThrough, // 防止?jié)L動(dòng)穿透
    closeBtn, // 關(guān)閉按鈕，可以自定義。值為 true 顯示默認(rèn)關(guān)閉按鈕，值為 false 不顯示關(guān)閉按鈕。值類(lèi)型為 string 則直接顯示值，如：“關(guān)閉”。值類(lèi)型為 TNode，則表示呈現(xiàn)自定義按鈕示例
    closeOnEscKeydown, // 按下 ESC 時(shí)是否觸發(fā)對(duì)話框關(guān)閉事件
    confirmOnEnter, // 是否在按下回車(chē)鍵時(shí)，觸發(fā)確認(rèn)事件
    closeOnOverlayClick, // 點(diǎn)擊蒙層時(shí)是否觸發(fā)關(guān)閉事件
    destroyOnClose, // 是否在關(guān)閉彈框的時(shí)候銷(xiāo)毀子元素
    showInAttachedElement, // 僅在掛載元素中顯示抽屜，默認(rèn)在瀏覽器可視區(qū)域顯示。父元素需要有定位屬性，如：position: relative
  } = props;
  const wrap = useRef<HTMLDivElement>(); // 掛載到包裹彈框的dom上，包裹了好幾層。。。
  const dialog = useRef<HTMLDivElement>(); // 引用彈窗dom
  const dialogPosition = useRef<HTMLDivElement>(); // 包裹彈窗，用于定位的dom引用
  const maskRef = useRef<HTMLDivElement>(); // 蒙層的dom引用
  const bodyOverflow = useRef<string>(); 
  const bodyCssTextRef = useRef<string>();
  const contentClickRef = useRef(false);
  const isModal = mode === 'modal';
  const isNormal = mode === 'normal';
  const canDraggable = props.draggable && mode === 'modeless';
  const dialogOpenClass = `${prefixCls}__${mode}`;
  useLayoutEffect(() => {
    bodyOverflow.current = document.body.style.overflow;
    bodyCssTextRef.current = document.body.style.cssText;
  }, []);
  useLayoutEffect(() => {
    if (visible) {
      if (isModal && bodyOverflow.current !== 'hidden' && preventScrollThrough && !showInAttachedElement) {
        const scrollWidth = window.innerWidth - document.body.offsetWidth;
        // 減少回流
        if (bodyCssTextRef.current === '') {
          let bodyCssText = 'overflow: hidden;';
          if (scrollWidth > 0) {
            bodyCssText += `position: relative;width: calc(100% - ${scrollWidth}px);`;
          }
          document.body.style.cssText = bodyCssText;
        } else {
          if (scrollWidth > 0) {
            document.body.style.width = `calc(100% - ${scrollWidth}px)`;
            document.body.style.position = 'relative';
          }
          document.body.style.overflow = 'hidden';
        }
      }
      if (wrap.current) {
        wrap.current.focus();
      }
    } else if (isModal) {
      const openDialogDom = document.querySelectorAll(`${prefixCls}__mode`);
      if (openDialogDom.length < 1) {
        document.body.style.cssText = bodyCssTextRef.current;
      }
    }
    // 組件銷(xiāo)毀后重置 body 樣式
    return () => {
      if (isModal) {
        // 此處只能查詢(xún) mode 模式的 dialog 個(gè)數(shù) 因?yàn)?modeless 會(huì)點(diǎn)擊透?jìng)?normal 是正常文檔流
        const openDialogDom = document.querySelectorAll(`${prefixCls}__mode`);
        if (openDialogDom.length < 1) {
          document.body.style.cssText = bodyCssTextRef.current;
          document.body.style.overflow = bodyOverflow.current;
        }
      } else {
        document.body.style.cssText = bodyCssTextRef.current;
        document.body.style.overflow = bodyOverflow.current;
      }
    };
  }, [preventScrollThrough, attach, visible, mode, isModal, showInAttachedElement, prefixCls]);
  const onAnimateLeave = () => {
    if (wrap.current) {
      wrap.current.style.display = 'none';
    }
    if (isModal && preventScrollThrough) {
      // 還原 body 的滾動(dòng)條
      const openDialogDom = document.querySelectorAll(`${prefixCls}__mode`);
      if (isModal && openDialogDom.length < 1) {
        document.body.style.overflow = bodyOverflow.current;
      }
    }
    if (!isModal) {
      // 關(guān)閉彈窗 清空拖拽設(shè)置的相關(guān) css
      const { style } = dialog.current;
      style.position = 'relative';
      style.left = 'unset';
      style.top = 'unset';
    }
    onClosed && onClosed();
  };
  const onMaskClick = (e: React.MouseEvent<HTMLDivElement>) => {
    if (showOverlay && (closeOnOverlayClick ?? local.closeOnOverlayClick)) {
      // 判斷點(diǎn)擊事件初次點(diǎn)擊是否為內(nèi)容區(qū)域
      if (contentClickRef.current) {
        contentClickRef.current = false;
      } else if (e.target === dialogPosition.current) {
        onOverlayClick({ e });
        onClose({ e, trigger: 'overlay' });
      }
    }
  };
  const handleCloseBtnClick = (e: React.MouseEvent<HTMLDivElement>) => {
    onCloseBtnClick({ e });
    onClose({ e, trigger: 'close-btn' });
  };
  const handleKeyDown = (e: React.KeyboardEvent<HTMLDivElement>) => {
    // https://developer.mozilla.org/en-US/docs/Web/API/KeyboardEvent/keyCode
    if (e.key === 'Escape') {
      e.stopPropagation();
      onEscKeydown({ e });
      if (closeOnEscKeydown ?? local.closeOnEscKeydown) {
        onClose({ e, trigger: 'esc' });
      }
    } else if (e.key === 'Enter' || e.key === 'NumpadEnter') {
      // 回車(chē)鍵觸發(fā)點(diǎn)擊確認(rèn)事件
      e.stopPropagation();
      if (confirmOnEnter) {
        onConfirm({ e });
      }
    }
  };
  // 渲染Dialog主體
  const renderDialog = () => {
    const dest: any = {};
    // 把width變?yōu)橛衟x結(jié)尾的字符串
    if (props.width !== undefined) {
      dest.width = GetCSSValue(props.width);
    }
    // normal 場(chǎng)景下，需要設(shè)置 zindex 為auto 避免出現(xiàn)多個(gè) dialog，normal 出現(xiàn)在最上層
    if (props.mode === 'normal') {
      dest.zIndex = 'auto';
    }
    // 獲取footer
    const footer = props.footer ? <div className={`${prefixCls}__footer`}>{props.footer}</div> : null;
    // 獲取header
    const { header } = props;
    // 獲取Dialog body
    const body = <div className={`${prefixCls}__body`}>{props.body || props.children}</div>;
    // 關(guān)閉按鈕，可以自定義。值為 true 顯示默認(rèn)關(guān)閉按鈕，值為 false 不顯示關(guān)閉按鈕。值類(lèi)型為 string 則直接顯示值，如：“關(guān)閉”。
    const closer = closeBtn && (
      <span onClick={handleCloseBtnClick} className={`${prefixCls}__close`}>
        {closeBtn}
      </span>
    );
    const validWindow = typeof window === 'object';
    // 獲取屏幕高度
    const screenHeight = validWindow ? window.innerHeight || document.documentElement.clientHeight : undefined;
    // 獲取屏幕寬度
    const screenWidth = validWindow ? window.innerWidth || document.documentElement.clientWidth : undefined;
    // 設(shè)置style
    const style = { ...dest, ...props.style };
    let dialogOffset = { x: 0, y: 0 };
    // 拖拽代碼實(shí)現(xiàn)部分
    const onDialogMove = (e: MouseEvent) => {
      // offsetWidth是指元素的寬 + padding + border的總和
      const { style, offsetWidth, offsetHeight } = dialog.current;
      // diffX是指彈框部分距離body左邊部分
      let diffX = e.clientX - dialogOffset.x;
      let diffY = e.clientY - dialogOffset.y;
      // 拖拽上左邊界限制
      if (diffX < 0) diffX = 0;
      if (diffY < 0) diffY = 0;
      // 右邊的限制
      if (screenWidth - offsetWidth - diffX < 0) diffX = screenWidth - offsetWidth;
      // 下邊的限制
      if (screenHeight - offsetHeight - diffY < 0) diffY = screenHeight - offsetHeight;
      style.position = 'absolute';
      style.left = `${diffX}px`;
      style.top = `${diffY}px`;
    };
    const onDialogMoveEnd = () => {
      // 恢復(fù)指針樣式為默認(rèn)，并且注銷(xiāo)mousemove， mouseup事件
      dialog.current.style.cursor = 'default';
      document.removeEventListener('mousemove', onDialogMove);
      document.removeEventListener('mouseup', onDialogMoveEnd);
    };
    // 拖拽開(kāi)始，對(duì)應(yīng)mouseDown事件
    const onDialogMoveStart = (e: React.MouseEvent<HTMLDivElement>) => {
      contentClickRef.current = true;
      // 阻止事件冒泡， mode === 'modeless才能拖拽
      if (canDraggable && e.currentTarget === e.target) {
        const { offsetLeft, offsetTop, offsetHeight, offsetWidth } = dialog.current;
        // 如果彈出框超出屏幕范圍 不能進(jìn)行拖拽
        if (offsetWidth > screenWidth || offsetHeight > screenHeight) return;
        // 拖拽樣式設(shè)置為move
        dialog.current.style.cursor = 'move';
        // 計(jì)算鼠標(biāo) e.clientX是鼠標(biāo)在屏幕的坐標(biāo)，offsetLeft是Dialog主體跟body的距離
        // 所以e.clientX - offsetLeft就是鼠標(biāo)在是Dialog主體上的橫坐標(biāo)
        const diffX = e.clientX - offsetLeft;
        const diffY = e.clientY - offsetTop;
        dialogOffset = {
          x: diffX,
          y: diffY,
        };
        // 此時(shí)把mousemove和mouseup事件也綁定一下，其實(shí)不建議綁定在這里直接操作dom
        document.addEventListener('mousemove', onDialogMove);
        document.addEventListener('mouseup', onDialogMoveEnd);
      }
    };
    // 頂部定位實(shí)現(xiàn)
    const positionStyle: any = {};
    if (props.top) {
      const topValue = GetCSSValue(props.top);
      positionStyle.paddingTop = topValue;
    }
    // 此處獲取定位方式 top 優(yōu)先級(jí)較高 存在時(shí) 默認(rèn)使用 top 定位
    const positionClass = classnames(
      `${prefixCls}__position`,
      { [`${prefixCls}--top`]: !!props.top },
      `${props.placement && !props.top ? `${prefixCls}--${props.placement}` : ''}`,
    );
    const dialogElement = (
      <div className={isNormal ? '' : `${prefixCls}__wrap`}>
        <div className={isNormal ? '' : positionClass} style={positionStyle} onClick={onMaskClick} ref={dialogPosition}>
          <div
            ref={dialog}
            style={style}
            className={classnames(`${prefixCls}`, `${prefixCls}--default`)}
            onMouseDown={onDialogMoveStart}
          >
            <div className={classnames(`${prefixCls}__header`)}>
              {header}
              {closer}
            </div>
            {body}
            {footer}
          </div>
        </div>
      </div>
    );
    return (
      <CSSTransition
        in={props.visible}
        appear
        mountOnEnter
        unmountOnExit={destroyOnClose}
        timeout={transitionTime}
        classNames={`${prefixCls}-zoom`}
        onEntered={props.onOpened}
        onExited={onAnimateLeave}
        nodeRef={dialog}
      >
        {dialogElement}
      </CSSTransition>
    );
  };
  const renderMask = () => {
    let maskElement;
    if (showOverlay) {
      maskElement = (
        <CSSTransition
          in={visible}
          appear
          timeout={transitionTime}
          classNames={`${prefixCls}-fade`}
          mountOnEnter
          unmountOnExit
          nodeRef={maskRef}
        >
          <div ref={maskRef} className={`${prefixCls}__mask`} />
        </CSSTransition>
      );
    }
    return maskElement;
  };
  const render = () => {
    const style: CSSProperties = {};
    if (visible) {
      style.display = 'block';
    }
    const wrapStyle = {
      ...style,
      zIndex,
    };
    const dialogBody = renderDialog();
    const wrapClass = classnames(
      props.className,
      `${prefixCls}__ctx`,
      !isNormal ? `${prefixCls}__ctx--fixed` : '',
      visible ? dialogOpenClass : '',
      isModal && showInAttachedElement ? `${prefixCls}__ctx--absolute` : '',
      props.mode === 'modeless' ? `${prefixCls}__ctx--modeless` : '',
    );
    // 如果不是 modal 模式 默認(rèn)沒(méi)有 mask 也就沒(méi)有相關(guān)點(diǎn)擊 mask 事件
    const dialog = (
      <div ref={wrap} className={wrapClass} style={wrapStyle} onKeyDown={handleKeyDown} tabIndex={0}>
        {mode === 'modal' && renderMask()}
        {dialogBody}
      </div>
    );
    let dom = null;
    if (visible || wrap.current) {
      // normal 模式 attach 無(wú)效
      if (attach === '' || isNormal) {
        dom = dialog;
      } else {
        dom = (
          <CSSTransition
            in={visible}
            appear
            timeout={transitionTime}
            mountOnEnter
            unmountOnExit={destroyOnClose}
            nodeRef={portalRef}
          >
            <Portal attach={attach} ref={portalRef}>
              {dialog}
            </Portal>
          </CSSTransition>
        );
      }
    }
    return dom;
  };
  return render();
});
RenderDialog.defaultProps = dialogDefaultProps;
export default RenderDialog;

結(jié)束，react組件庫(kù)繼續(xù)搞起！

以上就是比ant更豐富Modal組件功能實(shí)現(xiàn)示例詳解的詳細(xì)內(nèi)容，更多關(guān)于ant Modal組件功能的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: