From 6d252785e83218eeb5d080836281c90b86ca0e03 Mon Sep 17 00:00:00 2001
From: Emil Fresk <emil.fresk@gmail.com>
Date: Sun, 8 Jan 2023 21:10:06 +0100
Subject: Support 0 prio tasks

---
 macros/src/codegen/main.rs | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

(limited to 'macros/src/codegen/main.rs')

diff --git a/macros/src/codegen/main.rs b/macros/src/codegen/main.rs
index 90f09ae..8e7138f 100644
--- a/macros/src/codegen/main.rs
+++ b/macros/src/codegen/main.rs
@@ -16,11 +16,14 @@ pub fn codegen(app: &App, analysis: &Analysis) -> TokenStream2 {
         let name = &idle.name;
         quote!(#name(#name::Context::new()))
     } else {
-        // TODO: No idle defined, check for 0-priority tasks and generate an executor if needed
-
-        quote!(loop {
-            rtic::export::nop()
-        })
+        if analysis.channels.get(&0).is_some() {
+            let dispatcher = util::zero_prio_dispatcher_ident();
+            quote!(#dispatcher();)
+        } else {
+            quote!(loop {
+                rtic::export::nop()
+            })
+        }
     };
 
     let main = util::suffixed("main");
-- 
cgit v1.2.3