feat(apollo_infra_utils): add function to format Cairo0 source files

Author: dorimedini-starkware

Cargo.lock

@@ -7,7 +7,7 @@ license-file.workspace = true
 description = "Infrastructure utility."
 
 [features]
-testing = ["cached", "colored", "dep:assert-json-diff", "socket2", "toml"]
+testing = ["cached", "colored", "dep:assert-json-diff", "socket2", "tempfile", "toml"]
 
 [lints]
 workspace = true
@@ -20,6 +20,7 @@ num_enum.workspace = true
 serde = { workspace = true, features = ["derive"] }
 serde_json.workspace = true
 socket2 = { workspace = true, optional = true }
+tempfile = { workspace = true, optional = true }
 thiserror.workspace = true
 tokio = { workspace = true, features = ["process", "rt", "time"] }
 toml = { workspace = true, optional = true }
@@ -33,6 +34,7 @@ nix.workspace = true
 pretty_assertions.workspace = true
 rstest.workspace = true
 socket2.workspace = true
+tempfile.workspace = true
 tokio = { workspace = true, features = ["macros", "rt", "signal", "sync"] }
 toml.workspace = true
 tracing-subscriber = { workspace = true, features = ["env-filter"] }

crates/apollo_infra_utils/Cargo.toml

@@ -1,4 +1,6 @@
 #[cfg(any(test, feature = "testing"))]
+use std::io::Write;
+#[cfg(any(test, feature = "testing"))]
 use std::path::PathBuf;
 use std::process::Command;
 #[cfg(any(test, feature = "testing"))]
@@ -80,3 +82,23 @@ pub fn verify_cairo0_compiler_deps() {
         PIP_REQUIREMENTS_FILE.to_str().expect("Path to requirements.txt is valid unicode.")
     );
 }
+
+/// Runs the Cairo0 formatter on the input source code.
+#[cfg(any(test, feature = "testing"))]
+pub fn cairo0_format(unformatted: &String) -> String {
+    verify_cairo0_compiler_deps();
+
+    // Dump string to temporary file.
+    let mut temp_file = tempfile::NamedTempFile::new().unwrap();
+    temp_file.write_all(unformatted.as_bytes()).unwrap();
+
+    // Run formatter.
+    let mut command = Command::new("cairo-format");
+    command.arg(temp_file.path().to_str().unwrap());
+    let format_output = command.output().unwrap();
+    let stderr_output = String::from_utf8(format_output.stderr).unwrap();
+    assert!(format_output.status.success(), "{stderr_output}");
+
+    // Return formatted file.
+    String::from_utf8_lossy(format_output.stdout.as_slice()).to_string()
+}