import * as ATPAPI from "@atproto/api"; import { infiniteQueryOptions, type QueryFunctionContext, queryOptions, useInfiniteQuery, useQuery, type UseQueryResult} from "@tanstack/react-query"; import { useAtom } from "jotai"; import { constellationURLAtom, slingshotURLAtom } from "./atoms"; export function constructIdentityQuery(didorhandle?: string, slingshoturl?: string) { return queryOptions({ queryKey: ["identity", didorhandle], queryFn: async () => { if (!didorhandle) return undefined as undefined const res = await fetch( `https://${slingshoturl}/xrpc/com.bad-example.identity.resolveMiniDoc?identifier=${encodeURIComponent(didorhandle)}` ); if (!res.ok) throw new Error("Failed to fetch post"); try { return (await res.json()) as { did: string; handle: string; pds: string; signing_key: string; }; } catch (_e) { return undefined; } }, staleTime: /*0,//*/ 5 * 60 * 1000, // 5 minutes gcTime: /*0//*/5 * 60 * 1000, }); } export function useQueryIdentity(didorhandle: string): UseQueryResult< { did: string; handle: string; pds: string; signing_key: string; }, Error >; export function useQueryIdentity(): UseQueryResult< undefined, Error > export function useQueryIdentity(didorhandle?: string): UseQueryResult< { did: string; handle: string; pds: string; signing_key: string; } | undefined, Error > export function useQueryIdentity(didorhandle?: string) { const [slingshoturl] = useAtom(slingshotURLAtom) return useQuery(constructIdentityQuery(didorhandle, slingshoturl)); } export function constructPostQuery(uri?: string, slingshoturl?: string) { return queryOptions({ queryKey: ["post", uri], queryFn: async () => { if (!uri) return undefined as undefined const res = await fetch( `https://${slingshoturl}/xrpc/com.bad-example.repo.getUriRecord?at_uri=${encodeURIComponent(uri)}` ); let data: any; try { data = await res.json(); } catch { return undefined; } if (res.status === 400) return undefined; if (data?.error === "InvalidRequest" && data.message?.includes("Could not find repo")) { return undefined; // cache “not found” } try { if (!res.ok) throw new Error("Failed to fetch post"); return (data) as { uri: string; cid: string; value: any; }; } catch (_e) { return undefined; } }, retry: (failureCount, error) => { // dont retry 400 errors if ((error as any)?.message?.includes("400")) return false; return failureCount < 2; }, staleTime: /*0,//*/ 5 * 60 * 1000, // 5 minutes gcTime: /*0//*/5 * 60 * 1000, }); } export function useQueryPost(uri: string): UseQueryResult< { uri: string; cid: string; value: ATPAPI.AppBskyFeedPost.Record; }, Error >; export function useQueryPost(): UseQueryResult< undefined, Error > export function useQueryPost(uri?: string): UseQueryResult< { uri: string; cid: string; value: ATPAPI.AppBskyFeedPost.Record; } | undefined, Error > export function useQueryPost(uri?: string) { const [slingshoturl] = useAtom(slingshotURLAtom) return useQuery(constructPostQuery(uri, slingshoturl)); } export function constructProfileQuery(uri?: string, slingshoturl?: string) { return queryOptions({ queryKey: ["profile", uri], queryFn: async () => { if (!uri) return undefined as undefined const res = await fetch( `https://${slingshoturl}/xrpc/com.bad-example.repo.getUriRecord?at_uri=${encodeURIComponent(uri)}` ); let data: any; try { data = await res.json(); } catch { return undefined; } if (res.status === 400) return undefined; if (data?.error === "InvalidRequest" && data.message?.includes("Could not find repo")) { return undefined; // cache “not found” } try { if (!res.ok) throw new Error("Failed to fetch post"); return (data) as { uri: string; cid: string; value: any; }; } catch (_e) { return undefined; } }, retry: (failureCount, error) => { // dont retry 400 errors if ((error as any)?.message?.includes("400")) return false; return failureCount < 2; }, staleTime: /*0,//*/ 5 * 60 * 1000, // 5 minutes gcTime: /*0//*/5 * 60 * 1000, }); } export function useQueryProfile(uri: string): UseQueryResult< { uri: string; cid: string; value: ATPAPI.AppBskyActorProfile.Record; }, Error >; export function useQueryProfile(): UseQueryResult< undefined, Error >; export function useQueryProfile(uri?: string): UseQueryResult< { uri: string; cid: string; value: ATPAPI.AppBskyActorProfile.Record; } | undefined, Error > export function useQueryProfile(uri?: string) { const [slingshoturl] = useAtom(slingshotURLAtom) return useQuery(constructProfileQuery(uri, slingshoturl)); } // export function constructConstellationQuery( // method: "/links", // target: string, // collection: string, // path: string, // cursor?: string // ): QueryOptions; // export function constructConstellationQuery( // method: "/links/distinct-dids", // target: string, // collection: string, // path: string, // cursor?: string // ): QueryOptions; // export function constructConstellationQuery( // method: "/links/count", // target: string, // collection: string, // path: string, // cursor?: string // ): QueryOptions; // export function constructConstellationQuery( // method: "/links/count/distinct-dids", // target: string, // collection: string, // path: string, // cursor?: string // ): QueryOptions; // export function constructConstellationQuery( // method: "/links/all", // target: string // ): QueryOptions; export function constructConstellationQuery(query?:{ constellation: string, method: | "/links" | "/links/distinct-dids" | "/links/count" | "/links/count/distinct-dids" | "/links/all" | "undefined", target: string, collection?: string, path?: string, cursor?: string, dids?: string[] } ) { // : QueryOptions< // | linksRecordsResponse // | linksDidsResponse // | linksCountResponse // | linksAllResponse // | undefined, // Error // > return queryOptions({ queryKey: ["constellation", query?.method, query?.target, query?.collection, query?.path, query?.cursor, query?.dids] as const, queryFn: async () => { if (!query || query.method === "undefined") return undefined as undefined const method = query.method const target = query.target const collection = query?.collection const path = query?.path const cursor = query.cursor const dids = query?.dids const res = await fetch( `https://${query.constellation}${method}?target=${encodeURIComponent(target)}${collection ? `&collection=${encodeURIComponent(collection)}` : ""}${path ? `&path=${encodeURIComponent(path)}` : ""}${cursor ? `&cursor=${encodeURIComponent(cursor)}` : ""}${dids ? dids.map((did) => `&did=${encodeURIComponent(did)}`).join("") : ""}` ); if (!res.ok) throw new Error("Failed to fetch post"); try { switch (method) { case "/links": return (await res.json()) as linksRecordsResponse; case "/links/distinct-dids": return (await res.json()) as linksDidsResponse; case "/links/count": return (await res.json()) as linksCountResponse; case "/links/count/distinct-dids": return (await res.json()) as linksCountResponse; case "/links/all": return (await res.json()) as linksAllResponse; default: return undefined; } } catch (_e) { return undefined; } }, // enforce short lifespan staleTime: /*0,//*/ 5 * 60 * 1000, // 5 minutes gcTime: /*0//*/5 * 60 * 1000, }); } export function useQueryConstellation(query: { method: "/links"; target: string; collection: string; path: string; cursor?: string; dids?: string[]; }): UseQueryResult; export function useQueryConstellation(query: { method: "/links/distinct-dids"; target: string; collection: string; path: string; cursor?: string; }): UseQueryResult; export function useQueryConstellation(query: { method: "/links/count"; target: string; collection: string; path: string; cursor?: string; }): UseQueryResult; export function useQueryConstellation(query: { method: "/links/count/distinct-dids"; target: string; collection: string; path: string; cursor?: string; }): UseQueryResult; export function useQueryConstellation(query: { method: "/links/all"; target: string; }): UseQueryResult; export function useQueryConstellation(): undefined; export function useQueryConstellation(query: { method: "undefined"; target: string; }): undefined; export function useQueryConstellation(query?: { method: | "/links" | "/links/distinct-dids" | "/links/count" | "/links/count/distinct-dids" | "/links/all" | "undefined"; target: string; collection?: string; path?: string; cursor?: string; dids?: string[]; }): | UseQueryResult< | linksRecordsResponse | linksDidsResponse | linksCountResponse | linksAllResponse | undefined, Error > | undefined { //if (!query) return; const [constellationurl] = useAtom(constellationURLAtom) return useQuery( constructConstellationQuery(query && {constellation: constellationurl, ...query}) ); } type linksRecord = { did: string; collection: string; rkey: string; }; export type linksRecordsResponse = { total: string; linking_records: linksRecord[]; cursor?: string; }; type linksDidsResponse = { total: string; linking_dids: string[]; cursor?: string; }; type linksCountResponse = { total: string; }; export type linksAllResponse = { links: Record< string, Record< string, { records: number; distinct_dids: number; } > >; }; export function constructFeedSkeletonQuery(options?: { feedUri: string; agent?: ATPAPI.Agent; isAuthed: boolean; pdsUrl?: string; feedServiceDid?: string; }) { return queryOptions({ // The query key includes all dependencies to ensure it refetches when they change queryKey: ["feedSkeleton", options?.feedUri, { isAuthed: options?.isAuthed, did: options?.agent?.did }], queryFn: async () => { if (!options) return undefined as undefined const { feedUri, agent, isAuthed, pdsUrl, feedServiceDid } = options; if (isAuthed) { // Authenticated flow if (!agent || !pdsUrl || !feedServiceDid) { throw new Error("Missing required info for authenticated feed fetch."); } const url = `${pdsUrl}/xrpc/app.bsky.feed.getFeedSkeleton?feed=${encodeURIComponent(feedUri)}`; const res = await agent.fetchHandler(url, { method: "GET", headers: { "atproto-proxy": `${feedServiceDid}#bsky_fg`, "Content-Type": "application/json", }, }); if (!res.ok) throw new Error(`Authenticated feed fetch failed: ${res.statusText}`); return (await res.json()) as ATPAPI.AppBskyFeedGetFeedSkeleton.OutputSchema; } else { // Unauthenticated flow (using a public PDS/AppView) const url = `https://discover.bsky.app/xrpc/app.bsky.feed.getFeedSkeleton?feed=${encodeURIComponent(feedUri)}`; const res = await fetch(url); if (!res.ok) throw new Error(`Public feed fetch failed: ${res.statusText}`); return (await res.json()) as ATPAPI.AppBskyFeedGetFeedSkeleton.OutputSchema; } }, //enabled: !!feedUri && (isAuthed ? !!agent && !!pdsUrl && !!feedServiceDid : true), }); } export function useQueryFeedSkeleton(options?: { feedUri: string; agent?: ATPAPI.Agent; isAuthed: boolean; pdsUrl?: string; feedServiceDid?: string; }) { return useQuery(constructFeedSkeletonQuery(options)); } export function constructPreferencesQuery(agent?: ATPAPI.Agent | undefined, pdsUrl?: string | undefined) { return queryOptions({ queryKey: ['preferences', agent?.did], queryFn: async () => { if (!agent || !pdsUrl) throw new Error("Agent or PDS URL not available"); const url = `${pdsUrl}/xrpc/app.bsky.actor.getPreferences`; const res = await agent.fetchHandler(url, { method: "GET" }); if (!res.ok) throw new Error("Failed to fetch preferences"); return res.json(); }, }); } export function useQueryPreferences(options: { agent?: ATPAPI.Agent | undefined, pdsUrl?: string | undefined }) { return useQuery(constructPreferencesQuery(options.agent, options.pdsUrl)); } export function constructArbitraryQuery(uri?: string, slingshoturl?: string) { return queryOptions({ queryKey: ["arbitrary", uri], queryFn: async () => { if (!uri) return undefined as undefined const res = await fetch( `https://${slingshoturl}/xrpc/com.bad-example.repo.getUriRecord?at_uri=${encodeURIComponent(uri)}` ); let data: any; try { data = await res.json(); } catch { return undefined; } if (res.status === 400) return undefined; if (data?.error === "InvalidRequest" && data.message?.includes("Could not find repo")) { return undefined; // cache “not found” } try { if (!res.ok) throw new Error("Failed to fetch post"); return (data) as { uri: string; cid: string; value: any; }; } catch (_e) { return undefined; } }, retry: (failureCount, error) => { // dont retry 400 errors if ((error as any)?.message?.includes("400")) return false; return failureCount < 2; }, staleTime: /*0,//*/ 5 * 60 * 1000, // 5 minutes gcTime: /*0//*/5 * 60 * 1000, }); } export function useQueryArbitrary(uri: string): UseQueryResult< { uri: string; cid: string; value: any; }, Error >; export function useQueryArbitrary(): UseQueryResult< undefined, Error >; export function useQueryArbitrary(uri?: string): UseQueryResult< { uri: string; cid: string; value: any; } | undefined, Error >; export function useQueryArbitrary(uri?: string) { const [slingshoturl] = useAtom(slingshotURLAtom) return useQuery(constructArbitraryQuery(uri, slingshoturl)); } export function constructFallbackNothingQuery(){ return queryOptions({ queryKey: ["nothing"], queryFn: async () => { return undefined }, }); } type ListRecordsResponse = { cursor?: string; records: { uri: string; cid: string; value: ATPAPI.AppBskyFeedPost.Record; }[]; }; export function constructAuthorFeedQuery(did: string, pdsUrl: string) { return queryOptions({ queryKey: ['authorFeed', did], queryFn: async ({ pageParam }: QueryFunctionContext) => { const limit = 25; const cursor = pageParam as string | undefined; const cursorParam = cursor ? `&cursor=${cursor}` : ''; const url = `${pdsUrl}/xrpc/com.atproto.repo.listRecords?repo=${did}&collection=app.bsky.feed.post&limit=${limit}${cursorParam}`; const res = await fetch(url); if (!res.ok) throw new Error("Failed to fetch author's posts"); return res.json() as Promise; }, }); } export function useInfiniteQueryAuthorFeed(did: string | undefined, pdsUrl: string | undefined) { const { queryKey, queryFn } = constructAuthorFeedQuery(did!, pdsUrl!); return useInfiniteQuery({ queryKey, queryFn, initialPageParam: undefined as never, // ???? what is this shit getNextPageParam: (lastPage) => lastPage.cursor as null | undefined, enabled: !!did && !!pdsUrl, }); } type FeedSkeletonPage = ATPAPI.AppBskyFeedGetFeedSkeleton.OutputSchema; export function constructInfiniteFeedSkeletonQuery(options: { feedUri: string; agent?: ATPAPI.Agent; isAuthed: boolean; pdsUrl?: string; feedServiceDid?: string; }) { const { feedUri, agent, isAuthed, pdsUrl, feedServiceDid } = options; return queryOptions({ queryKey: ["feedSkeleton", feedUri, { isAuthed, did: agent?.did }], queryFn: async ({ pageParam }: QueryFunctionContext): Promise => { const cursorParam = pageParam ? `&cursor=${pageParam}` : ""; if (isAuthed) { if (!agent || !pdsUrl || !feedServiceDid) { throw new Error("Missing required info for authenticated feed fetch."); } const url = `${pdsUrl}/xrpc/app.bsky.feed.getFeedSkeleton?feed=${encodeURIComponent(feedUri)}${cursorParam}`; const res = await agent.fetchHandler(url, { method: "GET", headers: { "atproto-proxy": `${feedServiceDid}#bsky_fg`, "Content-Type": "application/json", }, }); if (!res.ok) throw new Error(`Authenticated feed fetch failed: ${res.statusText}`); return (await res.json()) as FeedSkeletonPage; } else { const url = `https://discover.bsky.app/xrpc/app.bsky.feed.getFeedSkeleton?feed=${encodeURIComponent(feedUri)}${cursorParam}`; const res = await fetch(url); if (!res.ok) throw new Error(`Public feed fetch failed: ${res.statusText}`); return (await res.json()) as FeedSkeletonPage; } }, }); } export function useInfiniteQueryFeedSkeleton(options: { feedUri: string; agent?: ATPAPI.Agent; isAuthed: boolean; pdsUrl?: string; feedServiceDid?: string; }) { const { queryKey, queryFn } = constructInfiniteFeedSkeletonQuery(options); return {...useInfiniteQuery({ queryKey, queryFn, initialPageParam: undefined as never, getNextPageParam: (lastPage) => lastPage.cursor as null | undefined, staleTime: Infinity, refetchOnWindowFocus: false, enabled: !!options.feedUri && (options.isAuthed ? !!options.agent && !!options.pdsUrl && !!options.feedServiceDid : true), }), queryKey: queryKey}; } export function yknowIReallyHateThisButWhateverGuardedConstructConstellationInfiniteQueryLinks(query?: { constellation: string, method: '/links' target?: string collection: string path: string }) { console.log( 'yknowIReallyHateThisButWhateverGuardedConstructConstellationInfiniteQueryLinks', query, ) return infiniteQueryOptions({ enabled: !!query?.target, queryKey: [ 'reddwarf_constellation', query?.method, query?.target, query?.collection, query?.path, ] as const, queryFn: async ({pageParam}: {pageParam?: string}) => { if (!query || !query?.target) return undefined const method = query.method const target = query.target const collection = query.collection const path = query.path const cursor = pageParam const res = await fetch( `https://${query.constellation}${method}?target=${encodeURIComponent(target)}${ collection ? `&collection=${encodeURIComponent(collection)}` : '' }${path ? `&path=${encodeURIComponent(path)}` : ''}${ cursor ? `&cursor=${encodeURIComponent(cursor)}` : '' }`, ) if (!res.ok) throw new Error('Failed to fetch') return (await res.json()) as linksRecordsResponse }, getNextPageParam: lastPage => { return (lastPage as any)?.cursor ?? undefined }, initialPageParam: undefined, staleTime: 5 * 60 * 1000, gcTime: 5 * 60 * 1000, }) }