fix bug

ari-party · Jul 3, 2024 · 420aa3a · 420aa3a
1 parent 58ab58d
commit 420aa3a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/hooks/data/useProjectile.ts b/src/hooks/data/useProjectile.ts
@@ -5,9 +5,13 @@ import type { Projectile } from '@/config/projectiles';
 
 export default function useProjectile(): Projectile {
   const projectileData = useDataStore((s) => s.projectile);
+  const setProjectile = useDataStore((s) => s.setProjectile);
 
-  const projectile =
-    guns[projectileData.gunKey].projectiles[projectileData.index];
+  const gun = guns[projectileData.gunKey];
+  if (!gun) setProjectile(Object.keys(guns)[0], 0);
+
+  const projectile = gun.projectiles[projectileData.index];
+  if (!projectile) setProjectile(projectileData.gunKey, 0);
 
   return projectile;
 }