add colors on warnings with colored package

dojoengine · Mar 10, 2024 · c706133 · c706133
1 parent 56f7336
commit c706133
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 3 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/bin/sozo/Cargo.toml b/bin/sozo/Cargo.toml
@@ -23,6 +23,7 @@ camino.workspace = true
 clap-verbosity-flag = "2.0.1"
 clap.workspace = true
 clap_complete.workspace = true
+colored = "2.1.0"
 console.workspace = true
 dojo-bindgen.workspace = true
 dojo-lang.workspace = true

diff --git a/bin/sozo/src/ops/auth.rs b/bin/sozo/src/ops/auth.rs
@@ -1,4 +1,5 @@
 use anyhow::{Context, Result};
+use colored::Colorize;
 use dojo_world::contracts::model::ModelReader;
 use dojo_world::contracts::world::WorldContract;
 use dojo_world::contracts::WorldContractReader;
@@ -31,14 +32,21 @@ pub async fn execute(command: AuthCommand, env_metadata: Option<Environment>) ->
                     let model_address = model.contract_address();
 
                     if model_address == FieldElement::ZERO {
-                        println!("[WARN] Unknown model '{}' => IGNORED", model_name);
+                        println!(
+                            "{} Unknown model {} => {}",
+                            "[WARN]".yellow(),
+                            model_name.blue(),
+                            "IGNORED".yellow()
+                        );
                     } else {
                         let contract = get_contract_address(&world, mc.contract).await?;
 
                         if model_address != contract {
                             println!(
-                                "[WARN] Wrong address for model '{}' 0x{:x} => IGNORED",
-                                model_name, contract
+                                "{} Wrong address for model {} => {}",
+                                "[WARN]".yellow(),
+                                model_name.blue(),
+                                "IGNORED".yellow()
                             );
                         } else {
                             calls.push(world.grant_writer_getcall(&mc.model, &contract.into()));