Use foundry UI in sncast

crates/cheatnet/Cargo.toml

@@ -42,6 +42,7 @@ k256.workspace = true
 p256.workspace = true
 shared.workspace = true
 rand.workspace = true
+foundry-ui = { path = "../foundry-ui" }
 
 [dev-dependencies]
 ctor.workspace = true

crates/cheatnet/tests/common/mod.rs

@@ -27,6 +27,7 @@ use cheatnet::runtime_extensions::forge_runtime_extension::contracts_data::Contr
 use cheatnet::state::CheatnetState;
 use conversions::IntoConv;
 use conversions::string::TryFromHexStr;
+use foundry_ui::UI;
 use runtime::starknet::constants::TEST_ADDRESS;
 use runtime::starknet::context::build_context;
 use scarb_api::metadata::MetadataCommandExt;
@@ -93,8 +94,9 @@ pub fn get_contracts() -> ContractsData {
 
     let package = scarb_metadata.packages.first().unwrap();
 
+    let ui = UI::default();
     let contracts =
-        get_contracts_artifacts_and_source_sierra_paths(&target_dir, package, false).unwrap();
+        get_contracts_artifacts_and_source_sierra_paths(&target_dir, package, false, &ui).unwrap();
     ContractsData::try_from(contracts).unwrap()
 }
 

crates/forge/Cargo.toml

@@ -51,6 +51,7 @@ futures.workspace = true
 url.workspace = true
 indoc.workspace = true
 derive_more.workspace = true
+foundry-ui = { path = "../foundry-ui" }
 
 [[bin]]
 name = "snforge"

crates/forge/src/lib.rs

@@ -7,6 +7,7 @@ use derive_more::Display;
 use forge_runner::CACHE_DIR;
 use forge_runner::debugging::TraceVerbosity;
 use forge_runner::forge_config::ForgeTrackedResource;
+use foundry_ui::UI;
 use run_tests::workspace::run_for_workspace;
 use scarb_api::{ScarbCommand, metadata::MetadataCommandExt};
 use scarb_ui::args::{FeaturesSpec, PackagesFilter};
