import { ObjectMap, useLoader } from '@react-three/fiber';
import { Mesh, MeshStandardMaterial } from 'three';
import { GLTF, GLTFLoader } from 'three-stdlib';
import { asset } from '../core/blitzkrieg/asset';

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

export function useModel(id: number) {
  const gltf = useLoader(GLTFLoader, asset(`3d/tanks/models/${id}.glb`));
  if (!cache[id]) {
    cache[id] = {
      gltf,
      hasPbr: Object.values(gltf.nodes).some(
        (node) =>
          node instanceof Mesh &&
          node.material instanceof MeshStandardMaterial &&
          node.material.roughnessMap !== null,
      ),
    };
  }
  return cache[id];
}