diff --git a/frontend/src/api/crowdsec.ts b/frontend/src/api/crowdsec.ts index 839bb6cc..77b1b425 100644 --- a/frontend/src/api/crowdsec.ts +++ b/frontend/src/api/crowdsec.ts @@ -156,4 +156,31 @@ export async function getCrowdsecKeyStatus(): Promise { return resp.data } -export default { startCrowdsec, stopCrowdsec, statusCrowdsec, importCrowdsecConfig, exportCrowdsecConfig, listCrowdsecFiles, readCrowdsecFile, writeCrowdsecFile, listCrowdsecDecisions, banIP, unbanIP, getCrowdsecKeyStatus } +export interface CrowdSecWhitelistEntry { + uuid: string + ip_or_cidr: string + reason: string + created_at: string + updated_at: string +} + +export interface AddWhitelistPayload { + ip_or_cidr: string + reason: string +} + +export const listWhitelists = async (): Promise => { + const resp = await client.get<{ whitelist: CrowdSecWhitelistEntry[] }>('/admin/crowdsec/whitelist') + return resp.data.whitelist +} + +export const addWhitelist = async (data: AddWhitelistPayload): Promise => { + const resp = await client.post('/admin/crowdsec/whitelist', data) + return resp.data +} + +export const deleteWhitelist = async (uuid: string): Promise => { + await client.delete(`/admin/crowdsec/whitelist/${uuid}`) +} + +export default { startCrowdsec, stopCrowdsec, statusCrowdsec, importCrowdsecConfig, exportCrowdsecConfig, listCrowdsecFiles, readCrowdsecFile, writeCrowdsecFile, listCrowdsecDecisions, banIP, unbanIP, getCrowdsecKeyStatus, listWhitelists, addWhitelist, deleteWhitelist } diff --git a/frontend/src/hooks/useCrowdSecWhitelist.ts b/frontend/src/hooks/useCrowdSecWhitelist.ts new file mode 100644 index 00000000..a36bfceb --- /dev/null +++ b/frontend/src/hooks/useCrowdSecWhitelist.ts @@ -0,0 +1,38 @@ +import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query' + +import { listWhitelists, addWhitelist, deleteWhitelist, type AddWhitelistPayload } from '../api/crowdsec' +import { toast } from '../utils/toast' + +export const useWhitelistEntries = () => + useQuery({ + queryKey: ['crowdsec-whitelist'], + queryFn: listWhitelists, + }) + +export const useAddWhitelist = () => { + const queryClient = useQueryClient() + return useMutation({ + mutationFn: (data: AddWhitelistPayload) => addWhitelist(data), + onSuccess: () => { + toast.success('Whitelist entry added') + queryClient.invalidateQueries({ queryKey: ['crowdsec-whitelist'] }) + }, + onError: (err: unknown) => { + toast.error(err instanceof Error ? err.message : 'Failed to add whitelist entry') + }, + }) +} + +export const useDeleteWhitelist = () => { + const queryClient = useQueryClient() + return useMutation({ + mutationFn: (uuid: string) => deleteWhitelist(uuid), + onSuccess: () => { + toast.success('Whitelist entry removed') + queryClient.invalidateQueries({ queryKey: ['crowdsec-whitelist'] }) + }, + onError: (err: unknown) => { + toast.error(err instanceof Error ? err.message : 'Failed to remove whitelist entry') + }, + }) +}