fix(Attachments): fix Attachments onRemove no work (#555)

* fix(Attachments): fix Attachments onRemove no work

* fix(Attachments): fix Attachments onRemove no work

* Update index.test.tsx

修复单测

---------

Co-authored-by: jin.gao <jin.gao@mobilemd.cn>
Co-authored-by: afc163 <afc163@gmail.com>

Author: 3 people

components/attachments/__tests__/index.test.tsx

@@ -53,6 +53,30 @@ describe('attachments', () => {
     expect(onChange.mock.calls[0][0].fileList).toHaveLength(0);
   });
 
+  it('add and remove file but can stop remove', async () => {
+    const onChange = jest.fn();
+    const { container } = render(
+      renderAttachments({
+        onChange,
+        onRemove: () => false,
+      }),
+    );
+
+    // Add file
+    fireEvent.change(container.querySelector('input')!, {
+      target: { files: [{ file: 'foo.png' }] },
+    });
+    await waitFakeTimer();
+    expect(onChange.mock.calls[0][0].fileList).toHaveLength(1);
+    onChange.mockClear();
+
+    // Remove file
+    fireEvent.click(container.querySelector('.ant-attachment-list-card-remove')!);
+    await waitFakeTimer();
+    expect(container.querySelectorAll('.ant-attachment-list-card')).toHaveLength(1);
+    expect(onChange).not.toHaveBeenCalled();
+  });
+
   it('overflow: scrollX', () => {
     const { container } = render(
       renderAttachments({

components/attachments/index.tsx

@@ -65,6 +65,7 @@ function Attachments(props: AttachmentsProps, ref: React.Ref<AttachmentsRef>) {
     getDropContainer,
     placeholder,
     onChange,
+    onRemove,
     overflow,
     disabled,
     classNames = {},
@@ -124,14 +125,20 @@ function Attachments(props: AttachmentsProps, ref: React.Ref<AttachmentsRef>) {
     onChange: triggerChange,
   };
 
-  const onItemRemove = (item: Attachment) => {
-    const newFileList = fileList.filter((fileItem) => fileItem.uid !== item.uid);
-    triggerChange({
-      file: item,
-      fileList: newFileList,
-    });
-  };
+  const onItemRemove = (item: Attachment) =>
+    Promise.resolve(typeof onRemove === 'function' ? onRemove(item) : onRemove).then((ret) => {
+      // Prevent removing file
+      if (ret === false) {
+        return;
+      }
 
+      const newFileList = fileList.filter((fileItem) => fileItem.uid !== item.uid);
+
+      triggerChange({
+        file: { ...item, status: 'removed' },
+        fileList: newFileList,
+      });
+    });
   // ============================ Render ============================
   let renderChildren: React.ReactElement;