From 85d699dbfc1eb5d893c98d7081ef39ad2cf140f2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tr=C3=A8sAbhi?= <43380238+tresabhi@users.noreply.github.com>
Date: Sat, 6 Apr 2024 19:04:57 -0500
Subject: [PATCH] feat: provisions

---
 src/app/tools/compare/page.tsx                | 115 +++++++++++-------
 .../Characteristics/components/Provisions.tsx |  41 ++-----
 src/components/ProvisionsManager.tsx          |  49 ++++++++
 3 files changed, 132 insertions(+), 73 deletions(-)
 create mode 100644 src/components/ProvisionsManager.tsx

diff --git a/src/app/tools/compare/page.tsx b/src/app/tools/compare/page.tsx
index 4d095df7d..f39307c4b 100644
--- a/src/app/tools/compare/page.tsx
+++ b/src/app/tools/compare/page.tsx
@@ -23,7 +23,9 @@ import { usePathname, useRouter, useSearchParams } from 'next/navigation';
 import { use, useCallback, useEffect, useMemo, useState } from 'react';
 import { EquipmentManager } from '../../../components/EquipmentManager';
 import PageWrapper from '../../../components/PageWrapper';
+import { ProvisionsManager } from '../../../components/ProvisionsManager';
 import { asset } from '../../../core/blitzkrieg/asset';
+import { availableProvisions } from '../../../core/blitzkrieg/availableProvisions';
 import { equipmentDefinitions } from '../../../core/blitzkrieg/equipmentDefinitions';
 import { modelDefinitions } from '../../../core/blitzkrieg/modelDefinitions';
 import { provisionDefinitions } from '../../../core/blitzkrieg/provisionDefinitions';
