aboutsummaryrefslogtreecommitdiff
path: root/rtic-macros/src/codegen/init.rs
diff options
context:
space:
mode:
Diffstat (limited to 'rtic-macros/src/codegen/init.rs')
-rw-r--r--rtic-macros/src/codegen/init.rs12
1 files changed, 8 insertions, 4 deletions
diff --git a/rtic-macros/src/codegen/init.rs b/rtic-macros/src/codegen/init.rs
index 6e1059f7..b667ae0f 100644
--- a/rtic-macros/src/codegen/init.rs
+++ b/rtic-macros/src/codegen/init.rs
@@ -18,7 +18,9 @@ pub fn codegen(app: &App, analysis: &Analysis) -> TokenStream2 {
let attrs = &init.attrs;
let stmts = &init.stmts;
let shared = &init.user_shared_struct;
+ let shared_vis = &app.shared_resources_vis;
let local = &init.user_local_struct;
+ let local_vis = &app.local_resources_vis;
let shared_resources: Vec<_> = app
.shared_resources
@@ -27,10 +29,11 @@ pub fn codegen(app: &App, analysis: &Analysis) -> TokenStream2 {
let ty = &v.ty;
let cfgs = &v.cfgs;
let docs = &v.docs;
+ let vis = &v.vis;
quote!(
#(#cfgs)*
#(#docs)*
- #k: #ty,
+ #vis #k: #ty,
)
})
.collect();
@@ -41,20 +44,21 @@ pub fn codegen(app: &App, analysis: &Analysis) -> TokenStream2 {
let ty = &v.ty;
let cfgs = &v.cfgs;
let docs = &v.docs;
+ let vis = &v.vis;
quote!(
#(#cfgs)*
#(#docs)*
- #k: #ty,
+ #vis #k: #ty,
)
})
.collect();
root_init.push(quote! {
- struct #shared {
+ #shared_vis struct #shared {
#(#shared_resources)*
}
- struct #local {
+ #local_vis struct #local {
#(#local_resources)*
}
});