From 7366b4156cfc2166706b876f91e01bb77c529d04 Mon Sep 17 00:00:00 2001
From: Hassan Ali <helfi92@gmail.com>
Date: Tue, 23 Oct 2018 13:43:36 -0400
Subject: [PATCH] Provide visual feedback when terminating ec2 instances

---
 src/components/Snackbar/index.jsx             |  5 +--
 .../AwsProvisioner/ViewWorkerType/index.jsx   | 40 +++++++++++++++++--
 2 files changed, 39 insertions(+), 6 deletions(-)

diff --git a/src/components/Snackbar/index.jsx b/src/components/Snackbar/index.jsx
index cd2b3092..d72c12e4 100644
--- a/src/components/Snackbar/index.jsx
+++ b/src/components/Snackbar/index.jsx
@@ -59,12 +59,11 @@ export default class Snackbar extends Component {
     /** Properties applied to the SnackbarContent element. */
     snackbarContentProps: object,
     /** Callback fired when the component requests to be closed. */
-    onClose: func,
+    onClose: func.isRequired,
   };
 
   static defaultProps = {
     variant: 'success',
-    onClose: null,
     snackbarContentProps: null,
   };
 
@@ -81,7 +80,7 @@ export default class Snackbar extends Component {
     const isWarning = variant === 'warning';
 
     return (
-      <MuiSnackbar {...props}>
+      <MuiSnackbar onClose={onClose} {...props}>
         <SnackbarContent
           className={classes[variant]}
           action={
diff --git a/src/views/AwsProvisioner/ViewWorkerType/index.jsx b/src/views/AwsProvisioner/ViewWorkerType/index.jsx
index 7a2e7940..f88ee684 100644
--- a/src/views/AwsProvisioner/ViewWorkerType/index.jsx
+++ b/src/views/AwsProvisioner/ViewWorkerType/index.jsx
@@ -13,6 +13,7 @@ import Button from '../../../components/Button';
 import AwsProvisionerErrorsTable from '../../../components/AwsProvisionerErrorsTable';
 import AwsProvisionerHealthTable from '../../../components/AwsProvisionerHealthTable';
 import AwsProvisionerWorkerTypeStatus from '../../../components/AwsProvisionerWorkerTypeStatus';
+import Snackbar from '../../../components/Snackbar';
 import Ec2ResourcesTable from '../../../components/Ec2ResourcesTable';
 import formatError from '../../../utils/formatError';
 import workerTypeQuery from './workerType.graphql';
@@ -43,6 +44,8 @@ export default class ViewWorkerType extends Component {
   state = {
     currentTab: 0,
     actionLoading: false,
+    showTerminateAllInstancesSnackbar: false,
+    showTerminateInstanceSnackbar: false,
   };
 
   handleTabChange = (e, currentTab) => {
@@ -65,7 +68,10 @@ export default class ViewWorkerType extends Component {
         variables: { workerType },
       });
 
-      this.setState({ actionLoading: false });
+      this.setState({
+        actionLoading: false,
+        showTerminateAllInstancesSnackbar: true,
+      });
     } catch (error) {
       this.setState({ actionLoading: false, error: formatError(error) });
     }
@@ -80,12 +86,22 @@ export default class ViewWorkerType extends Component {
         variables: { region, instanceId: id },
       });
 
-      this.setState({ actionLoading: false });
+      this.setState({
+        actionLoading: false,
+        showTerminateInstanceSnackbar: true,
+      });
     } catch (error) {
       this.setState({ actionLoading: false, error: formatError(error) });
     }
   };
 
+  handleSnackbarClose = () => {
+    this.setState({
+      showTerminateAllInstancesSnackbar: false,
+      showTerminateInstanceSnackbar: false,
+    });
+  };
+
   render() {
     const {
       classes,
@@ -101,7 +117,13 @@ export default class ViewWorkerType extends Component {
         params: { workerType },
       },
     } = this.props;
-    const { currentTab, actionLoading } = this.state;
+    const {
+      currentTab,
+      actionLoading,
+      showTerminateAllInstancesSnackbar,
+      showTerminateInstanceSnackbar,
+    } = this.state;
+    const FIVE_SECONDS = 5000;
 
     return (
       <Dashboard title={`AWS Provisioner ${workerType}`}>
@@ -166,6 +188,18 @@ export default class ViewWorkerType extends Component {
               </div>
             </Tooltip>
           )}
+        <Snackbar
+          open={showTerminateAllInstancesSnackbar}
+          onClose={this.handleSnackbarClose}
+          autoHideDuration={FIVE_SECONDS}
+          message="A request has been sent to terminate all instances."
+        />
+        <Snackbar
+          open={showTerminateInstanceSnackbar}
+          onClose={this.handleSnackbarClose}
+          autoHideDuration={FIVE_SECONDS}
+          message="A request has been sent to terminate the instance."
+        />
       </Dashboard>
     );
   }