From 49f601d200f76236a6be656d2d13c8948b3fdd95 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tr=C3=A8sAbhi?= <43380238+tresabhi@users.noreply.github.com>
Date: Sun, 1 Oct 2023 16:37:51 -0500
Subject: [PATCH] feat: added reset prompt

---
 src/app/tools/session/page.tsx | 83 +++++++++++++++++++++++++++++-----
 src/stores/session.ts          | 19 ++++++--
 2 files changed, 87 insertions(+), 15 deletions(-)

diff --git a/src/app/tools/session/page.tsx b/src/app/tools/session/page.tsx
index 97023ac9a..d9a10c27c 100644
--- a/src/app/tools/session/page.tsx
+++ b/src/app/tools/session/page.tsx
@@ -1,7 +1,15 @@
 'use client';
 
 import { CopyIcon, PersonIcon, ReloadIcon } from '@radix-ui/react-icons';
-import { Button, DropdownMenu, TextField } from '@radix-ui/themes';
+import {
+  AlertDialog,
+  Button,
+  Checkbox,
+  DropdownMenu,
+  Flex,
+  Text,
+  TextField,
+} from '@radix-ui/themes';
 import { debounce } from 'lodash';
 import { ChangeEvent, useRef, useState } from 'react';
 import PageWrapper from '../../../components/PageWrapper';
@@ -63,6 +71,12 @@ export default function Page() {
     });
   }
 
+  function reset() {
+    const session = useSession.getState();
+    if (!session.isTracking) return;
+    setSession(session.region, session.id, session.nickname);
+  }
+
   return (
     <PageWrapper>
       <div className={styles.toolBar}>
@@ -132,17 +146,62 @@ export default function Page() {
         </div>
 
         <div style={{ display: 'flex', gap: 8 }}>
-          <Button
-            className={styles.toolbarButton}
-            color="red"
-            onClick={async () => {
-              const session = useSession.getState();
-              if (!session.isTracking) return;
-              setSession(session.region, session.id, session.nickname);
-            }}
-          >
-            <ReloadIcon width="16" height="16" /> Reset
-          </Button>
+          <AlertDialog.Root>
+            <AlertDialog.Trigger>
+              <Button
+                onClick={(event) => {
+                  if (!session.promptBeforeReset) {
+                    event.preventDefault();
+                    reset();
+                  }
+                }}
+                className={styles.toolbarButton}
+                color="red"
+              >
+                <ReloadIcon width="16" height="16" /> Reset
+              </Button>
+            </AlertDialog.Trigger>
+
+            <AlertDialog.Content>
+              <Flex direction="column" gap="4">
+                <Flex direction="column" gap="0">
+                  <AlertDialog.Title>
+                    Are you sure you want to reset?
+                  </AlertDialog.Title>
+                  <AlertDialog.Description>
+                    This'll discard your current session and start tracking from
+                    scratch again.
+                  </AlertDialog.Description>
+                </Flex>
+
+                <Text size="2">
+                  <Checkbox
+                    mr="1"
+                    onCheckedChange={(checked) =>
+                      useSession.setState({
+                        promptBeforeReset: !checked,
+                      })
+                    }
+                  />
+                  Don't ask me again
+                </Text>
+              </Flex>
+
+              <Flex gap="3" mt="4">
+                <AlertDialog.Cancel>
+                  <Button variant="soft" color="gray">
+                    Cancel
+                  </Button>
+                </AlertDialog.Cancel>
+
+                <AlertDialog.Action>
+                  <Button variant="solid" color="red" onClick={reset}>
+                    Reset
+                  </Button>
+                </AlertDialog.Action>
+              </Flex>
+            </AlertDialog.Content>
+          </AlertDialog.Root>
           <Button
             className={styles.toolbarButton}
             onClick={() =>
diff --git a/src/stores/session.ts b/src/stores/session.ts
index aeeb82ac5..02cd818d3 100644
--- a/src/stores/session.ts
+++ b/src/stores/session.ts
@@ -3,7 +3,7 @@ import { devtools, persist } from 'zustand/middleware';
 import { Region } from '../constants/regions';
 import { NormalizedTankStats } from '../types/tanksStats';
 
-type Session =
+type Session = (
   | {
       isTracking: false;
     }
@@ -15,8 +15,21 @@ type Session =
       nickname: string;
       tankStats: NormalizedTankStats;
       time: number;
-    };
+    }
+) & {
+  promptBeforeReset: boolean;
+};
 
 export const useSession = create<Session>()(
-  devtools(persist((set) => ({ isTracking: false }), { name: 'session' })),
+  devtools(
+    persist(
+      (set) => ({
+        isTracking: false,
+        promptBeforeReset: true,
+      }),
+      {
+        name: 'session',
+      },
+    ),
+  ),
 );