coder/site/src/pages/WorkspacesPage/batchActions.tsx

import { useMutation } from "react-query";
import {
  deleteWorkspace,
  deleteFavoriteWorkspace,
  putFavoriteWorkspace,
  startWorkspace,
  stopWorkspace,
  updateWorkspace,
} from "api/api";
import type { Workspace } from "api/typesGenerated";
import { displayError } from "components/GlobalSnackbar/utils";

interface UseBatchActionsProps {
  onSuccess: () => Promise<void>;
}

export function useBatchActions(options: UseBatchActionsProps) {
  const { onSuccess } = options;

  const startAllMutation = useMutation({
    mutationFn: (workspaces: readonly Workspace[]) => {
      return Promise.all(
        workspaces.map((w) =>
          startWorkspace(w.id, w.latest_build.template_version_id),
        ),
      );
    },
    onSuccess,
    onError: () => {
      displayError("Failed to start workspaces");
    },
  });

  const stopAllMutation = useMutation({
    mutationFn: (workspaces: readonly Workspace[]) => {
      return Promise.all(workspaces.map((w) => stopWorkspace(w.id)));
    },
    onSuccess,
    onError: () => {
      displayError("Failed to stop workspaces");
    },
  });

  const deleteAllMutation = useMutation({
    mutationFn: (workspaces: readonly Workspace[]) => {
      return Promise.all(workspaces.map((w) => deleteWorkspace(w.id)));
    },
    onSuccess,
    onError: () => {
      displayError("Failed to delete some workspaces");
    },
  });

  const updateAllMutation = useMutation({
    mutationFn: (workspaces: readonly Workspace[]) => {
      return Promise.all(
        workspaces
          .filter((w) => w.outdated && !w.dormant_at)
          .map((w) => updateWorkspace(w)),
      );
    },
    onSuccess,
    onError: () => {
      displayError("Failed to update some workspaces");
    },
  });

  const favoriteAllMutation = useMutation({
    mutationFn: (workspaces: readonly Workspace[]) => {
      return Promise.all(
        workspaces
          .filter((w) => !w.favorite)
          .map((w) => putFavoriteWorkspace(w.id)),
      );
    },
    onSuccess,
    onError: () => {
      displayError("Failed to favorite some workspaces");
    },
  });

  const unfavoriteAllMutation = useMutation({
    mutationFn: (workspaces: readonly Workspace[]) => {
      return Promise.all(
        workspaces
          .filter((w) => w.favorite)
          .map((w) => deleteFavoriteWorkspace(w.id)),
      );
    },
    onSuccess,
    onError: () => {
      displayError("Failed to unfavorite some workspaces");
    },
  });

  return {
    favoriteAll: favoriteAllMutation.mutateAsync,
    unfavoriteAll: unfavoriteAllMutation.mutateAsync,
    startAll: startAllMutation.mutateAsync,
    stopAll: stopAllMutation.mutateAsync,
    deleteAll: deleteAllMutation.mutateAsync,
    updateAll: updateAllMutation.mutateAsync,
    isLoading:
      favoriteAllMutation.isLoading ||
      unfavoriteAllMutation.isLoading ||
      startAllMutation.isLoading ||
      stopAllMutation.isLoading ||
      deleteAllMutation.isLoading,
  };
}