import { useState } from 'react'; import { msg } from '@lingui/core/macro'; import { useLingui } from '@lingui/react'; import { Trans } from '@lingui/react/macro'; import { useRevalidator } from 'react-router'; import { match } from 'ts-pattern'; import { AppError, AppErrorCode } from '@documenso/lib/errors/app-error'; import { trpc } from '@documenso/trpc/react'; import type { TGetUserResponse } from '@documenso/trpc/server/admin-router/get-user.types'; import { Alert, AlertDescription, AlertTitle } from '@documenso/ui/primitives/alert'; import { Button } from '@documenso/ui/primitives/button'; import { Dialog, DialogContent, DialogDescription, DialogFooter, DialogHeader, DialogTitle, DialogTrigger, } from '@documenso/ui/primitives/dialog'; import { Input } from '@documenso/ui/primitives/input'; import { useToast } from '@documenso/ui/primitives/use-toast'; export type AdminUserResetTwoFactorDialogProps = { className?: string; user: TGetUserResponse; }; export const AdminUserResetTwoFactorDialog = ({ className, user, }: AdminUserResetTwoFactorDialogProps) => { const { _ } = useLingui(); const { toast } = useToast(); const { revalidate } = useRevalidator(); const [email, setEmail] = useState(''); const [open, setOpen] = useState(false); const { mutateAsync: resetTwoFactor, isPending: isResettingTwoFactor } = trpc.admin.user.resetTwoFactor.useMutation(); const onResetTwoFactor = async () => { try { await resetTwoFactor({ userId: user.id, }); toast({ title: _(msg`2FA Reset`), description: _(msg`The user's two factor authentication has been reset successfully.`), duration: 5000, }); await revalidate(); setOpen(false); } catch (err) { const error = AppError.parseError(err); const errorMessage = match(error.code) .with(AppErrorCode.NOT_FOUND, () => msg`User not found.`) .with( AppErrorCode.UNAUTHORIZED, () => msg`You are not authorized to reset two factor authentcation for this user.`, ) .otherwise( () => msg`An error occurred while resetting two factor authentication for the user.`, ); toast({ title: _(msg`Error`), description: _(errorMessage), variant: 'destructive', duration: 7500, }); } }; const handleOpenChange = (newOpen: boolean) => { setOpen(newOpen); if (!newOpen) { setEmail(''); } }; return (