Expose cli_utils module through library for CLI access

Author: 0x4543

src/bin/cli.rs

@@ -2,6 +2,7 @@ use anyhow::Result;
 use clap::{Parser, Subcommand};
 use std::env;
 use std::process::Command;
+use zeta_crypto::cli_utils;
 use zeta_crypto::{MnemonicHelper, Signer, Wallet, WalletConnectSession, ZetaConfig};
 
 #[derive(Parser)]
@@ -171,9 +172,9 @@ fn main() -> Result<()> {
             if input.trim().to_lowercase() == "yes" {
                 let _ = std::fs::remove_file(dir.join("logs.txt"));
                 let _ = std::fs::remove_file(dir.join("session.json"));
-                println!("Cleanup completed.");
+                cli_utils::success("Cleanup completed.");
             } else {
-                println!("Aborted.");
+                cli_utils::fail("Aborted.");
             }
         }
     }

src/cli_utils.rs

@@ -0,0 +1,11 @@
+use std::process;
+
+pub fn success(msg: &str) {
+    println!("\x1b[32m{}\x1b[0m", msg);
+    process::exit(0);
+}
+
+pub fn fail(msg: &str) {
+    eprintln!("\x1b[31mError:\x1b[0m {}", msg);
+    process::exit(1);
+}

src/lib.rs

@@ -4,6 +4,7 @@ pub mod mnemonic;
 pub mod signer;
 pub mod wallet;
 pub mod walletconnect;
+pub mod cli_utils;
 
 pub use config::ZetaConfig;
 pub use key_derivation::{derive_key_hkdf, derive_key_pbkdf2};