Redesign TableActions component to be more generic

ui/src/components/RetryTasksTable.tsx

@@ -15,6 +15,9 @@ import Box from "@material-ui/core/Box";
 import Checkbox from "@material-ui/core/Checkbox";
 import Collapse from "@material-ui/core/Collapse";
 import IconButton from "@material-ui/core/IconButton";
+import PlayArrowIcon from "@material-ui/icons/PlayArrow";
+import DeleteIcon from "@material-ui/icons/Delete";
+import ArchiveIcon from "@material-ui/icons/Archive";
 import KeyboardArrowUpIcon from "@material-ui/icons/KeyboardArrowUp";
 import KeyboardArrowDownIcon from "@material-ui/icons/KeyboardArrowDown";
 import Typography from "@material-ui/core/Typography";
@@ -174,15 +177,44 @@ function RetryTasksTable(props: Props & ReduxProps) {
   return (
     <div>
       <TableActions
-        allActionPending={props.allActionPending}
-        batchActionPending={props.batchActionPending}
-        showBatchActions={numSelected > 0}
-        onRunAllClick={handleRunAllClick}
-        onDeleteAllClick={handleDeleteAllClick}
-        onKillAllClick={handleKillAllClick}
-        onBatchRunClick={handleBatchRunClick}
-        onBatchDeleteClick={handleBatchDeleteClick}
-        onBatchKillClick={handleBatchKillClick}
+        showIconButtons={numSelected > 0}
+        iconButtonActions={[
+          {
+            tooltip: "Delete",
+            icon: <DeleteIcon />,
+            onClick: handleBatchDeleteClick,
+            disabled: props.batchActionPending,
+          },
+          {
+            tooltip: "Kill",
+            icon: <ArchiveIcon />,
+            onClick: handleBatchKillClick,
+            disabled: props.batchActionPending,
+          },
+          {
+            tooltip: "Run",
+            icon: <PlayArrowIcon />,
+            onClick: handleBatchRunClick,
+            disabled: props.batchActionPending,
+          },
+        ]}
+        menuItemActions={[
+          {
+            label: "Delete All",
+            onClick: handleDeleteAllClick,
+            disabled: props.allActionPending,
+          },
+          {
+            label: "Kill All",
+            onClick: handleKillAllClick,
+            disabled: props.allActionPending,
+          },
+          {
+            label: "Run All",
+            onClick: handleRunAllClick,
+            disabled: props.allActionPending,
+          },
+        ]}
       />
       <TableContainer component={Paper}>
         <Table