Merge pull request #594 from CREDO23/feat/add-jotai-tanstack-connectors

[Feat] Connectors| Add jotai & tanstack

surfsense_web/atoms/connectors/connector-mutation.atoms.ts

@@ -0,0 +1,100 @@
+import { atomWithMutation } from "jotai-tanstack-query";
+import { toast } from "sonner";
+import type {
+	CreateConnectorRequest,
+	DeleteConnectorRequest,
+	GetConnectorsResponse,
+	IndexConnectorRequest,
+	IndexConnectorResponse,
+	UpdateConnectorRequest,
+} from "@/contracts/types/connector.types";
+import { connectorsApiService } from "@/lib/apis/connectors-api.service";
+import { cacheKeys } from "@/lib/query-client/cache-keys";
+import { queryClient } from "@/lib/query-client/client";
+import { activeSearchSpaceIdAtom } from "../search-spaces/search-space-query.atoms";
+
+export const createConnectorMutationAtom = atomWithMutation((get) => {
+	const searchSpaceId = get(activeSearchSpaceIdAtom);
+
+	return {
+		mutationKey: cacheKeys.connectors.all(searchSpaceId!),
+		enabled: !!searchSpaceId,
+		mutationFn: async (request: CreateConnectorRequest) => {
+			return connectorsApiService.createConnector(request);
+		},
+
+		onSuccess: () => {
+			queryClient.invalidateQueries({
+				queryKey: cacheKeys.connectors.all(searchSpaceId!),
+			});
+		},
+	};
+});
+
+export const updateConnectorMutationAtom = atomWithMutation((get) => {
+	const searchSpaceId = get(activeSearchSpaceIdAtom);
+
+	return {
+		mutationKey: cacheKeys.connectors.all(searchSpaceId!),
+		enabled: !!searchSpaceId,
+		mutationFn: async (request: UpdateConnectorRequest) => {
+			return connectorsApiService.updateConnector(request);
+		},
+
+		onSuccess: (_, request: UpdateConnectorRequest) => {
+			queryClient.invalidateQueries({
+				queryKey: cacheKeys.connectors.all(searchSpaceId!),
+			});
+			queryClient.invalidateQueries({
+				queryKey: cacheKeys.connectors.byId(String(request.id)),
+			});
+		},
+	};
+});
+
+export const deleteConnectorMutationAtom = atomWithMutation((get) => {
+	const searchSpaceId = get(activeSearchSpaceIdAtom);
+
+	return {
+		mutationKey: cacheKeys.connectors.all(searchSpaceId!),
+		enabled: !!searchSpaceId,
+		mutationFn: async (request: DeleteConnectorRequest) => {
+			return connectorsApiService.deleteConnector(request);
+		},
+
+		onSuccess: (_, request: DeleteConnectorRequest) => {
+			queryClient.setQueryData(
+				cacheKeys.connectors.all(searchSpaceId!),
+				(oldData: GetConnectorsResponse | undefined) => {
+					if (!oldData) return oldData;
+					return oldData.filter((connector) => connector.id !== request.id);
+				}
+			);
+			queryClient.invalidateQueries({
+				queryKey: cacheKeys.connectors.byId(String(request.id)),
+			});
+		},
+	};
+});
+
+export const indexConnectorMutationAtom = atomWithMutation((get) => {
+	const searchSpaceId = get(activeSearchSpaceIdAtom);
+
+	return {
+		mutationKey: cacheKeys.connectors.index(),
+		enabled: !!searchSpaceId,
+		mutationFn: async (request: IndexConnectorRequest) => {
+			return connectorsApiService.indexConnector(request);
+		},
+
+		onSuccess: (response: IndexConnectorResponse) => {
+			toast.success(response.message);
+			queryClient.invalidateQueries({
+				queryKey: cacheKeys.connectors.all(searchSpaceId!),
+			});
+			queryClient.invalidateQueries({
+				queryKey: cacheKeys.connectors.byId(String(response.connector_id)),
+			});
+		},
+	};
+});

surfsense_web/atoms/connectors/connector-query.atoms.ts

@@ -0,0 +1,21 @@
+import { atomWithQuery } from "jotai-tanstack-query";
+import { connectorsApiService } from "@/lib/apis/connectors-api.service";
+import { cacheKeys } from "@/lib/query-client/cache-keys";
+import { activeSearchSpaceIdAtom } from "../search-spaces/search-space-query.atoms";
+
+export const connectorsAtom = atomWithQuery((get) => {
+	const searchSpaceId = get(activeSearchSpaceIdAtom);
+
+	return {
+		queryKey: cacheKeys.connectors.all(searchSpaceId!),
+		enabled: !!searchSpaceId,
+		staleTime: 5 * 60 * 1000, // 5 minutes
+		queryFn: async () => {
+			return connectorsApiService.getConnectors({
+				queryParams: {
+					search_space_id: searchSpaceId!,
+				},
+			});
+		},
+	};
+});

surfsense_web/atoms/connectors/ui.atoms.ts

@@ -0,0 +1,7 @@
+import { atom } from "jotai";
+import type { GetConnectorsRequest } from "@/contracts/types/connector.types";
+
+export const globalConnectorsQueryParamsAtom = atom<GetConnectorsRequest["queryParams"]>({
+	skip: 0,
+	limit: 10,
+});