fix(blueprint-proc-macro): set tokio crate in sdk::main

So that blueprints don't have to explicitly depend on `tokio`.
tangle-network · Nov 27, 2024 · f1aec01 · f1aec01
1 parent f35d0c1
commit f1aec01
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/macros/blueprint-proc-macro/src/sdk_main.rs b/macros/blueprint-proc-macro/src/sdk_main.rs
@@ -26,9 +26,9 @@ pub(crate) struct SdkMainArgs {
 
 pub(crate) fn sdk_main_impl(args: &SdkMainArgs, input: &ItemFn) -> syn::Result<TokenStream> {
     let tokio_args = if let Some(args) = &args.tokio_args {
-        quote! { ( #(#args),* ) }
+        quote! { ( crate = "::gadget_sdk::tokio", #(#args),* ) }
     } else {
-        quote! {}
+        quote! { ( crate = "::gadget_sdk::tokio" ) }
     };
 
     let env_function_signature = if args.env {