From ad1ad8002622af3ea45166d5347f7d1c0cf7c85f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tr=C3=A8sAbhi?= <43380238+tresabhi@users.noreply.github.com>
Date: Thu, 25 Jan 2024 11:50:43 -0600
Subject: [PATCH] feat: consumable buttons

---
 .vscode/settings.json                         |  1 +
 scripts/buildAssets/definitions.ts            |  9 +++-
 .../tankopedia/[id]/components/Configure.tsx  | 33 +++++++++++++++
 src/components/ModuleButton.tsx               | 41 +++++++++++--------
 .../blitzkrieg/checkConsumableInclusivity.ts  | 40 ++++++++++++++++++
 src/core/blitzkrieg/consumablesDefinitions.ts | 20 +++++----
 src/stores/tankopedia.ts                      |  2 +
 7 files changed, 120 insertions(+), 26 deletions(-)
 create mode 100644 src/core/blitzkrieg/checkConsumableInclusivity.ts

diff --git a/.vscode/settings.json b/.vscode/settings.json
index 8de5aaa43..c5334eed5 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -46,6 +46,7 @@
     "glsl",
     "GLTF",
     "immer",
+    "Inclusivity",
     "JOINTINDEX",
     "JOINTWEIGHT",
     "kranvagn",
diff --git a/scripts/buildAssets/definitions.ts b/scripts/buildAssets/definitions.ts
index f02e7070b..eae2b8548 100644
--- a/scripts/buildAssets/definitions.ts
+++ b/scripts/buildAssets/definitions.ts
@@ -7,7 +7,10 @@ import { readYAML } from '../../src/core/blitz/readYAML';
 import { readYAMLDVPL } from '../../src/core/blitz/readYAMLDVPL';
 import { toUniqueId } from '../../src/core/blitz/toUniqueId';
 import { commitAssets } from '../../src/core/blitzkrieg/commitAssets';
-import { ConsumableDefinitions } from '../../src/core/blitzkrieg/consumablesDefinitions';
+import {
+  ConsumableDefinitionFilterCategory,
+  ConsumableDefinitions,
+} from '../../src/core/blitzkrieg/consumablesDefinitions';
 import {
   EquipmentDefinitions,
   EquipmentPreset,
@@ -666,7 +669,9 @@ export async function buildDefinitions(production: boolean) {
       } else if ('extendedTags' in excludeRaw) {
         consumableDefinitions[consumable.id].exclude!.push({
           type: 'category',
-          categories: excludeRaw.extendedTags.split(' '),
+          categories: excludeRaw.extendedTags.split(
+            ' ',
+          ) as ConsumableDefinitionFilterCategory[],
         });
       } else throw new SyntaxError('Unhandled exclude type');
     }
diff --git a/src/app/tools/tankopedia/[id]/components/Configure.tsx b/src/app/tools/tankopedia/[id]/components/Configure.tsx
index 1ae2712bb..48d1a6f3f 100644
--- a/src/app/tools/tankopedia/[id]/components/Configure.tsx
+++ b/src/app/tools/tankopedia/[id]/components/Configure.tsx
@@ -1,6 +1,8 @@
 import { Flex, Heading } from '@radix-ui/themes';
 import { use } from 'react';
 import { ModuleButton } from '../../../../../components/ModuleButton';
+import { checkConsumableInclusivity } from '../../../../../core/blitzkrieg/checkConsumableInclusivity';
+import { consumableDefinitions } from '../../../../../core/blitzkrieg/consumablesDefinitions';
 import { equipmentDefinitions } from '../../../../../core/blitzkrieg/equipmentDefinitions';
 import { TIER_ROMAN_NUMERALS } from '../../../../../core/blitzkrieg/tankDefinitions';
 import { mutateDuel, useDuel } from '../../../../../stores/duel';
