DS Boyce
103 lines
2.6 KiB
TypeScript
103 lines
2.6 KiB
TypeScript
import type AtpAgent from '@atproto/api'
|
|
import {
|
|
type AppBskyActorDefs,
|
|
type ChatBskyActorDeclaration,
|
|
} from '@atproto/api'
|
|
import {useMutation, useQueryClient} from '@tanstack/react-query'
|
|
|
|
import {logger} from '#/logger'
|
|
import {useAgent, useSession} from '#/state/session'
|
|
import {RQKEY as PROFILE_RKEY} from '../profile'
|
|
|
|
export function useUpdateActorDeclaration({
|
|
onSuccess,
|
|
onError,
|
|
}: {
|
|
onSuccess?: () => void
|
|
onError?: (error: Error) => void
|
|
}) {
|
|
const queryClient = useQueryClient()
|
|
const {currentAccount} = useSession()
|
|
const agent = useAgent()
|
|
|
|
return useMutation({
|
|
mutationFn: async (allowIncoming: 'all' | 'none' | 'following') => {
|
|
if (!currentAccount) throw new Error('Not signed in')
|
|
const result = await agent.com.atproto.repo.putRecord({
|
|
repo: currentAccount.did,
|
|
collection: 'chat.bsky.actor.declaration',
|
|
rkey: 'self',
|
|
record: {
|
|
$type: 'chat.bsky.actor.declaration',
|
|
allowIncoming,
|
|
},
|
|
})
|
|
return result
|
|
},
|
|
onMutate: allowIncoming => {
|
|
if (!currentAccount) return
|
|
queryClient.setQueryData(
|
|
PROFILE_RKEY(currentAccount?.did),
|
|
(old?: AppBskyActorDefs.ProfileViewDetailed) => {
|
|
if (!old) return old
|
|
return {
|
|
...old,
|
|
associated: {
|
|
...old.associated,
|
|
chat: {
|
|
allowIncoming,
|
|
},
|
|
},
|
|
} satisfies AppBskyActorDefs.ProfileViewDetailed
|
|
},
|
|
)
|
|
},
|
|
onSuccess,
|
|
onError: error => {
|
|
logger.error(error)
|
|
if (currentAccount) {
|
|
void queryClient.invalidateQueries({
|
|
queryKey: PROFILE_RKEY(currentAccount.did),
|
|
})
|
|
}
|
|
onError?.(error)
|
|
},
|
|
})
|
|
}
|
|
|
|
// for use in the settings screen for testing
|
|
export function useDeleteActorDeclaration() {
|
|
const {currentAccount} = useSession()
|
|
const agent = useAgent()
|
|
|
|
return useMutation({
|
|
mutationFn: async () => {
|
|
if (!currentAccount) throw new Error('Not signed in')
|
|
const result = await agent.api.com.atproto.repo.deleteRecord({
|
|
repo: currentAccount.did,
|
|
collection: 'chat.bsky.actor.declaration',
|
|
rkey: 'self',
|
|
})
|
|
return result
|
|
},
|
|
})
|
|
}
|
|
|
|
export async function fetchActorDeclarationRecord({
|
|
agent,
|
|
did,
|
|
}: {
|
|
agent: AtpAgent
|
|
did?: string
|
|
}) {
|
|
if (!did) return
|
|
const res = await agent.com.atproto.repo
|
|
.getRecord({
|
|
repo: did,
|
|
collection: 'chat.bsky.actor.declaration',
|
|
rkey: 'self',
|
|
})
|
|
.catch(_e => undefined)
|
|
return res?.data.value as ChatBskyActorDeclaration.Main
|
|
}
|