From f6b4bb39236beae6d1951b33640e181530ab08be Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tr=C3=A8sAbhi?= <43380238+tresabhi@users.noreply.github.com>
Date: Wed, 10 Jan 2024 16:25:59 -0600
Subject: [PATCH] feat: no armor unload

---
 .../tankopedia/[id]/components/Model/TankDisplay.tsx      | 2 +-
 .../[id]/components/Model/components/TankArmor.tsx        | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/app/tools/tankopedia/[id]/components/Model/TankDisplay.tsx b/src/app/tools/tankopedia/[id]/components/Model/TankDisplay.tsx
index b3fdfe27e..73183d7de 100644
--- a/src/app/tools/tankopedia/[id]/components/Model/TankDisplay.tsx
+++ b/src/app/tools/tankopedia/[id]/components/Model/TankDisplay.tsx
@@ -134,7 +134,7 @@ export function TankDisplay() {
                   }
                 >
                   <TankModel />
-                  {mode === 'armor' && controlsEnabled && <TankArmor />}
+                  {mode === 'armor' && <TankArmor visible={controlsEnabled} />}
                 </Suspense>
               </Canvas>
             </div>
diff --git a/src/app/tools/tankopedia/[id]/components/Model/components/TankArmor.tsx b/src/app/tools/tankopedia/[id]/components/Model/components/TankArmor.tsx
index 8b208c885..ff063f71a 100644
--- a/src/app/tools/tankopedia/[id]/components/Model/components/TankArmor.tsx
+++ b/src/app/tools/tankopedia/[id]/components/Model/components/TankArmor.tsx
@@ -1,4 +1,4 @@
-import { useLoader } from '@react-three/fiber';
+import { GroupProps, useLoader } from '@react-three/fiber';
 import { useEffect, useState } from 'react';
 import { Euler, Mesh, Vector3 } from 'three';
 import { GLTFLoader } from 'three-stdlib';
@@ -12,9 +12,9 @@ import {
 } from '../../../../../../../core/blitzkrieg/modelDefinitions';
 import { useTankopedia } from '../../../../../../../stores/tankopedia';
 
-interface TankArmorProps {}
+interface TankArmorProps extends GroupProps {}
 
-export function TankArmor() {
+export function TankArmor(props: TankArmorProps) {
   const [awaitedModelDefinitions, setAwaitedModelDefinitions] = useState<
     ModelDefinitions | undefined
   >(undefined);
@@ -81,7 +81,7 @@ export function TankArmor() {
 
   return (
     <HeadsUpDisplay>
-      <group rotation={[-Math.PI / 2, 0, model.hullYaw]}>
+      <group {...props} rotation={[-Math.PI / 2, 0, model.hullYaw]}>
         {nodes.map((node) => {
           const isHull = node.name.startsWith('hull_');
           const armorId = parseInt(node.name.match(/.+_armor_(\d+)/)![1]);