windows cli color fix (MystenLabs#2987)

cat-kitty-it · Jul 5, 2022 · 224c374 · 224c374
1 parent 344cb5c
commit 224c374
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/crates/sui/src/console.rs b/crates/sui/src/console.rs
@@ -52,7 +52,7 @@ pub async fn start_console(
     writeln!(out)?;
 
     let mut shell = Shell::new(
-        "sui>-$ ".bold().green(),
+        "sui>-$ ",
         context,
         ClientCommandHandler,
         CommandStructure::from_clap(&install_shell_plugins(app)),

diff --git a/crates/sui/src/main.rs b/crates/sui/src/main.rs
@@ -13,6 +13,9 @@ mod cli_tests;
 
 #[tokio::main]
 async fn main() {
+    #[cfg(windows)]
+    colored::control::set_virtual_terminal(true).unwrap();
+
     let bin_name = env!("CARGO_BIN_NAME");
     let cmd: SuiCommand = SuiCommand::parse();
     let _guard = match cmd {

diff --git a/crates/sui/src/shell.rs b/crates/sui/src/shell.rs
@@ -73,7 +73,7 @@ impl<P: Display, S: Send, H: AsyncHandler<S>> Shell<P, S, H> {
         }));
 
         loop {
-            write!(out, "{}", self.prompt)?;
+            write!(out, "{}", format!("{}", self.prompt).bold().green())?;
             out.flush()?;
 
             // Read a line