Async tasks can now take arguments at spawn again

1 files changed, 10 insertions, 8 deletions

diff --git a/macros/src/syntax/parse/idle.rs b/macros/src/syntax/parse/idle.rs
index f049cca..124c136 100644
--- a/macros/src/syntax/parse/idle.rs
+++ b/macros/src/syntax/parse/idle.rs
@@ -21,14 +21,16 @@ impl Idle {
         let name = item.sig.ident.to_string();
 
         if valid_signature {
-            if let Some(context) = util::parse_inputs(item.sig.inputs, &name) {
-                return Ok(Idle {
-                    args,
-                    attrs: item.attrs,
-                    context,
-                    name: item.sig.ident,
-                    stmts: item.block.stmts,
-                });
+            if let Some((context, Ok(rest))) = util::parse_inputs(item.sig.inputs, &name) {
+                if rest.is_empty() {
+                    return Ok(Idle {
+                        args,
+                        attrs: item.attrs,
+                        context,
+                        name: item.sig.ident,
+                        stmts: item.block.stmts,
+                    });
+                }
             }
         }