feat: add afterOpen prop

Author: Amour1688

package.json

@@ -53,6 +53,7 @@
     "rc-util": "^5.6.1"
   },
   "devDependencies": {
+    "@types/classnames": "^2.2.11",
     "@types/enzyme": "^3.10.7",
     "@types/jest": "^26.0.14",
     "@types/react": "^16.9.2",

src/Dialog/index.tsx

@@ -32,6 +32,7 @@ export default function Dialog(props: IDialogChildProps) {
     wrapClassName,
     wrapProps,
     onClose,
+    afterOpen,
     afterClose,
 
     // Dialog
@@ -68,6 +69,7 @@ export default function Dialog(props: IDialogChildProps) {
         lastOutSideActiveElementRef.current = document.activeElement as HTMLElement;
         contentRef.current?.focus();
       }
+      afterOpen?.();
     } else {
       // Clean up scroll bar & focus back
       setAnimatedVisible(false);

src/IDialogPropTypes.tsx

@@ -8,6 +8,7 @@ export type IDialogPropTypes = {
   mask?: boolean;
   children?: any;
   afterClose?: () => any;
+  afterOpen?: () => any;
   onClose?: (e: SyntheticEvent) => any;
   closable?: boolean;
   maskClosable?: boolean;
@@ -41,4 +42,4 @@ export type IDialogPropTypes = {
   // https://github.com/ant-design/ant-design/issues/19771
   // https://github.com/react-component/dialog/issues/95
   focusTriggerAfterClose?: boolean;
-}
+};

tests/index.spec.tsx

@@ -385,6 +385,20 @@ describe('dialog', () => {
     });
   });
 
+  describe('afterOpen', () => {
+    it('should trigger afterOpen when set visible to true', () => {
+      const afterOpen = jest.fn();
+
+      const wrapper = mount(<Dialog afterOpen={afterOpen} visible={false} />);
+      jest.runAllTimers();
+
+      wrapper.setProps({ visible: true });
+      jest.runAllTimers();
+
+      expect(afterOpen).toHaveBeenCalledTimes(1);
+    });
+  });
+
   describe('afterClose', () => {
     it('should trigger afterClose when set visible to false', () => {
       const afterClose = jest.fn();