From 505e62aa610e870053985b1c890e3f52c7183c72 Mon Sep 17 00:00:00 2001
From: Rainnny7 <braydonrainnny@gmail.com>
Date: Thu, 19 Sep 2024 22:00:51 -0400
Subject: [PATCH] Basic disabling of TFA

---
 .../sidebar/organization-selector.tsx         |  2 +-
 .../dashboard/user/profile/tier-setting.tsx   |  2 +-
 .../user/settings/tfa/tfa-setting.tsx         | 60 ++++++++++++++-----
 .../dashboard/user/user-settings-header.tsx   |  2 +-
 4 files changed, 47 insertions(+), 19 deletions(-)

diff --git a/src/components/dashboard/sidebar/organization-selector.tsx b/src/components/dashboard/sidebar/organization-selector.tsx
index a85e905..af0bd6f 100644
--- a/src/components/dashboard/sidebar/organization-selector.tsx
+++ b/src/components/dashboard/sidebar/organization-selector.tsx
@@ -95,7 +95,7 @@ const OrganizationSelector = (): ReactElement => {
                     <ChevronsUpDownIcon className="ml-2 w-4 h-4 shrink-0 opacity-50" />
                 </Button>
             </PopoverTrigger>
-            <PopoverContent className="p-0 w-52">
+            <PopoverContent className="w-56 p-0">
                 <Command>
                     <CommandInput placeholder="Search organization..." />
                     <CommandList>
diff --git a/src/components/dashboard/user/profile/tier-setting.tsx b/src/components/dashboard/user/profile/tier-setting.tsx
index 4335591..1747c6e 100644
--- a/src/components/dashboard/user/profile/tier-setting.tsx
+++ b/src/components/dashboard/user/profile/tier-setting.tsx
@@ -33,7 +33,7 @@ const TierSetting = (): ReactElement => {
                     {capitalizeWords(user?.tier)}
                 </span>
 
-                <Link href="/#pricing">
+                <Link href="/#pricing" draggable={false}>
                     <Button
                         className="bg-background/30"
                         size="sm"
diff --git a/src/components/dashboard/user/settings/tfa/tfa-setting.tsx b/src/components/dashboard/user/settings/tfa/tfa-setting.tsx
index 2661565..719ed7b 100644
--- a/src/components/dashboard/user/settings/tfa/tfa-setting.tsx
+++ b/src/components/dashboard/user/settings/tfa/tfa-setting.tsx
@@ -39,11 +39,28 @@ const TFASetting = (): ReactElement => {
     const user: User | undefined = useUserContext(
         (state: UserState) => state.user
     );
+    const router: AppRouterInstance = useRouter();
+
     const [tfaResponse, setTfaResponse] = useState<
         UserSetupTfaResponse | undefined
     >(undefined);
     const [enabledTfa, setEnabledTfa] = useState<boolean>(false);
-    const router: AppRouterInstance = useRouter();
+    const [disabling, setDisabling] = useState<boolean>(false);
+
+    const onDialogStateChange = async (open: boolean) => {
+        if (open) {
+            await setupTfa();
+        } else if (enabledTfa) {
+            toast("Two-Factor Auth", {
+                icon: "🎉",
+                description:
+                    "Successfully enabled two-factor auth on your account!",
+            });
+            setTimeout(() => {
+                window.location.reload();
+            }, 1500);
+        }
+    };
 
     /**
      * Start setting up two-factor auth.
@@ -57,6 +74,25 @@ const TFASetting = (): ReactElement => {
         setTfaResponse(data);
     };
 
+    /**
+     * Disable two-factor auth.
+     */
+    const disableTfa = async () => {
+        setDisabling(true);
+        await apiRequest<void>({
+            endpoint: "/user/disable-tfa",
+            method: "POST",
+            session,
+        });
+        toast("Two-Factor Auth", {
+            icon: "🔓",
+            description: "Two-factor auth has been disabled for your account.",
+        });
+        setTimeout(() => {
+            window.location.reload();
+        }, 1500);
+    };
+
     return (
         <div className="px-5 flex items-center">
             {/* Name & Description */}
@@ -71,24 +107,16 @@ const TFASetting = (): ReactElement => {
 
             {/* Setting */}
             {hasFlag(user as User, UserFlag.TFA_ENABLED) ? (
-                <Button size="sm" variant="destructive">
+                <Button
+                    size="sm"
+                    variant="destructive"
+                    onClick={disableTfa}
+                    disabled={disabling}
+                >
                     Disable
                 </Button>
             ) : (
-                <Dialog
-                    onOpenChange={async (open: boolean) => {
-                        if (open) {
-                            setupTfa();
-                        } else if (enabledTfa) {
-                            toast("Two-Factor Auth", {
-                                icon: "🎉",
-                                description:
-                                    "Successfully enabled two-factor auth on your account!",
-                            });
-                            router.push("/dashboard");
-                        }
-                    }}
-                >
+                <Dialog onOpenChange={onDialogStateChange}>
                     <DialogTrigger>
                         <Button
                             className="bg-background/30"
diff --git a/src/components/dashboard/user/user-settings-header.tsx b/src/components/dashboard/user/user-settings-header.tsx
index 6f2ed54..bae084c 100644
--- a/src/components/dashboard/user/user-settings-header.tsx
+++ b/src/components/dashboard/user/user-settings-header.tsx
@@ -36,7 +36,7 @@ const UserSettingsHeader = ({ title }: { title: string }): ReactElement => {
             <Breadcrumb>
                 <BreadcrumbList>
                     <BreadcrumbItem>
-                        <BreadcrumbLink href="/dashboard">
+                        <BreadcrumbLink href="/dashboard" draggable={false}>
                             Dashboard
                         </BreadcrumbLink>
                     </BreadcrumbItem>