import { useMutation } from '@tanstack/react-query'; import { save } from '@tauri-apps/plugin-dialog'; import slugify from 'slugify'; import { trackEvent } from '../lib/analytics'; import type { HttpResponse } from '@yaakapp/api'; import { invokeCmd } from '../lib/tauri'; import { useActiveCookieJar } from './useActiveCookieJar'; import { useActiveEnvironment } from './useActiveEnvironment'; import { useAlert } from './useAlert'; import { useHttpRequests } from './useHttpRequests'; export function useSendAnyHttpRequest(options: { download?: boolean } = {}) { const environment = useActiveEnvironment(); const alert = useAlert(); const { activeCookieJar } = useActiveCookieJar(); const requests = useHttpRequests(); return useMutation({ mutationKey: ['send_any_request'], mutationFn: async (id) => { const request = requests.find((r) => r.id === id) ?? null; if (request == null) { return null; } let downloadDir: string | null = null; if (options.download) { downloadDir = await save({ title: 'Select Download Destination', defaultPath: slugify(request.name, { lower: true, trim: true, strict: true }), }); if (downloadDir == null) { return null; } } return invokeCmd('cmd_send_http_request', { request, environmentId: environment?.id, downloadDir: downloadDir, cookieJarId: activeCookieJar?.id, }); }, onSettled: () => trackEvent('http_request', 'send'), onError: (err) => alert({ id: 'send-failed', title: 'Send Failed', body: err }), }); }