blitzkit/packages/website/src/hooks/useArmor.ts

import { asset } from '@blitzkit/core';
import { type ObjectMap, useLoader } from '@react-three/fiber';
import { type GLTF, GLTFLoader } from 'three-stdlib';

const cache: Record<
  number,
  { gltf: GLTF & ObjectMap; hasDynamicArmor: boolean }
> = {};

export function useArmor(id: number) {
  const gltf = useLoader(GLTFLoader, asset(`3d/tanks/armor/${id}.glb`));

  if (!cache[id]) {
    cache[id] = {
      gltf,
      hasDynamicArmor: Object.values(gltf.nodes).some(
        (node) =>
          node.name.includes('state_00') || node.name.includes('state_01'),
      ),
    };
  }
  return cache[id];
}