yaak-mountain-loop/src-web/hooks/useRequest.ts

import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
import { invoke } from '@tauri-apps/api';
import { convertDates, HttpRequest } from '../lib/models';
import { responsesQueryKey } from './useResponses';

export function useRequests(workspaceId: string) {
  return useQuery(['requests'], async () => {
    const requests = (await invoke('requests', { workspaceId })) as HttpRequest[];
    return requests.map(convertDates);
  });
}

export function useRequestUpdate(request: HttpRequest | null) {
  const queryClient = useQueryClient();
  return useMutation<HttpRequest, unknown, Partial<HttpRequest>>({
    mutationFn: async (patch) => {
      if (request == null) {
        throw new Error("Can't update a null request");
      }

      const updatedRequest = { ...request, ...patch } as any;

      // TODO: Figure out why this is necessary
      updatedRequest.createdAt = updatedRequest.createdAt.toISOString().replace('Z', '');
      updatedRequest.updatedAt = updatedRequest.updatedAt.toISOString().replace('Z', '');

      const req = await invoke('update_request', { request: updatedRequest });
      return convertDates(req as HttpRequest);
    },
    onSuccess: (req) => {
      queryClient.setQueryData(['requests'], (requests: HttpRequest[] = []) =>
        requests.map((r) => (r.id === req.id ? req : r)),
      );
    },
  });
}

export function useRequestCreate(workspaceId: string) {
  const queryClient = useQueryClient();
  return useMutation<HttpRequest, unknown, Pick<HttpRequest, 'name'>>({
    mutationFn: async (patch) => {
      const req = await invoke('create_request', {
        ...patch,
        workspaceId,
      });
      return convertDates(req as HttpRequest);
    },
    onSuccess: (req) => {
      queryClient.setQueryData(['requests'], (requests: HttpRequest[] = []) => [...requests, req]);
    },
  });
}

export function useSendRequest(request: HttpRequest | null) {
  const queryClient = useQueryClient();
  return useMutation<void, string>({
    mutationFn: async () => {
      if (request == null) return;
      await invoke('send_request', { requestId: request.id });
    },
    onSuccess: async () => {
      if (request == null) return;
      await queryClient.invalidateQueries(responsesQueryKey(request.id));
    },
  });
}