From d6d58b0eb88242cf63724e1420bd29f8a4489916 Mon Sep 17 00:00:00 2001
From: Emil Fresk <emil.fresk@gmail.com>
Date: Tue, 10 Jan 2023 21:03:10 +0100
Subject: Async tasks can now take arguments at spawn again

---
 macros/src/syntax/parse/init.rs | 22 ++++++++++++----------
 1 file changed, 12 insertions(+), 10 deletions(-)

(limited to 'macros/src/syntax/parse/init.rs')

diff --git a/macros/src/syntax/parse/init.rs b/macros/src/syntax/parse/init.rs
index 23130c8..0aea20b 100644
--- a/macros/src/syntax/parse/init.rs
+++ b/macros/src/syntax/parse/init.rs
@@ -25,16 +25,18 @@ impl Init {
             if let Ok((user_shared_struct, user_local_struct)) =
                 util::type_is_init_return(&item.sig.output)
             {
-                if let Some(context) = util::parse_inputs(item.sig.inputs, &name) {
-                    return Ok(Init {
-                        args,
-                        attrs: item.attrs,
-                        context,
-                        name: item.sig.ident,
-                        stmts: item.block.stmts,
-                        user_shared_struct,
-                        user_local_struct,
-                    });
+                if let Some((context, Ok(rest))) = util::parse_inputs(item.sig.inputs, &name) {
+                    if rest.is_empty() {
+                        return Ok(Init {
+                            args,
+                            attrs: item.attrs,
+                            context,
+                            name: item.sig.ident,
+                            stmts: item.block.stmts,
+                            user_shared_struct,
+                            user_local_struct,
+                        });
+                    }
                 }
             }
         }
-- 
cgit v1.2.3