src/components/dialogs/EmailDialog/data/useManageEmail2FA.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

import {useMutation} from '@tanstack/react-query'

import {useAgent, useSession} from '#/state/session'
import {useUpdateAccountEmailStateQueryCache} from '#/components/dialogs/EmailDialog/data/useAccountEmailState'

export function useManageEmail2FA() {
  const agent = useAgent()
  const {currentAccount} = useSession()
  const updateAccountEmailStateQueryCache =
    useUpdateAccountEmailStateQueryCache()

  return useMutation({
    mutationFn: async ({
      enabled,
      token,
    }:
      | {enabled: true; token?: undefined}
      | {enabled: false; token: string}) => {
      if (!currentAccount?.email) {
        throw new Error('No email found for the current account')
      }

      await agent.com.atproto.server.updateEmail({
        email: currentAccount.email,
        emailAuthFactor: enabled,
        token,
      })
      const {data} = await agent.resumeSession(agent.session!)
      updateAccountEmailStateQueryCache({
        isEmailVerified: !!data.emailConfirmed,
        email2FAEnabled: !!data.emailAuthFactor,
      })
    },
  })
}