lenooby09.tech
mirror of https://git.lenooby09.tech/LeNooby09/social-app.git
hailey.at
1import {
2 AppBskyActorDefs,
3 AppBskyGraphDefs,
4 AppBskyGraphGetList,
5 BskyAgent,
6} from '@atproto/api'
7import {
8 InfiniteData,
9 QueryClient,
10 QueryKey,
11 useInfiniteQuery,
12 useQuery,
13} from '@tanstack/react-query'
14
15import {STALE} from '#/state/queries'
16import {useAgent} from '#/state/session'
17
18const PAGE_SIZE = 30
19type RQPageParam = string | undefined
20
21const RQKEY_ROOT = 'list-members'
22const RQKEY_ROOT_ALL = 'list-members-all'
23export const RQKEY = (uri: string) => [RQKEY_ROOT, uri]
24export const RQKEY_ALL = (uri: string) => [RQKEY_ROOT_ALL, uri]
25
26export function useListMembersQuery(uri?: string, limit: number = PAGE_SIZE) {
27 const agent = useAgent()
28 return useInfiniteQuery<
29 AppBskyGraphGetList.OutputSchema,
30 Error,
31 InfiniteData<AppBskyGraphGetList.OutputSchema>,
32 QueryKey,
33 RQPageParam
34 >({
35 staleTime: STALE.MINUTES.ONE,
36 queryKey: RQKEY(uri ?? ''),
37 async queryFn({pageParam}: {pageParam: RQPageParam}) {
38 const res = await agent.app.bsky.graph.getList({
39 list: uri!, // the enabled flag will prevent this from running until uri is set
40 limit,
41 cursor: pageParam,
42 })
43 return res.data
44 },
45 initialPageParam: undefined,
46 getNextPageParam: lastPage => lastPage.cursor,
47 enabled: Boolean(uri),
48 })
49}
50
51export function useAllListMembersQuery(uri?: string) {
52 const agent = useAgent()
53 return useQuery({
54 staleTime: STALE.MINUTES.ONE,
55 queryKey: RQKEY_ALL(uri ?? ''),
56 queryFn: async () => {
57 return getAllListMembers(agent, uri!)
58 },
59 enabled: Boolean(uri),
60 })
61}
62
63export async function getAllListMembers(agent: BskyAgent, uri: string) {
64 let hasMore = true
65 let cursor: string | undefined
66 const listItems: AppBskyGraphDefs.ListItemView[] = []
67 // We want to cap this at 6 pages, just for anything weird happening with the api
68 let i = 0
69 while (hasMore && i < 6) {
70 const res = await agent.app.bsky.graph.getList({
71 list: uri,
72 limit: 50,
73 cursor,
74 })
75 listItems.push(...res.data.items)
76 hasMore = Boolean(res.data.cursor)
77 cursor = res.data.cursor
78 i++
79 }
80 return listItems
81}
82
83export async function invalidateListMembersQuery({
84 queryClient,
85 uri,
86}: {
87 queryClient: QueryClient
88 uri: string
89}) {
90 await queryClient.invalidateQueries({queryKey: RQKEY(uri)})
91}
92
93export function* findAllProfilesInQueryData(
94 queryClient: QueryClient,
95 did: string,
96): Generator<AppBskyActorDefs.ProfileView, void> {
97 const queryDatas = queryClient.getQueriesData<
98 InfiniteData<AppBskyGraphGetList.OutputSchema>
99 >({
100 queryKey: [RQKEY_ROOT],
101 })
102 for (const [_queryKey, queryData] of queryDatas) {
103 if (!queryData) {
104 continue
105 }
106 for (const [_queryKey, queryData] of queryDatas) {
107 if (!queryData?.pages) {
108 continue
109 }
110 for (const page of queryData?.pages) {
111 if (page.list.creator.did === did) {
112 yield page.list.creator
113 }
114 for (const item of page.items) {
115 if (item.subject.did === did) {
116 yield item.subject
117 }
118 }
119 }
120 }
121 }
122
123 const allQueryData = queryClient.getQueriesData<
124 AppBskyGraphDefs.ListItemView[]
125 >({
126 queryKey: [RQKEY_ROOT_ALL],
127 })
128 for (const [_queryKey, queryData] of allQueryData) {
129 if (!queryData) {
130 continue
131 }
132 for (const item of queryData) {
133 if (item.subject.did === did) {
134 yield item.subject
135 }
136 }
137 }
138}