Add walnut flag to sozo execute and migrate apply commands

Cargo.toml

@@ -39,6 +39,7 @@ members = [
 	"crates/saya/core",
 	"crates/saya/provider",
 	"crates/sozo/signers",
+	"crates/sozo/walnut",
 	"crates/torii/client",
 	"crates/torii/server",
 	"crates/torii/types-test",
@@ -111,6 +112,7 @@ saya-provider = { path = "crates/saya/provider" }
 # sozo
 sozo-ops = { path = "crates/sozo/ops" }
 sozo-signers = { path = "crates/sozo/signers" }
+sozo-walnut = { path = "crates/sozo/walnut" }
 
 anyhow = "1.0.80"
 assert_fs = "1.1"
@@ -207,6 +209,7 @@ tower-http = "0.4.4"
 tracing = "0.1.34"
 tracing-subscriber = { version = "0.3.16", features = [ "env-filter", "json" ] }
 url = { version = "2.4.0", features = [ "serde" ] }
+walkdir = "2.5.0"
 
 # server
 hyper = "0.14.27"

bin/sozo/Cargo.toml

@@ -51,6 +51,7 @@ serde.workspace = true
 serde_json.workspace = true
 smol_str.workspace = true
 sozo-ops.workspace = true
+sozo-walnut.workspace = true
 starknet.workspace = true
 starknet-crypto.workspace = true
 thiserror.workspace = true

bin/sozo/src/commands/auth.rs

@@ -5,6 +5,7 @@
 use scarb::core::Config;
 use scarb_ui::Ui;
 use sozo_ops::auth;
+use sozo_walnut::WalnutDebugger;
 use tracing::trace;
 
 use super::options::account::AccountOptions;
@@ -149,24 +150,41 @@
 ) -> Result<()> {
     trace!(?kind, ?world, ?starknet, ?account, ?transaction, "Executing Grant command.");
     let world =
-        utils::world_from_env_metadata(world, account, starknet, &env_metadata, config).await?;
+        utils::world_from_env_metadata(world, account, &starknet, &env_metadata, config).await?;
+
+    let walnut_debugger =
+        WalnutDebugger::new_from_flag(transaction.walnut, starknet.url(env_metadata.as_ref())?);
 
     match kind {
         AuthKind::Writer { models_contracts } => {
             trace!(
                 contracts=?models_contracts,
                 "Granting Writer permissions."
             );
-            auth::grant_writer(ui, &world, &models_contracts, transaction.into(), default_namespace)
-                .await
+            auth::grant_writer(
+                ui,
+                &world,
+                &models_contracts,
+                &transaction.into(),
+                default_namespace,
+                &walnut_debugger,
+            )
+            .await
         }
         AuthKind::Owner { owners_resources } => {
             trace!(
                 resources=?owners_resources,
                 "Granting Owner permissions."
             );
-            auth::grant_owner(ui, &world, &owners_resources, transaction.into(), default_namespace)
-                .await
+            auth::grant_owner(
+                ui,
+                &world,
+                &owners_resources,
+                &transaction.into(),
+                default_namespace,
+                &walnut_debugger,
+            )
+            .await
         }
     }
 }
@@ -185,7 +203,10 @@
 ) -> Result<()> {
     trace!(?kind, ?world, ?starknet, ?account, ?transaction, "Executing Revoke command.");
     let world =
-        utils::world_from_env_metadata(world, account, starknet, &env_metadata, config).await?;
+        utils::world_from_env_metadata(world, account, &starknet, &env_metadata, config).await?;
+
+    let walnut_debugger =
+        WalnutDebugger::new_from_flag(transaction.walnut, starknet.url(env_metadata.as_ref())?);
 
     match kind {
         AuthKind::Writer { models_contracts } => {
@@ -197,8 +218,9 @@
                 ui,
                 &world,
                 &models_contracts,
-                transaction.into(),
+                &transaction.into(),
                 default_namespace,
+                &walnut_debugger,
             )
             .await
         }
@@ -207,8 +229,15 @@
                 resources=?owners_resources,
                 "Revoking Owner permissions."
             );