@@ -439,9 +441,14 @@ export default function Page() {
                 <Table.Cell />
 
                 {members.map(
-                  ({ equipmentMatrix, tank, key, provisions }, index) => {
+                  ({ equipmentMatrix, tank, key, provisions, gun }, index) => {
                     const equipmentPreset =
                       awaitedEquipmentDefinitions.presets[tank.equipment];
+                    const provisionsList = availableProvisions(
+                      tank,
+                      gun,
+                      awaitedProvisionDefinitions,
+                    );
 
                     return (
                       <Table.Cell key={key}>
@@ -559,48 +566,70 @@ export default function Page() {
                             </Popover.Content>
                           </Popover.Root>
 
-                          <Button
-                            variant="ghost"
-                            radius="large"
-                            style={{
-                              height: '100%',
-                              width: 16,
-                              position: 'relative',
-                            }}
-                          >
-                            <Flex direction="column">
-                              {provisions.map((provision, index) => (
-                                <img
-                                  key={provision}
-                                  src={asset(
-                                    `/icons/provisions/${provision}.webp`,
-                                  )}
-                                  style={{
-                                    left: '50%',
-                                    /**
-                                     * max = 100% - 16px - 4px
-                                     * min = 4px
-                                     * diff = max - min
-                                     *      = 100% - 16px - 4px - 4px
-                                     *      = 100% - 24px
-                                     * coefficient = index / length - 1
-                                     * position = min + coefficient * diff
-                                     */
-                                    top: `calc(4px + ${
-                                      provisions.length === 1
-                                        ? 0.5
-                                        : index / (provisions.length - 1)
-                                    } * (100% - 24px))`,
-                                    transform: 'translateX(-50%)',
-                                    position: 'absolute',
-                                    width: 16,
-                                    height: 16,
-                                    objectFit: 'contain',
-                                  }}
-                                />
-                              ))}
-                            </Flex>
-                          </Button>
+                          <Popover.Root>
+                            <Popover.Trigger>
+                              <Button
+                                variant="ghost"
+                                radius="large"
+                                style={{
+                                  height: '100%',
+                                  width: 16,
+                                  position: 'relative',
+                                }}
+                              >
+                                <Flex direction="column">
+                                  {provisions.map((provision, index) => (
+                                    <img
+                                      key={provision}
+                                      src={asset(
+                                        `/icons/provisions/${provision}.webp`,
+                                      )}
+                                      style={{
+                                        left: '50%',
+                                        /**
+                                         * max = 100% - 16px - 4px
+                                         * min = 4px
+                                         * diff = max - min
+                                         *      = 100% - 16px - 4px - 4px
+                                         *      = 100% - 24px
+                                         * coefficient = index / length - 1
+                                         * position = min + coefficient * diff
+                                         */
+                                        top: `calc(4px + ${
+                                          provisions.length === 1
+                                            ? 0.5
+                                            : index / (provisions.length - 1)
+                                        } * (100% - 24px))`,
+                                        transform: 'translateX(-50%)',
+                                        position: 'absolute',
+                                        width: 16,
+                                        height: 16,
+                                        objectFit: 'contain',
+                                      }}
+                                    />
+                                  ))}
+                                </Flex>
+                              </Button>
+                            </Popover.Trigger>
+
+                            <Popover.Content>
+                              <ProvisionsManager
+                                provisions={provisionsList.map(
+                                  (provision) => provision.id,
+                                )}
+                                selected={provisions}
+                                disabled={
+                                  tank.provisions === provisionsList.length
+                                }
+                                onChange={(provisions) => {
+                                  mutateCompareTemporary((draft) => {
+                                    draft.members[index].provisions =
+                                      provisions;
+                                  });
+                                }}
+                              />
+                            </Popover.Content>
+                          </Popover.Root>
                         </Flex>
                       </Table.Cell>
                     );
diff --git a/src/app/tools/tankopedia/[id]/components/Characteristics/components/Provisions.tsx b/src/app/tools/tankopedia/[id]/components/Characteristics/components/Provisions.tsx
index daf7eafde..677bdb5ee 100644
--- a/src/app/tools/tankopedia/[id]/components/Characteristics/components/Provisions.tsx
+++ b/src/app/tools/tankopedia/[id]/components/Characteristics/components/Provisions.tsx
@@ -1,6 +1,6 @@
 import { Button, Flex, Heading } from '@radix-ui/themes';
 import { use } from 'react';
-import { ProvisionButton } from '../../../../../../../components/ModuleButtons/ProvisionButton';
+import { ProvisionsManager } from '../../../../../../../components/ProvisionsManager';
 import { availableProvisions } from '../../../../../../../core/blitzkrieg/availableProvisions';
 import { provisionDefinitions } from '../../../../../../../core/blitzkrieg/provisionDefinitions';
 import { mutateDuel, useDuel } from '../../../../../../../stores/duel';
@@ -33,35 +33,16 @@ export function Provisions() {
         </Button>
       </Flex>
 
-      <Flex wrap="wrap">
-        {provisionsList.map((provision, index) => {
-          const selected = provisions.includes(provision.id);
-
-          return (
-            <ProvisionButton
-              key={provision.id}
-              first={index === 0}
-              last={index === provisionsList.length - 1}
-              rowChild
-              disabled={tank.provisions === provisions.length && !selected}
-              provision={provision.id}
-              selected={selected}
-              onClick={() => {
-                mutateDuel((draft) => {
-                  if (selected) {
-                    draft.protagonist!.provisions =
-                      draft.protagonist!.provisions.filter(
-                        (id) => id !== provision.id,
-                      );
-                  } else {
-                    draft.protagonist!.provisions.push(provision.id);
-                  }
-                });
-              }}
-            />
-          );
-        })}
-      </Flex>
+      <ProvisionsManager
+        provisions={provisionsList.map((provision) => provision.id)}
+        selected={provisions}
+        disabled={tank.provisions === provisionsList.length}
+        onChange={(provisions) => {
+          mutateDuel((draft) => {
+            draft.protagonist!.provisions = provisions;
+          });
+        }}
+      />
     </ConfigurationChildWrapper>
   );
 }
diff --git a/src/components/ProvisionsManager.tsx b/src/components/ProvisionsManager.tsx
new file mode 100644
index 000000000..b8aa7bf8d
--- /dev/null
+++ b/src/components/ProvisionsManager.tsx
@@ -0,0 +1,49 @@
+import { Flex } from '@radix-ui/themes';
+import { ProvisionButton } from './ModuleButtons/ProvisionButton';
+
+interface ProvisionsManagerProps {
+  provisions: number[];
+  selected: number[];
+  disabled?: boolean;
+  onChange?: (provisions: number[]) => void;
+}
+
+export function ProvisionsManager({
+  provisions,
+  selected,
+  onChange,
+  disabled,
+}: ProvisionsManagerProps) {
+  return (
+    <Flex wrap="wrap">
+      {provisions.map((provision, index) => {
+        const isSelected = selected.includes(provision);
+
+        return (
+          <ProvisionButton
+            key={provision}
+            first={index === 0}
+            last={index === provisions.length - 1}
+            rowChild
+            disabled={disabled && !selected}
+            provision={provision}
+            selected={isSelected}
+            onClick={() => {
+              if (!onChange) return;
+
+              const draft = [...selected];
+
+              if (isSelected) {
+                draft.splice(draft.indexOf(provision), 1);
+              } else {
+                draft.push(provision);
+              }
+
+              onChange(draft);
+            }}
+          />
+        );
+      })}
+    </Flex>
+  );
+}