Merge pull request mbrn#2335 from TeamDiverst/remove_pdf

allow control over export buttons
jayalath-jknr · Aug 23, 2020 · 2384f5a · 2384f5a
2 parents 6b2e168 + 1a4a048
commit 2384f5a
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 9 deletions.
diff --git a/src/components/m-table-toolbar.js b/src/components/m-table-toolbar.js
@@ -256,12 +256,16 @@ export class MTableToolbar extends React.Component {
               open={Boolean(this.state.exportButtonAnchorEl)}
               onClose={() => this.setState({ exportButtonAnchorEl: null })}
             >
-              <MenuItem key="export-csv" onClick={this.exportCsv}>
-                {localization.exportCSVName}
-              </MenuItem>
-              <MenuItem key="export-pdf" onClick={this.exportPdf}>
-                {localization.exportPDFName}
-              </MenuItem>
+              {(this.props.exportButton === true || this.props.exportButton.csv ) && (
+                <MenuItem key="export-csv" onClick={this.exportCsv}>
+                  {localization.exportCSVName}
+                </MenuItem>
+              )}
+              {(this.props.exportButton === true || this.props.exportButton.pdf ) && (
+                <MenuItem key="export-pdf" onClick={this.exportPdf}>
+                  {localization.exportPDFName}
+                </MenuItem>
+              )}
             </Menu>
           </span>
         )}
@@ -419,7 +423,7 @@ MTableToolbar.propTypes = {
   renderData: PropTypes.array,
   data: PropTypes.array,
   exportAllData: PropTypes.bool,
-  exportButton: PropTypes.bool,
+  exportButton: PropTypes.oneOfType([PropTypes.bool, PropTypes.shape({ csv: PropTypes.bool, pdf: PropTypes.bool })]),
   exportDelimiter: PropTypes.string,
   exportFileName: PropTypes.oneOfType([PropTypes.string, PropTypes.func]),
   exportCsv: PropTypes.func,

diff --git a/src/prop-types.js b/src/prop-types.js
@@ -308,7 +308,10 @@ export const propTypes = {
     doubleHorizontalScroll: PropTypes.bool,
     emptyRowsWhenPaging: PropTypes.bool,
     exportAllData: PropTypes.bool,
-    exportButton: PropTypes.bool,
+    exportButton: PropTypes.oneOfType([
+      PropTypes.bool,
+      PropTypes.shape({ csv: PropTypes.bool, pdf: PropTypes.bool }),
+    ]),
     exportDelimiter: PropTypes.string,
     exportFileName: PropTypes.oneOfType([PropTypes.string, PropTypes.func]),
     exportCsv: PropTypes.func,

diff --git a/types/index.d.ts b/types/index.d.ts
@@ -306,7 +306,7 @@ export interface Options<RowData extends object> {
   draggable?: boolean;
   emptyRowsWhenPaging?: boolean;
   exportAllData?: boolean;
-  exportButton?: boolean;
+  exportButton?: boolean | { csv?: boolean; pdf?: boolean };
   exportDelimiter?: string;
   exportFileName?:
     | string