Merge #416

416: Move entry-point main into a separate module r=korken89 a=AfoHT

Prevents conflict with user provided tasks named main

Curious what the full test suite will make of this

Co-authored-by: Henrik Tjäder <henrik@tjaders.com>

2 files changed, 37 insertions, 8 deletions

diff --git a/examples/task_named_main.rs b/examples/task_named_main.rs
new file mode 100644
index 0000000..c3d21b5
--- /dev/null
+++ b/examples/task_named_main.rs
@@ -0,0 +1,26 @@
+//! examples/task_named_main.rs
+
+#![deny(unsafe_code)]
+#![deny(warnings)]
+#![no_main]
+#![no_std]
+
+use panic_semihosting as _;
+
+#[rtic::app(device = lm3s6965, dispatchers = [SSI0])]
+mod app {
+    use cortex_m_semihosting::{debug, hprintln};
+
+    #[init]
+    fn init(_: init::Context) -> init::LateResources {
+        main::spawn().unwrap();
+
+        init::LateResources {}
+    }
+
+    #[task]
+    fn main(_: main::Context) {
+        hprintln!("This task is named main, useful for rust-analyzer").unwrap();
+        debug::exit(debug::EXIT_SUCCESS);
+    }
+}
diff --git a/macros/src/codegen.rs b/macros/src/codegen.rs
index 8309473..3cddf57 100644
--- a/macros/src/codegen.rs
+++ b/macros/src/codegen.rs
@@ -57,19 +57,22 @@ pub fn app(app: &App, analysis: &Analysis, extra: &Extra) -> TokenStream2 {
 
     let main = util::suffixed("main");
     mains.push(quote!(
-        #[no_mangle]
-        unsafe extern "C" fn #main() -> ! {
-            let _TODO: () = ();
+        mod rtic_ext {
+            use super::*;
+            #[no_mangle]
+            unsafe extern "C" fn #main() -> ! {
+                let _TODO: () = ();
 
-            #(#assertion_stmts)*
+                #(#assertion_stmts)*
 
-            #(#pre_init_stmts)*
+                #(#pre_init_stmts)*
 
-            #call_init
+                #call_init
 
-            #(#post_init_stmts)*
+                #(#post_init_stmts)*
 
-            #call_idle
+                #call_idle
+            }
         }
     ));