Samuel Newman
88 lines
2.9 KiB
TypeScript
88 lines
2.9 KiB
TypeScript
import type AtpAgent from '@atproto/api'
|
|
import {type ChatBskyActorDeclaration} from '@atproto/api'
|
|
|
|
import {networkRetry} from '#/lib/async/retry'
|
|
import {logger} from '#/logger'
|
|
import {
|
|
getDidFromAgentSession,
|
|
getOtherRequiredDataFromCache,
|
|
setOtherRequiredDataActorDeclarationCache,
|
|
} from '#/ageAssurance/data'
|
|
|
|
/**
|
|
* Updates the chat actor declaration record to restrict who can contact the
|
|
* user. Both restrictions write to the same record (`rkey: 'self'`), so this
|
|
* is a single helper to avoid two concurrent `putRecord` calls racing each
|
|
* other and clobbering one another's changes.
|
|
*
|
|
* - `restrictIncoming`: sets `allowIncoming: 'none'` (used when a user isn't
|
|
* age-assured).
|
|
* - `restrictGroupInvites`: sets `allowGroupInvites: 'none'` (used for under-18
|
|
* users, who per spec cannot participate in group chats).
|
|
*
|
|
* Dimensions that aren't being restricted preserve their cached value, falling
|
|
* back to the lexicon defaults when the cache is empty.
|
|
*/
|
|
export async function restrictChatSettings({
|
|
agent,
|
|
restrictIncoming = false,
|
|
restrictGroupInvites = false,
|
|
}: {
|
|
agent: AtpAgent
|
|
restrictIncoming?: boolean
|
|
restrictGroupInvites?: boolean
|
|
}): Promise<void> {
|
|
const did = getDidFromAgentSession(agent)
|
|
if (!did) return
|
|
|
|
const cached = getOtherRequiredDataFromCache({did})?.actorDeclaration
|
|
|
|
// When the cache is empty we fall back to defaults for any dimension we're
|
|
// not explicitly restricting, which could drop/downgrade a value the user
|
|
// has actually set server-side. The cache should be hydrated by
|
|
// prefetchOtherRequiredData before any of these paths fire, so log if that
|
|
// assumption ever breaks. (Restricting both dimensions is unaffected, so
|
|
// don't warn for the common signup/birthdate-change case.)
|
|
if (!cached && (!restrictIncoming || !restrictGroupInvites)) {
|
|
logger.warn(
|
|
`restrictChatSettings: cache miss, falling back to defaults for unrestricted dimensions`,
|
|
)
|
|
}
|
|
|
|
const record: ChatBskyActorDeclaration.Main = {
|
|
$type: 'chat.bsky.actor.declaration',
|
|
allowIncoming: restrictIncoming
|
|
? 'none'
|
|
: (cached?.allowIncoming ?? 'following'),
|
|
allowGroupInvites: restrictGroupInvites
|
|
? 'none'
|
|
: cached?.allowGroupInvites,
|
|
}
|
|
|
|
// Nothing to do if the record already reflects the desired restrictions.
|
|
if (
|
|
cached?.allowIncoming === record.allowIncoming &&
|
|
cached?.allowGroupInvites === record.allowGroupInvites
|
|
) {
|
|
return
|
|
}
|
|
|
|
try {
|
|
await networkRetry(3, () =>
|
|
agent.com.atproto.repo.putRecord({
|
|
repo: did,
|
|
collection: 'chat.bsky.actor.declaration',
|
|
rkey: 'self',
|
|
record,
|
|
}),
|
|
)
|
|
// important, update local cache to avoid running this again
|
|
setOtherRequiredDataActorDeclarationCache({
|
|
did,
|
|
actorDeclaration: record,
|
|
})
|
|
} catch {
|
|
logger.error(`restrictChatSettings: failed to set chat declaration`)
|
|
}
|
|
}
|