Various search fixes (#2145)

* Add posts-search query to shadow cache search

* Update user search to use correct endpoint

* Fix: include cursor in post search
This commit is contained in:
Paul Frazee 2023-12-07 17:20:17 -08:00 committed by GitHub
parent 448a403c81
commit 7b50331188
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 88 additions and 34 deletions

View file

@ -1,7 +1,13 @@
import {AppBskyFeedSearchPosts} from '@atproto/api'
import {useInfiniteQuery, InfiniteData, QueryKey} from '@tanstack/react-query'
import {AppBskyFeedDefs, AppBskyFeedSearchPosts} from '@atproto/api'
import {
useInfiniteQuery,
InfiniteData,
QueryKey,
QueryClient,
} from '@tanstack/react-query'
import {getAgent} from '#/state/session'
import {embedViewRecordToPostView, getEmbeddedPost} from './util'
const searchPostsQueryKey = ({query}: {query: string}) => [
'search-posts',
@ -17,10 +23,11 @@ export function useSearchPostsQuery({query}: {query: string}) {
string | undefined
>({
queryKey: searchPostsQueryKey({query}),
queryFn: async () => {
queryFn: async ({pageParam}) => {
const res = await getAgent().app.bsky.feed.searchPosts({
q: query,
limit: 25,
cursor: pageParam,
})
return res.data
},
@ -28,3 +35,30 @@ export function useSearchPostsQuery({query}: {query: string}) {
getNextPageParam: lastPage => lastPage.cursor,
})
}
export function* findAllPostsInQueryData(
queryClient: QueryClient,
uri: string,
): Generator<AppBskyFeedDefs.PostView, undefined> {
const queryDatas = queryClient.getQueriesData<
InfiniteData<AppBskyFeedSearchPosts.OutputSchema>
>({
queryKey: ['search-posts'],
})
for (const [_queryKey, queryData] of queryDatas) {
if (!queryData?.pages) {
continue
}
for (const page of queryData?.pages) {
for (const post of page.posts) {
if (post.uri === uri) {
yield post
}
const quotedPost = getEmbeddedPost(post.embed)
if (quotedPost?.uri === uri) {
yield embedViewRecordToPostView(quotedPost)
}
}
}
}
}