From db6d181f782de95d21ed423a94a747afe082381d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tr=C3=A8sAbhi?= <43380238+tresabhi@users.noreply.github.com>
Date: Sat, 20 Dec 2025 09:59:26 -0600
Subject: [PATCH] feat: use max stage

---
 packages/website-ue/src/pages/tanks/[slug]/_index.tsx | 2 ++
 packages/website-ue/src/stores/tankopedia.ts          | 6 +++---
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/packages/website-ue/src/pages/tanks/[slug]/_index.tsx b/packages/website-ue/src/pages/tanks/[slug]/_index.tsx
index b8b82d202..46f56c86a 100644
--- a/packages/website-ue/src/pages/tanks/[slug]/_index.tsx
+++ b/packages/website-ue/src/pages/tanks/[slug]/_index.tsx
@@ -18,6 +18,8 @@ export function Page({ id }: { id: string }) {
 function Content({ id }: { id: string }) {
   const tank = useAwait(() => api.tank(id), `tank-${id}`);
 
+  Tankopedia.useInitialization(tank.upgrade_stages.length);
+
   const protagonist = Tankopedia.use((state) => state.protagonist);
   const parameters = useMemo(
     () =>
diff --git a/packages/website-ue/src/stores/tankopedia.ts b/packages/website-ue/src/stores/tankopedia.ts
index 29e5a68d1..80b7740eb 100644
--- a/packages/website-ue/src/stores/tankopedia.ts
+++ b/packages/website-ue/src/stores/tankopedia.ts
@@ -5,6 +5,6 @@ interface Tankopedia {
   protagonist: TankState;
 }
 
-export const Tankopedia = new Varuna<Tankopedia>({
-  protagonist: { stage: 0, shell: 0 },
-});
+export const Tankopedia = new Varuna<Tankopedia, number>((maxStage) => ({
+  protagonist: { stage: maxStage, shell: 0 },
+}));