fix: focusTriggerAfterClose not working

close #280
ant-design/ant-design#32923

Author: afc163

src/Dialog/index.tsx

@@ -53,12 +53,9 @@ export default function Dialog(props: IDialogPropTypes) {
 
   // ========================= Events =========================
   function onDialogVisibleChanged(newVisible: boolean) {
+    // Try to focus
     if (newVisible) {
-      // Try to focus
-      if (!contains(wrapperRef.current, document.activeElement)) {
-        lastOutSideActiveElementRef.current = document.activeElement as HTMLElement;
-        contentRef.current?.focus();
-      }
+      contentRef.current?.focus();
     } else {
       // Clean up scroll bar & focus back
       setAnimatedVisible(false);
@@ -127,10 +124,17 @@ export default function Dialog(props: IDialogPropTypes) {
     }
   }
 
+  function saveLastOutSideActiveElementRef() {
+    if (!contains(wrapperRef.current, document.activeElement)) {
+      lastOutSideActiveElementRef.current = document.activeElement as HTMLElement;
+    }
+  }
+
   // ========================= Effect =========================
   useEffect(() => {
     if (visible) {
       setAnimatedVisible(true);
+      saveLastOutSideActiveElementRef();
     }
   }, [visible]);