kjuulh/rhai
1
0
Fork 0
Code Issues 1 Pull Requests 1 Packages Projects Releases Wiki Activity

Reflect visibility to generated modules.

Browse Source
This commit is contained in:
Stephen Chung
2021-02-21 13:51:24 +08:00
parent 03c31a969a
commit fc7697d504
3 changed files with 19 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
codegen/src/module.rs
View File

@@ -257,6 +257,7 @@ impl Module {
params,
..
} = self;
let mod_vis = mod_all.vis;
let mod_name = mod_all.ident.clone();
let (_, orig_content) = mod_all.content.take().unwrap();
let mod_attrs = mem::take(&mut mod_all.attrs);
@@ -282,7 +283,7 @@ impl Module {
// Regenerate the module with the new content added.
Ok(quote! {
#(#mod_attrs)*
pub mod #mod_name {
#mod_vis mod #mod_name {
#(#orig_content)*
#(#inner_modules)*
#mod_gen
@@ -292,7 +293,7 @@ impl Module {
// Regenerate the original module as-is.
Ok(quote! {
#(#mod_attrs)*
pub mod #mod_name {
#mod_vis mod #mod_name {
#(#orig_content)*
}
})