-            auth::revoke_owner(ui, &world, &owners_resources, transaction.into(), default_namespace)
-                .await
+            auth::revoke_owner(
+                ui,
+                &world,
+                &owners_resources,
+                &transaction.into(),
+                default_namespace,
+                &walnut_debugger,
+            )
+            .await
         }
     }
 }

bin/sozo/src/commands/execute.rs

@@ -4,6 +4,7 @@
 use dojo_world::metadata::get_default_namespace_from_ws;
 use scarb::core::Config;
 use sozo_ops::execute;
+use sozo_walnut::WalnutDebugger;
 use tracing::trace;
 
 use super::calldata_decoder;
@@ -61,11 +62,16 @@
             ensure_namespace(&self.tag_or_address, &default_namespace)
         };
 
+        let walnut_debugger = WalnutDebugger::new_from_flag(
+            self.transaction.walnut,
+            self.starknet.url(env_metadata.as_ref())?,
+        );
+
         config.tokio_handle().block_on(async {
             let world = utils::world_from_env_metadata(
                 self.world,
                 self.account,
-                self.starknet,
+                &self.starknet,
                 &env_metadata,
                 config,
             )
@@ -93,6 +99,7 @@
                 calldata,
                 &world,
                 &tx_config,
+                &walnut_debugger,
             )
             .await
         })

bin/sozo/src/commands/options/transaction.rs

@@ -37,6 +37,11 @@
     )]
     #[arg(global = true)]
     pub receipt: bool,
+
+    #[arg(long)]
+    #[arg(help = "Display the link to debug the transaction with Walnut.")]
+    #[arg(global = true)]
+    pub walnut: bool,
 }
 
 impl TransactionOptions {
@@ -52,10 +57,11 @@
             (true, false) => Ok(TxnAction::Estimate),
             (false, true) => Ok(TxnAction::Simulate),
             (false, false) => Ok(TxnAction::Send {
-                wait: self.wait,
+                wait: self.wait || self.walnut,
                 receipt: self.receipt,
                 max_fee_raw: self.max_fee_raw,
                 fee_estimate_multiplier: self.fee_estimate_multiplier,
+                walnut: self.walnut,
             }),
         }
     }
@@ -71,9 +77,10 @@
         );
         Self {
             fee_estimate_multiplier: value.fee_estimate_multiplier,
-            wait: value.wait,
+            wait: value.wait || value.walnut,
             receipt: value.receipt,
             max_fee_raw: value.max_fee_raw,
+            walnut: value.walnut,
         }
     }
 }

bin/sozo/src/commands/register.rs

@@ -3,6 +3,7 @@ use clap::{Args, Subcommand};
 use dojo_world::contracts::WorldContractReader;
 use scarb::core::Config;
 use sozo_ops::register;
+use sozo_walnut::WalnutDebugger;
 use starknet::accounts::ConnectedAccount;
 use starknet::core::types::{BlockId, BlockTag, Felt};
 use tracing::trace;
