support for module comments

bevyengine · Oct 10, 2024 · 6330223 · 6330223
1 parent 86b89c8
commit 6330223
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/compose/mod.rs b/src/compose/mod.rs
@@ -690,7 +690,6 @@ impl Composer {
                     }
                 })?,
         };
-
         Ok(IrBuildResult {
             module,
             start_offset,
@@ -964,7 +963,6 @@ impl Composer {
             module_definition.name,
             source.len()
         );
-
         let IrBuildResult {
             module: mut source_ir,
             start_offset,
@@ -1164,7 +1162,6 @@ impl Composer {
         }
         // Copy comments, should be done after each commented item has been imported.
         module_builder.import_comments(&source_ir.comments);
-
         let module_ir = module_builder.into_module_with_entrypoints();
         let mut header_ir: naga::Module = header_builder.into();
 

diff --git a/src/derive.rs b/src/derive.rs
@@ -817,7 +817,9 @@ impl<'a> DerivedModule<'a> {
             constants,
             global_variables,
             struct_members,
+            module,
         } = comments;
+        self.comments.module = module.clone();
         for comment in types.iter() {
             if let Some(new_handle) = self.type_map.get(comment.0) {
                 self.comments.types.insert(*new_handle, comment.1.clone());