yaak/src-web/hooks/useCreateEnvironment.ts

import { useMutation, useQueryClient } from '@tanstack/react-query';
import { invoke } from '@tauri-apps/api';
import type { Environment } from '../lib/models';
import { environmentsQueryKey } from './useEnvironments';
import { useActiveWorkspaceId } from './useActiveWorkspaceId';

export function useCreateEnvironment() {
  const workspaceId = useActiveWorkspaceId();
  const queryClient = useQueryClient();
  return useMutation<Environment, unknown, Pick<Environment, 'name'>>({
    mutationFn: (patch) => {
      return invoke('create_environment', { ...patch, workspaceId });
    },
    onSuccess: async (environment) => {
      if (workspaceId == null) return;
      queryClient.setQueryData<Environment[]>(environmentsQueryKey({ workspaceId }), (environments) => [
        ...(environments ?? []),
        environment,
      ]);
    },
  });
}