@@ -58,9 +59,12 @@ impl RegisterArgs {
         let world_address = world.world_address.unwrap_or_default();
         trace!(?world_address, "Using world address.");
 
+        let walnut_debugger =
+            WalnutDebugger::new_from_flag(transaction.walnut, starknet.url(env_metadata.as_ref())?);
+
         config.tokio_handle().block_on(async {
             let world =
-                utils::world_from_env_metadata(world, account, starknet, &env_metadata, config)
+                utils::world_from_env_metadata(world, account, &starknet, &env_metadata, config)
                     .await?;
             let provider = world.account.provider();
             let mut world_reader = WorldContractReader::new(world_address, &provider);
@@ -69,10 +73,11 @@ impl RegisterArgs {
             register::model_register(
                 models,
                 &world,
-                transaction.into(),
+                &transaction.into(),
                 world_reader,
                 world_address,
                 config,
+                &walnut_debugger,
             )
             .await
         })

bin/sozo/src/utils.rs

@@ -52,7 +52,7 @@ pub fn load_metadata_from_config(config: &Config) -> Result<Option<Environment>,
 pub async fn world_from_env_metadata(
     world: WorldOptions,
     account: AccountOptions,
-    starknet: StarknetOptions,
+    starknet: &StarknetOptions,
     env_metadata: &Option<Environment>,
     config: &Config,
 ) -> Result<WorldContract<SozoAccount<JsonRpcClient<HttpTransport>>>, Error> {
@@ -64,7 +64,7 @@ pub async fn world_from_env_metadata(
         .account(
             provider,
             WorldAddressOrName::Address(world_address),
-            &starknet,
+            starknet,
             env_metadata,
             config,
         )

crates/dojo-lang/src/compiler.rs

@@ -162,7 +162,6 @@ impl Compiler for DojoCompiler {
 
         for (decl, contract_class, debug_info) in izip!(contracts, classes, debug_info_classes) {
             let contract_name = decl.submodule_id.name(db.upcast_mut());
-
             // note that the qualified path is in snake case while
             // the `full_path()` method of StructId uses the original struct name case.
             // (see in `get_dojo_model_artifacts`)

crates/dojo-utils/src/tx/mod.rs

@@ -18,6 +18,7 @@ pub struct TxnConfig {
     pub wait: bool,
     pub receipt: bool,
     pub max_fee_raw: Option<Felt>,
+    pub walnut: bool,
 }
 
 #[derive(Debug, Copy, Clone)]
@@ -29,6 +30,7 @@ pub enum TxnAction {
         /// The multiplier for how much the actual transaction max fee should be relative to the
         /// estimated fee. If `None` is provided, the multiplier is set to `1.1`.
         fee_estimate_multiplier: Option<f64>,
+        walnut: bool,
     },
     Estimate,
     Simulate,

crates/dojo-world/Cargo.toml

@@ -36,7 +36,7 @@ scarb = { workspace = true, optional = true }
 tokio = { version = "1.32.0", features = [ "time" ], default-features = false, optional = true }
 toml.workspace = true
 url = { workspace = true, optional = true }
-walkdir = "2.5.0"
+walkdir.workspace = true
 
 [dev-dependencies]
 assert_fs.workspace = true
@@ -52,4 +52,4 @@ tokio.workspace = true
 contracts = [ "dep:dojo-types", "dep:http", "dep:num-traits" ]
 manifest = [ "contracts", "dep:dojo-types", "dep:scarb", "dep:url" ]
 metadata = [ "dep:ipfs-api-backend-hyper", "dep:scarb", "dep:url" ]
-migration = [ "dep:dojo-utils", "dep:scarb", "dep:tokio" ]
+migration = [ "dep:dojo-utils", "dep:scarb", "dep:tokio", "manifest" ]

crates/sozo/ops/Cargo.toml

@@ -50,6 +50,7 @@ tokio.workspace = true
 toml.workspace = true
 tracing.workspace = true
 url.workspace = true
+sozo-walnut.workspace = true
 
 dojo-test-utils = { workspace = true, features = [ "build-examples" ], optional = true }
 katana-runner = { workspace = true, optional = true }

crates/sozo/ops/src/account.rs

@@ -248,7 +248,7 @@
     };
 
     match txn_action {
-        TxnAction::Send { wait, receipt, max_fee_raw, fee_estimate_multiplier } => {
+        TxnAction::Send { wait, receipt, max_fee_raw, fee_estimate_multiplier, walnut } => {
             let max_fee = if let Some(max_fee_raw) = max_fee_raw {
                 MaxFeeType::Manual { max_fee: max_fee_raw }
             } else {
@@ -277,7 +277,8 @@
             };
 
             let account_deployment = account_deployment.max_fee(max_fee.max_fee());
-            let txn_config = TxnConfig { fee_estimate_multiplier, wait, receipt, max_fee_raw };
+            let txn_config =
+                TxnConfig { fee_estimate_multiplier, wait, receipt, max_fee_raw, walnut };
             do_account_deploy(
                 max_fee,
                 txn_config,