From a0637a8b21e4f2f2fa0e40404fe8c9369cac19ea Mon Sep 17 00:00:00 2001
From: Ali Cherry <32630359+alichry@users.noreply.github.com>
Date: Thu, 1 Apr 2021 12:16:04 +0300
Subject: [PATCH] Modify previewChipProps type in DropzoneAreaBase and
 PreviewList

---
 src/components/DropzoneAreaBase.js | 2 +-
 src/components/PreviewList.js      | 8 ++++++--
 src/index.d.ts                     | 2 +-
 3 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/components/DropzoneAreaBase.js b/src/components/DropzoneAreaBase.js
index c9fe83d0..5297cdaf 100644
--- a/src/components/DropzoneAreaBase.js
+++ b/src/components/DropzoneAreaBase.js
@@ -405,7 +405,7 @@ DropzoneAreaBase.propTypes = {
      *
      * @see See [Material-UI Chip](https://material-ui.com/api/chip/#props) for available values.
      */
-    previewChipProps: PropTypes.object,
+    previewChipProps: PropTypes.oneOfType([PropTypes.object, PropTypes.func]),
     /**
      * Custom CSS classNames for preview Grid components.<br/>
      * Should be in the form {container: string, item: string, image: string}.
diff --git a/src/components/PreviewList.js b/src/components/PreviewList.js
index d6693618..6f5b854f 100644
--- a/src/components/PreviewList.js
+++ b/src/components/PreviewList.js
@@ -68,6 +68,10 @@ function PreviewList({
                 className={clsx(classes.root, previewGridClasses.container)}
             >
                 {fileObjects.map((fileObject, i) => {
+                    let chipProps = previewChipProps;
+                    if (typeof previewChipProps === 'function') {
+                        chipProps = previewChipProps(fileObject);
+                    }
                     return (
                         <Grid
                             {...previewGridProps.item}
@@ -77,7 +81,7 @@ function PreviewList({
                         >
                             <Chip
                                 variant="outlined"
-                                {...previewChipProps}
+                                {...chipProps}
                                 label={fileObject.file.name}
                                 onDelete={handleRemove(i)}
                             />
@@ -131,7 +135,7 @@ PreviewList.propTypes = {
     fileObjects: PropTypes.arrayOf(PropTypes.object).isRequired,
     getPreviewIcon: PropTypes.func.isRequired,
     handleRemove: PropTypes.func.isRequired,
-    previewChipProps: PropTypes.object,
+    previewChipProps: PropTypes.oneOfType([PropTypes.object, PropTypes.func]),
     previewGridClasses: PropTypes.object,
     previewGridProps: PropTypes.object,
     showFileNames: PropTypes.bool,
diff --git a/src/index.d.ts b/src/index.d.ts
index e52beb4f..9949214f 100644
--- a/src/index.d.ts
+++ b/src/index.d.ts
@@ -49,7 +49,7 @@ export type DropzoneAreaBaseProps = {
   showFileNamesInPreview?: boolean;
   showFileNames?: boolean;
   useChipsForPreview?: boolean;
-  previewChipProps?: ChipProps;
+  previewChipProps?: ChipProps | ((fileObject: FileObject) => ChipProps);
   previewGridClasses?: {
     container?: string;
     item?: string;