SurfSense/surfsense_web/atoms/search-spaces/search-space-query.atoms.ts

import { atomWithQuery } from "jotai-tanstack-query";
import { atom } from "jotai";
import type { GetSearchSpacesRequest } from "@/contracts/types/search-space.types";
import { searchSpacesApiService } from "@/lib/apis/search-spaces-api.service";
import { cacheKeys } from "@/lib/query-client/cache-keys";

export const activeSearchSpaceIdAtom = atom<string | null>(null);

export const searchSpacesQueryParamsAtom = atom<GetSearchSpacesRequest["queryParams"]>({
	skip: 0,
	limit: 10,
	owned_only: false,
});

export const searchSpacesAtom = atomWithQuery((get) => {
	const queryParams = get(searchSpacesQueryParamsAtom);

	return {
		queryKey: cacheKeys.searchSpaces.withQueryParams(queryParams),
		staleTime: 5 * 60 * 1000,
		queryFn: async () => {
			return searchSpacesApiService.getSearchSpaces({
				queryParams,
			});
		},
	};
});

export const communityPromptsAtom = atomWithQuery(() => {
	return {
		queryKey: cacheKeys.searchSpaces.communityPrompts,
		staleTime: 30 * 60 * 1000,
		queryFn: async () => {
			return searchSpacesApiService.getCommunityPrompts();
		},
	};
});