29 lines
950 B
TypeScript
29 lines
950 B
TypeScript
import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query'
|
|
|
|
import { enrollConsole, getConsoleStatus, clearConsoleEnrollment, type ConsoleEnrollPayload, type ConsoleEnrollmentStatus } from '../api/consoleEnrollment'
|
|
|
|
export function useConsoleStatus(enabled = true) {
|
|
return useQuery<ConsoleEnrollmentStatus>({ queryKey: ['crowdsec-console-status'], queryFn: getConsoleStatus, enabled })
|
|
}
|
|
|
|
export function useEnrollConsole() {
|
|
const qc = useQueryClient()
|
|
return useMutation({
|
|
mutationFn: (payload: ConsoleEnrollPayload) => enrollConsole(payload),
|
|
onSuccess: () => {
|
|
qc.invalidateQueries({ queryKey: ['crowdsec-console-status'] })
|
|
},
|
|
})
|
|
}
|
|
|
|
export function useClearConsoleEnrollment() {
|
|
const queryClient = useQueryClient()
|
|
|
|
return useMutation({
|
|
mutationFn: clearConsoleEnrollment,
|
|
onSuccess: () => {
|
|
queryClient.invalidateQueries({ queryKey: ['crowdsec-console-status'] })
|
|
},
|
|
})
|
|
}
|