fix `idle::Resources::new`

it assumed that all resources were "early" resources

2 files changed, 13 insertions, 2 deletions

diff --git a/examples/late-resources.rs b/examples/late-resources.rs
index b5dba146..b60c7740 100644
--- a/examples/late-resources.rs
+++ b/examples/late-resources.rs
@@ -32,6 +32,11 @@ app! {
         static PORT: u16;
     },
 
+    idle: {
+        // Test that late resources can be used in idle
+        resources: [IP_ADDRESS],
+    }
+
     tasks: {
         SYS_TICK: {
             priority: 1,
diff --git a/macros/src/trans.rs b/macros/src/trans.rs
index 45841e73..96ff770b 100644
--- a/macros/src/trans.rs
+++ b/macros/src/trans.rs
@@ -74,8 +74,14 @@ fn idle(
                     });
 
                     let _name = Ident::new(format!("_{}", name.as_ref()));
-                    rexprs.push(quote! {
-                        #name: &mut #super_::#_name,
+                    rexprs.push(if resource.expr.is_some() {
+                        quote! {
+                            #name: &mut #super_::#_name,
+                        }
+                    } else {
+                        quote! {
+                            #name: #super_::#_name.as_mut(),
+                        }
                     });
                 } else {
                     rfields.push(quote! {