From 54e012f7368863cb7c81a9f188091bb5ed851d6e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tr=C3=A8sAbhi?= <43380238+tresabhi@users.noreply.github.com>
Date: Mon, 1 Apr 2024 19:05:55 -0500
Subject: [PATCH] feat: dropdown modules

---
 scripts/buildAssets/definitions.ts            |   2 +
 .../Characteristics/components/Modules.tsx    | 265 ++++++++++++------
 .../ModuleButtons/TankComponentButton.tsx     |  10 +-
 src/core/blitzkrieg/tankDefinitions/index.ts  |   1 +
 4 files changed, 189 insertions(+), 89 deletions(-)

diff --git a/scripts/buildAssets/definitions.ts b/scripts/buildAssets/definitions.ts
index d1b9a5eab..5635291fc 100644
--- a/scripts/buildAssets/definitions.ts
+++ b/scripts/buildAssets/definitions.ts
@@ -118,6 +118,7 @@ interface VehicleDefinitions {
       weight: number;
       terrainResistance: string;
       rotationSpeed: number;
+      userString: string;
       shotDispersionFactors: {
         vehicleMovement: number;
         vehicleRotation: number;
@@ -566,6 +567,7 @@ export async function definitions(production: boolean) {
           tankDefinitions[tankId].tracks.push({
             id: trackId,
             weight: track.weight,
+            name: strings[track.userString],
             traverseSpeed: track.rotationSpeed,
             dispersion: {
               move: track.shotDispersionFactors.vehicleMovement,
diff --git a/src/app/tools/tankopedia/[id]/components/Characteristics/components/Modules.tsx b/src/app/tools/tankopedia/[id]/components/Characteristics/components/Modules.tsx
index ff1564892..780b66ac8 100644
--- a/src/app/tools/tankopedia/[id]/components/Characteristics/components/Modules.tsx
+++ b/src/app/tools/tankopedia/[id]/components/Characteristics/components/Modules.tsx
@@ -1,4 +1,5 @@
-import { Button, Flex, Heading } from '@radix-ui/themes';
+import { Button, DropdownMenu, Flex, Heading } from '@radix-ui/themes';
+import { useState } from 'react';
 import { ModuleButton } from '../../../../../../../components/ModuleButtons/ModuleButton';
 import { TIER_ROMAN_NUMERALS } from '../../../../../../../core/blitzkrieg/tankDefinitions/constants';
 import { mutateDuel, useDuel } from '../../../../../../../stores/duel';
@@ -12,6 +13,10 @@ export function Modules() {
     protagonist.tank.turrets[0].guns.length > 1 ||
     protagonist.tank.engines.length > 1 ||
     protagonist.tank.tracks.length > 1;
+  const [turretMenuOpen, setTurretMenuOpen] = useState(false);
+  const [gunMenuOpen, setGunMenuOpen] = useState(false);
+  const [engineMenuOpen, setEngineMenuOpen] = useState(false);
+  const [trackMenuOpen, setTrackMenuOpen] = useState(false);
 
   return (
     <ConfigurationChildWrapper>
@@ -59,96 +64,180 @@ export function Modules() {
         )}
       </Flex>
 
-      <Flex gap="2" wrap="wrap">
-        <Flex>
-          {protagonist.tank.turrets.map((turret, index) => {
-            return (
-              <ModuleButton
-                key={turret.id}
-                selected={protagonist.turret.id === turret.id}
-                module="turret"
-                discriminator={TIER_ROMAN_NUMERALS[turret.tier]}
-                first={index === 0}
-                last={index === protagonist.tank.turrets.length - 1}
-                rowChild
-                onClick={() => {
-                  mutateDuel((draft) => {
-                    draft.protagonist!.turret = turret;
-                    draft.protagonist!.gun = turret.guns.at(-1)!;
-                    draft.protagonist!.shell = turret.guns.at(-1)!.shells[0];
-                  });
-                  mutateTankopediaTemporary((draft) => {
-                    draft.shot = undefined;
-                  });
-                }}
-              />
-            );
-          })}
-        </Flex>
+      <Flex gap="1" wrap="wrap">
+        <ModuleButton
+          module="turret"
+          discriminator={TIER_ROMAN_NUMERALS[protagonist.turret.tier]}
+          first
+          last
+          rowChild
+          onClick={() => setTurretMenuOpen(true)}
+        />
 
-        <Flex>
-          {protagonist.turret.guns.map((gun, index) => {
-            return (
-              <ModuleButton
-                key={gun.id}
-                module="gun"
-                discriminator={TIER_ROMAN_NUMERALS[gun.tier]}
-                selected={protagonist.gun.id === gun.id}
-                first={index === 0}
-                last={index === protagonist.turret.guns.length - 1}
-                rowChild
-                onClick={() => {
-                  mutateDuel((draft) => {
-                    draft.protagonist!.gun = gun;
-                    draft.protagonist!.shell = gun.shells[0];
-                  });
-                }}
-              />
-            );
-          })}
-        </Flex>
+        <DropdownMenu.Root
+          open={turretMenuOpen}
+          onOpenChange={setTurretMenuOpen}
+        >
+          <DropdownMenu.Trigger>
+            <div />
+          </DropdownMenu.Trigger>
 
-        <Flex>
-          {protagonist.tank.engines.map((engine, index) => {
-            return (
-              <ModuleButton
-                key={engine.id}
-                module="engine"
-                discriminator={TIER_ROMAN_NUMERALS[engine.tier]}
-                selected={protagonist.engine.id === engine.id}
-                first={index === 0}
-                last={index === protagonist.tank.engines.length - 1}
-                rowChild
-                onClick={() => {
-                  mutateDuel((draft) => {
-                    draft.protagonist!.engine = engine;
-                  });
-                }}
-              />
-            );
-          })}
-        </Flex>
+          <DropdownMenu.Content>
+            <DropdownMenu.RadioGroup value={`${protagonist.turret.id}`}>
+              {[...protagonist.tank.turrets].reverse().map((turret) => {
+                return (
+                  <DropdownMenu.RadioItem
+                    key={turret.id}
+                    value={`${turret.id}`}
+                    onClick={() => {
+                      mutateDuel((draft) => {
+                        draft.protagonist!.turret = turret;
+                        draft.protagonist!.gun = turret.guns.at(-1)!;
+                        draft.protagonist!.shell =
+                          turret.guns.at(-1)!.shells[0];
+                      });
+                      mutateTankopediaTemporary((draft) => {
+                        draft.shot = undefined;
+                      });
+                      setTurretMenuOpen(false);
+                    }}
+                  >
+                    <ModuleButton
+                      variant="ghost"
+                      module="turret"
+                      discriminator={TIER_ROMAN_NUMERALS[turret.tier]}
+                    />{' '}
+                    {turret.name}
+                  </DropdownMenu.RadioItem>
+                );
+              })}
+            </DropdownMenu.RadioGroup>
+          </DropdownMenu.Content>
+        </DropdownMenu.Root>
 
-        <Flex>
-          {protagonist.tank.tracks.map((track, index) => {
-            return (
-              <ModuleButton
-                key={track.id}
-                module="chassis"
-                discriminator={TIER_ROMAN_NUMERALS[track.tier]}
-                selected={protagonist.track.id === track.id}
-                first={index === 0}
-                last={index === protagonist.tank.tracks.length - 1}
-                rowChild
-                onClick={() => {
-                  mutateDuel((draft) => {
-                    draft.protagonist!.track = track;
-                  });
-                }}
-              />
-            );
-          })}
-        </Flex>
+        <ModuleButton
+          module="gun"
+          discriminator={TIER_ROMAN_NUMERALS[protagonist.gun.tier]}
+          first
+          last
+          rowChild
+          onClick={() => setGunMenuOpen(true)}
+        />
+
+        <DropdownMenu.Root open={gunMenuOpen} onOpenChange={setGunMenuOpen}>
+          <DropdownMenu.Trigger>
+            <div />
+          </DropdownMenu.Trigger>
+
+          <DropdownMenu.Content>
+            <DropdownMenu.RadioGroup value={`${protagonist.gun.id}`}>
+              {[...protagonist.turret.guns].reverse().map((gun) => {
+                return (
+                  <DropdownMenu.RadioItem
+                    key={gun.id}
+                    value={`${gun.id}`}
+                    onClick={() => {
+                      mutateDuel((draft) => {
+                        draft.protagonist!.gun = gun;
+                        draft.protagonist!.shell = gun.shells[0];
+                      });
+                    }}
+                  >
+                    <ModuleButton
+                      variant="ghost"
+                      module="gun"
+                      discriminator={TIER_ROMAN_NUMERALS[gun.tier]}
+                    />{' '}
+                    {gun.name}
+                  </DropdownMenu.RadioItem>
+                );
+              })}
+            </DropdownMenu.RadioGroup>
+          </DropdownMenu.Content>
+        </DropdownMenu.Root>
+
+        <ModuleButton
+          module="engine"
+          discriminator={TIER_ROMAN_NUMERALS[protagonist.engine.tier]}
+          first
+          last
+          rowChild
+          onClick={() => setEngineMenuOpen(true)}
+        />
+
+        <DropdownMenu.Root
+          open={engineMenuOpen}
+          onOpenChange={setEngineMenuOpen}
+        >
+          <DropdownMenu.Trigger>
+            <div />
+          </DropdownMenu.Trigger>
+
+          <DropdownMenu.Content>
+            <DropdownMenu.RadioGroup value={`${protagonist.engine.id}`}>
+              {[...protagonist.tank.engines].reverse().map((engine) => {
+                return (
+                  <DropdownMenu.RadioItem
+                    key={engine.id}
+                    value={`${engine.id}`}
+                    onClick={() => {
+                      mutateDuel((draft) => {
+                        draft.protagonist!.engine = engine;
+                      });
+                    }}
+                  >
+                    <ModuleButton
+                      variant="ghost"
+                      module="engine"
+                      discriminator={TIER_ROMAN_NUMERALS[engine.tier]}
+                    />{' '}
+                    {engine.name}
+                  </DropdownMenu.RadioItem>
+                );
+              })}
+            </DropdownMenu.RadioGroup>
+          </DropdownMenu.Content>
+        </DropdownMenu.Root>
+
+        <ModuleButton
+          module="chassis"
+          discriminator={TIER_ROMAN_NUMERALS[protagonist.track.tier]}
+          first
+          last
+          rowChild
+          onClick={() => setTrackMenuOpen(true)}
+        />
+
+        <DropdownMenu.Root open={trackMenuOpen} onOpenChange={setTrackMenuOpen}>
+          <DropdownMenu.Trigger>
+            <div />
+          </DropdownMenu.Trigger>
+
+          <DropdownMenu.Content>
+            <DropdownMenu.RadioGroup value={`${protagonist.track.id}`}>
+              {[...protagonist.tank.tracks].reverse().map((track) => {
+                return (
+                  <DropdownMenu.RadioItem
+                    key={track.id}
+                    value={`${track.id}`}
+                    onClick={() => {
+                      mutateDuel((draft) => {
+                        draft.protagonist!.track = track;
+                      });
+                    }}
+                  >
+                    <ModuleButton
+                      variant="ghost"
+                      module="chassis"
+                      discriminator={TIER_ROMAN_NUMERALS[track.tier]}
+                    />{' '}
+                    {track.name}
+                  </DropdownMenu.RadioItem>
+                );
+              })}
+            </DropdownMenu.RadioGroup>
+          </DropdownMenu.Content>
+        </DropdownMenu.Root>
       </Flex>
     </ConfigurationChildWrapper>
   );
diff --git a/src/components/ModuleButtons/TankComponentButton.tsx b/src/components/ModuleButtons/TankComponentButton.tsx
index 676de80e7..95efbb659 100644
--- a/src/components/ModuleButtons/TankComponentButton.tsx
+++ b/src/components/ModuleButtons/TankComponentButton.tsx
@@ -15,6 +15,7 @@ export interface TankComponentButtonProps
 }
 
 export function TankComponentButton({
+  variant,
   selected,
   first = false,
   last = false,
@@ -23,13 +24,15 @@ export function TankComponentButton({
   special,
   children,
   banner,
+  disabled,
+  style,
   ...props
 }: TankComponentButtonProps) {
   return (
     <Button
       radius="medium"
       color={selected ? (special ? 'amber' : undefined) : 'gray'}
-      variant={selected ? 'surface' : 'soft'}
+      variant={variant ?? (selected ? 'surface' : 'soft')}
       style={{
         padding: 0,
         minWidth: 48,
@@ -41,7 +44,11 @@ export function TankComponentButton({
         borderBottomRightRadius: last ? undefined : 0,
         margin: rowChild ? -0.5 : 'unset',
         overflow: 'hidden',
+        cursor: 'unset',
+        pointerEvents: variant === 'ghost' ? 'none' : undefined,
+        ...style,
       }}
+      disabled={variant === 'ghost' || disabled}
       {...props}
     >
       {banner !== undefined && (
@@ -69,6 +76,7 @@ export function TankComponentButton({
             right: 8,
             textAlign: 'right',
           }}
+          color={variant === 'ghost' ? 'gray' : undefined}
         >
           {discriminator}
         </Text>
diff --git a/src/core/blitzkrieg/tankDefinitions/index.ts b/src/core/blitzkrieg/tankDefinitions/index.ts
index 47ac938d8..76ae491d4 100644
--- a/src/core/blitzkrieg/tankDefinitions/index.ts
+++ b/src/core/blitzkrieg/tankDefinitions/index.ts
@@ -55,6 +55,7 @@ export interface TankDefinition {
 export interface TrackDefinition {
   id: number;
   tier: Tier;
+  name: string;
   weight: number;
   traverseSpeed: number;
   xp?: number;