fix: dialog reappear after closed

react-component · Dec 24, 2022 · 39b7801 · 39b7801
1 parent 56a414f
commit 39b7801
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/Dialog/Content/index.tsx b/src/Dialog/Content/index.tsx
@@ -37,7 +37,8 @@ const Content = React.forwardRef<ContentRef, ContentProps>((props, ref) => {
     contentStyle.transformOrigin = transformOrigin;
   }
 
-  function onPrepare() {
+  function onPrepare(dialog) {
+    dialog.style.display = null;
     const elementOffset = offset(dialogRef.current);
 
     setTransformOrigin(
@@ -47,6 +48,10 @@ const Content = React.forwardRef<ContentRef, ContentProps>((props, ref) => {
     );
   }
 
+  function onLeave(dialog) {
+    dialog.style.display = 'none';
+  }
+
   // ============================= Render =============================
   return (
     <CSSMotion
@@ -57,6 +62,7 @@ const Content = React.forwardRef<ContentRef, ContentProps>((props, ref) => {
       forceRender={forceRender}
       motionName={motionName}
       removeOnLeave={destroyOnClose}
+      onLeaveEnd={onLeave}
       ref={dialogRef}
     >
       {({ className: motionClassName, style: motionStyle }, motionRef) => (