From a7653cc05092aa5b009172c531e4f729c678a858 Mon Sep 17 00:00:00 2001
From: Vixu <lonevixu@gmail.com>
Date: Thu, 8 Jun 2023 13:44:49 +0200
Subject: allow init and idle to be externed

---
 rtic-macros/src/codegen/idle.rs | 24 ++++++++++++++----------
 1 file changed, 14 insertions(+), 10 deletions(-)

(limited to 'rtic-macros/src/codegen/idle.rs')

diff --git a/rtic-macros/src/codegen/idle.rs b/rtic-macros/src/codegen/idle.rs
index 0c833ef..9e608cb 100644
--- a/rtic-macros/src/codegen/idle.rs
+++ b/rtic-macros/src/codegen/idle.rs
@@ -34,16 +34,20 @@ pub fn codegen(app: &App, analysis: &Analysis) -> TokenStream2 {
         let attrs = &idle.attrs;
         let context = &idle.context;
         let stmts = &idle.stmts;
-        let user_idle = Some(quote!(
-            #(#attrs)*
-            #[allow(non_snake_case)]
-            fn #name(#context: #name::Context) -> ! {
-                use rtic::Mutex as _;
-                use rtic::mutex::prelude::*;
-
-                #(#stmts)*
-            }
-        ));
+        let user_idle = if !idle.is_extern {
+            Some(quote!(
+                #(#attrs)*
+                #[allow(non_snake_case)]
+                fn #name(#context: #name::Context) -> ! {
+                    use rtic::Mutex as _;
+                    use rtic::mutex::prelude::*;
+
+                    #(#stmts)*
+                }
+            ))
+        } else {
+            None
+        };
 
         quote!(
             #(#mod_app)*
-- 
cgit v1.2.3