properly handle `#[cfg]` (conditional compilation) on tasks

1 files changed, 4 insertions, 1 deletions

diff --git a/macros/src/syntax.rs b/macros/src/syntax.rs
index 0e6c606..b9424fb 100644
--- a/macros/src/syntax.rs
+++ b/macros/src/syntax.rs
@@ -1047,6 +1047,7 @@ impl Static {
 
 pub struct Task {
     pub args: TaskArgs,
+    pub cfgs: Vec<Attribute>,
     pub attrs: Vec<Attribute>,
     pub unsafety: Option<Token![unsafe]>,
     pub inputs: Vec<ArgCaptured>,
@@ -1098,9 +1099,11 @@ impl Task {
             _ => {}
         }
 
+        let (cfgs, attrs) = extract_cfgs(item.attrs);
         Ok(Task {
             args,
-            attrs: item.attrs,
+            cfgs,
+            attrs,
             unsafety: item.unsafety,
             inputs,
             statics: Static::parse(statics)?,