From 74ffcc6215475c4e612463efd0968d591d48159a Mon Sep 17 00:00:00 2001
From: "Jingwei.Zhang" <zhang.jingwei@99cloud.net>
Date: Wed, 6 Sep 2023 17:27:04 +0800
Subject: [PATCH] feat: update ActionButton component

1. refactor ActionButton
2. support ok/cancel button props for confirm/modal action

Change-Id: I07ba1e7b3ef7c7ef998aacf54bd5329d01737d0e
---
 .../Tables/Base/ActionButton/index.jsx        | 105 ++++++++++++------
 1 file changed, 69 insertions(+), 36 deletions(-)

diff --git a/src/components/Tables/Base/ActionButton/index.jsx b/src/components/Tables/Base/ActionButton/index.jsx
index 57578344..13e18338 100644
--- a/src/components/Tables/Base/ActionButton/index.jsx
+++ b/src/components/Tables/Base/ActionButton/index.jsx
@@ -15,7 +15,7 @@
 import React, { Component } from 'react';
 import { inject, observer } from 'mobx-react';
 import { Modal, Button, Tooltip } from 'antd';
-import { isArray, isFunction, isBoolean } from 'lodash';
+import { isArray, isFunction, isBoolean, isEmpty } from 'lodash';
 import Confirm from 'components/Confirm';
 import PropTypes from 'prop-types';
 import Notify from 'components/Notify';
@@ -23,7 +23,7 @@ import classnames from 'classnames';
 import { firstUpperCase, allSettled } from 'utils';
 import styles from './index.less';
 
-function getDefaultMsg(action, data) {
+export const getDefaultMsg = (action, data) => {
   const { actionName = '', title = '' } = action;
   const name = isArray(data) ? data.map((it) => it.name).join(', ') : data.name;
   const submitErrorMsg = t('Unable to {action} {name}.', {
@@ -50,7 +50,7 @@ function getDefaultMsg(action, data) {
     confirmContext,
     performErrorMsg,
   };
-}
+};
 
 export class ActionButton extends Component {
   static propTypes() {
@@ -253,21 +253,40 @@ export class ActionButton extends Component {
     this.onCallback(false, true);
   };
 
-  onShowConfirm = async () => {
-    const {
-      perform,
-      title,
-      confirmContext,
-      okText,
-      cancelText,
-      onSubmit,
-      afterSubmit,
-    } = this.props.action;
-    const { item, items, isBatch, containerProps, onCancelAction } = this.props;
-    const data = isBatch ? items : item;
-    const content = confirmContext
+  getConfirmOkButtonProps = (data, action) => {
+    const { disableSubmit = false, okButtonProps } = action;
+    if (okButtonProps) {
+      return okButtonProps;
+    }
+    return {
+      disabled: disableSubmit,
+    };
+  };
+
+  getConfirmCancelButtonProps = (data, action) => {
+    const { cancelButtonProps } = action;
+    if (cancelButtonProps) {
+      return cancelButtonProps;
+    }
+    return {};
+  };
+
+  getConfirmContent = (data, action) => {
+    const { confirmContext } = action;
+    return confirmContext
       ? confirmContext(data)
-      : getDefaultMsg(this.props.action, data).confirmContext;
+      : getDefaultMsg(action, data).confirmContext;
+  };
+
+  onShowConfirm = async () => {
+    const { item, items, isBatch, containerProps, onCancelAction, action } =
+      this.props;
+    const { perform, title, okText, cancelText, onSubmit, afterSubmit } =
+      action;
+    const data = isBatch ? items : item;
+    const content = this.getConfirmContent(data, action);
+    const okButtonProps = this.getConfirmOkButtonProps(data, action);
+    const cancelButtonProps = this.getConfirmCancelButtonProps(data, action);
     try {
       perform(data).then(
         () => {
@@ -276,6 +295,8 @@ export class ActionButton extends Component {
             content,
             okText,
             cancelText,
+            okButtonProps,
+            cancelButtonProps,
             onOk: () => {
               return this.onConfirmOK(
                 data,
@@ -292,8 +313,7 @@ export class ActionButton extends Component {
           });
         },
         (error) => {
-          const message =
-            error || getDefaultMsg(this.props.action, data).performErrorMsg;
+          const message = error || getDefaultMsg(action, data).performErrorMsg;
           Confirm.error({
             content: message,
           });
@@ -302,8 +322,7 @@ export class ActionButton extends Component {
     } catch (error) {
       // eslint-disable-next-line no-console
       console.log(error);
-      const message =
-        error || getDefaultMsg(this.props.action, data).performErrorMsg;
+      const message = error || getDefaultMsg(action, data).performErrorMsg;
       Confirm.error({
         content: message,
       });
@@ -444,6 +463,29 @@ export class ActionButton extends Component {
     }
   };
 
+  getModalOkButtonProps = (item, action) => {
+    const { disableSubmit = false, okButtonProps } = action;
+    if (okButtonProps) {
+      return okButtonProps;
+    }
+    return {
+      disabled: disableSubmit,
+    };
+  };
+
+  getModalCancelButtonProps = (item, action) => {
+    const { readOnly, cancelButtonProps } = action;
+    if (cancelButtonProps) {
+      return cancelButtonProps;
+    }
+    if (readOnly) {
+      return {
+        style: { display: 'none' },
+      };
+    }
+    return {};
+  };
+
   showModalAction() {
     this.setState({
       visible: true,
@@ -465,15 +507,10 @@ export class ActionButton extends Component {
       onCancelAction,
     } = this.props;
     const ActionComponent = action;
-    const {
-      okText,
-      cancelText,
-      id,
-      className,
-      readOnly,
-      disableSubmit = false,
-    } = action;
+    const { okText, cancelText, id, className } = action;
     const width = this.getModalWidth(action);
+    const okButtonProps = this.getModalOkButtonProps(item, action);
+    const cancelButtonProps = this.getModalCancelButtonProps(item, action);
     const modalProps = {
       title,
       visible,
@@ -481,18 +518,14 @@ export class ActionButton extends Component {
       width,
       onOk: () => this.onClickModalActionOk(),
       onCancel: this.onClickModalActionCancel,
-      okButtonProps: {
-        disabled: disableSubmit,
-      },
+      okButtonProps,
       confirmLoading: submitLoading,
       okText,
       cancelText,
       maskClosable: false,
     };
-    if (readOnly) {
-      modalProps.cancelButtonProps = {
-        style: { display: 'none' },
-      };
+    if (!isEmpty(cancelButtonProps)) {
+      modalProps.cancelButtonProps = cancelButtonProps;
     }
     return (
       <Modal {...modalProps}>