From 05f1872ef7c4802ecf72d24532e7406fd9beced1 Mon Sep 17 00:00:00 2001
From: Eric Bailey <git@esb.lol>
Date: Sun, 27 Apr 2025 18:31:03 -0500
Subject: [PATCH] Add new state

---
 src/components/dialogs/EmailDialog/index.tsx |  8 +++-----
 src/components/dialogs/EmailDialog/types.ts  | 12 ++++++++++--
 2 files changed, 13 insertions(+), 7 deletions(-)

diff --git a/src/components/dialogs/EmailDialog/index.tsx b/src/components/dialogs/EmailDialog/index.tsx
index b59b2be9a0..72c87a0d12 100644
--- a/src/components/dialogs/EmailDialog/index.tsx
+++ b/src/components/dialogs/EmailDialog/index.tsx
@@ -13,13 +13,11 @@ import {useRefreshSession} from '#/components/dialogs/EmailDialog/data/useRefres
  */
 import {Update} from '#/components/dialogs/EmailDialog/screens/Update'
 import {Verify} from '#/components/dialogs/EmailDialog/screens/Verify'
-import {type Screen,ScreenID} from '#/components/dialogs/EmailDialog/types'
+import {type Screen, ScreenID} from '#/components/dialogs/EmailDialog/types'
 
 export {useDialogControl} from '#/components/Dialog'
-export {
-  ScreenID as EmailDialogScreenID,
-  type Screen,
-} from '#/components/dialogs/EmailDialog/types'
+export type {Screen} from '#/components/dialogs/EmailDialog/types'
+export {ScreenID as EmailDialogScreenID} from '#/components/dialogs/EmailDialog/types'
 
 export function useEmailDialogControl() {
   return useStatefulDialogControl<Screen>()
diff --git a/src/components/dialogs/EmailDialog/types.ts b/src/components/dialogs/EmailDialog/types.ts
index 65124d9c32..3ca0afa51b 100644
--- a/src/components/dialogs/EmailDialog/types.ts
+++ b/src/components/dialogs/EmailDialog/types.ts
@@ -17,8 +17,16 @@ export type Screen =
       instructions?: ReactNode[]
       hideInitialCodeButton?: boolean
     }
+  | {
+      id: ScreenID.Enable2FA
+    }
+  | {
+      id: ScreenID.Disable2FA
+    }
 
 export enum ScreenID {
-  Update = 'Update', // normal, instructs to click link first
-  Verify = 'Verify', // as a separate step, instructs to click link first
+  Update = 'Update',
+  Verify = 'Verify',
+  Enable2FA = 'Enable2FA',
+  Disable2FA = 'Disable2FA',
 }