From 92b329a0fb269280cc6f13f64ddc4cbd72842e44 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tr=C3=A8sAbhi?= <43380238+tresabhi@users.noreply.github.com>
Date: Mon, 8 Apr 2024 19:04:07 -0500
Subject: [PATCH] feat: module quick buttons

---
 src/app/tools/compare/page.tsx | 38 ++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/src/app/tools/compare/page.tsx b/src/app/tools/compare/page.tsx
index c8a7e0c70..851660951 100644
--- a/src/app/tools/compare/page.tsx
+++ b/src/app/tools/compare/page.tsx
@@ -815,6 +815,44 @@ export default function Page() {
                                     });
                                   }}
                                 />
+
+                                <Flex justify="end" mt="4" gap="4">
+                                  <Button
+                                    variant="ghost"
+                                    onClick={() => {
+                                      mutateCompareTemporary((draft) => {
+                                        const member = draft.members[index];
+
+                                        member.turret = member.tank.turrets[0];
+                                        member.gun = member.turret.guns[0];
+                                        member.shell = member.gun.shells[0];
+                                        member.engine = member.tank.engines[0];
+                                        member.track = member.tank.tracks[0];
+                                      });
+                                    }}
+                                  >
+                                    Stock
+                                  </Button>
+                                  <Button
+                                    variant="ghost"
+                                    onClick={() => {
+                                      mutateCompareTemporary((draft) => {
+                                        const member = draft.members[index];
+
+                                        member.turret =
+                                          member.tank.turrets.at(-1)!;
+                                        member.gun = member.turret.guns.at(-1)!;
+                                        member.shell = member.gun.shells[0];
+                                        member.engine =
+                                          member.tank.engines.at(-1)!;
+                                        member.track =
+                                          member.tank.tracks.at(-1)!;
+                                      });
+                                    }}
+                                  >
+                                    Upgrade
+                                  </Button>
+                                </Flex>
                               </Popover.Content>
                             </Popover.Root>
                           </Button>