107 lines
3.1 KiB
TypeScript
107 lines
3.1 KiB
TypeScript
import React from 'react'
|
|
import {AppBskyActorDefs, moderateProfile, ModerationOpts} from '@atproto/api'
|
|
import {keepPreviousData, useQuery, useQueryClient} from '@tanstack/react-query'
|
|
|
|
import {isJustAMute} from '#/lib/moderation'
|
|
import {logger} from '#/logger'
|
|
import {STALE} from '#/state/queries'
|
|
import {useAgent} from '#/state/session'
|
|
import {useModerationOpts} from '../preferences/moderation-opts'
|
|
import {DEFAULT_LOGGED_OUT_PREFERENCES} from './preferences'
|
|
|
|
const DEFAULT_MOD_OPTS = {
|
|
userDid: undefined,
|
|
prefs: DEFAULT_LOGGED_OUT_PREFERENCES.moderationPrefs,
|
|
}
|
|
|
|
const RQKEY_ROOT = 'actor-autocomplete'
|
|
export const RQKEY = (prefix: string) => [RQKEY_ROOT, prefix]
|
|
|
|
export function useActorAutocompleteQuery(
|
|
prefix: string,
|
|
maintainData?: boolean,
|
|
limit?: number,
|
|
) {
|
|
const moderationOpts = useModerationOpts()
|
|
const agent = useAgent()
|
|
|
|
prefix = prefix.toLowerCase().trim()
|
|
if (prefix.endsWith('.')) {
|
|
// Going from "foo" to "foo." should not clear matches.
|
|
prefix = prefix.slice(0, -1)
|
|
}
|
|
|
|
return useQuery<AppBskyActorDefs.ProfileViewBasic[]>({
|
|
staleTime: STALE.MINUTES.ONE,
|
|
queryKey: RQKEY(prefix || ''),
|
|
async queryFn() {
|
|
const res = prefix
|
|
? await agent.searchActorsTypeahead({
|
|
q: prefix,
|
|
limit: limit || 8,
|
|
})
|
|
: undefined
|
|
return res?.data.actors || []
|
|
},
|
|
select: React.useCallback(
|
|
(data: AppBskyActorDefs.ProfileViewBasic[]) => {
|
|
return computeSuggestions(data, moderationOpts || DEFAULT_MOD_OPTS)
|
|
},
|
|
[moderationOpts],
|
|
),
|
|
placeholderData: maintainData ? keepPreviousData : undefined,
|
|
})
|
|
}
|
|
|
|
export type ActorAutocompleteFn = ReturnType<typeof useActorAutocompleteFn>
|
|
export function useActorAutocompleteFn() {
|
|
const queryClient = useQueryClient()
|
|
const moderationOpts = useModerationOpts()
|
|
const agent = useAgent()
|
|
|
|
return React.useCallback(
|
|
async ({query, limit = 8}: {query: string; limit?: number}) => {
|
|
query = query.toLowerCase()
|
|
let res
|
|
if (query) {
|
|
try {
|
|
res = await queryClient.fetchQuery({
|
|
staleTime: STALE.MINUTES.ONE,
|
|
queryKey: RQKEY(query || ''),
|
|
queryFn: () =>
|
|
agent.searchActorsTypeahead({
|
|
q: query,
|
|
limit,
|
|
}),
|
|
})
|
|
} catch (e) {
|
|
logger.error('useActorSearch: searchActorsTypeahead failed', {
|
|
message: e,
|
|
})
|
|
}
|
|
}
|
|
|
|
return computeSuggestions(
|
|
res?.data.actors,
|
|
moderationOpts || DEFAULT_MOD_OPTS,
|
|
)
|
|
},
|
|
[queryClient, moderationOpts, agent],
|
|
)
|
|
}
|
|
|
|
function computeSuggestions(
|
|
searched: AppBskyActorDefs.ProfileViewBasic[] = [],
|
|
moderationOpts: ModerationOpts,
|
|
) {
|
|
let items: AppBskyActorDefs.ProfileViewBasic[] = []
|
|
for (const item of searched) {
|
|
if (!items.find(item2 => item2.handle === item.handle)) {
|
|
items.push(item)
|
|
}
|
|
}
|
|
return items.filter(profile => {
|
|
const modui = moderateProfile(profile, moderationOpts).ui('profileList')
|
|
return !modui.filter || isJustAMute(modui)
|
|
})
|
|
}
|