feat(sozo-walnut): remove sensitive data from dojo configuration file

crates/sozo/walnut/Cargo.toml

@@ -7,19 +7,20 @@ version.workspace = true
 
 [dependencies]
 anyhow.workspace = true
+clap.workspace = true
 console.workspace = true
+dojo-utils.workspace = true
 reqwest.workspace = true
 scarb.workspace = true
 scarb-ui.workspace = true
 serde.workspace = true
 serde_json.workspace = true
 starknet.workspace = true
 thiserror.workspace = true
+toml.workspace = true
 url.workspace = true
 urlencoding = "2.1.3"
 walkdir.workspace = true
-dojo-utils.workspace = true
-clap.workspace = true
 
 [dev-dependencies]
 starknet.workspace = true

crates/sozo/walnut/src/lib.rs

@@ -45,6 +45,9 @@ pub enum Error {
     #[error("Invalid file name")]
     InvalidFileName,
 
+    #[error("Failed to serialize toml: {0}")]
+    TomlSerializationError(#[from] toml::ser::Error),
+
     #[error("Namespace prefix not found in file name")]
     NamespacePrefixNotFound,
 

crates/sozo/walnut/src/verification.rs

@@ -86,11 +86,41 @@
                         // Safe to unwrap here because we're iterating over files within root_dir,
                         // so path will always have root_dir as a prefix
                         let relative_path = path.strip_prefix(root_dir).unwrap();
-                        let file_content = std::fs::read_to_string(path)?;
-                        file_data.insert(
-                            relative_path.to_string_lossy().into_owned(),
-                            serde_json::Value::String(file_content),
-                        );
+                        let mut file_content = std::fs::read_to_string(path)?;
+
+                        // Check if the file is a TOML file and its name starts with "dojo_"
+                        if extension == "toml"
+                            && path
+                                .file_stem()
+                                .and_then(OsStr::to_str)
+                                .map_or(false, |name| name.starts_with("dojo_"))
+                        {
+                            if let Ok(mut toml_data) = file_content.parse::<toml::Value>() {
+                                if let Some(table) = toml_data.as_table_mut() {
+                                    // Remove the "env" table if it exists
+                                    table.remove("env");
+
+                                    // Serialize the modified TOML data back into a string, and
+                                    // handle any serialization error
+                                    file_content = toml::to_string(&toml_data)
+                                        .map_err(Error::TomlSerializationError)?;
+
+                                    // Insert the updated content into file_data, using the relative
+                                    // path as the key
+                                    file_data.insert(
+                                        relative_path.to_string_lossy().into_owned(),
+                                        Value::String(file_content),
+                                    );
+                                }
+                            }
+                        } else {
+                            // If the file is not a "dojo_" prefixed TOML file, just insert the
+                            // original content
+                            file_data.insert(
+                                relative_path.to_string_lossy().into_owned(),
+                                Value::String(file_content),
+                            );
+                        }
                     }
                 }
             }