From 4d85aab3e627f8f5b457aa63d183c3af46b4006d Mon Sep 17 00:00:00 2001
From: Niklas Eicker <git@nikl.me>
Date: Thu, 29 Feb 2024 11:22:51 +0100
Subject: [PATCH] Move to configured failure state when dynamic asset file
 fails loading

---
 .../loading_state/dynamic_asset_systems.rs    | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/bevy_asset_loader/src/loading_state/dynamic_asset_systems.rs b/bevy_asset_loader/src/loading_state/dynamic_asset_systems.rs
index 2533865..0e644c0 100644
--- a/bevy_asset_loader/src/loading_state/dynamic_asset_systems.rs
+++ b/bevy_asset_loader/src/loading_state/dynamic_asset_systems.rs
@@ -65,9 +65,17 @@ pub(crate) fn check_dynamic_asset_collections<S: States, C: DynamicAssetCollecti
         if loading_collections.is_none() {
             return;
         }
+        let config = asset_loader_config
+            .state_configurations
+            .get_mut(state.get())
+            .expect("No asset loader configuration for current state");
         let loading_collections = loading_collections.as_mut().unwrap();
         for handle in &loading_collections.handles {
             if let Some(load_state) = asset_server.get_load_state(handle.id()) {
+                if load_state == LoadState::Failed {
+                    config.loading_failed = true;
+                    continue;
+                }
                 if load_state != LoadState::Loaded {
                     return;
                 }
@@ -76,15 +84,10 @@ pub(crate) fn check_dynamic_asset_collections<S: States, C: DynamicAssetCollecti
             }
         }
         for collection in loading_collections.handles.drain(..) {
-            let collection = dynamic_asset_collections
-                .get(collection.typed::<C>())
-                .unwrap();
-            collection.register(&mut asset_keys);
+            if let Some(collection) = dynamic_asset_collections.get(collection.typed::<C>()) {
+                collection.register(&mut asset_keys);
+            }
         }
-        let config = asset_loader_config
-            .state_configurations
-            .get_mut(state.get())
-            .expect("No asset loader configuration for current state");
         config
             .loading_dynamic_collections
             .remove(&TypeId::of::<C>());