Files
Dorod-Sky/skyvern-frontend/src/hooks/useOnePasswordToken.ts
2025-08-05 23:33:08 +08:00

63 lines
2.0 KiB
TypeScript

import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
import { getClient } from "@/api/AxiosClient";
import { useCredentialGetter } from "./useCredentialGetter";
import {
CreateOnePasswordTokenRequest,
CreateOnePasswordTokenResponse,
OnePasswordTokenApiResponse,
} from "@/api/types";
import { useToast } from "@/components/ui/use-toast";
export function useOnePasswordToken() {
const credentialGetter = useCredentialGetter();
const queryClient = useQueryClient();
const { toast } = useToast();
const { data: onePasswordToken, isLoading } =
useQuery<OnePasswordTokenApiResponse>({
queryKey: ["onePasswordToken"],
queryFn: async () => {
const client = await getClient(credentialGetter, "sans-api-v1");
return await client
.get("/credentials/onepassword/get")
.then((response) => response.data.token)
.catch(() => null);
},
});
const createOrUpdateTokenMutation = useMutation({
mutationFn: async (data: CreateOnePasswordTokenRequest) => {
const client = await getClient(credentialGetter, "sans-api-v1");
return await client
.post("/credentials/onepassword/create", data)
.then((response) => response.data as CreateOnePasswordTokenResponse);
},
onSuccess: () => {
queryClient.invalidateQueries({ queryKey: ["onePasswordToken"] });
toast({
title: "Success",
description: "1Password service account token updated successfully",
});
},
onError: (error: unknown) => {
const message =
(error as { response?: { data?: { detail?: string } } })?.response?.data
?.detail ||
(error as Error)?.message ||
"Failed to update 1Password token";
toast({
title: "Error",
description: message,
variant: "destructive",
});
},
});
return {
onePasswordToken,
isLoading,
createOrUpdateToken: createOrUpdateTokenMutation.mutate,
isUpdating: createOrUpdateTokenMutation.isPending,
};
}