feat: batch workspace updates

coder · aslilac · Jan 18, 2024 · Jan 11, 2024 · Jan 12, 2024 · Jan 12, 2024
commit 07cdbc47c3e66fa9c42985c2f8eff63eb2600e31
diff --git a/site/src/pages/WorkspacesPage/BatchActions.tsx b/site/src/pages/WorkspacesPage/BatchActions.tsx
@@ -6,7 +6,12 @@ import "dayjs/plugin/relativeTime";
 import { type Interpolation, type Theme } from "@emotion/react";
 import { type FC, type ReactNode, useState } from "react";
 import { useMutation } from "react-query";
-import { deleteWorkspace, startWorkspace, stopWorkspace } from "api/api";
+import {
+  deleteWorkspace,
+  startWorkspace,
+  stopWorkspace,
+  updateWorkspace,
+} from "api/api";
 import type { Workspace } from "api/typesGenerated";
 import { ConfirmDialog } from "components/Dialogs/ConfirmDialog/ConfirmDialog";
 import { displayError } from "components/GlobalSnackbar/utils";
@@ -50,14 +55,25 @@ export function useBatchActions(options: UseBatchActionsProps) {
     },
     onSuccess,
     onError: () => {
-      displayError("Failed to delete workspaces");
+      displayError("Failed to delete some workspaces");
+    },
+  });
+
+  const updateAllMutation = useMutation({
+    mutationFn: async (workspaces: Workspace[]) => {
+      return Promise.all(workspaces.map((w) => updateWorkspace(w)));
+    },
+    onSuccess,
+    onError: () => {
+      displayError("Failed to update some workspaces");
     },
   });
 
   return {
     startAll: startAllMutation.mutateAsync,
     stopAll: stopAllMutation.mutateAsync,
     deleteAll: deleteAllMutation.mutateAsync,
+    updateAll: updateAllMutation.mutateAsync,
     isLoading:
       startAllMutation.isLoading ||
       stopAllMutation.isLoading ||

diff --git a/site/src/pages/WorkspacesPage/WorkspacesPage.tsx b/site/src/pages/WorkspacesPage/WorkspacesPage.tsx
@@ -99,6 +99,7 @@ const WorkspacesPage: FC = () => {
         onDeleteAll={() => {
           setIsConfirmingDeleteAll(true);
         }}
+        onUpdateAll={() => batchActions.updateAll(checkedWorkspaces)}
         onStartAll={() => batchActions.startAll(checkedWorkspaces)}
         onStopAll={() => batchActions.stopAll(checkedWorkspaces)}
       />

diff --git a/site/src/pages/WorkspacesPage/WorkspacesPageView.tsx b/site/src/pages/WorkspacesPage/WorkspacesPageView.tsx
@@ -15,6 +15,7 @@ import { WorkspacesButton } from "./WorkspacesButton";
 import { UseQueryResult } from "react-query";
 import StopOutlined from "@mui/icons-material/StopOutlined";
 import PlayArrowOutlined from "@mui/icons-material/PlayArrowOutlined";
+import CloudQueue from "@mui/icons-material/CloudQueue";
 import {
   MoreMenu,
   MoreMenuContent,
@@ -51,6 +52,7 @@ export interface WorkspacesPageViewProps {
   onCheckChange: (checkedWorkspaces: Workspace[]) => void;
   isRunningBatchAction: boolean;
   onDeleteAll: () => void;
+  onUpdateAll: () => void;
   onStartAll: () => void;
   onStopAll: () => void;
   canCheckWorkspaces: boolean;
@@ -71,6 +73,7 @@ export const WorkspacesPageView = ({
   checkedWorkspaces,
   onCheckChange,
   onDeleteAll,
+  onUpdateAll,
   onStopAll,
   onStartAll,
   isRunningBatchAction,
@@ -150,6 +153,9 @@ export const WorkspacesPageView = ({
                   <StopOutlined /> Stop
                 </MoreMenuItem>
                 <Divider />
+                <MoreMenuItem onClick={onUpdateAll}>
+                  <CloudQueue /> Update&hellip;
+                </MoreMenuItem>
                 <MoreMenuItem danger onClick={onDeleteAll}>
                   <DeleteOutlined /> Delete&hellip;
                 </MoreMenuItem>

diff --git a/site/src/theme/light/mui.ts b/site/src/theme/light/mui.ts
@@ -10,7 +10,7 @@ const muiTheme = createTheme({
     mode: "light",
     primary: {
       main: tw.sky[600],
-      contrastText: tw.sky[50],
+      contrastText: tw.sky[950],
       light: tw.sky[400],
       dark: tw.sky[500],
     },