Make nested immediate loads also return the handle to the asset.

Author: andriyDev

crates/bevy_asset/src/lib.rs

@@ -719,7 +719,7 @@ mod tests {
             let mut ron: CoolTextRon = ron::de::from_bytes(&bytes)?;
             let mut embedded = String::new();
             for dep in ron.embedded_dependencies {
-                let loaded = load_context
+                let (_, loaded) = load_context
                     .loader()
                     .immediate()
                     .load::<CoolText>(&dep)

crates/bevy_asset/src/loader_builders.rs

@@ -386,23 +386,29 @@ impl<'builder, 'reader, T> NestedLoader<'_, 'builder, T, Immediate<'builder, 're
         self,
         path: &AssetPath<'static>,
         asset_type_id: Option<TypeId>,
-    ) -> Result<(Arc<dyn ErasedAssetLoader>, NestedErasedAssetRef<'builder>), LoadDirectError> {
+    ) -> Result<
+        (
+            UntypedHandle,
+            Arc<dyn ErasedAssetLoader>,
+            NestedErasedAssetRef<'builder>,
+        ),
+        LoadDirectError,
+    > {
         if path.label().is_some() {
             return Err(LoadDirectError::RequestedSubasset(path.clone()));
         }
+        let asset_server = self.load_context.asset_server.clone();
         let (mut meta, loader, mut reader) = if let Some(reader) = self.mode.reader {
             let loader = if let Some(asset_type_id) = asset_type_id {
-                self.load_context
-                    .asset_server
+                asset_server
                     .get_asset_loader_with_asset_type_id(asset_type_id)
                     .await
                     .map_err(|error| LoadDirectError::LoadError {
                         dependency: path.clone(),
                         error: error.into(),
                     })?
             } else {
-                self.load_context
-                    .asset_server
+                asset_server
                     .get_path_asset_loader(path)
                     .await
                     .map_err(|error| LoadDirectError::LoadError {
@@ -413,9 +419,7 @@ impl<'builder, 'reader, T> NestedLoader<'_, 'builder, T, Immediate<'builder, 're
             let meta = loader.default_meta();
             (meta, loader, ReaderRef::Borrowed(reader))
         } else {
-            let (meta, loader, reader) = self
-                .load_context
-                .asset_server
+            let (meta, loader, reader) = asset_server
                 .get_meta_loader_and_reader(path, asset_type_id)
                 .await
                 .map_err(|error| LoadDirectError::LoadError {
@@ -433,7 +437,13 @@ impl<'builder, 'reader, T> NestedLoader<'_, 'builder, T, Immediate<'builder, 're
             .load_context
             .load_direct_internal(path.clone(), meta.as_ref(), &*loader, reader.as_mut())
             .await?;
-        Ok((loader, asset))
+
+        let handle = asset_server.get_or_create_path_handle_erased(
+            path,
+            asset.get_asset().asset_type_id(),
+            None,
+        );
+        Ok((handle, loader, asset))
     }
 }
 
@@ -451,11 +461,11 @@ impl<'builder> NestedLoader<'_, 'builder, StaticTyped, Immediate<'builder, '_>>
     pub async fn load<'p, A: Asset>(
         self,
         path: impl Into<AssetPath<'p>>,
-    ) -> Result<NestedAssetRef<'builder, A>, LoadDirectError> {
+    ) -> Result<(Handle<A>, NestedAssetRef<'builder, A>), LoadDirectError> {
         let path = path.into().into_owned();
         self.load_internal(&path, Some(TypeId::of::<A>()))
             .await
-            .and_then(move |(loader, untyped_asset)| {
+            .and_then(move |(handle, loader, untyped_asset)| {
                 untyped_asset
                     .downcast::<A>()
                     .map_err(|_| LoadDirectError::LoadError {
@@ -467,6 +477,7 @@ impl<'builder> NestedLoader<'_, 'builder, StaticTyped, Immediate<'builder, '_>>
                             loader_name: loader.type_name(),
                         },
                     })
+                    .map(|asset| (handle.typed(), asset))
             })
     }
 }
@@ -481,12 +492,12 @@ impl<'builder> NestedLoader<'_, 'builder, DynamicTyped, Immediate<'builder, '_>>
     pub async fn load(
         self,
         path: impl Into<AssetPath<'_>>,
-    ) -> Result<NestedErasedAssetRef<'builder>, LoadDirectError> {
+    ) -> Result<(UntypedHandle, NestedErasedAssetRef<'builder>), LoadDirectError> {
         let path = path.into().into_owned();
         let asset_type_id = Some(self.typing.asset_type_id);
         self.load_internal(&path, asset_type_id)
             .await
-            .map(|(_, asset)| asset)
+            .map(|(handle, _, asset)| (handle, asset))
     }
 }
 
@@ -497,10 +508,10 @@ impl<'builder> NestedLoader<'_, 'builder, UnknownTyped, Immediate<'builder, '_>>
     pub async fn load(
         self,
         path: impl Into<AssetPath<'_>>,
-    ) -> Result<NestedErasedAssetRef<'builder>, LoadDirectError> {
+    ) -> Result<(UntypedHandle, NestedErasedAssetRef<'builder>), LoadDirectError> {
         let path = path.into().into_owned();
         self.load_internal(&path, None)
             .await
-            .map(|(_, asset)| asset)
+            .map(|(handle, _, asset)| (handle, asset))
     }
 }