feat: module selection

Co-authored-by: Copilot <[email protected]>

packages/website-ue/src/core/tankopedia/computeCharacteristics.ts

@@ -1,4 +1,4 @@
-import type { Tank } from "@blitzkit/closed/protos/blitzkit/tank";
+import type { Tank } from "@protos/blitzkit/tank";
 import {
   ShellUpgradeSingleChange_AttributeName,
   TankAttributeChange_AttributeName,

packages/website-ue/src/core/tankopedia/createTankState.ts

@@ -1,16 +1,26 @@
-import type { TankCatalogComponent } from "@protos/blitz_static_tank_component";
-import { TerrainHardness } from "./tankState";
+import type { Tank } from "@protos/blitzkit/tank";
+import { TerrainHardness, type TankState } from "./tankState";
+
+export function createTankState({ tank, id }: Tank) {
+  const upgrades: Record<string, number> = {};
+
+  for (const upgradeLine of tank!.upgrade_lines) {
+    upgrades[upgradeLine.name] = upgradeLine.stages.length - 1;
+  }
 
-export function createTankState(tank: TankCatalogComponent) {
   return {
-    stage: tank.upgrade_stages.length,
+    id,
+
+    upgrades,
+
     shell: 0,
-    isGunDamaged: false,
-    isHullTraversing: false,
-    isShooting: false,
-    isTurretTraversing: false,
     speed: 0,
     terrainHardness: TerrainHardness.Hard,
+
+    isShooting: false,
+    isGunDamaged: false,
+    isTurretTraversing: false,
+    isHullTraversing: false,
     isCaughtOnFire: false,
-  };
+  } satisfies TankState;
 }

packages/website-ue/src/core/tankopedia/tankState.ts

@@ -5,13 +5,17 @@ export enum TerrainHardness {
 }
 
 export interface TankState {
-  stage: number;
+  id: string;
+
+  upgrades: Record<string, number>;
+
   shell: number;
   speed: number;
-  isHullTraversing: boolean;
-  isTurretTraversing: boolean;
+  terrainHardness: TerrainHardness;
+
   isShooting: boolean;
   isGunDamaged: boolean;
-  terrainHardness: TerrainHardness;
+  isTurretTraversing: boolean;
+  isHullTraversing: boolean;
   isCaughtOnFire: boolean;
 }