@@ -255,6 +256,7 @@ pub enum ExitStatus {
 }
 
 pub fn main_execution() -> Result<ExitStatus> {
+    let ui = UI::default();
     let cli = Cli::parse();
 
     match cli.subcommand {
@@ -297,7 +299,7 @@ pub fn main_execution() -> Result<ExitStatus> {
                 .enable_all()
                 .build()?;
 
-            rt.block_on(run_for_workspace(args))
+            rt.block_on(run_for_workspace(args, &ui))
         }
         ForgeSubcommand::CheckRequirements => {
             check_requirements(true, ForgeTrackedResource::default())?;

crates/forge/src/run_tests/package.rs

@@ -30,6 +30,7 @@ use forge_runner::{
     test_case_summary::AnyTestCaseSummary,
     test_target_summary::TestTargetSummary,
 };
+use foundry_ui::UI;
 use scarb_api::get_contracts_artifacts_and_source_sierra_paths;
 use scarb_metadata::{Metadata, PackageMetadata};
 use std::sync::Arc;
@@ -49,6 +50,7 @@ impl RunForPackageArgs {
         args: &TestArgs,
         cache_dir: &Utf8PathBuf,
         artifacts_dir: &Utf8Path,
+        ui: &UI,
     ) -> Result<RunForPackageArgs> {
         let raw_test_targets = load_test_artifacts(artifacts_dir, &package)?;
 
@@ -59,6 +61,7 @@ impl RunForPackageArgs {
                 &scarb_metadata.app_version_info.version,
                 args.no_optimization,
             ),
+            ui,
         )?;
         let contracts_data = ContractsData::try_from(contracts)?;
 

crates/forge/src/run_tests/workspace.rs

@@ -11,6 +11,7 @@ use forge_runner::{
     coverage_api::can_coverage_be_generated,
     test_case_summary::{AnyTestCaseSummary, TestCaseSummary},
 };
+use foundry_ui::UI;
 use scarb_api::{
     ScarbCommand,
     metadata::{Metadata, MetadataCommandExt, PackageMetadata},
@@ -20,7 +21,7 @@ use scarb_ui::args::PackagesFilter;
 use shared::consts::SNFORGE_TEST_FILTER;
 use std::env;
 
-pub async fn run_for_workspace(args: TestArgs) -> Result<ExitStatus> {
+pub async fn run_for_workspace(args: TestArgs, ui: &UI) -> Result<ExitStatus> {
     match args.color {
         // SAFETY: This runs in a single-threaded environment.
         ColorOption::Always => unsafe { env::set_var("CLICOLOR_FORCE", "1") },
@@ -81,6 +82,7 @@ pub async fn run_for_workspace(args: TestArgs) -> Result<ExitStatus> {
             &args,
             &cache_dir,
             &artifacts_dir_path,
+            ui,
         )?;
 
         let tests_file_summaries =

crates/forge/test_utils/Cargo.toml

@@ -25,3 +25,4 @@ forge = { path = "../../forge" }
 scarb-api = { path = "../../scarb-api" }
 forge_runner = { path = "../../forge-runner" }
 cheatnet = { path = "../../cheatnet" }
+foundry-ui = { path = "../../foundry-ui" }

crates/forge/test_utils/src/runner.rs

@@ -12,6 +12,7 @@ use forge_runner::{
     test_case_summary::{AnyTestCaseSummary, TestCaseSummary},
     test_target_summary::TestTargetSummary,
 };
+use foundry_ui::UI;
 use indoc::formatdoc;
 use scarb_api::{
     ScarbCommand, StarknetContractArtifacts, get_contracts_artifacts_and_source_sierra_paths,
@@ -57,7 +58,7 @@ impl Contract {
         })
     }
 
-    fn generate_sierra_and_casm(self) -> Result<(String, String)> {
+    fn generate_sierra_and_casm(self, ui: &UI) -> Result<(String, String)> {
         let dir = tempdir_with_tool_versions()?;
 
         let contract_path = dir.child("src/lib.cairo");
@@ -101,7 +102,7 @@ impl Contract {
         let artifacts_dir = target_dir_for_workspace(&scarb_metadata).join("dev");
 
         let contract =
-            get_contracts_artifacts_and_source_sierra_paths(&artifacts_dir, package, false)
+            get_contracts_artifacts_and_source_sierra_paths(&artifacts_dir, package, false, ui)
                 .unwrap()
                 .remove(&self.name)
                 .ok_or(anyhow!("there is no contract with name {}", self.name))?
@@ -194,13 +195,16 @@ impl<'a> TestCase {
         ]
     }
 
-    pub fn contracts(&self) -> Result<HashMap<String, (StarknetContractArtifacts, Utf8PathBuf)>> {
+    pub fn contracts(
+        &self,
+        ui: &UI,
+    ) -> Result<HashMap<String, (StarknetContractArtifacts, Utf8PathBuf)>> {
         self.contracts
             .clone()
             .into_iter()
             .map(|contract| {
                 let name = contract.name.clone();
-                let (sierra, casm) = contract.generate_sierra_and_casm()?;
+                let (sierra, casm) = contract.generate_sierra_and_casm(ui)?;
 
                 Ok((
                     name,

crates/forge/test_utils/src/running_tests.rs

@@ -13,6 +13,7 @@ use forge_runner::forge_config::{
     ExecutionDataToSave, ForgeConfig, ForgeTrackedResource, OutputConfig, TestRunnerConfig,
 };
 use forge_runner::test_target_summary::TestTargetSummary;
+use foundry_ui::UI;
 use scarb_api::{ScarbCommand, metadata::MetadataCommandExt};
 use std::num::NonZeroU32;
 use std::sync::Arc;
@@ -46,6 +47,7 @@ pub fn run_test_case(
     let raw_test_targets =
         load_test_artifacts(&test.path().unwrap().join("target/dev"), package).unwrap();
 
+    let ui = UI::default();
     rt.block_on(run_for_package(
         RunForPackageArgs {
             test_targets: raw_test_targets,
@@ -68,7 +70,7 @@ pub fn run_test_case(
                     cache_dir: Utf8PathBuf::from_path_buf(tempdir().unwrap().keep())
                         .unwrap()
                         .join(CACHE_DIR),
-                    contracts_data: ContractsData::try_from(test.contracts().unwrap()).unwrap(),
+                    contracts_data: ContractsData::try_from(test.contracts(&ui).unwrap()).unwrap(),
                     tracked_resource,
                     environment_variables: test.env().clone(),
                 }),

crates/forge/tests/integration/setup_fork.rs

@@ -1,4 +1,5 @@
 use cheatnet::runtime_extensions::forge_config_extension::config::BlockId;
+use foundry_ui::UI;
 use indoc::{formatdoc, indoc};
 use std::num::NonZeroU32;
 use std::path::Path;
@@ -129,6 +130,7 @@ fn fork_aliased_decorator() {
     let raw_test_targets =
         load_test_artifacts(&test.path().unwrap().join("target/dev"), package).unwrap();
 
+    let ui = UI::default();
     let result = rt
         .block_on(run_for_package(
             RunForPackageArgs {
@@ -152,7 +154,8 @@ fn fork_aliased_decorator() {
                         cache_dir: Utf8PathBuf::from_path_buf(tempdir().unwrap().keep())
                             .unwrap()
                             .join(CACHE_DIR),
-                        contracts_data: ContractsData::try_from(test.contracts().unwrap()).unwrap(),
+                        contracts_data: ContractsData::try_from(test.contracts(&ui).unwrap())
+                            .unwrap(),
                         tracked_resource: ForgeTrackedResource::CairoSteps,
                         environment_variables: test.env().clone(),
                     }),
@@ -216,6 +219,7 @@ fn fork_aliased_decorator_overrding() {
     let raw_test_targets =
         load_test_artifacts(&test.path().unwrap().join("target/dev"), package).unwrap();
 
+    let ui = UI::default();
     let result = rt
         .block_on(run_for_package(
             RunForPackageArgs {
@@ -239,7 +243,8 @@ fn fork_aliased_decorator_overrding() {
                         cache_dir: Utf8PathBuf::from_path_buf(tempdir().unwrap().keep())
                             .unwrap()
                             .join(CACHE_DIR),
-                        contracts_data: ContractsData::try_from(test.contracts().unwrap()).unwrap(),
+                        contracts_data: ContractsData::try_from(test.contracts(&ui).unwrap())
+                            .unwrap(),
                         tracked_resource: ForgeTrackedResource::CairoSteps,
                         environment_variables: test.env().clone(),
                     }),

crates/foundry-ui/src/lib.rs

@@ -27,13 +27,6 @@ impl UI {
         Self { output_format }
     }
 
-    // TODO: Will be removed in 3022-3-use-foundry-ui-in-sncast
-    /// Get the output format of this [`UI`] instance.
-    #[must_use]
-    pub fn output_format(&self) -> OutputFormat {
-        self.output_format
-    }
-
     /// Create a [`String`] representation of the given message based on the configured output format.
     fn format_message(&self, message: &impl Message) -> String {
         match self.output_format {

crates/scarb-api/Cargo.toml

@@ -18,6 +18,7 @@ regex.workspace = true
 rayon.workspace = true
 itertools.workspace = true
 universal-sierra-compiler-api = { path = "../universal-sierra-compiler-api" }
+foundry-ui = { path = "../foundry-ui" }
 
 [dev-dependencies]
 assert_fs.workspace = true