@@ -12,11 +14,13 @@ import {
 export function Configure() {
   const protagonist = useDuel((state) => state.protagonist!);
   const awaitedEquipmentDefinitions = use(equipmentDefinitions);
+  const awaitedConsumableDefinitions = use(consumableDefinitions);
   const equipmentRows =
     awaitedEquipmentDefinitions.presets[protagonist.tank.equipment];
   const equipmentMatrix = useTankopediaTemporary(
     (state) => state.equipmentMatrix,
   );
+  const consumables = useTankopediaTemporary((state) => state.consumables);
 
   return (
     <Flex gap="4" direction="column">
@@ -118,6 +122,35 @@ export function Configure() {
 
         <Flex gap="2" direction="column">
           <Heading size="4">Consumables</Heading>
+
+          {Object.values(awaitedConsumableDefinitions)
+            .filter((consumable) =>
+              checkConsumableInclusivity(
+                consumable,
+                protagonist.tank,
+                protagonist.gun,
+              ),
+            )
+            .map((consumable) => (
+              <ModuleButton
+                first
+                last
+                type="consumable"
+                consumable={consumable.id}
+                selected={consumables.includes(consumable.id)}
+                onClick={() => {
+                  mutateTankopediaTemporary((draft) => {
+                    if (draft.consumables.includes(consumable.id)) {
+                      draft.consumables = draft.consumables.filter(
+                        (id) => id !== consumable.id,
+                      );
+                    } else {
+                      draft.consumables.push(consumable.id);
+                    }
+                  });
+                }}
+              />
+            ))}
         </Flex>
       </Flex>
 
diff --git a/src/components/ModuleButton.tsx b/src/components/ModuleButton.tsx
index 024289bd2..a0f90c8c3 100644
--- a/src/components/ModuleButton.tsx
+++ b/src/components/ModuleButton.tsx
@@ -1,5 +1,5 @@
 import { Button, Text } from '@radix-ui/themes';
-import { ComponentProps } from 'react';
+import { CSSProperties, ComponentProps } from 'react';
 import { asset } from '../core/blitzkrieg/asset';
 
 type ModuleButtonProps = Omit<ComponentProps<typeof Button>, 'type'> & {
@@ -21,6 +21,10 @@ type ModuleButtonProps = Omit<ComponentProps<typeof Button>, 'type'> & {
         type: 'equipment';
         equipment: number;
       }
+    | {
+        type: 'consumable';
+        consumable: number;
+      }
   );
 
 export function ModuleButton({
@@ -30,10 +34,26 @@ export function ModuleButton({
   last = false,
   rowChild,
   discriminator,
+  disabled,
   ...props
 }: ModuleButtonProps) {
+  const imageStyles: CSSProperties = {
+    width: 32,
+    height: 32,
+    position: 'absolute',
+    opacity: disabled ? 0.5 : 1,
+    transform: 'scale(0.75)',
+  };
+
+  if (props.type === 'module') {
+    imageStyles.top = '50%';
+    imageStyles.left = '50%';
+    imageStyles.transform = 'translate(calc(-50% + 2px), calc(-50% + 2px))';
+  }
+
   return (
     <Button
+      disabled={disabled}
       radius="medium"
       color={selected ? undefined : 'gray'}
       variant={selected ? 'surface' : 'soft'}
@@ -57,22 +77,11 @@ export function ModuleButton({
             ? `icons/modules/${props.module}.webp`
             : props.type === 'shell'
               ? `icons/shells/${props.shell}.webp`
-              : `icons/equipment/${props.equipment}.webp`,
+              : props.type === 'consumable'
+                ? `icons/consumables/${props.consumable}.webp`
+                : `icons/equipment/${props.equipment}.webp`,
         )}
-        style={{
-          width: 32,
-          height: 32,
-          position: 'absolute',
-          ...(props.type === 'module'
-            ? {
-                top: '50%',
-                left: '50%',
-                transform: 'translate(calc(-50% + 2px), calc(-50% + 2px))',
-              }
-            : {
-                transform: 'scale(0.75)',
-              }),
-        }}
+        style={imageStyles}
       />
       {discriminator !== undefined && (
         <Text
diff --git a/src/core/blitzkrieg/checkConsumableInclusivity.ts b/src/core/blitzkrieg/checkConsumableInclusivity.ts
new file mode 100644
index 000000000..02f8175e2
--- /dev/null
+++ b/src/core/blitzkrieg/checkConsumableInclusivity.ts
@@ -0,0 +1,40 @@
+import { ConsumableEntry } from './consumablesDefinitions';
+import { GunDefinition, TankDefinition } from './tankDefinitions';
+
+export function checkConsumableInclusivity(
+  consumable: ConsumableEntry,
+  tank: TankDefinition,
+  gun: GunDefinition,
+) {
+  const included = consumable.include.every((rule) => {
+    switch (rule.type) {
+      case 'tier':
+        return rule.min <= tank.tier && tank.tier <= rule.max;
+
+      case 'ids':
+        return rule.ids.includes(tank.id);
+
+      case 'category':
+        throw new SyntaxError('Category filtering found in include rule');
+    }
+  });
+  const excluded = consumable.exclude?.some((rule) => {
+    switch (rule.type) {
+      case 'tier':
+        return rule.min <= tank.tier && tank.tier <= rule.max;
+
+      case 'ids':
+        return rule.ids.includes(tank.id);
+
+      case 'category':
+        return rule.categories.some((category) => {
+          switch (category) {
+            case 'clip':
+              return gun.type !== 'regular';
+          }
+        });
+    }
+  });
+
+  return included && !excluded;
+}
diff --git a/src/core/blitzkrieg/consumablesDefinitions.ts b/src/core/blitzkrieg/consumablesDefinitions.ts
index ca2200c96..fb5d8b7d4 100644
--- a/src/core/blitzkrieg/consumablesDefinitions.ts
+++ b/src/core/blitzkrieg/consumablesDefinitions.ts
@@ -1,5 +1,7 @@
 import { asset } from './asset';
 
+export type ConsumableDefinitionFilterCategory = 'clip';
+
 type ConsumableDefinitionFilter =
   | {
       type: 'tier';
@@ -12,17 +14,19 @@ type ConsumableDefinitionFilter =
     }
   | {
       type: 'category';
-      categories: string[];
+      categories: ConsumableDefinitionFilterCategory[];
     };
 
+export interface ConsumableEntry {
+  id: number;
+  name: string;
+  effects?: Record<string, number>;
+  include: ConsumableDefinitionFilter[];
+  exclude?: ConsumableDefinitionFilter[];
+}
+
 export interface ConsumableDefinitions {
-  [key: string]: {
-    id: number;
-    name: string;
-    effects?: Record<string, number>;
-    include: ConsumableDefinitionFilter[];
-    exclude?: ConsumableDefinitionFilter[];
-  };
+  [key: string]: ConsumableEntry;
 }
 
 export const consumableDefinitions = fetch(
diff --git a/src/stores/tankopedia.ts b/src/stores/tankopedia.ts
index 53be5d307..ef56760d5 100644
--- a/src/stores/tankopedia.ts
+++ b/src/stores/tankopedia.ts
@@ -57,6 +57,7 @@ interface TankopediaTemporary {
   };
   equipmentMatrix: EquipmentMatrix;
   mode: TankopediaMode;
+  consumables: number[];
 }
 
 export const useTankopediaPersistent = create<TankopediaPersistent>()(
@@ -104,6 +105,7 @@ export const useTankopediaTemporary = create<TankopediaTemporary>()(
       [0, 1, 0],
     ],
     mode: 'model',
+    consumables: [],
   })),
 );