62 lines
1.5 KiB
TypeScript
62 lines
1.5 KiB
TypeScript
|
import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
|
||
|
|
import {
|
||
|
|
getWgClients,
|
||
|
|
getWgInterface,
|
||
|
|
createWgClient,
|
||
|
|
deleteWgClient,
|
||
|
|
enableWgClient,
|
||
|
|
disableWgClient,
|
||
|
|
type WgClient,
|
||
|
|
type WgInterface,
|
||
|
|
} from "src/api/backend/wireguard";
|
||
|
|
|
||
|
|
export const useWgClients = (options = {}) => {
|
||
|
|
return useQuery<WgClient[], Error>({
|
||
|
|
queryKey: ["wg-clients"],
|
||
|
|
queryFn: getWgClients,
|
||
|
|
refetchInterval: 5000,
|
||
|
|
staleTime: 3000,
|
||
|
|
...options,
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
export const useWgInterface = (options = {}) => {
|
||
|
|
return useQuery<WgInterface, Error>({
|
||
|
|
queryKey: ["wg-interface"],
|
||
|
|
queryFn: getWgInterface,
|
||
|
|
staleTime: 60 * 1000,
|
||
|
|
...options,
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
export const useCreateWgClient = () => {
|
||
|
|
const queryClient = useQueryClient();
|
||
|
|
return useMutation({
|
||
|
|
mutationFn: (data: { name: string }) => createWgClient(data),
|
||
|
|
onSuccess: () => {
|
||
|
|
queryClient.invalidateQueries({ queryKey: ["wg-clients"] });
|
||
|
|
},
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
export const useDeleteWgClient = () => {
|
||
|
|
const queryClient = useQueryClient();
|
||
|
|
return useMutation({
|
||
|
|
mutationFn: (id: number) => deleteWgClient(id),
|
||
|
|
onSuccess: () => {
|
||
|
|
queryClient.invalidateQueries({ queryKey: ["wg-clients"] });
|
||
|
|
},
|
||
|
|
});
|
||
|
|
};
|
||
|
|
|
||
|
|
export const useToggleWgClient = () => {
|
||
|
|
const queryClient = useQueryClient();
|
||
|
|
return useMutation({
|
||
|
|
mutationFn: ({ id, enabled }: { id: number; enabled: boolean }) =>
|
||
|
|
enabled ? enableWgClient(id) : disableWgClient(id),
|
||
|
|
onSuccess: () => {
|
||
|
|
queryClient.invalidateQueries({ queryKey: ["wg-clients"] });
|
||
|
|
},
|
||
|
|
});
|
||
|
|
};
|